squiggle/packages/squiggle-lang/scripts/run-file.mjs

#!/usr/bin/env node
import { SqProject } from "@quri/squiggle-lang";
import fs from "fs";

import { Command } from "commander";

const measure = (cb, times = 1) => {
  const t1 = new Date();

  for (let i = 1; i <= times; i++) {
    cb();
  }
  const t2 = new Date();
  return (t2 - t1) / 1000;
};

const red = (str) => `\x1b[31m${str}\x1b[0m`;
const green = (str) => `\x1b[32m${str}\x1b[0m`;

const program = new Command();

program.option("-o, --output");
program.arguments("<string>");

const options = program.parse(process.argv);

const project = SqProject.create();
const sampleCount = process.env.SAMPLE_COUNT;
if (sampleCount) {
  project.setEnvironment({
    sampleCount: Number(sampleCount),
    xyPointLength: Number(sampleCount),
  });
}

const src = fs.readFileSync(program.args[0], "utf-8");
if (!src) {
  throw new Error("Expected src");
}

project.setSource("main", src);
const time = measure(() => project.run("main"));

const bindings = project.getBindings("main");
const result = project.getResult("main");

if (options.output) {
  console.log("Result:", result.tag, result.value.toString());
  console.log("Bindings:", bindings.toString());
}

console.log(
  "Time:",
  String(time),
  result.tag === "Error" ? red(result.tag) : green(result.tag),
  result.tag === "Error" ? result.value.toString() : ""
);