fix: match assemblyscript v22 enum identifiers to TitleCase

Author: HerrCai0907

index.mjs

@@ -1,6 +1,7 @@
 #!/usr/bin/env node
 
 import assemblyscript from "assemblyscript";
+import asc from "assemblyscript/asc";
 import prettier from "prettier";
 import * as fs from "fs";
 import { Command } from "commander";
@@ -10,6 +11,8 @@ import chalk from "chalk";
 import ignore from "ignore";
 import { SingleBar } from "cli-progress";
 
+const ascMajorVersion = parseInt(asc.version.split(".")[0]);
+
 const readFile = fs.promises.readFile;
 const writeFile = fs.promises.writeFile;
 
@@ -31,35 +34,69 @@ function preProcess(code) {
   function visitDecorators(node) {
     let list = [];
     let _visit = (_node) => {
-      switch (_node.kind) {
-        case NodeKind.SOURCE: {
-          _node.statements.forEach((statement) => {
-            _visit(statement);
-          });
-          break;
-        }
-        case NodeKind.CLASSDECLARATION:
-        case NodeKind.INTERFACEDECLARATION:
-        case NodeKind.NAMESPACEDECLARATION: {
-          _node.members.forEach((statement) => {
-            _visit(statement);
-          });
-          break;
+      if (ascMajorVersion < 22) {
+        switch (_node.kind) {
+          case NodeKind.SOURCE: {
+            _node.statements.forEach((statement) => {
+              _visit(statement);
+            });
+            break;
+          }
+          case NodeKind.CLASSDECLARATION:
+          case NodeKind.INTERFACEDECLARATION:
+          case NodeKind.NAMESPACEDECLARATION: {
+            _node.members.forEach((statement) => {
+              _visit(statement);
+            });
+            break;
+          }
+          case NodeKind.ENUMDECLARATION:
+          case NodeKind.METHODDECLARATION:
+          case NodeKind.FUNCTIONDECLARATION: {
+            if (_node.decorators) {
+              list.push(
+                ..._node.decorators.map((decorator) => {
+                  return {
+                    start: decorator.range.start,
+                    end: decorator.range.end,
+                  };
+                })
+              );
+            }
+            break;
+          }
         }
-        case NodeKind.ENUMDECLARATION:
-        case NodeKind.METHODDECLARATION:
-        case NodeKind.FUNCTIONDECLARATION: {
-          if (_node.decorators) {
-            list.push(
-              ..._node.decorators.map((decorator) => {
-                return {
-                  start: decorator.range.start,
-                  end: decorator.range.end,
-                };
-              })
-            );
+      } else {
+        switch (_node.kind) {
+          case NodeKind.Source: {
+            _node.statements.forEach((statement) => {
+              _visit(statement);
+            });
+            break;
+          }
+          case NodeKind.ClassDeclaration:
+          case NodeKind.InterfaceDeclaration:
+          case NodeKind.NamespaceDeclaration: {
+            _node.members.forEach((statement) => {
+              _visit(statement);
+            });
+            break;
+          }
+          case NodeKind.EnumDeclaration:
+          case NodeKind.MethodDeclaration:
+          case NodeKind.FunctionDeclaration: {
+            if (_node.decorators) {
+              list.push(
+                ..._node.decorators.map((decorator) => {
+                  return {
+                    start: decorator.range.start,
+                    end: decorator.range.end,
+                  };
+                })
+              );
+            }
+            break;
           }
-          break;
         }
       }
     };
@@ -103,8 +140,8 @@ async function format(path) {
   const tsCode = preProcess(code);
   let config = await resolveConfig(path);
   const formatTsCode = prettier.format(tsCode, config);
-  const foramtCode = postProcess(formatTsCode);
-  return foramtCode;
+  const formatCode = postProcess(formatTsCode);
+  return formatCode;
 }
 async function check(path) {
   const code = await readFile(path, { encoding: "utf8" });

package.json

@@ -28,7 +28,7 @@
     "url": "https://github.com/HerrCai0907/assemblyscript-prettier/issues"
   },
   "peerDependencies": {
-    "assemblyscript": "^0.21.0",
+    "assemblyscript": ">=0.20.0",
     "prettier": "^2.7.1"
   },
   "dependencies": {