From fdbf212515d018a98db3932d429f98417dc070ed Mon Sep 17 00:00:00 2001
From: Jarred Sumner <jarred@jarredsumner.com>
Date: Fri, 23 Apr 2021 12:37:34 -0700
Subject: [PATCH] decorator func enum

---
 src/js_parser.zig | 107 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 106 insertions(+), 1 deletion(-)

diff --git a/src/js_parser.zig b/src/js_parser.zig
index f435d50fa3..4e2e4e6ad4 100644
--- a/src/js_parser.zig
+++ b/src/js_parser.zig
@@ -317,7 +317,7 @@ const DeferredTsDecorators = struct {
 
     // If this turns out to be a "declare class" statement, we need to undo the
     // scopes that were potentially pushed while parsing the decorator arguments.
-    scopeIndex: usize,
+    scope_index: usize,
 };
 
 const LexicalDecl = enum(u8) { forbid, allow_all, allow_fn_inside_if, allow_fn_inside_label };
@@ -1549,6 +1549,106 @@ const P = struct {
                 }
             },
 
+            .t_function => {
+                p.lexer.next();
+                return p.parseFnStmt(loc, opts, null);
+            },
+            .t_enum => {
+                if (!p.options.ts) {
+                    p.lexer.unexpected();
+                }
+                return p.parseTypescriptEnumStmt(loc, opts);
+            },
+            .t_at => {
+                // Parse decorators before class statements, which are potentially exported
+                if (p.options.ts) {
+                    const scope_index = p.scopes_in_order.items.len;
+                    const ts_decorators = p.parseTypeScriptDecorators();
+
+                    // If this turns out to be a "declare class" statement, we need to undo the
+                    // scopes that were potentially pushed while parsing the decorator arguments.
+                    // That can look like any one of the following:
+                    //
+                    //   "@decorator declare class Foo {}"
+                    //   "@decorator declare abstract class Foo {}"
+                    //   "@decorator export declare class Foo {}"
+                    //   "@decorator export declare abstract class Foo {}"
+                    //
+                    opts.ts_decorators = DeferredTsDecorators{
+                        .values = ts_decorators,
+                        .scope_index = scope_index,
+                    };
+
+                    // "@decorator class Foo {}"
+                    // "@decorator abstract class Foo {}"
+                    // "@decorator declare class Foo {}"
+                    // "@decorator declare abstract class Foo {}"
+                    // "@decorator export class Foo {}"
+                    // "@decorator export abstract class Foo {}"
+                    // "@decorator export declare class Foo {}"
+                    // "@decorator export declare abstract class Foo {}"
+                    // "@decorator export default class Foo {}"
+                    // "@decorator export default abstract class Foo {}"
+                    if (p.lexer.token != .t_class and p.lexer.token != .t_export and !p.lexer.isContextualKeyword("abstract") and !p.lexer.isContextualKeyword("declare")) {
+                        p.lexer.expected(.t_class);
+                    }
+
+                    return p.parseStmt(opts);
+                }
+                notimpl();
+            },
+            .t_class => {
+                notimpl();
+            },
+            .t_var => {
+                notimpl();
+            },
+            .t_const => {
+                notimpl();
+            },
+            .t_if => {
+                notimpl();
+            },
+            .t_do => {
+                notimpl();
+            },
+            .t_while => {
+                notimpl();
+            },
+            .t_with => {
+                notimpl();
+            },
+            .t_switch => {
+                notimpl();
+            },
+            .t_try => {
+                notimpl();
+            },
+            .t_for => {
+                notimpl();
+            },
+            .t_import => {
+                notimpl();
+            },
+            .t_break => {
+                notimpl();
+            },
+            .t_continue => {
+                notimpl();
+            },
+            .t_return => {
+                notimpl();
+            },
+            .t_throw => {
+                notimpl();
+            },
+            .t_debugger => {
+                notimpl();
+            },
+            .t_open_brace => {
+                notimpl();
+            },
+
             else => {
                 notimpl();
             },
@@ -1557,6 +1657,11 @@ const P = struct {
         return js_ast.Stmt.empty();
     }
 
+    pub fn parseTypescriptEnumStmt(p: *P, loc: logger.Loc, opts: *ParseStatementOptions) Stmt {
+        notimpl();
+        // return Stmt.empty();
+    }
+
     pub fn parseExportClause(p: *P) ExportClauseResult {
         var items = List(js_ast.ClauseItem).initCapacity(p.allocator, 1) catch unreachable;
         var first_keyword_item_loc = logger.Loc{};