Lifei Zhou
63 lines
2.3 KiB
TypeScript
63 lines
2.3 KiB
TypeScript
import fs from 'node:fs';
|
|
import path from 'node:path';
|
|
import os from 'node:os';
|
|
import log from './logger';
|
|
|
|
/**
|
|
* Ensures Windows shims are available in %LOCALAPPDATA%\Goose\bin
|
|
* This allows the bundled executables to be found via PATH regardless of where Goose is installed
|
|
*/
|
|
export async function ensureWinShims(): Promise<void> {
|
|
if (process.platform !== 'win32') return;
|
|
|
|
const srcDir = path.join(process.resourcesPath, 'bin'); // existing dir
|
|
const tgtDir = path.join(
|
|
process.env.LOCALAPPDATA ?? path.join(os.homedir(), 'AppData', 'Local'),
|
|
'Goose',
|
|
'bin'
|
|
);
|
|
|
|
try {
|
|
await fs.promises.mkdir(tgtDir, { recursive: true });
|
|
|
|
// Copy command-line tools, NOT goosed.exe (which should always be used locally)
|
|
const shims = ['uvx.exe', 'uv.exe', 'npx.cmd'];
|
|
|
|
await Promise.all(
|
|
shims.map(async (shim) => {
|
|
const src = path.join(srcDir, shim);
|
|
const dst = path.join(tgtDir, shim);
|
|
try {
|
|
// Check if source file exists before attempting to copy
|
|
await fs.promises.access(src);
|
|
await fs.promises.copyFile(src, dst); // overwrites with newer build
|
|
log.info(`Copied Windows shim: ${shim} to ${dst}`);
|
|
} catch (e) {
|
|
log.error(`Failed to copy shim ${shim}`, e);
|
|
}
|
|
})
|
|
);
|
|
|
|
// Prepend to PATH **for this process & all children only**.
|
|
// This does NOT modify the user's permanent system PATH.
|
|
const currentPath = process.env.PATH ?? '';
|
|
if (!currentPath.toLowerCase().includes(tgtDir.toLowerCase())) {
|
|
process.env.PATH = `${tgtDir}${path.delimiter}${currentPath}`;
|
|
log.info(`Added ${tgtDir} to PATH for Goose processes only`);
|
|
} else {
|
|
// If it's already in PATH, make sure it's at the beginning
|
|
const pathParts = currentPath.split(path.delimiter);
|
|
const binDirIndex = pathParts.findIndex((p) => p.toLowerCase() === tgtDir.toLowerCase());
|
|
|
|
if (binDirIndex > 0) {
|
|
// Remove it from its current position and add to beginning
|
|
pathParts.splice(binDirIndex, 1);
|
|
process.env.PATH = `${tgtDir}${path.delimiter}${pathParts.join(path.delimiter)}`;
|
|
log.info(`Moved ${tgtDir} to beginning of PATH for Goose processes only`);
|
|
}
|
|
}
|
|
} catch (error) {
|
|
log.error('Failed to ensure Windows shims:', error);
|
|
}
|
|
}
|