| ... | ... | @@ -19,17 +19,21 @@ pub fn render(allocator: &mem.Allocator, stream: var, tree: &ast.Tree) (@typeOf( |
| 19 | 19 | while (it.next()) |decl| { |
| 20 | 20 | try renderTopLevelDecl(allocator, stream, tree, 0, decl.*); |
| 21 | 21 | if (it.peek()) |next_decl| { |
| 22 | | const n = if (nodeLineOffset(tree, decl.*, next_decl.*) >= 2) u8(2) else u8(1); |
| 23 | | try stream.writeByteNTimes('\n', n); |
| 22 | try renderExtraNewline(tree, stream, next_decl.*); |
| 24 | 23 | } |
| 25 | 24 | } |
| 26 | | try stream.write("\n"); |
| 27 | 25 | } |
| 28 | 26 | |
| 29 | | fn nodeLineOffset(tree: &ast.Tree, a: &ast.Node, b: &ast.Node) usize { |
| 30 | | const a_last_token = tree.tokens.at(a.lastToken()); |
| 31 | | const loc = tree.tokenLocation(a_last_token.end, b.firstToken()); |
| 32 | | return loc.line; |
| 27 | fn renderExtraNewline(tree: &ast.Tree, stream: var, node: &ast.Node) !void { |
| 28 | var first_token = node.firstToken(); |
| 29 | while (tree.tokens.at(first_token - 1).id == Token.Id.DocComment) { |
| 30 | first_token -= 1; |
| 31 | } |
| 32 | const prev_token_end = tree.tokens.at(first_token - 1).end; |
| 33 | const loc = tree.tokenLocation(prev_token_end, first_token); |
| 34 | if (loc.line >= 2) { |
| 35 | try stream.writeByte('\n'); |
| 36 | } |
| 33 | 37 | } |
| 34 | 38 | |
| 35 | 39 | fn renderTopLevelDecl(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, indent: usize, decl: &ast.Node) (@typeOf(stream).Child.Error || Error)!void { |
| ... | ... | @@ -37,119 +41,124 @@ fn renderTopLevelDecl(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, i |
| 37 | 41 | ast.Node.Id.FnProto => { |
| 38 | 42 | const fn_proto = @fieldParentPtr(ast.Node.FnProto, "base", decl); |
| 39 | 43 | |
| 40 | | try renderComments(tree, stream, fn_proto, indent); |
| 41 | | try renderExpression(allocator, stream, tree, indent, decl); |
| 44 | try renderDocComments(tree, stream, fn_proto, indent); |
| 42 | 45 | |
| 43 | 46 | if (fn_proto.body_node) |body_node| { |
| 44 | | try stream.write(" "); |
| 45 | | try renderExpression(allocator, stream, tree, indent, body_node); |
| 47 | try renderExpression(allocator, stream, tree, indent, decl, Space.Space); |
| 48 | try renderExpression(allocator, stream, tree, indent, body_node, Space.Newline); |
| 46 | 49 | } else { |
| 47 | | try stream.write(";"); |
| 50 | try renderExpression(allocator, stream, tree, indent, decl, Space.None); |
| 51 | try renderToken(tree, stream, tree.nextToken(decl.lastToken()), indent, Space.Newline); |
| 48 | 52 | } |
| 49 | 53 | }, |
| 54 | |
| 50 | 55 | ast.Node.Id.Use => { |
| 51 | 56 | const use_decl = @fieldParentPtr(ast.Node.Use, "base", decl); |
| 52 | 57 | |
| 53 | 58 | if (use_decl.visib_token) |visib_token| { |
| 54 | | try stream.print("{} ", tree.tokenSlice(visib_token)); |
| 59 | try renderToken(tree, stream, visib_token, indent, Space.Space); // pub |
| 55 | 60 | } |
| 56 | | try stream.write("use "); |
| 57 | | try renderExpression(allocator, stream, tree, indent, use_decl.expr); |
| 58 | | try stream.write(";"); |
| 61 | try renderToken(tree, stream, use_decl.use_token, indent, Space.Space); // use |
| 62 | try renderExpression(allocator, stream, tree, indent, use_decl.expr, Space.None); |
| 63 | try renderToken(tree, stream, use_decl.semicolon_token, indent, Space.Newline); // ; |
| 59 | 64 | }, |
| 65 | |
| 60 | 66 | ast.Node.Id.VarDecl => { |
| 61 | 67 | const var_decl = @fieldParentPtr(ast.Node.VarDecl, "base", decl); |
| 62 | 68 | |
| 63 | | try renderComments(tree, stream, var_decl, indent); |
| 69 | try renderDocComments(tree, stream, var_decl, indent); |
| 64 | 70 | try renderVarDecl(allocator, stream, tree, indent, var_decl); |
| 65 | 71 | }, |
| 72 | |
| 66 | 73 | ast.Node.Id.TestDecl => { |
| 67 | 74 | const test_decl = @fieldParentPtr(ast.Node.TestDecl, "base", decl); |
| 68 | 75 | |
| 69 | | try renderComments(tree, stream, test_decl, indent); |
| 70 | | try stream.write("test "); |
| 71 | | try renderExpression(allocator, stream, tree, indent, test_decl.name); |
| 72 | | try stream.write(" "); |
| 73 | | try renderExpression(allocator, stream, tree, indent, test_decl.body_node); |
| 76 | try renderDocComments(tree, stream, test_decl, indent); |
| 77 | try renderToken(tree, stream, test_decl.test_token, indent, Space.Space); |
| 78 | try renderExpression(allocator, stream, tree, indent, test_decl.name, Space.Space); |
| 79 | try renderExpression(allocator, stream, tree, indent, test_decl.body_node, Space.Newline); |
| 74 | 80 | }, |
| 81 | |
| 75 | 82 | ast.Node.Id.StructField => { |
| 76 | 83 | const field = @fieldParentPtr(ast.Node.StructField, "base", decl); |
| 77 | 84 | |
| 78 | | try renderComments(tree, stream, field, indent); |
| 85 | try renderDocComments(tree, stream, field, indent); |
| 79 | 86 | if (field.visib_token) |visib_token| { |
| 80 | | try stream.print("{} ", tree.tokenSlice(visib_token)); |
| 87 | try renderToken(tree, stream, visib_token, indent, Space.Space); // pub |
| 81 | 88 | } |
| 82 | | try stream.print("{}: ", tree.tokenSlice(field.name_token)); |
| 83 | | try renderExpression(allocator, stream, tree, indent, field.type_expr); |
| 84 | | try renderToken(tree, stream, field.lastToken() + 1, indent, true, true); |
| 89 | try renderToken(tree, stream, field.name_token, indent, Space.None); // name |
| 90 | try renderToken(tree, stream, tree.nextToken(field.name_token), indent, Space.Space); // : |
| 91 | try renderExpression(allocator, stream, tree, indent, field.type_expr, Space.None); // type |
| 92 | try renderToken(tree, stream, tree.nextToken(field.lastToken()), indent, Space.Newline); // , |
| 85 | 93 | }, |
| 94 | |
| 86 | 95 | ast.Node.Id.UnionTag => { |
| 87 | 96 | const tag = @fieldParentPtr(ast.Node.UnionTag, "base", decl); |
| 88 | 97 | |
| 89 | | try renderComments(tree, stream, tag, indent); |
| 90 | | try stream.print("{}", tree.tokenSlice(tag.name_token)); |
| 98 | try renderDocComments(tree, stream, tag, indent); |
| 99 | |
| 100 | const name_space = if (tag.type_expr == null and tag.value_expr != null) Space.Space else Space.None; |
| 101 | try renderToken(tree, stream, tag.name_token, indent, name_space); // name |
| 91 | 102 | |
| 92 | 103 | if (tag.type_expr) |type_expr| { |
| 93 | | try stream.print(": "); |
| 94 | | try renderExpression(allocator, stream, tree, indent, type_expr); |
| 104 | try renderToken(tree, stream, tree.nextToken(tag.name_token), indent, Space.Space); // : |
| 105 | |
| 106 | const after_type_space = if (tag.value_expr == null) Space.None else Space.Space; |
| 107 | try renderExpression(allocator, stream, tree, indent, type_expr, after_type_space); |
| 95 | 108 | } |
| 96 | 109 | |
| 97 | 110 | if (tag.value_expr) |value_expr| { |
| 98 | | try stream.print(" = "); |
| 99 | | try renderExpression(allocator, stream, tree, indent, value_expr); |
| 111 | try renderToken(tree, stream, tree.prevToken(value_expr.firstToken()), indent, Space.Space); // = |
| 112 | try renderExpression(allocator, stream, tree, indent, value_expr, Space.None); |
| 100 | 113 | } |
| 101 | 114 | |
| 102 | | try stream.write(","); |
| 115 | try renderToken(tree, stream, tree.nextToken(decl.lastToken()), indent, Space.Newline); // , |
| 103 | 116 | }, |
| 117 | |
| 104 | 118 | ast.Node.Id.EnumTag => { |
| 105 | 119 | const tag = @fieldParentPtr(ast.Node.EnumTag, "base", decl); |
| 106 | 120 | |
| 107 | | try renderComments(tree, stream, tag, indent); |
| 108 | | try stream.print("{}", tree.tokenSlice(tag.name_token)); |
| 121 | try renderDocComments(tree, stream, tag, indent); |
| 122 | |
| 123 | const after_name_space = if (tag.value == null) Space.None else Space.Space; |
| 124 | try renderToken(tree, stream, tag.name_token, indent, after_name_space); // name |
| 109 | 125 | |
| 110 | 126 | if (tag.value) |value| { |
| 111 | | try stream.print(" = "); |
| 112 | | try renderExpression(allocator, stream, tree, indent, value); |
| 127 | try renderToken(tree, stream, tree.nextToken(tag.name_token), indent, Space.Space); // = |
| 128 | try renderExpression(allocator, stream, tree, indent, value, Space.None); |
| 113 | 129 | } |
| 114 | 130 | |
| 115 | | try stream.write(","); |
| 131 | try renderToken(tree, stream, tree.nextToken(decl.lastToken()), indent, Space.Newline); // , |
| 116 | 132 | }, |
| 117 | | ast.Node.Id.ErrorTag => { |
| 118 | | const tag = @fieldParentPtr(ast.Node.ErrorTag, "base", decl); |
| 119 | 133 | |
| 120 | | try renderComments(tree, stream, tag, indent); |
| 121 | | try stream.print("{}", tree.tokenSlice(tag.name_token)); |
| 122 | | }, |
| 123 | 134 | ast.Node.Id.Comptime => { |
| 124 | | try renderExpression(allocator, stream, tree, indent, decl); |
| 125 | | try maybeRenderSemicolon(stream, tree, indent, decl); |
| 126 | | }, |
| 127 | | ast.Node.Id.LineComment => { |
| 128 | | const line_comment_node = @fieldParentPtr(ast.Node.LineComment, "base", decl); |
| 129 | | |
| 130 | | try stream.write(tree.tokenSlice(line_comment_node.token)); |
| 135 | assert(!decl.requireSemiColon()); |
| 136 | try renderExpression(allocator, stream, tree, indent, decl, Space.Newline); |
| 131 | 137 | }, |
| 132 | 138 | else => unreachable, |
| 133 | 139 | } |
| 134 | 140 | } |
| 135 | 141 | |
| 136 | | fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, indent: usize, base: &ast.Node) (@typeOf(stream).Child.Error || Error)!void { |
| 142 | fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, indent: usize, base: &ast.Node, space: Space) (@typeOf(stream).Child.Error || Error)!void { |
| 137 | 143 | switch (base.id) { |
| 138 | 144 | ast.Node.Id.Identifier => { |
| 139 | 145 | const identifier = @fieldParentPtr(ast.Node.Identifier, "base", base); |
| 140 | | try stream.print("{}", tree.tokenSlice(identifier.token)); |
| 146 | try renderToken(tree, stream, identifier.token, indent, space); |
| 141 | 147 | }, |
| 142 | 148 | ast.Node.Id.Block => { |
| 143 | 149 | const block = @fieldParentPtr(ast.Node.Block, "base", base); |
| 150 | |
| 144 | 151 | if (block.label) |label| { |
| 145 | | try stream.print("{}: ", tree.tokenSlice(label)); |
| 152 | try renderToken(tree, stream, label, indent, Space.None); |
| 153 | try renderToken(tree, stream, tree.nextToken(label), indent, Space.Space); |
| 146 | 154 | } |
| 147 | 155 | |
| 148 | 156 | if (block.statements.len == 0) { |
| 149 | | try stream.write("{}"); |
| 157 | try renderToken(tree, stream, block.lbrace, indent + indent_delta, Space.None); |
| 158 | try renderToken(tree, stream, block.rbrace, indent, space); |
| 150 | 159 | } else { |
| 151 | | try stream.write("{\n"); |
| 152 | 160 | const block_indent = indent + indent_delta; |
| 161 | try renderToken(tree, stream, block.lbrace, block_indent, Space.Newline); |
| 153 | 162 | |
| 154 | 163 | var it = block.statements.iterator(0); |
| 155 | 164 | while (it.next()) |statement| { |
| ... | ... | @@ -157,114 +166,85 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 157 | 166 | try renderStatement(allocator, stream, tree, block_indent, statement.*); |
| 158 | 167 | |
| 159 | 168 | if (it.peek()) |next_statement| { |
| 160 | | const n = if (nodeLineOffset(tree, statement.*, next_statement.*) >= 2) u8(2) else u8(1); |
| 161 | | try stream.writeByteNTimes('\n', n); |
| 169 | try renderExtraNewline(tree, stream, next_statement.*); |
| 162 | 170 | } |
| 163 | 171 | } |
| 164 | 172 | |
| 165 | | try stream.write("\n"); |
| 166 | 173 | try stream.writeByteNTimes(' ', indent); |
| 167 | | try stream.write("}"); |
| 174 | try renderToken(tree, stream, block.rbrace, indent, space); |
| 168 | 175 | } |
| 169 | 176 | }, |
| 170 | 177 | ast.Node.Id.Defer => { |
| 171 | 178 | const defer_node = @fieldParentPtr(ast.Node.Defer, "base", base); |
| 172 | | try stream.print("{} ", tree.tokenSlice(defer_node.defer_token)); |
| 173 | | try renderExpression(allocator, stream, tree, indent, defer_node.expr); |
| 179 | |
| 180 | try renderToken(tree, stream, defer_node.defer_token, indent, Space.Space); |
| 181 | try renderExpression(allocator, stream, tree, indent, defer_node.expr, space); |
| 174 | 182 | }, |
| 175 | 183 | ast.Node.Id.Comptime => { |
| 176 | 184 | const comptime_node = @fieldParentPtr(ast.Node.Comptime, "base", base); |
| 177 | | try stream.print("{} ", tree.tokenSlice(comptime_node.comptime_token)); |
| 178 | | try renderExpression(allocator, stream, tree, indent, comptime_node.expr); |
| 185 | |
| 186 | try renderToken(tree, stream, comptime_node.comptime_token, indent, Space.Space); |
| 187 | try renderExpression(allocator, stream, tree, indent, comptime_node.expr, space); |
| 179 | 188 | }, |
| 189 | |
| 180 | 190 | ast.Node.Id.AsyncAttribute => { |
| 181 | 191 | const async_attr = @fieldParentPtr(ast.Node.AsyncAttribute, "base", base); |
| 182 | | try stream.print("{}", tree.tokenSlice(async_attr.async_token)); |
| 183 | 192 | |
| 184 | 193 | if (async_attr.allocator_type) |allocator_type| { |
| 185 | | try stream.write("<"); |
| 186 | | try renderExpression(allocator, stream, tree, indent, allocator_type); |
| 187 | | try stream.write(">"); |
| 194 | try renderToken(tree, stream, async_attr.async_token, indent, Space.None); |
| 195 | |
| 196 | try renderToken(tree, stream, tree.nextToken(async_attr.async_token), indent, Space.None); |
| 197 | try renderExpression(allocator, stream, tree, indent, allocator_type, Space.None); |
| 198 | try renderToken(tree, stream, tree.nextToken(allocator_type.lastToken()), indent, space); |
| 199 | } else { |
| 200 | try renderToken(tree, stream, async_attr.async_token, indent, space); |
| 188 | 201 | } |
| 189 | 202 | }, |
| 203 | |
| 190 | 204 | ast.Node.Id.Suspend => { |
| 191 | 205 | const suspend_node = @fieldParentPtr(ast.Node.Suspend, "base", base); |
| 206 | |
| 192 | 207 | if (suspend_node.label) |label| { |
| 193 | | try stream.print("{}: ", tree.tokenSlice(label)); |
| 208 | try renderToken(tree, stream, label, indent, Space.None); |
| 209 | try renderToken(tree, stream, tree.nextToken(label), indent, Space.Space); |
| 194 | 210 | } |
| 195 | | try stream.print("{}", tree.tokenSlice(suspend_node.suspend_token)); |
| 196 | 211 | |
| 197 | 212 | if (suspend_node.payload) |payload| { |
| 198 | | try stream.write(" "); |
| 199 | | try renderExpression(allocator, stream, tree, indent, payload); |
| 200 | | } |
| 201 | | |
| 202 | | if (suspend_node.body) |body| { |
| 203 | | try stream.write(" "); |
| 204 | | try renderExpression(allocator, stream, tree, indent, body); |
| 213 | if (suspend_node.body) |body| { |
| 214 | try renderToken(tree, stream, suspend_node.suspend_token, indent, Space.Space); |
| 215 | try renderExpression(allocator, stream, tree, indent, payload, Space.Space); |
| 216 | try renderExpression(allocator, stream, tree, indent, body, space); |
| 217 | } else { |
| 218 | try renderToken(tree, stream, suspend_node.suspend_token, indent, Space.Space); |
| 219 | try renderExpression(allocator, stream, tree, indent, payload, space); |
| 220 | } |
| 221 | } else if (suspend_node.body) |body| { |
| 222 | try renderToken(tree, stream, suspend_node.suspend_token, indent, Space.Space); |
| 223 | try renderExpression(allocator, stream, tree, indent, body, space); |
| 224 | } else { |
| 225 | try renderToken(tree, stream, suspend_node.suspend_token, indent, space); |
| 205 | 226 | } |
| 206 | 227 | }, |
| 207 | 228 | |
| 208 | 229 | ast.Node.Id.InfixOp => { |
| 209 | 230 | const infix_op_node = @fieldParentPtr(ast.Node.InfixOp, "base", base); |
| 210 | 231 | |
| 211 | | try renderExpression(allocator, stream, tree, indent, infix_op_node.lhs); |
| 212 | | |
| 213 | | const text = switch (infix_op_node.op) { |
| 214 | | ast.Node.InfixOp.Op.Add => " + ", |
| 215 | | ast.Node.InfixOp.Op.AddWrap => " +% ", |
| 216 | | ast.Node.InfixOp.Op.ArrayCat => " ++ ", |
| 217 | | ast.Node.InfixOp.Op.ArrayMult => " ** ", |
| 218 | | ast.Node.InfixOp.Op.Assign => " = ", |
| 219 | | ast.Node.InfixOp.Op.AssignBitAnd => " &= ", |
| 220 | | ast.Node.InfixOp.Op.AssignBitOr => " |= ", |
| 221 | | ast.Node.InfixOp.Op.AssignBitShiftLeft => " <<= ", |
| 222 | | ast.Node.InfixOp.Op.AssignBitShiftRight => " >>= ", |
| 223 | | ast.Node.InfixOp.Op.AssignBitXor => " ^= ", |
| 224 | | ast.Node.InfixOp.Op.AssignDiv => " /= ", |
| 225 | | ast.Node.InfixOp.Op.AssignMinus => " -= ", |
| 226 | | ast.Node.InfixOp.Op.AssignMinusWrap => " -%= ", |
| 227 | | ast.Node.InfixOp.Op.AssignMod => " %= ", |
| 228 | | ast.Node.InfixOp.Op.AssignPlus => " += ", |
| 229 | | ast.Node.InfixOp.Op.AssignPlusWrap => " +%= ", |
| 230 | | ast.Node.InfixOp.Op.AssignTimes => " *= ", |
| 231 | | ast.Node.InfixOp.Op.AssignTimesWarp => " *%= ", |
| 232 | | ast.Node.InfixOp.Op.BangEqual => " != ", |
| 233 | | ast.Node.InfixOp.Op.BitAnd => " & ", |
| 234 | | ast.Node.InfixOp.Op.BitOr => " | ", |
| 235 | | ast.Node.InfixOp.Op.BitShiftLeft => " << ", |
| 236 | | ast.Node.InfixOp.Op.BitShiftRight => " >> ", |
| 237 | | ast.Node.InfixOp.Op.BitXor => " ^ ", |
| 238 | | ast.Node.InfixOp.Op.BoolAnd => " and ", |
| 239 | | ast.Node.InfixOp.Op.BoolOr => " or ", |
| 240 | | ast.Node.InfixOp.Op.Div => " / ", |
| 241 | | ast.Node.InfixOp.Op.EqualEqual => " == ", |
| 242 | | ast.Node.InfixOp.Op.ErrorUnion => "!", |
| 243 | | ast.Node.InfixOp.Op.GreaterOrEqual => " >= ", |
| 244 | | ast.Node.InfixOp.Op.GreaterThan => " > ", |
| 245 | | ast.Node.InfixOp.Op.LessOrEqual => " <= ", |
| 246 | | ast.Node.InfixOp.Op.LessThan => " < ", |
| 247 | | ast.Node.InfixOp.Op.MergeErrorSets => " || ", |
| 248 | | ast.Node.InfixOp.Op.Mod => " % ", |
| 249 | | ast.Node.InfixOp.Op.Mult => " * ", |
| 250 | | ast.Node.InfixOp.Op.MultWrap => " *% ", |
| 251 | | ast.Node.InfixOp.Op.Period => ".", |
| 252 | | ast.Node.InfixOp.Op.Sub => " - ", |
| 253 | | ast.Node.InfixOp.Op.SubWrap => " -% ", |
| 254 | | ast.Node.InfixOp.Op.UnwrapMaybe => " ?? ", |
| 255 | | ast.Node.InfixOp.Op.Range => " ... ", |
| 256 | | ast.Node.InfixOp.Op.Catch => |maybe_payload| blk: { |
| 257 | | try stream.write(" catch "); |
| 258 | | if (maybe_payload) |payload| { |
| 259 | | try renderExpression(allocator, stream, tree, indent, payload); |
| 260 | | try stream.write(" "); |
| 261 | | } |
| 262 | | break :blk ""; |
| 263 | | }, |
| 232 | const op_token = tree.tokens.at(infix_op_node.op_token); |
| 233 | const op_space = switch (infix_op_node.op) { |
| 234 | ast.Node.InfixOp.Op.Period, ast.Node.InfixOp.Op.ErrorUnion => Space.None, |
| 235 | else => Space.Space, |
| 264 | 236 | }; |
| 237 | try renderExpression(allocator, stream, tree, indent, infix_op_node.lhs, op_space); |
| 238 | try renderToken(tree, stream, infix_op_node.op_token, indent, op_space); |
| 239 | |
| 240 | switch (infix_op_node.op) { |
| 241 | ast.Node.InfixOp.Op.Catch => |maybe_payload| if (maybe_payload) |payload| { |
| 242 | try renderExpression(allocator, stream, tree, indent, payload, Space.Space); |
| 243 | }, |
| 244 | else => {}, |
| 245 | } |
| 265 | 246 | |
| 266 | | try stream.write(text); |
| 267 | | try renderExpression(allocator, stream, tree, indent, infix_op_node.rhs); |
| 247 | try renderExpression(allocator, stream, tree, indent, infix_op_node.rhs, space); |
| 268 | 248 | }, |
| 269 | 249 | |
| 270 | 250 | ast.Node.Id.PrefixOp => { |
| ... | ... | @@ -272,52 +252,73 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 272 | 252 | |
| 273 | 253 | switch (prefix_op_node.op) { |
| 274 | 254 | ast.Node.PrefixOp.Op.AddrOf => |addr_of_info| { |
| 275 | | try stream.write("&"); |
| 255 | try renderToken(tree, stream, prefix_op_node.op_token, indent, Space.None); // & |
| 276 | 256 | if (addr_of_info.align_expr) |align_expr| { |
| 277 | | try stream.write("align("); |
| 278 | | try renderExpression(allocator, stream, tree, indent, align_expr); |
| 279 | | try stream.write(") "); |
| 257 | const align_token = tree.nextToken(prefix_op_node.op_token); |
| 258 | try renderToken(tree, stream, align_token, indent, Space.None); // align |
| 259 | |
| 260 | const lparen_token = tree.prevToken(align_expr.firstToken()); |
| 261 | try renderToken(tree, stream, lparen_token, indent, Space.None); // ( |
| 262 | |
| 263 | try renderExpression(allocator, stream, tree, indent, align_expr, Space.None); |
| 264 | |
| 265 | const rparen_token = tree.nextToken(align_expr.lastToken()); |
| 266 | try renderToken(tree, stream, rparen_token, indent, Space.Space); // ) |
| 280 | 267 | } |
| 281 | | if (addr_of_info.const_token != null) { |
| 282 | | try stream.write("const "); |
| 268 | if (addr_of_info.const_token) |const_token| { |
| 269 | try renderToken(tree, stream, const_token, indent, Space.Space); // const |
| 283 | 270 | } |
| 284 | | if (addr_of_info.volatile_token != null) { |
| 285 | | try stream.write("volatile "); |
| 271 | if (addr_of_info.volatile_token) |volatile_token| { |
| 272 | try renderToken(tree, stream, volatile_token, indent, Space.Space); // volatile |
| 286 | 273 | } |
| 287 | 274 | }, |
| 288 | 275 | ast.Node.PrefixOp.Op.SliceType => |addr_of_info| { |
| 289 | | try stream.write("[]"); |
| 276 | try renderToken(tree, stream, prefix_op_node.op_token, indent, Space.None); // [ |
| 277 | try renderToken(tree, stream, tree.nextToken(prefix_op_node.op_token), indent, Space.None); // ] |
| 278 | |
| 290 | 279 | if (addr_of_info.align_expr) |align_expr| { |
| 291 | | try stream.print("align("); |
| 292 | | try renderExpression(allocator, stream, tree, indent, align_expr); |
| 293 | | try stream.print(") "); |
| 280 | const align_token = tree.nextToken(prefix_op_node.op_token); |
| 281 | try renderToken(tree, stream, align_token, indent, Space.None); // align |
| 282 | |
| 283 | const lparen_token = tree.prevToken(align_expr.firstToken()); |
| 284 | try renderToken(tree, stream, lparen_token, indent, Space.None); // ( |
| 285 | |
| 286 | try renderExpression(allocator, stream, tree, indent, align_expr, Space.None); |
| 287 | |
| 288 | const rparen_token = tree.nextToken(align_expr.lastToken()); |
| 289 | try renderToken(tree, stream, rparen_token, indent, Space.Space); // ) |
| 294 | 290 | } |
| 295 | | if (addr_of_info.const_token != null) { |
| 296 | | try stream.print("const "); |
| 291 | if (addr_of_info.const_token) |const_token| { |
| 292 | try renderToken(tree, stream, const_token, indent, Space.Space); |
| 297 | 293 | } |
| 298 | | if (addr_of_info.volatile_token != null) { |
| 299 | | try stream.print("volatile "); |
| 294 | if (addr_of_info.volatile_token) |volatile_token| { |
| 295 | try renderToken(tree, stream, volatile_token, indent, Space.Space); |
| 300 | 296 | } |
| 301 | 297 | }, |
| 302 | 298 | ast.Node.PrefixOp.Op.ArrayType => |array_index| { |
| 303 | | try stream.print("["); |
| 304 | | try renderExpression(allocator, stream, tree, indent, array_index); |
| 305 | | try stream.print("]"); |
| 299 | try renderToken(tree, stream, prefix_op_node.op_token, indent, Space.None); // [ |
| 300 | try renderExpression(allocator, stream, tree, indent, array_index, Space.None); |
| 301 | try renderToken(tree, stream, tree.nextToken(array_index.lastToken()), indent, Space.None); // ] |
| 302 | }, |
| 303 | ast.Node.PrefixOp.Op.BitNot, |
| 304 | ast.Node.PrefixOp.Op.BoolNot, |
| 305 | ast.Node.PrefixOp.Op.Negation, |
| 306 | ast.Node.PrefixOp.Op.NegationWrap, |
| 307 | ast.Node.PrefixOp.Op.UnwrapMaybe, |
| 308 | ast.Node.PrefixOp.Op.MaybeType, |
| 309 | ast.Node.PrefixOp.Op.PointerType => { |
| 310 | try renderToken(tree, stream, prefix_op_node.op_token, indent, Space.None); |
| 306 | 311 | }, |
| 307 | | ast.Node.PrefixOp.Op.BitNot => try stream.write("~"), |
| 308 | | ast.Node.PrefixOp.Op.BoolNot => try stream.write("!"), |
| 309 | | ast.Node.PrefixOp.Op.Negation => try stream.write("-"), |
| 310 | | ast.Node.PrefixOp.Op.NegationWrap => try stream.write("-%"), |
| 311 | | ast.Node.PrefixOp.Op.Try => try stream.write("try "), |
| 312 | | ast.Node.PrefixOp.Op.UnwrapMaybe => try stream.write("??"), |
| 313 | | ast.Node.PrefixOp.Op.MaybeType => try stream.write("?"), |
| 314 | | ast.Node.PrefixOp.Op.PointerType => try stream.write("*"), |
| 315 | | ast.Node.PrefixOp.Op.Await => try stream.write("await "), |
| 316 | | ast.Node.PrefixOp.Op.Cancel => try stream.write("cancel "), |
| 317 | | ast.Node.PrefixOp.Op.Resume => try stream.write("resume "), |
| 318 | | } |
| 319 | | |
| 320 | | try renderExpression(allocator, stream, tree, indent, prefix_op_node.rhs); |
| 312 | |
| 313 | ast.Node.PrefixOp.Op.Try, |
| 314 | ast.Node.PrefixOp.Op.Await, |
| 315 | ast.Node.PrefixOp.Op.Cancel, |
| 316 | ast.Node.PrefixOp.Op.Resume => { |
| 317 | try renderToken(tree, stream, prefix_op_node.op_token, indent, Space.Space); |
| 318 | }, |
| 319 | } |
| 320 | |
| 321 | try renderExpression(allocator, stream, tree, indent, prefix_op_node.rhs, space); |
| 321 | 322 | }, |
| 322 | 323 | |
| 323 | 324 | ast.Node.Id.SuffixOp => { |
| ... | ... | @@ -326,123 +327,139 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 326 | 327 | switch (suffix_op.op) { |
| 327 | 328 | @TagType(ast.Node.SuffixOp.Op).Call => |*call_info| { |
| 328 | 329 | if (call_info.async_attr) |async_attr| { |
| 329 | | try renderExpression(allocator, stream, tree, indent, &async_attr.base); |
| 330 | | try stream.write(" "); |
| 330 | try renderExpression(allocator, stream, tree, indent, &async_attr.base, Space.Space); |
| 331 | 331 | } |
| 332 | 332 | |
| 333 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 334 | | try stream.write("("); |
| 333 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 334 | |
| 335 | const lparen = tree.nextToken(suffix_op.lhs.lastToken()); |
| 336 | try renderToken(tree, stream, lparen, indent, Space.None); |
| 335 | 337 | |
| 336 | 338 | var it = call_info.params.iterator(0); |
| 337 | 339 | while (it.next()) |param_node| { |
| 338 | | try renderExpression(allocator, stream, tree, indent, param_node.*); |
| 340 | try renderExpression(allocator, stream, tree, indent, param_node.*, Space.None); |
| 341 | |
| 339 | 342 | if (it.peek() != null) { |
| 340 | | try stream.write(", "); |
| 343 | const comma = tree.nextToken(param_node.*.lastToken()); |
| 344 | try renderToken(tree, stream, comma, indent, Space.Space); |
| 341 | 345 | } |
| 342 | 346 | } |
| 343 | 347 | |
| 344 | | try stream.write(")"); |
| 348 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); |
| 345 | 349 | }, |
| 346 | 350 | |
| 347 | 351 | ast.Node.SuffixOp.Op.ArrayAccess => |index_expr| { |
| 348 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 349 | | try stream.write("["); |
| 350 | | try renderExpression(allocator, stream, tree, indent, index_expr); |
| 351 | | try stream.write("]"); |
| 352 | const lbracket = tree.prevToken(index_expr.firstToken()); |
| 353 | const rbracket = tree.nextToken(index_expr.lastToken()); |
| 354 | |
| 355 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 356 | try renderToken(tree, stream, lbracket, indent, Space.None); // [ |
| 357 | try renderExpression(allocator, stream, tree, indent, index_expr, Space.None); |
| 358 | try renderToken(tree, stream, rbracket, indent, space); // ] |
| 352 | 359 | }, |
| 353 | 360 | |
| 354 | 361 | ast.Node.SuffixOp.Op.Deref => { |
| 355 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 356 | | try stream.write(".*"); |
| 362 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 363 | try renderToken(tree, stream, tree.prevToken(suffix_op.rtoken), indent, Space.None); // . |
| 364 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); // * |
| 357 | 365 | }, |
| 358 | 366 | |
| 359 | 367 | @TagType(ast.Node.SuffixOp.Op).Slice => |range| { |
| 360 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 361 | | try stream.write("["); |
| 362 | | try renderExpression(allocator, stream, tree, indent, range.start); |
| 363 | | try stream.write(".."); |
| 368 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 369 | |
| 370 | const lbracket = tree.prevToken(range.start.firstToken()); |
| 371 | const dotdot = tree.nextToken(range.start.lastToken()); |
| 372 | |
| 373 | try renderToken(tree, stream, lbracket, indent, Space.None); // [ |
| 374 | try renderExpression(allocator, stream, tree, indent, range.start, Space.None); |
| 375 | try renderToken(tree, stream, dotdot, indent, Space.None); // .. |
| 364 | 376 | if (range.end) |end| { |
| 365 | | try renderExpression(allocator, stream, tree, indent, end); |
| 377 | try renderExpression(allocator, stream, tree, indent, end, Space.None); |
| 366 | 378 | } |
| 367 | | try stream.write("]"); |
| 379 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); // ] |
| 368 | 380 | }, |
| 369 | 381 | |
| 370 | 382 | ast.Node.SuffixOp.Op.StructInitializer => |*field_inits| { |
| 383 | const lbrace = tree.nextToken(suffix_op.lhs.lastToken()); |
| 384 | |
| 371 | 385 | if (field_inits.len == 0) { |
| 372 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 373 | | try stream.write("{}"); |
| 386 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 387 | try renderToken(tree, stream, lbrace, indent, Space.None); |
| 388 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); |
| 374 | 389 | return; |
| 375 | 390 | } |
| 376 | 391 | |
| 377 | 392 | if (field_inits.len == 1) { |
| 378 | 393 | const field_init = field_inits.at(0).*; |
| 379 | 394 | |
| 380 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 381 | | try stream.write("{ "); |
| 382 | | try renderExpression(allocator, stream, tree, indent, field_init); |
| 383 | | try stream.write(" }"); |
| 395 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 396 | try renderToken(tree, stream, lbrace, indent, Space.Space); |
| 397 | try renderExpression(allocator, stream, tree, indent, field_init, Space.Space); |
| 398 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); |
| 384 | 399 | return; |
| 385 | 400 | } |
| 386 | 401 | |
| 387 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 388 | | try stream.write("{\n"); |
| 402 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 403 | try renderToken(tree, stream, lbrace, indent, Space.Newline); |
| 389 | 404 | |
| 390 | 405 | const new_indent = indent + indent_delta; |
| 391 | 406 | |
| 392 | 407 | var it = field_inits.iterator(0); |
| 393 | 408 | while (it.next()) |field_init| { |
| 394 | 409 | try stream.writeByteNTimes(' ', new_indent); |
| 395 | | try renderExpression(allocator, stream, tree, new_indent, field_init.*); |
| 396 | | if ((field_init.*).id != ast.Node.Id.LineComment) { |
| 397 | | try stream.write(","); |
| 398 | | } |
| 410 | try renderExpression(allocator, stream, tree, new_indent, field_init.*, Space.None); |
| 411 | |
| 412 | const comma = tree.nextToken(field_init.*.lastToken()); |
| 413 | try renderToken(tree, stream, comma, new_indent, Space.Newline); |
| 414 | |
| 399 | 415 | if (it.peek()) |next_field_init| { |
| 400 | | const n = if (nodeLineOffset(tree, field_init.*, next_field_init.*) >= 2) u8(2) else u8(1); |
| 401 | | try stream.writeByteNTimes('\n', n); |
| 416 | try renderExtraNewline(tree, stream, next_field_init.*); |
| 402 | 417 | } |
| 403 | 418 | } |
| 404 | 419 | |
| 405 | | try stream.write("\n"); |
| 406 | 420 | try stream.writeByteNTimes(' ', indent); |
| 407 | | try stream.write("}"); |
| 421 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); |
| 408 | 422 | }, |
| 409 | 423 | |
| 410 | 424 | ast.Node.SuffixOp.Op.ArrayInitializer => |*exprs| { |
| 425 | const lbrace = tree.nextToken(suffix_op.lhs.lastToken()); |
| 426 | |
| 411 | 427 | if (exprs.len == 0) { |
| 412 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 413 | | try stream.write("{}"); |
| 428 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 429 | try renderToken(tree, stream, lbrace, indent, Space.None); |
| 430 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); |
| 414 | 431 | return; |
| 415 | 432 | } |
| 416 | 433 | if (exprs.len == 1) { |
| 417 | 434 | const expr = exprs.at(0).*; |
| 418 | 435 | |
| 419 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 420 | | try stream.write("{"); |
| 421 | | try renderExpression(allocator, stream, tree, indent, expr); |
| 422 | | try stream.write("}"); |
| 436 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 437 | try renderToken(tree, stream, lbrace, indent, Space.None); |
| 438 | try renderExpression(allocator, stream, tree, indent, expr, Space.None); |
| 439 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); |
| 423 | 440 | return; |
| 424 | 441 | } |
| 425 | 442 | |
| 426 | | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs); |
| 443 | try renderExpression(allocator, stream, tree, indent, suffix_op.lhs, Space.None); |
| 427 | 444 | |
| 428 | 445 | const new_indent = indent + indent_delta; |
| 429 | | try stream.write("{\n"); |
| 446 | try renderToken(tree, stream, lbrace, new_indent, Space.Newline); |
| 430 | 447 | |
| 431 | 448 | var it = exprs.iterator(0); |
| 432 | 449 | while (it.next()) |expr| { |
| 433 | 450 | try stream.writeByteNTimes(' ', new_indent); |
| 434 | | try renderExpression(allocator, stream, tree, new_indent, expr.*); |
| 435 | | try stream.write(","); |
| 451 | try renderExpression(allocator, stream, tree, new_indent, expr.*, Space.None); |
| 452 | |
| 453 | const comma = tree.nextToken(expr.*.lastToken()); |
| 454 | try renderToken(tree, stream, comma, new_indent, Space.Newline); // , |
| 436 | 455 | |
| 437 | 456 | if (it.peek()) |next_expr| { |
| 438 | | const n = if (nodeLineOffset(tree, expr.*, next_expr.*) >= 2) u8(2) else u8(1); |
| 439 | | try stream.writeByteNTimes('\n', n); |
| 457 | try renderExtraNewline(tree, stream, next_expr.*); |
| 440 | 458 | } |
| 441 | 459 | } |
| 442 | 460 | |
| 443 | | try stream.write("\n"); |
| 444 | 461 | try stream.writeByteNTimes(' ', indent); |
| 445 | | try stream.write("}"); |
| 462 | try renderToken(tree, stream, suffix_op.rtoken, indent, space); |
| 446 | 463 | }, |
| 447 | 464 | } |
| 448 | 465 | }, |
| ... | ... | @@ -452,159 +469,182 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 452 | 469 | |
| 453 | 470 | switch (flow_expr.kind) { |
| 454 | 471 | ast.Node.ControlFlowExpression.Kind.Break => |maybe_label| { |
| 455 | | try stream.print("break"); |
| 472 | const kw_space = if (maybe_label != null or flow_expr.rhs != null) Space.Space else space; |
| 473 | try renderToken(tree, stream, flow_expr.ltoken, indent, kw_space); |
| 456 | 474 | if (maybe_label) |label| { |
| 457 | | try stream.print(" :"); |
| 458 | | try renderExpression(allocator, stream, tree, indent, label); |
| 475 | const colon = tree.nextToken(flow_expr.ltoken); |
| 476 | try renderToken(tree, stream, colon, indent, Space.None); |
| 477 | |
| 478 | const expr_space = if (flow_expr.rhs != null) Space.Space else space; |
| 479 | try renderExpression(allocator, stream, tree, indent, label, expr_space); |
| 459 | 480 | } |
| 460 | 481 | }, |
| 461 | 482 | ast.Node.ControlFlowExpression.Kind.Continue => |maybe_label| { |
| 462 | | try stream.print("continue"); |
| 483 | const kw_space = if (maybe_label != null or flow_expr.rhs != null) Space.Space else space; |
| 484 | try renderToken(tree, stream, flow_expr.ltoken, indent, kw_space); |
| 463 | 485 | if (maybe_label) |label| { |
| 464 | | try stream.print(" :"); |
| 465 | | try renderExpression(allocator, stream, tree, indent, label); |
| 486 | const colon = tree.nextToken(flow_expr.ltoken); |
| 487 | try renderToken(tree, stream, colon, indent, Space.None); |
| 488 | |
| 489 | const expr_space = if (flow_expr.rhs != null) Space.Space else space; |
| 490 | try renderExpression(allocator, stream, tree, indent, label, space); |
| 466 | 491 | } |
| 467 | 492 | }, |
| 468 | 493 | ast.Node.ControlFlowExpression.Kind.Return => { |
| 469 | | try stream.print("return"); |
| 494 | const kw_space = if (flow_expr.rhs != null) Space.Space else space; |
| 495 | try renderToken(tree, stream, flow_expr.ltoken, indent, kw_space); |
| 470 | 496 | }, |
| 471 | 497 | } |
| 472 | 498 | |
| 473 | 499 | if (flow_expr.rhs) |rhs| { |
| 474 | | try stream.write(" "); |
| 475 | | try renderExpression(allocator, stream, tree, indent, rhs); |
| 500 | try renderExpression(allocator, stream, tree, indent, rhs, space); |
| 476 | 501 | } |
| 477 | 502 | }, |
| 478 | 503 | |
| 479 | 504 | ast.Node.Id.Payload => { |
| 480 | 505 | const payload = @fieldParentPtr(ast.Node.Payload, "base", base); |
| 481 | 506 | |
| 482 | | try stream.write("|"); |
| 483 | | try renderExpression(allocator, stream, tree, indent, payload.error_symbol); |
| 484 | | try stream.write("|"); |
| 507 | try renderToken(tree, stream, payload.lpipe, indent, Space.None); |
| 508 | try renderExpression(allocator, stream, tree, indent, payload.error_symbol, Space.None); |
| 509 | try renderToken(tree, stream, payload.rpipe, indent, space); |
| 485 | 510 | }, |
| 486 | 511 | |
| 487 | 512 | ast.Node.Id.PointerPayload => { |
| 488 | 513 | const payload = @fieldParentPtr(ast.Node.PointerPayload, "base", base); |
| 489 | 514 | |
| 490 | | try stream.write("|"); |
| 515 | try renderToken(tree, stream, payload.lpipe, indent, Space.None); |
| 491 | 516 | if (payload.ptr_token) |ptr_token| { |
| 492 | | try stream.write(tree.tokenSlice(ptr_token)); |
| 517 | try renderToken(tree, stream, ptr_token, indent, Space.None); |
| 493 | 518 | } |
| 494 | | try renderExpression(allocator, stream, tree, indent, payload.value_symbol); |
| 495 | | try stream.write("|"); |
| 519 | try renderExpression(allocator, stream, tree, indent, payload.value_symbol, Space.None); |
| 520 | try renderToken(tree, stream, payload.rpipe, indent, space); |
| 496 | 521 | }, |
| 497 | 522 | |
| 498 | 523 | ast.Node.Id.PointerIndexPayload => { |
| 499 | 524 | const payload = @fieldParentPtr(ast.Node.PointerIndexPayload, "base", base); |
| 500 | 525 | |
| 501 | | try stream.write("|"); |
| 526 | try renderToken(tree, stream, payload.lpipe, indent, Space.None); |
| 502 | 527 | if (payload.ptr_token) |ptr_token| { |
| 503 | | try stream.write(tree.tokenSlice(ptr_token)); |
| 528 | try renderToken(tree, stream, ptr_token, indent, Space.None); |
| 504 | 529 | } |
| 505 | | try renderExpression(allocator, stream, tree, indent, payload.value_symbol); |
| 530 | try renderExpression(allocator, stream, tree, indent, payload.value_symbol, Space.None); |
| 506 | 531 | |
| 507 | 532 | if (payload.index_symbol) |index_symbol| { |
| 508 | | try stream.write(", "); |
| 509 | | try renderExpression(allocator, stream, tree, indent, index_symbol); |
| 533 | const comma = tree.nextToken(payload.value_symbol.lastToken()); |
| 534 | |
| 535 | try renderToken(tree, stream, comma, indent, Space.Space); |
| 536 | try renderExpression(allocator, stream, tree, indent, index_symbol, Space.None); |
| 510 | 537 | } |
| 511 | 538 | |
| 512 | | try stream.write("|"); |
| 539 | try renderToken(tree, stream, payload.rpipe, indent, space); |
| 513 | 540 | }, |
| 514 | 541 | |
| 515 | 542 | ast.Node.Id.GroupedExpression => { |
| 516 | 543 | const grouped_expr = @fieldParentPtr(ast.Node.GroupedExpression, "base", base); |
| 517 | 544 | |
| 518 | | try renderToken(tree, stream, grouped_expr.lparen, indent, false, false); |
| 519 | | try renderExpression(allocator, stream, tree, indent, grouped_expr.expr); |
| 520 | | try renderToken(tree, stream, grouped_expr.rparen, indent, false, false); |
| 545 | try renderToken(tree, stream, grouped_expr.lparen, indent, Space.None); |
| 546 | try renderExpression(allocator, stream, tree, indent, grouped_expr.expr, Space.None); |
| 547 | try renderToken(tree, stream, grouped_expr.rparen, indent, space); |
| 521 | 548 | }, |
| 522 | 549 | |
| 523 | 550 | ast.Node.Id.FieldInitializer => { |
| 524 | 551 | const field_init = @fieldParentPtr(ast.Node.FieldInitializer, "base", base); |
| 525 | 552 | |
| 526 | | try stream.print(".{} = ", tree.tokenSlice(field_init.name_token)); |
| 527 | | try renderExpression(allocator, stream, tree, indent, field_init.expr); |
| 553 | try renderToken(tree, stream, field_init.period_token, indent, Space.None); // . |
| 554 | try renderToken(tree, stream, field_init.name_token, indent, Space.Space); // name |
| 555 | try renderToken(tree, stream, tree.nextToken(field_init.name_token), indent, Space.Space); // = |
| 556 | try renderExpression(allocator, stream, tree, indent, field_init.expr, space); |
| 528 | 557 | }, |
| 529 | 558 | |
| 530 | 559 | ast.Node.Id.IntegerLiteral => { |
| 531 | 560 | const integer_literal = @fieldParentPtr(ast.Node.IntegerLiteral, "base", base); |
| 532 | | try renderToken(tree, stream, integer_literal.token, indent, false, false); |
| 561 | try renderToken(tree, stream, integer_literal.token, indent, space); |
| 533 | 562 | }, |
| 534 | 563 | ast.Node.Id.FloatLiteral => { |
| 535 | 564 | const float_literal = @fieldParentPtr(ast.Node.FloatLiteral, "base", base); |
| 536 | | try stream.print("{}", tree.tokenSlice(float_literal.token)); |
| 565 | try renderToken(tree, stream, float_literal.token, indent, space); |
| 537 | 566 | }, |
| 538 | 567 | ast.Node.Id.StringLiteral => { |
| 539 | 568 | const string_literal = @fieldParentPtr(ast.Node.StringLiteral, "base", base); |
| 540 | | try renderToken(tree, stream, string_literal.token, indent, false, false); |
| 569 | try renderToken(tree, stream, string_literal.token, indent, space); |
| 541 | 570 | }, |
| 542 | 571 | ast.Node.Id.CharLiteral => { |
| 543 | 572 | const char_literal = @fieldParentPtr(ast.Node.CharLiteral, "base", base); |
| 544 | | try stream.print("{}", tree.tokenSlice(char_literal.token)); |
| 573 | try renderToken(tree, stream, char_literal.token, indent, space); |
| 545 | 574 | }, |
| 546 | 575 | ast.Node.Id.BoolLiteral => { |
| 547 | 576 | const bool_literal = @fieldParentPtr(ast.Node.CharLiteral, "base", base); |
| 548 | | try stream.print("{}", tree.tokenSlice(bool_literal.token)); |
| 577 | try renderToken(tree, stream, bool_literal.token, indent, space); |
| 549 | 578 | }, |
| 550 | 579 | ast.Node.Id.NullLiteral => { |
| 551 | 580 | const null_literal = @fieldParentPtr(ast.Node.NullLiteral, "base", base); |
| 552 | | try stream.print("{}", tree.tokenSlice(null_literal.token)); |
| 581 | try renderToken(tree, stream, null_literal.token, indent, space); |
| 553 | 582 | }, |
| 554 | 583 | ast.Node.Id.ThisLiteral => { |
| 555 | 584 | const this_literal = @fieldParentPtr(ast.Node.ThisLiteral, "base", base); |
| 556 | | try stream.print("{}", tree.tokenSlice(this_literal.token)); |
| 585 | try renderToken(tree, stream, this_literal.token, indent, space); |
| 557 | 586 | }, |
| 558 | 587 | ast.Node.Id.Unreachable => { |
| 559 | 588 | const unreachable_node = @fieldParentPtr(ast.Node.Unreachable, "base", base); |
| 560 | | try stream.print("{}", tree.tokenSlice(unreachable_node.token)); |
| 589 | try renderToken(tree, stream, unreachable_node.token, indent, space); |
| 561 | 590 | }, |
| 562 | 591 | ast.Node.Id.ErrorType => { |
| 563 | 592 | const error_type = @fieldParentPtr(ast.Node.ErrorType, "base", base); |
| 564 | | try stream.print("{}", tree.tokenSlice(error_type.token)); |
| 593 | try renderToken(tree, stream, error_type.token, indent, space); |
| 565 | 594 | }, |
| 566 | 595 | ast.Node.Id.VarType => { |
| 567 | 596 | const var_type = @fieldParentPtr(ast.Node.VarType, "base", base); |
| 568 | | try stream.print("{}", tree.tokenSlice(var_type.token)); |
| 597 | try renderToken(tree, stream, var_type.token, indent, space); |
| 569 | 598 | }, |
| 570 | 599 | ast.Node.Id.ContainerDecl => { |
| 571 | 600 | const container_decl = @fieldParentPtr(ast.Node.ContainerDecl, "base", base); |
| 572 | 601 | |
| 573 | | switch (container_decl.layout) { |
| 574 | | ast.Node.ContainerDecl.Layout.Packed => try stream.print("packed "), |
| 575 | | ast.Node.ContainerDecl.Layout.Extern => try stream.print("extern "), |
| 576 | | ast.Node.ContainerDecl.Layout.Auto => {}, |
| 577 | | } |
| 578 | | |
| 579 | | switch (container_decl.kind) { |
| 580 | | ast.Node.ContainerDecl.Kind.Struct => try stream.print("struct"), |
| 581 | | ast.Node.ContainerDecl.Kind.Enum => try stream.print("enum"), |
| 582 | | ast.Node.ContainerDecl.Kind.Union => try stream.print("union"), |
| 602 | if (container_decl.layout_token) |layout_token| { |
| 603 | try renderToken(tree, stream, layout_token, indent, Space.Space); |
| 583 | 604 | } |
| 584 | 605 | |
| 585 | 606 | switch (container_decl.init_arg_expr) { |
| 586 | | ast.Node.ContainerDecl.InitArg.None => try stream.write(" "), |
| 607 | ast.Node.ContainerDecl.InitArg.None => { |
| 608 | try renderToken(tree, stream, container_decl.kind_token, indent, Space.Space); // union |
| 609 | }, |
| 587 | 610 | ast.Node.ContainerDecl.InitArg.Enum => |enum_tag_type| { |
| 611 | try renderToken(tree, stream, container_decl.kind_token, indent, Space.None); // union |
| 612 | |
| 613 | const lparen = tree.nextToken(container_decl.kind_token); |
| 614 | const enum_token = tree.nextToken(lparen); |
| 615 | |
| 616 | try renderToken(tree, stream, lparen, indent, Space.None); // ( |
| 617 | try renderToken(tree, stream, enum_token, indent, Space.None); // enum |
| 618 | |
| 588 | 619 | if (enum_tag_type) |expr| { |
| 589 | | try stream.write("(enum("); |
| 590 | | try renderExpression(allocator, stream, tree, indent, expr); |
| 591 | | try stream.write(")) "); |
| 620 | try renderToken(tree, stream, tree.nextToken(enum_token), indent, Space.None); // ( |
| 621 | try renderExpression(allocator, stream, tree, indent, expr, Space.None); |
| 622 | |
| 623 | const rparen = tree.nextToken(expr.lastToken()); |
| 624 | try renderToken(tree, stream, rparen, indent, Space.None); // ) |
| 625 | try renderToken(tree, stream, tree.nextToken(rparen), indent, Space.Space); // ) |
| 592 | 626 | } else { |
| 593 | | try stream.write("(enum) "); |
| 627 | try renderToken(tree, stream, tree.nextToken(enum_token), indent, Space.Space); // ) |
| 594 | 628 | } |
| 595 | 629 | }, |
| 596 | 630 | ast.Node.ContainerDecl.InitArg.Type => |type_expr| { |
| 597 | | try stream.write("("); |
| 598 | | try renderExpression(allocator, stream, tree, indent, type_expr); |
| 599 | | try stream.write(") "); |
| 631 | try renderToken(tree, stream, container_decl.kind_token, indent, Space.None); // union |
| 632 | |
| 633 | const lparen = tree.nextToken(container_decl.kind_token); |
| 634 | const rparen = tree.nextToken(type_expr.lastToken()); |
| 635 | |
| 636 | try renderToken(tree, stream, lparen, indent, Space.None); // ( |
| 637 | try renderExpression(allocator, stream, tree, indent, type_expr, Space.None); |
| 638 | try renderToken(tree, stream, rparen, indent, Space.Space); // ) |
| 600 | 639 | }, |
| 601 | 640 | } |
| 602 | 641 | |
| 603 | 642 | if (container_decl.fields_and_decls.len == 0) { |
| 604 | | try stream.write("{}"); |
| 643 | try renderToken(tree, stream, container_decl.lbrace_token, indent + indent_delta, Space.None); // { |
| 644 | try renderToken(tree, stream, container_decl.rbrace_token, indent, space); // } |
| 605 | 645 | } else { |
| 606 | | try stream.write("{\n"); |
| 607 | 646 | const new_indent = indent + indent_delta; |
| 647 | try renderToken(tree, stream, container_decl.lbrace_token, new_indent, Space.Newline); // { |
| 608 | 648 | |
| 609 | 649 | var it = container_decl.fields_and_decls.iterator(0); |
| 610 | 650 | while (it.next()) |decl| { |
| ... | ... | @@ -612,22 +652,24 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 612 | 652 | try renderTopLevelDecl(allocator, stream, tree, new_indent, decl.*); |
| 613 | 653 | |
| 614 | 654 | if (it.peek()) |next_decl| { |
| 615 | | const n = if (nodeLineOffset(tree, decl.*, next_decl.*) >= 2) u8(2) else u8(1); |
| 616 | | try stream.writeByteNTimes('\n', n); |
| 655 | try renderExtraNewline(tree, stream, next_decl.*); |
| 617 | 656 | } |
| 618 | 657 | } |
| 619 | 658 | |
| 620 | | try stream.write("\n"); |
| 621 | 659 | try stream.writeByteNTimes(' ', indent); |
| 622 | | try stream.write("}"); |
| 660 | try renderToken(tree, stream, container_decl.rbrace_token, indent, space); // } |
| 623 | 661 | } |
| 624 | 662 | }, |
| 625 | 663 | |
| 626 | 664 | ast.Node.Id.ErrorSetDecl => { |
| 627 | 665 | const err_set_decl = @fieldParentPtr(ast.Node.ErrorSetDecl, "base", base); |
| 628 | 666 | |
| 667 | const lbrace = tree.nextToken(err_set_decl.error_token); |
| 668 | |
| 629 | 669 | if (err_set_decl.decls.len == 0) { |
| 630 | | try stream.write("error{}"); |
| 670 | try renderToken(tree, stream, err_set_decl.error_token, indent, Space.None); |
| 671 | try renderToken(tree, stream, lbrace, indent, Space.None); |
| 672 | try renderToken(tree, stream, err_set_decl.rbrace_token, indent, space); |
| 631 | 673 | return; |
| 632 | 674 | } |
| 633 | 675 | |
| ... | ... | @@ -642,62 +684,80 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 642 | 684 | break :blk; |
| 643 | 685 | } |
| 644 | 686 | |
| 645 | | try stream.write("error{"); |
| 646 | | try renderTopLevelDecl(allocator, stream, tree, indent, node); |
| 647 | | try stream.write("}"); |
| 687 | try renderToken(tree, stream, err_set_decl.error_token, indent, Space.None); // error |
| 688 | try renderToken(tree, stream, lbrace, indent, Space.None); // { |
| 689 | try renderExpression(allocator, stream, tree, indent, node, Space.None); |
| 690 | try renderToken(tree, stream, err_set_decl.rbrace_token, indent, space); // } |
| 648 | 691 | return; |
| 649 | 692 | } |
| 650 | 693 | |
| 651 | | try stream.write("error{\n"); |
| 694 | try renderToken(tree, stream, err_set_decl.error_token, indent, Space.None); // error |
| 695 | try renderToken(tree, stream, lbrace, indent, Space.Newline); // { |
| 652 | 696 | const new_indent = indent + indent_delta; |
| 653 | 697 | |
| 654 | 698 | var it = err_set_decl.decls.iterator(0); |
| 655 | 699 | while (it.next()) |node| { |
| 656 | 700 | try stream.writeByteNTimes(' ', new_indent); |
| 657 | | try renderTopLevelDecl(allocator, stream, tree, new_indent, node.*); |
| 658 | | if ((node.*).id != ast.Node.Id.LineComment) { |
| 659 | | try stream.write(","); |
| 660 | | } |
| 701 | try renderExpression(allocator, stream, tree, new_indent, node.*, Space.None); |
| 702 | try renderToken(tree, stream, tree.nextToken(node.*.lastToken()), new_indent, Space.Newline); // , |
| 703 | |
| 661 | 704 | if (it.peek()) |next_node| { |
| 662 | | const n = if (nodeLineOffset(tree, node.*, next_node.*) >= 2) u8(2) else u8(1); |
| 663 | | try stream.writeByteNTimes('\n', n); |
| 705 | try renderExtraNewline(tree, stream, next_node.*); |
| 664 | 706 | } |
| 665 | 707 | } |
| 666 | 708 | |
| 667 | | try stream.write("\n"); |
| 668 | 709 | try stream.writeByteNTimes(' ', indent); |
| 669 | | try stream.write("}"); |
| 710 | try renderToken(tree, stream, err_set_decl.rbrace_token, indent, space); // } |
| 711 | }, |
| 712 | |
| 713 | ast.Node.Id.ErrorTag => { |
| 714 | const tag = @fieldParentPtr(ast.Node.ErrorTag, "base", base); |
| 715 | |
| 716 | try renderDocComments(tree, stream, tag, indent); |
| 717 | try renderToken(tree, stream, tag.name_token, indent, space); // name |
| 670 | 718 | }, |
| 671 | 719 | |
| 672 | 720 | ast.Node.Id.MultilineStringLiteral => { |
| 673 | 721 | const multiline_str_literal = @fieldParentPtr(ast.Node.MultilineStringLiteral, "base", base); |
| 674 | | try stream.print("\n"); |
| 722 | |
| 723 | var skip_first_indent = true; |
| 724 | if (tree.tokens.at(multiline_str_literal.firstToken() - 1).id != Token.Id.LineComment) { |
| 725 | try stream.print("\n"); |
| 726 | skip_first_indent = false; |
| 727 | } |
| 675 | 728 | |
| 676 | 729 | var i: usize = 0; |
| 677 | 730 | while (i < multiline_str_literal.lines.len) : (i += 1) { |
| 678 | 731 | const t = multiline_str_literal.lines.at(i).*; |
| 679 | | try stream.writeByteNTimes(' ', indent + indent_delta); |
| 680 | | try stream.print("{}", tree.tokenSlice(t)); |
| 732 | if (!skip_first_indent) { |
| 733 | try stream.writeByteNTimes(' ', indent + indent_delta); |
| 734 | } |
| 735 | try renderToken(tree, stream, t, indent, Space.None); |
| 736 | skip_first_indent = false; |
| 681 | 737 | } |
| 682 | 738 | try stream.writeByteNTimes(' ', indent); |
| 683 | 739 | }, |
| 684 | 740 | ast.Node.Id.UndefinedLiteral => { |
| 685 | 741 | const undefined_literal = @fieldParentPtr(ast.Node.UndefinedLiteral, "base", base); |
| 686 | | try stream.print("{}", tree.tokenSlice(undefined_literal.token)); |
| 742 | try renderToken(tree, stream, undefined_literal.token, indent, space); |
| 687 | 743 | }, |
| 688 | 744 | |
| 689 | 745 | ast.Node.Id.BuiltinCall => { |
| 690 | 746 | const builtin_call = @fieldParentPtr(ast.Node.BuiltinCall, "base", base); |
| 691 | | try stream.print("{}(", tree.tokenSlice(builtin_call.builtin_token)); |
| 747 | |
| 748 | try renderToken(tree, stream, builtin_call.builtin_token, indent, Space.None); // @name |
| 749 | try renderToken(tree, stream, tree.nextToken(builtin_call.builtin_token), indent, Space.None); // ( |
| 692 | 750 | |
| 693 | 751 | var it = builtin_call.params.iterator(0); |
| 694 | 752 | while (it.next()) |param_node| { |
| 695 | | try renderExpression(allocator, stream, tree, indent, param_node.*); |
| 753 | try renderExpression(allocator, stream, tree, indent, param_node.*, Space.None); |
| 754 | |
| 696 | 755 | if (it.peek() != null) { |
| 697 | | try stream.write(", "); |
| 756 | const comma_token = tree.nextToken(param_node.*.lastToken()); |
| 757 | try renderToken(tree, stream, comma_token, indent, Space.Space); // , |
| 698 | 758 | } |
| 699 | 759 | } |
| 700 | | try stream.write(")"); |
| 760 | try renderToken(tree, stream, builtin_call.rparen_token, indent, space); // ) |
| 701 | 761 | }, |
| 702 | 762 | |
| 703 | 763 | ast.Node.Id.FnProto => { |
| ... | ... | @@ -706,108 +766,123 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 706 | 766 | if (fn_proto.visib_token) |visib_token_index| { |
| 707 | 767 | const visib_token = tree.tokens.at(visib_token_index); |
| 708 | 768 | assert(visib_token.id == Token.Id.Keyword_pub or visib_token.id == Token.Id.Keyword_export); |
| 709 | | try stream.print("{} ", tree.tokenSlice(visib_token_index)); |
| 769 | |
| 770 | try renderToken(tree, stream, visib_token_index, indent, Space.Space); // pub |
| 710 | 771 | } |
| 711 | 772 | |
| 712 | 773 | if (fn_proto.extern_export_inline_token) |extern_export_inline_token| { |
| 713 | | try stream.print("{} ", tree.tokenSlice(extern_export_inline_token)); |
| 774 | try renderToken(tree, stream, extern_export_inline_token, indent, Space.Space); // extern/export |
| 714 | 775 | } |
| 715 | 776 | |
| 716 | 777 | if (fn_proto.lib_name) |lib_name| { |
| 717 | | try renderExpression(allocator, stream, tree, indent, lib_name); |
| 718 | | try stream.write(" "); |
| 778 | try renderExpression(allocator, stream, tree, indent, lib_name, Space.Space); |
| 719 | 779 | } |
| 720 | 780 | |
| 721 | 781 | if (fn_proto.cc_token) |cc_token| { |
| 722 | | try stream.print("{} ", tree.tokenSlice(cc_token)); |
| 782 | try renderToken(tree, stream, cc_token, indent, Space.Space); // stdcallcc |
| 723 | 783 | } |
| 724 | 784 | |
| 725 | 785 | if (fn_proto.async_attr) |async_attr| { |
| 726 | | try renderExpression(allocator, stream, tree, indent, &async_attr.base); |
| 727 | | try stream.write(" "); |
| 786 | try renderExpression(allocator, stream, tree, indent, &async_attr.base, Space.Space); |
| 728 | 787 | } |
| 729 | 788 | |
| 730 | | try stream.write("fn"); |
| 731 | | |
| 732 | | if (fn_proto.name_token) |name_token| { |
| 733 | | try stream.print(" {}", tree.tokenSlice(name_token)); |
| 789 | if (fn_proto.name_token) |name_token| blk: { |
| 790 | try renderToken(tree, stream, fn_proto.fn_token, indent, Space.Space); // fn |
| 791 | try renderToken(tree, stream, name_token, indent, Space.None); // name |
| 792 | try renderToken(tree, stream, tree.nextToken(name_token), indent, Space.None); // ( |
| 793 | } else blk: { |
| 794 | try renderToken(tree, stream, fn_proto.fn_token, indent, Space.None); // fn |
| 795 | try renderToken(tree, stream, tree.nextToken(fn_proto.fn_token), indent, Space.None); // ( |
| 734 | 796 | } |
| 735 | 797 | |
| 736 | | try stream.write("("); |
| 737 | | |
| 738 | 798 | var it = fn_proto.params.iterator(0); |
| 739 | 799 | while (it.next()) |param_decl_node| { |
| 740 | 800 | try renderParamDecl(allocator, stream, tree, indent, param_decl_node.*); |
| 741 | 801 | |
| 742 | 802 | if (it.peek() != null) { |
| 743 | | try stream.write(", "); |
| 803 | const comma = tree.nextToken(param_decl_node.*.lastToken()); |
| 804 | try renderToken(tree, stream, comma, indent, Space.Space); // , |
| 744 | 805 | } |
| 745 | 806 | } |
| 746 | 807 | |
| 747 | | try stream.write(") "); |
| 808 | const rparen = tree.prevToken(switch (fn_proto.return_type) { |
| 809 | ast.Node.FnProto.ReturnType.Explicit => |node| node.firstToken(), |
| 810 | ast.Node.FnProto.ReturnType.InferErrorSet => |node| tree.prevToken(node.firstToken()), |
| 811 | }); |
| 812 | try renderToken(tree, stream, rparen, indent, Space.Space); // ) |
| 748 | 813 | |
| 749 | 814 | if (fn_proto.align_expr) |align_expr| { |
| 750 | | try stream.write("align("); |
| 751 | | try renderExpression(allocator, stream, tree, indent, align_expr); |
| 752 | | try stream.write(") "); |
| 815 | const align_rparen = tree.nextToken(align_expr.lastToken()); |
| 816 | const align_lparen = tree.prevToken(align_expr.firstToken()); |
| 817 | const align_kw = tree.prevToken(align_lparen); |
| 818 | |
| 819 | try renderToken(tree, stream, align_kw, indent, Space.None); // align |
| 820 | try renderToken(tree, stream, align_lparen, indent, Space.None); // ( |
| 821 | try renderExpression(allocator, stream, tree, indent, align_expr, Space.None); |
| 822 | try renderToken(tree, stream, align_rparen, indent, Space.Space); // ) |
| 753 | 823 | } |
| 754 | 824 | |
| 755 | 825 | switch (fn_proto.return_type) { |
| 756 | 826 | ast.Node.FnProto.ReturnType.Explicit => |node| { |
| 757 | | try renderExpression(allocator, stream, tree, indent, node); |
| 827 | try renderExpression(allocator, stream, tree, indent, node, space); |
| 758 | 828 | }, |
| 759 | 829 | ast.Node.FnProto.ReturnType.InferErrorSet => |node| { |
| 760 | | try stream.write("!"); |
| 761 | | try renderExpression(allocator, stream, tree, indent, node); |
| 830 | try renderToken(tree, stream, tree.prevToken(node.firstToken()), indent, Space.None); // ! |
| 831 | try renderExpression(allocator, stream, tree, indent, node, space); |
| 762 | 832 | }, |
| 763 | 833 | } |
| 764 | 834 | }, |
| 765 | 835 | |
| 766 | 836 | ast.Node.Id.PromiseType => { |
| 767 | 837 | const promise_type = @fieldParentPtr(ast.Node.PromiseType, "base", base); |
| 768 | | try stream.write(tree.tokenSlice(promise_type.promise_token)); |
| 838 | |
| 769 | 839 | if (promise_type.result) |result| { |
| 770 | | try stream.write(tree.tokenSlice(result.arrow_token)); |
| 771 | | try renderExpression(allocator, stream, tree, indent, result.return_type); |
| 840 | try renderToken(tree, stream, promise_type.promise_token, indent, Space.None); // promise |
| 841 | try renderToken(tree, stream, result.arrow_token, indent, Space.None); // -> |
| 842 | try renderExpression(allocator, stream, tree, indent, result.return_type, space); |
| 843 | } else { |
| 844 | try renderToken(tree, stream, promise_type.promise_token, indent, space); // promise |
| 772 | 845 | } |
| 773 | 846 | }, |
| 774 | 847 | |
| 775 | | ast.Node.Id.LineComment => { |
| 776 | | const line_comment_node = @fieldParentPtr(ast.Node.LineComment, "base", base); |
| 777 | | try stream.write(tree.tokenSlice(line_comment_node.token)); |
| 778 | | }, |
| 779 | | |
| 780 | 848 | ast.Node.Id.DocComment => unreachable, // doc comments are attached to nodes |
| 781 | 849 | |
| 782 | 850 | ast.Node.Id.Switch => { |
| 783 | 851 | const switch_node = @fieldParentPtr(ast.Node.Switch, "base", base); |
| 784 | 852 | |
| 785 | | try stream.print("{} (", tree.tokenSlice(switch_node.switch_token)); |
| 853 | try renderToken(tree, stream, switch_node.switch_token, indent, Space.Space); // switch |
| 854 | try renderToken(tree, stream, tree.nextToken(switch_node.switch_token), indent, Space.None); // ( |
| 855 | |
| 856 | const rparen = tree.nextToken(switch_node.expr.lastToken()); |
| 857 | const lbrace = tree.nextToken(rparen); |
| 858 | |
| 786 | 859 | if (switch_node.cases.len == 0) { |
| 787 | | try renderExpression(allocator, stream, tree, indent, switch_node.expr); |
| 788 | | try stream.write(") {}"); |
| 860 | try renderExpression(allocator, stream, tree, indent, switch_node.expr, Space.None); |
| 861 | try renderToken(tree, stream, rparen, indent, Space.Space); // ) |
| 862 | try renderToken(tree, stream, lbrace, indent, Space.None); // { |
| 863 | try renderToken(tree, stream, switch_node.rbrace, indent, space); // } |
| 789 | 864 | return; |
| 790 | 865 | } |
| 791 | 866 | |
| 792 | | try renderExpression(allocator, stream, tree, indent, switch_node.expr); |
| 793 | | try stream.write(") {\n"); |
| 867 | try renderExpression(allocator, stream, tree, indent, switch_node.expr, Space.None); |
| 868 | |
| 869 | try renderToken(tree, stream, rparen, indent, Space.Space); // ) |
| 870 | try renderToken(tree, stream, lbrace, indent, Space.Newline); // { |
| 794 | 871 | |
| 795 | 872 | const new_indent = indent + indent_delta; |
| 796 | 873 | |
| 797 | 874 | var it = switch_node.cases.iterator(0); |
| 798 | 875 | while (it.next()) |node| { |
| 799 | 876 | try stream.writeByteNTimes(' ', new_indent); |
| 800 | | try renderExpression(allocator, stream, tree, new_indent, node.*); |
| 877 | try renderExpression(allocator, stream, tree, new_indent, node.*, Space.Newline); |
| 801 | 878 | |
| 802 | 879 | if (it.peek()) |next_node| { |
| 803 | | const n = if (nodeLineOffset(tree, node.*, next_node.*) >= 2) u8(2) else u8(1); |
| 804 | | try stream.writeByteNTimes('\n', n); |
| 880 | try renderExtraNewline(tree, stream, next_node.*); |
| 805 | 881 | } |
| 806 | 882 | } |
| 807 | 883 | |
| 808 | | try stream.write("\n"); |
| 809 | 884 | try stream.writeByteNTimes(' ', indent); |
| 810 | | try stream.write("}"); |
| 885 | try renderToken(tree, stream, switch_node.rbrace, indent, space); // } |
| 811 | 886 | }, |
| 812 | 887 | |
| 813 | 888 | ast.Node.Id.SwitchCase => { |
| ... | ... | @@ -815,54 +890,50 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 815 | 890 | |
| 816 | 891 | var it = switch_case.items.iterator(0); |
| 817 | 892 | while (it.next()) |node| { |
| 818 | | try renderExpression(allocator, stream, tree, indent, node.*); |
| 893 | if (it.peek()) |next_node| { |
| 894 | try renderExpression(allocator, stream, tree, indent, node.*, Space.None); |
| 819 | 895 | |
| 820 | | if (it.peek() != null) { |
| 821 | | try stream.write(",\n"); |
| 896 | const comma_token = tree.nextToken(node.*.lastToken()); |
| 897 | try renderToken(tree, stream, comma_token, indent, Space.Newline); // , |
| 898 | try renderExtraNewline(tree, stream, next_node.*); |
| 822 | 899 | try stream.writeByteNTimes(' ', indent); |
| 900 | } else { |
| 901 | try renderExpression(allocator, stream, tree, indent, node.*, Space.Space); |
| 823 | 902 | } |
| 824 | 903 | } |
| 825 | 904 | |
| 826 | | try stream.write(" => "); |
| 905 | try renderToken(tree, stream, switch_case.arrow_token, indent, Space.Space); // => |
| 827 | 906 | |
| 828 | 907 | if (switch_case.payload) |payload| { |
| 829 | | try renderExpression(allocator, stream, tree, indent, payload); |
| 830 | | try stream.write(" "); |
| 908 | try renderExpression(allocator, stream, tree, indent, payload, Space.Space); |
| 831 | 909 | } |
| 832 | 910 | |
| 833 | | try renderExpression(allocator, stream, tree, indent, switch_case.expr); |
| 834 | | { |
| 835 | | // Handle missing comma after last switch case |
| 836 | | var index = switch_case.lastToken() + 1; |
| 837 | | switch (tree.tokens.at(index).id) { |
| 838 | | Token.Id.RBrace => { |
| 839 | | try stream.write(","); |
| 840 | | }, |
| 841 | | Token.Id.LineComment => { |
| 842 | | try stream.write(", "); |
| 843 | | try renderToken(tree, stream, index, indent, true, true); |
| 844 | | }, |
| 845 | | else => try renderToken(tree, stream, index, indent, true, true), |
| 846 | | } |
| 911 | // add a trailing comma if necessary |
| 912 | const end_token = switch_case.lastToken() + 1; |
| 913 | switch (tree.tokens.at(end_token).id) { |
| 914 | Token.Id.Comma => { |
| 915 | try renderExpression(allocator, stream, tree, indent, switch_case.expr, Space.None); |
| 916 | try renderToken(tree, stream, end_token, indent, space); // , |
| 917 | }, |
| 918 | Token.Id.LineComment => { |
| 919 | try renderExpression(allocator, stream, tree, indent, switch_case.expr, Space.NoComment); |
| 920 | try stream.write(", "); |
| 921 | try renderToken(tree, stream, end_token, indent, space); |
| 922 | }, |
| 923 | else => { |
| 924 | try renderExpression(allocator, stream, tree, indent, switch_case.expr, Space.None); |
| 925 | try stream.write(",\n"); |
| 926 | assert(space == Space.Newline); |
| 927 | }, |
| 847 | 928 | } |
| 848 | 929 | }, |
| 849 | 930 | ast.Node.Id.SwitchElse => { |
| 850 | 931 | const switch_else = @fieldParentPtr(ast.Node.SwitchElse, "base", base); |
| 851 | | try stream.print("{}", tree.tokenSlice(switch_else.token)); |
| 932 | try renderToken(tree, stream, switch_else.token, indent, space); |
| 852 | 933 | }, |
| 853 | 934 | ast.Node.Id.Else => { |
| 854 | 935 | const else_node = @fieldParentPtr(ast.Node.Else, "base", base); |
| 855 | 936 | |
| 856 | | var prev_tok_index = else_node.else_token - 1; |
| 857 | | while (tree.tokens.at(prev_tok_index).id == Token.Id.LineComment) : (prev_tok_index -= 1) { } |
| 858 | | prev_tok_index += 1; |
| 859 | | while (prev_tok_index < else_node.else_token) : (prev_tok_index += 1) { |
| 860 | | try stream.print("{}\n", tree.tokenSlice(prev_tok_index)); |
| 861 | | try stream.writeByteNTimes(' ', indent); |
| 862 | | } |
| 863 | | |
| 864 | | try stream.print("{}", tree.tokenSlice(else_node.else_token)); |
| 865 | | |
| 866 | 937 | const block_body = switch (else_node.body.id) { |
| 867 | 938 | ast.Node.Id.Block, |
| 868 | 939 | ast.Node.Id.If, |
| ... | ... | @@ -872,21 +943,19 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 872 | 943 | else => false, |
| 873 | 944 | }; |
| 874 | 945 | |
| 875 | | if (block_body) { |
| 876 | | try stream.write(" "); |
| 877 | | } |
| 946 | const after_else_space = if (block_body or else_node.payload != null) Space.Space else Space.Newline; |
| 947 | try renderToken(tree, stream, else_node.else_token, indent, after_else_space); |
| 878 | 948 | |
| 879 | 949 | if (else_node.payload) |payload| { |
| 880 | | try renderExpression(allocator, stream, tree, indent, payload); |
| 881 | | try stream.write(" "); |
| 950 | const payload_space = if (block_body) Space.Space else Space.Newline; |
| 951 | try renderExpression(allocator, stream, tree, indent, payload, Space.Space); |
| 882 | 952 | } |
| 883 | 953 | |
| 884 | 954 | if (block_body) { |
| 885 | | try renderExpression(allocator, stream, tree, indent, else_node.body); |
| 955 | try renderExpression(allocator, stream, tree, indent, else_node.body, space); |
| 886 | 956 | } else { |
| 887 | | try stream.write("\n"); |
| 888 | 957 | try stream.writeByteNTimes(' ', indent + indent_delta); |
| 889 | | try renderExpression(allocator, stream, tree, indent, else_node.body); |
| 958 | try renderExpression(allocator, stream, tree, indent, else_node.body, space); |
| 890 | 959 | } |
| 891 | 960 | }, |
| 892 | 961 | |
| ... | ... | @@ -894,103 +963,134 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 894 | 963 | const while_node = @fieldParentPtr(ast.Node.While, "base", base); |
| 895 | 964 | |
| 896 | 965 | if (while_node.label) |label| { |
| 897 | | try stream.print("{}: ", tree.tokenSlice(label)); |
| 966 | try renderToken(tree, stream, label, indent, Space.None); // label |
| 967 | try renderToken(tree, stream, tree.nextToken(label), indent, Space.Space); // : |
| 898 | 968 | } |
| 899 | 969 | |
| 900 | 970 | if (while_node.inline_token) |inline_token| { |
| 901 | | try stream.print("{} ", tree.tokenSlice(inline_token)); |
| 971 | try renderToken(tree, stream, inline_token, indent, Space.Space); // inline |
| 902 | 972 | } |
| 903 | 973 | |
| 904 | | try stream.print("{} (", tree.tokenSlice(while_node.while_token)); |
| 905 | | try renderExpression(allocator, stream, tree, indent, while_node.condition); |
| 906 | | try stream.write(")"); |
| 974 | try renderToken(tree, stream, while_node.while_token, indent, Space.Space); // while |
| 975 | try renderToken(tree, stream, tree.nextToken(while_node.while_token), indent, Space.None); // ( |
| 976 | try renderExpression(allocator, stream, tree, indent, while_node.condition, Space.None); |
| 977 | |
| 978 | { |
| 979 | const rparen = tree.nextToken(while_node.condition.lastToken()); |
| 980 | const rparen_space = if (while_node.payload != null or while_node.continue_expr != null or |
| 981 | while_node.body.id == ast.Node.Id.Block) Space.Space else Space.Newline; |
| 982 | try renderToken(tree, stream, rparen, indent, rparen_space); // ) |
| 983 | } |
| 907 | 984 | |
| 908 | 985 | if (while_node.payload) |payload| { |
| 909 | | try stream.write(" "); |
| 910 | | try renderExpression(allocator, stream, tree, indent, payload); |
| 986 | try renderExpression(allocator, stream, tree, indent, payload, Space.Space); |
| 911 | 987 | } |
| 912 | 988 | |
| 913 | 989 | if (while_node.continue_expr) |continue_expr| { |
| 914 | | try stream.write(" : ("); |
| 915 | | try renderExpression(allocator, stream, tree, indent, continue_expr); |
| 916 | | try stream.write(")"); |
| 990 | const rparen = tree.nextToken(continue_expr.lastToken()); |
| 991 | const lparen = tree.prevToken(continue_expr.firstToken()); |
| 992 | const colon = tree.prevToken(lparen); |
| 993 | |
| 994 | try renderToken(tree, stream, colon, indent, Space.Space); // : |
| 995 | try renderToken(tree, stream, lparen, indent, Space.None); // ( |
| 996 | |
| 997 | try renderExpression(allocator, stream, tree, indent, continue_expr, Space.None); |
| 998 | |
| 999 | const rparen_space = if (while_node.body.id == ast.Node.Id.Block) Space.Space else Space.Newline; |
| 1000 | try renderToken(tree, stream, rparen, indent, rparen_space); // ) |
| 917 | 1001 | } |
| 918 | 1002 | |
| 1003 | const body_space = blk: { |
| 1004 | if (while_node.@"else" != null) { |
| 1005 | break :blk if (while_node.body.id == ast.Node.Id.Block) Space.Space else Space.Newline; |
| 1006 | } else { |
| 1007 | break :blk space; |
| 1008 | } |
| 1009 | }; |
| 1010 | |
| 919 | 1011 | if (while_node.body.id == ast.Node.Id.Block) { |
| 920 | | try stream.write(" "); |
| 921 | | try renderExpression(allocator, stream, tree, indent, while_node.body); |
| 1012 | try renderExpression(allocator, stream, tree, indent, while_node.body, body_space); |
| 922 | 1013 | } else { |
| 923 | | try stream.write("\n"); |
| 924 | 1014 | try stream.writeByteNTimes(' ', indent + indent_delta); |
| 925 | | try renderExpression(allocator, stream, tree, indent, while_node.body); |
| 1015 | try renderExpression(allocator, stream, tree, indent, while_node.body, body_space); |
| 926 | 1016 | } |
| 927 | 1017 | |
| 928 | 1018 | if (while_node.@"else") |@"else"| { |
| 929 | 1019 | if (while_node.body.id == ast.Node.Id.Block) { |
| 930 | | try stream.write(" "); |
| 931 | 1020 | } else { |
| 932 | | try stream.write("\n"); |
| 933 | 1021 | try stream.writeByteNTimes(' ', indent); |
| 934 | 1022 | } |
| 935 | 1023 | |
| 936 | | try renderExpression(allocator, stream, tree, indent, &@"else".base); |
| 1024 | try renderExpression(allocator, stream, tree, indent, &@"else".base, space); |
| 937 | 1025 | } |
| 938 | 1026 | }, |
| 939 | 1027 | |
| 940 | 1028 | ast.Node.Id.For => { |
| 941 | 1029 | const for_node = @fieldParentPtr(ast.Node.For, "base", base); |
| 1030 | |
| 942 | 1031 | if (for_node.label) |label| { |
| 943 | | try stream.print("{}: ", tree.tokenSlice(label)); |
| 1032 | try renderToken(tree, stream, label, indent, Space.None); // label |
| 1033 | try renderToken(tree, stream, tree.nextToken(label), indent, Space.Space); // : |
| 944 | 1034 | } |
| 945 | 1035 | |
| 946 | 1036 | if (for_node.inline_token) |inline_token| { |
| 947 | | try stream.print("{} ", tree.tokenSlice(inline_token)); |
| 1037 | try renderToken(tree, stream, inline_token, indent, Space.Space); // inline |
| 948 | 1038 | } |
| 949 | 1039 | |
| 950 | | try stream.print("{} (", tree.tokenSlice(for_node.for_token)); |
| 951 | | try renderExpression(allocator, stream, tree, indent, for_node.array_expr); |
| 952 | | try stream.write(")"); |
| 1040 | try renderToken(tree, stream, for_node.for_token, indent, Space.Space); // for |
| 1041 | try renderToken(tree, stream, tree.nextToken(for_node.for_token), indent, Space.None); // ( |
| 1042 | try renderExpression(allocator, stream, tree, indent, for_node.array_expr, Space.None); |
| 1043 | |
| 1044 | const rparen = tree.nextToken(for_node.array_expr.lastToken()); |
| 1045 | const rparen_space = if (for_node.payload != null or |
| 1046 | for_node.body.id == ast.Node.Id.Block) Space.Space else Space.Newline; |
| 1047 | try renderToken(tree, stream, rparen, indent, rparen_space); // ) |
| 953 | 1048 | |
| 954 | 1049 | if (for_node.payload) |payload| { |
| 955 | | try stream.write(" "); |
| 956 | | try renderExpression(allocator, stream, tree, indent, payload); |
| 1050 | const payload_space = if (for_node.body.id == ast.Node.Id.Block) Space.Space else Space.Newline; |
| 1051 | try renderExpression(allocator, stream, tree, indent, payload, payload_space); |
| 957 | 1052 | } |
| 958 | 1053 | |
| 1054 | const body_space = blk: { |
| 1055 | if (for_node.@"else" != null) { |
| 1056 | if (for_node.body.id == ast.Node.Id.Block) { |
| 1057 | break :blk Space.Space; |
| 1058 | } else { |
| 1059 | break :blk Space.Newline; |
| 1060 | } |
| 1061 | } else { |
| 1062 | break :blk space; |
| 1063 | } |
| 1064 | }; |
| 959 | 1065 | if (for_node.body.id == ast.Node.Id.Block) { |
| 960 | | try stream.write(" "); |
| 961 | | try renderExpression(allocator, stream, tree, indent, for_node.body); |
| 1066 | try renderExpression(allocator, stream, tree, indent, for_node.body, body_space); |
| 962 | 1067 | } else { |
| 963 | | try stream.write("\n"); |
| 964 | 1068 | try stream.writeByteNTimes(' ', indent + indent_delta); |
| 965 | | try renderExpression(allocator, stream, tree, indent, for_node.body); |
| 1069 | try renderExpression(allocator, stream, tree, indent, for_node.body, body_space); |
| 966 | 1070 | } |
| 967 | 1071 | |
| 968 | 1072 | if (for_node.@"else") |@"else"| { |
| 969 | | if (for_node.body.id == ast.Node.Id.Block) { |
| 970 | | try stream.write(" "); |
| 971 | | } else { |
| 972 | | try stream.write("\n"); |
| 1073 | if (for_node.body.id != ast.Node.Id.Block) { |
| 973 | 1074 | try stream.writeByteNTimes(' ', indent); |
| 974 | 1075 | } |
| 975 | 1076 | |
| 976 | | try renderExpression(allocator, stream, tree, indent, &@"else".base); |
| 1077 | try renderExpression(allocator, stream, tree, indent, &@"else".base, space); |
| 977 | 1078 | } |
| 978 | 1079 | }, |
| 979 | 1080 | |
| 980 | 1081 | ast.Node.Id.If => { |
| 981 | 1082 | const if_node = @fieldParentPtr(ast.Node.If, "base", base); |
| 982 | | try stream.print("{} (", tree.tokenSlice(if_node.if_token)); |
| 983 | 1083 | |
| 984 | | try renderExpression(allocator, stream, tree, indent, if_node.condition); |
| 985 | | try renderToken(tree, stream, if_node.condition.lastToken() + 1, indent, false, true); |
| 1084 | try renderToken(tree, stream, if_node.if_token, indent, Space.Space); |
| 1085 | try renderToken(tree, stream, tree.prevToken(if_node.condition.firstToken()), indent, Space.None); |
| 1086 | |
| 1087 | try renderExpression(allocator, stream, tree, indent, if_node.condition, Space.None); |
| 1088 | try renderToken(tree, stream, tree.nextToken(if_node.condition.lastToken()), indent, Space.Space); |
| 986 | 1089 | |
| 987 | 1090 | if (if_node.payload) |payload| { |
| 988 | | try renderExpression(allocator, stream, tree, indent, payload); |
| 989 | | try stream.write(" "); |
| 1091 | try renderExpression(allocator, stream, tree, indent, payload, Space.Space); |
| 990 | 1092 | } |
| 991 | 1093 | |
| 992 | | try renderExpression(allocator, stream, tree, indent, if_node.body); |
| 993 | | |
| 994 | 1094 | switch (if_node.body.id) { |
| 995 | 1095 | ast.Node.Id.Block, |
| 996 | 1096 | ast.Node.Id.If, |
| ... | ... | @@ -999,25 +1099,29 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 999 | 1099 | ast.Node.Id.Switch => { |
| 1000 | 1100 | if (if_node.@"else") |@"else"| { |
| 1001 | 1101 | if (if_node.body.id == ast.Node.Id.Block) { |
| 1002 | | try stream.write(" "); |
| 1102 | try renderExpression(allocator, stream, tree, indent, if_node.body, Space.Space); |
| 1003 | 1103 | } else { |
| 1004 | | try stream.write("\n"); |
| 1104 | try renderExpression(allocator, stream, tree, indent, if_node.body, Space.Newline); |
| 1005 | 1105 | try stream.writeByteNTimes(' ', indent); |
| 1006 | 1106 | } |
| 1007 | 1107 | |
| 1008 | | try renderExpression(allocator, stream, tree, indent, &@"else".base); |
| 1108 | try renderExpression(allocator, stream, tree, indent, &@"else".base, space); |
| 1109 | } else { |
| 1110 | try renderExpression(allocator, stream, tree, indent, if_node.body, space); |
| 1009 | 1111 | } |
| 1010 | 1112 | }, |
| 1011 | 1113 | else => { |
| 1012 | 1114 | if (if_node.@"else") |@"else"| { |
| 1013 | | try stream.print(" {} ", tree.tokenSlice(@"else".else_token)); |
| 1115 | try renderExpression(allocator, stream, tree, indent, if_node.body, Space.Space); |
| 1116 | try renderToken(tree, stream, @"else".else_token, indent, Space.Space); |
| 1014 | 1117 | |
| 1015 | 1118 | if (@"else".payload) |payload| { |
| 1016 | | try renderExpression(allocator, stream, tree, indent, payload); |
| 1017 | | try stream.write(" "); |
| 1119 | try renderExpression(allocator, stream, tree, indent, payload, Space.Space); |
| 1018 | 1120 | } |
| 1019 | 1121 | |
| 1020 | | try renderExpression(allocator, stream, tree, indent, @"else".body); |
| 1122 | try renderExpression(allocator, stream, tree, indent, @"else".body, space); |
| 1123 | } else { |
| 1124 | try renderExpression(allocator, stream, tree, indent, if_node.body, space); |
| 1021 | 1125 | } |
| 1022 | 1126 | }, |
| 1023 | 1127 | } |
| ... | ... | @@ -1025,15 +1129,17 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 1025 | 1129 | |
| 1026 | 1130 | ast.Node.Id.Asm => { |
| 1027 | 1131 | const asm_node = @fieldParentPtr(ast.Node.Asm, "base", base); |
| 1028 | | try stream.print("{} ", tree.tokenSlice(asm_node.asm_token)); |
| 1132 | |
| 1133 | try renderToken(tree, stream, asm_node.asm_token, indent, Space.Space); // asm |
| 1029 | 1134 | |
| 1030 | 1135 | if (asm_node.volatile_token) |volatile_token| { |
| 1031 | | try stream.print("{} ", tree.tokenSlice(volatile_token)); |
| 1136 | try renderToken(tree, stream, volatile_token, indent, Space.Space); // volatile |
| 1137 | try renderToken(tree, stream, tree.nextToken(volatile_token), indent, Space.None); // ( |
| 1138 | } else { |
| 1139 | try renderToken(tree, stream, tree.nextToken(asm_node.asm_token), indent, Space.None); // ( |
| 1032 | 1140 | } |
| 1033 | 1141 | |
| 1034 | | try stream.print("("); |
| 1035 | | try renderExpression(allocator, stream, tree, indent, asm_node.template); |
| 1036 | | try stream.print("\n"); |
| 1142 | try renderExpression(allocator, stream, tree, indent, asm_node.template, Space.Newline); |
| 1037 | 1143 | const indent_once = indent + indent_delta; |
| 1038 | 1144 | try stream.writeByteNTimes(' ', indent_once); |
| 1039 | 1145 | try stream.print(": "); |
| ... | ... | @@ -1043,13 +1149,15 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 1043 | 1149 | var it = asm_node.outputs.iterator(0); |
| 1044 | 1150 | while (it.next()) |asm_output| { |
| 1045 | 1151 | const node = &(asm_output.*).base; |
| 1046 | | try renderExpression(allocator, stream, tree, indent_extra, node); |
| 1152 | try renderExpression(allocator, stream, tree, indent_extra, node, Space.None); |
| 1047 | 1153 | |
| 1048 | 1154 | if (it.peek()) |next_asm_output| { |
| 1049 | 1155 | const next_node = &(next_asm_output.*).base; |
| 1050 | | const n = if (nodeLineOffset(tree, node, next_node) >= 2) u8(2) else u8(1); |
| 1051 | | try stream.writeByte(','); |
| 1052 | | try stream.writeByteNTimes('\n', n); |
| 1156 | |
| 1157 | const comma = tree.prevToken(next_asm_output.*.firstToken()); |
| 1158 | try renderToken(tree, stream, comma, indent_extra, Space.Newline); // , |
| 1159 | try renderExtraNewline(tree, stream, next_node); |
| 1160 | |
| 1053 | 1161 | try stream.writeByteNTimes(' ', indent_extra); |
| 1054 | 1162 | } |
| 1055 | 1163 | } |
| ... | ... | @@ -1063,13 +1171,15 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 1063 | 1171 | var it = asm_node.inputs.iterator(0); |
| 1064 | 1172 | while (it.next()) |asm_input| { |
| 1065 | 1173 | const node = &(asm_input.*).base; |
| 1066 | | try renderExpression(allocator, stream, tree, indent_extra, node); |
| 1174 | try renderExpression(allocator, stream, tree, indent_extra, node, Space.None); |
| 1067 | 1175 | |
| 1068 | 1176 | if (it.peek()) |next_asm_input| { |
| 1069 | 1177 | const next_node = &(next_asm_input.*).base; |
| 1070 | | const n = if (nodeLineOffset(tree, node, next_node) >= 2) u8(2) else u8(1); |
| 1071 | | try stream.writeByte(','); |
| 1072 | | try stream.writeByteNTimes('\n', n); |
| 1178 | |
| 1179 | const comma = tree.prevToken(next_asm_input.*.firstToken()); |
| 1180 | try renderToken(tree, stream, comma, indent_extra, Space.Newline); // , |
| 1181 | try renderExtraNewline(tree, stream, next_node); |
| 1182 | |
| 1073 | 1183 | try stream.writeByteNTimes(' ', indent_extra); |
| 1074 | 1184 | } |
| 1075 | 1185 | } |
| ... | ... | @@ -1082,7 +1192,7 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 1082 | 1192 | { |
| 1083 | 1193 | var it = asm_node.clobbers.iterator(0); |
| 1084 | 1194 | while (it.next()) |node| { |
| 1085 | | try renderExpression(allocator, stream, tree, indent_once, node.*); |
| 1195 | try renderExpression(allocator, stream, tree, indent_once, node.*, Space.None); |
| 1086 | 1196 | |
| 1087 | 1197 | if (it.peek() != null) { |
| 1088 | 1198 | try stream.write(", "); |
| ... | ... | @@ -1090,47 +1200,46 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 1090 | 1200 | } |
| 1091 | 1201 | } |
| 1092 | 1202 | |
| 1093 | | try stream.write(")"); |
| 1203 | try renderToken(tree, stream, asm_node.rparen, indent, space); |
| 1094 | 1204 | }, |
| 1095 | 1205 | |
| 1096 | 1206 | ast.Node.Id.AsmInput => { |
| 1097 | 1207 | const asm_input = @fieldParentPtr(ast.Node.AsmInput, "base", base); |
| 1098 | 1208 | |
| 1099 | 1209 | try stream.write("["); |
| 1100 | | try renderExpression(allocator, stream, tree, indent, asm_input.symbolic_name); |
| 1210 | try renderExpression(allocator, stream, tree, indent, asm_input.symbolic_name, Space.None); |
| 1101 | 1211 | try stream.write("] "); |
| 1102 | | try renderExpression(allocator, stream, tree, indent, asm_input.constraint); |
| 1212 | try renderExpression(allocator, stream, tree, indent, asm_input.constraint, Space.None); |
| 1103 | 1213 | try stream.write(" ("); |
| 1104 | | try renderExpression(allocator, stream, tree, indent, asm_input.expr); |
| 1105 | | try stream.write(")"); |
| 1214 | try renderExpression(allocator, stream, tree, indent, asm_input.expr, Space.None); |
| 1215 | try renderToken(tree, stream, asm_input.lastToken(), indent, space); // ) |
| 1106 | 1216 | }, |
| 1107 | 1217 | |
| 1108 | 1218 | ast.Node.Id.AsmOutput => { |
| 1109 | 1219 | const asm_output = @fieldParentPtr(ast.Node.AsmOutput, "base", base); |
| 1110 | 1220 | |
| 1111 | 1221 | try stream.write("["); |
| 1112 | | try renderExpression(allocator, stream, tree, indent, asm_output.symbolic_name); |
| 1222 | try renderExpression(allocator, stream, tree, indent, asm_output.symbolic_name, Space.None); |
| 1113 | 1223 | try stream.write("] "); |
| 1114 | | try renderExpression(allocator, stream, tree, indent, asm_output.constraint); |
| 1224 | try renderExpression(allocator, stream, tree, indent, asm_output.constraint, Space.None); |
| 1115 | 1225 | try stream.write(" ("); |
| 1116 | 1226 | |
| 1117 | 1227 | switch (asm_output.kind) { |
| 1118 | 1228 | ast.Node.AsmOutput.Kind.Variable => |variable_name| { |
| 1119 | | try renderExpression(allocator, stream, tree, indent, &variable_name.base); |
| 1229 | try renderExpression(allocator, stream, tree, indent, &variable_name.base, Space.None); |
| 1120 | 1230 | }, |
| 1121 | 1231 | ast.Node.AsmOutput.Kind.Return => |return_type| { |
| 1122 | 1232 | try stream.write("-> "); |
| 1123 | | try renderExpression(allocator, stream, tree, indent, return_type); |
| 1233 | try renderExpression(allocator, stream, tree, indent, return_type, Space.None); |
| 1124 | 1234 | }, |
| 1125 | 1235 | } |
| 1126 | 1236 | |
| 1127 | | try stream.write(")"); |
| 1237 | try renderToken(tree, stream, asm_output.lastToken(), indent, space); // ) |
| 1128 | 1238 | }, |
| 1129 | 1239 | |
| 1130 | 1240 | ast.Node.Id.StructField, |
| 1131 | 1241 | ast.Node.Id.UnionTag, |
| 1132 | 1242 | ast.Node.Id.EnumTag, |
| 1133 | | ast.Node.Id.ErrorTag, |
| 1134 | 1243 | ast.Node.Id.Root, |
| 1135 | 1244 | ast.Node.Id.VarDecl, |
| 1136 | 1245 | ast.Node.Id.Use, |
| ... | ... | @@ -1139,70 +1248,74 @@ fn renderExpression(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, ind |
| 1139 | 1248 | } |
| 1140 | 1249 | } |
| 1141 | 1250 | |
| 1142 | | fn renderVarDecl(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, indent: usize, var_decl: &ast.Node.VarDecl) (@typeOf(stream).Child.Error || Error)!void { |
| 1251 | fn renderVarDecl(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, indent: usize, |
| 1252 | var_decl: &ast.Node.VarDecl) (@typeOf(stream).Child.Error || Error)!void |
| 1253 | { |
| 1143 | 1254 | if (var_decl.visib_token) |visib_token| { |
| 1144 | | try renderToken(tree, stream, visib_token, indent, false, true); |
| 1255 | try renderToken(tree, stream, visib_token, indent, Space.Space); // pub |
| 1145 | 1256 | } |
| 1146 | 1257 | |
| 1147 | 1258 | if (var_decl.extern_export_token) |extern_export_token| { |
| 1148 | | try renderToken(tree, stream, extern_export_token, indent, false, true); |
| 1259 | try renderToken(tree, stream, extern_export_token, indent, Space.Space); // extern |
| 1149 | 1260 | |
| 1150 | 1261 | if (var_decl.lib_name) |lib_name| { |
| 1151 | | try renderExpression(allocator, stream, tree, indent, lib_name); |
| 1152 | | try stream.write(" "); |
| 1262 | try renderExpression(allocator, stream, tree, indent, lib_name, Space.Space); // "lib" |
| 1153 | 1263 | } |
| 1154 | 1264 | } |
| 1155 | 1265 | |
| 1156 | 1266 | if (var_decl.comptime_token) |comptime_token| { |
| 1157 | | try renderToken(tree, stream, comptime_token, indent, false, true); |
| 1267 | try renderToken(tree, stream, comptime_token, indent, Space.Space); // comptime |
| 1158 | 1268 | } |
| 1159 | 1269 | |
| 1160 | | try renderToken(tree, stream, var_decl.mut_token, indent, false, true); |
| 1161 | | try renderToken(tree, stream, var_decl.name_token, indent, false, false); |
| 1270 | try renderToken(tree, stream, var_decl.mut_token, indent, Space.Space); // var |
| 1271 | |
| 1272 | const name_space = if (var_decl.type_node == null and (var_decl.align_node != null or |
| 1273 | var_decl.init_node != null)) Space.Space else Space.None; |
| 1274 | try renderToken(tree, stream, var_decl.name_token, indent, name_space); |
| 1162 | 1275 | |
| 1163 | 1276 | if (var_decl.type_node) |type_node| { |
| 1164 | | try stream.write(": "); |
| 1165 | | try renderExpression(allocator, stream, tree, indent, type_node); |
| 1277 | try renderToken(tree, stream, tree.nextToken(var_decl.name_token), indent, Space.Space); |
| 1278 | const s = if (var_decl.align_node != null or var_decl.init_node != null) Space.Space else Space.None; |
| 1279 | try renderExpression(allocator, stream, tree, indent, type_node, s); |
| 1166 | 1280 | } |
| 1167 | 1281 | |
| 1168 | 1282 | if (var_decl.align_node) |align_node| { |
| 1169 | | try stream.write(" align("); |
| 1170 | | try renderExpression(allocator, stream, tree, indent, align_node); |
| 1171 | | try stream.write(")"); |
| 1283 | const lparen = tree.prevToken(align_node.firstToken()); |
| 1284 | const align_kw = tree.prevToken(lparen); |
| 1285 | const rparen = tree.nextToken(align_node.lastToken()); |
| 1286 | try renderToken(tree, stream, align_kw, indent, Space.None); // align |
| 1287 | try renderToken(tree, stream, lparen, indent, Space.None); // ( |
| 1288 | try renderExpression(allocator, stream, tree, indent, align_node, Space.None); |
| 1289 | const s = if (var_decl.init_node != null) Space.Space else Space.None; |
| 1290 | try renderToken(tree, stream, rparen, indent, s); // ) |
| 1172 | 1291 | } |
| 1173 | 1292 | |
| 1174 | 1293 | if (var_decl.init_node) |init_node| { |
| 1175 | | const text = if (init_node.id == ast.Node.Id.MultilineStringLiteral) " =" else " = "; |
| 1176 | | try stream.write(text); |
| 1177 | | try renderExpression(allocator, stream, tree, indent, init_node); |
| 1294 | const s = if (init_node.id == ast.Node.Id.MultilineStringLiteral) Space.None else Space.Space; |
| 1295 | try renderToken(tree, stream, var_decl.eq_token, indent, s); // = |
| 1296 | try renderExpression(allocator, stream, tree, indent, init_node, Space.None); |
| 1178 | 1297 | } |
| 1179 | 1298 | |
| 1180 | | try renderToken(tree, stream, var_decl.semicolon_token, indent, true, false); |
| 1181 | | } |
| 1182 | | |
| 1183 | | fn maybeRenderSemicolon(stream: var, tree: &ast.Tree, indent: usize, base: &ast.Node) (@typeOf(stream).Child.Error || Error)!void { |
| 1184 | | if (base.requireSemiColon()) { |
| 1185 | | const semicolon_index = base.lastToken() + 1; |
| 1186 | | assert(tree.tokens.at(semicolon_index).id == Token.Id.Semicolon); |
| 1187 | | try renderToken(tree, stream, semicolon_index, indent, true, true); |
| 1188 | | } |
| 1299 | try renderToken(tree, stream, var_decl.semicolon_token, indent, Space.Newline); |
| 1189 | 1300 | } |
| 1190 | 1301 | |
| 1191 | 1302 | fn renderParamDecl(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, indent: usize, base: &ast.Node) (@typeOf(stream).Child.Error || Error)!void { |
| 1192 | 1303 | const param_decl = @fieldParentPtr(ast.Node.ParamDecl, "base", base); |
| 1304 | |
| 1193 | 1305 | if (param_decl.comptime_token) |comptime_token| { |
| 1194 | | try stream.print("{} ", tree.tokenSlice(comptime_token)); |
| 1306 | try renderToken(tree, stream, comptime_token, indent, Space.Space); |
| 1195 | 1307 | } |
| 1196 | 1308 | if (param_decl.noalias_token) |noalias_token| { |
| 1197 | | try stream.print("{} ", tree.tokenSlice(noalias_token)); |
| 1309 | try renderToken(tree, stream, noalias_token, indent, Space.Space); |
| 1198 | 1310 | } |
| 1199 | 1311 | if (param_decl.name_token) |name_token| { |
| 1200 | | try stream.print("{}: ", tree.tokenSlice(name_token)); |
| 1312 | try renderToken(tree, stream, name_token, indent, Space.None); |
| 1313 | try renderToken(tree, stream, tree.nextToken(name_token), indent, Space.Space); // : |
| 1201 | 1314 | } |
| 1202 | 1315 | if (param_decl.var_args_token) |var_args_token| { |
| 1203 | | try stream.print("{}", tree.tokenSlice(var_args_token)); |
| 1316 | try renderToken(tree, stream, var_args_token, indent, Space.None); |
| 1204 | 1317 | } else { |
| 1205 | | try renderExpression(allocator, stream, tree, indent, param_decl.type_node); |
| 1318 | try renderExpression(allocator, stream, tree, indent, param_decl.type_node, Space.None); |
| 1206 | 1319 | } |
| 1207 | 1320 | } |
| 1208 | 1321 | |
| ... | ... | @@ -1213,41 +1326,104 @@ fn renderStatement(allocator: &mem.Allocator, stream: var, tree: &ast.Tree, inde |
| 1213 | 1326 | try renderVarDecl(allocator, stream, tree, indent, var_decl); |
| 1214 | 1327 | }, |
| 1215 | 1328 | else => { |
| 1216 | | try renderExpression(allocator, stream, tree, indent, base); |
| 1217 | | try maybeRenderSemicolon(stream, tree, indent, base); |
| 1329 | if (base.requireSemiColon()) { |
| 1330 | try renderExpression(allocator, stream, tree, indent, base, Space.None); |
| 1331 | |
| 1332 | const semicolon_index = tree.nextToken(base.lastToken()); |
| 1333 | assert(tree.tokens.at(semicolon_index).id == Token.Id.Semicolon); |
| 1334 | try renderToken(tree, stream, semicolon_index, indent, Space.Newline); |
| 1335 | } else { |
| 1336 | try renderExpression(allocator, stream, tree, indent, base, Space.Newline); |
| 1337 | } |
| 1218 | 1338 | }, |
| 1219 | 1339 | } |
| 1220 | 1340 | } |
| 1221 | 1341 | |
| 1222 | | fn renderToken(tree: &ast.Tree, stream: var, token_index: ast.TokenIndex, indent: usize, line_break: bool, space: bool) (@typeOf(stream).Child.Error || Error)!void { |
| 1223 | | const token = tree.tokens.at(token_index); |
| 1342 | const Space = enum { |
| 1343 | None, |
| 1344 | Newline, |
| 1345 | Space, |
| 1346 | NoNewline, |
| 1347 | NoIndent, |
| 1348 | NoComment, |
| 1349 | }; |
| 1350 | |
| 1351 | fn renderToken(tree: &ast.Tree, stream: var, token_index: ast.TokenIndex, indent: usize, space: Space) (@typeOf(stream).Child.Error || Error)!void { |
| 1352 | var token = tree.tokens.at(token_index); |
| 1224 | 1353 | try stream.write(tree.tokenSlicePtr(token)); |
| 1225 | 1354 | |
| 1226 | | const next_token = tree.tokens.at(token_index + 1); |
| 1227 | | if (next_token.id == Token.Id.LineComment) { |
| 1228 | | const loc = tree.tokenLocationPtr(token.end, next_token); |
| 1229 | | if (loc.line == 0) { |
| 1230 | | try stream.print(" {}", tree.tokenSlicePtr(next_token)); |
| 1231 | | if (!line_break) { |
| 1232 | | try stream.write("\n"); |
| 1233 | | |
| 1234 | | const after_comment_token = tree.tokens.at(token_index + 2); |
| 1235 | | const next_line_indent = switch (after_comment_token.id) { |
| 1236 | | Token.Id.RParen, Token.Id.RBrace, Token.Id.RBracket => indent, |
| 1237 | | else => indent + indent_delta, |
| 1238 | | }; |
| 1239 | | try stream.writeByteNTimes(' ', next_line_indent); |
| 1240 | | return; |
| 1355 | if (space == Space.NoComment) return; |
| 1356 | |
| 1357 | var next_token = tree.tokens.at(token_index + 1); |
| 1358 | if (next_token.id != Token.Id.LineComment) { |
| 1359 | switch (space) { |
| 1360 | Space.None, Space.NoNewline, Space.NoIndent => return, |
| 1361 | Space.Newline => return stream.write("\n"), |
| 1362 | Space.Space => return stream.writeByte(' '), |
| 1363 | Space.NoComment => unreachable, |
| 1364 | } |
| 1365 | } |
| 1366 | |
| 1367 | var loc = tree.tokenLocationPtr(token.end, next_token); |
| 1368 | var offset: usize = 1; |
| 1369 | if (loc.line == 0) { |
| 1370 | try stream.print(" {}", tree.tokenSlicePtr(next_token)); |
| 1371 | offset = 2; |
| 1372 | token = next_token; |
| 1373 | next_token = tree.tokens.at(token_index + offset); |
| 1374 | if (next_token.id != Token.Id.LineComment) { |
| 1375 | switch (space) { |
| 1376 | Space.None, Space.Space => { |
| 1377 | try stream.writeByte('\n'); |
| 1378 | const after_comment_token = tree.tokens.at(token_index + offset); |
| 1379 | const next_line_indent = switch (after_comment_token.id) { |
| 1380 | Token.Id.RParen, Token.Id.RBrace, Token.Id.RBracket => indent, |
| 1381 | else => indent + indent_delta, |
| 1382 | }; |
| 1383 | try stream.writeByteNTimes(' ', next_line_indent); |
| 1384 | }, |
| 1385 | Space.Newline, Space.NoIndent => try stream.write("\n"), |
| 1386 | Space.NoNewline => {}, |
| 1387 | Space.NoComment => unreachable, |
| 1241 | 1388 | } |
| 1389 | return; |
| 1242 | 1390 | } |
| 1391 | loc = tree.tokenLocationPtr(token.end, next_token); |
| 1243 | 1392 | } |
| 1244 | 1393 | |
| 1245 | | if (!line_break and space) { |
| 1246 | | try stream.writeByte(' '); |
| 1394 | while (true) { |
| 1395 | assert(loc.line != 0); |
| 1396 | const newline_count = if (loc.line == 1) u8(1) else u8(2); |
| 1397 | try stream.writeByteNTimes('\n', newline_count); |
| 1398 | try stream.writeByteNTimes(' ', indent); |
| 1399 | try stream.write(tree.tokenSlicePtr(next_token)); |
| 1400 | |
| 1401 | offset += 1; |
| 1402 | token = next_token; |
| 1403 | next_token = tree.tokens.at(token_index + offset); |
| 1404 | if (next_token.id != Token.Id.LineComment) { |
| 1405 | switch (space) { |
| 1406 | Space.Newline, Space.NoIndent => try stream.writeByte('\n'), |
| 1407 | Space.None, Space.Space => { |
| 1408 | try stream.writeByte('\n'); |
| 1409 | |
| 1410 | const after_comment_token = tree.tokens.at(token_index + offset); |
| 1411 | const next_line_indent = switch (after_comment_token.id) { |
| 1412 | Token.Id.RParen, Token.Id.RBrace, Token.Id.RBracket => indent, |
| 1413 | else => indent, |
| 1414 | }; |
| 1415 | try stream.writeByteNTimes(' ', next_line_indent); |
| 1416 | }, |
| 1417 | Space.NoNewline => {}, |
| 1418 | Space.NoComment => unreachable, |
| 1419 | } |
| 1420 | return; |
| 1421 | } |
| 1422 | loc = tree.tokenLocationPtr(token.end, next_token); |
| 1247 | 1423 | } |
| 1248 | 1424 | } |
| 1249 | 1425 | |
| 1250 | | fn renderComments(tree: &ast.Tree, stream: var, node: var, indent: usize) (@typeOf(stream).Child.Error || Error)!void { |
| 1426 | fn renderDocComments(tree: &ast.Tree, stream: var, node: var, indent: usize) (@typeOf(stream).Child.Error || Error)!void { |
| 1251 | 1427 | const comment = node.doc_comments ?? return; |
| 1252 | 1428 | var it = comment.lines.iterator(0); |
| 1253 | 1429 | while (it.next()) |line_token_index| { |