123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197
#!/usr/bin/env -S deno run --allow-run=deno --allow-read --allow-write --allow-net=deno.land --no-check
// CLI utility to build/list/extract/run ESZIPs
import { build, Parser } from "./mod.ts";import { dirname, join } from "https://deno.land/std@0.127.0/path/mod.ts";import { assertStrictEquals } from "https://deno.land/std@0.127.0/testing/asserts.ts";
function hasV2Header(bytes: Uint8Array) {  const magicV2 = new TextDecoder().decode(bytes.slice(0, 8));  return magicV2 === "ESZIP_V2";}
interface ESZIP {  extract(dest: string): Promise<void>;  list(): string[];}
interface V1Entry {  Source: {    source: string;    transpiled: string;  };}
class V1 {  inner: Record<string, V1Entry>;
  constructor(bytes: Uint8Array) {    const json = new TextDecoder().decode(bytes);    const eszip = JSON.parse(json);    assertStrictEquals(eszip.version, 1);    this.inner = eszip;  }
  static load(bytes: Uint8Array) {    return Promise.resolve(new V1(bytes));  }
  *entries() {    for (      const [        url,        {          Source: { source, transpiled },        },      ] of Object.entries(this.inner.modules)    ) {      yield { url, source, transpiled };    }  }
  async extract(dest: string) {    for (const { url, source, transpiled } of this.entries()) {      await write(join(dest, "source", url2path(url)), source);      await write(        join(dest, "transpiled", url2path(url)),        transpiled ?? source,      );    }  }
  list() {    return Array.from(this.entries()).map((e) => e.url);  }}
class V2 {  parser: Parser;  specifiers: string[];
  constructor(parser: Parser, specifiers: string[]) {    this.parser = parser;    this.specifiers = specifiers;  }
  static async load(bytes: Uint8Array) {    const parser = await Parser.createInstance();    const specifiers = await parser.parseBytes(bytes);    await parser.load();    return new V2(parser, specifiers as string[]);  }
  async extract(dest: string) {    const imports: Record<string, string> = {};
    for (const specifier of this.specifiers) {      const module = await this.parser.getModuleSource(specifier);      await write(join(dest, "source", url2path(specifier)), module);      // Track import      imports[specifier] = `./${url2path(specifier)}`;    }    // Write import map    const importMap = JSON.stringify({ imports }, null, 2);    await Deno.writeTextFile(      join(dest, "source", "import_map.json"),      importMap,    );  }
  list() {    return this.specifiers;  }}
async function loadESZIP(filename: string): Promise<ESZIP> {  const bytes = await Deno.readFile(filename);  if (hasV2Header(bytes)) {    console.log("Loading eszip v2");    return await V2.load(bytes);  }  console.log("Loading eszip v1");  return await V1.load(bytes);}
function url2path(url: string) {  return join(...(new URL(url).pathname.split("/").filter(Boolean)));}
async function write(path: string, content: string) {  await Deno.mkdir(dirname(path), { recursive: true });  await Deno.writeTextFile(path, content);}
async function run(eszip: ESZIP, specifier: string) {  // Extract to tmp directory  const tmpDir = await Deno.makeTempDir({ prefix: "esz" });  try {    // Extract    await eszip.extract(tmpDir);    const importMap = join(tmpDir, "source", "import_map.json");    // Run    const p = Deno.run({      cmd: [        "deno",        "run",        "-A",        "--no-check",        "--import-map",        importMap,        specifier,      ],    });    await p.status();  } finally {    // Cleanup    await Deno.remove(tmpDir, { recursive: true });  }}
// Mainasync function main() {  const args = Deno.args;  const [subcmd, filename, ...rest] = args;
  if (subcmd === "help") {    return console.log("TODO");  }
  switch (subcmd) {    case "build":    case "b": {      const eszip = await build([filename]);      let out = rest[0];      if (!out) {        // Create outfile name from url filename        out = new URL(filename).pathname.split("/").pop() || "out";      }      console.log(`${out}.eszip: ${eszip.length} bytes`);      await Deno.writeFile(`${out}.eszip`, eszip);      return;    }    case "x":    case "extract": {      const eszip = await loadESZIP(filename);      return await eszip.extract(rest[0] ?? Deno.cwd());    }    case "l":    case "ls":    case "list": {      const eszip = await loadESZIP(filename);      return console.log(eszip.list().join("\n"));    }    case "r":    case "run": {      const eszip = await loadESZIP(filename);      const specifier = rest[0];      if (!specifier) {        return console.error("Please provide a specifier to run");      }      return await run(eszip, specifier);    }  }}
await main();