/**
 * @file Compiler frontend for node.js
 *
 * Uses the low-level API exported from src/index.ts so it works with the compiler compiled to
 * JavaScript as well as the compiler compiled to WebAssembly (eventually). Runs the sources
 * directly through ts-node if distribution files are not present (indicated by a `-dev` version).
 *
 * Can also be packaged as a bundle suitable for in-browser use with the standard library injected
 * in the build step. See dist/asc.js for the bundle and webpack.config.js for building details.
 */

const fs = require("fs");
const path = require("path");
const os = require("os");

// Use distribution files if present, otherwise run the sources directly
var assemblyscript, isDev;
(() => {
  try {
    assemblyscript = require("../dist/assemblyscript.js");
    isDev = false;
    try { require("source-map-support").install(); } catch (e) {/* optional */}
  } catch (e) {
    try {
      require("ts-node").register({ project: path.join(__dirname, "..", "src", "tsconfig.json") });
      require("../src/glue/js");
      assemblyscript = require("../src");
      isDev = true;
    } catch (e) {
      // last resort: same directory CommonJS
      assemblyscript = eval("require('./assemblyscript')");
      isDev = false;
    }
  }
})();

/** Whether this is a webpack bundle or not. */
exports.isBundle = typeof BUNDLE_VERSION === "string";

/** Whether asc runs the sources directly or not. */
exports.isDev = isDev;

/** AssemblyScript veresion. */
exports.version = exports.isBundle ? BUNDLE_VERSION : require("../package.json").version;

/** Available options. */
exports.options = require("./asc.json");

/** Common root used in source maps. */
exports.sourceMapRoot = "assemblyscript:///";

/** Prefix used for library files. */
exports.libraryPrefix = assemblyscript.LIBRARY_PREFIX;

/** Default Binaryen optimization level. */
exports.defaultOptimizeLevel = 2;

/** Default Binaryen shrink level. */
exports.defaultShrinkLevel = 1;

/** Bundled library files, if any. */
exports.libraryFiles = exports.isBundle ? BUNDLE_LIBRARY : {};

/** Bundled definition files, if any. */
exports.definitionFiles = exports.isBundle ? BUNDLE_DEFINITIONS : {};

