|
|
@@ -0,0 +1,434 @@
|
|
|
+import { createHash } from 'node:crypto';
|
|
|
+import {
|
|
|
+ chmodSync,
|
|
|
+ copyFileSync,
|
|
|
+ existsSync,
|
|
|
+ mkdirSync,
|
|
|
+ mkdtempSync,
|
|
|
+ readFileSync,
|
|
|
+ renameSync,
|
|
|
+ rmSync,
|
|
|
+ writeFileSync,
|
|
|
+} from 'node:fs';
|
|
|
+import { homedir, platform, tmpdir } from 'node:os';
|
|
|
+import * as path from 'node:path';
|
|
|
+import { setTimeout as delay } from 'node:timers/promises';
|
|
|
+import type { CompanionConfig } from '../config/schema';
|
|
|
+import { crossSpawn } from '../utils/compat';
|
|
|
+import { log } from '../utils/logger';
|
|
|
+
|
|
|
+export interface CompanionManifest {
|
|
|
+ version: string;
|
|
|
+ tag: string;
|
|
|
+ repo: string;
|
|
|
+ checksums?: Record<string, string>;
|
|
|
+}
|
|
|
+
|
|
|
+interface CompanionInstallMetadata {
|
|
|
+ version: string;
|
|
|
+ tag: string;
|
|
|
+ target: string;
|
|
|
+ installedAt: string;
|
|
|
+ archiveName: string;
|
|
|
+ checksum?: string;
|
|
|
+}
|
|
|
+
|
|
|
+export type CompanionUpdateResult =
|
|
|
+ | { status: 'installed'; binaryPath: string; version: string }
|
|
|
+ | { status: 'current'; binaryPath: string; version: string }
|
|
|
+ | { status: 'skipped'; reason: string; binaryPath?: string }
|
|
|
+ | { status: 'failed'; error: string; binaryPath: string };
|
|
|
+
|
|
|
+const DOWNLOAD_TIMEOUT_MS = 30_000;
|
|
|
+const LOCK_TIMEOUT_MS = 2_000;
|
|
|
+const FIRST_METADATA_VERSION = '0.1.2';
|
|
|
+
|
|
|
+export const COMPANION_MANIFEST: CompanionManifest = {
|
|
|
+ version: '0.1.2',
|
|
|
+ tag: 'companion-v0.1.2',
|
|
|
+ repo: 'alvinunreal/oh-my-opencode-slim',
|
|
|
+ checksums: {
|
|
|
+ 'oh-my-opencode-slim-companion-v0.1.2-aarch64-apple-darwin.tar.gz':
|
|
|
+ '6f66fdfe895ab39b96ddf1f395c4547cb428cfe68c94a1a3a9f11c1ab46863c2',
|
|
|
+ 'oh-my-opencode-slim-companion-v0.1.2-aarch64-unknown-linux-gnu.tar.gz':
|
|
|
+ 'af979907a429904db4449a79e0d8d99edf994ce373d298284b440549d1eb507b',
|
|
|
+ 'oh-my-opencode-slim-companion-v0.1.2-x86_64-pc-windows-msvc.zip':
|
|
|
+ 'fca87451bcc3a3f5eb5ab6dcd83999bb35e2a4f30d5a666c427b7bb892ac1eb0',
|
|
|
+ 'oh-my-opencode-slim-companion-v0.1.2-x86_64-unknown-linux-gnu.tar.gz':
|
|
|
+ 'e6352a4e6c71773617ca2831af2102b011d3114c0fb8d40635b6af8076b757bc',
|
|
|
+ },
|
|
|
+};
|
|
|
+
|
|
|
+export function getCompanionTarget(): string | null {
|
|
|
+ const p = process.platform;
|
|
|
+ const a = process.arch;
|
|
|
+ if (p === 'darwin') {
|
|
|
+ if (a === 'arm64') return 'aarch64-apple-darwin';
|
|
|
+ } else if (p === 'linux') {
|
|
|
+ if (a === 'x64') return 'x86_64-unknown-linux-gnu';
|
|
|
+ if (a === 'arm64') return 'aarch64-unknown-linux-gnu';
|
|
|
+ } else if (p === 'win32') {
|
|
|
+ if (a === 'x64') return 'x86_64-pc-windows-msvc';
|
|
|
+ }
|
|
|
+ return null;
|
|
|
+}
|
|
|
+
|
|
|
+export function getCompanionBinaryPath(): string {
|
|
|
+ const xdg = process.env.XDG_DATA_HOME?.trim();
|
|
|
+ const base =
|
|
|
+ xdg && path.isAbsolute(xdg) ? xdg : path.join(homedir(), '.local', 'share');
|
|
|
+ return path.join(
|
|
|
+ base,
|
|
|
+ 'opencode',
|
|
|
+ 'storage',
|
|
|
+ 'oh-my-opencode-slim',
|
|
|
+ 'bin',
|
|
|
+ platform() === 'win32'
|
|
|
+ ? 'oh-my-opencode-slim-companion.exe'
|
|
|
+ : 'oh-my-opencode-slim-companion',
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+export function loadCompanionManifestFromPackageRoot(
|
|
|
+ packageRoot: string,
|
|
|
+): CompanionManifest | null {
|
|
|
+ const manifestPath = path.join(
|
|
|
+ packageRoot,
|
|
|
+ 'src',
|
|
|
+ 'companion',
|
|
|
+ 'companion-manifest.json',
|
|
|
+ );
|
|
|
+ try {
|
|
|
+ const parsed = JSON.parse(
|
|
|
+ readFileSync(manifestPath, 'utf8'),
|
|
|
+ ) as Partial<CompanionManifest>;
|
|
|
+ if (parsed.version && parsed.tag && parsed.repo) {
|
|
|
+ return {
|
|
|
+ version: parsed.version,
|
|
|
+ tag: parsed.tag,
|
|
|
+ repo: parsed.repo,
|
|
|
+ checksums: parsed.checksums,
|
|
|
+ };
|
|
|
+ }
|
|
|
+ } catch {}
|
|
|
+ return null;
|
|
|
+}
|
|
|
+
|
|
|
+export async function ensureCompanionVersion(options: {
|
|
|
+ config?: CompanionConfig;
|
|
|
+ manifest?: CompanionManifest;
|
|
|
+ dryRun?: boolean;
|
|
|
+ downloadTimeoutMs?: number;
|
|
|
+ lockTimeoutMs?: number;
|
|
|
+}): Promise<CompanionUpdateResult> {
|
|
|
+ const { config, dryRun = false } = options;
|
|
|
+ const manifest = options.manifest ?? COMPANION_MANIFEST;
|
|
|
+ const binaryPath = getCompanionBinaryPath();
|
|
|
+
|
|
|
+ if (config?.enabled !== true) {
|
|
|
+ return { status: 'skipped', reason: 'disabled', binaryPath };
|
|
|
+ }
|
|
|
+
|
|
|
+ if (config.binaryPath?.trim()) {
|
|
|
+ return { status: 'skipped', reason: 'custom-binary', binaryPath };
|
|
|
+ }
|
|
|
+
|
|
|
+ const target = getCompanionTarget();
|
|
|
+ if (!target) {
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath,
|
|
|
+ error: `Unsupported platform/architecture: ${process.platform} ${process.arch}`,
|
|
|
+ };
|
|
|
+ }
|
|
|
+
|
|
|
+ const current = readInstallMetadata(binaryPath);
|
|
|
+ if (
|
|
|
+ existsSync(binaryPath) &&
|
|
|
+ !current &&
|
|
|
+ manifest.version === FIRST_METADATA_VERSION
|
|
|
+ ) {
|
|
|
+ const archiveName = companionArchiveName(manifest.version, target);
|
|
|
+ writeInstallMetadata(binaryPath, {
|
|
|
+ version: manifest.version,
|
|
|
+ tag: manifest.tag,
|
|
|
+ target,
|
|
|
+ installedAt: new Date().toISOString(),
|
|
|
+ archiveName,
|
|
|
+ checksum: manifest.checksums?.[archiveName],
|
|
|
+ });
|
|
|
+ return { status: 'current', binaryPath, version: manifest.version };
|
|
|
+ }
|
|
|
+
|
|
|
+ if (
|
|
|
+ existsSync(binaryPath) &&
|
|
|
+ current?.target === target &&
|
|
|
+ compareSemver(current.version, manifest.version) >= 0
|
|
|
+ ) {
|
|
|
+ return { status: 'current', binaryPath, version: current.version };
|
|
|
+ }
|
|
|
+
|
|
|
+ if (dryRun) {
|
|
|
+ return { status: 'installed', binaryPath, version: manifest.version };
|
|
|
+ }
|
|
|
+
|
|
|
+ return withCompanionInstallLock(
|
|
|
+ binaryPath,
|
|
|
+ options.lockTimeoutMs,
|
|
|
+ async () => {
|
|
|
+ const lockedCurrent = readInstallMetadata(binaryPath);
|
|
|
+ if (
|
|
|
+ existsSync(binaryPath) &&
|
|
|
+ !lockedCurrent &&
|
|
|
+ manifest.version === FIRST_METADATA_VERSION
|
|
|
+ ) {
|
|
|
+ const archiveName = companionArchiveName(manifest.version, target);
|
|
|
+ writeInstallMetadata(binaryPath, {
|
|
|
+ version: manifest.version,
|
|
|
+ tag: manifest.tag,
|
|
|
+ target,
|
|
|
+ installedAt: new Date().toISOString(),
|
|
|
+ archiveName,
|
|
|
+ checksum: manifest.checksums?.[archiveName],
|
|
|
+ });
|
|
|
+ return { status: 'current', binaryPath, version: manifest.version };
|
|
|
+ }
|
|
|
+
|
|
|
+ if (
|
|
|
+ existsSync(binaryPath) &&
|
|
|
+ lockedCurrent?.target === target &&
|
|
|
+ compareSemver(lockedCurrent.version, manifest.version) >= 0
|
|
|
+ ) {
|
|
|
+ return {
|
|
|
+ status: 'current',
|
|
|
+ binaryPath,
|
|
|
+ version: lockedCurrent.version,
|
|
|
+ };
|
|
|
+ }
|
|
|
+
|
|
|
+ return installCompanionArchive(
|
|
|
+ binaryPath,
|
|
|
+ target,
|
|
|
+ manifest,
|
|
|
+ options.downloadTimeoutMs ?? DOWNLOAD_TIMEOUT_MS,
|
|
|
+ );
|
|
|
+ },
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+async function installCompanionArchive(
|
|
|
+ finalBinaryPath: string,
|
|
|
+ target: string,
|
|
|
+ manifest: CompanionManifest,
|
|
|
+ downloadTimeoutMs: number,
|
|
|
+): Promise<CompanionUpdateResult> {
|
|
|
+ const isWindows = process.platform === 'win32';
|
|
|
+ const archiveName = companionArchiveName(manifest.version, target, isWindows);
|
|
|
+ const downloadUrl = `https://github.com/${manifest.repo}/releases/download/${manifest.tag}/${archiveName}`;
|
|
|
+ const expectedChecksum = manifest.checksums?.[archiveName];
|
|
|
+ if (!expectedChecksum) {
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath: finalBinaryPath,
|
|
|
+ error: `Missing SHA256 checksum for companion archive: ${archiveName}`,
|
|
|
+ };
|
|
|
+ }
|
|
|
+
|
|
|
+ let buffer: ArrayBuffer;
|
|
|
+ const controller = new AbortController();
|
|
|
+ const timeout = setTimeout(() => controller.abort(), downloadTimeoutMs);
|
|
|
+ try {
|
|
|
+ const res = await fetch(downloadUrl, { signal: controller.signal });
|
|
|
+ if (!res.ok) {
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath: finalBinaryPath,
|
|
|
+ error: `Failed to download companion binary (HTTP ${res.status}): ${res.statusText}`,
|
|
|
+ };
|
|
|
+ }
|
|
|
+ buffer = await res.arrayBuffer();
|
|
|
+ } catch (err) {
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath: finalBinaryPath,
|
|
|
+ error: `Failed to fetch companion archive: ${formatError(err)}`,
|
|
|
+ };
|
|
|
+ } finally {
|
|
|
+ clearTimeout(timeout);
|
|
|
+ }
|
|
|
+
|
|
|
+ const checksum = createHash('sha256')
|
|
|
+ .update(Buffer.from(buffer))
|
|
|
+ .digest('hex');
|
|
|
+ if (checksum !== expectedChecksum) {
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath: finalBinaryPath,
|
|
|
+ error: 'Companion archive checksum mismatch',
|
|
|
+ };
|
|
|
+ }
|
|
|
+
|
|
|
+ let tempDir = '';
|
|
|
+ try {
|
|
|
+ tempDir = mkdtempSync(path.join(tmpdir(), 'companion-install-'));
|
|
|
+ const archivePath = path.join(tempDir, archiveName);
|
|
|
+ writeFileSync(archivePath, Buffer.from(buffer));
|
|
|
+
|
|
|
+ const extractedDir = path.join(tempDir, 'extracted');
|
|
|
+ mkdirSync(extractedDir, { recursive: true });
|
|
|
+
|
|
|
+ if (isWindows) {
|
|
|
+ const { extractZip } = await import('../utils/zip-extractor');
|
|
|
+ await extractZip(archivePath, extractedDir);
|
|
|
+ } else {
|
|
|
+ const proc = crossSpawn(['tar', '-xzf', archivePath, '-C', extractedDir]);
|
|
|
+ const exitCode = await proc.exited;
|
|
|
+ if (exitCode !== 0) {
|
|
|
+ const stderr = await proc.stderr();
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath: finalBinaryPath,
|
|
|
+ error: `Archive extraction failed (tar exited with ${exitCode}): ${stderr}`,
|
|
|
+ };
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ const binaryName = isWindows
|
|
|
+ ? 'oh-my-opencode-slim-companion.exe'
|
|
|
+ : 'oh-my-opencode-slim-companion';
|
|
|
+ const extractedBinaryPath = path.join(extractedDir, binaryName);
|
|
|
+
|
|
|
+ if (!existsSync(extractedBinaryPath)) {
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath: finalBinaryPath,
|
|
|
+ error: `Binary ${binaryName} not found in extracted archive`,
|
|
|
+ };
|
|
|
+ }
|
|
|
+
|
|
|
+ const binDir = path.dirname(finalBinaryPath);
|
|
|
+ mkdirSync(binDir, { recursive: true });
|
|
|
+
|
|
|
+ const tmpFinalPath = `${finalBinaryPath}.tmp`;
|
|
|
+ copyFileSync(extractedBinaryPath, tmpFinalPath);
|
|
|
+
|
|
|
+ if (!isWindows) {
|
|
|
+ chmodSync(tmpFinalPath, 0o755);
|
|
|
+ }
|
|
|
+
|
|
|
+ renameSync(tmpFinalPath, finalBinaryPath);
|
|
|
+ writeInstallMetadata(finalBinaryPath, {
|
|
|
+ version: manifest.version,
|
|
|
+ tag: manifest.tag,
|
|
|
+ target,
|
|
|
+ installedAt: new Date().toISOString(),
|
|
|
+ archiveName,
|
|
|
+ checksum,
|
|
|
+ });
|
|
|
+
|
|
|
+ return {
|
|
|
+ status: 'installed',
|
|
|
+ binaryPath: finalBinaryPath,
|
|
|
+ version: manifest.version,
|
|
|
+ };
|
|
|
+ } catch (err) {
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath: finalBinaryPath,
|
|
|
+ error: `Failed to install companion: ${formatError(err)}`,
|
|
|
+ };
|
|
|
+ } finally {
|
|
|
+ if (tempDir) {
|
|
|
+ try {
|
|
|
+ rmSync(tempDir, { recursive: true, force: true });
|
|
|
+ } catch {}
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+function readInstallMetadata(
|
|
|
+ binaryPath: string,
|
|
|
+): CompanionInstallMetadata | null {
|
|
|
+ try {
|
|
|
+ const parsed = JSON.parse(
|
|
|
+ readFileSync(metadataPath(binaryPath), 'utf8'),
|
|
|
+ ) as Partial<CompanionInstallMetadata> | null;
|
|
|
+ if (parsed?.version && parsed.tag && parsed.target) {
|
|
|
+ return parsed as CompanionInstallMetadata;
|
|
|
+ }
|
|
|
+ } catch {}
|
|
|
+ return null;
|
|
|
+}
|
|
|
+
|
|
|
+function writeInstallMetadata(
|
|
|
+ binaryPath: string,
|
|
|
+ metadata: CompanionInstallMetadata,
|
|
|
+): void {
|
|
|
+ writeFileSync(metadataPath(binaryPath), JSON.stringify(metadata, null, 2));
|
|
|
+}
|
|
|
+
|
|
|
+function metadataPath(binaryPath: string): string {
|
|
|
+ return `${binaryPath}.json`;
|
|
|
+}
|
|
|
+
|
|
|
+async function withCompanionInstallLock(
|
|
|
+ binaryPath: string,
|
|
|
+ timeoutMs: number | undefined,
|
|
|
+ run: () => Promise<CompanionUpdateResult>,
|
|
|
+): Promise<CompanionUpdateResult> {
|
|
|
+ const lock = `${binaryPath}.lock`;
|
|
|
+ const deadline = Date.now() + (timeoutMs ?? LOCK_TIMEOUT_MS);
|
|
|
+ mkdirSync(path.dirname(binaryPath), { recursive: true });
|
|
|
+ while (Date.now() <= deadline) {
|
|
|
+ try {
|
|
|
+ mkdirSync(lock);
|
|
|
+ try {
|
|
|
+ return await run();
|
|
|
+ } finally {
|
|
|
+ try {
|
|
|
+ rmSync(lock, { recursive: true, force: true });
|
|
|
+ } catch {}
|
|
|
+ }
|
|
|
+ } catch (err) {
|
|
|
+ const code = (err as NodeJS.ErrnoException).code;
|
|
|
+ if (code !== 'EEXIST') throw err;
|
|
|
+ await delay(25);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ log('[companion] install lock timed out', lock);
|
|
|
+ return {
|
|
|
+ status: 'failed',
|
|
|
+ binaryPath,
|
|
|
+ error: 'Timed out waiting for companion install lock',
|
|
|
+ };
|
|
|
+}
|
|
|
+
|
|
|
+function companionArchiveName(
|
|
|
+ version: string,
|
|
|
+ target: string,
|
|
|
+ isWindows = process.platform === 'win32',
|
|
|
+): string {
|
|
|
+ const ext = isWindows ? 'zip' : 'tar.gz';
|
|
|
+ return `oh-my-opencode-slim-companion-v${version}-${target}.${ext}`;
|
|
|
+}
|
|
|
+
|
|
|
+function compareSemver(a: string, b: string): number {
|
|
|
+ const left = parseSemver(a);
|
|
|
+ const right = parseSemver(b);
|
|
|
+ if (!left || !right) return a.localeCompare(b);
|
|
|
+ for (let i = 0; i < 3; i++) {
|
|
|
+ const diff = left[i] - right[i];
|
|
|
+ if (diff !== 0) return diff;
|
|
|
+ }
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+function parseSemver(version: string): [number, number, number] | null {
|
|
|
+ const match = version.match(/^(\d+)\.(\d+)\.(\d+)(?:[-+].*)?$/);
|
|
|
+ if (!match) return null;
|
|
|
+ return [Number(match[1]), Number(match[2]), Number(match[3])];
|
|
|
+}
|
|
|
+
|
|
|
+function formatError(err: unknown): string {
|
|
|
+ return err instanceof Error ? err.message : String(err);
|
|
|
+}
|
Alvin Unreal