Add Visitor Pass

cli/asc.js

@@ -583,12 +583,14 @@ exports.main = function main(argv, options, callback) {
       assemblyscript.enableFeature(compilerOptions, flag);
     }
   }
-
+
+  var compiler
   var module;
   stats.compileCount++;
   try {
     stats.compileTime += measure(() => {
-      module = assemblyscript.compileProgram(program, compilerOptions);
+      compiler = assemblyscript.initializeCompiler(program, compilerOptions);
+      module = compiler.compile();
     });
   } catch (e) {
     return callback(e);
@@ -598,6 +600,24 @@ exports.main = function main(argv, options, callback) {
     return callback(Error("Compile error"));
   }
 
+  // Call each visitor
+  if (args.postCompile) {
+    let visitors = []
+    try {
+      for (let filename of args.postCompile) {
+        let _filename = path.isAbsolute(filename = filename.trim())
+        ? filename
+        : path.join(process.cwd(), filename)
+        visitors.push(require(_filename).default);
+      }
+      visitors.forEach(visitor => {
+        (new visitor(parser, compiler, stderr)).start();
+      });
+    } catch (e) {
+      return callback(e);
+    }
+  }
+
   // Validate the module if requested
   if (args.validate) {
     stats.validateCount++;

cli/asc.json

@@ -242,5 +242,26 @@
   "-O0z": { "value": { "optimizeLevel": 0, "shrinkLevel": 2 } },
   "-O1z": { "value": { "optimizeLevel": 1, "shrinkLevel": 2 } },
   "-O2z": { "value": { "optimizeLevel": 2, "shrinkLevel": 2 } },
-  "-O3z": { "value": { "optimizeLevel": 3, "shrinkLevel": 2 } }
+  "-O3z": { "value": { "optimizeLevel": 3, "shrinkLevel": 2 } },
+  "path": {
+    "description": [
+      "Comma separated paths to look for dependencies.",
+      "Looks for folders with package.json that includes a 'ascMain' field",
+      "or defaults to having an '/assembly' folder."],
+    "type": "S"
+  },
+  "traceResolution": {
+    "description": "Enable tracing of package resolution.",
+    "type": "b",
+    "default": false
+  },
+  "listFiles": {
+    "description": "List files to be compiled and exit.",
+    "type": "b",
+    "default": false
+  },
+  "postCompile": {
+    "description": "A pass over the program after compiling to Binaryen IR",
+    "type": "S"
+  }
 }

cli/postCompile.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Definitions for custom compiler transforms that can be applied with the `--postCompile` option.
+ * @module cli/postTransform
+ *//***/
+
+import { Parser } from "../src/parser";
+import { Compiler } from "../src/compiler";
+
+interface stderr {
+    write: (str: string)=> void;
+}
+export interface Vistor {
+    new(parser: Parser, compiler: Compiler, writer: stderr);
+    start(): void;
+}   
+

lib/visitor/as-pect.config.js

@@ -0,0 +1,76 @@
+module.exports = {
+  /**
+   * A set of globs passed to the glob package that qualify typescript files for testing.
+   */
+  include: ["assembly/__tests__/**/*.spec.ts"],
+  /**
+   * A set of globs passed to the glob package that quality files to be added to each test.
+   */
+  add: ["assembly/__tests__/**/*.include.ts"],
+  /**
+   * All the compiler flags needed for this test suite. Make sure that a binary file is output.
+   */
+  flags: {
+    "--validate": [],
+    "--debug": [],
+    /** This is required. Do not change this. The filename is ignored, but required by the compiler. */
+    "--binaryFile": ["output.wasm"],
+    /** To enable wat file output, use the following flag. The filename is ignored, but required by the compiler. */
+    "--textFile": ["output.wat"],
+    /** To select an appropriate runtime, use the --runtime compiler flag. */
+    "--runtime": ["stub"], // Acceptable values are: full, half, stub (arena), and none,
+    "--baseDir": process.cwd(),
+    "--postCompile": "dist/instances/virtual.js",
+  },
+  /**
+   * A set of regexp that will disclude source files from testing.
+   */
+  disclude: [/node_modules/],
+  /**
+   * Add your required AssemblyScript imports here.
+   */
+  imports: {},
+  /**
+   * All performance statistics reporting can be configured here.
+   */
+  performance: {
+    /** Enable performance statistics gathering for every test. */
+    enabled: false,
+    /** Set the maximum number of samples to run for every test. */
+    maxSamples: 10000,
+    /** Set the maximum test run time in milliseconds for every test. */
+    maxTestRunTime: 5000,
+    /** Report the median time in the default reporter for every test. */
+    reportMedian: true,
+    /** Report the average time in milliseconds for every test. */
+    reportAverage: true,
+    /** Report the standard deviation for every test. */
+    reportStandardDeviation: false,
+    /** Report the maximum run time in milliseconds for every test. */
+    reportMax: false,
+    /** Report the minimum run time in milliseconds for every test. */
+    reportMin: false,
+  },
+  /**
+   * Add a custom reporter here if you want one. The following example is in typescript.
+   *
+   * @example
+   * import { TestReporter, TestGroup, TestResult, TestContext } from "as-pect";
+   *
+   * export class CustomReporter extends TestReporter {
+   *   // implement each abstract method here
+   *   public abstract onStart(suite: TestContext): void;
+   *   public abstract onGroupStart(group: TestGroup): void;
+   *   public abstract onGroupFinish(group: TestGroup): void;
+   *   public abstract onTestStart(group: TestGroup, result: TestResult): void;
+   *   public abstract onTestFinish(group: TestGroup, result: TestResult): void;
+   *   public abstract onFinish(suite: TestContext): void;
+   * }
+   */
+  // reporter: new CustomReporter(),
+  /**
+   * Specify if the binary wasm file should be written to the file system.
+   */
+  outputBinary: false,
+  compiler: "../.."
+};