/** Runs the command line utility using the specified arguments array. */
exports.main = function main(argv, options, callback) {
  if (typeof options === "function") {
    callback = options;
    options = {};
  } else if (!options) {
    options = {};
  }

  const stdout = options.stdout || process.stdout;
  const stderr = options.stderr || process.stderr;
  const readFile = options.readFile || readFileNode;
  const writeFile = options.writeFile || writeFileNode;
  const listFiles = options.listFiles || listFilesNode;
  const stats = options.stats || createStats();

  // All of the above must be specified in browser environments
  if (!stdout) throw Error("'options.stdout' must be specified");
  if (!stderr) throw Error("'options.stderr' must be specified");
  if (!fs.readFileSync) {
    if (readFile === readFileNode) throw Error("'options.readFile' must be specified");
    if (writeFile === writeFileNode) throw Error("'options.writeFile' must be specified");
    if (listFiles === listFilesNode) throw Error("'options.listFiles' must be specified");
  }

  const args = parseArguments(argv);
  const indent = 24;

  // Use default callback if none is provided
  if (!callback) callback = function defaultCallback(err) {
    var code = 0;
    if (err) {
      stderr.write(err.stack + os.EOL);
      code = 1;
    }
    return code;
  };

  // Just print the version if requested
  if (args.version) {
    stdout.write("Version " + exports.version + (isDev ? "-dev" : "") + os.EOL);
    return callback(null);
  }
  // Print the help message if requested or no source files are provided
  if (args.help || args._.length < 1) {
    const opts = [];
    Object.keys(exports.options).forEach(name => {
      var option = exports.options[name];
      var text = " ";
      text += "--" + name;
      if (option.aliases && option.aliases[0].length === 1) {
        text += ", -" + option.aliases[0];
      }
      while (text.length < indent) {
        text += " ";
      }
      if (Array.isArray(option.desc)) {
        opts.push(text + option.desc[0] + option.desc.slice(1).map(line => {
          for (let i = 0; i < indent; ++i) {
            line = " " + line;
          }
          return os.EOL + line;
        }).join(""));
      } else {
        opts.push(text + option.desc);
      }
    });

    (args.help ? stdout : stderr).write([
      "Version " + exports.version + (isDev ? "-dev" : ""),
      "Syntax:   asc [entryFile ...] [options]",
      "",
      "Examples: asc hello.ts",
      "          asc hello.ts -b hello.wasm -t hello.wat",
      "          asc hello1.ts hello2.ts -b -O > hello.wasm",
      "",
      "Options:"
    ].concat(opts).join(os.EOL) + os.EOL);
    return callback(null);
  }

  // Set up base directory
  const baseDir = args.baseDir ? path.resolve(args.baseDir) : process.cwd();

  // Include standard library if --noLib isn't set
  const stdLibDir = path.join(__dirname, "..", "std", "assembly");
  const libDirs = args.noLib ? [] : [ stdLibDir ];

  // Include custom library components (with or without stdlib)
  if (args.lib) {
    if (typeof args.lib === "string") {
      args.lib = args.lib.split(",");
    }
    Array.prototype.push.apply(libDirs, args.lib.map(lib => lib.trim()));
  }

  // Begin parsing
  var parser = null;

  // Include entry files
  for (let i = 0, k = args._.length; i < k; ++i) {
    const filename = args._[i];

    let sourcePath = filename.replace(/\\/g, "/").replace(/(\.ts|\/)$/, "");

    // Try entryPath.ts, then entryPath/index.ts
    let sourceText = readFile(path.join(baseDir, sourcePath) + ".ts");
    if (sourceText === null) {
      sourceText = readFile(path.join(baseDir, sourcePath, "index.ts"));
      if (sourceText === null) {
        return callback(Error("Entry file '" + sourcePath + ".ts' not found."));
      } else {
        sourcePath += "/index.ts";
      }
    } else {
      sourcePath += ".ts";
    }

    stats.parseCount++;
    stats.parseTime += measure(() => {
      parser = assemblyscript.parseFile(sourceText, sourcePath, true, parser);
    });

    while ((sourcePath = parser.nextFile()) != null) {
      let found = false;

      // Load library file if explicitly requested
      if (sourcePath.startsWith(exports.libraryPrefix)) {
        for (let i = 0, k = libDirs.length; i < k; ++i) {
          if (exports.libraryFiles.hasOwnProperty(sourcePath)) {
            sourceText = exports.libraryFiles[sourcePath];
          } else {
            sourceText = readFile(path.join(
              libDirs[i],
              sourcePath.substring(exports.libraryPrefix.length) + ".ts")
            );
            if (sourceText !== null) {
              sourcePath += ".ts";
              break;
            }
          }
        }

      // Otherwise try nextFile.ts, nextFile/index.ts, (lib)/nextFile.ts
      } else {
        sourceText = readFile(path.join(baseDir, sourcePath + ".ts"));
        if (sourceText === null) {
          sourceText = readFile(path.join(baseDir, sourcePath, "index.ts"));
          if (sourceText === null) {
            for (let i = 0, k = libDirs.length; i < k; ++i) {
              const dir = libDirs[i];
              const key = exports.libraryPrefix + sourcePath;
              if (exports.libraryFiles.hasOwnProperty(key)) {
                sourceText = exports.libraryFiles[key];
              } else {
                sourceText = readFile(path.join(dir, sourcePath + ".ts"));
                if (sourceText !== null) {
                  sourcePath = exports.libraryPrefix + sourcePath + ".ts";
                  break;
                }
              }
            }
            if (sourceText === null) {
              return callback(
                Error("Import file '" + sourcePath + ".ts' not found.")
              );
            }
          } else {
            sourcePath += "/index.ts";
          }
        } else {
          sourcePath += ".ts";
        }
      }
      stats.parseCount++;
      stats.parseTime += measure(() => {
        assemblyscript.parseFile(sourceText, sourcePath, false, parser);
      });
    }
    if (checkDiagnostics(parser, stderr)) {
      return callback(Error("Parse error"));
    }
  }

  // Include (other) library components
  var hasBundledLibrary = false;
  if (!args.noLib)
    Object.keys(exports.libraryFiles).forEach(libPath => {
      if (libPath.lastIndexOf("/") >= exports.libraryPrefix.length) return;
      stats.parseCount++;
      stats.parseTime += measure(() => {
        parser = assemblyscript.parseFile(
          exports.libraryFiles[libPath],
          libPath + ".ts",
          false,
          parser
        );
      });
      hasBundledLibrary = true;
    });
  for (let i = 0, k = libDirs.length; i < k; ++i) {
    if (i === 0 && hasBundledLibrary) continue;
    let libDir = libDirs[i];
    let libFiles;
    if (libDir.endsWith(".ts")) {
      libFiles = [path.basename(libDir)];
      libDir = path.dirname(libDir);
    } else {
      libFiles = listFiles(libDir);
    }
    for (let j = 0, l = libFiles.length; j < l; ++j) {
      let libPath = libFiles[j];
      let libText = readFile(path.join(libDir, libPath));
      if (libText === null) {
        return callback(Error("Library file '" + libPath + "' not found."));
      }
      stats.parseCount++;
      stats.parseTime += measure(() => {
        parser = assemblyscript.parseFile(
          libText,
          exports.libraryPrefix + libPath,
          false,
          parser
        );
      });
    }
  }

  // Finish parsing
  const program = assemblyscript.finishParsing(parser);

  // Begin compilation
  const compilerOptions = assemblyscript.createOptions();
  assemblyscript.setTarget(compilerOptions, 0);
  assemblyscript.setNoTreeShaking(compilerOptions, !!args.noTreeShaking);
  assemblyscript.setNoAssert(compilerOptions, !!args.noAssert);
  assemblyscript.setNoMemory(compilerOptions, !!args.noMemory);
  assemblyscript.setImportMemory(compilerOptions, !!args.importMemory);
  assemblyscript.setMemoryBase(compilerOptions, args.memoryBase >>> 0);
  assemblyscript.setSourceMap(compilerOptions, args.sourceMap != null);

  var module;
  stats.compileCount++;
  (() => {
    try {
      stats.compileTime += measure(() => {
        module = assemblyscript.compileProgram(program, compilerOptions);
      });
    } catch (e) {
      return callback(e);
    }
  })();
  if (checkDiagnostics(parser, stderr)) {
    if (module) module.dispose();
    return callback(Error("Compile error"));
  }

  // Validate the module if requested
  if (args.validate) {
    stats.validateCount++;
    stats.validateTime += measure(() => {
      if (!module.validate()) {
        module.dispose();
        return callback(Error("Validate error"));
      }
    });
  }

  // Set Binaryen-specific options
  if (args.trapMode === "clamp") {
    stats.optimizeCount++;
    stats.optimizeTime += measure(() => {
      module.runPasses([ "trap-mode-clamp" ]);
    });
  } else if (args.trapMode === "js") {
    stats.optimizeCount++;
    stats.optimizeTime += measure(() => {
      module.runPasses([ "trap-mode-js" ]);
    });
  } else if (args.trapMode !== "allow") {
    module.dispose();
    return callback(Error("Unsupported trap mode"));
  }

  var optimizeLevel = -1;
  var shrinkLevel = 0;
  var debugInfo = !args.noDebug;

  if (args.optimize !== false) {
    if (typeof args.optimize === "number") {
      optimizeLevel = args.optimize;
    } else if (args["0"]) {
      optimizeLevel = 0;
    } else if (args["1"]) {
      optimizeLevel = 1;
    } else if (args["2"]) {
      optimizeLevel = 2;
    } else if (args["3"]) {
      optimizeLevel = 3;
    } else if (args.optimize === true) {
      optimizeLevel = exports.defaultOptimizeLevel;
      shrinkLevel = exports.defaultShrinkLevel;
    } else
      optimizeLevel = 0;
  }

  if (args["s"]) {
    shrinkLevel = 1;
  } else if (args["z"]) {
    shrinkLevel = 2;
  }

  if (typeof args.optimizeLevel === "number") {
    optimizeLevel = args.optimizeLevel;
  }

  if (typeof args.shrinkLevel === "number") {
    shrinkLevel = args.shrinkLevel;
  } else if (args.shrinkLevel === "s") {
    shrinkLevel = 1;
  } else if (args.shrinkLevel === "z") {
    shrinkLevel = 2;
  }

  module.setOptimizeLevel(optimizeLevel > 0 ? optimizeLevel : 0);
  module.setShrinkLevel(shrinkLevel);
  module.setDebugInfo(debugInfo);

  var runPasses = [];
  if (args.runPasses) {
    if (typeof args.runPasses === "string") {
      args.runPasses = args.runPasses.split(",");
    }
    if (args.runPasses.length) {
      args.runPasses.forEach(pass => {
        if (runPasses.indexOf(pass) < 0)
          runPasses.push(pass);
      });
    }
  }

  // Optimize the module if requested
  if (optimizeLevel >= 0) {
    stats.optimizeCount++;
    stats.optimizeTime += measure(() => {
      module.optimize();
    });
  }

  // Run additional passes if requested
  if (runPasses.length) {
    stats.optimizeCount++;
    stats.optimizeTime += measure(() => {
      module.runPasses(runPasses.map(pass => pass.trim()));
    });
  }

  // Prepare output
  if (!args.noEmit) {
    let hasStdout = false;

    if (args.outFile != null) {
      if (/\.was?t$/.test(args.outFile) && args.textFile == null) {
        args.textFile = args.outFile;
      } else if (/\.js$/.test(args.outFile) && args.asmjsFile == null) {
        args.asmjsFile = args.outFile;
      } else if (args.binaryFile == null) {
        args.binaryFile = args.outFile;
      }
    }

    // Write binary
    if (args.binaryFile != null) {
      let sourceMapURL = args.sourceMap != null
        ? args.sourceMap.length
          ? args.sourceMap
          : path.basename(args.binaryFile) + ".map"
        : null;

      let wasm;
      stats.emitCount++;
      stats.emitTime += measure(() => {
        wasm = module.toBinary(sourceMapURL)
      });

      if (args.binaryFile.length) {
        writeFile(path.join(baseDir, args.binaryFile), wasm.output);
      } else {
        writeStdout(wasm.output);
        hasStdout = true;
      }

      // Post-process source map
      if (wasm.sourceMap != null) {
        if (args.binaryFile.length) {
          let sourceMap = JSON.parse(wasm.sourceMap);
          sourceMap.sourceRoot = exports.sourceMapRoot;
          sourceMap.sources.forEach((name, index) => {
            let text = null;
            if (name.startsWith(exports.libraryPrefix)) {
              for (let i = 0, k = libDirs.length; i < k; ++i) {
                text = readFile(path.join(
                  libDirs[i],
                  name.substring(exports.libraryPrefix.length))
                );
                if (text !== null) break;
              }
            } else {
              text = readFile(path.join(baseDir, name));
            }
            if (text === null) {
              return callback(Error("Source file '" + name + "' not found."));
            }
            if (!sourceMap.sourceContents) sourceMap.sourceContents = [];
            sourceMap.sourceContents[index] = text;
          });
          writeFile(path.join(
            baseDir,
            path.dirname(args.binaryFile),
            path.basename(sourceMapURL)
          ), JSON.stringify(sourceMap));
        } else {
          stderr.write("Skipped source map (stdout already occupied)" + os.EOL);
        }
      }
    }

    // Write asm.js
    if (args.asmjsFile != null) {
      let asm;
      if (args.asmjsFile.length) {
        stats.emitCount++;
        stats.emitTime += measure(() => {
          asm = module.toAsmjs();
        });
        writeFile(path.join(baseDir, args.asmjsFile), asm);
      } else if (!hasStdout) {
        stats.emitCount++;
        stats.emitTime += measure(() => {
          asm = module.toAsmjs();
        });
        writeStdout(asm);
        hasStdout = true;
      }
    }

    // Write WebIDL
    if (args.idlFile != null) {
      let idl;
      if (args.idlFile.length) {
        stats.emitCount++;
        stats.emitTime += measure(() => {
          idl = assemblyscript.buildIDL(program);
        });
        writeFile(path.join(baseDir, args.idlFile), idl);
      } else if (!hasStdout) {
        stats.emitCount++;
        stats.emitTime += measure(() => {
          idl = assemblyscript.buildIDL(program);
        });
        writeStdout(idl);
        hasStdout = true;
      }
    }

    // Write TypeScript definition
    if (args.tsdFile != null) {
      let tsd;
      if (args.tsdFile.length) {
        stats.emitCount++;
        stats.emitTime += measure(() => {
          tsd = assemblyscript.buildTSD(program);
        });
        writeFile(path.join(baseDir, args.tsdFile), tsd);
      } else if (!hasStdout) {
        stats.emitCount++;
        stats.emitTime += measure(() => {
          tsd = assemblyscript.buildTSD(program);
        });
        writeStdout(tsd);
        hasStdout = true;
      }
    }

    // Write text (must be last)
    if (args.textFile != null) {
      let wat;
      if (args.textFile && args.textFile.length) {
        stats.emitCount++;
        stats.emitTime += measure(() => {
          wat = module.toText();
        });
        writeFile(path.join(baseDir, args.textFile), wat);
      } else if (!hasStdout) {
        stats.emitCount++;
        stats.emitTime += measure(() => {
          wat = module.toText()
        });
        writeStdout(wat);
      }
    }
  }

  module.dispose();
  if (args.measure) {
    printStats(stats, stderr);
  }
  return callback(null);

  function readFileNode(filename) {
    try {
      let text;
      stats.readCount++;
      stats.readTime += measure(() => {
        text = fs.readFileSync(filename, { encoding: "utf8" });
      });
      return text;
    } catch (e) {
      return null;
    }
  }

  function writeFileNode(filename, contents) {
    try {
      stats.writeCount++;
      stats.writeTime += measure(() => {
        if (typeof contents === "string") {
          fs.writeFileSync(filename, contents, { encoding: "utf8" } );
        } else {
          fs.writeFileSync(filename, contents);
        }
      });
      return true;
    } catch (e) {
      return false;
    }
  }

  function listFilesNode(dirname) {
    var files;
    try {
      stats.readTime += measure(() => {
        files = require("glob").sync("*.ts", { cwd: dirname });
      });
      return files;
    } catch (e) {
      return [];
    }
  }

  function writeStdout(contents) {
    if (!writeStdout.used) {
      stats.writeCount++;
      writeStdout.used = true;
    }
    stats.writeTime += measure(() => {
      if (typeof contents === "string") {
        stdout.write(contents, { encoding: "utf8" });
      } else {
        stdout.write(contents);
      }
    });
  }
}

