compile.ts

#!/usr/bin/env node
import chalk from "chalk";
import { execSync } from "child_process";
import chokidar from "chokidar";
import { Command } from "commander";
import { build } from "esbuild";
import fs from "fs";
import os from "os";
import path from "path";
import glob from "fast-glob";
import http from "http";
import finalhandler from "finalhandler";
import serveStatic from "serve-static";

const SOURCE_FOLDER_NAME = "plugins";
const DESTINATION_FOLDER_NAME = "dist";
const SOURCE_FILE_GLOB = `${SOURCE_FOLDER_NAME}/**/*.(ts|md)`;
const UTIL_FILE_GLOB = `util/index.ts`;
const PORT = 3691;

enum PLUGINS_DEV_STATE {
  USE_LOCAL_PLUGINS = "useLocalPlugins",
  PLUGINS_LAST_UPDATED = "localPluginsLastUpdated",
  PLUGIN_DIRECTORY = "pluginDirectory",
  PLUGIN_LOCAL_PORT = "localPluginPort",
}

const setDeveloperState = (key: string, value: string | null) => {
  try {
    return execSync(
      `fig _ local-state plugins.developer.${key} ${value === null ? "--delete" : value}`
    );
  } catch (_) {
    return null;
  }
};

const buildFiles = (files: string[], format: "cjs" | "esm") =>
  build({
    entryPoints: files,
    entryNames: "[dir]/[name]",
    outbase: ".",
    outdir: `${DESTINATION_FOLDER_NAME}/${format}`,
    bundle: true,
    format,
    minify: true,
    sourcemap: "inline"
  })

const compile = async (files: string[], compileMarkdown = false) => {
  const jsFiles = files.filter((file) => file.endsWith("ts"));
  const dirs = files.map((file) => file.slice(0, file.lastIndexOf("/")));

  if (jsFiles.length > 0) {
    const fileName = jsFiles.length === 1 ? jsFiles[0] : `${jsFiles.length} plugins`;
    await Promise.all([
      buildFiles(jsFiles, "esm"),
      buildFiles(jsFiles, "cjs")
    ]).catch((e: Error) => console.log(`Error building ${fileName}: ${e.message}`));
    console.log(`Built ${fileName}`);
  }

  if (compileMarkdown) {
    const mdFiles = files.filter((file) => file.endsWith("md"));
    if (mdFiles.length > 0) {
      mdFiles.forEach((mdFile) => {
        fs.copyFileSync(mdFile, `${DESTINATION_FOLDER_NAME}/esm/${mdFile}`);
      });
      console.log(`Copied ${mdFiles.length} md files`);
    }
  }

  setDeveloperState(PLUGINS_DEV_STATE.PLUGINS_LAST_UPDATED, `${Date.now()}`);
};

const warn = (warning: string, ...help: string[]) => {
  const text = help.map((x) => ` ${x}`).join("\n");
  console.log(
    `\n******\n\n${chalk.bold(chalk.yellow(` WARNING: ${warning}`))}\n\n${text}\n\n******\n`
  );
};

async function runProgram(options: { watch?: boolean }) {
  console.clear();
  const files = await glob(SOURCE_FILE_GLOB);
  await compile(files, Boolean(options.watch));

  if (!options.watch) {
    const utils = await glob(UTIL_FILE_GLOB);
    await compile(utils);
  }

  if (options.watch) {
    const isMacOS = os.type() === "Darwin";
    if (isMacOS) {
      const globalFigAppPath = path.join("/", "Applications", "Fig.app");
      const localFigAppPath = path.join(os.homedir(), "Applications", "Fig.app");

      if (!fs.existsSync(globalFigAppPath) && !fs.existsSync(localFigAppPath)) {
        warn("Fig is not installed", chalk.bold(chalk.cyan("Download Fig at:")), "https://fig.io/");
      } else {
        try {
          execSync("fig --version");
        } catch (e) {
          warn(
            "Fig cli is not installed",
            chalk.bold(
              chalk.cyan(
                "1. Run the install and update script ( ◧ > Integrations > Developer > Run install and update script)"
              )
            ),
            chalk.bold(chalk.cyan("2. Create a new terminal session"))
          );
        }
      }
    } else {
      warn(
        "Looks like you're not on macOS. We're working on linux / windows builds!",
        chalk.bold(
          chalk.yellow(
            "You can still build and contribute to completion specs, but you won't be able to test them unless you are on a mac or a macosVM"
          )
        )
      );
    }

    console.log(
      `Welcome to ${chalk.magenta("Dev Mode")}\n\n`,
      `1. Edit your plugin(s) in the ${chalk.bold("plugins/")} directory\n`,
      `2. On save, they will compile to the local ${chalk.bold("dist/")} directory\n`,
      `3. Test your changes ${chalk.bold("instantly")} in fig\n`,
      `4. When done, hit ${chalk.bold("ctrl + c")} to exit\n`
    );

    ["SIGINT", "SIGTERM", "SIGQUIT"].forEach((signal) =>
      process.on(signal, () => {
        console.log("\nPlugins dev mode disabled\n");
        setDeveloperState(PLUGINS_DEV_STATE.USE_LOCAL_PLUGINS, null);
        process.exit();
      })
    );
    
    setDeveloperState(PLUGINS_DEV_STATE.USE_LOCAL_PLUGINS, "true");
    setDeveloperState(PLUGINS_DEV_STATE.PLUGINS_LAST_UPDATED, `${Date.now()}`);
    setDeveloperState(
      PLUGINS_DEV_STATE.PLUGIN_DIRECTORY,
      `${path.join(process.cwd(), DESTINATION_FOLDER_NAME, "esm", "plugins")}`
    );
    setDeveloperState(PLUGINS_DEV_STATE.PLUGIN_LOCAL_PORT, `${PORT}`);

    const watcher = chokidar.watch(SOURCE_FILE_GLOB, { ignoreInitial: true });
    watcher.on("change", (file: string) => compile([file], true));
    watcher.on("add", (file: string) => compile([file], true));

    const serve = serveStatic("./plugins/");

    const server = http.createServer((req, res) => {
      serve(req, res, () => {
        finalhandler(req, res)("");
      });
    });

    server.listen(PORT);
  }
}

new Command()
  .name("plugin-compiler")
  .option("-w, --watch", "Watch files and re-compile on change")
  .action(runProgram)
  .parse(process.argv);