kuddev
12f0ef278f
Persist native AI recovery targets and per-pane launch contexts without treating command submission as provider confirmation. Keep window partitions and the active window in the durable session snapshot for immediate and scheduled updates. Coordinate draft approval, snapshot writes, helper readiness, process exit and installation with a committed transaction. Reverify packages and retain recovery state on failure. Isolate acceptance installers from production integrations and reject same-version setup that leaves the executable unchanged. Validation: Windows GPUI suite 1552 passed; settings 55 passed; i18n 20 passed; 9 native helper scenarios; 77 installer migration checks and 41 static invariants. Full interactive packaged upgrade acceptance remains separate.
75 lines
3.7 KiB
JavaScript
75 lines
3.7 KiB
JavaScript
// Execute the shipped bridge, including its bounded native-header reader.
|
|
import assert from 'node:assert/strict';
|
|
import { mkdtempSync, readFileSync, writeFileSync, mkdirSync, rmSync } from 'node:fs';
|
|
import { join } from 'node:path';
|
|
import { stripTypeScriptTypes } from 'node:module';
|
|
import { fileURLToPath } from 'node:url';
|
|
import { test, after } from 'node:test';
|
|
|
|
const root = new URL('../../', import.meta.url);
|
|
const rust = readFileSync(new URL('nebula_app/src/ai_hook/win.rs', root), 'utf8');
|
|
const embedded = rust.match(/const PI_EXTENSION_TS: &str = r#"([\s\S]*?)"#;/)?.[1];
|
|
assert.ok(embedded, 'use the actual generated bridge');
|
|
const events = [];
|
|
globalThis.__pebrelBridgeSpawn = (_hook, args) => {
|
|
events.push(JSON.parse(args[1]));
|
|
return { unref() {} };
|
|
};
|
|
const executable = stripTypeScriptTypes(embedded.replace(
|
|
'import { spawn } from "node:child_process";',
|
|
'const spawn = globalThis.__pebrelBridgeSpawn;',
|
|
) + '\nexport { sessionFor };');
|
|
const bridge = await import('data:text/javascript;base64,' + Buffer.from(executable).toString('base64'));
|
|
mkdirSync(new URL('tmp/', root), { recursive: true });
|
|
const directory = mkdtempSync(fileURLToPath(new URL('tmp/pi-bridge-', root)));
|
|
after(() => rmSync(directory, { recursive: true }));
|
|
const context = (id, file) => ({
|
|
sessionManager: { getSessionId: () => id, getSessionFile: () => file },
|
|
});
|
|
|
|
test('native header replaces filename guessing; process badges are never recovery IDs', () => {
|
|
const file = join(directory, 'timestamp_native-id.jsonl');
|
|
writeFileSync(file, JSON.stringify({ type: 'session', id: 'native-id' }) + '\n');
|
|
assert.deepEqual(bridge.sessionFor(context(undefined, file)), {
|
|
session_id: 'native-id', session_file: file,
|
|
});
|
|
assert.deepEqual(bridge.sessionFor(context(undefined, undefined)), {});
|
|
assert.deepEqual(bridge.sessionFor(context(undefined, file + '.missing')), {});
|
|
assert.deepEqual(bridge.sessionFor({ sessionManager: {
|
|
getSessionId() { throw new Error('closing'); }, getSessionFile: () => file,
|
|
} }), { session_id: 'native-id', session_file: file });
|
|
});
|
|
|
|
test('stale or corrupt file cannot replace direct native identity', () => {
|
|
const file = join(directory, 'mismatch.jsonl');
|
|
writeFileSync(file, JSON.stringify({ type: 'session', id: 'old' }) + '\n');
|
|
assert.deepEqual(bridge.sessionFor(context('current', file)), { session_id: 'current' });
|
|
for (const content of ['invalid\n', '{"type":"message","id":"wrong"}\n', 'x'.repeat(20000)]) {
|
|
writeFileSync(file, content);
|
|
assert.deepEqual(bridge.sessionFor(context(undefined, file)), {});
|
|
}
|
|
});
|
|
|
|
test('session switch retains one process ordering stream and reports identity before turns', async () => {
|
|
const previous = process.env.PEBREL_HOOK_EXE;
|
|
process.env.PEBREL_HOOK_EXE = 'test-hook';
|
|
try {
|
|
const callbacks = new Map();
|
|
bridge.default({ on: (kind, callback) => callbacks.set(kind, callback) });
|
|
await callbacks.get('session_start')({}, context('first'));
|
|
await callbacks.get('session_start')({}, context('second'));
|
|
await callbacks.get('agent_start')({}, context('second'));
|
|
await callbacks.get('agent_end')({}, context('second'));
|
|
assert.deepEqual(events.map(event => event.session_id), ['first', 'second', 'second', 'second']);
|
|
assert.deepEqual(events.map(event => event.kind), ['session-start', 'session-start', 'prompt', 'done']);
|
|
assert.equal(new Set(events.map(event => event.bridge_instance)).size, 1);
|
|
for (let i = 1; i < events.length; i++) {
|
|
assert.ok(BigInt(events[i].bridge_sequence) > BigInt(events[i - 1].bridge_sequence));
|
|
assert.notEqual(events[i].event_id, events[i - 1].event_id);
|
|
}
|
|
} finally {
|
|
if (previous === undefined) delete process.env.PEBREL_HOOK_EXE;
|
|
else process.env.PEBREL_HOOK_EXE = previous;
|
|
}
|
|
});
|