/** Parses the specified command line arguments. */
function parseArguments(argv) {
  const opts = {};
  Object.keys(exports.options).forEach(key => {
    const opt = exports.options[key];
    if (opt.aliases) {
      (opts.alias || (opts.alias = {}))[key] = opt.aliases;
    }
    if (opt.default !== undefined) {
      (opts.default || (opts.default = {}))[key] = opt.default;
    }
    if (opt.type === "string") {
      (opts.string || (opts.string = [])).push(key);
    } else if (opt.type === "boolean") {
      (opts.boolean || (opts.boolean = [])).push(key);
    }
  });
  return require("minimist")(argv, opts);
}

exports.parseArguments = parseArguments;

/** Checks diagnostics emitted so far for errors. */
function checkDiagnostics(emitter, stderr) {
  var diagnostic;
  var hasErrors = false;
  while ((diagnostic = assemblyscript.nextDiagnostic(emitter)) != null) {
    stderr.write(
      assemblyscript.formatDiagnostic(diagnostic, stderr.isTTY, true) +
      os.EOL + os.EOL
    );
    if (assemblyscript.isError(diagnostic)) hasErrors = true;
  }
  return hasErrors;
}

exports.checkDiagnostics = checkDiagnostics;

/** Creates an empty set of stats. */
function createStats() {
  return {
    readTime: 0,
    readCount: 0,
    writeTime: 0,
    writeCount: 0,
    parseTime: 0,
    parseCount: 0,
    compileTime: 0,
    compileCount: 0,
    emitTime: 0,
    emitCount: 0,
    validateTime: 0,
    validateCount: 0,
    optimizeTime: 0,
    optimizeCount: 0
  };
}

