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 · 377 lines
JavaScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378import { spawnSync } from "node:child_process";import { chmodSync, mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "node:fs";import { tmpdir } from "node:os";import { delimiter, join } from "node:path";
const installerSource = readFileSync("install.sh", "utf-8");const mainCall = '\nmain "$@"';const mainCallIndex = installerSource.lastIndexOf(mainCall);const ansiPattern = /\x1b\[[0-?]*[ -/]*[@-~]/g;const syncEnd = "\x1b[?2026l";const failures = [];
if (mainCallIndex === -1) { console.error('Installer check failed: could not find final main "$@" call.'); process.exit(1);}
const harnessSource = `${installerSource.slice(0, mainCallIndex)}
prime_agent_test_cols=80prime_agent_test_rows=24
prime_agent_read_terminal_size() { prime_agent_screen_cols="$prime_agent_test_cols" prime_agent_screen_rows="$prime_agent_test_rows"}
print_render_meta() { label="$1" if prime_agent_show_logo; then visible=1 else visible=0 fi content_height=$(prime_agent_content_height) printf '__META__ %s cols=%s rows=%s layout_show_logo=%s lab_width=%s render_lab_width=%s compact=%s visible=%s content_height=%s\\n' \\ "$label" "$prime_agent_screen_cols" "$prime_agent_screen_rows" "$prime_agent_screen_layout_show_logo" \\ "$prime_agent_screen_layout_lab_width" "$prime_agent_screen_render_lab_width" "$prime_agent_screen_compact" "$visible" "$content_height"}
render_case() { prime_agent_screen_title="Installing Prime Agent" prime_agent_screen_detail="Fetching the verified package." prime_agent_screen_question= prime_agent_screen_frame=1 prime_agent_screen_cols="$1" prime_agent_screen_rows="$2" prime_agent_screen_layout_ready=0 prime_agent_screen_layout_show_logo=0 prime_agent_screen_layout_lab_width=0 prime_agent_screen_render_lab_width=0 prime_agent_screen_compact=0 prime_agent_init_screen_layout prime_agent_refresh_screen_layout_mode print_render_meta first printf '__RENDER_START__ first\\n' prime_agent_render_screen printf '__RENDER_END__ first\\n'
prime_agent_screen_frame=2 prime_agent_screen_cols="$3" prime_agent_screen_rows="$4" prime_agent_refresh_screen_layout_mode print_render_meta second printf '__RENDER_START__ second\\n' prime_agent_render_screen printf '__RENDER_END__ second\\n'}
screen_case() { prime_agent_screen_enabled=1 prime_agent_screen_drawn=0 prime_agent_screen_last_cols=0 prime_agent_screen_last_rows=0 prime_agent_screen_layout_ready=0 prime_agent_screen_layout_show_logo=0 prime_agent_screen_layout_lab_width=0 prime_agent_screen_render_lab_width=0 prime_agent_screen_compact=0 prime_agent_screen_frame=0
prime_agent_test_cols="$1" prime_agent_test_rows="$2" printf '__SCREEN_START__ first\\n' >&2 prime_agent_screen "Installing Prime Agent" "Installing Prime Agent" "Fetching the verified package." "" printf '__SCREEN_END__ first\\n' >&2
prime_agent_test_cols="$3" prime_agent_test_rows="$4" printf '__SCREEN_START__ second\\n' >&2 prime_agent_screen "Installing Prime Agent" "Installing Prime Agent" "Fetching the verified package." "" printf '__SCREEN_END__ second\\n' >&2}
progress_case() { progress_details="Preparing global install.Linking command binaries.Finalizing npm install." for progress_frame in 1 24 25 48 49 200; do prime_agent_animation_frame="$progress_frame" printf '__PROGRESS__ %s\t%s\t%s\\n' "$progress_frame" "$(prime_agent_animation_status "Installing Prime Agent" "$progress_details" static)" "$(prime_agent_animation_detail "$progress_details")" done}
render_case "$@"screen_case "$@"progress_case`;
const tempDir = mkdtempSync(join(tmpdir(), "prime-agent-installer-render-"));const harnessPath = join(tempDir, "harness.sh");
try { writeFileSync(harnessPath, harnessSource, "utf-8");
const stableVisible = runCase("stable visible logo", 100, 30, 90, 30); check(stableVisible.meta.first.visible === "1", "expected the initial large render to show the logo"); check(stableVisible.meta.second.visible === "1", "expected a safe resize to keep showing the logo"); check( stableVisible.meta.first.lab_width === stableVisible.meta.second.lab_width, "expected logo lab width to stay stable across a safe resize", ); assertInstallerProgress(stableVisible.progress);
const stableExpand = runCase("stable expanded logo", 60, 24, 120, 32); check(stableExpand.meta.first.visible === "1", "expected the initial medium render to show the logo"); check(stableExpand.meta.second.visible === "1", "expected terminal growth to keep showing the logo"); check( stableExpand.meta.first.lab_width === stableExpand.meta.second.lab_width, "expected logo lab width not to grow after terminal expansion", );
const noLogoStart = runCase("small initial terminal", 41, 24, 100, 30); check(noLogoStart.meta.first.layout_show_logo === "0", "expected a too-narrow initial terminal to freeze text-only layout"); check(noLogoStart.meta.second.visible === "0", "expected terminal growth not to enable a logo after text-only layout was frozen");
const narrowLogo = runCase("narrow logo on width shrink", 100, 30, 60, 24); check(narrowLogo.meta.first.visible === "1", "expected the initial wide render to show the logo"); check(narrowLogo.meta.second.compact === "0", "expected shrink below frozen lab width to keep rendering the logo"); check(narrowLogo.meta.second.visible === "1", "expected narrow width mode to keep showing the logo"); check( Number(narrowLogo.meta.second.render_lab_width) <= 59, "expected narrow width mode to keep the rendered lab width inside the resized terminal", );
const compactWidth = runCase("compact on severe width shrink", 100, 30, 32, 24); check(compactWidth.meta.first.visible === "1", "expected the initial wide render to show the logo"); check(compactWidth.meta.second.compact === "1", "expected shrink below logo width to use compact mode"); check(compactWidth.meta.second.visible === "0", "expected severe compact width mode to hide the logo");
const compactRows = runCase("compact on row shrink", 100, 30, 100, 10); check(compactRows.meta.first.visible === "1", "expected the initial tall render to show the logo"); check(compactRows.meta.second.compact === "1", "expected shrink below frozen splash height to use compact mode"); check(compactRows.meta.second.visible === "0", "expected compact row mode to hide the logo");
checkNpmInstallPolicies();} finally { rmSync(tempDir, { recursive: true, force: true });}
if (failures.length > 0) { console.error(["Installer check failed:", ...failures.map((failure) => `- ${failure}`)].join("\n")); process.exit(1);}
console.log("Installer check passed.");
function checkNpmInstallPolicies() { const binDir = join(tempDir, "bin"); const installHarnessPath = join(tempDir, "install-harness.sh"); const npmPath = join(binDir, "npm"); const tarballPath = join(tempDir, "verified release package.tgz"); const installHarnessSource = `${installerSource.slice(0, mainCallIndex)}
prime_agent_npm_install "$1"`; const npmSource = `#!/bin/shset -eu
if [ "\${1:-}" = "--version" ]; then printf '%s\\n' "$FAKE_NPM_VERSION" exit 0fi[ "\${1:-}" = install ] || exit 1
remote_policy=script_policy=target=for arg in "$@"; do case "$arg" in --allow-remote=*) remote_policy=\${arg#*=} ;; --allow-scripts=*) script_policy=\${arg#*=} ;; "$FAKE_NPM_TARBALL") target="$arg" ;; esacdone[ "$target" = "$FAKE_NPM_TARBALL" ] || exit 1
npm_major=\${FAKE_NPM_VERSION%%.*}if [ "$npm_major" -ge 12 ]; then [ "$remote_policy" = all ] && [ "$script_policy" = "$FAKE_NPM_TARBALL" ] || exit 1else [ -z "$remote_policy" ] && [ -z "$script_policy" ] || exit 1fi`;
mkdirSync(binDir); writeFileSync(installHarnessPath, installHarnessSource, "utf-8"); writeFileSync(npmPath, npmSource, "utf-8"); writeFileSync(tarballPath, "verified fixture", "utf-8"); chmodSync(npmPath, 0o755);
for (const npmVersion of ["10.9.8", "11.12.1", "12.0.2"]) { const result = spawnSync("sh", [installHarnessPath, tarballPath], { encoding: "utf-8", env: { ...process.env, FAKE_NPM_TARBALL: tarballPath, FAKE_NPM_VERSION: npmVersion, PATH: `${binDir}${delimiter}${process.env.PATH ?? ""}`, }, }); if (result.status !== 0) { failures.push(`npm ${npmVersion}: install policy check failed\n${result.stderr}${result.stdout}`); } }}
function runCase(name, initialCols, initialRows, resizedCols, resizedRows) { const result = spawnSync("sh", [harnessPath, String(initialCols), String(initialRows), String(resizedCols), String(resizedRows)], { detached: true, encoding: "utf-8", }); if (result.status !== 0) { failures.push(`${name}: harness exited with ${result.status ?? "unknown"}\n${result.stderr}${result.stdout}`); return emptyParsedCase(); }
const parsed = parseRenderOutput(result.stdout); parsed.screens = parseScreenOutput(result.stderr); assertLineWidths(name, "first", parsed, initialCols, initialRows); assertLineWidths(name, "second", parsed, resizedCols, resizedRows); assertScreenFrame(name, "first", parsed, initialCols, initialRows); assertScreenFrame(name, "second", parsed, resizedCols, resizedRows); return parsed;}
function parseRenderOutput(output) { const parsed = emptyParsedCase(); let activeRender = null;
for (const rawLine of output.split("\n")) { const line = rawLine.replace(/\r$/, ""); if (line.startsWith("__META__ ")) { const [, label, ...fields] = line.split(" "); parsed.meta[label] = Object.fromEntries(fields.map((field) => field.split("="))); continue; } if (line.startsWith("__RENDER_START__ ")) { activeRender = line.slice("__RENDER_START__ ".length); parsed.renders[activeRender] = []; continue; } if (line.startsWith("__RENDER_END__ ")) { activeRender = null; continue; } if (line.startsWith("__PROGRESS__ ")) { const [frame, status, detail] = line.slice("__PROGRESS__ ".length).split("\t"); parsed.progress.push({ frame: Number(frame), status, detail }); continue; } if (activeRender) { parsed.renders[activeRender].push(line.replace(ansiPattern, "")); } }
return parsed;}
function parseScreenOutput(output) { const screens = {}; for (const label of ["first", "second"]) { const startToken = `__SCREEN_START__ ${label}\n`; const endToken = `__SCREEN_END__ ${label}\n`; const startIndex = output.indexOf(startToken); if (startIndex === -1) { failures.push(`missing ${label} screen start marker`); continue; } const contentStart = startIndex + startToken.length; const endIndex = output.indexOf(endToken, contentStart); if (endIndex === -1) { failures.push(`missing ${label} screen end marker`); continue; } screens[label] = output.slice(contentStart, endIndex); } return screens;}
function assertInstallerProgress(progress) { check(progress.length === 6, `expected six progress samples, got ${progress.length}`); if (progress.length !== 6) return;
const expectedDetails = [ "Preparing global install.", "Preparing global install.", "Linking command binaries.", "Linking command binaries.", "Finalizing npm install.", "Finalizing npm install.", ]; for (const [index, expectedDetail] of expectedDetails.entries()) { check( progress[index].detail === expectedDetail, `expected progress sample ${index + 1} to show "${expectedDetail}", got "${progress[index].detail}"`, ); check( progress[index].status === "Installing Prime Agent...", `expected progress sample ${index + 1} to use indeterminate status`, ); check(!progress[index].status.includes("%"), `expected progress sample ${index + 1} not to include a percent`); }}
function assertLineWidths(name, label, parsed, cols, rows) { const lines = parsed.renders[label] ?? []; check(lines.length === rows, `${name}: expected ${label} render to have ${rows} rows, got ${lines.length}`);
const maxWidth = Math.max(cols - 1, 0); for (const [index, line] of lines.entries()) { check(line.length <= maxWidth, `${name}: ${label} render line ${index + 1} reached ${line.length} columns in a ${cols}-column terminal`); }}
function assertScreenFrame(name, label, parsed, cols, rows) { const screen = parsed.screens[label] ?? ""; check(screen.endsWith(syncEnd), `${name}: expected ${label} screen frame to end with synchronized update close`); check(!screen.endsWith(`\n${syncEnd}`), `${name}: expected ${label} screen frame not to emit a trailing row newline`); check(countNewlines(screen) === rows - 1, `${name}: expected ${label} screen frame to contain ${rows - 1} line breaks`);
const lines = screen.replace(ansiPattern, "").split("\n"); check(lines.length === rows, `${name}: expected ${label} screen frame to contain ${rows} rows, got ${lines.length}`); const maxWidth = Math.max(cols - 1, 0); for (const [index, line] of lines.entries()) { check(line.length <= maxWidth, `${name}: ${label} screen line ${index + 1} reached ${line.length} columns in a ${cols}-column terminal`); }}
function countNewlines(text) { let count = 0; for (const char of text) { if (char === "\n") count++; } return count;}
function check(condition, message) { if (!condition) { failures.push(message); }}
function emptyParsedCase() { return { meta: { first: {}, second: {}, }, renders: { first: [], second: [], }, screens: {}, progress: [], };}