as-test 0.4.0 → 0.4.2

package/transform/src/coverage.ts

@@ -18,10 +18,9 @@ import {
   ArrowKind,
   Node,
 } from "assemblyscript/dist/assemblyscript.js";
-
-import { BaseVisitor, SimpleParser } from "visitor-as/dist/index.js";
-import { RangeTransform } from "visitor-as/dist/transformRange.js";
-import { isStdlib } from "visitor-as/dist/utils.js";
+import { RangeTransform } from "./range.js";
+import { isStdlib, SimpleParser } from "./util.js";
+import { Visitor } from "./visitor.js";
 
 enum CoverType {
   Function,
@@ -38,7 +37,7 @@ class CoverPoint {
   public executed: boolean = false;
 }
 
-export class CoverageTransform extends BaseVisitor {
+export class CoverageTransform extends Visitor {
   public mustImport: boolean = false;
   public points: Map<string, CoverPoint> = new Map<string, CoverPoint>();
   public globalStatements: Statement[] = [];
@@ -359,14 +358,14 @@ export class CoverageTransform extends BaseVisitor {
         if (ifTrue.visited) return;
         // @ts-ignore
         ifTrue.visited = true;
-        this._visit(ifTrue);
+        this.visit(ifTrue);
       }
       if (visitIfFalse) {
         // @ts-ignore
         if (ifFalse.visited) return;
         // @ts-ignore
         ifFalse.visited = true;
-        this._visit(ifFalse!);
+        this.visit(ifFalse!);
       }
     } else {
       super.visitIfStatement(node);

package/transform/src/index.ts

@@ -7,10 +7,9 @@ import {
   Source,
   Tokenizer,
 } from "assemblyscript/dist/assemblyscript.js";
-
-import { isStdlib } from "visitor-as/dist/utils.js";
 import { CoverageTransform } from "./coverage.js";
 import { MockTransform } from "./mock.js";
+import { isStdlib } from "./util.js";
 
 export default class Transformer extends Transform {
   // Trigger the transform after parse.

package/transform/src/linker.ts

@@ -0,0 +1,41 @@
+import {
+  ClassDeclaration,
+  ImportStatement,
+  NodeKind,
+  Parser,
+  Source,
+} from "assemblyscript/dist/assemblyscript.js";
+
+export function getImports(source: Source): ImportStatement[] {
+  return source.statements.filter(
+    (v) => v.kind === NodeKind.Import,
+  ) as ImportStatement[];
+}
+
+export function getImportedClass(
+  name: string,
+  source: Source,
+  parser: Parser,
+): ClassDeclaration | null {
+  for (const stmt of getImports(source)) {
+    const externalSource = parser.sources.find(
+      (src) => src.internalPath === stmt.internalPath,
+    );
+    if (!externalSource) continue;
+
+    const classDeclaration = externalSource.statements.find(
+      (s) =>
+        s.kind === NodeKind.ClassDeclaration &&
+        (<ClassDeclaration>s).name.text === name,
+    ) as ClassDeclaration | null;
+
+    if (classDeclaration) return classDeclaration;
+  }
+  return null;
+}
+
+export function getClasses(source: Source): ClassDeclaration[] {
+  return source.statements.filter(
+    (v) => v.kind === NodeKind.ClassDeclaration,
+  ) as ClassDeclaration[];
+}

package/transform/src/mock.ts

@@ -11,11 +11,10 @@ import {
   Expression,
   FunctionDeclaration,
 } from "assemblyscript/dist/assemblyscript.js";
-
-import { BaseVisitor } from "visitor-as/dist/index.js";
-import { isStdlib, toString } from "visitor-as/dist/utils.js";
-export class MockTransform extends BaseVisitor {
-  public currentSource: Source;
+import { Visitor } from "./visitor.js";
+import { toString } from "./util.js";
+export class MockTransform extends Visitor {
+  public srcCurrent: Source | null = null;
   public globalStatements: Statement[] = [];
   public mocked = new Set<string>();
   public importFns: FunctionDeclaration[] = [];
@@ -58,7 +57,7 @@ export class MockTransform extends BaseVisitor {
       cb.range,
     );
 
-    const stmts = this.currentSource.statements;
+    const stmts = this.srcCurrent.statements;
     let index = -1;
     for (let i = 0; i < stmts.length; i++) {
       const stmt = stmts[i];
@@ -79,7 +78,7 @@ export class MockTransform extends BaseVisitor {
   }
   visitSource(node: Source): void {
     this.mocked = new Set<string>();
-    this.currentSource = node;
+    this.srcCurrent = node;
     this.importFns = [];
     super.visitSource(node);
 
@@ -96,10 +95,10 @@ export class MockTransform extends BaseVisitor {
           .join(".");
       else if (dec.args[0])
         path =
-          this.currentSource.simplePath +
+          this.srcCurrent.simplePath +
           "." +
           (dec.args[0] as StringLiteralExpression).value;
-      else path = this.currentSource.simplePath + "." + node.name.text;
+      else path = this.srcCurrent.simplePath + "." + node.name.text;
       if (!this.importMocked.has(path)) return;
 
       let args: Expression[] = [
@@ -145,7 +144,7 @@ export class MockTransform extends BaseVisitor {
         node.range,
       );
 
-      const stmts = this.currentSource.statements;
+      const stmts = this.srcCurrent.statements;
       let index = -1;
       for (let i = 0; i < stmts.length; i++) {
         const stmt = stmts[i];

package/transform/src/range.ts

@@ -0,0 +1,12 @@
+import { Node } from "assemblyscript/dist/assemblyscript.js";
+import { Visitor } from "./visitor.js";
+
+export class RangeTransform extends Visitor {
+  constructor(private node: Node) {
+    super();
+  }
+  _visit(node: Node, ref: Node | null): void {
+    node.range = this.node.range;
+    return super._visit(node, ref);
+  }
+}

package/transform/src/types.ts

@@ -0,0 +1,35 @@
+import {
+  ClassDeclaration,
+  Expression,
+  FieldDeclaration,
+} from "assemblyscript/dist/assemblyscript.js";
+
+export enum PropertyFlags {
+  OmitNull,
+  OmitIf,
+  Raw,
+}
+
+export class Property {
+  public name: string = "";
+  public alias: string | null = null;
+  public type: string = "";
+  public value: string | null = null;
+  public flags: Map<PropertyFlags, Expression | null> = new Map<
+    PropertyFlags,
+    Expression | null
+  >();
+  public node!: FieldDeclaration;
+  public byteSize: number = 0;
+}
+
+export class Schema {
+  public static: boolean = true;
+  public name: string = "";
+  public members: Property[] = [];
+  public parent: Schema | null = null;
+  public node!: ClassDeclaration;
+  public needsLink: string | null = null;
+  public byteSize: number = 0;
+  public deps: Schema[] = [];
+}

package/transform/src/util.ts

@@ -0,0 +1,81 @@
+// Taken from https://github.com/as-pect/visitor-as/blob/master/src/simpleParser.ts
+import {
+  Parser,
+  Tokenizer,
+  Source,
+  SourceKind,
+  Expression,
+  Statement,
+  NamespaceDeclaration,
+  ClassDeclaration,
+  DeclarationStatement,
+  Range,
+  Node,
+} from "assemblyscript/dist/assemblyscript.js";
+import { ASTBuilder } from "./builder.js";
+
+export class SimpleParser {
+  private static get parser(): Parser {
+    return new Parser();
+  }
+
+  private static getTokenizer(s: string, file: string = "index.ts"): Tokenizer {
+    return new Tokenizer(new Source(SourceKind.User, file, s));
+  }
+
+  static parseExpression(s: string): Expression {
+    const res = this.parser.parseExpression(this.getTokenizer(s));
+    if (res == null) {
+      throw new Error("Failed to parse the expression: '" + s + "'");
+    }
+    return res;
+  }
+
+  static parseStatement(s: string, topLevel = false): Statement {
+    const res = this.parser.parseStatement(this.getTokenizer(s), topLevel);
+    if (res == null) {
+      throw new Error("Failed to parse the statement: '" + s + "'");
+    }
+    return res;
+  }
+
+  static parseTopLevelStatement(
+    s: string,
+    namespace?: NamespaceDeclaration | null,
+  ): Statement {
+    const res = this.parser.parseTopLevelStatement(
+      this.getTokenizer(s),
+      namespace,
+    );
+    if (res == null) {
+      throw new Error("Failed to parse the top level statement: '" + s + "'");
+    }
+    return res;
+  }
+
+  static parseClassMember(
+    s: string,
+    _class: ClassDeclaration,
+  ): DeclarationStatement {
+    let res = this.parser.parseClassMember(
+      this.getTokenizer(s, _class.range.source.normalizedPath),
+      _class,
+    );
+    if (res == null) {
+      throw new Error("Failed to parse the class member: '" + s + "'");
+    }
+    return <DeclarationStatement>res;
+  }
+}
+
+let isStdlibRegex =
+  /\~lib\/(?:array|arraybuffer|atomics|builtins|crypto|console|compat|dataview|date|diagnostics|error|function|iterator|map|math|number|object|process|reference|regexp|set|staticarray|string|symbol|table|typedarray|vector|rt\/?|bindings\/|shared\/typeinfo)|util\/|uri|polyfills|memory/;
+
+export function isStdlib(s: Source | { range: Range }): boolean {
+  let source = s instanceof Source ? s : s.range.source;
+  return isStdlibRegex.test(source.internalPath);
+}
+
+export function toString(node: Node): string {
+  return ASTBuilder.build(node);
+}