exports.createStats = createStats;

if (!process.hrtime) process.hrtime = require("browser-process-hrtime");

/** Measures the execution time of the specified function.  */
function measure(fn) {
  const start = process.hrtime();
  fn();
  const times = process.hrtime(start);
  return times[0] * 1e9 + times[1];
}

exports.measure = measure;

/** Formats and prints out the contents of a set of stats. */
function printStats(stats, output) {
  function format(time, count) {
    return time ? (time / 1e6).toFixed(3) + " ms" : "N/A";
  }
  (output || process.stdout).write([
    "I/O Read  : " + format(stats.readTime, stats.readCount),
    "I/O Write : " + format(stats.writeTime, stats.writeCount),
    "Parse     : " + format(stats.parseTime, stats.parseCount),
    "Compile   : " + format(stats.compileTime, stats.compileCount),
    "Emit      : " + format(stats.emitTime, stats.emitCount),
    "Validate  : " + format(stats.validateTime, stats.validateCount),
    "Optimize  : " + format(stats.optimizeTime, stats.optimizeCount)
  ].join(os.EOL) + os.EOL);
}

exports.printStats = printStats;

/** Creates a memory stream that can be used in place of stdout/stderr. */
function createMemoryStream(fn) {
  var stream = [];
  stream.write = function(chunk) {
    if (typeof chunk === "string") {
      this.push(Buffer.from(chunk, "utf8"));
    } else {
      this.push(chunk);
    }
    if (fn) fn(chunk);
  };
  stream.toBuffer = function() {
    return Buffer.concat(this);
  };
  stream.toString = function() {
    return this.toBuffer().toString("utf8");
  };
  return stream;
}

exports.createMemoryStream = createMemoryStream;

/** Compatible TypeScript compiler options. */
exports.tscOptions = {
  alwaysStrict: true,
  noImplicitAny: true,
  noImplicitReturns: true,
  noImplicitThis: true,
  noEmitOnError: true,
  strictNullChecks: true,
  experimentalDecorators: true,
  target: "esnext",
  module: "commonjs",
  noLib: true,
  types: [],
  allowJs: false
};