feat: Add script to build Python sidecar executable

2026-05-08 04:13:13 +02:00
parent e4028f58a8
commit 978919b410
1 changed files with 110 additions and 0 deletions
--- a/scripts/build-python-sidecar.mjs
+++ b/scripts/build-python-sidecar.mjs
@@ -0,0 +1,110 @@
+import { existsSync, mkdirSync, statSync } from "node:fs";
+import { dirname, join, resolve } from "node:path";
+import { execFileSync, spawnSync } from "node:child_process";
+import { fileURLToPath } from "node:url";
+
+const root = resolve(dirname(fileURLToPath(import.meta.url)), "..");
+const python = existsSync(join(root, ".venv", "bin", "python"))
+  ? join(root, ".venv", "bin", "python")
+  : "python3";
+
+function run(command, args, options = {}) {
+  const result = spawnSync(command, args, {
+    cwd: root,
+    stdio: "inherit",
+    env: process.env,
+    ...options,
+  });
+  if (result.status !== 0) {
+    process.exit(result.status ?? 1);
+  }
+}
+
+function read(command, args) {
+  return execFileSync(command, args, { cwd: root, encoding: "utf8" }).trim();
+}
+
+function targetTriple() {
+  const explicit =
+    process.env.TAURI_ENV_TARGET_TRIPLE ||
+    process.env.CARGO_BUILD_TARGET ||
+    process.env.TARGET;
+  if (explicit) return explicit;
+
+  const rustc = read("rustc", ["-vV"]);
+  const host = rustc
+    .split(/\r?\n/)
+    .find((line) => line.startsWith("host:"))
+    ?.replace("host:", "")
+    .trim();
+  if (!host) {
+    throw new Error("Could not determine Rust target triple.");
+  }
+  return host;
+}
+
+function newestMtime(paths) {
+  return Math.max(...paths.map((path) => statSync(path).mtimeMs));
+}
+
+const target = targetTriple();
+const binariesDir = join(root, "src-tauri", "binaries");
+const sidecar = join(binariesDir, `concept-api-${target}${target.includes("windows") ? ".exe" : ""}`);
+const sources = [
+  join(root, "concept_api.py"),
+  join(root, "websearch.py"),
+];
+
+mkdirSync(binariesDir, { recursive: true });
+
+if (
+  existsSync(sidecar) &&
+  statSync(sidecar).mtimeMs >= newestMtime(sources) &&
+  process.argv.includes("--force") === false
+) {
+  console.log(`Python sidecar is up to date: ${sidecar}`);
+  process.exit(0);
+}
+
+const pyinstallerCheck = spawnSync(python, ["-m", "PyInstaller", "--version"], {
+  cwd: root,
+  stdio: "ignore",
+});
+if (pyinstallerCheck.status !== 0) {
+  run(python, ["-m", "pip", "install", "pyinstaller"]);
+}
+
+run(python, [
+  "-m",
+  "PyInstaller",
+  "--noconfirm",
+  "--clean",
+  "--onefile",
+  "--name",
+  `concept-api-${target}`,
+  "--distpath",
+  binariesDir,
+  "--workpath",
+  join(root, ".sidecar-build", "work"),
+  "--specpath",
+  join(root, ".sidecar-build"),
+  "--paths",
+  root,
+  "--hidden-import",
+  "fitz",
+  "--hidden-import",
+  "bs4",
+  "--hidden-import",
+  "PIL",
+  "--hidden-import",
+  "PIL.Image",
+  "--exclude-module",
+  "torch",
+  "--exclude-module",
+  "transformers",
+  "--exclude-module",
+  "whisper",
+  "--exclude-module",
+  "cv2",
+  join(root, "concept_api.py"),
+]);