72 lines
2.4 KiB
TypeScript
72 lines
2.4 KiB
TypeScript
import { existsSync, openSync, closeSync, readSync } from "node:fs";
|
|
import path from "node:path";
|
|
|
|
const outputDir = path.resolve(import.meta.dir, "../../web/public");
|
|
const outputs = {
|
|
windows: path.join(outputDir, "OpenAICanvas-ComfyBridge.exe"),
|
|
linuxAmd64: path.join(outputDir, "OpenAICanvas-ComfyBridge-linux-amd64"),
|
|
linuxArm64: path.join(outputDir, "OpenAICanvas-ComfyBridge-linux-arm64"),
|
|
} as const;
|
|
|
|
function isWindowsExecutable(file: string) {
|
|
if (!existsSync(file)) return false;
|
|
const descriptor = openSync(file, "r");
|
|
try {
|
|
const header = Buffer.alloc(2);
|
|
return readSync(descriptor, header, 0, header.length, 0) === 2 && header[0] === 0x4d && header[1] === 0x5a;
|
|
} finally {
|
|
closeSync(descriptor);
|
|
}
|
|
}
|
|
|
|
function isLinuxExecutable(file: string) {
|
|
if (!existsSync(file)) return false;
|
|
const descriptor = openSync(file, "r");
|
|
try {
|
|
const header = Buffer.alloc(4);
|
|
return readSync(descriptor, header, 0, header.length, 0) === 4
|
|
&& header[0] === 0x7f
|
|
&& header[1] === 0x45
|
|
&& header[2] === 0x4c
|
|
&& header[3] === 0x46;
|
|
} finally {
|
|
closeSync(descriptor);
|
|
}
|
|
}
|
|
|
|
function assertBuildOutput(target: keyof typeof outputs) {
|
|
const valid = target === "windows"
|
|
? isWindowsExecutable(outputs[target])
|
|
: isLinuxExecutable(outputs[target]);
|
|
if (!valid) throw new Error(`Bridge 构建产物缺失或格式无效：${path.basename(outputs[target])}`);
|
|
}
|
|
|
|
function build(target: keyof typeof outputs, goos: string, goarch: string) {
|
|
const command = Bun.spawnSync({
|
|
cmd: ["go", "build", "-trimpath", "-ldflags=-s -w", "-o", outputs[target], "."],
|
|
cwd: path.resolve(import.meta.dir, "../native/comfy-bridge"),
|
|
env: {
|
|
...process.env,
|
|
GOOS: goos,
|
|
GOARCH: goarch,
|
|
CGO_ENABLED: "0",
|
|
GOTELEMETRY: "off",
|
|
},
|
|
stdout: "inherit",
|
|
stderr: "inherit",
|
|
});
|
|
if (command.exitCode !== 0) process.exit(command.exitCode);
|
|
assertBuildOutput(target);
|
|
}
|
|
|
|
if (process.env.CANVAS_PREBUILT_BRIDGE === "1") {
|
|
assertBuildOutput("windows");
|
|
assertBuildOutput("linuxAmd64");
|
|
assertBuildOutput("linuxArm64");
|
|
process.exit(0);
|
|
}
|
|
|
|
build("windows", "windows", "amd64");
|
|
build("linuxAmd64", "linux", "amd64");
|
|
build("linuxArm64", "linux", "arm64");
|