blitzkit/packages/scripts/src/protos.ts

import { execFile } from "child_process";
import { existsSync } from "fs";
import { mkdir, readdir, rm, writeFile } from "fs/promises";

const TS_PROTO_EXECUTABLE_LOCATIONS = [
  "./node_modules/.bin/protoc-gen-ts_proto",
  "../../node_modules/.bin/protoc-gen-ts_proto",
];
const ROOT = "../closed/protos";

let tsProtoExecutableLocation: string | undefined = undefined;

for (const location of TS_PROTO_EXECUTABLE_LOCATIONS) {
  if (existsSync(location)) {
    tsProtoExecutableLocation = location;
    break;
  } else if (existsSync(`${location}.exe`)) {
    tsProtoExecutableLocation = `${location}.exe`;
    break;
  }
}

if (!tsProtoExecutableLocation) {
  throw new Error(
    `Could not find ts-proto executable. Checked locations: ${TS_PROTO_EXECUTABLE_LOCATIONS.join(
      ", ",
    )}`,
  );
}

await mkdir("../../temp", { recursive: true });

let args = "";

args += `--plugin=${tsProtoExecutableLocation}\n`;
args += "--ts_proto_opt=esModuleInterop=true\n";
args += "--ts_proto_opt=oneof=unions-value\n";
args += "--ts_proto_opt=unrecognizedEnum=false\n";
args += "--ts_proto_opt=snakeToCamel=false\n";
args += `--ts_proto_out=${ROOT}\n`;

args += `-I=${ROOT}\n`;
// for (const dir of await readdir(ROOT)) {
//   args += `-I=${ROOT}/${dir}\n`;
// }

const filesRaw = await readdir(`${ROOT}`, { recursive: true });
const files: string[] = [];

for (const file of filesRaw) {
  if (!file.startsWith("blitz/")) continue;

  if (file.endsWith(".proto")) {
    files.push(file);
    args += `${ROOT}/${file}\n`;
  }

  if (file.endsWith(".ts")) await rm(`${ROOT}/${file}`);
}

await writeFile("../../temp/protoc.txt", args);

execFile("protoc", ["@../../temp/protoc.txt"], (error, stdout, stderr) => {
  if (error) throw new Error(error.message);
  if (stderr) console.error(stderr);
  if (stdout) console.log(stdout);
});