Something went wrong. Try again.
fork of prime agent with changes i want but are not upstreamable probqbly (and some fixes)
Something went wrong. Try again.
13 kB · 396 lines
JavaScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397#!/usr/bin/env node
import { spawnSync } from "node:child_process";import { createHash } from "node:crypto";import { cpSync, existsSync, mkdirSync, renameSync, readFileSync, rmSync, statSync, writeFileSync,} from "node:fs";import { basename, dirname, isAbsolute, join, relative, resolve } from "node:path";import { fileURLToPath, pathToFileURL } from "node:url";import { assembleBinaryArchives } from "./assemble-release-archives.mjs";import { manifestV1Platforms } from "./release-platforms.mjs";
const root = resolve(dirname(fileURLToPath(import.meta.url)), "..");const defaultOutputDir = join(root, "packages", "coding-agent", "release");const defaultBaseUrl = process.env.PRIME_AGENT_DOWNLOAD_BASE_URL;const publicPackageName = process.env.PRIME_AGENT_PACKAGE_NAME || "prime-agent";const publicCommandName = process.env.PRIME_AGENT_CMD || "prime-agent";const releaseChannels = new Set(["stable", "beta"]);
const releasePackages = [ { packageDir: "ai", publicName: undefined, artifactName: "prime-agent-ai" }, { packageDir: "tui", publicName: undefined, artifactName: "prime-agent-tui" }, { packageDir: "agent", publicName: undefined, artifactName: "prime-agent-core" }, { packageDir: "coding-agent", publicName: publicPackageName, artifactName: publicPackageName },];
function parseArgs(args) { const parsed = { baseUrl: defaultBaseUrl, channel: "stable", outDir: defaultOutputDir, version: undefined, binaryDir: undefined, };
for (let i = 0; i < args.length; i += 1) { const arg = args[i]; switch (arg) { case "--binary-dir": { const value = args[i + 1]; if (!value) throw new Error("--binary-dir requires a value"); parsed.binaryDir = resolve(root, value); i += 1; break; } case "--channel": { const value = args[i + 1]; if (!value || !releaseChannels.has(value)) { throw new Error("--channel must be stable or beta"); } parsed.channel = value; i += 1; break; } case "--base-url": { const value = args[i + 1]; if (!value) throw new Error("--base-url requires a value"); parsed.baseUrl = value; i += 1; break; } case "--out-dir": { const value = args[i + 1]; if (!value) throw new Error("--out-dir requires a value"); parsed.outDir = resolve(root, value); i += 1; break; } case "--version": { const value = args[i + 1]; if (!value) throw new Error("--version requires a value"); parsed.version = normalizeVersion(value); i += 1; break; } case "--help": case "-h": printHelp(); process.exit(0); break; default: throw new Error(`Unknown argument: ${arg}`); } }
if (!parsed.baseUrl) { throw new Error("--base-url or PRIME_AGENT_DOWNLOAD_BASE_URL is required"); }
parsed.baseUrl = parsed.baseUrl.replace(/\/+$/, ""); return parsed;}
function printHelp() { console.log(`Usage: node scripts/pack-prime-agent-release.mjs --base-url url [--channel stable|beta] [--version x.y.z] [--out-dir path] [--binary-dir path]
Creates private npm tarballs for R2 distribution:
<out-dir>/artifacts/prime-agent-<version>.tgz <out-dir>/artifacts/prime-agent-ai-<version>.tgz <out-dir>/artifacts/prime-agent-core-<version>.tgz <out-dir>/artifacts/prime-agent-tui-<version>.tgz <out-dir>/artifacts/SHA256SUMS <out-dir>/artifacts/<channel> <out-dir>/artifacts/latest.json (stable) or beta.json (beta)
Add --binary-dir packages/coding-agent/binaries to include every standalone platform archive.`);}
function normalizeVersion(version) { const normalized = version.startsWith("v") ? version.slice(1) : version; if (!/^[0-9A-Za-z.-]+$/.test(normalized)) { throw new Error(`Invalid release version: ${version}`); } return normalized;}
function readJson(path) { return JSON.parse(readFileSync(path, "utf8"));}
function writeJson(path, value) { writeFileSync(path, `${JSON.stringify(value, null, 2)}\n`);}
function packagePath(packageDir) { return join(root, "packages", packageDir);}
function assertSafeOutputDir(outDir) { const relativeToReleaseRoot = relative(defaultOutputDir, outDir); if (relativeToReleaseRoot === "" || (!relativeToReleaseRoot.startsWith("..") && !isAbsolute(relativeToReleaseRoot))) { return; } throw new Error(`Refusing to remove output directory outside ${defaultOutputDir}: ${outDir}`);}
function packageJsonPath(packageDir) { return join(packagePath(packageDir), "package.json");}
function requireBuiltPackage(packageDir) { const dist = join(packagePath(packageDir), "dist"); if (!existsSync(dist)) { throw new Error(`Missing ${dist}. Run npm run build before packing a release.`); }}
function copyIfExists(source, target) { if (existsSync(source)) { cpSync(source, target, { recursive: true }); }}
function npmTarballName(packageName, version) { return `${packageName.replace(/^@/, "").replace("/", "-")}-${version}.tgz`;}
function releaseTarballUrl(baseUrl, version, tarballFile) { return `${baseUrl}/releases/v${version}/${tarballFile}`;}
function rewriteInternalDependencies(dependencies, internalPackageUrls) { if (!dependencies) return undefined; const rewritten = {}; for (const [name, range] of Object.entries(dependencies)) { rewritten[name] = internalPackageUrls.get(name) || range; } return rewritten;}
function releaseScripts(sourceScripts) { if (!sourceScripts?.postinstall) return undefined; return { postinstall: sourceScripts.postinstall, };}
function createReleasePackageJson(sourcePackage, packageName, releaseVersion, internalPackageUrls) { const packageJson = { ...sourcePackage, name: packageName, version: releaseVersion, dependencies: rewriteInternalDependencies(sourcePackage.dependencies, internalPackageUrls), optionalDependencies: rewriteInternalDependencies(sourcePackage.optionalDependencies, internalPackageUrls), scripts: releaseScripts(sourcePackage.scripts), };
delete packageJson.devDependencies; delete packageJson.overrides; delete packageJson.private;
if (packageName === publicPackageName) { packageJson.bin = { [publicCommandName]: "dist/bundle/cli.js", }; packageJson.piConfig = { ...(packageJson.piConfig || {}), name: publicCommandName, configDir: ".prime/agent", }; }
return packageJson;}
function copyPackageContents(sourceDir, targetDir, packageJson) { mkdirSync(targetDir, { recursive: true }); writeJson(join(targetDir, "package.json"), packageJson);
for (const entry of ["dist", "docs", "examples", "skills", "postinstall.cjs", "README.md", "CHANGELOG.md"]) { copyIfExists(join(sourceDir, entry), join(targetDir, entry)); }}
function run(command, args, cwd) { const result = spawnSync(command, args, { cwd, stdio: "pipe", encoding: "utf8", });
if (result.status !== 0) { if (result.stdout) process.stdout.write(result.stdout); if (result.stderr) process.stderr.write(result.stderr); throw new Error(`${command} ${args.join(" ")} failed with exit code ${result.status}`); }
if (result.stderr) process.stderr.write(result.stderr); return result.stdout.trim();}
function sha256File(path) { const hash = createHash("sha256"); hash.update(readFileSync(path)); return hash.digest("hex");}
export function writeReleaseMetadata({ artifactsDir, channel, releaseVersion, codingAgentTarball, tarballs, binaries,}) { writeFileSync( join(artifactsDir, "SHA256SUMS"), [...tarballs, ...binaries].map((artifact) => `${artifact.sha256} ${artifact.file}`).join("\n") + "\n", ); writeFileSync(join(artifactsDir, channel), `v${releaseVersion}\n`);
const manifestName = channel === "stable" ? "latest.json" : "beta.json"; const manifestV1PlatformSet = new Set(manifestV1Platforms); const binariesV1 = binaries.filter((entry) => manifestV1PlatformSet.has(entry.platform)); writeJson(join(artifactsDir, manifestName), { version: `v${releaseVersion}`, package: publicPackageName, tarball: `releases/v${releaseVersion}/${codingAgentTarball}`, ...(binariesV1.length > 0 ? { binaries: binariesV1 } : {}), ...(binaries.length > 0 ? { binariesV2: binaries } : {}), tarballs: tarballs.map((tarball) => ({ package: tarball.name, file: tarball.file, sha256: tarball.sha256, })), });}
function main() { const args = parseArgs(process.argv.slice(2)); const sourcePackages = new Map( releasePackages.map((releasePackage) => [ releasePackage.packageDir, readJson(packageJsonPath(releasePackage.packageDir)), ]), ); const cliPackage = sourcePackages.get("coding-agent"); const releaseVersion = args.version || normalizeVersion(process.env.PRIME_AGENT_VERSION || cliPackage.version);
for (const releasePackage of releasePackages) { requireBuiltPackage(releasePackage.packageDir); }
// Dependency keys stay on the source package names so existing compiled imports // keep resolving, while release package names and artifact filenames are branded. const sourcePackageNames = new Map(); const packageNames = new Map(); const artifactFiles = new Map(); for (const releasePackage of releasePackages) { const sourcePackage = sourcePackages.get(releasePackage.packageDir); const packageName = releasePackage.publicName || releasePackage.artifactName || sourcePackage.name; sourcePackageNames.set(releasePackage.packageDir, sourcePackage.name); packageNames.set(releasePackage.packageDir, packageName); artifactFiles.set( releasePackage.packageDir, npmTarballName(releasePackage.artifactName || packageName, releaseVersion), ); }
const internalPackageUrls = new Map(); for (const releasePackage of releasePackages) { if (releasePackage.packageDir === "coding-agent") continue; const sourcePackageName = sourcePackageNames.get(releasePackage.packageDir); const artifactFile = artifactFiles.get(releasePackage.packageDir); internalPackageUrls.set(sourcePackageName, releaseTarballUrl(args.baseUrl, releaseVersion, artifactFile)); }
const stagingRoot = join(args.outDir, "packages"); const artifactsDir = join(args.outDir, "artifacts"); assertSafeOutputDir(args.outDir); rmSync(args.outDir, { force: true, recursive: true }); mkdirSync(stagingRoot, { recursive: true }); mkdirSync(artifactsDir, { recursive: true });
const tarballs = []; for (const releasePackage of releasePackages) { const sourcePackage = sourcePackages.get(releasePackage.packageDir); const packageName = packageNames.get(releasePackage.packageDir); const stagingDir = join(stagingRoot, releasePackage.packageDir); const packageJson = createReleasePackageJson( sourcePackage, packageName, releaseVersion, internalPackageUrls, );
copyPackageContents(packagePath(releasePackage.packageDir), stagingDir, packageJson); if (releasePackage.packageDir === "coding-agent" && args.binaryDir) { const bundleDir = join(stagingDir, "dist/bundle"); renameSync(join(bundleDir, "cli.js"), join(bundleDir, "cli-node.js")); cpSync(join(stagingDir, "dist/cli/npm-native-bridge.js"), join(bundleDir, "cli.js")); cpSync(join(root, "install.sh"), join(stagingDir, "dist/install.sh")); writeJson(join(stagingDir, "dist/native-release.json"), { baseUrl: args.baseUrl, version: releaseVersion }); }
const tarballName = run("npm", ["pack", stagingDir, "--pack-destination", artifactsDir, "--silent"], root) .split("\n") .at(-1); if (!tarballName) { throw new Error(`npm pack did not report a tarball name for ${packageName}`); }
const tarballPath = join(artifactsDir, basename(tarballName)); if (!existsSync(tarballPath) || !statSync(tarballPath).isFile()) { throw new Error(`npm pack did not create ${tarballPath}`); }
const artifactFile = artifactFiles.get(releasePackage.packageDir); const artifactPath = join(artifactsDir, artifactFile); if (tarballPath !== artifactPath) { rmSync(artifactPath, { force: true }); renameSync(tarballPath, artifactPath); }
tarballs.push({ name: packageName, file: artifactFile, sha256: sha256File(artifactPath), }); }
tarballs.sort((left, right) => left.file.localeCompare(right.file)); const binaries = args.binaryDir ? assembleBinaryArchives({ binaryDir: args.binaryDir, artifactsDir, version: releaseVersion }) : []; writeReleaseMetadata({ artifactsDir, channel: args.channel, releaseVersion, codingAgentTarball: artifactFiles.get("coding-agent"), tarballs, binaries, });
for (const artifact of [...tarballs, ...binaries]) { console.log(`Created ${join(artifactsDir, artifact.file)}`); }}
if (process.argv[1] && import.meta.url === pathToFileURL(resolve(process.argv[1])).href) { try { main(); } catch (error) { console.error(error instanceof Error ? error.message : String(error)); process.exit(1); }}