| ... | ... | @@ -346,20 +346,27 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 346 | 346 | const next_token_tag = tree.tokenTag(next_token); |
| 347 | 347 | |
| 348 | 348 | // dedent the next thing that comes after a multiline string literal |
| 349 | | if (!ais.indentStackEmpty() and |
| 350 | | next_token_tag != .colon and |
| 351 | | ((next_token_tag != .semicolon and next_token_tag != .comma) or |
| 352 | | ais.lastSpaceModeIndent() < ais.currentIndent())) |
| 349 | if (next_token_tag != .colon and |
| 350 | !ais.indentStackEmpty() and |
| 351 | ais.lastSpaceModeIndent() < ais.currentIndent()) |
| 353 | 352 | { |
| 354 | | ais.popIndent(); |
| 355 | | try ais.pushIndent(.normal); |
| 353 | const indent_top = &ais.indent_stack.items[ais.indent_stack.items.len - 1]; |
| 354 | if (indent_top.realized) { |
| 355 | indent_top.realized = false; |
| 356 | ais.indent_count -= 1; |
| 357 | } |
| 356 | 358 | } |
| 357 | 359 | |
| 358 | 360 | switch (space) { |
| 359 | | .none, .space, .newline, .skip => {}, |
| 360 | | .semicolon => if (next_token_tag == .semicolon) try renderTokenOverrideSpaceMode(r, next_token, .newline, .semicolon), |
| 361 | | .comma => if (next_token_tag == .comma) try renderTokenOverrideSpaceMode(r, next_token, .newline, .comma), |
| 362 | | .comma_space => if (next_token_tag == .comma) try renderToken(r, next_token, .space), |
| 361 | .none, .space, .newline, .maybe_space, .skip => {}, |
| 362 | .semicolon => if (next_token_tag == .semicolon) |
| 363 | try renderTokenOverrideSpaceMode(r, next_token, .newline, .semicolon), |
| 364 | .comma => if (next_token_tag == .comma) |
| 365 | try renderTokenOverrideSpaceMode(r, next_token, .newline, .comma), |
| 366 | .comma_space => if (next_token_tag == .comma) |
| 367 | try renderToken(r, next_token, .space), |
| 368 | .comma_maybe_space => if (next_token_tag == .comma) |
| 369 | try renderToken(r, next_token, .maybe_space), |
| 363 | 370 | } |
| 364 | 371 | }, |
| 365 | 372 | |
| ... | ... | @@ -384,11 +391,11 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 384 | 391 | const defer_token = tree.nodeMainToken(node); |
| 385 | 392 | const maybe_payload_token, const expr = tree.nodeData(node).opt_token_and_node; |
| 386 | 393 | |
| 387 | | try renderToken(r, defer_token, .space); |
| 394 | try renderToken(r, defer_token, .maybe_space); |
| 388 | 395 | if (maybe_payload_token.unwrap()) |payload_token| { |
| 389 | 396 | try renderToken(r, payload_token - 1, .none); // | |
| 390 | 397 | try renderIdentifier(r, payload_token, .none, .preserve_when_shadowing); // identifier |
| 391 | | try renderToken(r, payload_token + 1, .space); // | |
| 398 | try renderToken(r, payload_token + 1, .maybe_space); // | |
| 392 | 399 | } |
| 393 | 400 | return renderExpression(r, expr, space); |
| 394 | 401 | }, |
| ... | ... | @@ -400,7 +407,7 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 400 | 407 | => { |
| 401 | 408 | const main_token = tree.nodeMainToken(node); |
| 402 | 409 | const item = tree.nodeData(node).node; |
| 403 | | try renderToken(r, main_token, .space); |
| 410 | try renderToken(r, main_token, .maybe_space); |
| 404 | 411 | return renderExpression(r, item, space); |
| 405 | 412 | }, |
| 406 | 413 | |
| ... | ... | @@ -409,8 +416,9 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 409 | 416 | const lhs, const rhs = tree.nodeData(node).node_and_node; |
| 410 | 417 | const fallback_first = tree.firstToken(rhs); |
| 411 | 418 | |
| 412 | | const same_line = tree.tokensOnSameLine(main_token, fallback_first); |
| 413 | | const after_op_space = if (same_line) Space.space else Space.newline; |
| 419 | const seperate_line = !tree.tokensOnSameLine(main_token, fallback_first) or |
| 420 | tree.tokenTag(fallback_first) == .multiline_string_literal_line; |
| 421 | const after_op_space: Space = if (seperate_line) .newline else .space; |
| 414 | 422 | |
| 415 | 423 | try renderExpression(r, lhs, .space); // target |
| 416 | 424 | |
| ... | ... | @@ -433,13 +441,18 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 433 | 441 | const dot_token = name_token - 1; |
| 434 | 442 | |
| 435 | 443 | try ais.pushIndent(.field_access); |
| 436 | | try renderExpression(r, lhs, .none); |
| 437 | 444 | |
| 438 | | // Allow a line break between the lhs and the dot if the lhs and rhs |
| 439 | | // are on different lines. |
| 440 | 445 | const lhs_last_token = tree.lastToken(lhs); |
| 441 | 446 | const same_line = tree.tokensOnSameLine(lhs_last_token, name_token); |
| 442 | | if (!same_line and !hasComment(tree, lhs_last_token, dot_token)) try ais.insertNewline(); |
| 447 | // Keeping a space after the number ensures it will not turn into a decimal number |
| 448 | // (e.g. "0xF .A"). |
| 449 | const number_space: Space = if (tree.tokenTag(lhs_last_token) == .number_literal and same_line) .space else .none; |
| 450 | try renderExpression(r, lhs, number_space); |
| 451 | |
| 452 | // Allow a line break between the lhs and the dot if the lhs and rhs |
| 453 | // are on different lines. |
| 454 | if (!same_line and !hasComment(tree, lhs_last_token, dot_token)) |
| 455 | try ais.insertNewline(); |
| 443 | 456 | |
| 444 | 457 | try renderToken(r, dot_token, .none); |
| 445 | 458 | |
| ... | ... | @@ -489,11 +502,9 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 489 | 502 | try renderExpression(r, lhs, .space); |
| 490 | 503 | const op_token = tree.nodeMainToken(node); |
| 491 | 504 | try ais.pushIndent(.after_equals); |
| 492 | | if (tree.tokensOnSameLine(op_token, op_token + 1)) { |
| 493 | | try renderToken(r, op_token, .space); |
| 494 | | } else { |
| 495 | | try renderToken(r, op_token, .newline); |
| 496 | | } |
| 505 | const rhs_seperate_line = !tree.tokensOnSameLine(op_token, op_token + 1) or |
| 506 | tree.tokenTag(op_token + 1) == .multiline_string_literal_line; |
| 507 | try renderToken(r, op_token, if (rhs_seperate_line) .newline else .space); |
| 497 | 508 | try renderExpression(r, rhs, space); |
| 498 | 509 | ais.popIndent(); |
| 499 | 510 | }, |
| ... | ... | @@ -532,11 +543,9 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 532 | 543 | try renderExpression(r, lhs, .space); |
| 533 | 544 | const op_token = tree.nodeMainToken(node); |
| 534 | 545 | try ais.pushIndent(.binop); |
| 535 | | if (tree.tokensOnSameLine(op_token, op_token + 1)) { |
| 536 | | try renderToken(r, op_token, .space); |
| 537 | | } else { |
| 538 | | try renderToken(r, op_token, .newline); |
| 539 | | } |
| 546 | const rhs_seperate_line = !tree.tokensOnSameLine(op_token, op_token + 1) or |
| 547 | tree.tokenTag(op_token + 1) == .multiline_string_literal_line; |
| 548 | try renderToken(r, op_token, if (rhs_seperate_line) .newline else .space); |
| 540 | 549 | try renderExpression(r, rhs, space); |
| 541 | 550 | ais.popIndent(); |
| 542 | 551 | }, |
| ... | ... | @@ -544,11 +553,11 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 544 | 553 | .assign_destructure => { |
| 545 | 554 | const full = tree.assignDestructure(node); |
| 546 | 555 | if (full.comptime_token) |comptime_token| { |
| 547 | | try renderToken(r, comptime_token, .space); |
| 556 | try renderToken(r, comptime_token, .maybe_space); |
| 548 | 557 | } |
| 549 | 558 | |
| 550 | 559 | for (full.ast.variables, 0..) |variable_node, i| { |
| 551 | | const variable_space: Space = if (i == full.ast.variables.len - 1) .space else .comma_space; |
| 560 | const variable_space: Space = if (i == full.ast.variables.len - 1) .maybe_space else .comma_maybe_space; |
| 552 | 561 | switch (tree.nodeTag(variable_node)) { |
| 553 | 562 | .global_var_decl, |
| 554 | 563 | .local_var_decl, |
| ... | ... | @@ -561,11 +570,10 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 561 | 570 | } |
| 562 | 571 | } |
| 563 | 572 | try ais.pushIndent(.after_equals); |
| 564 | | if (tree.tokensOnSameLine(full.ast.equal_token, full.ast.equal_token + 1)) { |
| 565 | | try renderToken(r, full.ast.equal_token, .space); |
| 566 | | } else { |
| 567 | | try renderToken(r, full.ast.equal_token, .newline); |
| 568 | | } |
| 573 | const expr_seperate_line = |
| 574 | !tree.tokensOnSameLine(full.ast.equal_token, full.ast.equal_token + 1) or |
| 575 | tree.tokenTag(full.ast.equal_token + 1) == .multiline_string_literal_line; |
| 576 | try renderToken(r, full.ast.equal_token, if (expr_seperate_line) .newline else .space); |
| 569 | 577 | try renderExpression(r, full.ast.value_expr, space); |
| 570 | 578 | ais.popIndent(); |
| 571 | 579 | }, |
| ... | ... | @@ -584,7 +592,7 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 584 | 592 | .@"try", |
| 585 | 593 | .@"resume", |
| 586 | 594 | => { |
| 587 | | try renderToken(r, tree.nodeMainToken(node), .space); |
| 595 | try renderToken(r, tree.nodeMainToken(node), .maybe_space); |
| 588 | 596 | return renderExpression(r, tree.nodeData(node).node, space); |
| 589 | 597 | }, |
| 590 | 598 | |
| ... | ... | @@ -637,7 +645,8 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 637 | 645 | const lhs, const rhs = tree.nodeData(node).node_and_node; |
| 638 | 646 | const lbracket = tree.firstToken(rhs) - 1; |
| 639 | 647 | const rbracket = tree.lastToken(rhs) + 1; |
| 640 | | const one_line = tree.tokensOnSameLine(lbracket, rbracket); |
| 648 | const one_line = tree.tokensOnSameLine(lbracket, rbracket) and |
| 649 | !becomesMultilineExpr(tree, rhs); |
| 641 | 650 | const inner_space = if (one_line) Space.none else Space.newline; |
| 642 | 651 | try renderExpression(r, lhs, .none); |
| 643 | 652 | try ais.pushIndent(.normal); |
| ... | ... | @@ -668,30 +677,23 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 668 | 677 | .@"break", .@"continue" => { |
| 669 | 678 | const main_token = tree.nodeMainToken(node); |
| 670 | 679 | const opt_label_token, const opt_target = tree.nodeData(node).opt_token_and_opt_node; |
| 671 | | if (opt_label_token == .none and opt_target == .none) { |
| 672 | | try renderToken(r, main_token, space); // break/continue |
| 673 | | } else if (opt_label_token == .none and opt_target != .none) { |
| 674 | | const target = opt_target.unwrap().?; |
| 675 | | try renderToken(r, main_token, .space); // break/continue |
| 676 | | try renderExpression(r, target, space); |
| 677 | | } else if (opt_label_token != .none and opt_target == .none) { |
| 678 | | const label_token = opt_label_token.unwrap().?; |
| 679 | | try renderToken(r, main_token, .space); // break/continue |
| 680 | | try renderToken(r, label_token - 1, .none); // : |
| 681 | | try renderIdentifier(r, label_token, space, .eagerly_unquote); // identifier |
| 682 | | } else if (opt_label_token != .none and opt_target != .none) { |
| 683 | | const label_token = opt_label_token.unwrap().?; |
| 684 | | const target = opt_target.unwrap().?; |
| 685 | | try renderToken(r, main_token, .space); // break/continue |
| 680 | |
| 681 | const before_target_space: Space = if (opt_target != .none) .maybe_space else space; |
| 682 | const before_label_space: Space = if (opt_label_token != .none) .space else before_target_space; |
| 683 | |
| 684 | try renderToken(r, main_token, before_label_space); |
| 685 | if (opt_label_token.unwrap()) |label_token| { |
| 686 | 686 | try renderToken(r, label_token - 1, .none); // : |
| 687 | | try renderIdentifier(r, label_token, .space, .eagerly_unquote); // identifier |
| 687 | try renderIdentifier(r, label_token, before_target_space, .eagerly_unquote); // identifier |
| 688 | } |
| 689 | if (opt_target.unwrap()) |target| { |
| 688 | 690 | try renderExpression(r, target, space); |
| 689 | | } else unreachable; |
| 691 | } |
| 690 | 692 | }, |
| 691 | 693 | |
| 692 | 694 | .@"return" => { |
| 693 | 695 | if (tree.nodeData(node).opt_node.unwrap()) |expr| { |
| 694 | | try renderToken(r, tree.nodeMainToken(node), .space); |
| 696 | try renderToken(r, tree.nodeMainToken(node), .maybe_space); |
| 695 | 697 | try renderExpression(r, expr, space); |
| 696 | 698 | } else { |
| 697 | 699 | try renderToken(r, tree.nodeMainToken(node), space); |
| ... | ... | @@ -740,33 +742,36 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void { |
| 740 | 742 | try renderToken(r, lbrace, .none); |
| 741 | 743 | try renderIdentifier(r, lbrace + 1, .none, .eagerly_unquote); // identifier |
| 742 | 744 | return renderToken(r, rbrace, space); |
| 743 | | } else if (tree.tokenTag(rbrace - 1) == .comma) { |
| 744 | | // There is a trailing comma so render each member on a new line. |
| 745 | } else if (!isOneLineErrorSetDecl(tree, lbrace, rbrace)) { |
| 746 | // Render each member on a new line. |
| 745 | 747 | try ais.pushIndent(.normal); |
| 746 | 748 | try renderToken(r, lbrace, .newline); |
| 747 | 749 | var i = lbrace + 1; |
| 748 | 750 | while (i < rbrace) : (i += 1) { |
| 751 | const tag = tree.tokenTag(i); |
| 752 | if (tag == .comma) { |
| 753 | assert(tree.tokenTag(i - 1) == .identifier); |
| 754 | continue; |
| 755 | } |
| 749 | 756 | if (i > lbrace + 1) try renderExtraNewlineToken(r, i); |
| 750 | | switch (tree.tokenTag(i)) { |
| 757 | switch (tag) { |
| 751 | 758 | .doc_comment => try renderToken(r, i, .newline), |
| 752 | 759 | .identifier => { |
| 753 | 760 | try ais.pushSpace(.comma); |
| 754 | 761 | try renderIdentifier(r, i, .comma, .eagerly_unquote); |
| 755 | 762 | ais.popSpace(); |
| 756 | 763 | }, |
| 757 | | .comma => {}, |
| 758 | 764 | else => unreachable, |
| 759 | 765 | } |
| 760 | 766 | } |
| 761 | 767 | ais.popIndent(); |
| 762 | 768 | return renderToken(r, rbrace, space); |
| 763 | 769 | } else { |
| 764 | | // There is no trailing comma so render everything on one line. |
| 770 | // Render each member on one line. |
| 765 | 771 | try renderToken(r, lbrace, .space); |
| 766 | 772 | var i = lbrace + 1; |
| 767 | 773 | while (i < rbrace) : (i += 1) { |
| 768 | 774 | switch (tree.tokenTag(i)) { |
| 769 | | .doc_comment => unreachable, // TODO |
| 770 | 775 | .identifier => try renderIdentifier(r, i, .comma_space, .eagerly_unquote), |
| 771 | 776 | .comma => {}, |
| 772 | 777 | else => unreachable, |
| ... | ... | @@ -926,6 +931,382 @@ fn renderExpressionFixup(r: *Render, node: Ast.Node.Index, space: Space) Error!v |
| 926 | 931 | } |
| 927 | 932 | } |
| 928 | 933 | |
| 934 | /// Same as becomesMultilineExpr, but returns false when `node == .none` |
| 935 | fn optBecomesMultilineExpr(tree: Ast, node: Ast.Node.OptionalIndex) bool { |
| 936 | return if (node.unwrap()) |payload| becomesMultilineExpr(tree, payload) else false; |
| 937 | } |
| 938 | |
| 939 | /// May return false if `node` is already multiline |
| 940 | fn becomesMultilineExpr(tree: Ast, node: Ast.Node.Index) bool { |
| 941 | // Conditions related to comments, doc comments, and multiline string literals are ignored |
| 942 | // since they always go to the end of the line, which already make them a multi-line |
| 943 | // expression (since they contain a newline). |
| 944 | switch (tree.nodeTag(node)) { |
| 945 | .identifier, |
| 946 | .number_literal, |
| 947 | .char_literal, |
| 948 | .unreachable_literal, |
| 949 | .anyframe_literal, |
| 950 | .string_literal, |
| 951 | .multiline_string_literal, |
| 952 | .error_value, |
| 953 | .enum_literal, |
| 954 | => return false, |
| 955 | .container_decl_trailing, |
| 956 | .container_decl_arg_trailing, |
| 957 | .container_decl_two_trailing, |
| 958 | .tagged_union_trailing, |
| 959 | .tagged_union_enum_tag_trailing, |
| 960 | .tagged_union_two_trailing, |
| 961 | .switch_comma, |
| 962 | .builtin_call_two_comma, |
| 963 | .builtin_call_comma, |
| 964 | .call_one_comma, |
| 965 | .call_comma, |
| 966 | .struct_init_one_comma, |
| 967 | .struct_init_dot_two_comma, |
| 968 | .struct_init_dot_comma, |
| 969 | .struct_init_comma, |
| 970 | .array_init_one_comma, |
| 971 | .array_init_dot_two_comma, |
| 972 | .array_init_dot_comma, |
| 973 | .array_init_comma, |
| 974 | // The following always have a non-zero amount of members |
| 975 | // which is also the condition for them to be multi-line. |
| 976 | .block, |
| 977 | .block_semicolon, |
| 978 | => return true, |
| 979 | .block_two, |
| 980 | .block_two_semicolon, |
| 981 | => return tree.nodeData(node).opt_node_and_opt_node[0] != .none, |
| 982 | .container_decl, |
| 983 | .container_decl_arg, |
| 984 | .container_decl_two, |
| 985 | .tagged_union, |
| 986 | .tagged_union_enum_tag, |
| 987 | .tagged_union_two, |
| 988 | => { |
| 989 | var buf: [2]Ast.Node.Index = undefined; |
| 990 | const full = tree.fullContainerDecl(&buf, node).?; |
| 991 | if (full.ast.arg.unwrap()) |arg| { |
| 992 | if (becomesMultilineExpr(tree, arg)) |
| 993 | return true; |
| 994 | } |
| 995 | // This does the same checks as `isOneLineContainerDecl`, however it avoids unnecessary |
| 996 | // checks related to comments and multiline strings, which would mean the container is |
| 997 | // already multiple lines. |
| 998 | for (full.ast.members) |member| { |
| 999 | if (tree.fullContainerField(member)) |field_full| { |
| 1000 | for ([_]Ast.Node.OptionalIndex{ |
| 1001 | field_full.ast.type_expr, |
| 1002 | field_full.ast.align_expr, |
| 1003 | field_full.ast.value_expr, |
| 1004 | }) |opt_expr| { |
| 1005 | if (opt_expr.unwrap()) |expr| { |
| 1006 | if (becomesMultilineExpr(tree, expr)) |
| 1007 | return true; |
| 1008 | } |
| 1009 | } |
| 1010 | } else return true; |
| 1011 | } |
| 1012 | return false; |
| 1013 | }, |
| 1014 | .error_set_decl => { |
| 1015 | const lbrace, const rbrace = tree.nodeData(node).token_and_token; |
| 1016 | return !isOneLineErrorSetDecl(tree, lbrace, rbrace); |
| 1017 | }, |
| 1018 | .@"switch" => { |
| 1019 | const op, const extra_index = tree.nodeData(node).node_and_extra; |
| 1020 | const case_range = tree.extraData(extra_index, Ast.Node.SubRange); |
| 1021 | return @intFromEnum(case_range.end) - @intFromEnum(case_range.start) != 0 or |
| 1022 | becomesMultilineExpr(tree, op); |
| 1023 | }, |
| 1024 | .for_simple, .@"for" => { |
| 1025 | const full = tree.fullFor(node).?; |
| 1026 | if (becomesMultilineExpr(tree, full.ast.then_expr) or |
| 1027 | optBecomesMultilineExpr(tree, full.ast.else_expr)) |
| 1028 | return true; |
| 1029 | |
| 1030 | for (full.ast.inputs) |expr| { |
| 1031 | if (if (tree.nodeTag(expr) == .for_range) blk: { |
| 1032 | const lhs, const rhs = tree.nodeData(expr).node_and_opt_node; |
| 1033 | break :blk becomesMultilineExpr(tree, lhs) or optBecomesMultilineExpr(tree, rhs); |
| 1034 | } else becomesMultilineExpr(tree, expr)) |
| 1035 | return true; |
| 1036 | } |
| 1037 | const final_input_expr = full.ast.inputs[full.ast.inputs.len - 1]; |
| 1038 | if (tree.tokenTag(tree.lastToken(final_input_expr) + 1) == .comma) |
| 1039 | return true; |
| 1040 | |
| 1041 | const token_tags = tree.tokens.items(.tag); |
| 1042 | const payload = full.payload_token; |
| 1043 | const pipe = std.mem.indexOfScalarPos(Token.Tag, token_tags, payload, .pipe).?; |
| 1044 | return token_tags[@intCast(pipe - 1)] == .comma; |
| 1045 | }, |
| 1046 | .while_simple, |
| 1047 | .while_cont, |
| 1048 | .@"while", |
| 1049 | => { |
| 1050 | const full = tree.fullWhile(node).?; |
| 1051 | return becomesMultilineExpr(tree, full.ast.cond_expr) or |
| 1052 | becomesMultilineExpr(tree, full.ast.then_expr) or |
| 1053 | optBecomesMultilineExpr(tree, full.ast.cont_expr) or |
| 1054 | optBecomesMultilineExpr(tree, full.ast.else_expr); |
| 1055 | }, |
| 1056 | .if_simple, |
| 1057 | .@"if", |
| 1058 | => { |
| 1059 | const full = tree.fullIf(node).?; |
| 1060 | return becomesMultilineExpr(tree, full.ast.cond_expr) or |
| 1061 | becomesMultilineExpr(tree, full.ast.then_expr) or |
| 1062 | optBecomesMultilineExpr(tree, full.ast.else_expr); |
| 1063 | }, |
| 1064 | .fn_proto_simple, |
| 1065 | .fn_proto_multi, |
| 1066 | .fn_proto_one, |
| 1067 | .fn_proto, |
| 1068 | => { |
| 1069 | var buf: [1]Ast.Node.Index = undefined; |
| 1070 | const fn_proto = tree.fullFnProto(&buf, node).?; |
| 1071 | |
| 1072 | for ([_]Ast.Node.OptionalIndex{ |
| 1073 | fn_proto.ast.return_type, |
| 1074 | fn_proto.ast.align_expr, |
| 1075 | fn_proto.ast.addrspace_expr, |
| 1076 | fn_proto.ast.section_expr, |
| 1077 | fn_proto.ast.callconv_expr, |
| 1078 | }) |opt_expr| { |
| 1079 | if (opt_expr.unwrap()) |expr| { |
| 1080 | if (becomesMultilineExpr(tree, expr)) |
| 1081 | return true; |
| 1082 | } |
| 1083 | } |
| 1084 | for (fn_proto.ast.params) |expr| { |
| 1085 | if (becomesMultilineExpr(tree, expr)) |
| 1086 | return true; |
| 1087 | } |
| 1088 | |
| 1089 | const lparen = fn_proto.ast.fn_token + 1; |
| 1090 | const return_type = fn_proto.ast.return_type.unwrap().?; |
| 1091 | const maybe_bang = tree.firstToken(return_type) - 1; |
| 1092 | const rparen = fnProtoRparen(tree, fn_proto, maybe_bang); |
| 1093 | return !isOneLineFnProto(tree, fn_proto, lparen, rparen); |
| 1094 | }, |
| 1095 | .asm_simple, |
| 1096 | => { |
| 1097 | const lhs = tree.nodeData(node).node_and_token[0]; |
| 1098 | return becomesMultilineExpr(tree, lhs); |
| 1099 | }, |
| 1100 | .@"asm", |
| 1101 | => { |
| 1102 | const lhs, const extra_index = tree.nodeData(node).node_and_extra; |
| 1103 | const asm_extra = tree.extraData(extra_index, Ast.Node.Asm); |
| 1104 | return @intFromEnum(asm_extra.items_end) - @intFromEnum(asm_extra.items_start) != 0 or |
| 1105 | becomesMultilineExpr(tree, lhs) or optBecomesMultilineExpr(tree, asm_extra.clobbers); |
| 1106 | }, |
| 1107 | .array_type, .array_type_sentinel => { |
| 1108 | const array_type = tree.fullArrayType(node).?; |
| 1109 | const rbracket = tree.firstToken(array_type.ast.elem_type) - 1; |
| 1110 | return !isOneLineArrayType(tree, array_type, rbracket) or |
| 1111 | becomesMultilineExpr(tree, array_type.ast.elem_type); |
| 1112 | }, |
| 1113 | .array_access => { |
| 1114 | const lhs, const rhs = tree.nodeData(node).node_and_node; |
| 1115 | const lbracket = tree.firstToken(rhs) - 1; |
| 1116 | const rbracket = tree.lastToken(rhs) + 1; |
| 1117 | return !tree.tokensOnSameLine(lbracket, rbracket) or |
| 1118 | becomesMultilineExpr(tree, lhs) or |
| 1119 | becomesMultilineExpr(tree, rhs); |
| 1120 | }, |
| 1121 | .call_one, |
| 1122 | .call, |
| 1123 | .builtin_call_two, |
| 1124 | .builtin_call, |
| 1125 | .array_init_one, |
| 1126 | .array_init_dot_two, |
| 1127 | .array_init_dot, |
| 1128 | .array_init, |
| 1129 | .struct_init_one, |
| 1130 | .struct_init_dot_two, |
| 1131 | .struct_init_dot, |
| 1132 | .struct_init, |
| 1133 | => |tag| { |
| 1134 | var buf: [2]Ast.Node.Index = undefined; |
| 1135 | const opt_lhs: Ast.Node.OptionalIndex, const items = switch (tag) { |
| 1136 | .call_one, .call => blk: { |
| 1137 | const full = tree.fullCall(buf[0..1], node).?; |
| 1138 | break :blk .{ full.ast.fn_expr.toOptional(), full.ast.params }; |
| 1139 | }, |
| 1140 | .builtin_call_two, .builtin_call => .{ .none, tree.builtinCallParams(&buf, node).? }, |
| 1141 | .array_init_one, |
| 1142 | .array_init_dot_two, |
| 1143 | .array_init_dot, |
| 1144 | .array_init, |
| 1145 | => blk: { |
| 1146 | const full = tree.fullArrayInit(&buf, node).?; |
| 1147 | break :blk .{ full.ast.type_expr, full.ast.elements }; |
| 1148 | }, |
| 1149 | .struct_init_one, |
| 1150 | .struct_init_dot_two, |
| 1151 | .struct_init_dot, |
| 1152 | .struct_init, |
| 1153 | => blk: { |
| 1154 | const full = tree.fullStructInit(&buf, node).?; |
| 1155 | break :blk .{ full.ast.type_expr, full.ast.fields }; |
| 1156 | }, |
| 1157 | else => unreachable, |
| 1158 | }; |
| 1159 | if (opt_lhs.unwrap()) |lhs| { |
| 1160 | if (becomesMultilineExpr(tree, lhs)) |
| 1161 | return true; |
| 1162 | } |
| 1163 | for (items) |expr| { |
| 1164 | if (becomesMultilineExpr(tree, expr)) |
| 1165 | return true; |
| 1166 | } |
| 1167 | return false; |
| 1168 | }, |
| 1169 | .assign_destructure => { |
| 1170 | const full = tree.assignDestructure(node); |
| 1171 | for (full.ast.variables) |expr| { |
| 1172 | if (becomesMultilineExpr(tree, expr)) |
| 1173 | return true; |
| 1174 | } |
| 1175 | return becomesMultilineExpr(tree, full.ast.value_expr); |
| 1176 | }, |
| 1177 | .ptr_type_aligned, |
| 1178 | .ptr_type_sentinel, |
| 1179 | .ptr_type, |
| 1180 | .ptr_type_bit_range, |
| 1181 | => { |
| 1182 | const full = tree.fullPtrType(node).?; |
| 1183 | return becomesMultilineExpr(tree, full.ast.child_type) or |
| 1184 | optBecomesMultilineExpr(tree, full.ast.sentinel) or |
| 1185 | optBecomesMultilineExpr(tree, full.ast.align_node) or |
| 1186 | optBecomesMultilineExpr(tree, full.ast.addrspace_node) or |
| 1187 | optBecomesMultilineExpr(tree, full.ast.bit_range_start) or |
| 1188 | optBecomesMultilineExpr(tree, full.ast.bit_range_end); |
| 1189 | }, |
| 1190 | .slice_open, |
| 1191 | .slice, |
| 1192 | .slice_sentinel, |
| 1193 | => { |
| 1194 | const full = tree.fullSlice(node).?; |
| 1195 | return becomesMultilineExpr(tree, full.ast.sliced) or |
| 1196 | becomesMultilineExpr(tree, full.ast.start) or |
| 1197 | optBecomesMultilineExpr(tree, full.ast.end) or |
| 1198 | optBecomesMultilineExpr(tree, full.ast.sentinel); |
| 1199 | }, |
| 1200 | .@"comptime", |
| 1201 | .@"nosuspend", |
| 1202 | .@"suspend", |
| 1203 | .@"resume", |
| 1204 | .bit_not, |
| 1205 | .bool_not, |
| 1206 | .negation, |
| 1207 | .negation_wrap, |
| 1208 | .optional_type, |
| 1209 | .address_of, |
| 1210 | .deref, |
| 1211 | .@"try", |
| 1212 | => return becomesMultilineExpr(tree, tree.nodeData(node).node), |
| 1213 | .@"return" => return optBecomesMultilineExpr(tree, tree.nodeData(node).opt_node), |
| 1214 | .field_access, |
| 1215 | .unwrap_optional, |
| 1216 | .grouped_expression, |
| 1217 | => return becomesMultilineExpr(tree, tree.nodeData(node).node_and_token[0]), |
| 1218 | .add, |
| 1219 | .add_wrap, |
| 1220 | .add_sat, |
| 1221 | .array_cat, |
| 1222 | .array_mult, |
| 1223 | .bang_equal, |
| 1224 | .bit_and, |
| 1225 | .bit_or, |
| 1226 | .shl, |
| 1227 | .shl_sat, |
| 1228 | .shr, |
| 1229 | .bit_xor, |
| 1230 | .bool_and, |
| 1231 | .bool_or, |
| 1232 | .div, |
| 1233 | .equal_equal, |
| 1234 | .greater_or_equal, |
| 1235 | .greater_than, |
| 1236 | .less_or_equal, |
| 1237 | .less_than, |
| 1238 | .merge_error_sets, |
| 1239 | .mod, |
| 1240 | .mul, |
| 1241 | .mul_wrap, |
| 1242 | .mul_sat, |
| 1243 | .sub, |
| 1244 | .sub_wrap, |
| 1245 | .sub_sat, |
| 1246 | .@"orelse", |
| 1247 | .@"catch", |
| 1248 | .error_union, |
| 1249 | .assign, |
| 1250 | .assign_bit_and, |
| 1251 | .assign_bit_or, |
| 1252 | .assign_shl, |
| 1253 | .assign_shl_sat, |
| 1254 | .assign_shr, |
| 1255 | .assign_bit_xor, |
| 1256 | .assign_div, |
| 1257 | .assign_sub, |
| 1258 | .assign_sub_wrap, |
| 1259 | .assign_sub_sat, |
| 1260 | .assign_mod, |
| 1261 | .assign_add, |
| 1262 | .assign_add_wrap, |
| 1263 | .assign_add_sat, |
| 1264 | .assign_mul, |
| 1265 | .assign_mul_wrap, |
| 1266 | .assign_mul_sat, |
| 1267 | => { |
| 1268 | const lhs, const rhs = tree.nodeData(node).node_and_node; |
| 1269 | return becomesMultilineExpr(tree, lhs) or becomesMultilineExpr(tree, rhs); |
| 1270 | }, |
| 1271 | .@"break", .@"continue" => { |
| 1272 | const opt_expr = tree.nodeData(node).opt_token_and_opt_node[1]; |
| 1273 | return optBecomesMultilineExpr(tree, opt_expr); |
| 1274 | }, |
| 1275 | .anyframe_type => return becomesMultilineExpr(tree, tree.nodeData(node).token_and_node[1]), |
| 1276 | .@"errdefer", |
| 1277 | .@"defer", |
| 1278 | .for_range, |
| 1279 | .switch_range, |
| 1280 | .switch_case_one, |
| 1281 | .switch_case_inline_one, |
| 1282 | .switch_case, |
| 1283 | .switch_case_inline, |
| 1284 | .asm_output, |
| 1285 | .asm_input, |
| 1286 | .fn_decl, |
| 1287 | .container_field, |
| 1288 | .container_field_init, |
| 1289 | .container_field_align, |
| 1290 | .root, |
| 1291 | .global_var_decl, |
| 1292 | .local_var_decl, |
| 1293 | .simple_var_decl, |
| 1294 | .aligned_var_decl, |
| 1295 | .test_decl, |
| 1296 | => unreachable, |
| 1297 | } |
| 1298 | } |
| 1299 | |
| 1300 | fn isOneLineArrayType( |
| 1301 | tree: Ast, |
| 1302 | array_type: Ast.full.ArrayType, |
| 1303 | rbracket: Ast.TokenIndex, |
| 1304 | ) bool { |
| 1305 | return tree.tokensOnSameLine(array_type.ast.lbracket, rbracket) and |
| 1306 | !becomesMultilineExpr(tree, array_type.ast.elem_count) and |
| 1307 | !optBecomesMultilineExpr(tree, array_type.ast.sentinel); |
| 1308 | } |
| 1309 | |
| 929 | 1310 | fn renderArrayType( |
| 930 | 1311 | r: *Render, |
| 931 | 1312 | array_type: Ast.full.ArrayType, |
| ... | ... | @@ -934,7 +1315,7 @@ fn renderArrayType( |
| 934 | 1315 | const tree = r.tree; |
| 935 | 1316 | const ais = r.ais; |
| 936 | 1317 | const rbracket = tree.firstToken(array_type.ast.elem_type) - 1; |
| 937 | | const one_line = tree.tokensOnSameLine(array_type.ast.lbracket, rbracket); |
| 1318 | const one_line = isOneLineArrayType(tree, array_type, rbracket); |
| 938 | 1319 | const inner_space = if (one_line) Space.none else Space.newline; |
| 939 | 1320 | try ais.pushIndent(.normal); |
| 940 | 1321 | try renderToken(r, array_type.ast.lbracket, inner_space); // lbracket |
| ... | ... | @@ -951,6 +1332,7 @@ fn renderArrayType( |
| 951 | 1332 | fn renderPtrType(r: *Render, ptr_type: Ast.full.PtrType, space: Space) Error!void { |
| 952 | 1333 | const tree = r.tree; |
| 953 | 1334 | const main_token = ptr_type.ast.main_token; |
| 1335 | |
| 954 | 1336 | switch (ptr_type.size) { |
| 955 | 1337 | .one => { |
| 956 | 1338 | // Since ** tokens exist and the same token is shared by two |
| ... | ... | @@ -997,11 +1379,36 @@ fn renderPtrType(r: *Render, ptr_type: Ast.full.PtrType, space: Space) Error!voi |
| 997 | 1379 | }, |
| 998 | 1380 | } |
| 999 | 1381 | |
| 1382 | // .maybe_space cannot be used at the end of each qualifier since they may be reordered |
| 1383 | const final_qual: enum { |
| 1384 | @"volatile", |
| 1385 | @"const", |
| 1386 | @"addrspace", |
| 1387 | @"align", |
| 1388 | @"allowzero", |
| 1389 | none, |
| 1390 | } = if (ptr_type.volatile_token != null) |
| 1391 | .@"volatile" |
| 1392 | else if (ptr_type.const_token != null) |
| 1393 | .@"const" |
| 1394 | else if (ptr_type.ast.addrspace_node != .none) |
| 1395 | .@"addrspace" |
| 1396 | else if (ptr_type.ast.align_node != .none) |
| 1397 | .@"align" |
| 1398 | else if (ptr_type.allowzero_token != null) |
| 1399 | .@"allowzero" |
| 1400 | else |
| 1401 | .none; |
| 1402 | const final_qual_space: Space = if (tree.tokenTag(tree.firstToken(ptr_type.ast.child_type)) != |
| 1403 | .multiline_string_literal_line) .space else .none; |
| 1404 | |
| 1000 | 1405 | if (ptr_type.allowzero_token) |allowzero_token| { |
| 1001 | | try renderToken(r, allowzero_token, .space); |
| 1406 | const this_space: Space = if (final_qual == .@"allowzero") final_qual_space else .space; |
| 1407 | try renderToken(r, allowzero_token, this_space); |
| 1002 | 1408 | } |
| 1003 | 1409 | |
| 1004 | 1410 | if (ptr_type.ast.align_node.unwrap()) |align_node| { |
| 1411 | const this_space: Space = if (final_qual == .@"align") final_qual_space else .space; |
| 1005 | 1412 | const align_first = tree.firstToken(align_node); |
| 1006 | 1413 | try renderToken(r, align_first - 2, .none); // align |
| 1007 | 1414 | try renderToken(r, align_first - 1, .none); // lparen |
| ... | ... | @@ -1012,26 +1419,29 @@ fn renderPtrType(r: *Render, ptr_type: Ast.full.PtrType, space: Space) Error!voi |
| 1012 | 1419 | try renderExpression(r, bit_range_start, .none); |
| 1013 | 1420 | try renderToken(r, tree.firstToken(bit_range_end) - 1, .none); // colon |
| 1014 | 1421 | try renderExpression(r, bit_range_end, .none); |
| 1015 | | try renderToken(r, tree.lastToken(bit_range_end) + 1, .space); // rparen |
| 1422 | try renderToken(r, tree.lastToken(bit_range_end) + 1, this_space); // rparen |
| 1016 | 1423 | } else { |
| 1017 | | try renderToken(r, tree.lastToken(align_node) + 1, .space); // rparen |
| 1424 | try renderToken(r, tree.lastToken(align_node) + 1, this_space); // rparen |
| 1018 | 1425 | } |
| 1019 | 1426 | } |
| 1020 | 1427 | |
| 1021 | 1428 | if (ptr_type.ast.addrspace_node.unwrap()) |addrspace_node| { |
| 1429 | const this_space: Space = if (final_qual == .@"addrspace") final_qual_space else .space; |
| 1022 | 1430 | const addrspace_first = tree.firstToken(addrspace_node); |
| 1023 | 1431 | try renderToken(r, addrspace_first - 2, .none); // addrspace |
| 1024 | 1432 | try renderToken(r, addrspace_first - 1, .none); // lparen |
| 1025 | 1433 | try renderExpression(r, addrspace_node, .none); |
| 1026 | | try renderToken(r, tree.lastToken(addrspace_node) + 1, .space); // rparen |
| 1434 | try renderToken(r, tree.lastToken(addrspace_node) + 1, this_space); // rparen |
| 1027 | 1435 | } |
| 1028 | 1436 | |
| 1029 | 1437 | if (ptr_type.const_token) |const_token| { |
| 1030 | | try renderToken(r, const_token, .space); |
| 1438 | const this_space: Space = if (final_qual == .@"const") final_qual_space else .space; |
| 1439 | try renderToken(r, const_token, this_space); |
| 1031 | 1440 | } |
| 1032 | 1441 | |
| 1033 | 1442 | if (ptr_type.volatile_token) |volatile_token| { |
| 1034 | | try renderToken(r, volatile_token, .space); |
| 1443 | const this_space: Space = if (final_qual == .@"volatile") final_qual_space else unreachable; |
| 1444 | try renderToken(r, volatile_token, this_space); |
| 1035 | 1445 | } |
| 1036 | 1446 | |
| 1037 | 1447 | try renderExpression(r, ptr_type.ast.child_type, space); |
| ... | ... | @@ -1044,12 +1454,14 @@ fn renderSlice( |
| 1044 | 1454 | space: Space, |
| 1045 | 1455 | ) Error!void { |
| 1046 | 1456 | const tree = r.tree; |
| 1047 | | const after_start_space_bool = nodeCausesSliceOpSpace(tree.nodeTag(slice.ast.start)) or |
| 1457 | const space_around_dots = nodeCausesSliceOpSpace(tree.nodeTag(slice.ast.start)) or |
| 1048 | 1458 | if (slice.ast.end.unwrap()) |end| nodeCausesSliceOpSpace(tree.nodeTag(end)) else false; |
| 1049 | | const after_start_space = if (after_start_space_bool) Space.space else Space.none; |
| 1050 | | const after_dots_space = if (slice.ast.end != .none) |
| 1051 | | after_start_space |
| 1052 | | else if (slice.ast.sentinel != .none) Space.space else Space.none; |
| 1459 | const after_start_space: Space = if (space_around_dots) .space else .none; |
| 1460 | const before_sentinel_space: Space = if (slice.ast.sentinel != .none) .space else .none; |
| 1461 | const after_dots_space: Space = if (slice.ast.end != .none) |
| 1462 | if (space_around_dots) .maybe_space else .none |
| 1463 | else |
| 1464 | before_sentinel_space; |
| 1053 | 1465 | |
| 1054 | 1466 | try renderExpression(r, slice.ast.sliced, .none); |
| 1055 | 1467 | try renderToken(r, slice.ast.lbracket, .none); // lbracket |
| ... | ... | @@ -1059,8 +1471,7 @@ fn renderSlice( |
| 1059 | 1471 | try renderToken(r, start_last + 1, after_dots_space); // ellipsis2 ("..") |
| 1060 | 1472 | |
| 1061 | 1473 | if (slice.ast.end.unwrap()) |end| { |
| 1062 | | const after_end_space = if (slice.ast.sentinel != .none) Space.space else Space.none; |
| 1063 | | try renderExpression(r, end, after_end_space); |
| 1474 | try renderExpression(r, end, before_sentinel_space); |
| 1064 | 1475 | } |
| 1065 | 1476 | |
| 1066 | 1477 | if (slice.ast.sentinel.unwrap()) |sentinel| { |
| ... | ... | @@ -1088,7 +1499,7 @@ fn renderAsmOutput( |
| 1088 | 1499 | |
| 1089 | 1500 | if (tree.tokenTag(symbolic_name + 4) == .arrow) { |
| 1090 | 1501 | const type_expr, const rparen = tree.nodeData(asm_output).opt_node_and_token; |
| 1091 | | try renderToken(r, symbolic_name + 4, .space); // -> |
| 1502 | try renderToken(r, symbolic_name + 4, .maybe_space); // -> |
| 1092 | 1503 | try renderExpression(r, type_expr.unwrap().?, Space.none); |
| 1093 | 1504 | return renderToken(r, rparen, space); |
| 1094 | 1505 | } else { |
| ... | ... | @@ -1169,33 +1580,38 @@ fn renderVarDeclWithoutFixups( |
| 1169 | 1580 | |
| 1170 | 1581 | try renderToken(r, var_decl.ast.mut_token, .space); // var |
| 1171 | 1582 | |
| 1172 | | if (var_decl.ast.type_node != .none or var_decl.ast.align_node != .none or |
| 1173 | | var_decl.ast.addrspace_node != .none or var_decl.ast.section_node != .none or |
| 1174 | | var_decl.ast.init_node != .none) |
| 1175 | | { |
| 1176 | | const name_space = if (var_decl.ast.type_node == .none and |
| 1177 | | (var_decl.ast.align_node != .none or |
| 1178 | | var_decl.ast.addrspace_node != .none or |
| 1179 | | var_decl.ast.section_node != .none or |
| 1180 | | var_decl.ast.init_node != .none)) |
| 1181 | | Space.space |
| 1182 | | else |
| 1183 | | Space.none; |
| 1184 | | |
| 1185 | | try renderIdentifier(r, var_decl.ast.mut_token + 1, name_space, .preserve_when_shadowing); // name |
| 1186 | | } else { |
| 1187 | | return renderIdentifier(r, var_decl.ast.mut_token + 1, space, .preserve_when_shadowing); // name |
| 1583 | const last_component: enum { |
| 1584 | value, |
| 1585 | @"linksection", |
| 1586 | @"addrspace", |
| 1587 | @"align", |
| 1588 | type, |
| 1589 | identifier, |
| 1590 | } = if (var_decl.ast.init_node != .none) |
| 1591 | .value |
| 1592 | else if (var_decl.ast.section_node != .none) |
| 1593 | .@"linksection" |
| 1594 | else if (var_decl.ast.addrspace_node != .none) |
| 1595 | .@"addrspace" |
| 1596 | else if (var_decl.ast.align_node != .none) |
| 1597 | .@"align" |
| 1598 | else if (var_decl.ast.type_node != .none) |
| 1599 | .type |
| 1600 | else |
| 1601 | .identifier; |
| 1602 | |
| 1603 | if (last_component == .identifier) { |
| 1604 | return renderIdentifier(r, var_decl.ast.mut_token + 1, space, .preserve_when_shadowing); |
| 1188 | 1605 | } |
| 1606 | const after_ident_space: Space = if (var_decl.ast.type_node != .none) .none else .space; |
| 1607 | try renderIdentifier(r, var_decl.ast.mut_token + 1, after_ident_space, .preserve_when_shadowing); |
| 1189 | 1608 | |
| 1190 | 1609 | if (var_decl.ast.type_node.unwrap()) |type_node| { |
| 1191 | | try renderToken(r, var_decl.ast.mut_token + 2, Space.space); // : |
| 1192 | | if (var_decl.ast.align_node != .none or var_decl.ast.addrspace_node != .none or |
| 1193 | | var_decl.ast.section_node != .none or var_decl.ast.init_node != .none) |
| 1194 | | { |
| 1195 | | try renderExpression(r, type_node, .space); |
| 1196 | | } else { |
| 1610 | try renderToken(r, var_decl.ast.mut_token + 2, .maybe_space); // : |
| 1611 | if (last_component == .type) { |
| 1197 | 1612 | return renderExpression(r, type_node, space); |
| 1198 | 1613 | } |
| 1614 | try renderExpression(r, type_node, .space); |
| 1199 | 1615 | } |
| 1200 | 1616 | |
| 1201 | 1617 | if (var_decl.ast.align_node.unwrap()) |align_node| { |
| ... | ... | @@ -1205,13 +1621,10 @@ fn renderVarDeclWithoutFixups( |
| 1205 | 1621 | try renderToken(r, align_kw, Space.none); // align |
| 1206 | 1622 | try renderToken(r, lparen, Space.none); // ( |
| 1207 | 1623 | try renderExpression(r, align_node, Space.none); |
| 1208 | | if (var_decl.ast.addrspace_node != .none or var_decl.ast.section_node != .none or |
| 1209 | | var_decl.ast.init_node != .none) |
| 1210 | | { |
| 1211 | | try renderToken(r, rparen, .space); // ) |
| 1212 | | } else { |
| 1624 | if (last_component == .@"align") { |
| 1213 | 1625 | return renderToken(r, rparen, space); // ) |
| 1214 | 1626 | } |
| 1627 | try renderToken(r, rparen, .space); // ) |
| 1215 | 1628 | } |
| 1216 | 1629 | |
| 1217 | 1630 | if (var_decl.ast.addrspace_node.unwrap()) |addrspace_node| { |
| ... | ... | @@ -1221,12 +1634,10 @@ fn renderVarDeclWithoutFixups( |
| 1221 | 1634 | try renderToken(r, addrspace_kw, Space.none); // addrspace |
| 1222 | 1635 | try renderToken(r, lparen, Space.none); // ( |
| 1223 | 1636 | try renderExpression(r, addrspace_node, Space.none); |
| 1224 | | if (var_decl.ast.section_node != .none or var_decl.ast.init_node != .none) { |
| 1225 | | try renderToken(r, rparen, .space); // ) |
| 1226 | | } else { |
| 1227 | | try renderToken(r, rparen, .none); // ) |
| 1228 | | return renderToken(r, rparen + 1, Space.newline); // ; |
| 1637 | if (last_component == .@"addrspace") { |
| 1638 | return renderToken(r, rparen, space); // ) |
| 1229 | 1639 | } |
| 1640 | try renderToken(r, rparen, .space); // ) |
| 1230 | 1641 | } |
| 1231 | 1642 | |
| 1232 | 1643 | if (var_decl.ast.section_node.unwrap()) |section_node| { |
| ... | ... | @@ -1236,17 +1647,19 @@ fn renderVarDeclWithoutFixups( |
| 1236 | 1647 | try renderToken(r, section_kw, Space.none); // linksection |
| 1237 | 1648 | try renderToken(r, lparen, Space.none); // ( |
| 1238 | 1649 | try renderExpression(r, section_node, Space.none); |
| 1239 | | if (var_decl.ast.init_node != .none) { |
| 1240 | | try renderToken(r, rparen, .space); // ) |
| 1241 | | } else { |
| 1650 | if (last_component == .@"linksection") { |
| 1242 | 1651 | return renderToken(r, rparen, space); // ) |
| 1243 | 1652 | } |
| 1653 | try renderToken(r, rparen, .space); // ) |
| 1244 | 1654 | } |
| 1245 | 1655 | |
| 1656 | assert(last_component == .value); |
| 1246 | 1657 | const init_node = var_decl.ast.init_node.unwrap().?; |
| 1247 | 1658 | |
| 1248 | 1659 | const eq_token = tree.firstToken(init_node) - 1; |
| 1249 | | const eq_space: Space = if (tree.tokensOnSameLine(eq_token, eq_token + 1)) .space else .newline; |
| 1660 | const rhs_seperate_line = !tree.tokensOnSameLine(eq_token, eq_token + 1) or |
| 1661 | tree.tokenTag(eq_token + 1) == .multiline_string_literal_line; |
| 1662 | const eq_space: Space = if (rhs_seperate_line) .newline else .space; |
| 1250 | 1663 | try ais.pushIndent(.after_equals); |
| 1251 | 1664 | try renderToken(r, eq_token, eq_space); // = |
| 1252 | 1665 | try renderExpression(r, init_node, space); // ; |
| ... | ... | @@ -1346,8 +1759,10 @@ fn renderThenElse( |
| 1346 | 1759 | const tree = r.tree; |
| 1347 | 1760 | const ais = r.ais; |
| 1348 | 1761 | const then_expr_is_block = nodeIsBlock(tree.nodeTag(then_expr)); |
| 1762 | const then_expr_first_token = tree.firstToken(then_expr); |
| 1349 | 1763 | const indent_then_expr = !then_expr_is_block and |
| 1350 | | !tree.tokensOnSameLine(last_prefix_token, tree.firstToken(then_expr)); |
| 1764 | (!tree.tokensOnSameLine(last_prefix_token, then_expr_first_token) or |
| 1765 | tree.tokenTag(then_expr_first_token) == .multiline_string_literal_line); |
| 1351 | 1766 | |
| 1352 | 1767 | if (indent_then_expr) try ais.pushIndent(.normal); |
| 1353 | 1768 | |
| ... | ... | @@ -1381,7 +1796,9 @@ fn renderThenElse( |
| 1381 | 1796 | |
| 1382 | 1797 | const indent_else_expr = indent_then_expr and |
| 1383 | 1798 | !nodeIsBlock(tree.nodeTag(else_expr)) and |
| 1384 | | !nodeIsIfForWhileSwitch(tree.nodeTag(else_expr)); |
| 1799 | !nodeIsIfForWhileSwitch(tree.nodeTag(else_expr)) or |
| 1800 | tree.tokenTag(tree.firstToken(else_expr)) == |
| 1801 | .multiline_string_literal_line; |
| 1385 | 1802 | if (indent_else_expr) { |
| 1386 | 1803 | try ais.pushIndent(.normal); |
| 1387 | 1804 | try renderToken(r, last_else_token, .newline); |
| ... | ... | @@ -1417,44 +1834,29 @@ fn renderFor(r: *Render, for_node: Ast.full.For, space: Space) Error!void { |
| 1417 | 1834 | try renderParamList(r, lparen, for_node.ast.inputs, .space); |
| 1418 | 1835 | |
| 1419 | 1836 | var cur = for_node.payload_token; |
| 1420 | | const pipe = std.mem.findScalarPos(std.zig.Token.Tag, token_tags, cur, .pipe).?; |
| 1421 | | if (tree.tokenTag(@intCast(pipe - 1)) == .comma) { |
| 1837 | const pipe = std.mem.findScalarPos(Token.Tag, token_tags, cur, .pipe).?; |
| 1838 | const capture_trailing_comma = token_tags[@intCast(pipe - 1)] == .comma; |
| 1839 | |
| 1840 | if (capture_trailing_comma) |
| 1422 | 1841 | try ais.pushIndent(.normal); |
| 1423 | | try renderToken(r, cur - 1, .newline); // | |
| 1424 | | while (true) { |
| 1425 | | if (tree.tokenTag(cur) == .asterisk) { |
| 1426 | | try renderToken(r, cur, .none); // * |
| 1427 | | cur += 1; |
| 1428 | | } |
| 1429 | | try renderIdentifier(r, cur, .none, .preserve_when_shadowing); // identifier |
| 1842 | try renderToken(r, cur - 1, if (capture_trailing_comma) .newline else .none); // | |
| 1843 | while (true) { |
| 1844 | if (token_tags[cur] == .asterisk) { |
| 1845 | try renderToken(r, cur, .none); // * |
| 1430 | 1846 | cur += 1; |
| 1431 | | if (tree.tokenTag(cur) == .comma) { |
| 1432 | | try renderToken(r, cur, .newline); // , |
| 1433 | | cur += 1; |
| 1434 | | } |
| 1435 | | if (tree.tokenTag(cur) == .pipe) { |
| 1436 | | break; |
| 1437 | | } |
| 1438 | 1847 | } |
| 1439 | | ais.popIndent(); |
| 1440 | | } else { |
| 1441 | | try renderToken(r, cur - 1, .none); // | |
| 1442 | | while (true) { |
| 1443 | | if (tree.tokenTag(cur) == .asterisk) { |
| 1444 | | try renderToken(r, cur, .none); // * |
| 1445 | | cur += 1; |
| 1446 | | } |
| 1447 | | try renderIdentifier(r, cur, .none, .preserve_when_shadowing); // identifier |
| 1848 | try renderIdentifier(r, cur, .none, .preserve_when_shadowing); // identifier |
| 1849 | cur += 1; |
| 1850 | if (token_tags[cur] == .comma) { |
| 1851 | try renderToken(r, cur, if (capture_trailing_comma) .newline else .space); // , |
| 1448 | 1852 | cur += 1; |
| 1449 | | if (tree.tokenTag(cur) == .comma) { |
| 1450 | | try renderToken(r, cur, .space); // , |
| 1451 | | cur += 1; |
| 1452 | | } |
| 1453 | | if (tree.tokenTag(cur) == .pipe) { |
| 1454 | | break; |
| 1455 | | } |
| 1853 | } |
| 1854 | if (token_tags[cur] == .pipe) { |
| 1855 | break; |
| 1456 | 1856 | } |
| 1457 | 1857 | } |
| 1858 | if (capture_trailing_comma) |
| 1859 | ais.popIndent(); |
| 1458 | 1860 | |
| 1459 | 1861 | try renderThenElse( |
| 1460 | 1862 | r, |
| ... | ... | @@ -1483,97 +1885,80 @@ fn renderContainerField( |
| 1483 | 1885 | }; |
| 1484 | 1886 | |
| 1485 | 1887 | if (field.comptime_token) |t| { |
| 1486 | | try renderToken(r, t, .space); // comptime |
| 1487 | | } |
| 1488 | | if (field.ast.type_expr == .none and field.ast.value_expr == .none) { |
| 1489 | | if (field.ast.align_expr.unwrap()) |align_expr| { |
| 1490 | | try renderIdentifier(r, field.ast.main_token, .space, quote); // name |
| 1491 | | const lparen_token = tree.firstToken(align_expr) - 1; |
| 1492 | | const align_kw = lparen_token - 1; |
| 1493 | | const rparen_token = tree.lastToken(align_expr) + 1; |
| 1494 | | try renderToken(r, align_kw, .none); // align |
| 1495 | | try renderToken(r, lparen_token, .none); // ( |
| 1496 | | try renderExpression(r, align_expr, .none); // alignment |
| 1497 | | return renderToken(r, rparen_token, .space); // ) |
| 1888 | try renderToken(r, t, .maybe_space); // comptime |
| 1889 | } |
| 1890 | |
| 1891 | const last_component: enum { |
| 1892 | value, |
| 1893 | @"align", |
| 1894 | type, |
| 1895 | identifier, |
| 1896 | } = if (field.ast.value_expr != .none) |
| 1897 | .value |
| 1898 | else if (field.ast.align_expr != .none) |
| 1899 | .@"align" |
| 1900 | else if (field.ast.type_expr != .none) |
| 1901 | .type |
| 1902 | else if (!field.ast.tuple_like) |
| 1903 | .identifier |
| 1904 | else |
| 1905 | unreachable; |
| 1906 | |
| 1907 | if (!field.ast.tuple_like) { |
| 1908 | if (last_component == .identifier) { |
| 1909 | return renderIdentifierComma(r, field.ast.main_token, space, quote); // name |
| 1498 | 1910 | } |
| 1499 | | return renderIdentifierComma(r, field.ast.main_token, space, quote); // name |
| 1911 | const this_space: Space = if (field.ast.type_expr != .none) .none else .space; |
| 1912 | try renderIdentifier(r, field.ast.main_token, this_space, quote); // name |
| 1500 | 1913 | } |
| 1501 | | if (field.ast.type_expr != .none and field.ast.value_expr == .none) { |
| 1502 | | const type_expr = field.ast.type_expr.unwrap().?; |
| 1914 | |
| 1915 | if (field.ast.type_expr.unwrap()) |type_expr| { |
| 1503 | 1916 | if (!field.ast.tuple_like) { |
| 1504 | | try renderIdentifier(r, field.ast.main_token, .none, quote); // name |
| 1505 | | try renderToken(r, field.ast.main_token + 1, .space); // : |
| 1917 | try renderToken(r, field.ast.main_token + 1, .maybe_space); // : |
| 1506 | 1918 | } |
| 1507 | 1919 | |
| 1508 | | if (field.ast.align_expr.unwrap()) |align_expr| { |
| 1509 | | try renderExpression(r, type_expr, .space); // type |
| 1510 | | const align_token = tree.firstToken(align_expr) - 2; |
| 1511 | | try renderToken(r, align_token, .none); // align |
| 1512 | | try renderToken(r, align_token + 1, .none); // ( |
| 1513 | | try renderExpression(r, align_expr, .none); // alignment |
| 1514 | | const rparen = tree.lastToken(align_expr) + 1; |
| 1515 | | return renderTokenComma(r, rparen, space); // ) |
| 1516 | | } else { |
| 1920 | if (last_component == .type) { |
| 1517 | 1921 | return renderExpressionComma(r, type_expr, space); // type |
| 1518 | 1922 | } |
| 1923 | try renderExpression(r, type_expr, .space); // type |
| 1519 | 1924 | } |
| 1520 | | if (field.ast.type_expr == .none and field.ast.value_expr != .none) { |
| 1521 | | const value_expr = field.ast.value_expr.unwrap().?; |
| 1522 | | |
| 1523 | | try renderIdentifier(r, field.ast.main_token, .space, quote); // name |
| 1524 | | if (field.ast.align_expr.unwrap()) |align_expr| { |
| 1525 | | const lparen_token = tree.firstToken(align_expr) - 1; |
| 1526 | | const align_kw = lparen_token - 1; |
| 1527 | | const rparen_token = tree.lastToken(align_expr) + 1; |
| 1528 | | try renderToken(r, align_kw, .none); // align |
| 1529 | | try renderToken(r, lparen_token, .none); // ( |
| 1530 | | try renderExpression(r, align_expr, .none); // alignment |
| 1531 | | try renderToken(r, rparen_token, .space); // ) |
| 1532 | | } |
| 1533 | | try renderToken(r, field.ast.main_token + 1, .space); // = |
| 1534 | | return renderExpressionComma(r, value_expr, space); // value |
| 1535 | | } |
| 1536 | | if (!field.ast.tuple_like) { |
| 1537 | | try renderIdentifier(r, field.ast.main_token, .none, quote); // name |
| 1538 | | try renderToken(r, field.ast.main_token + 1, .space); // : |
| 1539 | | } |
| 1540 | | |
| 1541 | | const type_expr = field.ast.type_expr.unwrap().?; |
| 1542 | | const value_expr = field.ast.value_expr.unwrap().?; |
| 1543 | | |
| 1544 | | try renderExpression(r, type_expr, .space); // type |
| 1545 | 1925 | |
| 1546 | 1926 | if (field.ast.align_expr.unwrap()) |align_expr| { |
| 1547 | | const lparen_token = tree.firstToken(align_expr) - 1; |
| 1548 | | const align_kw = lparen_token - 1; |
| 1549 | | const rparen_token = tree.lastToken(align_expr) + 1; |
| 1550 | | try renderToken(r, align_kw, .none); // align |
| 1551 | | try renderToken(r, lparen_token, .none); // ( |
| 1927 | const align_token = tree.firstToken(align_expr) - 2; |
| 1928 | try renderToken(r, align_token, .none); // align |
| 1929 | try renderToken(r, align_token + 1, .none); // ( |
| 1552 | 1930 | try renderExpression(r, align_expr, .none); // alignment |
| 1553 | | try renderToken(r, rparen_token, .space); // ) |
| 1931 | const rparen = tree.lastToken(align_expr) + 1; |
| 1932 | if (last_component == .@"align") { |
| 1933 | return renderTokenComma(r, rparen, space); // ) |
| 1934 | } |
| 1935 | try renderToken(r, rparen, .space); |
| 1554 | 1936 | } |
| 1555 | | const eq_token = tree.firstToken(value_expr) - 1; |
| 1556 | | const eq_space: Space = if (tree.tokensOnSameLine(eq_token, eq_token + 1)) .space else .newline; |
| 1557 | 1937 | |
| 1558 | | try ais.pushIndent(.after_equals); |
| 1559 | | try renderToken(r, eq_token, eq_space); // = |
| 1938 | if (field.ast.value_expr.unwrap()) |value_expr| { |
| 1939 | assert(last_component == .value); |
| 1940 | const eq_token = tree.firstToken(value_expr) - 1; |
| 1941 | const seperate_line = !tree.tokensOnSameLine(eq_token, eq_token + 1) or |
| 1942 | tree.tokenTag(eq_token + 1) == .multiline_string_literal_line; |
| 1943 | const eq_space: Space = if (seperate_line) .newline else .space; |
| 1560 | 1944 | |
| 1561 | | if (eq_space == .space) { |
| 1562 | | ais.popIndent(); |
| 1563 | | try renderExpressionComma(r, value_expr, space); // value |
| 1564 | | return; |
| 1565 | | } |
| 1566 | | |
| 1567 | | const maybe_comma = tree.lastToken(value_expr) + 1; |
| 1945 | try ais.pushIndent(.after_equals); |
| 1946 | try renderToken(r, eq_token, eq_space); // = |
| 1947 | if (eq_space == .space) { |
| 1948 | ais.popIndent(); |
| 1949 | return renderExpressionComma(r, value_expr, space); // value |
| 1950 | } |
| 1568 | 1951 | |
| 1569 | | if (tree.tokenTag(maybe_comma) == .comma) { |
| 1570 | | try renderExpression(r, value_expr, .none); // value |
| 1571 | | ais.popIndent(); |
| 1572 | | try renderToken(r, maybe_comma, .newline); |
| 1573 | | } else { |
| 1574 | | try renderExpression(r, value_expr, space); // value |
| 1575 | | ais.popIndent(); |
| 1576 | | } |
| 1952 | const maybe_comma = tree.lastToken(value_expr) + 1; |
| 1953 | if (tree.tokenTag(maybe_comma) == .comma) { |
| 1954 | try renderExpression(r, value_expr, .none); // value |
| 1955 | ais.popIndent(); |
| 1956 | try renderToken(r, maybe_comma, .newline); |
| 1957 | } else { |
| 1958 | try renderExpression(r, value_expr, space); // value |
| 1959 | ais.popIndent(); |
| 1960 | } |
| 1961 | } else unreachable; |
| 1577 | 1962 | } |
| 1578 | 1963 | |
| 1579 | 1964 | fn renderBuiltinCall( |
| ... | ... | @@ -1587,14 +1972,9 @@ fn renderBuiltinCall( |
| 1587 | 1972 | |
| 1588 | 1973 | try renderToken(r, builtin_token, .none); // @name |
| 1589 | 1974 | |
| 1590 | | if (params.len == 0) { |
| 1591 | | try renderToken(r, builtin_token + 1, .none); // ( |
| 1592 | | return renderToken(r, builtin_token + 2, space); // ) |
| 1593 | | } |
| 1594 | | |
| 1595 | 1975 | if (r.fixups.rebase_imported_paths) |prefix| { |
| 1596 | 1976 | const slice = tree.tokenSlice(builtin_token); |
| 1597 | | if (mem.eql(u8, slice, "@import")) f: { |
| 1977 | if (params.len != 0 and mem.eql(u8, slice, "@import")) f: { |
| 1598 | 1978 | const param = params[0]; |
| 1599 | 1979 | const str_lit_token = tree.nodeMainToken(param); |
| 1600 | 1980 | assert(tree.tokenTag(str_lit_token) == .string_literal); |
| ... | ... | @@ -1613,45 +1993,69 @@ fn renderBuiltinCall( |
| 1613 | 1993 | } |
| 1614 | 1994 | } |
| 1615 | 1995 | |
| 1616 | | const last_param = params[params.len - 1]; |
| 1617 | | const after_last_param_token = tree.lastToken(last_param) + 1; |
| 1618 | | |
| 1619 | | if (tree.tokenTag(after_last_param_token) != .comma) { |
| 1620 | | // Render all on one line, no trailing comma. |
| 1621 | | try renderToken(r, builtin_token + 1, .none); // ( |
| 1622 | | |
| 1623 | | for (params, 0..) |param_node, i| { |
| 1624 | | const first_param_token = tree.firstToken(param_node); |
| 1625 | | if (tree.tokenTag(first_param_token) == .multiline_string_literal_line or |
| 1626 | | hasSameLineComment(tree, first_param_token - 1)) |
| 1627 | | { |
| 1628 | | try ais.pushIndent(.normal); |
| 1629 | | try renderExpression(r, param_node, .none); |
| 1630 | | ais.popIndent(); |
| 1631 | | } else { |
| 1632 | | try renderExpression(r, param_node, .none); |
| 1633 | | } |
| 1996 | return renderParamList(r, builtin_token + 1, params, space); |
| 1997 | } |
| 1634 | 1998 | |
| 1635 | | if (i + 1 < params.len) { |
| 1636 | | const comma_token = tree.lastToken(param_node) + 1; |
| 1637 | | try renderToken(r, comma_token, .space); // , |
| 1638 | | } |
| 1999 | fn fnProtoRparen(tree: Ast, fn_proto: Ast.full.FnProto, maybe_bang: Ast.TokenIndex) Ast.TokenIndex { |
| 2000 | // These may appear in any order, so we have to check the token_starts array |
| 2001 | // to find out which is first. |
| 2002 | var rparen = if (tree.tokenTag(maybe_bang) == .bang) maybe_bang - 1 else maybe_bang; |
| 2003 | var smallest_start = tree.tokenStart(maybe_bang); |
| 2004 | if (fn_proto.ast.align_expr.unwrap()) |align_expr| { |
| 2005 | const tok = tree.firstToken(align_expr) - 3; |
| 2006 | const start = tree.tokenStart(tok); |
| 2007 | if (start < smallest_start) { |
| 2008 | rparen = tok; |
| 2009 | smallest_start = start; |
| 1639 | 2010 | } |
| 1640 | | return renderToken(r, after_last_param_token, space); // ) |
| 1641 | | } else { |
| 1642 | | // Render one param per line. |
| 1643 | | try ais.pushIndent(.normal); |
| 1644 | | try renderToken(r, builtin_token + 1, Space.newline); // ( |
| 1645 | | |
| 1646 | | for (params) |param_node| { |
| 1647 | | try ais.pushSpace(.comma); |
| 1648 | | try renderExpression(r, param_node, .comma); |
| 1649 | | ais.popSpace(); |
| 2011 | } |
| 2012 | if (fn_proto.ast.addrspace_expr.unwrap()) |addrspace_expr| { |
| 2013 | const tok = tree.firstToken(addrspace_expr) - 3; |
| 2014 | const start = tree.tokenStart(tok); |
| 2015 | if (start < smallest_start) { |
| 2016 | rparen = tok; |
| 2017 | smallest_start = start; |
| 2018 | } |
| 2019 | } |
| 2020 | if (fn_proto.ast.section_expr.unwrap()) |section_expr| { |
| 2021 | const tok = tree.firstToken(section_expr) - 3; |
| 2022 | const start = tree.tokenStart(tok); |
| 2023 | if (start < smallest_start) { |
| 2024 | rparen = tok; |
| 2025 | smallest_start = start; |
| 2026 | } |
| 2027 | } |
| 2028 | if (fn_proto.ast.callconv_expr.unwrap()) |callconv_expr| { |
| 2029 | const tok = tree.firstToken(callconv_expr) - 3; |
| 2030 | const start = tree.tokenStart(tok); |
| 2031 | if (start < smallest_start) { |
| 2032 | rparen = tok; |
| 2033 | smallest_start = start; |
| 1650 | 2034 | } |
| 1651 | | ais.popIndent(); |
| 1652 | | |
| 1653 | | return renderToken(r, after_last_param_token + 1, space); // ) |
| 1654 | 2035 | } |
| 2036 | assert(tree.tokenTag(rparen) == .r_paren); |
| 2037 | return rparen; |
| 2038 | } |
| 2039 | |
| 2040 | fn isOneLineFnProto( |
| 2041 | tree: Ast, |
| 2042 | fn_proto: Ast.full.FnProto, |
| 2043 | lparen: Ast.TokenIndex, |
| 2044 | rparen: Ast.TokenIndex, |
| 2045 | ) bool { |
| 2046 | const trailing_comma = tree.tokenTag(rparen - 1) == .comma; |
| 2047 | if (trailing_comma or hasComment(tree, lparen, rparen)) |
| 2048 | return false; |
| 2049 | |
| 2050 | // Check that there are no doc comments |
| 2051 | var after_last_param = lparen + 1; |
| 2052 | for (fn_proto.ast.params) |expr| { |
| 2053 | // Looking before each param is insufficient since anytype is not included in `params` |
| 2054 | if (hasDocComment(tree, after_last_param, tree.firstToken(expr))) |
| 2055 | return false; |
| 2056 | after_last_param = tree.lastToken(expr) + 1; |
| 2057 | } |
| 2058 | return !hasDocComment(tree, after_last_param, rparen); |
| 1655 | 2059 | } |
| 1656 | 2060 | |
| 1657 | 2061 | fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!void { |
| ... | ... | @@ -1671,51 +2075,11 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1671 | 2075 | |
| 1672 | 2076 | const return_type = fn_proto.ast.return_type.unwrap().?; |
| 1673 | 2077 | const maybe_bang = tree.firstToken(return_type) - 1; |
| 1674 | | const rparen = blk: { |
| 1675 | | // These may appear in any order, so we have to check the token_starts array |
| 1676 | | // to find out which is first. |
| 1677 | | var rparen = if (tree.tokenTag(maybe_bang) == .bang) maybe_bang - 1 else maybe_bang; |
| 1678 | | var smallest_start = tree.tokenStart(maybe_bang); |
| 1679 | | if (fn_proto.ast.align_expr.unwrap()) |align_expr| { |
| 1680 | | const tok = tree.firstToken(align_expr) - 3; |
| 1681 | | const start = tree.tokenStart(tok); |
| 1682 | | if (start < smallest_start) { |
| 1683 | | rparen = tok; |
| 1684 | | smallest_start = start; |
| 1685 | | } |
| 1686 | | } |
| 1687 | | if (fn_proto.ast.addrspace_expr.unwrap()) |addrspace_expr| { |
| 1688 | | const tok = tree.firstToken(addrspace_expr) - 3; |
| 1689 | | const start = tree.tokenStart(tok); |
| 1690 | | if (start < smallest_start) { |
| 1691 | | rparen = tok; |
| 1692 | | smallest_start = start; |
| 1693 | | } |
| 1694 | | } |
| 1695 | | if (fn_proto.ast.section_expr.unwrap()) |section_expr| { |
| 1696 | | const tok = tree.firstToken(section_expr) - 3; |
| 1697 | | const start = tree.tokenStart(tok); |
| 1698 | | if (start < smallest_start) { |
| 1699 | | rparen = tok; |
| 1700 | | smallest_start = start; |
| 1701 | | } |
| 1702 | | } |
| 1703 | | if (fn_proto.ast.callconv_expr.unwrap()) |callconv_expr| { |
| 1704 | | const tok = tree.firstToken(callconv_expr) - 3; |
| 1705 | | const start = tree.tokenStart(tok); |
| 1706 | | if (start < smallest_start) { |
| 1707 | | rparen = tok; |
| 1708 | | smallest_start = start; |
| 1709 | | } |
| 1710 | | } |
| 1711 | | break :blk rparen; |
| 1712 | | }; |
| 1713 | | assert(tree.tokenTag(rparen) == .r_paren); |
| 2078 | const rparen = fnProtoRparen(tree, fn_proto, maybe_bang); |
| 1714 | 2079 | |
| 1715 | 2080 | // The params list is a sparse set that does *not* include anytype or ... parameters. |
| 1716 | 2081 | |
| 1717 | | const trailing_comma = tree.tokenTag(rparen - 1) == .comma; |
| 1718 | | if (!trailing_comma and !hasComment(tree, lparen, rparen)) { |
| 2082 | if (isOneLineFnProto(tree, fn_proto, lparen, rparen)) { |
| 1719 | 2083 | // Render all on one line, no trailing comma. |
| 1720 | 2084 | try renderToken(r, lparen, .none); // ( |
| 1721 | 2085 | |
| ... | ... | @@ -1724,10 +2088,7 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1724 | 2088 | while (true) { |
| 1725 | 2089 | last_param_token += 1; |
| 1726 | 2090 | switch (tree.tokenTag(last_param_token)) { |
| 1727 | | .doc_comment => { |
| 1728 | | try renderToken(r, last_param_token, .newline); |
| 1729 | | continue; |
| 1730 | | }, |
| 2091 | .doc_comment => unreachable, |
| 1731 | 2092 | .ellipsis3 => { |
| 1732 | 2093 | try renderToken(r, last_param_token, .none); // ... |
| 1733 | 2094 | break; |
| ... | ... | @@ -1743,7 +2104,7 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1743 | 2104 | }, |
| 1744 | 2105 | .r_paren => break, |
| 1745 | 2106 | .comma => { |
| 1746 | | try renderToken(r, last_param_token, .space); // , |
| 2107 | try renderToken(r, last_param_token, .maybe_space); // , |
| 1747 | 2108 | continue; |
| 1748 | 2109 | }, |
| 1749 | 2110 | else => {}, // Parameter type without a name. |
| ... | ... | @@ -1753,7 +2114,7 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1753 | 2114 | { |
| 1754 | 2115 | try renderIdentifier(r, last_param_token, .none, .preserve_when_shadowing); // name |
| 1755 | 2116 | last_param_token = last_param_token + 1; |
| 1756 | | try renderToken(r, last_param_token, .space); // : |
| 2117 | try renderToken(r, last_param_token, .maybe_space); // : |
| 1757 | 2118 | last_param_token += 1; |
| 1758 | 2119 | } |
| 1759 | 2120 | if (tree.tokenTag(last_param_token) == .keyword_anytype) { |
| ... | ... | @@ -1822,7 +2183,7 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1822 | 2183 | ais.popIndent(); |
| 1823 | 2184 | } |
| 1824 | 2185 | |
| 1825 | | try renderToken(r, rparen, .space); // ) |
| 2186 | try renderToken(r, rparen, .maybe_space); // ) |
| 1826 | 2187 | |
| 1827 | 2188 | if (fn_proto.ast.align_expr.unwrap()) |align_expr| { |
| 1828 | 2189 | const align_lparen = tree.firstToken(align_expr) - 1; |
| ... | ... | @@ -1831,7 +2192,7 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1831 | 2192 | try renderToken(r, align_lparen - 1, .none); // align |
| 1832 | 2193 | try renderToken(r, align_lparen, .none); // ( |
| 1833 | 2194 | try renderExpression(r, align_expr, .none); |
| 1834 | | try renderToken(r, align_rparen, .space); // ) |
| 2195 | try renderToken(r, align_rparen, .maybe_space); // ) |
| 1835 | 2196 | } |
| 1836 | 2197 | |
| 1837 | 2198 | if (fn_proto.ast.addrspace_expr.unwrap()) |addrspace_expr| { |
| ... | ... | @@ -1841,7 +2202,7 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1841 | 2202 | try renderToken(r, align_lparen - 1, .none); // addrspace |
| 1842 | 2203 | try renderToken(r, align_lparen, .none); // ( |
| 1843 | 2204 | try renderExpression(r, addrspace_expr, .none); |
| 1844 | | try renderToken(r, align_rparen, .space); // ) |
| 2205 | try renderToken(r, align_rparen, .maybe_space); // ) |
| 1845 | 2206 | } |
| 1846 | 2207 | |
| 1847 | 2208 | if (fn_proto.ast.section_expr.unwrap()) |section_expr| { |
| ... | ... | @@ -1851,7 +2212,7 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1851 | 2212 | try renderToken(r, section_lparen - 1, .none); // section |
| 1852 | 2213 | try renderToken(r, section_lparen, .none); // ( |
| 1853 | 2214 | try renderExpression(r, section_expr, .none); |
| 1854 | | try renderToken(r, section_rparen, .space); // ) |
| 2215 | try renderToken(r, section_rparen, .maybe_space); // ) |
| 1855 | 2216 | } |
| 1856 | 2217 | |
| 1857 | 2218 | if (fn_proto.ast.callconv_expr.unwrap()) |callconv_expr| { |
| ... | ... | @@ -1865,7 +2226,7 @@ fn renderFnProto(r: *Render, fn_proto: Ast.full.FnProto, space: Space) Error!voi |
| 1865 | 2226 | try renderToken(r, callconv_lparen - 1, .none); // callconv |
| 1866 | 2227 | try renderToken(r, callconv_lparen, .none); // ( |
| 1867 | 2228 | try renderExpression(r, callconv_expr, .none); |
| 1868 | | try renderToken(r, callconv_rparen, .space); // ) |
| 2229 | try renderToken(r, callconv_rparen, .maybe_space); // ) |
| 1869 | 2230 | } |
| 1870 | 2231 | } |
| 1871 | 2232 | |
| ... | ... | @@ -1890,7 +2251,7 @@ fn renderSwitchCase( |
| 1890 | 2251 | |
| 1891 | 2252 | // render inline keyword |
| 1892 | 2253 | if (switch_case.inline_token) |some| { |
| 1893 | | try renderToken(r, some, .space); |
| 2254 | try renderToken(r, some, .maybe_space); |
| 1894 | 2255 | } |
| 1895 | 2256 | |
| 1896 | 2257 | // Render everything before the arrow |
| ... | ... | @@ -1904,33 +2265,26 @@ fn renderSwitchCase( |
| 1904 | 2265 | } else { |
| 1905 | 2266 | // Render on one line |
| 1906 | 2267 | for (switch_case.ast.values) |value_expr| { |
| 1907 | | try renderExpression(r, value_expr, .comma_space); |
| 2268 | try renderExpression(r, value_expr, .comma_maybe_space); |
| 1908 | 2269 | } |
| 1909 | 2270 | } |
| 1910 | 2271 | |
| 1911 | | // Render the arrow and everything after it |
| 1912 | | const pre_target_space = if (tree.nodeTag(switch_case.ast.target_expr) == .multiline_string_literal) |
| 1913 | | // Newline gets inserted when rendering the target expr. |
| 1914 | | Space.none |
| 1915 | | else |
| 1916 | | Space.space; |
| 1917 | | const after_arrow_space: Space = if (switch_case.payload_token == null) pre_target_space else .space; |
| 1918 | | try renderToken(r, switch_case.ast.arrow_token, after_arrow_space); // => |
| 2272 | try renderToken(r, switch_case.ast.arrow_token, .maybe_space); // => |
| 1919 | 2273 | |
| 1920 | 2274 | if (switch_case.payload_token) |payload_token| { |
| 1921 | 2275 | try renderToken(r, payload_token - 1, .none); // pipe |
| 1922 | | const ident = payload_token + @intFromBool(tree.tokenTag(payload_token) == .asterisk); |
| 1923 | | if (tree.tokenTag(payload_token) == .asterisk) { |
| 2276 | var ident = payload_token; |
| 2277 | if (tree.tokenTag(ident) == .asterisk) { |
| 1924 | 2278 | try renderToken(r, payload_token, .none); // asterisk |
| 2279 | ident += 1; |
| 1925 | 2280 | } |
| 1926 | 2281 | try renderIdentifier(r, ident, .none, .preserve_when_shadowing); // identifier |
| 1927 | 2282 | if (tree.tokenTag(ident + 1) == .comma) { |
| 1928 | | try renderToken(r, ident + 1, .space); // , |
| 1929 | | try renderIdentifier(r, ident + 2, .none, .preserve_when_shadowing); // identifier |
| 1930 | | try renderToken(r, ident + 3, pre_target_space); // pipe |
| 1931 | | } else { |
| 1932 | | try renderToken(r, ident + 1, pre_target_space); // pipe |
| 2283 | ident += 2; |
| 2284 | try renderToken(r, ident - 1, .space); // , |
| 2285 | try renderIdentifier(r, ident, .none, .preserve_when_shadowing); // identifier |
| 1933 | 2286 | } |
| 2287 | try renderToken(r, ident + 1, .maybe_space); // pipe |
| 1934 | 2288 | } |
| 1935 | 2289 | |
| 1936 | 2290 | try renderExpression(r, switch_case.ast.target_expr, space); |
| ... | ... | @@ -2018,26 +2372,13 @@ fn renderStructInit( |
| 2018 | 2372 | try ais.pushIndent(.normal); |
| 2019 | 2373 | try renderToken(r, struct_init.ast.lbrace, .newline); |
| 2020 | 2374 | |
| 2021 | | try renderToken(r, struct_init.ast.lbrace + 1, .none); // . |
| 2022 | | try renderIdentifier(r, struct_init.ast.lbrace + 2, .space, .eagerly_unquote); // name |
| 2023 | | // Don't output a space after the = if expression is a multiline string, |
| 2024 | | // since then it will start on the next line. |
| 2025 | | const field_node = struct_init.ast.fields[0]; |
| 2026 | | const expr = tree.nodeTag(field_node); |
| 2027 | | var space_after_equal: Space = if (expr == .multiline_string_literal) .none else .space; |
| 2028 | | try renderToken(r, struct_init.ast.lbrace + 3, space_after_equal); // = |
| 2029 | | |
| 2030 | | try ais.pushSpace(.comma); |
| 2031 | | try renderExpressionFixup(r, field_node, .comma); |
| 2032 | | ais.popSpace(); |
| 2033 | | |
| 2034 | | for (struct_init.ast.fields[1..]) |field_init| { |
| 2375 | for (0.., struct_init.ast.fields) |i, field_init| { |
| 2035 | 2376 | const init_token = tree.firstToken(field_init); |
| 2036 | | try renderExtraNewlineToken(r, init_token - 3); |
| 2377 | if (i != 0) |
| 2378 | try renderExtraNewlineToken(r, init_token - 3); |
| 2037 | 2379 | try renderToken(r, init_token - 3, .none); // . |
| 2038 | 2380 | try renderIdentifier(r, init_token - 2, .space, .eagerly_unquote); // name |
| 2039 | | space_after_equal = if (tree.nodeTag(field_init) == .multiline_string_literal) .none else .space; |
| 2040 | | try renderToken(r, init_token - 1, space_after_equal); // = |
| 2381 | try renderToken(r, init_token - 1, .maybe_space); // = |
| 2041 | 2382 | |
| 2042 | 2383 | try ais.pushSpace(.comma); |
| 2043 | 2384 | try renderExpressionFixup(r, field_init, .comma); |
| ... | ... | @@ -2053,8 +2394,7 @@ fn renderStructInit( |
| 2053 | 2394 | const init_token = tree.firstToken(field_init); |
| 2054 | 2395 | try renderToken(r, init_token - 3, .none); // . |
| 2055 | 2396 | try renderIdentifier(r, init_token - 2, .space, .eagerly_unquote); // name |
| 2056 | | const space_after_equal: Space = if (tree.nodeTag(field_init) == .multiline_string_literal) .none else .space; |
| 2057 | | try renderToken(r, init_token - 1, space_after_equal); // = |
| 2397 | try renderToken(r, init_token - 1, .maybe_space); // = |
| 2058 | 2398 | try renderExpressionFixup(r, field_init, .comma_space); |
| 2059 | 2399 | } |
| 2060 | 2400 | } |
| ... | ... | @@ -2122,174 +2462,176 @@ fn renderArrayInit( |
| 2122 | 2462 | |
| 2123 | 2463 | try ais.pushIndent(.normal); |
| 2124 | 2464 | try renderToken(r, array_init.ast.lbrace, .newline); |
| 2125 | | |
| 2126 | | var expr_index: usize = 0; |
| 2127 | | while (true) { |
| 2128 | | const row_size = rowSize(tree, array_init.ast.elements[expr_index..], rbrace); |
| 2129 | | const row_exprs = array_init.ast.elements[expr_index..]; |
| 2130 | | // A place to store the width of each expression and its column's maximum |
| 2131 | | const widths = try gpa.alloc(usize, row_exprs.len + row_size); |
| 2132 | | defer gpa.free(widths); |
| 2133 | | @memset(widths, 0); |
| 2134 | | |
| 2135 | | const expr_newlines = try gpa.alloc(bool, row_exprs.len); |
| 2136 | | defer gpa.free(expr_newlines); |
| 2137 | | @memset(expr_newlines, false); |
| 2138 | | |
| 2139 | | const expr_widths = widths[0..row_exprs.len]; |
| 2140 | | const column_widths = widths[row_exprs.len..]; |
| 2141 | | |
| 2142 | | // Find next row with trailing comment (if any) to end the current section. |
| 2143 | | const section_end = sec_end: { |
| 2144 | | var this_line_first_expr: usize = 0; |
| 2145 | | var this_line_size = rowSize(tree, row_exprs, rbrace); |
| 2146 | | for (row_exprs, 0..) |expr, i| { |
| 2147 | | // Ignore comment on first line of this section. |
| 2148 | | if (i == 0) continue; |
| 2149 | | const expr_last_token = tree.lastToken(expr); |
| 2150 | | if (tree.tokensOnSameLine(tree.firstToken(row_exprs[0]), expr_last_token)) |
| 2151 | | continue; |
| 2152 | | // Track start of line containing comment. |
| 2153 | | if (!tree.tokensOnSameLine(tree.firstToken(row_exprs[this_line_first_expr]), expr_last_token)) { |
| 2154 | | this_line_first_expr = i; |
| 2155 | | this_line_size = rowSize(tree, row_exprs[this_line_first_expr..], rbrace); |
| 2156 | | } |
| 2157 | | |
| 2158 | | const maybe_comma = expr_last_token + 1; |
| 2159 | | if (tree.tokenTag(maybe_comma) == .comma) { |
| 2160 | | if (hasSameLineComment(tree, maybe_comma)) |
| 2161 | | break :sec_end i - this_line_size + 1; |
| 2162 | | } |
| 2163 | | } |
| 2164 | | break :sec_end row_exprs.len; |
| 2165 | | }; |
| 2166 | | expr_index += section_end; |
| 2167 | | |
| 2168 | | const section_exprs = row_exprs[0..section_end]; |
| 2169 | | |
| 2170 | | var sub_expr_buffer: Writer.Allocating = .init(gpa); |
| 2171 | | defer sub_expr_buffer.deinit(); |
| 2172 | | |
| 2173 | | const sub_expr_buffer_starts = try gpa.alloc(usize, section_exprs.len + 1); |
| 2174 | | defer gpa.free(sub_expr_buffer_starts); |
| 2175 | | |
| 2176 | | var auto_indenting_stream: AutoIndentingStream = .init(gpa, &sub_expr_buffer.writer, indent_delta); |
| 2177 | | defer auto_indenting_stream.deinit(); |
| 2178 | | var sub_render: Render = .{ |
| 2465 | try ais.pushSpace(.comma); |
| 2466 | |
| 2467 | const expr_widths = try gpa.alloc(enum(usize) { |
| 2468 | /// The expression contains non-printable characters (e.g. unicode / newlines) |
| 2469 | /// or has formatting disabled at the start or end. |
| 2470 | nonprint = std.math.maxInt(usize), |
| 2471 | _, |
| 2472 | }, array_init.ast.elements.len); |
| 2473 | defer gpa.free(expr_widths); |
| 2474 | { |
| 2475 | var buf: Writer.Allocating = .init(gpa); |
| 2476 | defer buf.deinit(); |
| 2477 | var sub_ais: AutoIndentingStream = .init(gpa, &buf.writer, indent_delta); |
| 2478 | sub_ais.disabled_offset = ais.disabled_offset; |
| 2479 | defer sub_ais.deinit(); |
| 2480 | var sub_r: Render = .{ |
| 2179 | 2481 | .gpa = r.gpa, |
| 2180 | | .ais = &auto_indenting_stream, |
| 2482 | .ais = &sub_ais, |
| 2181 | 2483 | .tree = r.tree, |
| 2182 | 2484 | .fixups = r.fixups, |
| 2183 | 2485 | }; |
| 2184 | | |
| 2185 | | // Calculate size of columns in current section |
| 2186 | | var column_counter: usize = 0; |
| 2187 | | var single_line = true; |
| 2188 | | var contains_newline = false; |
| 2189 | | for (section_exprs, 0..) |expr, i| { |
| 2190 | | const start = sub_expr_buffer.written().len; |
| 2191 | | sub_expr_buffer_starts[i] = start; |
| 2192 | | |
| 2193 | | if (i + 1 < section_exprs.len) { |
| 2194 | | try renderExpression(&sub_render, expr, .none); |
| 2195 | | const written = sub_expr_buffer.written(); |
| 2196 | | const width = written.len - start; |
| 2197 | | const this_contains_newline = mem.findScalar(u8, written[start..], '\n') != null; |
| 2198 | | contains_newline = contains_newline or this_contains_newline; |
| 2199 | | expr_widths[i] = width; |
| 2200 | | expr_newlines[i] = this_contains_newline; |
| 2201 | | |
| 2202 | | if (!this_contains_newline) { |
| 2203 | | const column = column_counter % row_size; |
| 2204 | | column_widths[column] = @max(column_widths[column], width); |
| 2205 | | |
| 2206 | | const expr_last_token = tree.lastToken(expr) + 1; |
| 2207 | | const next_expr = section_exprs[i + 1]; |
| 2208 | | column_counter += 1; |
| 2209 | | if (!tree.tokensOnSameLine(expr_last_token, tree.firstToken(next_expr))) single_line = false; |
| 2210 | | } else { |
| 2211 | | single_line = false; |
| 2212 | | column_counter = 0; |
| 2213 | | } |
| 2486 | for (array_init.ast.elements, expr_widths) |e, *width| { |
| 2487 | const begin_disabled = sub_ais.disabled_offset != null; |
| 2488 | // `.skip` space so trailing commments aren't included |
| 2489 | try renderExpressionComma(&sub_r, e, .skip); |
| 2490 | if (!begin_disabled and sub_ais.disabled_offset == null) { |
| 2491 | const w = buf.written(); |
| 2492 | width.* = for (w) |c| { |
| 2493 | if (!std.ascii.isPrint(c)) |
| 2494 | break .nonprint; |
| 2495 | } else @enumFromInt(w.len - @intFromBool(w[w.len - 1] == ',')); |
| 2214 | 2496 | } else { |
| 2215 | | try ais.pushSpace(.comma); |
| 2216 | | try renderExpression(&sub_render, expr, .comma); |
| 2217 | | ais.popSpace(); |
| 2497 | width.* = .nonprint; |
| 2498 | } |
| 2218 | 2499 | |
| 2219 | | const written = sub_expr_buffer.written(); |
| 2220 | | const width = written.len - start - 2; |
| 2221 | | const this_contains_newline = mem.findScalar(u8, written[start .. written.len - 1], '\n') != null; |
| 2222 | | contains_newline = contains_newline or this_contains_newline; |
| 2223 | | expr_widths[i] = width; |
| 2224 | | expr_newlines[i] = contains_newline; |
| 2500 | // Write trailing comments since they may enable/disable zig fmt |
| 2501 | buf.clearRetainingCapacity(); |
| 2502 | var after_expr = tree.lastToken(e); |
| 2503 | after_expr += @intFromBool(tree.tokenTag(after_expr + 1) == .comma); |
| 2504 | try renderSpace(&sub_r, after_expr, tokenSliceForRender(tree, after_expr).len, .none); |
| 2225 | 2505 | |
| 2226 | | if (!contains_newline) { |
| 2227 | | const column = column_counter % row_size; |
| 2228 | | column_widths[column] = @max(column_widths[column], width); |
| 2229 | | } |
| 2230 | | } |
| 2506 | buf.clearRetainingCapacity(); |
| 2231 | 2507 | } |
| 2232 | | sub_expr_buffer_starts[section_exprs.len] = sub_expr_buffer.written().len; |
| 2233 | | |
| 2234 | | // Render exprs in current section. |
| 2235 | | column_counter = 0; |
| 2236 | | for (section_exprs, 0..) |expr, i| { |
| 2237 | | const start = sub_expr_buffer_starts[i]; |
| 2238 | | const end = sub_expr_buffer_starts[i + 1]; |
| 2239 | | const expr_text = sub_expr_buffer.written()[start..end]; |
| 2240 | | if (!expr_newlines[i]) { |
| 2241 | | try ais.writeAll(expr_text); |
| 2242 | | } else { |
| 2243 | | var by_line = std.mem.splitScalar(u8, expr_text, '\n'); |
| 2244 | | var last_line_was_empty = false; |
| 2245 | | try ais.writeAll(by_line.first()); |
| 2246 | | while (by_line.next()) |line| { |
| 2247 | | if (std.mem.startsWith(u8, line, "//") and last_line_was_empty) { |
| 2248 | | try ais.insertNewline(); |
| 2249 | | } else { |
| 2250 | | try ais.maybeInsertNewline(); |
| 2251 | | } |
| 2252 | | last_line_was_empty = (line.len == 0); |
| 2253 | | try ais.writeAll(line); |
| 2254 | | } |
| 2255 | | } |
| 2256 | | |
| 2257 | | if (i + 1 < section_exprs.len) { |
| 2258 | | const next_expr = section_exprs[i + 1]; |
| 2259 | | const comma = tree.lastToken(expr) + 1; |
| 2508 | } |
| 2260 | 2509 | |
| 2261 | | if (column_counter != row_size - 1) { |
| 2262 | | if (!expr_newlines[i] and !expr_newlines[i + 1]) { |
| 2263 | | // Neither the current or next expression is multiline |
| 2264 | | try renderToken(r, comma, .space); // , |
| 2265 | | assert(column_widths[column_counter % row_size] >= expr_widths[i]); |
| 2266 | | const padding = column_widths[column_counter % row_size] - expr_widths[i]; |
| 2267 | | try ais.splatByteAll(' ', padding); |
| 2510 | var remaining_exprs = array_init.ast.elements; |
| 2511 | var remaining_widths = expr_widths; |
| 2512 | while (remaining_exprs.len != 0) { |
| 2513 | var row_size: usize = 1; |
| 2514 | for (1.., remaining_exprs, remaining_widths) |len, e, w| { |
| 2515 | if (w == .nonprint) break; |
| 2516 | row_size = len; |
| 2268 | 2517 | |
| 2269 | | column_counter += 1; |
| 2270 | | continue; |
| 2271 | | } |
| 2272 | | } |
| 2518 | var after_expr = tree.lastToken(e); |
| 2519 | after_expr += @intFromBool(tree.tokenTag(after_expr + 1) == .comma); |
| 2520 | assert(tree.tokenTag(after_expr) == .comma or after_expr + 1 == rbrace); |
| 2521 | if (!tree.tokensOnSameLine(after_expr, after_expr + 1)) |
| 2522 | break; |
| 2523 | } else { |
| 2524 | // All the expressions are on the same line. |
| 2525 | // However, if there is a trailing comma, we put them each on their own line. |
| 2526 | if (tree.tokenTag(rbrace - 1) == .comma) |
| 2527 | row_size = 1; |
| 2528 | } |
| 2273 | 2529 | |
| 2274 | | if (single_line and row_size != 1) { |
| 2275 | | try renderToken(r, comma, .space); // , |
| 2276 | | continue; |
| 2530 | // Determine the size of this section |
| 2531 | const section_end = end: { |
| 2532 | var line_start = row_size; // Start after the first row to ignore comments on it |
| 2533 | break :end for (line_start.., remaining_exprs[line_start..]) |i, e| { |
| 2534 | const expr_first = tree.firstToken(e); |
| 2535 | // Any nonprint character terminates the line because they are always put on their |
| 2536 | // own line, so they will not end up on the same line as the trailing comment. |
| 2537 | if (expr_widths[i - 1] == .nonprint or !tree.tokensOnSameLine(expr_first - 1, expr_first)) { |
| 2538 | line_start = i; |
| 2277 | 2539 | } |
| 2278 | 2540 | |
| 2279 | | column_counter = 0; |
| 2280 | | try renderToken(r, comma, .newline); // , |
| 2281 | | try renderExtraNewline(r, next_expr); |
| 2541 | var after_expr = tree.lastToken(e); |
| 2542 | after_expr += @intFromBool(tree.tokenTag(after_expr + 1) == .comma); |
| 2543 | assert(tree.tokenTag(after_expr) == .comma or after_expr + 1 == rbrace); |
| 2544 | if (hasTrailingComment(tree, after_expr)) |
| 2545 | break line_start; |
| 2546 | } else remaining_exprs.len; |
| 2547 | }; |
| 2548 | const section_exprs = remaining_exprs[0..section_end]; |
| 2549 | const section_widths = remaining_widths[0..section_end]; |
| 2550 | remaining_exprs = remaining_exprs[section_end..]; |
| 2551 | remaining_widths = remaining_widths[section_end..]; |
| 2552 | |
| 2553 | // Determine the width of each column |
| 2554 | var col_widths = try gpa.alloc(usize, row_size); |
| 2555 | defer gpa.free(col_widths); |
| 2556 | @memset(col_widths, 0); |
| 2557 | |
| 2558 | var col: usize = 0; |
| 2559 | for (section_widths) |w| { |
| 2560 | if (w == .nonprint) { |
| 2561 | col = 0; |
| 2562 | continue; |
| 2563 | } |
| 2564 | col_widths[col] = @max(col_widths[col], @intFromEnum(w)); |
| 2565 | col += 1; |
| 2566 | if (col == row_size) { |
| 2567 | col = 0; |
| 2282 | 2568 | } |
| 2283 | 2569 | } |
| 2284 | 2570 | |
| 2285 | | if (expr_index == array_init.ast.elements.len) |
| 2286 | | break; |
| 2571 | // Render each expression |
| 2572 | col = 0; |
| 2573 | for (0.., section_exprs, section_widths) |i, e, w| { |
| 2574 | if (i + 1 == section_end or col + 1 == row_size or |
| 2575 | w == .nonprint or section_widths[i + 1] == .nonprint) |
| 2576 | { |
| 2577 | try renderExpression(r, e, .comma); |
| 2578 | col = 0; |
| 2579 | if (i + 1 != section_end) { |
| 2580 | try renderExtraNewline(r, section_exprs[i + 1]); |
| 2581 | } |
| 2582 | } else { |
| 2583 | try renderExpression(r, e, .comma_space); |
| 2584 | try ais.splatByteAll(' ', col_widths[col] - @intFromEnum(w)); |
| 2585 | col += 1; |
| 2586 | } |
| 2587 | } |
| 2287 | 2588 | } |
| 2288 | 2589 | |
| 2590 | ais.popSpace(); |
| 2289 | 2591 | ais.popIndent(); |
| 2290 | 2592 | return renderToken(r, rbrace, space); // rbrace |
| 2291 | 2593 | } |
| 2292 | 2594 | |
| 2595 | fn isOneLineErrorSetDecl( |
| 2596 | tree: Ast, |
| 2597 | lbrace: Ast.TokenIndex, |
| 2598 | rbrace: Ast.TokenIndex, |
| 2599 | ) bool { |
| 2600 | // If there is a trailing comma, comment, or document comment, then render each |
| 2601 | // item on its own line. |
| 2602 | return tree.tokenTag(rbrace - 1) != .comma and |
| 2603 | !hasDocComment(tree, lbrace + 1, rbrace) and |
| 2604 | !hasComment(tree, lbrace, rbrace); |
| 2605 | } |
| 2606 | |
| 2607 | fn isOneLineContainerDecl( |
| 2608 | tree: Ast, |
| 2609 | container_decl: Ast.full.ContainerDecl, |
| 2610 | lbrace: Ast.TokenIndex, |
| 2611 | rbrace: Ast.TokenIndex, |
| 2612 | ) bool { |
| 2613 | // We print all the members in one-line unless one of the following conditions are true: |
| 2614 | |
| 2615 | // 1. The container has comments or multiline strings. |
| 2616 | if (hasComment(tree, lbrace, rbrace) or hasMultilineString(tree, lbrace, rbrace)) { |
| 2617 | return false; |
| 2618 | } |
| 2619 | |
| 2620 | // 2. The container has a container comment. |
| 2621 | if (tree.tokenTag(lbrace + 1) == .container_doc_comment) return false; |
| 2622 | |
| 2623 | // 3. A member of the container has a doc comment. |
| 2624 | if (hasDocComment(tree, lbrace + 1, rbrace)) |
| 2625 | return false; |
| 2626 | |
| 2627 | // 4. The container has non-field members. |
| 2628 | for (container_decl.ast.members) |member| { |
| 2629 | if (tree.fullContainerField(member) == null) return false; |
| 2630 | } |
| 2631 | |
| 2632 | return true; |
| 2633 | } |
| 2634 | |
| 2293 | 2635 | fn renderContainerDecl( |
| 2294 | 2636 | r: *Render, |
| 2295 | 2637 | container_decl_node: Ast.Node.Index, |
| ... | ... | @@ -2354,27 +2696,7 @@ fn renderContainerDecl( |
| 2354 | 2696 | } |
| 2355 | 2697 | |
| 2356 | 2698 | const src_has_trailing_comma = tree.tokenTag(rbrace - 1) == .comma; |
| 2357 | | if (!src_has_trailing_comma) one_line: { |
| 2358 | | // We print all the members in-line unless one of the following conditions are true: |
| 2359 | | |
| 2360 | | // 1. The container has comments or multiline strings. |
| 2361 | | if (hasComment(tree, lbrace, rbrace) or hasMultilineString(tree, lbrace, rbrace)) { |
| 2362 | | break :one_line; |
| 2363 | | } |
| 2364 | | |
| 2365 | | // 2. The container has a container comment. |
| 2366 | | if (tree.tokenTag(lbrace + 1) == .container_doc_comment) break :one_line; |
| 2367 | | |
| 2368 | | // 3. A member of the container has a doc comment. |
| 2369 | | for (tree.tokens.items(.tag)[lbrace + 1 .. rbrace - 1]) |tag| { |
| 2370 | | if (tag == .doc_comment) break :one_line; |
| 2371 | | } |
| 2372 | | |
| 2373 | | // 4. The container has non-field members. |
| 2374 | | for (container_decl.ast.members) |member| { |
| 2375 | | if (tree.fullContainerField(member) == null) break :one_line; |
| 2376 | | } |
| 2377 | | |
| 2699 | if (!src_has_trailing_comma and isOneLineContainerDecl(tree, container_decl, lbrace, rbrace)) { |
| 2378 | 2700 | // Print all the declarations on the same line. |
| 2379 | 2701 | try renderToken(r, lbrace, .space); // lbrace |
| 2380 | 2702 | for (container_decl.ast.members) |member| { |
| ... | ... | @@ -2436,7 +2758,7 @@ fn renderAsm( |
| 2436 | 2758 | const first_clobber = tree.firstToken(clobbers); |
| 2437 | 2759 | try renderToken(r, first_clobber - 3, .none); |
| 2438 | 2760 | try renderToken(r, first_clobber - 2, .none); |
| 2439 | | try renderToken(r, first_clobber - 1, .space); |
| 2761 | try renderToken(r, first_clobber - 1, .maybe_space); |
| 2440 | 2762 | try renderExpression(r, clobbers, .none); |
| 2441 | 2763 | ais.popIndent(); |
| 2442 | 2764 | return renderToken(r, asm_node.ast.rparen, space); // rparen |
| ... | ... | @@ -2450,6 +2772,9 @@ fn renderAsm( |
| 2450 | 2772 | |
| 2451 | 2773 | try ais.forcePushIndent(.normal); |
| 2452 | 2774 | try renderExpression(r, asm_node.ast.template, .newline); |
| 2775 | ais.forceLastIndent(); // Might have been dedented by a multiline string literal |
| 2776 | assert(ais.current_line_empty); |
| 2777 | |
| 2453 | 2778 | ais.setIndentDelta(asm_indent_delta); |
| 2454 | 2779 | const colon1 = tree.lastToken(asm_node.ast.template) + 1; |
| 2455 | 2780 | |
| ... | ... | @@ -2527,9 +2852,10 @@ fn renderAsm( |
| 2527 | 2852 | unreachable; |
| 2528 | 2853 | }; |
| 2529 | 2854 | |
| 2530 | | try renderToken(r, colon3, .space); // : |
| 2855 | try renderToken(r, colon3, .maybe_space); // : |
| 2531 | 2856 | const clobbers = asm_node.ast.clobbers.unwrap().?; |
| 2532 | 2857 | try renderExpression(r, clobbers, .none); |
| 2858 | ais.forceLastIndent(); // Might have been dedented by a multiline string literal |
| 2533 | 2859 | ais.setIndentDelta(indent_delta); |
| 2534 | 2860 | ais.popIndent(); |
| 2535 | 2861 | return renderToken(r, asm_node.ast.rparen, space); // rparen |
| ... | ... | @@ -2555,7 +2881,7 @@ fn renderParamList( |
| 2555 | 2881 | |
| 2556 | 2882 | if (params.len == 0) { |
| 2557 | 2883 | try ais.pushIndent(.normal); |
| 2558 | | try renderToken(r, lparen, .none); |
| 2884 | try renderToken(r, lparen, .none); // ( |
| 2559 | 2885 | ais.popIndent(); |
| 2560 | 2886 | return renderToken(r, lparen + 1, space); // ) |
| 2561 | 2887 | } |
| ... | ... | @@ -2590,10 +2916,7 @@ fn renderParamList( |
| 2590 | 2916 | |
| 2591 | 2917 | if (i + 1 < params.len) { |
| 2592 | 2918 | const comma = tree.lastToken(param_node) + 1; |
| 2593 | | const next_multiline_string = |
| 2594 | | tree.tokenTag(tree.firstToken(params[i + 1])) == .multiline_string_literal_line; |
| 2595 | | const comma_space: Space = if (next_multiline_string) .none else .space; |
| 2596 | | try renderToken(r, comma, comma_space); |
| 2919 | try renderToken(r, comma, .maybe_space); |
| 2597 | 2920 | } |
| 2598 | 2921 | } |
| 2599 | 2922 | ais.popIndent(); |
| ... | ... | @@ -2655,6 +2978,13 @@ const Space = enum { |
| 2655 | 2978 | /// Additionally consume the next token if it is a semicolon. |
| 2656 | 2979 | /// In either case, a newline will be inserted afterwards. |
| 2657 | 2980 | semicolon, |
| 2981 | /// If the next token is not a multiline string literal, this acts as .space, |
| 2982 | /// otherwise this acts as .none. |
| 2983 | maybe_space, |
| 2984 | /// Additionally consume the next token if it is a comma. |
| 2985 | /// In either case, a space will be inserted afterwards |
| 2986 | /// if the following token is not a multiline string literal. |
| 2987 | comma_maybe_space, |
| 2658 | 2988 | /// Skip rendering whitespace and comments. If this is used, the caller |
| 2659 | 2989 | /// *must* handle whitespace and comments manually. |
| 2660 | 2990 | skip, |
| ... | ... | @@ -2719,6 +3049,16 @@ fn renderSpace(r: *Render, token_index: Ast.TokenIndex, lexeme_len: usize, space |
| 2719 | 3049 | try ais.insertNewline(); |
| 2720 | 3050 | }, |
| 2721 | 3051 | |
| 3052 | .maybe_space => if (!comment and next_token_tag != .multiline_string_literal_line) { |
| 3053 | try ais.writeByte(' '); |
| 3054 | }, |
| 3055 | |
| 3056 | .comma_maybe_space => if (next_token_tag == .comma) { |
| 3057 | try renderToken(r, token_index + 1, .maybe_space); |
| 3058 | } else if (!comment) { |
| 3059 | try ais.writeByte(' '); |
| 3060 | }, |
| 3061 | |
| 2722 | 3062 | .skip => unreachable, |
| 2723 | 3063 | } |
| 2724 | 3064 | } |
| ... | ... | @@ -2727,9 +3067,9 @@ fn renderOnlySpace(r: *Render, space: Space) Error!void { |
| 2727 | 3067 | const ais = r.ais; |
| 2728 | 3068 | switch (space) { |
| 2729 | 3069 | .none => {}, |
| 2730 | | .space => try ais.writeByte(' '), |
| 3070 | .space, .maybe_space => try ais.writeByte(' '), |
| 2731 | 3071 | .newline => try ais.insertNewline(), |
| 2732 | | .comma => try ais.writeAll(",\n"), |
| 3072 | .comma, .comma_maybe_space => try ais.writeAll(",\n"), |
| 2733 | 3073 | .comma_space => try ais.writeAll(", "), |
| 2734 | 3074 | .semicolon => try ais.writeAll(";\n"), |
| 2735 | 3075 | .skip => unreachable, |
| ... | ... | @@ -2794,7 +3134,7 @@ fn renderIdentifier(r: *Render, token_index: Ast.TokenIndex, space: Space, quote |
| 2794 | 3134 | }, |
| 2795 | 3135 | .failure => return renderQuotedIdentifier(r, token_index, space, false), |
| 2796 | 3136 | } |
| 2797 | | contents_i += esc_offset; |
| 3137 | contents_i = esc_offset; |
| 2798 | 3138 | continue; |
| 2799 | 3139 | }, |
| 2800 | 3140 | else => return renderQuotedIdentifier(r, token_index, space, false), |
| ... | ... | @@ -2884,6 +3224,9 @@ fn renderIdentifierContents(ais: *AutoIndentingStream, bytes: []const u8) !void |
| 2884 | 3224 | } |
| 2885 | 3225 | }, |
| 2886 | 3226 | .failure => { |
| 3227 | // Escape the stray backslash |
| 3228 | // This also avoids cases like "\x3\x39" becoming "\x39" |
| 3229 | try ais.writeByte('\\'); |
| 2887 | 3230 | try ais.writeAll(escape_sequence); |
| 2888 | 3231 | }, |
| 2889 | 3232 | } |
| ... | ... | @@ -2910,7 +3253,7 @@ fn hasComment(tree: Ast, start_token: Ast.TokenIndex, end_token: Ast.TokenIndex) |
| 2910 | 3253 | const token: Ast.TokenIndex = @intCast(i); |
| 2911 | 3254 | const start = tree.tokenStart(token) + tree.tokenSlice(token).len; |
| 2912 | 3255 | const end = tree.tokenStart(token + 1); |
| 2913 | | if (mem.find(u8, tree.source[start..end], "//") != null) return true; |
| 3256 | if (mem.findScalar(u8, tree.source[start..end], '/') != null) return true; |
| 2914 | 3257 | } |
| 2915 | 3258 | |
| 2916 | 3259 | return false; |
| ... | ... | @@ -2926,6 +3269,16 @@ fn hasMultilineString(tree: Ast, start_token: Ast.TokenIndex, end_token: Ast.Tok |
| 2926 | 3269 | ) != null; |
| 2927 | 3270 | } |
| 2928 | 3271 | |
| 3272 | /// Returns true if there exists a doc comment between the start |
| 3273 | /// of token `start_token` and the start of token `end_token`. |
| 3274 | fn hasDocComment(tree: Ast, start_token: Ast.TokenIndex, end_token: Ast.TokenIndex) bool { |
| 3275 | return std.mem.indexOfScalar( |
| 3276 | Token.Tag, |
| 3277 | tree.tokens.items(.tag)[start_token..end_token], |
| 3278 | .doc_comment, |
| 3279 | ) != null; |
| 3280 | } |
| 3281 | |
| 2929 | 3282 | /// Assumes that start is the first byte past the previous token and |
| 2930 | 3283 | /// that end is the last byte before the next token. |
| 2931 | 3284 | fn renderComments(r: *Render, start: usize, end: usize) Error!bool { |
| ... | ... | @@ -3095,25 +3448,10 @@ fn tokenSliceForRender(tree: Ast, token_index: Ast.TokenIndex) []const u8 { |
| 3095 | 3448 | return ret; |
| 3096 | 3449 | } |
| 3097 | 3450 | |
| 3098 | | fn writeStringLiteralAsIdentifier(r: *Render, token_index: Ast.TokenIndex) !usize { |
| 3099 | | const tree = r.tree; |
| 3100 | | const ais = r.ais; |
| 3101 | | assert(tree.tokenTag(token_index) == .string_literal); |
| 3102 | | const lexeme = tokenSliceForRender(tree, token_index); |
| 3103 | | const unquoted = lexeme[1..][0 .. lexeme.len - 2]; |
| 3104 | | if (std.zig.isValidId(unquoted)) { |
| 3105 | | try ais.writeAll(unquoted); |
| 3106 | | return unquoted.len; |
| 3107 | | } else { |
| 3108 | | try ais.writeByte('@'); |
| 3109 | | try ais.writeAll(lexeme); |
| 3110 | | return lexeme.len + 1; |
| 3111 | | } |
| 3112 | | } |
| 3113 | | |
| 3114 | | fn hasSameLineComment(tree: Ast, token_index: Ast.TokenIndex) bool { |
| 3115 | | const between_source = tree.source[tree.tokenStart(token_index)..tree.tokenStart(token_index + 1)]; |
| 3116 | | for (between_source) |byte| switch (byte) { |
| 3451 | fn hasTrailingComment(tree: Ast, t: Ast.TokenIndex) bool { |
| 3452 | const start = tree.tokenStart(t) + tree.tokenSlice(t).len; |
| 3453 | const between = tree.source[start..tree.tokenStart(t + 1)]; |
| 3454 | for (between) |byte| switch (byte) { |
| 3117 | 3455 | '\n' => return false, |
| 3118 | 3456 | '/' => return true, |
| 3119 | 3457 | else => continue, |
| ... | ... | @@ -3125,12 +3463,7 @@ fn hasSameLineComment(tree: Ast, token_index: Ast.TokenIndex) bool { |
| 3125 | 3463 | /// start_token and end_token. |
| 3126 | 3464 | fn anythingBetween(tree: Ast, start_token: Ast.TokenIndex, end_token: Ast.TokenIndex) bool { |
| 3127 | 3465 | if (start_token + 1 != end_token) return true; |
| 3128 | | const between_source = tree.source[tree.tokenStart(start_token)..tree.tokenStart(start_token + 1)]; |
| 3129 | | for (between_source) |byte| switch (byte) { |
| 3130 | | '/' => return true, |
| 3131 | | else => continue, |
| 3132 | | }; |
| 3133 | | return false; |
| 3466 | return hasComment(tree, start_token, end_token); |
| 3134 | 3467 | } |
| 3135 | 3468 | |
| 3136 | 3469 | fn writeFixingWhitespace(w: *Writer, slice: []const u8) Error!void { |
| ... | ... | @@ -3217,29 +3550,6 @@ fn nodeCausesSliceOpSpace(tag: Ast.Node.Tag) bool { |
| 3217 | 3550 | }; |
| 3218 | 3551 | } |
| 3219 | 3552 | |
| 3220 | | // Returns the number of nodes in `exprs` that are on the same line as `rtoken`. |
| 3221 | | fn rowSize(tree: Ast, exprs: []const Ast.Node.Index, rtoken: Ast.TokenIndex) usize { |
| 3222 | | const first_token = tree.firstToken(exprs[0]); |
| 3223 | | if (tree.tokensOnSameLine(first_token, rtoken)) { |
| 3224 | | const maybe_comma = rtoken - 1; |
| 3225 | | if (tree.tokenTag(maybe_comma) == .comma) |
| 3226 | | return 1; |
| 3227 | | return exprs.len; // no newlines |
| 3228 | | } |
| 3229 | | |
| 3230 | | var count: usize = 1; |
| 3231 | | for (exprs, 0..) |expr, i| { |
| 3232 | | if (i + 1 < exprs.len) { |
| 3233 | | const expr_last_token = tree.lastToken(expr) + 1; |
| 3234 | | if (!tree.tokensOnSameLine(expr_last_token, tree.firstToken(exprs[i + 1]))) return count; |
| 3235 | | count += 1; |
| 3236 | | } else { |
| 3237 | | return count; |
| 3238 | | } |
| 3239 | | } |
| 3240 | | unreachable; |
| 3241 | | } |
| 3242 | | |
| 3243 | 3553 | /// Automatically inserts indentation of written data by keeping |
| 3244 | 3554 | /// track of the current indentation level |
| 3245 | 3555 | /// |
| ... | ... | @@ -3477,11 +3787,19 @@ const AutoIndentingStream = struct { |
| 3477 | 3787 | |
| 3478 | 3788 | pub fn popIndent(ais: *AutoIndentingStream) void { |
| 3479 | 3789 | if (ais.indent_stack.pop().?.realized) { |
| 3480 | | assert(ais.indent_count > 0); |
| 3481 | 3790 | ais.indent_count -= 1; |
| 3482 | 3791 | } |
| 3483 | 3792 | } |
| 3484 | 3793 | |
| 3794 | /// Forces the last pushed indent to be realized |
| 3795 | pub fn forceLastIndent(ais: *AutoIndentingStream) void { |
| 3796 | const top = &ais.indent_stack.items[ais.indent_stack.items.len - 1]; |
| 3797 | if (!top.realized) { |
| 3798 | top.realized = true; |
| 3799 | ais.indent_count += 1; |
| 3800 | } |
| 3801 | } |
| 3802 | |
| 3485 | 3803 | pub fn indentStackEmpty(ais: *AutoIndentingStream) bool { |
| 3486 | 3804 | return ais.indent_stack.items.len == 0; |
| 3487 | 3805 | } |
| ... | ... | @@ -3489,8 +3807,8 @@ const AutoIndentingStream = struct { |
| 3489 | 3807 | /// Writes ' ' bytes if the current line is empty |
| 3490 | 3808 | fn applyIndent(ais: *AutoIndentingStream) Error!void { |
| 3491 | 3809 | const current_indent = ais.currentIndent(); |
| 3492 | | if (ais.current_line_empty and current_indent > 0) { |
| 3493 | | if (ais.disabled_offset == null) { |
| 3810 | if (ais.current_line_empty) { |
| 3811 | if (current_indent > 0 and ais.disabled_offset == null) { |
| 3494 | 3812 | try ais.underlying_writer.splatByteAll(' ', current_indent); |
| 3495 | 3813 | } |
| 3496 | 3814 | ais.applied_indent = current_indent; |