json-as 1.1.20 → 1.1.22

package/transform/src/types.ts

@@ -1,4 +1,4 @@
-import { ClassDeclaration, Expression, FieldDeclaration, Source, NodeKind, Node, NamespaceDeclaration, DeclarationStatement, TypeName, Parser, ImportStatement, CommonFlags } from "assemblyscript/dist/assemblyscript.js";
+import { ClassDeclaration, Expression, FieldDeclaration, Source, NodeKind, Node, NamespaceDeclaration, DeclarationStatement, TypeName, Parser, ImportStatement, CommonFlags, EnumDeclaration } from "assemblyscript/dist/assemblyscript.js";
 import { TypeAlias } from "./linkers/alias.js";
 import { stripNull } from "./index.js";
 
@@ -100,9 +100,10 @@ export class Src {
   public schemas: Schema[];
   public aliases: TypeAlias[];
   public exports: Schema[];
+  public imports: ImportStatement[] = [];
   private nodeMap: Map<Node, NamespaceDeclaration[]> = new Map<Node, NamespaceDeclaration[]>();
   private classes: Record<string, ClassDeclaration> = {};
-  private imports: ImportStatement[] = [];
+  private enums: Record<string, EnumDeclaration> = {};
 
   constructor(
     source: Source,
@@ -130,6 +131,10 @@ export class Src {
           const classDeclaration = node as ClassDeclaration;
           this.classes[this.qualifiedName(classDeclaration, path)] = classDeclaration;
           break;
+        case NodeKind.EnumDeclaration:
+          const enumDeclaration = node as EnumDeclaration;
+          this.enums[this.qualifiedName(enumDeclaration, path)] = enumDeclaration;
+          break;
         case NodeKind.Import:
           const importStatement = node as ImportStatement;
           this.imports.push(importStatement);
@@ -157,6 +162,15 @@ export class Src {
     return this.classes[qualifiedName] || null;
   }
 
+  /**
+   * Get an enum declaration by its qualified name.
+   * @param qualifiedName Qualified name (eg. "Namespace.MyEnum")
+   * @returns Enum declaration or null if not found.
+   */
+  getEnum(qualifiedName: string): EnumDeclaration | null {
+    return this.enums[qualifiedName] || null;
+  }
+
   /**
    * Get imported class from other sources in the parser.
    * @param qualifiedName Qualified name of class.
@@ -177,6 +191,26 @@ export class Src {
     return null;
   }
 
+  /**
+   * Get imported enum from other sources in the parser.
+   * @param qualifiedName Qualified name of enum.
+   * @param parser AssemblyScript parser.
+   * @returns Enum declaration or null if not found.
+   */
+  getImportedEnum(qualifiedName: string, parser: Parser): EnumDeclaration | null {
+    for (const stmt of this.imports) {
+      const externalSource = parser.sources.filter((src) => src.internalPath != this.internalPath).find((src) => src.internalPath == stmt.internalPath);
+      if (!externalSource) continue;
+
+      const source = this.sourceSet.get(externalSource);
+      const enumDeclaration = source.getEnum(qualifiedName);
+      if (enumDeclaration && enumDeclaration.flags & CommonFlags.Export) {
+        return enumDeclaration;
+      }
+    }
+    return null;
+  }
+
   /**
    * Gets a unique path string to the node by combining the internalPath with
    * the qualified name of the node.