import { $ } from "bun"
import path from "node:path"
import { promisify } from "node:util"
import { brotliCompress, constants } from "node:zlib"
import { AppArchive } from "../src/app-archive"
import { collectFiles } from "./files"

export async function buildAppArchive(channel: string, version: string, options?: { skipBuild?: boolean }) {
  if (options?.skipBuild) return AppArchive.encode([])
  const root = path.resolve(import.meta.dirname, "../../app")
  await $`bun run build`.cwd(root).env({
    ...process.env,
    OPENCODE_CHANNEL: channel,
    VITE_OPENCODE_SERVER_MODE: "origin",
    VITE_OPENCODE_VERSION: version,
  })
  return AppArchive.encode(
    await Promise.all(
      (await collectFiles(path.join(root, "dist")))
        .map((key) => key.replaceAll(path.sep, "/"))
        .filter((key) => !key.endsWith(".map"))
        .toSorted()
        .map(async (key) => {
          const body = Buffer.from(await Bun.file(path.join(root, "dist", key)).arrayBuffer())
          // Independent entries let the server materialize only assets the browser requests.
          return [key, await compress(body, channel)] as const
        }),
    ),
  )
}

// The assets compress once per release, so they take brotli's best quality: about 11% smaller than quality 6 for the
// web UI. Each runs on the thread pool, which keeps the build to about a minute. Beta builds on every push and only
// bundles the CLI into the desktop app, so it keeps quality 6, which takes a few seconds.
function compress(body: Buffer, channel: string) {
  return promisify(brotliCompress)(body, {
    params: {
      [constants.BROTLI_PARAM_QUALITY]: channel === "beta" ? 6 : 11,
      [constants.BROTLI_PARAM_SIZE_HINT]: body.length,
    },
  })
}