| ... | ... | @@ -55,7 +55,7 @@ pub const Parser = struct { |
| 55 | 55 | const TopLevelDeclCtx = struct { |
| 56 | 56 | decls: &ArrayList(&ast.Node), |
| 57 | 57 | visib_token: ?Token, |
| 58 | | extern_token: ?Token, |
| 58 | extern_export_inline_token: ?Token, |
| 59 | 59 | lib_name: ?&ast.Node, |
| 60 | 60 | }; |
| 61 | 61 | |
| ... | ... | @@ -142,6 +142,7 @@ pub const Parser = struct { |
| 142 | 142 | const State = union(enum) { |
| 143 | 143 | TopLevel, |
| 144 | 144 | TopLevelExtern: TopLevelDeclCtx, |
| 145 | TopLevelLibname: TopLevelDeclCtx, |
| 145 | 146 | TopLevelDecl: TopLevelDeclCtx, |
| 146 | 147 | ContainerExtern: ContainerExternCtx, |
| 147 | 148 | ContainerDecl: &ast.NodeContainerDecl, |
| ... | ... | @@ -171,12 +172,14 @@ pub const Parser = struct { |
| 171 | 172 | Semicolon: &const &const ast.Node, |
| 172 | 173 | AsmOutputItems: &ArrayList(&ast.NodeAsmOutput), |
| 173 | 174 | AsmInputItems: &ArrayList(&ast.NodeAsmInput), |
| 174 | | AsmClopperItems: &ArrayList(&ast.NodeStringLiteral), |
| 175 | AsmClopperItems: &ArrayList(&ast.Node), |
| 175 | 176 | ExprListItemOrEnd: ExprListCtx, |
| 176 | 177 | ExprListCommaOrEnd: ExprListCtx, |
| 177 | 178 | FieldInitListItemOrEnd: ListSave(&ast.NodeFieldInitializer), |
| 178 | 179 | FieldInitListCommaOrEnd: ListSave(&ast.NodeFieldInitializer), |
| 179 | 180 | FieldListCommaOrEnd: &ast.NodeContainerDecl, |
| 181 | IdentifierListItemOrEnd: ListSave(&ast.NodeIdentifier), |
| 182 | IdentifierListCommaOrEnd: ListSave(&ast.NodeIdentifier), |
| 180 | 183 | SwitchCaseOrEnd: ListSave(&ast.NodeSwitchCase), |
| 181 | 184 | SuspendBody: &ast.NodeSuspend, |
| 182 | 185 | AsyncEnd: AsyncEndCtx, |
| ... | ... | @@ -299,7 +302,11 @@ pub const Parser = struct { |
| 299 | 302 | Token.Id.Keyword_test => { |
| 300 | 303 | stack.append(State.TopLevel) catch unreachable; |
| 301 | 304 | |
| 302 | | const name_token = (try self.expectToken(&stack, Token.Id.StringLiteral)) ?? continue; |
| 305 | const name_token = self.getNextToken(); |
| 306 | const name = (try self.parseStringLiteral(arena, name_token)) ?? { |
| 307 | try self.parseError(&stack, name_token, "expected string literal, found {}", @tagName(name_token.id)); |
| 308 | continue; |
| 309 | }; |
| 303 | 310 | const lbrace = (try self.expectToken(&stack, Token.Id.LBrace)) ?? continue; |
| 304 | 311 | |
| 305 | 312 | const block = try self.createNode(arena, ast.NodeBlock, |
| ... | ... | @@ -315,7 +322,7 @@ pub const Parser = struct { |
| 315 | 322 | ast.NodeTestDecl { |
| 316 | 323 | .base = undefined, |
| 317 | 324 | .test_token = token, |
| 318 | | .name = &(try self.createLiteral(arena, ast.NodeStringLiteral, name_token)).base, |
| 325 | .name = name, |
| 319 | 326 | .body_node = &block.base, |
| 320 | 327 | } |
| 321 | 328 | ); |
| ... | ... | @@ -326,13 +333,13 @@ pub const Parser = struct { |
| 326 | 333 | root_node.eof_token = token; |
| 327 | 334 | return Tree {.root_node = root_node, .arena_allocator = arena_allocator}; |
| 328 | 335 | }, |
| 329 | | Token.Id.Keyword_pub, Token.Id.Keyword_export => { |
| 336 | Token.Id.Keyword_pub => { |
| 330 | 337 | stack.append(State.TopLevel) catch unreachable; |
| 331 | 338 | try stack.append(State { |
| 332 | 339 | .TopLevelExtern = TopLevelDeclCtx { |
| 333 | 340 | .decls = &root_node.decls, |
| 334 | 341 | .visib_token = token, |
| 335 | | .extern_token = null, |
| 342 | .extern_export_inline_token = null, |
| 336 | 343 | .lib_name = null, |
| 337 | 344 | } |
| 338 | 345 | }); |
| ... | ... | @@ -357,7 +364,7 @@ pub const Parser = struct { |
| 357 | 364 | .TopLevelExtern = TopLevelDeclCtx { |
| 358 | 365 | .decls = &root_node.decls, |
| 359 | 366 | .visib_token = null, |
| 360 | | .extern_token = null, |
| 367 | .extern_export_inline_token = null, |
| 361 | 368 | .lib_name = null, |
| 362 | 369 | } |
| 363 | 370 | }); |
| ... | ... | @@ -368,42 +375,24 @@ pub const Parser = struct { |
| 368 | 375 | State.TopLevelExtern => |ctx| { |
| 369 | 376 | const token = self.getNextToken(); |
| 370 | 377 | switch (token.id) { |
| 371 | | Token.Id.Keyword_use => { |
| 372 | | const node = try self.createAttachNode(arena, ctx.decls, ast.NodeUse, |
| 373 | | ast.NodeUse { |
| 374 | | .base = undefined, |
| 375 | | .visib_token = ctx.visib_token, |
| 376 | | .expr = undefined, |
| 377 | | .semicolon_token = undefined, |
| 378 | | } |
| 379 | | ); |
| 378 | Token.Id.Keyword_export, Token.Id.Keyword_inline => { |
| 380 | 379 | stack.append(State { |
| 381 | | .ExpectTokenSave = ExpectTokenSave { |
| 382 | | .id = Token.Id.Semicolon, |
| 383 | | .ptr = &node.semicolon_token, |
| 384 | | } |
| 380 | .TopLevelDecl = TopLevelDeclCtx { |
| 381 | .decls = ctx.decls, |
| 382 | .visib_token = ctx.visib_token, |
| 383 | .extern_export_inline_token = token, |
| 384 | .lib_name = null, |
| 385 | }, |
| 385 | 386 | }) catch unreachable; |
| 386 | | try stack.append(State { .Expression = DestPtr { .Field = &node.expr } }); |
| 387 | 387 | continue; |
| 388 | 388 | }, |
| 389 | 389 | Token.Id.Keyword_extern => { |
| 390 | | const lib_name_token = self.getNextToken(); |
| 391 | | const lib_name = blk: { |
| 392 | | if (lib_name_token.id == Token.Id.StringLiteral) { |
| 393 | | const res = try self.createLiteral(arena, ast.NodeStringLiteral, lib_name_token); |
| 394 | | break :blk &res.base; |
| 395 | | } else { |
| 396 | | self.putBackToken(lib_name_token); |
| 397 | | break :blk null; |
| 398 | | } |
| 399 | | }; |
| 400 | | |
| 401 | 390 | stack.append(State { |
| 402 | | .TopLevelDecl = TopLevelDeclCtx { |
| 391 | .TopLevelLibname = TopLevelDeclCtx { |
| 403 | 392 | .decls = ctx.decls, |
| 404 | 393 | .visib_token = ctx.visib_token, |
| 405 | | .extern_token = token, |
| 406 | | .lib_name = lib_name, |
| 394 | .extern_export_inline_token = token, |
| 395 | .lib_name = null, |
| 407 | 396 | }, |
| 408 | 397 | }) catch unreachable; |
| 409 | 398 | continue; |
| ... | ... | @@ -415,17 +404,67 @@ pub const Parser = struct { |
| 415 | 404 | } |
| 416 | 405 | } |
| 417 | 406 | }, |
| 407 | |
| 408 | State.TopLevelLibname => |ctx| { |
| 409 | const lib_name = blk: { |
| 410 | const lib_name_token = self.getNextToken(); |
| 411 | break :blk (try self.parseStringLiteral(arena, lib_name_token)) ?? { |
| 412 | self.putBackToken(lib_name_token); |
| 413 | break :blk null; |
| 414 | }; |
| 415 | }; |
| 416 | |
| 417 | stack.append(State { |
| 418 | .TopLevelDecl = TopLevelDeclCtx { |
| 419 | .decls = ctx.decls, |
| 420 | .visib_token = ctx.visib_token, |
| 421 | .extern_export_inline_token = ctx.extern_export_inline_token, |
| 422 | .lib_name = lib_name, |
| 423 | }, |
| 424 | }) catch unreachable; |
| 425 | }, |
| 426 | |
| 418 | 427 | State.TopLevelDecl => |ctx| { |
| 419 | 428 | const token = self.getNextToken(); |
| 420 | 429 | switch (token.id) { |
| 430 | Token.Id.Keyword_use => { |
| 431 | if (ctx.extern_export_inline_token != null) { |
| 432 | try self.parseError(&stack, token, "Invalid token {}", @tagName((??ctx.extern_export_inline_token).id)); |
| 433 | continue; |
| 434 | } |
| 435 | |
| 436 | const node = try self.createAttachNode(arena, ctx.decls, ast.NodeUse, |
| 437 | ast.NodeUse { |
| 438 | .base = undefined, |
| 439 | .visib_token = ctx.visib_token, |
| 440 | .expr = undefined, |
| 441 | .semicolon_token = undefined, |
| 442 | } |
| 443 | ); |
| 444 | stack.append(State { |
| 445 | .ExpectTokenSave = ExpectTokenSave { |
| 446 | .id = Token.Id.Semicolon, |
| 447 | .ptr = &node.semicolon_token, |
| 448 | } |
| 449 | }) catch unreachable; |
| 450 | try stack.append(State { .Expression = DestPtr { .Field = &node.expr } }); |
| 451 | continue; |
| 452 | }, |
| 421 | 453 | Token.Id.Keyword_var, Token.Id.Keyword_const => { |
| 454 | if (ctx.extern_export_inline_token) |extern_export_inline_token| { |
| 455 | if (extern_export_inline_token.id == Token.Id.Keyword_inline) { |
| 456 | try self.parseError(&stack, token, "Invalid token {}", @tagName(extern_export_inline_token.id)); |
| 457 | continue; |
| 458 | } |
| 459 | } |
| 460 | |
| 422 | 461 | const var_decl_node = try self.createAttachNode(arena, ctx.decls, ast.NodeVarDecl, |
| 423 | 462 | ast.NodeVarDecl { |
| 424 | 463 | .base = undefined, |
| 425 | 464 | .visib_token = ctx.visib_token, |
| 426 | 465 | .mut_token = token, |
| 427 | 466 | .comptime_token = null, |
| 428 | | .extern_token = ctx.extern_token, |
| 467 | .extern_export_token = ctx.extern_export_inline_token, |
| 429 | 468 | .type_node = null, |
| 430 | 469 | .align_node = null, |
| 431 | 470 | .init_node = null, |
| ... | ... | @@ -449,8 +488,7 @@ pub const Parser = struct { |
| 449 | 488 | .params = ArrayList(&ast.Node).init(arena), |
| 450 | 489 | .return_type = undefined, |
| 451 | 490 | .var_args_token = null, |
| 452 | | .extern_token = ctx.extern_token, |
| 453 | | .inline_token = null, |
| 491 | .extern_export_inline_token = ctx.extern_export_inline_token, |
| 454 | 492 | .cc_token = null, |
| 455 | 493 | .async_attr = null, |
| 456 | 494 | .body_node = null, |
| ... | ... | @@ -472,8 +510,7 @@ pub const Parser = struct { |
| 472 | 510 | .params = ArrayList(&ast.Node).init(arena), |
| 473 | 511 | .return_type = undefined, |
| 474 | 512 | .var_args_token = null, |
| 475 | | .extern_token = ctx.extern_token, |
| 476 | | .inline_token = null, |
| 513 | .extern_export_inline_token = ctx.extern_export_inline_token, |
| 477 | 514 | .cc_token = token, |
| 478 | 515 | .async_attr = null, |
| 479 | 516 | .body_node = null, |
| ... | ... | @@ -510,8 +547,7 @@ pub const Parser = struct { |
| 510 | 547 | .params = ArrayList(&ast.Node).init(arena), |
| 511 | 548 | .return_type = undefined, |
| 512 | 549 | .var_args_token = null, |
| 513 | | .extern_token = ctx.extern_token, |
| 514 | | .inline_token = null, |
| 550 | .extern_export_inline_token = ctx.extern_export_inline_token, |
| 515 | 551 | .cc_token = null, |
| 516 | 552 | .async_attr = async_node, |
| 517 | 553 | .body_node = null, |
| ... | ... | @@ -667,6 +703,7 @@ pub const Parser = struct { |
| 667 | 703 | const node = try self.createAttachNode(arena, &container_decl.fields_and_decls, ast.NodeStructField, |
| 668 | 704 | ast.NodeStructField { |
| 669 | 705 | .base = undefined, |
| 706 | .visib_token = null, |
| 670 | 707 | .name_token = token, |
| 671 | 708 | .type_expr = undefined, |
| 672 | 709 | } |
| ... | ... | @@ -719,13 +756,48 @@ pub const Parser = struct { |
| 719 | 756 | }, |
| 720 | 757 | } |
| 721 | 758 | }, |
| 722 | | Token.Id.Keyword_pub, Token.Id.Keyword_export => { |
| 759 | Token.Id.Keyword_pub => { |
| 760 | if (self.eatToken(Token.Id.Identifier)) |identifier| { |
| 761 | switch (container_decl.kind) { |
| 762 | ast.NodeContainerDecl.Kind.Struct => { |
| 763 | const node = try self.createAttachNode(arena, &container_decl.fields_and_decls, ast.NodeStructField, |
| 764 | ast.NodeStructField { |
| 765 | .base = undefined, |
| 766 | .visib_token = token, |
| 767 | .name_token = identifier, |
| 768 | .type_expr = undefined, |
| 769 | } |
| 770 | ); |
| 771 | |
| 772 | stack.append(State { .FieldListCommaOrEnd = container_decl }) catch unreachable; |
| 773 | try stack.append(State { .Expression = DestPtr { .Field = &node.type_expr } }); |
| 774 | try stack.append(State { .ExpectToken = Token.Id.Colon }); |
| 775 | continue; |
| 776 | }, |
| 777 | else => { |
| 778 | self.putBackToken(identifier); |
| 779 | } |
| 780 | } |
| 781 | } |
| 782 | |
| 783 | stack.append(State{ .ContainerDecl = container_decl }) catch unreachable; |
| 784 | try stack.append(State { |
| 785 | .TopLevelExtern = TopLevelDeclCtx { |
| 786 | .decls = &container_decl.fields_and_decls, |
| 787 | .visib_token = token, |
| 788 | .extern_export_inline_token = null, |
| 789 | .lib_name = null, |
| 790 | } |
| 791 | }); |
| 792 | continue; |
| 793 | }, |
| 794 | Token.Id.Keyword_export => { |
| 723 | 795 | stack.append(State{ .ContainerDecl = container_decl }) catch unreachable; |
| 724 | 796 | try stack.append(State { |
| 725 | 797 | .TopLevelExtern = TopLevelDeclCtx { |
| 726 | 798 | .decls = &container_decl.fields_and_decls, |
| 727 | 799 | .visib_token = token, |
| 728 | | .extern_token = null, |
| 800 | .extern_export_inline_token = null, |
| 729 | 801 | .lib_name = null, |
| 730 | 802 | } |
| 731 | 803 | }); |
| ... | ... | @@ -742,7 +814,7 @@ pub const Parser = struct { |
| 742 | 814 | .TopLevelExtern = TopLevelDeclCtx { |
| 743 | 815 | .decls = &container_decl.fields_and_decls, |
| 744 | 816 | .visib_token = null, |
| 745 | | .extern_token = null, |
| 817 | .extern_export_inline_token = null, |
| 746 | 818 | .lib_name = null, |
| 747 | 819 | } |
| 748 | 820 | }); |
| ... | ... | @@ -862,111 +934,11 @@ pub const Parser = struct { |
| 862 | 934 | stack.append(State { .Expression = DestPtr { .Field = &node.rhs } }) catch unreachable; |
| 863 | 935 | continue; |
| 864 | 936 | }, |
| 865 | | Token.Id.Keyword_suspend => { |
| 866 | | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeSuspend, |
| 867 | | ast.NodeSuspend { |
| 868 | | .base = undefined, |
| 869 | | .suspend_token = token, |
| 870 | | .payload = null, |
| 871 | | .body = null, |
| 872 | | } |
| 873 | | ); |
| 874 | | |
| 875 | | stack.append(State { .SuspendBody = node }) catch unreachable; |
| 876 | | try stack.append(State { .Payload = &node.payload }); |
| 877 | | continue; |
| 878 | | }, |
| 879 | | Token.Id.Keyword_if => { |
| 880 | | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeIf, |
| 881 | | ast.NodeIf { |
| 882 | | .base = undefined, |
| 883 | | .if_token = token, |
| 884 | | .condition = undefined, |
| 885 | | .payload = null, |
| 886 | | .body = undefined, |
| 887 | | .@"else" = null, |
| 888 | | } |
| 889 | | ); |
| 890 | | |
| 891 | | stack.append(State { .Else = &node.@"else" }) catch unreachable; |
| 892 | | try stack.append(State { .Expression = DestPtr { .Field = &node.body } }); |
| 893 | | try stack.append(State { .PointerPayload = &node.payload }); |
| 894 | | try stack.append(State { .ExpectToken = Token.Id.RParen }); |
| 895 | | try stack.append(State { .Expression = DestPtr { .Field = &node.condition } }); |
| 896 | | try stack.append(State { .ExpectToken = Token.Id.LParen }); |
| 897 | | continue; |
| 898 | | }, |
| 899 | | Token.Id.Keyword_while => { |
| 900 | | stack.append(State { |
| 901 | | .While = LoopCtx { |
| 902 | | .label = null, |
| 903 | | .inline_token = null, |
| 904 | | .loop_token = token, |
| 905 | | .dest_ptr = dest_ptr, |
| 906 | | } |
| 907 | | }) catch unreachable; |
| 908 | | continue; |
| 909 | | }, |
| 910 | | Token.Id.Keyword_for => { |
| 911 | | stack.append(State { |
| 912 | | .For = LoopCtx { |
| 913 | | .label = null, |
| 914 | | .inline_token = null, |
| 915 | | .loop_token = token, |
| 916 | | .dest_ptr = dest_ptr, |
| 917 | | } |
| 918 | | }) catch unreachable; |
| 919 | | continue; |
| 920 | | }, |
| 921 | | Token.Id.Keyword_switch => { |
| 922 | | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeSwitch, |
| 923 | | ast.NodeSwitch { |
| 924 | | .base = undefined, |
| 925 | | .switch_token = token, |
| 926 | | .expr = undefined, |
| 927 | | .cases = ArrayList(&ast.NodeSwitchCase).init(arena), |
| 928 | | .rbrace = undefined, |
| 929 | | } |
| 930 | | ); |
| 931 | | |
| 932 | | stack.append(State { |
| 933 | | .SwitchCaseOrEnd = ListSave(&ast.NodeSwitchCase) { |
| 934 | | .list = &node.cases, |
| 935 | | .ptr = &node.rbrace, |
| 936 | | }, |
| 937 | | }) catch unreachable; |
| 938 | | try stack.append(State { .ExpectToken = Token.Id.LBrace }); |
| 939 | | try stack.append(State { .ExpectToken = Token.Id.RParen }); |
| 940 | | try stack.append(State { .Expression = DestPtr { .Field = &node.expr } }); |
| 941 | | try stack.append(State { .ExpectToken = Token.Id.LParen }); |
| 942 | | }, |
| 943 | | Token.Id.Keyword_comptime => { |
| 944 | | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeComptime, |
| 945 | | ast.NodeComptime { |
| 946 | | .base = undefined, |
| 947 | | .comptime_token = token, |
| 948 | | .expr = undefined, |
| 949 | | } |
| 950 | | ); |
| 951 | | try stack.append(State { .Expression = DestPtr { .Field = &node.expr } }); |
| 952 | | continue; |
| 953 | | }, |
| 954 | | Token.Id.LBrace => { |
| 955 | | const block = try self.createToDestNode(arena, dest_ptr, ast.NodeBlock, |
| 956 | | ast.NodeBlock { |
| 957 | | .base = undefined, |
| 958 | | .label = null, |
| 959 | | .lbrace = token, |
| 960 | | .statements = ArrayList(&ast.Node).init(arena), |
| 961 | | .rbrace = undefined, |
| 962 | | } |
| 963 | | ); |
| 964 | | stack.append(State { .Block = block }) catch unreachable; |
| 965 | | continue; |
| 966 | | }, |
| 967 | 937 | else => { |
| 968 | | self.putBackToken(token); |
| 969 | | stack.append(State { .UnwrapExpressionBegin = dest_ptr }) catch unreachable; |
| 938 | if (!try self.parseBlockExpr(&stack, arena, dest_ptr, token)) { |
| 939 | self.putBackToken(token); |
| 940 | stack.append(State { .UnwrapExpressionBegin = dest_ptr }) catch unreachable; |
| 941 | } |
| 970 | 942 | continue; |
| 971 | 943 | } |
| 972 | 944 | } |
| ... | ... | @@ -1405,7 +1377,7 @@ pub const Parser = struct { |
| 1405 | 1377 | State.PrefixOpExpression => |dest_ptr| { |
| 1406 | 1378 | const token = self.getNextToken(); |
| 1407 | 1379 | if (tokenIdToPrefixOp(token.id)) |prefix_id| { |
| 1408 | | const node = try self.createToDestNode(arena, dest_ptr, ast.NodePrefixOp, |
| 1380 | var node = try self.createToDestNode(arena, dest_ptr, ast.NodePrefixOp, |
| 1409 | 1381 | ast.NodePrefixOp { |
| 1410 | 1382 | .base = undefined, |
| 1411 | 1383 | .op_token = token, |
| ... | ... | @@ -1413,6 +1385,20 @@ pub const Parser = struct { |
| 1413 | 1385 | .rhs = undefined, |
| 1414 | 1386 | } |
| 1415 | 1387 | ); |
| 1388 | |
| 1389 | if (token.id == Token.Id.AsteriskAsterisk) { |
| 1390 | const child = try self.createNode(arena, ast.NodePrefixOp, |
| 1391 | ast.NodePrefixOp { |
| 1392 | .base = undefined, |
| 1393 | .op_token = token, |
| 1394 | .op = prefix_id, |
| 1395 | .rhs = undefined, |
| 1396 | } |
| 1397 | ); |
| 1398 | node.rhs = &child.base; |
| 1399 | node = child; |
| 1400 | } |
| 1401 | |
| 1416 | 1402 | stack.append(State { .TypeExprBegin = DestPtr { .Field = &node.rhs } }) catch unreachable; |
| 1417 | 1403 | if (node.op == ast.NodePrefixOp.PrefixOp.AddrOf) { |
| 1418 | 1404 | try stack.append(State { .AddrOfModifiers = &node.op.AddrOf }); |
| ... | ... | @@ -1515,17 +1501,23 @@ pub const Parser = struct { |
| 1515 | 1501 | continue; |
| 1516 | 1502 | }, |
| 1517 | 1503 | Token.Id.Period => { |
| 1504 | const identifier = try self.createLiteral(arena, ast.NodeIdentifier, Token(undefined)); |
| 1518 | 1505 | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeInfixOp, |
| 1519 | 1506 | ast.NodeInfixOp { |
| 1520 | 1507 | .base = undefined, |
| 1521 | 1508 | .lhs = dest_ptr.get(), |
| 1522 | 1509 | .op_token = token, |
| 1523 | 1510 | .op = ast.NodeInfixOp.InfixOp.Period, |
| 1524 | | .rhs = undefined, |
| 1511 | .rhs = &identifier.base, |
| 1525 | 1512 | } |
| 1526 | 1513 | ); |
| 1527 | 1514 | stack.append(State { .SuffixOpExpressionEnd = dest_ptr }) catch unreachable; |
| 1528 | | try stack.append(State { .SuffixOpExpressionBegin = DestPtr { .Field = &node.rhs }}); |
| 1515 | try stack.append(State { |
| 1516 | .ExpectTokenSave = ExpectTokenSave { |
| 1517 | .id = Token.Id.Identifier, |
| 1518 | .ptr = &identifier.token |
| 1519 | } |
| 1520 | }); |
| 1529 | 1521 | continue; |
| 1530 | 1522 | }, |
| 1531 | 1523 | else => { |
| ... | ... | @@ -1546,10 +1538,6 @@ pub const Parser = struct { |
| 1546 | 1538 | dest_ptr.store(&(try self.createLiteral(arena, ast.NodeFloatLiteral, token)).base); |
| 1547 | 1539 | continue; |
| 1548 | 1540 | }, |
| 1549 | | Token.Id.StringLiteral => { |
| 1550 | | dest_ptr.store(&(try self.createLiteral(arena, ast.NodeStringLiteral, token)).base); |
| 1551 | | continue; |
| 1552 | | }, |
| 1553 | 1541 | Token.Id.CharLiteral => { |
| 1554 | 1542 | dest_ptr.store(&(try self.createLiteral(arena, ast.NodeCharLiteral, token)).base); |
| 1555 | 1543 | continue; |
| ... | ... | @@ -1578,24 +1566,8 @@ pub const Parser = struct { |
| 1578 | 1566 | dest_ptr.store(&(try self.createLiteral(arena, ast.NodeUnreachable, token)).base); |
| 1579 | 1567 | continue; |
| 1580 | 1568 | }, |
| 1581 | | Token.Id.MultilineStringLiteralLine => { |
| 1582 | | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeMultilineStringLiteral, |
| 1583 | | ast.NodeMultilineStringLiteral { |
| 1584 | | .base = undefined, |
| 1585 | | .tokens = ArrayList(Token).init(arena), |
| 1586 | | } |
| 1587 | | ); |
| 1588 | | try node.tokens.append(token); |
| 1589 | | while (true) { |
| 1590 | | const multiline_str = self.getNextToken(); |
| 1591 | | if (multiline_str.id != Token.Id.MultilineStringLiteralLine) { |
| 1592 | | self.putBackToken(multiline_str); |
| 1593 | | break; |
| 1594 | | } |
| 1595 | | |
| 1596 | | try node.tokens.append(multiline_str); |
| 1597 | | } |
| 1598 | | continue; |
| 1569 | Token.Id.StringLiteral, Token.Id.MultilineStringLiteralLine => { |
| 1570 | dest_ptr.store((try self.parseStringLiteral(arena, token)) ?? unreachable); |
| 1599 | 1571 | }, |
| 1600 | 1572 | Token.Id.LParen => { |
| 1601 | 1573 | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeGroupedExpression, |
| ... | ... | @@ -1677,11 +1649,8 @@ pub const Parser = struct { |
| 1677 | 1649 | |
| 1678 | 1650 | }, |
| 1679 | 1651 | Token.Id.Keyword_error => { |
| 1680 | | const next = self.getNextToken(); |
| 1681 | | |
| 1682 | | if (next.id != Token.Id.LBrace) { |
| 1652 | if (self.eatToken(Token.Id.LBrace) == null) { |
| 1683 | 1653 | dest_ptr.store(&(try self.createLiteral(arena, ast.NodeErrorType, token)).base); |
| 1684 | | self.putBackToken(next); |
| 1685 | 1654 | continue; |
| 1686 | 1655 | } |
| 1687 | 1656 | |
| ... | ... | @@ -1694,43 +1663,12 @@ pub const Parser = struct { |
| 1694 | 1663 | } |
| 1695 | 1664 | ); |
| 1696 | 1665 | |
| 1697 | | while (true) { |
| 1698 | | const t = self.getNextToken(); |
| 1699 | | switch (t.id) { |
| 1700 | | Token.Id.RBrace => { |
| 1701 | | node.rbrace_token = t; |
| 1702 | | break; |
| 1703 | | }, |
| 1704 | | Token.Id.Identifier => { |
| 1705 | | try node.decls.append( |
| 1706 | | try self.createLiteral(arena, ast.NodeIdentifier, t) |
| 1707 | | ); |
| 1708 | | }, |
| 1709 | | else => { |
| 1710 | | try self.parseError(&stack, token, "expected {} or {}, found {}", |
| 1711 | | @tagName(Token.Id.RBrace), |
| 1712 | | @tagName(Token.Id.Identifier), |
| 1713 | | @tagName(token.id)); |
| 1714 | | continue; |
| 1715 | | } |
| 1716 | | } |
| 1717 | | |
| 1718 | | const t2 = self.getNextToken(); |
| 1719 | | switch (t2.id) { |
| 1720 | | Token.Id.RBrace => { |
| 1721 | | node.rbrace_token = t; |
| 1722 | | break; |
| 1723 | | }, |
| 1724 | | Token.Id.Comma => continue, |
| 1725 | | else => { |
| 1726 | | try self.parseError(&stack, token, "expected {} or {}, found {}", |
| 1727 | | @tagName(Token.Id.RBrace), |
| 1728 | | @tagName(Token.Id.Comma), |
| 1729 | | @tagName(token.id)); |
| 1730 | | continue; |
| 1731 | | } |
| 1666 | stack.append(State { |
| 1667 | .IdentifierListItemOrEnd = ListSave(&ast.NodeIdentifier) { |
| 1668 | .list = &node.decls, |
| 1669 | .ptr = &node.rbrace_token, |
| 1732 | 1670 | } |
| 1733 | | } |
| 1671 | }) catch unreachable; |
| 1734 | 1672 | continue; |
| 1735 | 1673 | }, |
| 1736 | 1674 | Token.Id.Keyword_packed => { |
| ... | ... | @@ -1754,8 +1692,7 @@ pub const Parser = struct { |
| 1754 | 1692 | .params = ArrayList(&ast.Node).init(arena), |
| 1755 | 1693 | .return_type = undefined, |
| 1756 | 1694 | .var_args_token = null, |
| 1757 | | .extern_token = token, |
| 1758 | | .inline_token = null, |
| 1695 | .extern_export_inline_token = token, |
| 1759 | 1696 | .cc_token = null, |
| 1760 | 1697 | .async_attr = null, |
| 1761 | 1698 | .body_node = null, |
| ... | ... | @@ -1812,8 +1749,7 @@ pub const Parser = struct { |
| 1812 | 1749 | .params = ArrayList(&ast.Node).init(arena), |
| 1813 | 1750 | .return_type = undefined, |
| 1814 | 1751 | .var_args_token = null, |
| 1815 | | .extern_token = null, |
| 1816 | | .inline_token = null, |
| 1752 | .extern_export_inline_token = null, |
| 1817 | 1753 | .cc_token = null, |
| 1818 | 1754 | .async_attr = null, |
| 1819 | 1755 | .body_node = null, |
| ... | ... | @@ -1835,8 +1771,7 @@ pub const Parser = struct { |
| 1835 | 1771 | .params = ArrayList(&ast.Node).init(arena), |
| 1836 | 1772 | .return_type = undefined, |
| 1837 | 1773 | .var_args_token = null, |
| 1838 | | .extern_token = null, |
| 1839 | | .inline_token = null, |
| 1774 | .extern_export_inline_token = null, |
| 1840 | 1775 | .cc_token = token, |
| 1841 | 1776 | .async_attr = null, |
| 1842 | 1777 | .body_node = null, |
| ... | ... | @@ -1857,7 +1792,12 @@ pub const Parser = struct { |
| 1857 | 1792 | break :blk true; |
| 1858 | 1793 | }; |
| 1859 | 1794 | _ = (try self.expectToken(&stack, Token.Id.LParen)) ?? continue; |
| 1860 | | const template = (try self.expectToken(&stack, Token.Id.StringLiteral)) ?? continue; |
| 1795 | |
| 1796 | const template_token = self.getNextToken(); |
| 1797 | const template = (try self.parseStringLiteral(arena, template_token)) ?? { |
| 1798 | try self.parseError(&stack, template_token, "expected string literal, found {}", @tagName(template_token.id)); |
| 1799 | continue; |
| 1800 | }; |
| 1861 | 1801 | // TODO parse template |
| 1862 | 1802 | |
| 1863 | 1803 | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeAsm, |
| ... | ... | @@ -1869,7 +1809,7 @@ pub const Parser = struct { |
| 1869 | 1809 | //.tokens = ArrayList(ast.NodeAsm.AsmToken).init(arena), |
| 1870 | 1810 | .outputs = ArrayList(&ast.NodeAsmOutput).init(arena), |
| 1871 | 1811 | .inputs = ArrayList(&ast.NodeAsmInput).init(arena), |
| 1872 | | .cloppers = ArrayList(&ast.NodeStringLiteral).init(arena), |
| 1812 | .cloppers = ArrayList(&ast.Node).init(arena), |
| 1873 | 1813 | .rparen = undefined, |
| 1874 | 1814 | } |
| 1875 | 1815 | ); |
| ... | ... | @@ -1897,7 +1837,9 @@ pub const Parser = struct { |
| 1897 | 1837 | continue; |
| 1898 | 1838 | }, |
| 1899 | 1839 | else => { |
| 1900 | | try self.parseError(&stack, token, "expected primary expression, found {}", @tagName(token.id)); |
| 1840 | if (!try self.parseBlockExpr(&stack, arena, dest_ptr, token)) { |
| 1841 | try self.parseError(&stack, token, "expected primary expression, found {}", @tagName(token.id)); |
| 1842 | } |
| 1901 | 1843 | continue; |
| 1902 | 1844 | } |
| 1903 | 1845 | } |
| ... | ... | @@ -1955,7 +1897,12 @@ pub const Parser = struct { |
| 1955 | 1897 | |
| 1956 | 1898 | const symbolic_name = (try self.expectToken(&stack, Token.Id.Identifier)) ?? continue; |
| 1957 | 1899 | _ = (try self.expectToken(&stack, Token.Id.RBracket)) ?? continue; |
| 1958 | | const constraint = (try self.expectToken(&stack, Token.Id.StringLiteral)) ?? continue; |
| 1900 | |
| 1901 | const constraint_token = self.getNextToken(); |
| 1902 | const constraint = (try self.parseStringLiteral(arena, constraint_token)) ?? { |
| 1903 | try self.parseError(&stack, constraint_token, "expected string literal, found {}", @tagName(constraint_token.id)); |
| 1904 | continue; |
| 1905 | }; |
| 1959 | 1906 | |
| 1960 | 1907 | _ = (try self.expectToken(&stack, Token.Id.LParen)) ?? continue; |
| 1961 | 1908 | try stack.append(State { .ExpectToken = Token.Id.RParen }); |
| ... | ... | @@ -1964,7 +1911,7 @@ pub const Parser = struct { |
| 1964 | 1911 | ast.NodeAsmOutput { |
| 1965 | 1912 | .base = undefined, |
| 1966 | 1913 | .symbolic_name = try self.createLiteral(arena, ast.NodeIdentifier, symbolic_name), |
| 1967 | | .constraint = try self.createLiteral(arena, ast.NodeStringLiteral, constraint), |
| 1914 | .constraint = constraint, |
| 1968 | 1915 | .kind = undefined, |
| 1969 | 1916 | } |
| 1970 | 1917 | ); |
| ... | ... | @@ -2000,7 +1947,12 @@ pub const Parser = struct { |
| 2000 | 1947 | |
| 2001 | 1948 | const symbolic_name = (try self.expectToken(&stack, Token.Id.Identifier)) ?? continue; |
| 2002 | 1949 | _ = (try self.expectToken(&stack, Token.Id.RBracket)) ?? continue; |
| 2003 | | const constraint = (try self.expectToken(&stack, Token.Id.StringLiteral)) ?? continue; |
| 1950 | |
| 1951 | const constraint_token = self.getNextToken(); |
| 1952 | const constraint = (try self.parseStringLiteral(arena, constraint_token)) ?? { |
| 1953 | try self.parseError(&stack, constraint_token, "expected string literal, found {}", @tagName(constraint_token.id)); |
| 1954 | continue; |
| 1955 | }; |
| 2004 | 1956 | |
| 2005 | 1957 | _ = (try self.expectToken(&stack, Token.Id.LParen)) ?? continue; |
| 2006 | 1958 | try stack.append(State { .ExpectToken = Token.Id.RParen }); |
| ... | ... | @@ -2009,7 +1961,7 @@ pub const Parser = struct { |
| 2009 | 1961 | ast.NodeAsmInput { |
| 2010 | 1962 | .base = undefined, |
| 2011 | 1963 | .symbolic_name = try self.createLiteral(arena, ast.NodeIdentifier, symbolic_name), |
| 2012 | | .constraint = try self.createLiteral(arena, ast.NodeStringLiteral, constraint), |
| 1964 | .constraint = constraint, |
| 2013 | 1965 | .expr = undefined, |
| 2014 | 1966 | } |
| 2015 | 1967 | ); |
| ... | ... | @@ -2018,13 +1970,13 @@ pub const Parser = struct { |
| 2018 | 1970 | }, |
| 2019 | 1971 | |
| 2020 | 1972 | State.AsmClopperItems => |items| { |
| 2021 | | const string = self.getNextToken(); |
| 2022 | | if (string.id != Token.Id.StringLiteral) { |
| 2023 | | self.putBackToken(string); |
| 1973 | const string_token = self.getNextToken(); |
| 1974 | const string = (try self.parseStringLiteral(arena, string_token)) ?? { |
| 1975 | self.putBackToken(string_token); |
| 2024 | 1976 | continue; |
| 2025 | | } |
| 1977 | }; |
| 1978 | try items.append(string); |
| 2026 | 1979 | |
| 2027 | | try items.append(try self.createLiteral(arena, ast.NodeStringLiteral, string)); |
| 2028 | 1980 | stack.append(State { .AsmClopperItems = items }) catch unreachable; |
| 2029 | 1981 | try stack.append(State { .IfToken = Token.Id.Comma }); |
| 2030 | 1982 | }, |
| ... | ... | @@ -2076,6 +2028,24 @@ pub const Parser = struct { |
| 2076 | 2028 | }); |
| 2077 | 2029 | }, |
| 2078 | 2030 | |
| 2031 | State.IdentifierListItemOrEnd => |list_state| { |
| 2032 | if (self.eatToken(Token.Id.RBrace)) |rbrace| { |
| 2033 | *list_state.ptr = rbrace; |
| 2034 | continue; |
| 2035 | } |
| 2036 | |
| 2037 | const node = try self.createLiteral(arena, ast.NodeIdentifier, Token(undefined)); |
| 2038 | try list_state.list.append(node); |
| 2039 | |
| 2040 | stack.append(State { .IdentifierListCommaOrEnd = list_state }) catch unreachable; |
| 2041 | try stack.append(State { |
| 2042 | .ExpectTokenSave = ExpectTokenSave { |
| 2043 | .id = Token.Id.Identifier, |
| 2044 | .ptr = &node.token, |
| 2045 | } |
| 2046 | }); |
| 2047 | }, |
| 2048 | |
| 2079 | 2049 | State.SwitchCaseOrEnd => |list_state| { |
| 2080 | 2050 | if (self.eatToken(Token.Id.RBrace)) |rbrace| { |
| 2081 | 2051 | *list_state.ptr = rbrace; |
| ... | ... | @@ -2092,7 +2062,7 @@ pub const Parser = struct { |
| 2092 | 2062 | ); |
| 2093 | 2063 | try list_state.list.append(node); |
| 2094 | 2064 | stack.append(State { .SwitchCaseCommaOrEnd = list_state }) catch unreachable; |
| 2095 | | try stack.append(State { .Expression = DestPtr{ .Field = &node.expr } }); |
| 2065 | try stack.append(State { .AssignmentExpressionBegin = DestPtr{ .Field = &node.expr } }); |
| 2096 | 2066 | try stack.append(State { .PointerPayload = &node.payload }); |
| 2097 | 2067 | |
| 2098 | 2068 | const maybe_else = self.getNextToken(); |
| ... | ... | @@ -2133,6 +2103,11 @@ pub const Parser = struct { |
| 2133 | 2103 | continue; |
| 2134 | 2104 | }, |
| 2135 | 2105 | |
| 2106 | State.IdentifierListCommaOrEnd => |list_state| { |
| 2107 | try self.commaOrEnd(&stack, Token.Id.RBrace, list_state.ptr, State { .IdentifierListItemOrEnd = list_state }); |
| 2108 | continue; |
| 2109 | }, |
| 2110 | |
| 2136 | 2111 | State.SwitchCaseCommaOrEnd => |list_state| { |
| 2137 | 2112 | try self.commaOrEnd(&stack, Token.Id.RBrace, list_state.ptr, State { .SwitchCaseOrEnd = list_state }); |
| 2138 | 2113 | continue; |
| ... | ... | @@ -2355,12 +2330,14 @@ pub const Parser = struct { |
| 2355 | 2330 | }, |
| 2356 | 2331 | |
| 2357 | 2332 | State.FnProtoAlign => |fn_proto| { |
| 2333 | stack.append(State { .FnProtoReturnType = fn_proto }) catch unreachable; |
| 2334 | |
| 2358 | 2335 | if (self.eatToken(Token.Id.Keyword_align)) |align_token| { |
| 2359 | | @panic("TODO fn proto align"); |
| 2336 | try stack.append(State { .ExpectToken = Token.Id.RParen }); |
| 2337 | try stack.append(State { .Expression = DestPtr { .NullableField = &fn_proto.align_expr } }); |
| 2338 | try stack.append(State { .ExpectToken = Token.Id.LParen }); |
| 2360 | 2339 | } |
| 2361 | | stack.append(State { |
| 2362 | | .FnProtoReturnType = fn_proto, |
| 2363 | | }) catch unreachable; |
| 2340 | |
| 2364 | 2341 | continue; |
| 2365 | 2342 | }, |
| 2366 | 2343 | |
| ... | ... | @@ -2374,11 +2351,17 @@ pub const Parser = struct { |
| 2374 | 2351 | }) catch unreachable; |
| 2375 | 2352 | continue; |
| 2376 | 2353 | }, |
| 2377 | | Token.Id.Keyword_align => { |
| 2378 | | @panic("TODO fn proto align"); |
| 2379 | | continue; |
| 2380 | | }, |
| 2381 | 2354 | else => { |
| 2355 | // TODO: this is a special case. Remove this when #760 is fixed |
| 2356 | if (token.id == Token.Id.Keyword_error) { |
| 2357 | if (self.isPeekToken(Token.Id.LBrace)) { |
| 2358 | fn_proto.return_type = ast.NodeFnProto.ReturnType { |
| 2359 | .Explicit = &(try self.createLiteral(arena, ast.NodeErrorType, token)).base |
| 2360 | }; |
| 2361 | continue; |
| 2362 | } |
| 2363 | } |
| 2364 | |
| 2382 | 2365 | self.putBackToken(token); |
| 2383 | 2366 | fn_proto.return_type = ast.NodeFnProto.ReturnType { .Explicit = undefined }; |
| 2384 | 2367 | stack.append(State { |
| ... | ... | @@ -2628,7 +2611,7 @@ pub const Parser = struct { |
| 2628 | 2611 | .visib_token = null, |
| 2629 | 2612 | .mut_token = mut_token, |
| 2630 | 2613 | .comptime_token = next, |
| 2631 | | .extern_token = null, |
| 2614 | .extern_export_token = null, |
| 2632 | 2615 | .type_node = null, |
| 2633 | 2616 | .align_node = null, |
| 2634 | 2617 | .init_node = null, |
| ... | ... | @@ -2656,7 +2639,7 @@ pub const Parser = struct { |
| 2656 | 2639 | .visib_token = null, |
| 2657 | 2640 | .mut_token = next, |
| 2658 | 2641 | .comptime_token = null, |
| 2659 | | .extern_token = null, |
| 2642 | .extern_export_token = null, |
| 2660 | 2643 | .type_node = null, |
| 2661 | 2644 | .align_node = null, |
| 2662 | 2645 | .init_node = null, |
| ... | ... | @@ -2747,7 +2730,7 @@ pub const Parser = struct { |
| 2747 | 2730 | continue; |
| 2748 | 2731 | } |
| 2749 | 2732 | |
| 2750 | | n = while_node.body; |
| 2733 | return while_node.body.id != ast.Node.Id.Block; |
| 2751 | 2734 | }, |
| 2752 | 2735 | ast.Node.Id.For => { |
| 2753 | 2736 | const for_node = @fieldParentPtr(ast.NodeFor, "base", n); |
| ... | ... | @@ -2756,7 +2739,7 @@ pub const Parser = struct { |
| 2756 | 2739 | continue; |
| 2757 | 2740 | } |
| 2758 | 2741 | |
| 2759 | | n = for_node.body; |
| 2742 | return for_node.body.id != ast.Node.Id.Block; |
| 2760 | 2743 | }, |
| 2761 | 2744 | ast.Node.Id.If => { |
| 2762 | 2745 | const if_node = @fieldParentPtr(ast.NodeIf, "base", n); |
| ... | ... | @@ -2765,25 +2748,25 @@ pub const Parser = struct { |
| 2765 | 2748 | continue; |
| 2766 | 2749 | } |
| 2767 | 2750 | |
| 2768 | | n = if_node.body; |
| 2751 | return if_node.body.id != ast.Node.Id.Block; |
| 2769 | 2752 | }, |
| 2770 | 2753 | ast.Node.Id.Else => { |
| 2771 | 2754 | const else_node = @fieldParentPtr(ast.NodeElse, "base", n); |
| 2772 | 2755 | n = else_node.body; |
| 2756 | continue; |
| 2773 | 2757 | }, |
| 2774 | 2758 | ast.Node.Id.Defer => { |
| 2775 | 2759 | const defer_node = @fieldParentPtr(ast.NodeDefer, "base", n); |
| 2776 | | n = defer_node.expr; |
| 2760 | return defer_node.expr.id != ast.Node.Id.Block; |
| 2777 | 2761 | }, |
| 2778 | 2762 | ast.Node.Id.Comptime => { |
| 2779 | 2763 | const comptime_node = @fieldParentPtr(ast.NodeComptime, "base", n); |
| 2780 | | n = comptime_node.expr; |
| 2764 | return comptime_node.expr.id != ast.Node.Id.Block; |
| 2781 | 2765 | }, |
| 2782 | 2766 | ast.Node.Id.Suspend => { |
| 2783 | 2767 | const suspend_node = @fieldParentPtr(ast.NodeSuspend, "base", n); |
| 2784 | 2768 | if (suspend_node.body) |body| { |
| 2785 | | n = body; |
| 2786 | | continue; |
| 2769 | return body.id != ast.Node.Id.Block; |
| 2787 | 2770 | } |
| 2788 | 2771 | |
| 2789 | 2772 | return true; |
| ... | ... | @@ -2793,6 +2776,148 @@ pub const Parser = struct { |
| 2793 | 2776 | } |
| 2794 | 2777 | } |
| 2795 | 2778 | |
| 2779 | fn parseStringLiteral(self: &Parser, arena: &mem.Allocator, token: &const Token) !?&ast.Node { |
| 2780 | switch (token.id) { |
| 2781 | Token.Id.StringLiteral => { |
| 2782 | return &(try self.createLiteral(arena, ast.NodeStringLiteral, token)).base; |
| 2783 | }, |
| 2784 | Token.Id.MultilineStringLiteralLine => { |
| 2785 | const node = try self.createNode(arena, ast.NodeMultilineStringLiteral, |
| 2786 | ast.NodeMultilineStringLiteral { |
| 2787 | .base = undefined, |
| 2788 | .tokens = ArrayList(Token).init(arena), |
| 2789 | } |
| 2790 | ); |
| 2791 | try node.tokens.append(token); |
| 2792 | while (true) { |
| 2793 | const multiline_str = self.getNextToken(); |
| 2794 | if (multiline_str.id != Token.Id.MultilineStringLiteralLine) { |
| 2795 | self.putBackToken(multiline_str); |
| 2796 | break; |
| 2797 | } |
| 2798 | |
| 2799 | try node.tokens.append(multiline_str); |
| 2800 | } |
| 2801 | |
| 2802 | return &node.base; |
| 2803 | }, |
| 2804 | // TODO: We shouldn't need a cast, but: |
| 2805 | // zig: /home/jc/Documents/zig/src/ir.cpp:7962: TypeTableEntry* ir_resolve_peer_types(IrAnalyze*, AstNode*, IrInstruction**, size_t): Assertion `err_set_type != nullptr' failed. |
| 2806 | else => return (?&ast.Node)(null), |
| 2807 | } |
| 2808 | } |
| 2809 | |
| 2810 | fn parseBlockExpr(self: &Parser, stack: &ArrayList(State), arena: &mem.Allocator, dest_ptr: &const DestPtr, token: &const Token) !bool { |
| 2811 | switch (token.id) { |
| 2812 | Token.Id.Keyword_suspend => { |
| 2813 | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeSuspend, |
| 2814 | ast.NodeSuspend { |
| 2815 | .base = undefined, |
| 2816 | .suspend_token = *token, |
| 2817 | .payload = null, |
| 2818 | .body = null, |
| 2819 | } |
| 2820 | ); |
| 2821 | |
| 2822 | stack.append(State { .SuspendBody = node }) catch unreachable; |
| 2823 | try stack.append(State { .Payload = &node.payload }); |
| 2824 | return true; |
| 2825 | }, |
| 2826 | Token.Id.Keyword_if => { |
| 2827 | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeIf, |
| 2828 | ast.NodeIf { |
| 2829 | .base = undefined, |
| 2830 | .if_token = *token, |
| 2831 | .condition = undefined, |
| 2832 | .payload = null, |
| 2833 | .body = undefined, |
| 2834 | .@"else" = null, |
| 2835 | } |
| 2836 | ); |
| 2837 | |
| 2838 | stack.append(State { .Else = &node.@"else" }) catch unreachable; |
| 2839 | try stack.append(State { .Expression = DestPtr { .Field = &node.body } }); |
| 2840 | try stack.append(State { .PointerPayload = &node.payload }); |
| 2841 | try stack.append(State { .ExpectToken = Token.Id.RParen }); |
| 2842 | try stack.append(State { .Expression = DestPtr { .Field = &node.condition } }); |
| 2843 | try stack.append(State { .ExpectToken = Token.Id.LParen }); |
| 2844 | return true; |
| 2845 | }, |
| 2846 | Token.Id.Keyword_while => { |
| 2847 | stack.append(State { |
| 2848 | .While = LoopCtx { |
| 2849 | .label = null, |
| 2850 | .inline_token = null, |
| 2851 | .loop_token = *token, |
| 2852 | .dest_ptr = *dest_ptr, |
| 2853 | } |
| 2854 | }) catch unreachable; |
| 2855 | return true; |
| 2856 | }, |
| 2857 | Token.Id.Keyword_for => { |
| 2858 | stack.append(State { |
| 2859 | .For = LoopCtx { |
| 2860 | .label = null, |
| 2861 | .inline_token = null, |
| 2862 | .loop_token = *token, |
| 2863 | .dest_ptr = *dest_ptr, |
| 2864 | } |
| 2865 | }) catch unreachable; |
| 2866 | return true; |
| 2867 | }, |
| 2868 | Token.Id.Keyword_switch => { |
| 2869 | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeSwitch, |
| 2870 | ast.NodeSwitch { |
| 2871 | .base = undefined, |
| 2872 | .switch_token = *token, |
| 2873 | .expr = undefined, |
| 2874 | .cases = ArrayList(&ast.NodeSwitchCase).init(arena), |
| 2875 | .rbrace = undefined, |
| 2876 | } |
| 2877 | ); |
| 2878 | |
| 2879 | stack.append(State { |
| 2880 | .SwitchCaseOrEnd = ListSave(&ast.NodeSwitchCase) { |
| 2881 | .list = &node.cases, |
| 2882 | .ptr = &node.rbrace, |
| 2883 | }, |
| 2884 | }) catch unreachable; |
| 2885 | try stack.append(State { .ExpectToken = Token.Id.LBrace }); |
| 2886 | try stack.append(State { .ExpectToken = Token.Id.RParen }); |
| 2887 | try stack.append(State { .Expression = DestPtr { .Field = &node.expr } }); |
| 2888 | try stack.append(State { .ExpectToken = Token.Id.LParen }); |
| 2889 | return true; |
| 2890 | }, |
| 2891 | Token.Id.Keyword_comptime => { |
| 2892 | const node = try self.createToDestNode(arena, dest_ptr, ast.NodeComptime, |
| 2893 | ast.NodeComptime { |
| 2894 | .base = undefined, |
| 2895 | .comptime_token = *token, |
| 2896 | .expr = undefined, |
| 2897 | } |
| 2898 | ); |
| 2899 | try stack.append(State { .Expression = DestPtr { .Field = &node.expr } }); |
| 2900 | return true; |
| 2901 | }, |
| 2902 | Token.Id.LBrace => { |
| 2903 | const block = try self.createToDestNode(arena, dest_ptr, ast.NodeBlock, |
| 2904 | ast.NodeBlock { |
| 2905 | .base = undefined, |
| 2906 | .label = null, |
| 2907 | .lbrace = *token, |
| 2908 | .statements = ArrayList(&ast.Node).init(arena), |
| 2909 | .rbrace = undefined, |
| 2910 | } |
| 2911 | ); |
| 2912 | stack.append(State { .Block = block }) catch unreachable; |
| 2913 | return true; |
| 2914 | }, |
| 2915 | else => { |
| 2916 | return false; |
| 2917 | } |
| 2918 | } |
| 2919 | } |
| 2920 | |
| 2796 | 2921 | fn commaOrEnd(self: &Parser, stack: &ArrayList(State), end: &const Token.Id, maybe_ptr: ?&Token, state_after_comma: &const State) !void { |
| 2797 | 2922 | var token = self.getNextToken(); |
| 2798 | 2923 | switch (token.id) { |
| ... | ... | @@ -2884,7 +3009,7 @@ pub const Parser = struct { |
| 2884 | 3009 | Token.Id.Tilde => ast.NodePrefixOp.PrefixOp { .BitNot = void{} }, |
| 2885 | 3010 | Token.Id.Minus => ast.NodePrefixOp.PrefixOp { .Negation = void{} }, |
| 2886 | 3011 | Token.Id.MinusPercent => ast.NodePrefixOp.PrefixOp { .NegationWrap = void{} }, |
| 2887 | | Token.Id.Asterisk => ast.NodePrefixOp.PrefixOp { .Deref = void{} }, |
| 3012 | Token.Id.Asterisk, Token.Id.AsteriskAsterisk => ast.NodePrefixOp.PrefixOp { .Deref = void{} }, |
| 2888 | 3013 | Token.Id.Ampersand => ast.NodePrefixOp.PrefixOp { |
| 2889 | 3014 | .AddrOf = ast.NodePrefixOp.AddrOfInfo { |
| 2890 | 3015 | .align_expr = null, |
| ... | ... | @@ -2897,6 +3022,7 @@ pub const Parser = struct { |
| 2897 | 3022 | Token.Id.QuestionMark => ast.NodePrefixOp.PrefixOp { .MaybeType = void{} }, |
| 2898 | 3023 | Token.Id.QuestionMarkQuestionMark => ast.NodePrefixOp.PrefixOp { .UnwrapMaybe = void{} }, |
| 2899 | 3024 | Token.Id.Keyword_await => ast.NodePrefixOp.PrefixOp { .Await = void{} }, |
| 3025 | Token.Id.Keyword_try => ast.NodePrefixOp.PrefixOp { .Try = void{ } }, |
| 2900 | 3026 | else => null, |
| 2901 | 3027 | }; |
| 2902 | 3028 | } |
| ... | ... | @@ -3051,7 +3177,6 @@ pub const Parser = struct { |
| 3051 | 3177 | |
| 3052 | 3178 | const RenderState = union(enum) { |
| 3053 | 3179 | TopLevelDecl: &ast.Node, |
| 3054 | | FnProtoRParen: &ast.NodeFnProto, |
| 3055 | 3180 | ParamDecl: &ast.Node, |
| 3056 | 3181 | Text: []const u8, |
| 3057 | 3182 | Expression: &ast.Node, |
| ... | ... | @@ -3129,6 +3254,9 @@ pub const Parser = struct { |
| 3129 | 3254 | }, |
| 3130 | 3255 | ast.Node.Id.StructField => { |
| 3131 | 3256 | const field = @fieldParentPtr(ast.NodeStructField, "base", decl); |
| 3257 | if (field.visib_token) |visib_token| { |
| 3258 | try stream.print("{} ", self.tokenizer.getTokenSlice(visib_token)); |
| 3259 | } |
| 3132 | 3260 | try stream.print("{}: ", self.tokenizer.getTokenSlice(field.name_token)); |
| 3133 | 3261 | try stack.append(RenderState { .Expression = field.type_expr}); |
| 3134 | 3262 | }, |
| ... | ... | @@ -3190,13 +3318,13 @@ pub const Parser = struct { |
| 3190 | 3318 | try stack.append(RenderState { .Text = self.tokenizer.getTokenSlice(comptime_token) }); |
| 3191 | 3319 | } |
| 3192 | 3320 | |
| 3193 | | if (var_decl.extern_token) |extern_token| { |
| 3321 | if (var_decl.extern_export_token) |extern_export_token| { |
| 3194 | 3322 | if (var_decl.lib_name != null) { |
| 3195 | 3323 | try stack.append(RenderState { .Text = " " }); |
| 3196 | 3324 | try stack.append(RenderState { .Expression = ??var_decl.lib_name }); |
| 3197 | 3325 | } |
| 3198 | 3326 | try stack.append(RenderState { .Text = " " }); |
| 3199 | | try stack.append(RenderState { .Text = self.tokenizer.getTokenSlice(extern_token) }); |
| 3327 | try stack.append(RenderState { .Text = self.tokenizer.getTokenSlice(extern_export_token) }); |
| 3200 | 3328 | } |
| 3201 | 3329 | |
| 3202 | 3330 | if (var_decl.visib_token) |visib_token| { |
| ... | ... | @@ -3737,8 +3865,10 @@ pub const Parser = struct { |
| 3737 | 3865 | }, |
| 3738 | 3866 | } |
| 3739 | 3867 | |
| 3740 | | if (fn_proto.align_expr != null) { |
| 3741 | | @panic("TODO"); |
| 3868 | if (fn_proto.align_expr) |align_expr| { |
| 3869 | try stack.append(RenderState { .Text = ") " }); |
| 3870 | try stack.append(RenderState { .Expression = align_expr}); |
| 3871 | try stack.append(RenderState { .Text = "align(" }); |
| 3742 | 3872 | } |
| 3743 | 3873 | |
| 3744 | 3874 | try stack.append(RenderState { .Text = ") " }); |
| ... | ... | @@ -3774,9 +3904,9 @@ pub const Parser = struct { |
| 3774 | 3904 | try stack.append(RenderState { .Text = " " }); |
| 3775 | 3905 | try stack.append(RenderState { .Expression = lib_name }); |
| 3776 | 3906 | } |
| 3777 | | if (fn_proto.extern_token) |extern_token| { |
| 3907 | if (fn_proto.extern_export_inline_token) |extern_export_inline_token| { |
| 3778 | 3908 | try stack.append(RenderState { .Text = " " }); |
| 3779 | | try stack.append(RenderState { .Text = self.tokenizer.getTokenSlice(extern_token) }); |
| 3909 | try stack.append(RenderState { .Text = self.tokenizer.getTokenSlice(extern_export_inline_token) }); |
| 3780 | 3910 | } |
| 3781 | 3911 | |
| 3782 | 3912 | if (fn_proto.visib_token) |visib_token| { |
| ... | ... | @@ -4021,8 +4151,6 @@ pub const Parser = struct { |
| 4021 | 4151 | try stream.write("volatile "); |
| 4022 | 4152 | } |
| 4023 | 4153 | |
| 4024 | | try stream.print("({}", self.tokenizer.getTokenSlice(asm_node.template)); |
| 4025 | | |
| 4026 | 4154 | try stack.append(RenderState { .Indent = indent }); |
| 4027 | 4155 | try stack.append(RenderState { .Text = ")" }); |
| 4028 | 4156 | { |
| ... | ... | @@ -4030,7 +4158,7 @@ pub const Parser = struct { |
| 4030 | 4158 | var i = cloppers.len; |
| 4031 | 4159 | while (i != 0) { |
| 4032 | 4160 | i -= 1; |
| 4033 | | try stack.append(RenderState { .Expression = &cloppers[i].base }); |
| 4161 | try stack.append(RenderState { .Expression = cloppers[i] }); |
| 4034 | 4162 | |
| 4035 | 4163 | if (i != 0) { |
| 4036 | 4164 | try stack.append(RenderState { .Text = ", " }); |
| ... | ... | @@ -4099,6 +4227,8 @@ pub const Parser = struct { |
| 4099 | 4227 | try stack.append(RenderState.PrintIndent); |
| 4100 | 4228 | try stack.append(RenderState { .Indent = indent + indent_delta}); |
| 4101 | 4229 | try stack.append(RenderState { .Text = "\n" }); |
| 4230 | try stack.append(RenderState { .Expression = asm_node.template }); |
| 4231 | try stack.append(RenderState { .Text = "(" }); |
| 4102 | 4232 | }, |
| 4103 | 4233 | ast.Node.Id.AsmInput => { |
| 4104 | 4234 | const asm_input = @fieldParentPtr(ast.NodeAsmInput, "base", base); |
| ... | ... | @@ -4106,7 +4236,7 @@ pub const Parser = struct { |
| 4106 | 4236 | try stack.append(RenderState { .Text = ")"}); |
| 4107 | 4237 | try stack.append(RenderState { .Expression = asm_input.expr}); |
| 4108 | 4238 | try stack.append(RenderState { .Text = " ("}); |
| 4109 | | try stack.append(RenderState { .Expression = &asm_input.constraint.base}); |
| 4239 | try stack.append(RenderState { .Expression = asm_input.constraint }); |
| 4110 | 4240 | try stack.append(RenderState { .Text = "] "}); |
| 4111 | 4241 | try stack.append(RenderState { .Expression = &asm_input.symbolic_name.base}); |
| 4112 | 4242 | try stack.append(RenderState { .Text = "["}); |
| ... | ... | @@ -4125,7 +4255,7 @@ pub const Parser = struct { |
| 4125 | 4255 | }, |
| 4126 | 4256 | } |
| 4127 | 4257 | try stack.append(RenderState { .Text = " ("}); |
| 4128 | | try stack.append(RenderState { .Expression = &asm_output.constraint.base}); |
| 4258 | try stack.append(RenderState { .Expression = asm_output.constraint }); |
| 4129 | 4259 | try stack.append(RenderState { .Text = "] "}); |
| 4130 | 4260 | try stack.append(RenderState { .Expression = &asm_output.symbolic_name.base}); |
| 4131 | 4261 | try stack.append(RenderState { .Text = "["}); |
| ... | ... | @@ -4140,26 +4270,6 @@ pub const Parser = struct { |
| 4140 | 4270 | ast.Node.Id.TestDecl, |
| 4141 | 4271 | ast.Node.Id.ParamDecl => unreachable, |
| 4142 | 4272 | }, |
| 4143 | | RenderState.FnProtoRParen => |fn_proto| { |
| 4144 | | try stream.print(")"); |
| 4145 | | if (fn_proto.align_expr != null) { |
| 4146 | | @panic("TODO"); |
| 4147 | | } |
| 4148 | | try stream.print(" "); |
| 4149 | | if (fn_proto.body_node) |body_node| { |
| 4150 | | try stack.append(RenderState { .Expression = body_node}); |
| 4151 | | try stack.append(RenderState { .Text = " "}); |
| 4152 | | } |
| 4153 | | switch (fn_proto.return_type) { |
| 4154 | | ast.NodeFnProto.ReturnType.Explicit => |node| { |
| 4155 | | try stack.append(RenderState { .Expression = node}); |
| 4156 | | }, |
| 4157 | | ast.NodeFnProto.ReturnType.InferErrorSet => |node| { |
| 4158 | | try stream.print("!"); |
| 4159 | | try stack.append(RenderState { .Expression = node}); |
| 4160 | | }, |
| 4161 | | } |
| 4162 | | }, |
| 4163 | 4273 | RenderState.Statement => |base| { |
| 4164 | 4274 | if (base.comment) |comment| { |
| 4165 | 4275 | for (comment.lines.toSliceConst()) |line_token| { |
| ... | ... | @@ -4513,10 +4623,20 @@ test "zig fmt: var type" { |
| 4513 | 4623 | ); |
| 4514 | 4624 | } |
| 4515 | 4625 | |
| 4516 | | test "zig fmt: extern function" { |
| 4626 | test "zig fmt: functions" { |
| 4517 | 4627 | try testCanonical( |
| 4518 | 4628 | \\extern fn puts(s: &const u8) c_int; |
| 4519 | 4629 | \\extern "c" fn puts(s: &const u8) c_int; |
| 4630 | \\export fn puts(s: &const u8) c_int; |
| 4631 | \\inline fn puts(s: &const u8) c_int; |
| 4632 | \\pub extern fn puts(s: &const u8) c_int; |
| 4633 | \\pub extern "c" fn puts(s: &const u8) c_int; |
| 4634 | \\pub export fn puts(s: &const u8) c_int; |
| 4635 | \\pub inline fn puts(s: &const u8) c_int; |
| 4636 | \\pub extern fn puts(s: &const u8) align(2 + 2) c_int; |
| 4637 | \\pub extern "c" fn puts(s: &const u8) align(2 + 2) c_int; |
| 4638 | \\pub export fn puts(s: &const u8) align(2 + 2) c_int; |
| 4639 | \\pub inline fn puts(s: &const u8) align(2 + 2) c_int; |
| 4520 | 4640 | \\ |
| 4521 | 4641 | ); |
| 4522 | 4642 | } |
| ... | ... | @@ -4576,6 +4696,7 @@ test "zig fmt: struct declaration" { |
| 4576 | 4696 | \\const S = struct { |
| 4577 | 4697 | \\ const Self = this; |
| 4578 | 4698 | \\ f1: u8, |
| 4699 | \\ pub f3: u8, |
| 4579 | 4700 | \\ |
| 4580 | 4701 | \\ fn method(self: &Self) Self { |
| 4581 | 4702 | \\ return *self; |
| ... | ... | @@ -4586,14 +4707,14 @@ test "zig fmt: struct declaration" { |
| 4586 | 4707 | \\ |
| 4587 | 4708 | \\const Ps = packed struct { |
| 4588 | 4709 | \\ a: u8, |
| 4589 | | \\ b: u8, |
| 4710 | \\ pub b: u8, |
| 4590 | 4711 | \\ |
| 4591 | 4712 | \\ c: u8 |
| 4592 | 4713 | \\}; |
| 4593 | 4714 | \\ |
| 4594 | 4715 | \\const Es = extern struct { |
| 4595 | 4716 | \\ a: u8, |
| 4596 | | \\ b: u8, |
| 4717 | \\ pub b: u8, |
| 4597 | 4718 | \\ |
| 4598 | 4719 | \\ c: u8 |
| 4599 | 4720 | \\}; |
| ... | ... | @@ -4753,6 +4874,7 @@ test "zig fmt: switch" { |
| 4753 | 4874 | \\ const res = switch (0) { |
| 4754 | 4875 | \\ 0 => 0, |
| 4755 | 4876 | \\ 1 => 2, |
| 4877 | \\ 1 => a = 4, |
| 4756 | 4878 | \\ else => 4 |
| 4757 | 4879 | \\ }; |
| 4758 | 4880 | \\ |
| ... | ... | @@ -4898,6 +5020,7 @@ test "zig fmt: if" { |
| 4898 | 5020 | \\ } |
| 4899 | 5021 | \\ |
| 4900 | 5022 | \\ const is_world_broken = if (10 < 0) true else false; |
| 5023 | \\ const some_number = 1 + if (10 < 0) 2 else 3; |
| 4901 | 5024 | \\ |
| 4902 | 5025 | \\ const a: ?u8 = 10; |
| 4903 | 5026 | \\ const b: ?u8 = null; |
| ... | ... | @@ -5011,6 +5134,7 @@ test "zig fmt: inline asm" { |
| 5011 | 5134 | test "zig fmt: coroutines" { |
| 5012 | 5135 | try testCanonical( |
| 5013 | 5136 | \\async fn simpleAsyncFn() void { |
| 5137 | \\ const a = async a.b(); |
| 5014 | 5138 | \\ x += 1; |
| 5015 | 5139 | \\ suspend; |
| 5016 | 5140 | \\ x += 1; |
| ... | ... | @@ -5058,3 +5182,13 @@ test "zig fmt: string identifier" { |
| 5058 | 5182 | \\ |
| 5059 | 5183 | ); |
| 5060 | 5184 | } |
| 5185 | |
| 5186 | test "zig fmt: error return" { |
| 5187 | try testCanonical( |
| 5188 | \\fn err() error { |
| 5189 | \\ call(); |
| 5190 | \\ return error.InvalidArgs; |
| 5191 | \\} |
| 5192 | \\ |
| 5193 | ); |
| 5194 | } |