build.mjs

import esbuild from "esbuild";
import archiver from "archiver";
import fs from "fs-extra";
import tailwindcss from "tailwindcss";
import autoprefixer from "autoprefixer";
import postcssPlugin from "esbuild-style-plugin";
import copyStaticFilesPlugin from "esbuild-copy-files-plugin";
import path from 'path';


const buildDir = "build";
const minify = process.argv.includes("--minify");

async function cleanBuildDir() {
  const entries = await fs.readdir(buildDir);
  for (const entry of entries) {
    if (path.extname(entry) === ".zip") continue;
    await fs.remove(`${buildDir}/${entry}`);
  }
}

async function runEsbuild() {
  await esbuild.build({
    entryPoints: [
      "src/content-scripts/mainUI.tsx",
      "src/background/bg.ts",
      "src/options/options.tsx",
    ],
    outdir: buildDir,
    bundle: true,
    minify: minify,
    treeShaking: true,
    define: {
      "process.env.NODE_ENV": '"production"',
    },
    jsxFactory: "h",
    jsxFragment: "Fragment",
    jsx: "automatic",
    loader: {
      ".png": "dataurl",
    },
    plugins: [
      postcssPlugin({
        postcss: {
          plugins: [tailwindcss, autoprefixer],
        },
      }),
      copyStaticFilesPlugin({
        source: ["src/manifest.json", "src/assets/"],
        target: buildDir,
        copyWithFolder: false,
      }),
      copyStaticFilesPlugin({
        source: ["src/options/options.html"],
        target: buildDir + "/options",
        copyWithFolder: false,
      }),
      copyStaticFilesPlugin({
        source: ["src/_locales/"],
        target: buildDir,
        copyWithFolder: true,
      }),
    ],
  });
}

async function createZipExtensionForBrowser(browser) {
  const manifest = await fs.readJson(`${buildDir}/manifest.json`);
  const version = manifest.version;
  let archiveName = `build/webchatgpt-${version}-${browser}.zip`;

  const archive = archiver("zip", { zlib: { level: 9 } });
  const stream = fs.createWriteStream(archiveName);

  archive.pipe(stream);

  await addFilesToZip(archive, browser);

  console.log(`Creating ${archiveName}…`);
  archive.finalize();
}

async function addFilesToZip(archive, browser) {
  const entries = await fs.readdir("build");
  for (const entry of entries) {
    const entryStat = await fs.stat(`build/${entry}`);

    if (entryStat.isDirectory()) {
      archive.directory(`build/${entry}`, entry);
    } else {
      if (path.extname(entry) === ".zip") continue;
      if (entry === "manifest.json") continue;
      archive.file(`build/${entry}`, { name: entry });
    }
  }
  if (browser === "firefox") {
    archive.file("src/manifest.v2.json", { name: "manifest.json" });
  } else if (browser === "chrome") {
    archive.file("build/manifest.json", { name: "manifest.json" });
  }
}

async function build() {
  await cleanBuildDir();
  await runEsbuild();

  const createZips = process.argv.includes("--create-zips");
  if (createZips) {
    try {
      await deleteZipsInBuildFolder();

      await createZipExtensionForBrowser("chrome");
      await createZipExtensionForBrowser("firefox");
    } catch (error) {
      console.error(error);
    }
  }

  console.log("Build complete");

  async function deleteZipsInBuildFolder() {
    const entries = await fs.readdir("build");
    for (const entry of entries) {
      if (path.extname(entry) === ".zip") {
        await fs.remove(`build/${entry}`);
      }
    }
  }
}

build();