From 9aed926cda818cfb35dcb376212e14b8d3ffd41e Mon Sep 17 00:00:00 2001 From: mesalogo Date: Tue, 8 Sep 2026 01:09:14 +0800 Subject: [PATCH] fix: stabilize OpenCode reuse and session switching MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Check cached embedded servers before reuse, isolate concurrent cancellation, and serialize request MCP registration and cleanup. Preserve native controls across session switches and recover usable partial inventories. Share the bundled offline OpenCode configuration with desktop and Agent launch paths, including model bridge environment forwarding. Validation: focused tests and typecheck passed; lint passed before the final fixes. The full test run exceeded 120 seconds and reported a DS stdio MCP failure. Current-source Agent host and real suspend/resume validation remain pending. Release note: 修复切换会话后 OpenCode 原生控件消失及并发取消相互影响的问题，并完善内嵌服务复用检查和桌面、Agent 的离线依赖配置。 --- build/opencode-config.cjs | 149 + build/remote-runtime-bundle.cjs | 1017 +- build/runtime-hooks.cjs | 334 +- docs/features/assistant-workbar/README.md | 2 +- docs/features/assistant-workbar/progress.md | 18 + src/agent-daemon/model-bridge-helper.ts | 20 + src/agent-daemon/model-bridge.test.ts | 90 +- src/agent-daemon/opencode-runtime-profile.ts | 240 +- src/main/agent/opencode-runtime.test.ts | 5658 ++++--- src/main/agent/opencode-runtime.ts | 3671 ++-- src/renderer/src/App.test.tsx | 14885 ++++++++--------- src/renderer/src/App.tsx | 13475 ++++++++------- 12 files changed, 19417 insertions(+), 20142 deletions(-) create mode 100644 build/opencode-config.cjs diff --git a/build/opencode-config.cjs b/build/opencode-config.cjs new file mode 100644 index 00000000..60445c91 --- /dev/null +++ b/build/opencode-config.cjs @@ -0,0 +1,149 @@ +const { existsSync } = require("node:fs"); +const { + mkdirSync, + readFileSync, + renameSync, + rmSync, + statSync, + writeFileSync, +} = require("node:fs"); +const { join } = require("node:path"); +const { spawnSync } = require("node:child_process"); + +const opencodeVersion = "1.18.29"; +const opencodePluginPackage = "@opencode-ai/plugin"; + +function npmInvocation() { + const npmCli = process.env.npm_execpath; + if (npmCli) { + return { command: process.execPath, prefixArgs: [npmCli] }; + } + if (process.platform === "win32") { + throw new Error("npm_execpath is required to prepare OpenCode config"); + } + return { command: "npm", prefixArgs: [] }; +} + +function pluginIntegrity(projectDir) { + const lock = JSON.parse( + readFileSync(join(projectDir, "package-lock.json"), "utf8"), + ); + const entry = lock.packages?.[`node_modules/${opencodePluginPackage}`]; + if ( + entry?.version !== opencodeVersion || + typeof entry.integrity !== "string" + ) { + throw new Error( + `Missing locked ${opencodePluginPackage}@${opencodeVersion} integrity`, + ); + } + return entry.integrity; +} + +function prepareBundledOpenCodeConfig(projectDir) { + const targetDirectory = join( + projectDir, + ".runtime-resources", + "opencode-config", + ); + const identity = { + packageName: opencodePluginPackage, + version: opencodeVersion, + integrity: pluginIntegrity(projectDir), + }; + const markerPath = join(targetDirectory, ".goodbuddy-ready.json"); + try { + const ready = JSON.parse(readFileSync(markerPath, "utf8")); + if ( + ready.packageName === identity.packageName && + ready.version === identity.version && + ready.integrity === identity.integrity && + statSync( + join( + targetDirectory, + "node_modules", + "@opencode-ai", + "plugin", + "package.json", + ), + ).isFile() + ) { + return targetDirectory; + } + } catch { + // Rebuild an incomplete or stale offline config template. + } + + const stagingDirectory = `${targetDirectory}.staging-${process.pid}`; + rmSync(stagingDirectory, { recursive: true, force: true }); + mkdirSync(stagingDirectory, { recursive: true }); + try { + writeFileSync( + join(stagingDirectory, "package.json"), + `${JSON.stringify( + { + private: true, + dependencies: { + [opencodePluginPackage]: opencodeVersion, + }, + }, + null, + 2, + )}\n`, + "utf8", + ); + writeFileSync(join(stagingDirectory, ".gitignore"), "node_modules\n"); + const npm = npmInvocation(); + const environment = { ...process.env }; + delete environment.NODE_TLS_REJECT_UNAUTHORIZED; + const result = spawnSync( + npm.command, + [ + ...npm.prefixArgs, + "install", + "--ignore-scripts", + "--omit=dev", + "--omit=optional", + "--no-audit", + "--no-fund", + "--save-exact", + ], + { + cwd: stagingDirectory, + encoding: "utf8", + env: environment, + maxBuffer: 16 * 1024 * 1024, + shell: false, + windowsHide: true, + }, + ); + if (result.status !== 0) { + throw new Error( + `Unable to prepare bundled OpenCode config: ${ + result.error?.message || + result.stderr || + result.stdout || + `npm exited with ${result.status}` + }`, + ); + } + writeFileSync( + join(stagingDirectory, ".goodbuddy-ready.json"), + `${JSON.stringify(identity)}\n`, + "utf8", + ); + rmSync(targetDirectory, { recursive: true, force: true }); + mkdirSync(join(targetDirectory, ".."), { recursive: true }); + renameSync(stagingDirectory, targetDirectory); + return targetDirectory; + } finally { + if (existsSync(stagingDirectory)) { + rmSync(stagingDirectory, { recursive: true, force: true }); + } + } +} + +module.exports = { + opencodeVersion, + prepareBundledOpenCodeConfig, +}; diff --git a/build/remote-runtime-bundle.cjs b/build/remote-runtime-bundle.cjs index e913a3eb..dc8fb5a1 100644 --- a/build/remote-runtime-bundle.cjs +++ b/build/remote-runtime-bundle.cjs @@ -1,12 +1,8 @@ -const { - createHash, - createPublicKey, - sign, - verify -} = require('node:crypto') +const { createHash, createPublicKey, sign, verify } = require("node:crypto"); const { chmodSync, closeSync, + cpSync, copyFileSync, existsSync, lstatSync, @@ -19,9 +15,9 @@ const { renameSync, rmSync, statSync, - writeFileSync -} = require('node:fs') -const { tmpdir } = require('node:os') + writeFileSync, +} = require("node:fs"); +const { tmpdir } = require("node:os"); const { basename, dirname, @@ -29,450 +25,423 @@ const { posix, relative, resolve, - sep -} = require('node:path') -const tar = require('tar') + sep, +} = require("node:path"); +const tar = require("tar"); +const { targetName, assertTargetBinary } = require("./agent-build-target.cjs"); const { - targetName, - assertTargetBinary -} = require('./agent-build-target.cjs') -const { - preflightRegisteredProductionKey -} = require('./signing-key-preflight.cjs') + preflightRegisteredProductionKey, +} = require("./signing-key-preflight.cjs"); +const { prepareBundledOpenCodeConfig } = require("./opencode-config.cjs"); -const root = join(__dirname, '..') -const supportedArchitectures = Object.freeze(['x64', 'arm64']) -const manifestFileName = 'manifest.json' -const signatureFileName = 'manifest.sig' -const signingPrivateKeyEnvironment = - 'GOODBUDDY_SIGNING_PRIVATE_KEY' -const signingKeyIdEnvironment = - 'GOODBUDDY_SIGNING_KEY_ID' +const root = join(__dirname, ".."); +const supportedArchitectures = Object.freeze(["x64", "arm64"]); +const manifestFileName = "manifest.json"; +const signatureFileName = "manifest.sig"; +const signingPrivateKeyEnvironment = "GOODBUDDY_SIGNING_PRIVATE_KEY"; +const signingKeyIdEnvironment = "GOODBUDDY_SIGNING_KEY_ID"; const signatureDomain = Buffer.from( - 'GoodBuddy Remote Runtime Bundle Manifest Signature v1\0', - 'utf8' -) + "GoodBuddy Remote Runtime Bundle Manifest Signature v1\0", + "utf8", +); const profileDigests = Object.freeze({ adapter: - 'sha256:ac7696abe504bbea444d8ad80dd8faa96437e6dbb27ed987ce46e1045f3dd365', - acp: - 'sha256:0e3764ab897258bc0234162357c2eefc3faeb9918080ae81b60e2084ed7996a6' -}) + "sha256:ac7696abe504bbea444d8ad80dd8faa96437e6dbb27ed987ce46e1045f3dd365", + acp: "sha256:0e3764ab897258bc0234162357c2eefc3faeb9918080ae81b60e2084ed7996a6", +}); const fixedLimits = Object.freeze({ maximumPromptRuntimeMilliseconds: 10 * 60 * 1000, maximumPromptInputBytes: 16 * 1024 * 1024, - maximumPromptOutputBytes: 8 * 1024 * 1024 -}) + maximumPromptOutputBytes: 8 * 1024 * 1024, +}); function readJson(filePath, description) { try { - return JSON.parse(readFileSync(filePath, 'utf8')) + return JSON.parse(readFileSync(filePath, "utf8")); } catch (error) { throw new Error( `${description} is missing or invalid: ${ error instanceof Error ? error.message : String(error) }`, - { cause: error } - ) + { cause: error }, + ); } } function readRemoteRuntimeLock(projectRoot = root) { const lock = readJson( - join(projectRoot, 'remote-runtime-lock.json'), - 'Remote Runtime lock' - ) - const runtime = lock?.runtimes?.opencode + join(projectRoot, "remote-runtime-lock.json"), + "Remote Runtime lock", + ); + const runtime = lock?.runtimes?.opencode; if ( lock?.formatVersion !== 1 || - runtime?.version !== '1.18.29' || - runtime.provider !== 'opencode' || - runtime.entrypoint !== 'bin/opencode' || - runtime.entrypointIdentity !== 'opencode-acp' || - JSON.stringify(runtime.argvPrefix) !== JSON.stringify(['acp']) || + runtime?.version !== "1.18.29" || + runtime.provider !== "opencode" || + runtime.entrypoint !== "bin/opencode" || + runtime.entrypointIdentity !== "opencode-acp" || + JSON.stringify(runtime.argvPrefix) !== JSON.stringify(["acp"]) || !Number.isSafeInteger(runtime.protocol?.major) || !Number.isSafeInteger(runtime.protocol?.minor) ) { - throw new Error('Remote Runtime lock contract is invalid') + throw new Error("Remote Runtime lock contract is invalid"); } for (const architecture of supportedArchitectures) { - const target = runtime.targets?.[architecture] + const target = runtime.targets?.[architecture]; if ( - typeof target?.package !== 'string' || - typeof target.integrity !== 'string' || + typeof target?.package !== "string" || + typeof target.integrity !== "string" || !/^sha512-[A-Za-z0-9+/]{86}==$/u.test(target.integrity) ) { - throw new Error( - `Remote Runtime lock target is invalid: ${architecture}` - ) + throw new Error(`Remote Runtime lock target is invalid: ${architecture}`); } } - return lock + return lock; } function readTrustedKeyRegistry(projectRoot = root) { const registry = readJson( - join(projectRoot, 'resources', 'agent-release-keys.json'), - 'Runtime trusted key registry' - ) + join(projectRoot, "resources", "agent-release-keys.json"), + "Runtime trusted key registry", + ); if ( registry?.formatVersion !== 1 || !Array.isArray(registry.keys) || !Array.isArray(registry.revocations) ) { - throw new Error('Runtime trusted key registry contract is invalid') + throw new Error("Runtime trusted key registry contract is invalid"); } - const keyIds = new Set() + const keyIds = new Set(); for (const key of registry.keys) { if ( - typeof key?.keyId !== 'string' || + typeof key?.keyId !== "string" || !/^[a-z0-9][a-z0-9._-]{0,63}$/u.test(key.keyId) || keyIds.has(key.keyId) || - typeof key.publicKeySpkiBase64 !== 'string' || - !['production', 'test'].includes(key.environment) + typeof key.publicKeySpkiBase64 !== "string" || + !["production", "test"].includes(key.environment) ) { - throw new Error('Runtime trusted key registry entry is invalid') + throw new Error("Runtime trusted key registry entry is invalid"); } const publicKey = createPublicKey({ - key: Buffer.from(key.publicKeySpkiBase64, 'base64'), - format: 'der', - type: 'spki' - }) - if (publicKey.asymmetricKeyType !== 'ed25519') { - throw new Error(`Runtime trusted key is not Ed25519: ${key.keyId}`) + key: Buffer.from(key.publicKeySpkiBase64, "base64"), + format: "der", + type: "spki", + }); + if (publicKey.asymmetricKeyType !== "ed25519") { + throw new Error(`Runtime trusted key is not Ed25519: ${key.keyId}`); } - keyIds.add(key.keyId) + keyIds.add(key.keyId); } - return registry + return registry; } function sha256(contents) { - return createHash('sha256').update(contents).digest('hex') + return createHash("sha256").update(contents).digest("hex"); } function sha512FileIntegrity(filePath) { - const hash = createHash('sha512') - const descriptor = openSync(filePath, 'r') - const chunk = Buffer.allocUnsafe(1024 * 1024) + const hash = createHash("sha512"); + const descriptor = openSync(filePath, "r"); + const chunk = Buffer.allocUnsafe(1024 * 1024); try { while (true) { - const bytesRead = readSync( - descriptor, - chunk, - 0, - chunk.byteLength, - null - ) + const bytesRead = readSync(descriptor, chunk, 0, chunk.byteLength, null); if (bytesRead === 0) { - break + break; } - hash.update(chunk.subarray(0, bytesRead)) + hash.update(chunk.subarray(0, bytesRead)); } } finally { - closeSync(descriptor) + closeSync(descriptor); } - return `sha512-${hash.digest('base64')}` + return `sha512-${hash.digest("base64")}`; } function sha256FileSync(filePath) { - return sha256(readFileSync(filePath)) + return sha256(readFileSync(filePath)); } function canonicalJson(value) { if ( value === null || - typeof value === 'boolean' || - typeof value === 'string' + typeof value === "boolean" || + typeof value === "string" ) { - return JSON.stringify(value) + return JSON.stringify(value); } - if (typeof value === 'number') { + if (typeof value === "number") { if (!Number.isFinite(value)) { - throw new Error('Canonical Runtime value is not finite') + throw new Error("Canonical Runtime value is not finite"); } - return JSON.stringify(Object.is(value, -0) ? 0 : value) + return JSON.stringify(Object.is(value, -0) ? 0 : value); } if (Array.isArray(value)) { - return `[${value.map(canonicalJson).join(',')}]` + return `[${value.map(canonicalJson).join(",")}]`; } if ( - typeof value !== 'object' || + typeof value !== "object" || Object.getPrototypeOf(value) !== Object.prototype ) { - throw new Error('Canonical Runtime value is not plain JSON') + throw new Error("Canonical Runtime value is not plain JSON"); } return `{${Object.keys(value) .sort() - .map( - (key) => - `${JSON.stringify(key)}:${canonicalJson(value[key])}` - ) - .join(',')}}` + .map((key) => `${JSON.stringify(key)}:${canonicalJson(value[key])}`) + .join(",")}}`; } function digestCanonicalOperation({ method, scope, payload }) { return `sha256:${sha256( - Buffer.from(canonicalJson({ method, payload, scope }), 'utf8') - )}` + Buffer.from(canonicalJson({ method, payload, scope }), "utf8"), + )}`; } function digestRuntimeBundleIdentity(manifest) { - const { bundleDigest: _bundleDigest, ...identity } = manifest - void _bundleDigest + const { bundleDigest: _bundleDigest, ...identity } = manifest; + void _bundleDigest; return digestCanonicalOperation({ - method: 'runtime/bundleIdentity', + method: "runtime/bundleIdentity", scope: { - kind: 'installation', - installationId: manifest.runtimeId + kind: "installation", + installationId: manifest.runtimeId, }, - payload: identity - }) + payload: identity, + }); } function digestRuntimeBundleManifest(manifest) { return digestCanonicalOperation({ - method: 'runtime/bundleManifest', + method: "runtime/bundleManifest", scope: { - kind: 'installation', - installationId: manifest.runtimeId + kind: "installation", + installationId: manifest.runtimeId, }, - payload: manifest - }) + payload: manifest, + }); } function canonicalManifestBytes(manifest) { - return Buffer.from(`${JSON.stringify(manifest, null, 2)}\n`, 'utf8') + return Buffer.from(`${JSON.stringify(manifest, null, 2)}\n`, "utf8"); } function signaturePayload(manifestBytes) { - return Buffer.concat([signatureDomain, manifestBytes]) + return Buffer.concat([signatureDomain, manifestBytes]); } function publicKeySpkiBase64(key) { return createPublicKey(key) - .export({ format: 'der', type: 'spki' }) - .toString('base64') + .export({ format: "der", type: "spki" }) + .toString("base64"); } function productionSigningIdentity(environment = process.env) { - const keyId = environment[signingKeyIdEnvironment] - const privateKey = environment[signingPrivateKeyEnvironment] + const keyId = environment[signingKeyIdEnvironment]; + const privateKey = environment[signingPrivateKeyEnvironment]; if (!keyId || !privateKey) { throw new Error( - `${signingKeyIdEnvironment} and ${signingPrivateKeyEnvironment} are required to sign the bundled Runtime` - ) + `${signingKeyIdEnvironment} and ${signingPrivateKeyEnvironment} are required to sign the bundled Runtime`, + ); } - return { keyId, privateKey } + return { keyId, privateKey }; } function trustedKeyForSigning( signingIdentity, registry, - verificationEnvironment + verificationEnvironment, ) { const record = registry.keys.find( - (key) => key.keyId === signingIdentity.keyId - ) + (key) => key.keyId === signingIdentity.keyId, + ); if (!record) { throw new Error( - `Runtime signing key is not registered: ${signingIdentity.keyId}` - ) + `Runtime signing key is not registered: ${signingIdentity.keyId}`, + ); } if ( - verificationEnvironment === 'production' && - record.environment !== 'production' + verificationEnvironment === "production" && + record.environment !== "production" ) { throw new Error( - `Runtime signing key is not registered for production: ${record.keyId}` - ) + `Runtime signing key is not registered for production: ${record.keyId}`, + ); } if ( - registry.revocations.some( - (revocation) => revocation.keyId === record.keyId - ) + registry.revocations.some((revocation) => revocation.keyId === record.keyId) ) { - throw new Error('Runtime signing key is outside its trusted interval') + throw new Error("Runtime signing key is outside its trusted interval"); } if ( publicKeySpkiBase64(signingIdentity.privateKey) !== record.publicKeySpkiBase64 ) { throw new Error( - 'Runtime signing private key does not match the trusted registry' - ) + "Runtime signing private key does not match the trusted registry", + ); } - return record + return record; } function preflightProductionSigningKey(options = {}) { - const projectRoot = options.projectRoot ?? root - const environment = options.environment ?? process.env - const keyId = - options.keyId ?? environment[signingKeyIdEnvironment] - const registry = - options.registry ?? readTrustedKeyRegistry(projectRoot) + const projectRoot = options.projectRoot ?? root; + const environment = options.environment ?? process.env; + const keyId = options.keyId ?? environment[signingKeyIdEnvironment]; + const registry = options.registry ?? readTrustedKeyRegistry(projectRoot); return preflightRegisteredProductionKey({ - component: 'GoodBuddy', + component: "GoodBuddy", keyId, registry, - missingKeyIdMessage: - `${signingKeyIdEnvironment} is not configured; provision the production GoodBuddy signing key ID before building a release` - }) + missingKeyIdMessage: `${signingKeyIdEnvironment} is not configured; provision the production GoodBuddy signing key ID before building a release`, + }); } function assertSafeManifestPath(filePath) { if ( - typeof filePath !== 'string' || + typeof filePath !== "string" || filePath.length === 0 || filePath.length > 240 || - filePath.includes('\\') || - filePath.includes('\0') || + filePath.includes("\\") || + filePath.includes("\0") || posix.isAbsolute(filePath) || filePath - .split('/') - .some( - (part) => part.length === 0 || part === '.' || part === '..' - ) + .split("/") + .some((part) => part.length === 0 || part === "." || part === "..") ) { - throw new Error(`Unsafe Runtime manifest path: ${String(filePath)}`) + throw new Error(`Unsafe Runtime manifest path: ${String(filePath)}`); } } function compareUtf8(left, right) { - return Buffer.compare( - Buffer.from(left, 'utf8'), - Buffer.from(right, 'utf8') - ) + return Buffer.compare(Buffer.from(left, "utf8"), Buffer.from(right, "utf8")); } function listFiles(directory) { - const files = [] - const pending = [directory] + const files = []; + const pending = [directory]; while (pending.length > 0) { - const current = pending.pop() + const current = pending.pop(); const entries = readdirSync(current, { - withFileTypes: true - }).sort((left, right) => compareUtf8(right.name, left.name)) + withFileTypes: true, + }).sort((left, right) => compareUtf8(right.name, left.name)); for (const entry of entries) { - const absolutePath = join(current, entry.name) + const absolutePath = join(current, entry.name); if (entry.isSymbolicLink()) { throw new Error( - `Runtime bundle cannot contain a symlink: ${absolutePath}` - ) + `Runtime bundle cannot contain a symlink: ${absolutePath}`, + ); } if (entry.isDirectory()) { - pending.push(absolutePath) + pending.push(absolutePath); } else if (entry.isFile()) { - files.push(absolutePath) + files.push(absolutePath); } else { throw new Error( - `Runtime bundle contains an unsupported file: ${absolutePath}` - ) + `Runtime bundle contains an unsupported file: ${absolutePath}`, + ); } } } - return files.sort(compareUtf8) + return files.sort(compareUtf8); } function posixRelative(directory, filePath) { - return relative(directory, filePath).split(sep).join('/') + return relative(directory, filePath).split(sep).join("/"); } function modeString(filePath) { - return (statSync(filePath).mode & 0o777) - .toString(8) - .padStart(4, '0') + return (statSync(filePath).mode & 0o777).toString(8).padStart(4, "0"); } -function assertElfArchitecture(filePath, expectedArchitecture, platform = 'linux') { - assertTargetBinary(filePath, expectedArchitecture, platform, 'Runtime') +function assertElfArchitecture( + filePath, + expectedArchitecture, + platform = "linux", +) { + assertTargetBinary(filePath, expectedArchitecture, platform, "Runtime"); } function validateManifestShape(manifest) { if ( manifest?.formatVersion !== 2 || - manifest.product !== 'GoodBuddy' || - manifest.runtimeId !== 'opencode' || - manifest.runtimeVersion !== '1.18.29' || - manifest.provider !== 'opencode' || + manifest.product !== "GoodBuddy" || + manifest.runtimeId !== "opencode" || + manifest.runtimeVersion !== "1.18.29" || + manifest.provider !== "opencode" || !( - manifest.platform === 'linux' || - (manifest.platform === 'darwin' && manifest.architecture === 'arm64') + manifest.platform === "linux" || + (manifest.platform === "darwin" && manifest.architecture === "arm64") ) || !supportedArchitectures.includes(manifest.architecture) || - typeof manifest.signingKeyId !== 'string' || + typeof manifest.signingKeyId !== "string" || !/^sha256:[a-f0-9]{64}$/u.test(manifest.bundleDigest) || !/^sha256:[a-f0-9]{64}$/u.test(manifest.adapterDigest) || - !/^sha256:[a-f0-9]{64}$/u.test( - manifest.acpCapabilitiesDigest - ) || + !/^sha256:[a-f0-9]{64}$/u.test(manifest.acpCapabilitiesDigest) || !Array.isArray(manifest.files) || !Array.isArray(manifest.licenses) || !Array.isArray(manifest.allowedEnvironmentNames) ) { - throw new Error('Runtime manifest contract is invalid') + throw new Error("Runtime manifest contract is invalid"); } if ( - !Number.isSafeInteger( - manifest.limits?.maximumPromptRuntimeMilliseconds - ) || + !Number.isSafeInteger(manifest.limits?.maximumPromptRuntimeMilliseconds) || manifest.limits.maximumPromptRuntimeMilliseconds < 1 || !Number.isSafeInteger(manifest.limits.maximumPromptInputBytes) || manifest.limits.maximumPromptInputBytes < 1 || !Number.isSafeInteger(manifest.limits.maximumPromptOutputBytes) || manifest.limits.maximumPromptOutputBytes < 0 || Object.keys(manifest.limits).length !== 3 || - 'quotas' in manifest + "quotas" in manifest ) { - throw new Error('Runtime manifest limits contract is invalid') + throw new Error("Runtime manifest limits contract is invalid"); } if ( - manifest.entrypoint?.identity !== 'opencode-acp' || - manifest.entrypoint.path !== 'bin/opencode' || + manifest.entrypoint?.identity !== "opencode-acp" || + manifest.entrypoint.path !== "bin/opencode" || !/^[a-f0-9]{64}$/u.test(manifest.entrypoint.sha256) || - JSON.stringify(manifest.entrypoint.argvPrefix) !== - JSON.stringify(['acp']) + JSON.stringify(manifest.entrypoint.argvPrefix) !== JSON.stringify(["acp"]) ) { - throw new Error('Runtime manifest entrypoint is invalid') + throw new Error("Runtime manifest entrypoint is invalid"); } } -function trustedKeyForManifest( - manifest, - registry, - verificationEnvironment -) { +function trustedKeyForManifest(manifest, registry, verificationEnvironment) { const key = registry.keys.find( - (candidate) => candidate.keyId === manifest.signingKeyId - ) + (candidate) => candidate.keyId === manifest.signingKeyId, + ); if (!key) { throw new Error( - `Runtime manifest uses an unknown signing key: ${manifest.signingKeyId}` - ) + `Runtime manifest uses an unknown signing key: ${manifest.signingKeyId}`, + ); } if ( - verificationEnvironment === 'production' && - key.environment !== 'production' + verificationEnvironment === "production" && + key.environment !== "production" ) { throw new Error( - `Production Runtime verification rejects non-production key: ${key.keyId}` - ) + `Production Runtime verification rejects non-production key: ${key.keyId}`, + ); } if ( - registry.revocations.some( - (revocation) => revocation.keyId === key.keyId - ) + registry.revocations.some((revocation) => revocation.keyId === key.keyId) ) { - throw new Error('Runtime signing key is outside its trusted interval') + throw new Error("Runtime signing key is outside its trusted interval"); } - return key + return key; } -function assertManifestMatchesLock(manifest, lock, architecture, platform = 'linux') { - const expected = lock.runtimes.opencode - const target = expected.targets[platform === 'linux' ? architecture : targetName(architecture, platform)] +function assertManifestMatchesLock( + manifest, + lock, + architecture, + platform = "linux", +) { + const expected = lock.runtimes.opencode; + const target = + expected.targets[ + platform === "linux" ? architecture : targetName(architecture, platform) + ]; if ( - manifest.runtimeId !== 'opencode' || + manifest.runtimeId !== "opencode" || manifest.runtimeVersion !== expected.version || manifest.provider !== expected.provider || manifest.platform !== platform || @@ -491,296 +460,268 @@ function assertManifestMatchesLock(manifest, lock, architecture, platform = 'lin manifest.acpCapabilitiesDigest !== profileDigests.acp ) { throw new Error( - 'Runtime manifest does not match the locked OpenCode profile' - ) + "Runtime manifest does not match the locked OpenCode profile", + ); } } function verifyBundleDirectory(bundleDirectoryInput, options = {}) { - const bundleDirectory = resolve(bundleDirectoryInput) - const projectRoot = options.projectRoot ?? root - const lock = options.lock ?? readRemoteRuntimeLock(projectRoot) - const registry = - options.registry ?? readTrustedKeyRegistry(projectRoot) - const architecture = options.architecture - const platform = options.platform ?? 'linux' - targetName(architecture, platform) - const manifestBytes = readFileSync( - join(bundleDirectory, manifestFileName) - ) + const bundleDirectory = resolve(bundleDirectoryInput); + const projectRoot = options.projectRoot ?? root; + const lock = options.lock ?? readRemoteRuntimeLock(projectRoot); + const registry = options.registry ?? readTrustedKeyRegistry(projectRoot); + const architecture = options.architecture; + const platform = options.platform ?? "linux"; + targetName(architecture, platform); + const manifestBytes = readFileSync(join(bundleDirectory, manifestFileName)); const signatureText = readFileSync( join(bundleDirectory, signatureFileName), - 'utf8' - ) + "utf8", + ); if (!/^[A-Za-z0-9+/]{86}==\n$/u.test(signatureText)) { - throw new Error('Runtime detached signature encoding is invalid') + throw new Error("Runtime detached signature encoding is invalid"); } - const manifest = JSON.parse(manifestBytes.toString('utf8')) - validateManifestShape(manifest) + const manifest = JSON.parse(manifestBytes.toString("utf8")); + validateManifestShape(manifest); if (!canonicalManifestBytes(manifest).equals(manifestBytes)) { - throw new Error( - 'Runtime manifest is not in canonical deterministic form' - ) + throw new Error("Runtime manifest is not in canonical deterministic form"); } - assertManifestMatchesLock(manifest, lock, architecture, platform) - const environment = - options.verificationEnvironment ?? 'production' - const key = trustedKeyForManifest( - manifest, - registry, - environment - ) + assertManifestMatchesLock(manifest, lock, architecture, platform); + const environment = options.verificationEnvironment ?? "production"; + const key = trustedKeyForManifest(manifest, registry, environment); const publicKey = createPublicKey({ - key: Buffer.from(key.publicKeySpkiBase64, 'base64'), - format: 'der', - type: 'spki' - }) - const signatureBytes = Buffer.from(signatureText.trim(), 'base64') + key: Buffer.from(key.publicKeySpkiBase64, "base64"), + format: "der", + type: "spki", + }); + const signatureBytes = Buffer.from(signatureText.trim(), "base64"); if ( signatureBytes.length !== 64 || - !verify( - null, - signaturePayload(manifestBytes), - publicKey, - signatureBytes - ) + !verify(null, signaturePayload(manifestBytes), publicKey, signatureBytes) ) { - throw new Error('Runtime manifest signature verification failed') + throw new Error("Runtime manifest signature verification failed"); } if (digestRuntimeBundleIdentity(manifest) !== manifest.bundleDigest) { - throw new Error('Runtime bundle identity digest does not match') + throw new Error("Runtime bundle identity digest does not match"); } if ( options.requireDigestDirectory !== false && - basename(bundleDirectory) !== - manifest.bundleDigest.slice('sha256:'.length) + basename(bundleDirectory) !== manifest.bundleDigest.slice("sha256:".length) ) { throw new Error( - 'Runtime bundle digest does not match its managed directory' - ) + "Runtime bundle digest does not match its managed directory", + ); } - const declared = new Set() + const declared = new Set(); for (const file of manifest.files) { - assertSafeManifestPath(file.path) + assertSafeManifestPath(file.path); if ( declared.has(file.path) || !Number.isSafeInteger(file.size) || file.size < 0 || !/^[a-f0-9]{64}$/u.test(file.sha256) || - !['0644', '0755'].includes(file.mode) + !["0644", "0755"].includes(file.mode) ) { - throw new Error(`Runtime manifest file entry is invalid: ${file.path}`) + throw new Error(`Runtime manifest file entry is invalid: ${file.path}`); } - declared.add(file.path) - const filePath = join(bundleDirectory, ...file.path.split('/')) - const stat = lstatSync(filePath) + declared.add(file.path); + const filePath = join(bundleDirectory, ...file.path.split("/")); + const stat = lstatSync(filePath); if (!stat.isFile() || stat.isSymbolicLink()) { - throw new Error(`Runtime payload is not a regular file: ${file.path}`) + throw new Error(`Runtime payload is not a regular file: ${file.path}`); } if (stat.size !== file.size) { - throw new Error(`Runtime payload size mismatch: ${file.path}`) + throw new Error(`Runtime payload size mismatch: ${file.path}`); } if (sha256FileSync(filePath) !== file.sha256) { - throw new Error(`Runtime payload hash mismatch: ${file.path}`) + throw new Error(`Runtime payload hash mismatch: ${file.path}`); } if ( options.enforceFilesystemMode !== false && - process.platform !== 'win32' && + process.platform !== "win32" && modeString(filePath) !== file.mode ) { - throw new Error(`Runtime payload mode mismatch: ${file.path}`) + throw new Error(`Runtime payload mode mismatch: ${file.path}`); } } const actual = new Set( listFiles(bundleDirectory).map((filePath) => - posixRelative(bundleDirectory, filePath) - ) - ) - const expected = new Set([ - ...declared, - manifestFileName, - signatureFileName - ]) + posixRelative(bundleDirectory, filePath), + ), + ); + const expected = new Set([...declared, manifestFileName, signatureFileName]); if ( actual.size !== expected.size || [...actual].some((filePath) => !expected.has(filePath)) ) { - throw new Error( - 'Runtime bundle contains undeclared or missing files' - ) + throw new Error("Runtime bundle contains undeclared or missing files"); } for (const license of manifest.licenses) { - assertSafeManifestPath(license.path) + assertSafeManifestPath(license.path); if (!declared.has(license.path)) { - throw new Error('Runtime manifest license is not declared') + throw new Error("Runtime manifest license is not declared"); } } if ( !declared.has(manifest.entrypoint.path) || manifest.entrypoint.sha256 !== - manifest.files.find( - (file) => file.path === manifest.entrypoint.path - )?.sha256 + manifest.files.find((file) => file.path === manifest.entrypoint.path) + ?.sha256 ) { - throw new Error('Runtime entrypoint payload is missing') + throw new Error("Runtime entrypoint payload is missing"); } assertElfArchitecture( - join(bundleDirectory, ...manifest.entrypoint.path.split('/')), + join(bundleDirectory, ...manifest.entrypoint.path.split("/")), architecture, - platform - ) + platform, + ); return { bundleDirectory, manifest, - manifestDigest: digestRuntimeBundleManifest(manifest) - } + manifestDigest: digestRuntimeBundleManifest(manifest), + }; } -function lockedArchive(lock, architecture, archivePath, platform = 'linux') { - const input = resolveLockedRuntimeInput(lock, architecture, platform) +function lockedArchive(lock, architecture, archivePath, platform = "linux") { + const input = resolveLockedRuntimeInput(lock, architecture, platform); if (!archivePath || !existsSync(archivePath)) { - throw new Error( - `Locked OpenCode archive is required: ${input.archive}` - ) + throw new Error(`Locked OpenCode archive is required: ${input.archive}`); } if (!statSync(archivePath).isFile()) { - throw new Error(`Runtime archive is not a file: ${archivePath}`) + throw new Error(`Runtime archive is not a file: ${archivePath}`); } if (basename(archivePath) !== input.archive) { throw new Error( - `Runtime archive filename mismatch: expected ${input.archive}` - ) + `Runtime archive filename mismatch: expected ${input.archive}`, + ); } - const actualIntegrity = sha512FileIntegrity(archivePath) + const actualIntegrity = sha512FileIntegrity(archivePath); if (actualIntegrity !== input.integrity) { throw new Error( - `Runtime archive integrity mismatch for ${targetName(architecture)}` - ) + `Runtime archive integrity mismatch for ${targetName(architecture)}`, + ); } - return lock.runtimes.opencode.targets[platform === 'linux' ? architecture : targetName(architecture, platform)] + return lock.runtimes.opencode.targets[ + platform === "linux" ? architecture : targetName(architecture, platform) + ]; } -function resolveLockedRuntimeInput(lock, architecture, platform = 'linux') { - targetName(architecture, platform) - const runtime = lock.runtimes.opencode - const target = runtime.targets[platform === 'linux' ? architecture : targetName(architecture, platform)] +function resolveLockedRuntimeInput(lock, architecture, platform = "linux") { + targetName(architecture, platform); + const runtime = lock.runtimes.opencode; + const target = + runtime.targets[ + platform === "linux" ? architecture : targetName(architecture, platform) + ]; return { packageName: target.package, version: runtime.version, integrity: target.integrity, - archive: `${target.package}-${runtime.version}.tgz` - } + archive: `${target.package}-${runtime.version}.tgz`, + }; } function createManifest(bundleDirectory, metadata) { const files = listFiles(bundleDirectory) .filter((filePath) => { - const name = posixRelative(bundleDirectory, filePath) - return name !== manifestFileName && name !== signatureFileName + const name = posixRelative(bundleDirectory, filePath); + return name !== manifestFileName && name !== signatureFileName; }) .map((filePath) => { - const path = posixRelative(bundleDirectory, filePath) - assertSafeManifestPath(path) + const path = posixRelative(bundleDirectory, filePath); + assertSafeManifestPath(path); return { path, size: statSync(filePath).size, sha256: sha256FileSync(filePath), - mode: path === 'bin/opencode' ? '0755' : '0644' - } - }) - const entrypoint = files.find( - (file) => file.path === 'bin/opencode' - ) + mode: path === "bin/opencode" ? "0755" : "0644", + }; + }); + const entrypoint = files.find((file) => file.path === "bin/opencode"); if (!entrypoint) { - throw new Error('Runtime entrypoint is missing') + throw new Error("Runtime entrypoint is missing"); } const initial = { formatVersion: 2, - product: 'GoodBuddy', - runtimeId: 'opencode', + product: "GoodBuddy", + runtimeId: "opencode", runtimeVersion: metadata.runtime.version, provider: metadata.runtime.provider, - platform: metadata.platform ?? 'linux', + platform: metadata.platform ?? "linux", architecture: metadata.architecture, signingKeyId: metadata.signingKeyId, - bundleDigest: `sha256:${'0'.repeat(64)}`, + bundleDigest: `sha256:${"0".repeat(64)}`, adapterDigest: profileDigests.adapter, sourcePackage: metadata.sourcePackage, entrypoint: { identity: metadata.runtime.entrypointIdentity, path: metadata.runtime.entrypoint, sha256: entrypoint.sha256, - argvPrefix: metadata.runtime.argvPrefix + argvPrefix: metadata.runtime.argvPrefix, }, files, licenses: [ { - package: 'opencode-ai', + package: "opencode-ai", version: metadata.runtime.version, - spdx: 'MIT', - path: 'licenses/opencode-MIT.txt' - } + spdx: "MIT", + path: "licenses/opencode-MIT.txt", + }, ], - allowedEnvironmentNames: - metadata.runtime.allowedEnvironmentNames, + allowedEnvironmentNames: metadata.runtime.allowedEnvironmentNames, protocol: metadata.runtime.protocol, acpCapabilitiesDigest: profileDigests.acp, - limits: fixedLimits - } + limits: fixedLimits, + }; return { ...initial, - bundleDigest: digestRuntimeBundleIdentity(initial) - } + bundleDigest: digestRuntimeBundleIdentity(initial), + }; } function defaultOutputRoot(projectRoot = root) { - return join(projectRoot, '.remote-runtime-resources') + return join(projectRoot, ".remote-runtime-resources"); } -function bundleDirectory(outputRoot, architecture, digest, platform = 'linux') { +function bundleDirectory(outputRoot, architecture, digest, platform = "linux") { return join( outputRoot, targetName(architecture, platform), - 'opencode', - digest.slice('sha256:'.length) - ) + "opencode", + digest.slice("sha256:".length), + ); } function buildRuntimeBundle(options) { - const projectRoot = options.projectRoot ?? root - const architecture = options.architecture - const platform = options.platform ?? 'linux' - const lock = options.lock ?? readRemoteRuntimeLock(projectRoot) - const runtime = lock.runtimes.opencode + const projectRoot = options.projectRoot ?? root; + const architecture = options.architecture; + const platform = options.platform ?? "linux"; + const lock = options.lock ?? readRemoteRuntimeLock(projectRoot); + const runtime = lock.runtimes.opencode; const sourcePackage = lockedArchive( lock, architecture, options.runtimeArchive, - platform - ) - const registry = - options.registry ?? readTrustedKeyRegistry(projectRoot) + platform, + ); + const registry = options.registry ?? readTrustedKeyRegistry(projectRoot); const signingIdentity = options.testSigningIdentity ?? options.signingIdentity ?? - productionSigningIdentity(options.environment ?? process.env) - const verificationEnvironment = - options.testSigningIdentity ? 'test' : 'production' - trustedKeyForSigning( - signingIdentity, - registry, - verificationEnvironment - ) - const outputRoot = - options.outputRoot ?? defaultOutputRoot(projectRoot) - mkdirSync(outputRoot, { recursive: true }) + productionSigningIdentity(options.environment ?? process.env); + const verificationEnvironment = options.testSigningIdentity + ? "test" + : "production"; + trustedKeyForSigning(signingIdentity, registry, verificationEnvironment); + const outputRoot = options.outputRoot ?? defaultOutputRoot(projectRoot); + mkdirSync(outputRoot, { recursive: true }); const staging = mkdtempSync( - join(outputRoot, `.stage-${targetName(architecture)}-`) - ) + join(outputRoot, `.stage-${targetName(architecture)}-`), + ); const extracted = mkdtempSync( - join(tmpdir(), `goodbuddy-opencode-${targetName(architecture)}-`) - ) + join(tmpdir(), `goodbuddy-opencode-${targetName(architecture)}-`), + ); try { tar.x({ file: options.runtimeArchive, @@ -789,87 +730,82 @@ function buildRuntimeBundle(options) { strict: true, filter(path) { return ( - path === 'package/package.json' || - path === 'package/bin/opencode' - ) - } - }) + path === "package/package.json" || path === "package/bin/opencode" + ); + }, + }); const sourceMetadata = readJson( - join(extracted, 'package', 'package.json'), - 'OpenCode source package metadata' - ) + join(extracted, "package", "package.json"), + "OpenCode source package metadata", + ); if ( sourceMetadata.name !== sourcePackage.package && sourceMetadata.name !== sourcePackage.name ) { // The lock target itself carries the canonical package name. - if ( - sourceMetadata.name !== - sourcePackage.package - ) { - throw new Error('OpenCode source package name does not match the lock') + if (sourceMetadata.name !== sourcePackage.package) { + throw new Error("OpenCode source package name does not match the lock"); } } if (sourceMetadata.version !== runtime.version) { throw new Error( - 'OpenCode source package version does not match the lock' - ) + "OpenCode source package version does not match the lock", + ); } - mkdirSync(join(staging, 'bin'), { mode: 0o700 }) + mkdirSync(join(staging, "bin"), { mode: 0o700 }); copyFileSync( - join(extracted, 'package', 'bin', 'opencode'), - join(staging, 'bin', 'opencode') - ) - chmodSync(join(staging, 'bin', 'opencode'), 0o755) + join(extracted, "package", "bin", "opencode"), + join(staging, "bin", "opencode"), + ); + chmodSync(join(staging, "bin", "opencode"), 0o755); assertElfArchitecture( - join(staging, 'bin', 'opencode'), + join(staging, "bin", "opencode"), architecture, - platform - ) - mkdirSync(join(staging, 'licenses'), { mode: 0o700 }) + platform, + ); + const configTemplate = prepareBundledOpenCodeConfig(projectRoot); + cpSync(configTemplate, join(staging, "config", "opencode"), { + recursive: true, + }); + mkdirSync(join(staging, "licenses"), { mode: 0o700 }); copyFileSync( - join(projectRoot, 'node_modules', 'opencode-ai', 'LICENSE'), - join(staging, 'licenses', 'opencode-MIT.txt') - ) - chmodSync( - join(staging, 'licenses', 'opencode-MIT.txt'), - 0o644 - ) + join(projectRoot, "node_modules", "opencode-ai", "LICENSE"), + join(staging, "licenses", "opencode-MIT.txt"), + ); + chmodSync(join(staging, "licenses", "opencode-MIT.txt"), 0o644); const manifest = createManifest(staging, { platform, architecture, runtime, sourcePackage: { name: sourcePackage.package, - integrity: sourcePackage.integrity + integrity: sourcePackage.integrity, }, - signingKeyId: signingIdentity.keyId - }) - const manifestBytes = canonicalManifestBytes(manifest) + signingKeyId: signingIdentity.keyId, + }); + const manifestBytes = canonicalManifestBytes(manifest); const signatureBytes = sign( null, signaturePayload(manifestBytes), - signingIdentity.privateKey - ) - writeFileSync( - join(staging, manifestFileName), - manifestBytes, - { mode: 0o644 } - ) + signingIdentity.privateKey, + ); + writeFileSync(join(staging, manifestFileName), manifestBytes, { + mode: 0o644, + }); writeFileSync( join(staging, signatureFileName), - `${signatureBytes.toString('base64')}\n`, - { mode: 0o644 } - ) - chmodSync(join(staging, manifestFileName), 0o644) - chmodSync(join(staging, signatureFileName), 0o644) + `${signatureBytes.toString("base64")}\n`, + { mode: 0o644 }, + ); + chmodSync(join(staging, manifestFileName), 0o644); + chmodSync(join(staging, signatureFileName), 0o644); const destination = bundleDirectory( outputRoot, architecture, manifest.bundleDigest, - platform - ) - mkdirSync(dirname(destination), { recursive: true, mode: 0o700 }) + platform, + ); + mkdirSync(dirname(destination), { recursive: true, mode: 0o700 }); if (existsSync(destination)) { const existing = verifyBundleDirectory(destination, { platform, @@ -878,19 +814,17 @@ function buildRuntimeBundle(options) { lock, registry, verificationEnvironment, - enforceFilesystemMode: options.enforceFilesystemMode - }) - if ( - canonicalJson(existing.manifest) !== canonicalJson(manifest) - ) { + enforceFilesystemMode: options.enforceFilesystemMode, + }); + if (canonicalJson(existing.manifest) !== canonicalJson(manifest)) { throw new Error( - 'Existing Runtime digest directory has different content' - ) + "Existing Runtime digest directory has different content", + ); } - return existing + return existing; } - chmodSync(staging, 0o700) - renameSync(staging, destination) + chmodSync(staging, 0o700); + renameSync(staging, destination); return verifyBundleDirectory(destination, { platform, projectRoot, @@ -898,21 +832,21 @@ function buildRuntimeBundle(options) { lock, registry, verificationEnvironment, - enforceFilesystemMode: options.enforceFilesystemMode - }) + enforceFilesystemMode: options.enforceFilesystemMode, + }); } finally { - rmSync(staging, { recursive: true, force: true }) - rmSync(extracted, { recursive: true, force: true }) + rmSync(staging, { recursive: true, force: true }); + rmSync(extracted, { recursive: true, force: true }); } } function createRuntimeArchive(bundleDirectoryInput, archivePath) { - const directory = resolve(bundleDirectoryInput) + const directory = resolve(bundleDirectoryInput); const entries = listFiles(directory) .map((filePath) => posixRelative(directory, filePath)) - .sort(compareUtf8) - mkdirSync(dirname(archivePath), { recursive: true }) - rmSync(archivePath, { force: true }) + .sort(compareUtf8); + mkdirSync(dirname(archivePath), { recursive: true }); + rmSync(archivePath, { force: true }); tar.c( { cwd: directory, @@ -920,183 +854,176 @@ function createRuntimeArchive(bundleDirectoryInput, archivePath) { sync: true, portable: true, noPax: true, - mtime: new Date(0) + mtime: new Date(0), }, - entries - ) + entries, + ); } function importRuntimeArchive(archivePath, options) { if (!existsSync(archivePath)) { - throw new Error(`Runtime artifact archive is missing: ${archivePath}`) + throw new Error(`Runtime artifact archive is missing: ${archivePath}`); } - const projectRoot = options.projectRoot ?? root - const architecture = options.architecture - const outputRoot = - options.outputRoot ?? defaultOutputRoot(projectRoot) - mkdirSync(outputRoot, { recursive: true }) + const projectRoot = options.projectRoot ?? root; + const architecture = options.architecture; + const outputRoot = options.outputRoot ?? defaultOutputRoot(projectRoot); + mkdirSync(outputRoot, { recursive: true }); const staging = mkdtempSync( - join(outputRoot, `.import-${targetName(architecture)}-`) - ) + join(outputRoot, `.import-${targetName(architecture)}-`), + ); try { tar.x({ cwd: staging, file: archivePath, sync: true, strict: true, - preservePaths: false - }) + preservePaths: false, + }); const verified = verifyBundleDirectory(staging, { ...options, projectRoot, architecture, - requireDigestDirectory: false - }) + requireDigestDirectory: false, + }); const destination = bundleDirectory( outputRoot, architecture, - verified.manifest.bundleDigest - ) - mkdirSync(dirname(destination), { recursive: true, mode: 0o700 }) + verified.manifest.bundleDigest, + ); + mkdirSync(dirname(destination), { recursive: true, mode: 0o700 }); if (existsSync(destination)) { return verifyBundleDirectory(destination, { ...options, projectRoot, - architecture - }) + architecture, + }); } - chmodSync(staging, 0o700) - renameSync(staging, destination) + chmodSync(staging, 0o700); + renameSync(staging, destination); return verifyBundleDirectory(destination, { ...options, projectRoot, - architecture - }) + architecture, + }); } finally { - rmSync(staging, { recursive: true, force: true }) + rmSync(staging, { recursive: true, force: true }); } } function findOnlyBundle(outputRoot, architecture) { - const runtimeRoot = join( - outputRoot, - targetName(architecture), - 'opencode' - ) + const runtimeRoot = join(outputRoot, targetName(architecture), "opencode"); const entries = existsSync(runtimeRoot) ? readdirSync(runtimeRoot, { withFileTypes: true }) .filter((entry) => entry.isDirectory()) .map((entry) => join(runtimeRoot, entry.name)) - : [] + : []; if (entries.length !== 1) { throw new Error( - `Expected exactly one built Runtime bundle for ${targetName(architecture)}` - ) + `Expected exactly one built Runtime bundle for ${targetName(architecture)}`, + ); } - return entries[0] + return entries[0]; } function redactSecrets(value, environment = process.env) { - let output = String(value) + let output = String(value); for (const secret of [ environment[signingPrivateKeyEnvironment], - environment[signingKeyIdEnvironment] + environment[signingKeyIdEnvironment], ]) { if (secret && secret.length >= 4) { - output = output.split(secret).join('[redacted]') + output = output.split(secret).join("[redacted]"); } } return output.replace( /-----BEGIN PRIVATE KEY-----[\s\S]*?-----END PRIVATE KEY-----/gu, - '[redacted private key]' - ) + "[redacted private key]", + ); } function parseArguments(argv) { - const command = argv[0] - const options = {} + const command = argv[0]; + const options = {}; const allowed = new Set([ - '--arch', - '--runtime-archive', - '--archive', - '--output-root', - '--bundle' - ]) + "--arch", + "--runtime-archive", + "--archive", + "--output-root", + "--bundle", + ]); for (let index = 1; index < argv.length; index += 2) { - const flag = argv[index] - const value = argv[index + 1] + const flag = argv[index]; + const value = argv[index + 1]; if (!allowed.has(flag) || !value) { - throw new Error('Invalid Remote Runtime bundle argument') + throw new Error("Invalid Remote Runtime bundle argument"); } options[ - flag.slice(2).replace(/-([a-z])/gu, (_match, value) => - value.toUpperCase() - ) - ] = value + flag + .slice(2) + .replace(/-([a-z])/gu, (_match, value) => value.toUpperCase()) + ] = value; } - if (!['build', 'import', 'verify', 'preflight'].includes(command)) { + if (!["build", "import", "verify", "preflight"].includes(command)) { throw new Error( - 'Usage: remote-runtime-bundle.cjs --arch ' - ) + "Usage: remote-runtime-bundle.cjs --arch ", + ); } - if (command === 'preflight') { - return { command } + if (command === "preflight") { + return { command }; } - targetName(options.arch) - return { command, ...options } + targetName(options.arch); + return { command, ...options }; } function main(argv = process.argv.slice(2)) { - const options = parseArguments(argv) - if (options.command === 'preflight') { - const record = preflightProductionSigningKey() + const options = parseArguments(argv); + if (options.command === "preflight") { + const record = preflightProductionSigningKey(); console.log( - `Production GoodBuddy signing preflight passed for bundled Runtime: ${record.keyId}` - ) - return + `Production GoodBuddy signing preflight passed for bundled Runtime: ${record.keyId}`, + ); + return; } const outputRoot = options.outputRoot ? resolve(options.outputRoot) - : defaultOutputRoot() - if (options.command === 'build') { + : defaultOutputRoot(); + if (options.command === "build") { const verified = buildRuntimeBundle({ architecture: options.arch, - runtimeArchive: resolve(options.runtimeArchive ?? ''), - outputRoot - }) + runtimeArchive: resolve(options.runtimeArchive ?? ""), + outputRoot, + }); const archivePath = options.archive ? resolve(options.archive) : join( root, - 'dist', - 'remote-runtime-bundles', - `goodbuddy-opencode-${targetName(options.arch)}.tar` - ) - createRuntimeArchive(verified.bundleDirectory, archivePath) + "dist", + "remote-runtime-bundles", + `goodbuddy-opencode-${targetName(options.arch)}.tar`, + ); + createRuntimeArchive(verified.bundleDirectory, archivePath); console.log( - `Remote Runtime bundle built and verified: ${targetName(options.arch)}` - ) - return + `Remote Runtime bundle built and verified: ${targetName(options.arch)}`, + ); + return; } - if (options.command === 'import') { - importRuntimeArchive(resolve(options.archive ?? ''), { + if (options.command === "import") { + importRuntimeArchive(resolve(options.archive ?? ""), { architecture: options.arch, - outputRoot - }) + outputRoot, + }); console.log( - `Remote Runtime bundle imported and verified: ${targetName(options.arch)}` - ) - return + `Remote Runtime bundle imported and verified: ${targetName(options.arch)}`, + ); + return; } verifyBundleDirectory( options.bundle ? resolve(options.bundle) : findOnlyBundle(outputRoot, options.arch), - { architecture: options.arch } - ) - console.log( - `Remote Runtime bundle verified: ${targetName(options.arch)}` - ) + { architecture: options.arch }, + ); + console.log(`Remote Runtime bundle verified: ${targetName(options.arch)}`); } module.exports = { @@ -1121,18 +1048,16 @@ module.exports = { signaturePayload, supportedArchitectures, targetName, - verifyBundleDirectory -} + verifyBundleDirectory, +}; if (require.main === module) { try { - main() + main(); } catch (error) { console.error( - redactSecrets( - error instanceof Error ? error.message : String(error) - ) - ) - process.exitCode = 1 + redactSecrets(error instanceof Error ? error.message : String(error)), + ); + process.exitCode = 1; } } diff --git a/build/runtime-hooks.cjs b/build/runtime-hooks.cjs index 47e16275..9996488d 100644 --- a/build/runtime-hooks.cjs +++ b/build/runtime-hooks.cjs @@ -1,4 +1,4 @@ -const { createHash } = require('node:crypto') +const { createHash } = require("node:crypto"); const { chmod, mkdir, @@ -6,102 +6,100 @@ const { rename, rm, stat, - writeFile -} = require('node:fs/promises') -const { existsSync } = require('node:fs') -const { join } = require('node:path') -const { spawnSync } = require('node:child_process') -const tar = require('tar') -const { sha256File } = require('./file-hash.cjs') + writeFile, +} = require("node:fs/promises"); +const { existsSync } = require("node:fs"); +const { join } = require("node:path"); +const { spawnSync } = require("node:child_process"); +const tar = require("tar"); +const { sha256File } = require("./file-hash.cjs"); +const { + opencodeVersion, + prepareBundledOpenCodeConfig, +} = require("./opencode-config.cjs"); -const opencodeVersion = '1.18.29' -const opencodePluginPackage = '@opencode-ai/plugin' const architectureNames = { - 1: 'x64', - 3: 'arm64' -} + 1: "x64", + 3: "arm64", +}; const platformNames = { - darwin: 'darwin', - linux: 'linux', - win32: 'windows' -} + darwin: "darwin", + linux: "linux", + win32: "windows", +}; function sha512Integrity(contents) { - return `sha512-${createHash('sha512').update(contents).digest('base64')}` + return `sha512-${createHash("sha512").update(contents).digest("base64")}`; } async function lockedIntegrity(projectDir, packageName) { const lock = JSON.parse( - await readFile(join(projectDir, 'package-lock.json'), 'utf8') - ) - const entry = lock.packages?.[`node_modules/${packageName}`] + await readFile(join(projectDir, "package-lock.json"), "utf8"), + ); + const entry = lock.packages?.[`node_modules/${packageName}`]; if ( entry?.version !== opencodeVersion || - typeof entry.integrity !== 'string' + typeof entry.integrity !== "string" ) { throw new Error( - `Missing locked ${packageName}@${opencodeVersion} integrity` - ) + `Missing locked ${packageName}@${opencodeVersion} integrity`, + ); } - return entry.integrity + return entry.integrity; } function npmInvocation() { - const npmCli = process.env.npm_execpath + const npmCli = process.env.npm_execpath; if (npmCli) { return { command: process.execPath, - prefixArgs: [npmCli] - } + prefixArgs: [npmCli], + }; } - if (process.platform === 'win32') { - throw new Error('npm_execpath is required to prepare bundled runtimes') + if (process.platform === "win32") { + throw new Error("npm_execpath is required to prepare bundled runtimes"); } return { - command: 'npm', - prefixArgs: [] - } + command: "npm", + prefixArgs: [], + }; } async function downloadPackage(projectDir, packageName, integrity) { - const cacheDirectory = join( - projectDir, - '.runtime-resources', - 'cache' - ) - await mkdir(cacheDirectory, { recursive: true }) + const cacheDirectory = join(projectDir, ".runtime-resources", "cache"); + await mkdir(cacheDirectory, { recursive: true }); const archivePath = join( cacheDirectory, - `${packageName}-${opencodeVersion}.tgz` - ) + `${packageName}-${opencodeVersion}.tgz`, + ); if (existsSync(archivePath)) { - const cached = await readFile(archivePath) + const cached = await readFile(archivePath); if (sha512Integrity(cached) === integrity) { - return archivePath + return archivePath; } - await rm(archivePath, { force: true }) + await rm(archivePath, { force: true }); } - const npm = npmInvocation() + const npm = npmInvocation(); const result = spawnSync( npm.command, [ ...npm.prefixArgs, - 'pack', + "pack", `${packageName}@${opencodeVersion}`, - '--ignore-scripts', - '--json', - '--pack-destination', - cacheDirectory + "--ignore-scripts", + "--json", + "--pack-destination", + cacheDirectory, ], { cwd: projectDir, - encoding: 'utf8', + encoding: "utf8", maxBuffer: 16 * 1024 * 1024, shell: false, - windowsHide: true - } - ) + windowsHide: true, + }, + ); if (result.status !== 0) { throw new Error( `Unable to fetch ${packageName}: ${ @@ -109,219 +107,107 @@ async function downloadPackage(projectDir, packageName, integrity) { result.stderr || result.stdout || `npm exited with ${result.status}` - }` - ) + }`, + ); } - const output = JSON.parse(result.stdout) - const downloadedPath = join(cacheDirectory, output[0].filename) - const contents = await readFile(downloadedPath) + const output = JSON.parse(result.stdout); + const downloadedPath = join(cacheDirectory, output[0].filename); + const contents = await readFile(downloadedPath); if (sha512Integrity(contents) !== integrity) { - await rm(downloadedPath, { force: true }) - throw new Error(`Integrity verification failed for ${packageName}`) + await rm(downloadedPath, { force: true }); + throw new Error(`Integrity verification failed for ${packageName}`); } if (downloadedPath !== archivePath) { - await rm(archivePath, { force: true }) - await rename(downloadedPath, archivePath) - } - return archivePath -} - -async function prepareBundledOpenCodeConfig(projectDir) { - const targetDirectory = join( - projectDir, - '.runtime-resources', - 'opencode-config' - ) - const pluginIntegrity = await lockedIntegrity( - projectDir, - opencodePluginPackage - ) - const identity = { - packageName: opencodePluginPackage, - version: opencodeVersion, - integrity: pluginIntegrity - } - try { - const ready = JSON.parse( - await readFile(join(targetDirectory, '.goodbuddy-ready.json'), 'utf8') - ) - if ( - ready.packageName === identity.packageName && - ready.version === identity.version && - ready.integrity === identity.integrity && - (await stat( - join( - targetDirectory, - 'node_modules', - '@opencode-ai', - 'plugin', - 'package.json' - ) - )).isFile() - ) { - return - } - } catch { - // Rebuild an incomplete or stale offline config template. - } - - const stagingDirectory = `${targetDirectory}.staging-${process.pid}` - await rm(stagingDirectory, { recursive: true, force: true }) - await mkdir(stagingDirectory, { recursive: true }) - try { - await writeFile( - join(stagingDirectory, 'package.json'), - `${JSON.stringify( - { - private: true, - dependencies: { - [opencodePluginPackage]: opencodeVersion - } - }, - null, - 2 - )}\n`, - 'utf8' - ) - const npm = npmInvocation() - const environment = { ...process.env } - delete environment.NODE_TLS_REJECT_UNAUTHORIZED - const result = spawnSync( - npm.command, - [ - ...npm.prefixArgs, - 'install', - '--ignore-scripts', - '--omit=dev', - '--omit=optional', - '--no-audit', - '--no-fund', - '--save-exact' - ], - { - cwd: stagingDirectory, - encoding: 'utf8', - env: environment, - maxBuffer: 16 * 1024 * 1024, - shell: false, - windowsHide: true - } - ) - if (result.status !== 0) { - throw new Error( - `Unable to prepare bundled OpenCode config: ${ - result.error?.message || - result.stderr || - result.stdout || - `npm exited with ${result.status}` - }` - ) - } - await writeFile( - join(stagingDirectory, '.goodbuddy-ready.json'), - `${JSON.stringify(identity)}\n`, - 'utf8' - ) - await rm(targetDirectory, { recursive: true, force: true }) - await rename(stagingDirectory, targetDirectory) - } finally { - await rm(stagingDirectory, { recursive: true, force: true }) + await rm(archivePath, { force: true }); + await rename(downloadedPath, archivePath); } + return archivePath; } module.exports = async function prepareBundledRuntimes(context) { - const platform = context.electronPlatformName - const architecture = architectureNames[context.arch] - const packagePlatform = platformNames[platform] + const platform = context.electronPlatformName; + const architecture = architectureNames[context.arch]; + const packagePlatform = platformNames[platform]; if (!architecture || !packagePlatform) { throw new Error( - `Bundled OpenCode does not support ${platform}/${context.arch}` - ) + `Bundled OpenCode does not support ${platform}/${context.arch}`, + ); } const suffix = - architecture === 'x64' ? `${architecture}-baseline` : architecture - const packageName = `opencode-${packagePlatform}-${suffix}` - const projectDir = context.packager.projectDir - await prepareBundledOpenCodeConfig(projectDir) + architecture === "x64" ? `${architecture}-baseline` : architecture; + const packageName = `opencode-${packagePlatform}-${suffix}`; + const projectDir = context.packager.projectDir; + await prepareBundledOpenCodeConfig(projectDir); const projectPackage = JSON.parse( - await readFile(join(projectDir, 'package.json'), 'utf8') - ) + await readFile(join(projectDir, "package.json"), "utf8"), + ); await writeFile( - join(projectDir, 'out', 'main', 'package.json'), + join(projectDir, "out", "main", "package.json"), `${JSON.stringify( { - name: '@deepseek-ai/dsh-llm', - version: - projectPackage.dependencies['@deepseek-ai/dsh-llm'], + name: "@deepseek-ai/dsh-llm", + version: projectPackage.dependencies["@deepseek-ai/dsh-llm"], private: true, - type: 'module' + type: "module", }, null, - 2 + 2, )}\n`, - 'utf8' - ) - const integrity = await lockedIntegrity(projectDir, packageName) - const targetDirectory = join( - projectDir, - '.runtime-resources', - architecture - ) - const readyPath = join(targetDirectory, '.ready.json') - const executable = platform === 'win32' ? 'opencode.exe' : 'opencode' - const preparedPath = join(targetDirectory, executable) + "utf8", + ); + const integrity = await lockedIntegrity(projectDir, packageName); + const targetDirectory = join(projectDir, ".runtime-resources", architecture); + const readyPath = join(targetDirectory, ".ready.json"); + const executable = platform === "win32" ? "opencode.exe" : "opencode"; + const preparedPath = join(targetDirectory, executable); const identity = { packageName, version: opencodeVersion, - integrity - } + integrity, + }; try { - const ready = JSON.parse(await readFile(readyPath, 'utf8')) + const ready = JSON.parse(await readFile(readyPath, "utf8")); if ( ready.packageName === identity.packageName && ready.version === identity.version && ready.integrity === identity.integrity && - typeof ready.executableSha256 === 'string' && - (platform === 'win32' || + typeof ready.executableSha256 === "string" && + (platform === "win32" || ((await stat(preparedPath)).mode & 0o111) !== 0) && (await sha256File(preparedPath)) === ready.executableSha256 ) { - return + return; } } catch { // Rebuild an incomplete or stale runtime cache. } - const archivePath = await downloadPackage( - projectDir, - packageName, - integrity - ) - const stagingDirectory = `${targetDirectory}.staging-${process.pid}` - await rm(stagingDirectory, { recursive: true, force: true }) - await mkdir(stagingDirectory, { recursive: true }) + const archivePath = await downloadPackage(projectDir, packageName, integrity); + const stagingDirectory = `${targetDirectory}.staging-${process.pid}`; + await rm(stagingDirectory, { recursive: true, force: true }); + await mkdir(stagingDirectory, { recursive: true }); try { await tar.x({ file: archivePath, cwd: stagingDirectory, - strip: 1 - }) - const sourcePath = join(stagingDirectory, 'bin', executable) - const stagingExecutable = join(stagingDirectory, executable) - await rename(sourcePath, stagingExecutable) - if (platform !== 'win32') { - await chmod(stagingExecutable, 0o755) + strip: 1, + }); + const sourcePath = join(stagingDirectory, "bin", executable); + const stagingExecutable = join(stagingDirectory, executable); + await rename(sourcePath, stagingExecutable); + if (platform !== "win32") { + await chmod(stagingExecutable, 0o755); } - const executableSha256 = await sha256File(stagingExecutable) + const executableSha256 = await sha256File(stagingExecutable); await writeFile( - join(stagingDirectory, '.ready.json'), + join(stagingDirectory, ".ready.json"), JSON.stringify({ ...identity, executableSha256 }), - 'utf8' - ) - await rm(targetDirectory, { recursive: true, force: true }) - await rename(stagingDirectory, targetDirectory) + "utf8", + ); + await rm(targetDirectory, { recursive: true, force: true }); + await rename(stagingDirectory, targetDirectory); } finally { - await rm(stagingDirectory, { recursive: true, force: true }) + await rm(stagingDirectory, { recursive: true, force: true }); } -} +}; diff --git a/docs/features/assistant-workbar/README.md b/docs/features/assistant-workbar/README.md index 2bb2d9bd..3f253f44 100644 --- a/docs/features/assistant-workbar/README.md +++ b/docs/features/assistant-workbar/README.md @@ -8,7 +8,7 @@ | --- | --- | | [通用助手工作栏与执行空间 PRD](./prd.md) | 工作栏整体产品范围、执行空间和各工作面板归属 | | [多终端页签 PRD](./terminal-tabs-prd.md) | 多终端 Tab、目标选择、关闭确认和终端生命周期 | -| [实现与验证进度](./progress.md) | 工作区改动的验证证据与剩余阻塞 | +| [实现与验证进度](./progress.md) | 工作区与 Runtime 相关改动的实现事实、验证证据与剩余阻塞 | 当前尚未拆出独立 User Stories、功能逻辑、UI 设计和技术设计文档；相关内容仍保留在 两份现有 PRD 中，后续实际维护该功能时按职责拆分。 diff --git a/docs/features/assistant-workbar/progress.md b/docs/features/assistant-workbar/progress.md index c23962e9..1eb13e29 100644 --- a/docs/features/assistant-workbar/progress.md +++ b/docs/features/assistant-workbar/progress.md @@ -1,5 +1,23 @@ # 工作栏实现与验证进度 +## 2026-09-08 OpenCode 复用与原生控件恢复 + +- `OpenCodeRuntime` 复用缓存的 embedded server 前执行有超时的健康检查；健康则保留 + client，检查失败、抛错或超时则清除 client、会话映射、会话初始化及待答问题，关闭旧 + server 后重建。并发调用共享检查与重建，调用方取消只结束自身等待。 +- 请求级 MCP 注册与断开通过实例内 `mutateMcp` 队列串行执行。仅实际发出 `mcp.add` + 的名称进入清理列表；注册尚在排队时取消不追加断开。断开的 1 秒超时从出队后开始。 +- `App` 按本机/远程、连接方式、项目和 Runtime 选择组成 scope 缓存原生清单，用请求 + 代次阻止旧 scope 结果回写；同 scope 会话刷新不清空重取，切换会话重置消息级选择， + Runtime 切换状态也按 generation 隔离。清单请求失败或返回 `unavailable`/`partial` + 时保留已有可用缓存并有界重试；没有可用缓存时保留新返回的部分数据，因此 + `unavailable` 重试为可用 `partial` 后可显示 Agent 和 Command。 + +验证记录：首轮定向测试 273 项通过、5 项跳过；补充修复后的 Runtime 与生命周期测试 +77 项通过，界面定向测试 6 项通过，类型检查通过。修复前的 lint 检查通过。 +全量测试超过 120 秒被终止，期间出现一项 DS 本地 stdio MCP 测试失败，尚待定位。 +真实系统休眠恢复和当前源码的 GoodBuddy Agent 实机路径尚未完成验证。 + ## 2026-09-06 项目工作区 范围见 [PRD 工作区](./prd.md#76-工作区)。 diff --git a/src/agent-daemon/model-bridge-helper.ts b/src/agent-daemon/model-bridge-helper.ts index 27fe8c20..bde1ee5e 100644 --- a/src/agent-daemon/model-bridge-helper.ts +++ b/src/agent-daemon/model-bridge-helper.ts @@ -890,6 +890,26 @@ function credentialFreeHelperEnvironment( } environment[name] = value } + for (const name of [ + 'OPENCODE_CONFIG_DIR', + 'OPENCODE_DISABLE_AUTOUPDATE', + 'OPENCODE_DISABLE_CLAUDE_CODE_SKILLS', + 'OPENCODE_DISABLE_EMBEDDED_WEB_UI', + 'OPENCODE_DISABLE_EXTERNAL_SKILLS', + 'OPENCODE_EXPERIMENTAL_DISABLE_FILEWATCHER', + 'OPENCODE_DISABLE_LSP_DOWNLOAD', + 'OPENCODE_DISABLE_MODELS_FETCH', + 'OPENCODE_DISABLE_PROJECT_CONFIG', + 'OPENCODE_DISABLE_SHARE' + ]) { + const value = source[name] + if (value !== undefined) { + if (value.includes('\0')) { + throw new Error(`Invalid helper environment value for ${name}`) + } + environment[name] = value + } + } return environment } diff --git a/src/agent-daemon/model-bridge.test.ts b/src/agent-daemon/model-bridge.test.ts index 20fa4efc..0f380e8a 100644 --- a/src/agent-daemon/model-bridge.test.ts +++ b/src/agent-daemon/model-bridge.test.ts @@ -1,3 +1,4 @@ +import { EventEmitter } from 'node:events' import { chmodSync, existsSync, @@ -21,8 +22,12 @@ import { import { createOpenCodeModelBridgeProviderConfig, MODEL_BRIDGE_SDK_AUTH_SENTINEL, - ModelBridgeLoopbackProxy + ModelBridgeLoopbackProxy, + runOpenCodeModelBridgeHelper, + type ModelBridgeHelperSpawn } from './model-bridge-helper' +import { createOpenCodeLaunchProfile } from './opencode-runtime-profile' +import { createRuntimeBundleTestFixture } from './runtime-bundle-test-fixture' import { createUnixModelBridgeExchange, MODEL_BRIDGE_BROKER_SOCKET_NAME, @@ -51,6 +56,89 @@ afterEach(() => { }) describe('model bridge loopback helper', () => { + it.each(['ask', 'execute'] as const)( + 'preserves the %s launch profile offline resources in the final OpenCode spawn', + async (workMode) => { + const fixture = await createRuntimeBundleTestFixture() + temporaryPaths.push(fixture.root) + const socketPath = join(fixture.root, 'bridge', 'model.sock') + const profile = createOpenCodeLaunchProfile({ + ...fixture, + workspaceDirectory: fixture.root, + workMode, + modelBridge: { + agentExecutablePath: join(fixture.root, 'agent', 'goodbuddy-agent'), + bridgeDirectory: join(fixture.root, 'bridge'), + socketPath, + policy: { + protocol: 'anthropic-messages', + model: 'private-model', + modelProfileDigest: `sha256:${'9'.repeat(64)}`, + supportsImageInput: false + } + } + }) + const spawn = vi.fn(() => { + const child = new EventEmitter() + queueMicrotask(() => child.emit('close', 0, null)) + return child + }) + const resources = { + HOME: fixture.root, + PATH: join(fixture.root, 'tools'), + TMPDIR: join(fixture.root, 'tmp'), + XDG_CACHE_HOME: join(fixture.root, 'cache'), + XDG_CONFIG_HOME: join(fixture.root, 'config'), + XDG_DATA_HOME: join(fixture.root, 'data'), + XDG_STATE_HOME: join(fixture.root, 'state') + } + await expect(runOpenCodeModelBridgeHelper({ + socketPath, + protocol: 'anthropic-messages', + model: 'private-model', + supportsImageInput: false, + workMode, + opencodeEntrypoint: join(fixture.bundleDirectory, 'bin', 'opencode'), + environment: { + ...profile.env, + ...resources, + OPENCODE_CONFIG_CONTENT: '{"model":"must-not-be-used"}', + OPENCODE_CONFIG: '/unrelated/config.json', + ANTHROPIC_API_KEY: 'must-not-be-forwarded', + NODE_OPTIONS: '--inspect' + }, + spawn + })).resolves.toBe(0) + expect(spawn).toHaveBeenCalledOnce() + const [executable, args, options] = spawn.mock.calls[0]! + expect(executable).toBe(join(fixture.bundleDirectory, 'bin', 'opencode')) + expect(args).toEqual(['acp']) + expect(options).toMatchObject({ shell: false, stdio: 'inherit' }) + expect(options.env).toMatchObject({ + ...resources, + LANG: profile.env.LANG, + LC_ALL: profile.env.LC_ALL, + OPENCODE_CONFIG_DIR: join(fixture.bundleDirectory, 'config', 'opencode'), + OPENCODE_DISABLE_AUTOUPDATE: '1', + OPENCODE_DISABLE_CLAUDE_CODE_SKILLS: '1', + OPENCODE_DISABLE_EMBEDDED_WEB_UI: '1', + OPENCODE_DISABLE_EXTERNAL_SKILLS: '1', + OPENCODE_EXPERIMENTAL_DISABLE_FILEWATCHER: '1', + OPENCODE_DISABLE_LSP_DOWNLOAD: '1', + OPENCODE_DISABLE_MODELS_FETCH: '1', + OPENCODE_DISABLE_PROJECT_CONFIG: '1', + OPENCODE_DISABLE_SHARE: '1' + }) + expect(options.env.OPENCODE_CONFIG).toBeUndefined() + expect(options.env.ANTHROPIC_API_KEY).toBeUndefined() + expect(options.env.NODE_OPTIONS).toBeUndefined() + const config = JSON.parse(options.env.OPENCODE_CONFIG_CONTENT!) + expect(config.model).toBe('goodbuddy-anthropic/private-model') + expect(config.permission).toBe(workMode === 'ask' ? 'ask' : undefined) + expect(config.plugin).toHaveLength(1) + } + ) + it('forwards only the bounded request contract and returns the response', async () => { const exchange = vi.fn( async () => validResponse diff --git a/src/agent-daemon/opencode-runtime-profile.ts b/src/agent-daemon/opencode-runtime-profile.ts index e4baa4fd..66f85c8a 100644 --- a/src/agent-daemon/opencode-runtime-profile.ts +++ b/src/agent-daemon/opencode-runtime-profile.ts @@ -1,230 +1,216 @@ -import { - dirname, - isAbsolute, - join, - relative, - resolve, - sep -} from 'node:path' +import { dirname, isAbsolute, join, relative, resolve, sep } from "node:path"; import { OPENCODE_REMOTE_RUNTIME_ENVIRONMENT_NAMES, remoteRuntimeBundleManifestSchema, - type RemoteRuntimeBundleManifest -} from '../shared/remote-runtime-launch-contracts' + type RemoteRuntimeBundleManifest, +} from "../shared/remote-runtime-launch-contracts"; import { modelBridgePolicySchema, - type ModelBridgePolicy -} from '../shared/model-bridge-contracts' -import { - assertAbsoluteManagedPath -} from './managed-paths' + type ModelBridgePolicy, +} from "../shared/model-bridge-contracts"; +import { assertAbsoluteManagedPath } from "./managed-paths"; const OPEN_CODE_ASK_CONFIG = JSON.stringify({ - permission: 'ask', + permission: "ask", agent: { build: { - permission: 'ask' - } - } -}) + permission: "ask", + }, + }, +}); + +const OPEN_CODE_RUNTIME_ENVIRONMENT = Object.freeze({ + OPENCODE_DISABLE_AUTOUPDATE: "1", + OPENCODE_DISABLE_CLAUDE_CODE_SKILLS: "1", + OPENCODE_DISABLE_EMBEDDED_WEB_UI: "1", + OPENCODE_DISABLE_EXTERNAL_SKILLS: "1", + OPENCODE_EXPERIMENTAL_DISABLE_FILEWATCHER: "1", + OPENCODE_DISABLE_LSP_DOWNLOAD: "1", + OPENCODE_DISABLE_MODELS_FETCH: "1", + OPENCODE_DISABLE_PROJECT_CONFIG: "1", + OPENCODE_DISABLE_SHARE: "1", +}); export type OpenCodeLaunchProfile = { - executable: string - processExecutable: string - args: readonly string[] - cwd: string - env: Readonly - workMode: 'ask' | 'execute' -} + executable: string; + processExecutable: string; + args: readonly string[]; + cwd: string; + env: Readonly; + workMode: "ask" | "execute"; +}; export function createOpenCodeLaunchProfile(input: { - manifest: RemoteRuntimeBundleManifest - bundleDirectory: string - workspaceDirectory: string - workMode: 'ask' | 'execute' + manifest: RemoteRuntimeBundleManifest; + bundleDirectory: string; + workspaceDirectory: string; + workMode: "ask" | "execute"; modelBridge?: { - agentExecutablePath: string - bridgeDirectory: string - socketPath: string - policy: ModelBridgePolicy - } + agentExecutablePath: string; + bridgeDirectory: string; + socketPath: string; + policy: ModelBridgePolicy; + }; }): OpenCodeLaunchProfile { - const manifest = remoteRuntimeBundleManifestSchema.parse( - input.manifest - ) + const manifest = remoteRuntimeBundleManifestSchema.parse(input.manifest); if ( - manifest.runtimeId !== 'opencode' || - manifest.provider !== 'opencode' || - manifest.entrypoint.identity !== 'opencode-acp' || + manifest.runtimeId !== "opencode" || + manifest.provider !== "opencode" || + manifest.entrypoint.identity !== "opencode-acp" || manifest.entrypoint.argvPrefix.length !== 1 || - manifest.entrypoint.argvPrefix[0] !== 'acp' + manifest.entrypoint.argvPrefix[0] !== "acp" ) { - throw new Error('Runtime manifest is not the fixed OpenCode ACP profile') + throw new Error("Runtime manifest is not the fixed OpenCode ACP profile"); } if ( - [...manifest.allowedEnvironmentNames].sort().join('\0') !== - [...OPENCODE_REMOTE_RUNTIME_ENVIRONMENT_NAMES] - .sort() - .join('\0') + [...manifest.allowedEnvironmentNames].sort().join("\0") !== + [...OPENCODE_REMOTE_RUNTIME_ENVIRONMENT_NAMES].sort().join("\0") ) { throw new Error( - 'OpenCode Runtime environment allowlist does not match the fixed profile' - ) + "OpenCode Runtime environment allowlist does not match the fixed profile", + ); } const bundleDirectory = assertAbsoluteManagedPath( - resolve(input.bundleDirectory) - ) + resolve(input.bundleDirectory), + ); const workspaceDirectory = normalizedAbsolutePath( input.workspaceDirectory, - 'Workspace' - ) + "Workspace", + ); const modelBridge = input.modelBridge === undefined ? undefined : { agentExecutablePath: normalizedAbsolutePath( input.modelBridge.agentExecutablePath, - 'Agent executable' + "Agent executable", ), agentInstallationDirectory: normalizedAbsolutePath( dirname(input.modelBridge.agentExecutablePath), - 'Agent installation' + "Agent installation", ), bridgeDirectory: normalizedAbsolutePath( input.modelBridge.bridgeDirectory, - 'Model bridge' + "Model bridge", ), socketPath: normalizedAbsolutePath( input.modelBridge.socketPath, - 'Model bridge socket' + "Model bridge socket", ), - policy: modelBridgePolicySchema.parse( - input.modelBridge.policy - ) - } + policy: modelBridgePolicySchema.parse(input.modelBridge.policy), + }; if ( modelBridge !== undefined && - (!isStrictDescendant( - modelBridge.bridgeDirectory, - modelBridge.socketPath - ) || + (!isStrictDescendant(modelBridge.bridgeDirectory, modelBridge.socketPath) || !isStrictDescendant( modelBridge.agentInstallationDirectory, - modelBridge.agentExecutablePath + modelBridge.agentExecutablePath, )) ) { throw new Error( - 'Model bridge socket or Agent executable escapes its managed directory' - ) + "Model bridge socket or Agent executable escapes its managed directory", + ); } const executablePath = resolve( bundleDirectory, - ...manifest.entrypoint.path.split('/') - ) + ...manifest.entrypoint.path.split("/"), + ); if (!isStrictDescendant(bundleDirectory, executablePath)) { - throw new Error('OpenCode entrypoint escapes its bundle directory') + throw new Error("OpenCode entrypoint escapes its bundle directory"); } const runtimeCommand = modelBridge === undefined ? { executable: executablePath, - args: [...manifest.entrypoint.argvPrefix] + args: [...manifest.entrypoint.argvPrefix], } : { - executable: join( - modelBridge.agentInstallationDirectory, - 'node' - ), + executable: join(modelBridge.agentInstallationDirectory, "node"), args: [ - join( - modelBridge.agentInstallationDirectory, - 'lib', - 'agent.cjs' - ), - 'model-bridge-helper', - '--socket-path', + join(modelBridge.agentInstallationDirectory, "lib", "agent.cjs"), + "model-bridge-helper", + "--socket-path", modelBridge.socketPath, - '--protocol', + "--protocol", modelBridge.policy.protocol, - '--model', + "--model", modelBridge.policy.model, - '--supports-image-input', - modelBridge.policy.supportsImageInput ? 'true' : 'false', - '--work-mode', + "--supports-image-input", + modelBridge.policy.supportsImageInput ? "true" : "false", + "--work-mode", input.workMode, - '--opencode-entrypoint', - executablePath - ] - } + "--opencode-entrypoint", + executablePath, + ], + }; + const environmentBase = { + ...executeEnvironment(process.env, workspaceDirectory), + ...OPEN_CODE_RUNTIME_ENVIRONMENT, + OPENCODE_CONFIG_DIR: join(bundleDirectory, "config", "opencode"), + }; const environment = - input.workMode === 'ask' + input.workMode === "ask" ? { - ...executeEnvironment(process.env, workspaceDirectory), - OPENCODE_CONFIG_CONTENT: OPEN_CODE_ASK_CONFIG + ...environmentBase, + OPENCODE_CONFIG_CONTENT: OPEN_CODE_ASK_CONFIG, } - : executeEnvironment(process.env, workspaceDirectory) + : environmentBase; return { ...runtimeCommand, processExecutable: runtimeCommand.executable, cwd: workspaceDirectory, env: environment, - workMode: input.workMode - } + workMode: input.workMode, + }; } function normalizedAbsolutePath(value: string, label: string): string { - if ( - value.includes('\0') || - !isAbsolute(value) || - resolve(value) !== value - ) { - throw new Error(`${label} path must be a normalized absolute path`) + if (value.includes("\0") || !isAbsolute(value) || resolve(value) !== value) { + throw new Error(`${label} path must be a normalized absolute path`); } - return value + return value; } function executeEnvironment( source: Readonly, - workspaceDirectory: string + workspaceDirectory: string, ): Readonly { const home = usableEnvironmentValue(source.HOME) ? source.HOME! - : workspaceDirectory + : workspaceDirectory; const defaults = { HOME: home, - LANG: 'C.UTF-8', - LC_ALL: 'C.UTF-8', - PATH: '/usr/bin:/bin', - TMPDIR: '/tmp', - XDG_CACHE_HOME: join(home, '.cache'), - XDG_CONFIG_HOME: join(home, '.config'), - XDG_DATA_HOME: join(home, '.local', 'share'), - XDG_STATE_HOME: join(home, '.local', 'state') - } as const - const environment: NodeJS.ProcessEnv = { ...source } + LANG: "C.UTF-8", + LC_ALL: "C.UTF-8", + PATH: "/usr/bin:/bin", + TMPDIR: "/tmp", + XDG_CACHE_HOME: join(home, ".cache"), + XDG_CONFIG_HOME: join(home, ".config"), + XDG_DATA_HOME: join(home, ".local", "share"), + XDG_STATE_HOME: join(home, ".local", "state"), + } as const; + const environment: NodeJS.ProcessEnv = { ...source }; for (const [name, fallback] of Object.entries(defaults)) { if (!usableEnvironmentValue(environment[name])) { - environment[name] = fallback + environment[name] = fallback; } } - return environment + return environment; } -function usableEnvironmentValue( - value: string | undefined -): value is string { - return value !== undefined && value.length > 0 && !value.includes('\0') +function usableEnvironmentValue(value: string | undefined): value is string { + return value !== undefined && value.length > 0 && !value.includes("\0"); } function isStrictDescendant(parent: string, candidate: string): boolean { - const child = relative(parent, candidate) + const child = relative(parent, candidate); return ( child.length > 0 && - child !== '..' && + child !== ".." && !child.startsWith(`..${sep}`) && !isAbsolute(child) - ) + ); } diff --git a/src/main/agent/opencode-runtime.test.ts b/src/main/agent/opencode-runtime.test.ts index a5e1cb2c..6c98d887 100644 --- a/src/main/agent/opencode-runtime.test.ts +++ b/src/main/agent/opencode-runtime.test.ts @@ -1,106 +1,106 @@ -import { EventEmitter } from 'node:events' -import { createServer } from 'node:http' +import { EventEmitter } from "node:events"; +import { createServer } from "node:http"; import { mkdir, mkdtemp, readFile, rm, stat, - writeFile -} from 'node:fs/promises' -import { tmpdir } from 'node:os' -import { join, resolve } from 'node:path' -import { PassThrough } from 'node:stream' -import type { createOpencodeClient } from '@opencode-ai/sdk/v2' -import type spawn from 'cross-spawn' -import { describe, expect, it, vi } from 'vitest' -import type { KnowledgeMcpGateway } from './knowledge-mcp-gateway' -import type { RuntimeEvent } from './runtime' + writeFile, +} from "node:fs/promises"; +import { tmpdir } from "node:os"; +import { join, resolve } from "node:path"; +import { PassThrough } from "node:stream"; +import type { createOpencodeClient } from "@opencode-ai/sdk/v2"; +import type spawn from "cross-spawn"; +import { describe, expect, it, vi } from "vitest"; +import type { KnowledgeMcpGateway } from "./knowledge-mcp-gateway"; +import type { RuntimeEvent } from "./runtime"; import { OpenCodeRuntime, - type OpenCodeRuntimeDependencies -} from './opencode-runtime' + type OpenCodeRuntimeDependencies, +} from "./opencode-runtime"; -type SpawnedProcess = ReturnType +type SpawnedProcess = ReturnType; function fakeChild(pid = 42): SpawnedProcess { const child = new EventEmitter() as EventEmitter & { - stdout: PassThrough - stderr: PassThrough - exitCode: number | null - killed: boolean - pid: number - kill: ReturnType - unref: ReturnType - } - child.stdout = new PassThrough() - child.stderr = new PassThrough() - child.exitCode = null - child.killed = false - child.pid = pid + stdout: PassThrough; + stderr: PassThrough; + exitCode: number | null; + killed: boolean; + pid: number; + kill: ReturnType; + unref: ReturnType; + }; + child.stdout = new PassThrough(); + child.stderr = new PassThrough(); + child.exitCode = null; + child.killed = false; + child.pid = pid; child.kill = vi.fn(() => { - child.killed = true + child.killed = true; queueMicrotask(() => { - child.exitCode = 0 - child.emit('close', 0, null) - }) - return true - }) - child.unref = vi.fn(() => child) - return child as unknown as SpawnedProcess + child.exitCode = 0; + child.emit("close", 0, null); + }); + return true; + }); + child.unref = vi.fn(() => child); + return child as unknown as SpawnedProcess; } function fakeClient() { return { session: { - list: vi.fn().mockResolvedValue({ data: [], error: undefined }) - } - } as unknown as ReturnType + list: vi.fn().mockResolvedValue({ data: [], error: undefined }), + }, + } as unknown as ReturnType; } function stderrOf(child: SpawnedProcess): PassThrough { - return child.stderr as PassThrough + return child.stderr as PassThrough; } function closeChild(child: SpawnedProcess, code: number): void { - ;(child as unknown as { exitCode: number | null }).exitCode = code - child.emit('close', code, null) + (child as unknown as { exitCode: number | null }).exitCode = code; + child.emit("close", code, null); } function options( - overrides: Partial[0]> = {} + overrides: Partial[0]> = {}, ): ConstructorParameters[0] { return { embedded: true, - binaryPath: '', - configPath: '', + binaryPath: "", + configPath: "", defaultWorkspace: process.cwd(), - ...overrides - } + ...overrides, + }; } function dependencies( child: SpawnedProcess, - overrides: Partial = {} + overrides: Partial = {}, ): { - deps: Partial - spawnMock: ReturnType - detectBinary: ReturnType - validateBinary: ReturnType - createClient: ReturnType - checkServerHealth: ReturnType + deps: Partial; + spawnMock: ReturnType; + detectBinary: ReturnType; + validateBinary: ReturnType; + createClient: ReturnType; + checkServerHealth: ReturnType; } { - const spawnMock = vi.fn(() => child) + const spawnMock = vi.fn(() => child); const detectBinary = vi.fn().mockResolvedValue({ - path: 'opencode', - detail: 'OpenCode CLI 已就绪' - }) + path: "opencode", + detail: "OpenCode CLI 已就绪", + }); const validateBinary = vi.fn().mockResolvedValue({ - path: 'opencode', - detail: 'OpenCode CLI 已就绪' - }) - const createClient = vi.fn(() => fakeClient()) - const checkServerHealth = vi.fn().mockResolvedValue(true) + path: "opencode", + detail: "OpenCode CLI 已就绪", + }); + const createClient = vi.fn(() => fakeClient()); + const checkServerHealth = vi.fn().mockResolvedValue(true); return { deps: { spawn: spawnMock as unknown as typeof spawn, @@ -108,146 +108,140 @@ function dependencies( validateBinary, createClient: createClient as unknown as typeof createOpencodeClient, checkServerHealth, - platform: 'linux', + platform: "linux", startupTimeoutMs: 100, - ...overrides + ...overrides, }, spawnMock, detectBinary, validateBinary, createClient, - checkServerHealth - } + checkServerHealth, + }; } function permissionEvent( - overrides: Record = {} + overrides: Record = {}, ): Record { return { - id: 'event-1', - type: 'permission.asked', + id: "event-1", + type: "permission.asked", properties: { - id: 'permission-1', - sessionID: 'session-1', - permission: 'bash', - patterns: ['npm test'], - metadata: { command: 'npm test' }, - always: ['npm test'], + id: "permission-1", + sessionID: "session-1", + permission: "bash", + patterns: ["npm test"], + metadata: { command: "npm test" }, + always: ["npm test"], tool: { - messageID: 'message-1', - callID: 'call-1' + messageID: "message-1", + callID: "call-1", }, - ...overrides - } - } + ...overrides, + }, + }; } function completedToolEvent( - callId = 'call-1', - tool = 'bash' + callId = "call-1", + tool = "bash", ): Record { return { id: `event-tool-${callId}`, - type: 'message.part.updated', + type: "message.part.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", part: { id: `part-${callId}`, callID: callId, - type: 'tool', + type: "tool", tool, state: { - status: 'completed', + status: "completed", input: { - command: 'npm test', - token: 'visible-token' + command: "npm test", + token: "visible-token", }, - output: 'Tests passed\nAuthorization: Bearer secret-token' - } - } - } - } + output: "Tests passed\nAuthorization: Bearer secret-token", + }, + }, + }, + }; } function runClient(events: Record[]) { - const callOrder: string[] = [] + const callOrder: string[] = []; const permissionReply = vi.fn().mockResolvedValue({ data: true, - error: undefined - }) + error: undefined, + }); const questionReply = vi.fn().mockResolvedValue({ data: true, - error: undefined - }) + error: undefined, + }); const questionReject = vi.fn().mockResolvedValue({ data: true, - error: undefined - }) + error: undefined, + }); const client = { session: { list: vi.fn().mockResolvedValue({ data: [], error: undefined }), create: vi.fn().mockResolvedValue({ - data: { id: 'session-1' }, - error: undefined + data: { id: "session-1" }, + error: undefined, }), update: vi.fn().mockResolvedValue({ - data: { id: 'session-1' }, - error: undefined + data: { id: "session-1" }, + error: undefined, }), promptAsync: vi.fn().mockImplementation(async () => { - callOrder.push('prompt') - return { data: true, error: undefined } + callOrder.push("prompt"); + return { data: true, error: undefined }; }), abort: vi.fn().mockResolvedValue({ data: true, - error: undefined + error: undefined, }), delete: vi.fn().mockResolvedValue({ data: true, - error: undefined - }) + error: undefined, + }), }, event: { subscribe: vi.fn().mockImplementation(async () => { - callOrder.push('subscribe') + callOrder.push("subscribe"); return { stream: (async function* () { for (const event of events) { - yield event + yield event; } - })() - } - }) + })(), + }; + }), }, permission: { - reply: permissionReply + reply: permissionReply, }, question: { reply: questionReply, - reject: questionReject + reject: questionReject, }, mcp: { - add: vi - .fn() - .mockImplementation( - async (input: { name: string }) => ({ - data: { - [input.name]: { status: 'connected' } - }, - error: undefined - }) - ), - disconnect: vi - .fn() - .mockResolvedValue({ data: true, error: undefined }) + add: vi.fn().mockImplementation(async (input: { name: string }) => ({ + data: { + [input.name]: { status: "connected" }, + }, + error: undefined, + })), + disconnect: vi.fn().mockResolvedValue({ data: true, error: undefined }), }, tool: { ids: vi.fn().mockResolvedValue({ - data: ['read', 'write', 'bash', 'task'], - error: undefined - }) - } - } as unknown as ReturnType + data: ["read", "write", "bash", "task"], + error: undefined, + }), + }, + } as unknown as ReturnType; return { client, callOrder, @@ -256,2259 +250,2276 @@ function runClient(events: Record[]) { questionReject, session: client.session, event: client.event, - tool: client.tool - } + tool: client.tool, + }; } function embeddedRuntime( client: ReturnType, - overrides: Partial< - ConstructorParameters[0] - > = {}, - dependencyOverrides: Partial = {} + overrides: Partial[0]> = {}, + dependencyOverrides: Partial = {}, ): OpenCodeRuntime { - const child = fakeChild() + const child = fakeChild(); const { deps } = dependencies(child, { ...dependencyOverrides, - createClient: vi.fn( - () => client - ) as unknown as typeof createOpencodeClient - }) - return new OpenCodeRuntime(options(overrides), deps) + createClient: vi.fn(() => client) as unknown as typeof createOpencodeClient, + }); + return new OpenCodeRuntime(options(overrides), deps); } async function collectRun( runtime: OpenCodeRuntime, - workMode: 'ask' | 'execute' = 'execute' + workMode: "ask" | "execute" = "execute", ) { - const events = [] + const events = []; for await (const event of runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'test', - workMode + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "test", + workMode, }, - new AbortController().signal + new AbortController().signal, )) { - events.push(event) + events.push(event); } - return events + return events; } -describe('OpenCodeRuntime embedded launcher', () => { - it('checks embedded availability without starting OpenCode', async () => { - const child = fakeChild() +describe("OpenCodeRuntime embedded launcher", () => { + it("checks embedded availability without starting OpenCode", async () => { + const child = fakeChild(); const { deps, spawnMock, detectBinary, validateBinary, createClient, - checkServerHealth - } = dependencies(child) - const runtime = new OpenCodeRuntime(options(), deps) + checkServerHealth, + } = dependencies(child); + const runtime = new OpenCodeRuntime(options(), deps); await expect(runtime.getStatus()).resolves.toMatchObject({ available: true, - detail: 'OpenCode 已配置，将在首次使用时启动' - }) - expect(detectBinary).toHaveBeenCalledWith( - 'opencode', - '', - undefined - ) - expect(validateBinary).toHaveBeenCalledWith( - 'opencode', - 'opencode' - ) - expect(spawnMock).not.toHaveBeenCalled() - expect(createClient).not.toHaveBeenCalled() - expect(checkServerHealth).not.toHaveBeenCalled() + detail: "OpenCode 已配置，将在首次使用时启动", + }); + expect(detectBinary).toHaveBeenCalledWith("opencode", "", undefined); + expect(validateBinary).toHaveBeenCalledWith("opencode", "opencode"); + expect(spawnMock).not.toHaveBeenCalled(); + expect(createClient).not.toHaveBeenCalled(); + expect(checkServerHealth).not.toHaveBeenCalled(); - await runtime.dispose() - }) + await runtime.dispose(); + }); - it('reports an unavailable embedded binary without starting OpenCode', async () => { - const child = fakeChild() + it("reports an unavailable embedded binary without starting OpenCode", async () => { + const child = fakeChild(); const { deps, spawnMock, detectBinary, validateBinary, createClient, - checkServerHealth - } = dependencies(child) + checkServerHealth, + } = dependencies(child); detectBinary.mockResolvedValueOnce({ - path: '/missing/opencode', - detail: 'OpenCode CLI 已就绪' - }) + path: "/missing/opencode", + detail: "OpenCode CLI 已就绪", + }); validateBinary.mockResolvedValueOnce({ - detail: 'OpenCode CLI 未找到' - }) + detail: "OpenCode CLI 未找到", + }); const runtime = new OpenCodeRuntime( options({ - binaryPath: '/missing/opencode', - bundledBinaryPath: '/bundled/opencode' + binaryPath: "/missing/opencode", + bundledBinaryPath: "/bundled/opencode", }), - deps - ) + deps, + ); await expect(runtime.getStatus()).resolves.toEqual({ - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: false, supportsToolExecution: true, - detail: 'OpenCode CLI 未找到' - }) - expect(spawnMock).not.toHaveBeenCalled() + detail: "OpenCode CLI 未找到", + }); + expect(spawnMock).not.toHaveBeenCalled(); expect(detectBinary).toHaveBeenCalledWith( - 'opencode', - '/missing/opencode', - '/bundled/opencode' - ) + "opencode", + "/missing/opencode", + "/bundled/opencode", + ); expect(validateBinary).toHaveBeenCalledWith( - 'opencode', - '/missing/opencode' - ) - expect(createClient).not.toHaveBeenCalled() - expect(checkServerHealth).not.toHaveBeenCalled() + "opencode", + "/missing/opencode", + ); + expect(createClient).not.toHaveBeenCalled(); + expect(checkServerHealth).not.toHaveBeenCalled(); - await runtime.dispose() - }) + await runtime.dispose(); + }); - it('reports a missing embedded model credential without starting OpenCode', async () => { - const child = fakeChild() + it("reports a missing embedded model credential without starting OpenCode", async () => { + const child = fakeChild(); const { deps, spawnMock, detectBinary, validateBinary, createClient, - checkServerHealth - } = dependencies(child) + checkServerHealth, + } = dependencies(child); const runtime = new OpenCodeRuntime( options({ modelProfile: { - id: '00000000-0000-4000-8000-000000000011', - name: 'Independent model', - baseUrl: 'https://model.example', - modelName: 'private-model', - protocol: 'anthropic-messages', - authentication: 'api-key' - } + id: "00000000-0000-4000-8000-000000000011", + name: "Independent model", + baseUrl: "https://model.example", + modelName: "private-model", + protocol: "anthropic-messages", + authentication: "api-key", + }, }), - deps - ) + deps, + ); await expect(runtime.getStatus()).resolves.toEqual({ - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: false, supportsToolExecution: true, - detail: 'OpenCode 独立模型连接尚未配置 API Key' - }) - expect(detectBinary).not.toHaveBeenCalled() - expect(validateBinary).not.toHaveBeenCalled() - expect(spawnMock).not.toHaveBeenCalled() - expect(createClient).not.toHaveBeenCalled() - expect(checkServerHealth).not.toHaveBeenCalled() + detail: "OpenCode 独立模型连接尚未配置 API Key", + }); + expect(detectBinary).not.toHaveBeenCalled(); + expect(validateBinary).not.toHaveBeenCalled(); + expect(spawnMock).not.toHaveBeenCalled(); + expect(createClient).not.toHaveBeenCalled(); + expect(checkServerHealth).not.toHaveBeenCalled(); - await runtime.dispose() - }) + await runtime.dispose(); + }); - it('uses the detected binary and passes an absolute config path only through env', async () => { - const serverChild = fakeChild(314) - const killerChild = fakeChild(315) + it("uses the detected binary and passes an absolute config path only through env", async () => { + const serverChild = fakeChild(314); + const killerChild = fakeChild(315); const detectBinary = vi.fn().mockResolvedValue({ - path: 'C:\\Tools\\opencode.exe', - detail: 'OpenCode CLI 已就绪' - }) - const createClient = vi.fn(() => fakeClient()) - const checkServerHealth = vi.fn().mockResolvedValue(true) + path: "C:\\Tools\\opencode.exe", + detail: "OpenCode CLI 已就绪", + }); + const createClient = vi.fn(() => fakeClient()); + const checkServerHealth = vi.fn().mockResolvedValue(true); const spawnMock = vi.fn((command: string) => { - if (command === 'taskkill.exe') { + if (command === "taskkill.exe") { queueMicrotask(() => { - closeChild(serverChild, 0) - }) - return killerChild + closeChild(serverChild, 0); + }); + return killerChild; } - return serverChild - }) - const configPath = './private/opencode.json' + return serverChild; + }); + const configPath = "./private/opencode.json"; const runtime = new OpenCodeRuntime( options({ - binaryPath: 'C:\\Configured\\opencode.exe', + binaryPath: "C:\\Configured\\opencode.exe", configPath, launchEnvironmentProvider: () => Object.freeze({ - PATH: 'C:\\GoodBuddy\\tools;C:\\Windows', - OPENAI_API_KEY: 'must-not-leak', - ELECTRON_RUN_AS_NODE: '1' - }) + PATH: "C:\\GoodBuddy\\tools;C:\\Windows", + OPENAI_API_KEY: "must-not-leak", + ELECTRON_RUN_AS_NODE: "1", + }), }), { spawn: spawnMock as unknown as typeof spawn, detectBinary, createClient: createClient as unknown as typeof createOpencodeClient, checkServerHealth, - platform: 'win32' - } - ) + platform: "win32", + }, + ); await expect(runtime.testConnection()).resolves.toMatchObject({ available: true, - detail: - '由 GoodBuddy 以当前用户权限管理本机 OpenCode 进程' - }) + detail: "由 GoodBuddy 以当前用户权限管理本机 OpenCode 进程", + }); expect(detectBinary).toHaveBeenCalledWith( - 'opencode', - 'C:\\Configured\\opencode.exe', - undefined - ) + "opencode", + "C:\\Configured\\opencode.exe", + undefined, + ); expect(spawnMock).toHaveBeenNthCalledWith( 1, - 'C:\\Tools\\opencode.exe', - [ - 'serve', - '--hostname=127.0.0.1', - expect.stringMatching(/^--port=\d+$/u) - ], + "C:\\Tools\\opencode.exe", + ["serve", "--hostname=127.0.0.1", expect.stringMatching(/^--port=\d+$/u)], expect.objectContaining({ cwd: process.cwd(), shell: false, - stdio: ['ignore', 'pipe', 'pipe'], + stdio: ["ignore", "pipe", "pipe"], windowsHide: true, env: expect.objectContaining({ - OPENCODE_CONFIG: resolve(configPath) - }) - }) - ) + OPENCODE_CONFIG: resolve(configPath), + }), + }), + ); const firstSpawn = spawnMock.mock.calls[0] as unknown as - | [string, string[]] - | undefined + [string, string[]] | undefined; const serverPort = firstSpawn?.[1] - ?.find((argument) => argument.startsWith('--port=')) - ?.slice('--port='.length) - expect(serverPort).toMatch(/^\d+$/u) - const serverUrl = `http://127.0.0.1:${serverPort}` + ?.find((argument) => argument.startsWith("--port=")) + ?.slice("--port=".length); + expect(serverPort).toMatch(/^\d+$/u); + const serverUrl = `http://127.0.0.1:${serverPort}`; const clientOptions = ( createClient.mock.calls as unknown as Array< [ { - baseUrl?: string - directory?: string - headers?: Record - } + baseUrl?: string; + directory?: string; + headers?: Record; + }, ] > )[0]?.[0] as | { - baseUrl?: string - directory?: string - headers?: Record + baseUrl?: string; + directory?: string; + headers?: Record; } - | undefined + | undefined; expect(clientOptions).toMatchObject({ baseUrl: serverUrl, directory: process.cwd(), headers: { - Authorization: expect.stringMatching(/^Basic /u) - } - }) + Authorization: expect.stringMatching(/^Basic /u), + }, + }); expect(checkServerHealth).toHaveBeenCalledWith( serverUrl, clientOptions?.headers?.Authorization, - expect.any(AbortSignal) - ) + expect.any(AbortSignal), + ); const spawnOptions = ( spawnMock.mock.calls as unknown as Array< [string, string[], { env?: NodeJS.ProcessEnv }] > - )[0]?.[2] as - | { env?: NodeJS.ProcessEnv } - | undefined - expect(spawnOptions?.env?.OPENCODE_SERVER_USERNAME).toBe( - 'goodbuddy' - ) - expect(spawnOptions?.env?.OPENCODE_SERVER_PASSWORD).toBeTruthy() - expect(spawnOptions?.env?.PATH).toBe( - 'C:\\GoodBuddy\\tools;C:\\Windows' - ) - expect(spawnOptions?.env).not.toHaveProperty( - 'ELECTRON_RUN_AS_NODE' - ) + )[0]?.[2] as { env?: NodeJS.ProcessEnv } | undefined; + expect(spawnOptions?.env?.OPENCODE_SERVER_USERNAME).toBe("goodbuddy"); + expect(spawnOptions?.env?.OPENCODE_SERVER_PASSWORD).toBeTruthy(); + expect(spawnOptions?.env?.PATH).toBe("C:\\GoodBuddy\\tools;C:\\Windows"); + expect(spawnOptions?.env).not.toHaveProperty("ELECTRON_RUN_AS_NODE"); expect( Buffer.from( - clientOptions?.headers?.Authorization?.slice(6) ?? '', - 'base64' - ).toString() - ).toBe( - `goodbuddy:${spawnOptions?.env?.OPENCODE_SERVER_PASSWORD}` - ) - expect(runtime.requiresToolApproval).toBe(false) + clientOptions?.headers?.Authorization?.slice(6) ?? "", + "base64", + ).toString(), + ).toBe(`goodbuddy:${spawnOptions?.env?.OPENCODE_SERVER_PASSWORD}`); + expect(runtime.requiresToolApproval).toBe(false); - await runtime.dispose() + await runtime.dispose(); expect(spawnMock).toHaveBeenNthCalledWith( 2, - 'taskkill.exe', - ['/PID', '314', '/T', '/F'], + "taskkill.exe", + ["/PID", "314", "/T", "/F"], { shell: false, - stdio: 'ignore', - windowsHide: true - } - ) - expect(killerChild.unref).toHaveBeenCalledOnce() - }) + stdio: "ignore", + windowsHide: true, + }, + ); + expect(killerChild.unref).toHaveBeenCalledOnce(); + }); - it('registers only assigned Skill packages in an isolated config directory', async () => { + it("registers only assigned Skill packages in an isolated config directory", async () => { const sourceRoot = await mkdtemp( - join(tmpdir(), 'goodbuddy-opencode-skill-source-') - ) - const skillDirectory = join(sourceRoot, 'longdoc-docx') - await mkdir(join(skillDirectory, 'templates'), { - recursive: true - }) + join(tmpdir(), "goodbuddy-opencode-skill-source-"), + ); + const skillDirectory = join(sourceRoot, "longdoc-docx"); + await mkdir(join(skillDirectory, "templates"), { + recursive: true, + }); await writeFile( - join(skillDirectory, 'SKILL.md'), + join(skillDirectory, "SKILL.md"), [ - '---', - 'id: longdoc-docx', - 'name: 长文档', - 'description: Build a DOCX', - '---', - '', - '# Long document' - ].join('\n'), - 'utf8' - ) + "---", + "id: longdoc-docx", + "name: 长文档", + "description: Build a DOCX", + "---", + "", + "# Long document", + ].join("\n"), + "utf8", + ); await writeFile( - join(skillDirectory, 'templates', 'document.txt'), - 'template', - 'utf8' - ) - const child = fakeChild() - const { deps, spawnMock } = dependencies(child) + join(skillDirectory, "templates", "document.txt"), + "template", + "utf8", + ); + const child = fakeChild(); + const { deps, spawnMock } = dependencies(child); const runtime = new OpenCodeRuntime( options({ skillPackages: [ { - id: 'longdoc-docx', - directory: skillDirectory - } - ] + id: "longdoc-docx", + directory: skillDirectory, + }, + ], }), - deps - ) + deps, + ); await expect(runtime.testConnection()).resolves.toMatchObject({ - available: true - }) + available: true, + }); const spawnOptions = spawnMock.mock.calls[0]?.[2] as - | { env?: NodeJS.ProcessEnv } - | undefined - const configDirectory = spawnOptions?.env?.OPENCODE_CONFIG_DIR - expect(configDirectory).toBeTruthy() - const registrationRoot = resolve(configDirectory!, '..') - const registeredSkill = join( - configDirectory!, - 'skills', - 'longdoc-docx' - ) + { env?: NodeJS.ProcessEnv } | undefined; + const configDirectory = spawnOptions?.env?.OPENCODE_CONFIG_DIR; + expect(configDirectory).toBeTruthy(); + const registrationRoot = resolve(configDirectory!, ".."); + const registeredSkill = join(configDirectory!, "skills", "longdoc-docx"); try { await expect( - readFile( - join(registeredSkill, 'templates', 'document.txt'), - 'utf8' - ) - ).resolves.toBe('template') + readFile(join(registeredSkill, "templates", "document.txt"), "utf8"), + ).resolves.toBe("template"); const registeredManifest = await readFile( - join(registeredSkill, 'SKILL.md'), - 'utf8' - ) - expect(registeredManifest).toContain('name: longdoc-docx') - expect(registeredManifest).not.toContain('id: longdoc-docx') + join(registeredSkill, "SKILL.md"), + "utf8", + ); + expect(registeredManifest).toContain("name: longdoc-docx"); + expect(registeredManifest).not.toContain("id: longdoc-docx"); const config = JSON.parse( - spawnOptions?.env?.OPENCODE_CONFIG_CONTENT ?? '{}' - ) as Record + spawnOptions?.env?.OPENCODE_CONFIG_CONTENT ?? "{}", + ) as Record; expect(config).toEqual({ skills: { - paths: [join(configDirectory!, 'skills')], - urls: [] + paths: [join(configDirectory!, "skills")], + urls: [], }, permission: { skill: { - '*': 'deny', - 'longdoc-docx': 'allow' - } - } - }) + "*": "deny", + "longdoc-docx": "allow", + }, + }, + }); expect(spawnOptions?.env).toMatchObject({ - OPENCODE_DISABLE_CLAUDE_CODE_SKILLS: '1', - OPENCODE_DISABLE_EXTERNAL_SKILLS: '1', - OPENCODE_DISABLE_PROJECT_CONFIG: '1', - XDG_CACHE_HOME: join(registrationRoot, 'xdg-cache'), - XDG_CONFIG_HOME: join(registrationRoot, 'xdg-config'), - XDG_DATA_HOME: join(registrationRoot, 'xdg-data'), - XDG_STATE_HOME: join(registrationRoot, 'xdg-state') - }) + OPENCODE_DISABLE_CLAUDE_CODE_SKILLS: "1", + OPENCODE_DISABLE_EXTERNAL_SKILLS: "1", + OPENCODE_DISABLE_PROJECT_CONFIG: "1", + XDG_CACHE_HOME: join(registrationRoot, "xdg-cache"), + XDG_CONFIG_HOME: join(registrationRoot, "xdg-config"), + XDG_DATA_HOME: join(registrationRoot, "xdg-data"), + XDG_STATE_HOME: join(registrationRoot, "xdg-state"), + }); } finally { - await runtime.dispose() - await rm(sourceRoot, { recursive: true, force: true }) + await runtime.dispose(); + await rm(sourceRoot, { recursive: true, force: true }); } - await expect(stat(registrationRoot)).rejects.toThrow() - }) + await expect(stat(registrationRoot)).rejects.toThrow(); + }); - it('reuses shared OpenCode config and content-addressed Skills across projects', async () => { + it("reuses shared OpenCode config and content-addressed Skills across projects", async () => { const sourceRoot = await mkdtemp( - join(tmpdir(), 'goodbuddy-opencode-shared-skill-source-') - ) + join(tmpdir(), "goodbuddy-opencode-shared-skill-source-"), + ); const sharedCacheRoot = await mkdtemp( - join(tmpdir(), 'goodbuddy-opencode-shared-cache-') - ) - const skillDirectory = join(sourceRoot, 'shared-skill') - const bundledConfigPath = join(sourceRoot, 'bundled-config') - await mkdir(skillDirectory, { recursive: true }) + join(tmpdir(), "goodbuddy-opencode-shared-cache-"), + ); + const skillDirectory = join(sourceRoot, "shared-skill"); + const bundledConfigPath = join(sourceRoot, "bundled-config"); + await mkdir(skillDirectory, { recursive: true }); await mkdir( - join( - bundledConfigPath, - 'node_modules', - '@opencode-ai', - 'plugin' - ), - { recursive: true } - ) + join(bundledConfigPath, "node_modules", "@opencode-ai", "plugin"), + { recursive: true }, + ); await writeFile( - join(bundledConfigPath, '.goodbuddy-ready.json'), + join(bundledConfigPath, ".goodbuddy-ready.json"), '{"version":"1.18.29"}\n', - 'utf8' - ) + "utf8", + ); await writeFile( join( bundledConfigPath, - 'node_modules', - '@opencode-ai', - 'plugin', - 'package.json' + "node_modules", + "@opencode-ai", + "plugin", + "package.json", ), '{"name":"@opencode-ai/plugin","version":"1.18.29"}\n', - 'utf8' - ) + "utf8", + ); await writeFile( - join(skillDirectory, 'SKILL.md'), + join(skillDirectory, "SKILL.md"), [ - '---', - 'name: Shared Skill', - 'description: Shared test Skill', - '---', - '', - '# Shared Skill' - ].join('\n'), - 'utf8' - ) + "---", + "name: Shared Skill", + "description: Shared test Skill", + "---", + "", + "# Shared Skill", + ].join("\n"), + "utf8", + ); const skillPackages = [ { - id: 'shared-skill', + id: "shared-skill", directory: skillDirectory, - digest: 'shared-skill-digest' - } - ] - const firstChild = fakeChild() - const first = dependencies(firstChild) + digest: "shared-skill-digest", + }, + ]; + const firstChild = fakeChild(); + const first = dependencies(firstChild); const firstRuntime = new OpenCodeRuntime( options({ sharedCacheRoot, skillPackages, - bundledConfigPath + bundledConfigPath, }), - first.deps - ) - const secondChild = fakeChild() - const second = dependencies(secondChild) + first.deps, + ); + const secondChild = fakeChild(); + const second = dependencies(secondChild); const secondRuntime = new OpenCodeRuntime( options({ sharedCacheRoot, skillPackages, - bundledConfigPath + bundledConfigPath, }), - second.deps - ) + second.deps, + ); try { await expect(firstRuntime.testConnection()).resolves.toMatchObject({ - available: true - }) - const firstEnvironment = first.spawnMock.mock.calls[0]?.[2]?.env - const firstConfigDirectory = - firstEnvironment?.OPENCODE_CONFIG_DIR ?? '' + available: true, + }); + const firstEnvironment = first.spawnMock.mock.calls[0]?.[2]?.env; + const firstConfigDirectory = firstEnvironment?.OPENCODE_CONFIG_DIR ?? ""; const firstConfig = JSON.parse( - firstEnvironment?.OPENCODE_CONFIG_CONTENT ?? '{}' - ) as { skills?: { paths?: string[] } } - const firstSkillsRoot = firstConfig.skills?.paths?.[0] ?? '' + firstEnvironment?.OPENCODE_CONFIG_CONTENT ?? "{}", + ) as { skills?: { paths?: string[] } }; + const firstSkillsRoot = firstConfig.skills?.paths?.[0] ?? ""; expect(firstConfigDirectory).toBe( - join(sharedCacheRoot, 'config', 'opencode') - ) + join(sharedCacheRoot, "config", "opencode"), + ); expect(firstEnvironment?.XDG_CONFIG_HOME).toBe( - join(sharedCacheRoot, 'config') - ) + join(sharedCacheRoot, "config"), + ); await expect( readFile( join( firstConfigDirectory, - 'node_modules', - '@opencode-ai', - 'plugin', - 'package.json' + "node_modules", + "@opencode-ai", + "plugin", + "package.json", ), - 'utf8' - ) - ).resolves.toContain('1.18.29') + "utf8", + ), + ).resolves.toContain("1.18.29"); expect(firstEnvironment?.XDG_CACHE_HOME).toBe( - join(sharedCacheRoot, 'cache') - ) + join(sharedCacheRoot, "cache"), + ); expect(firstEnvironment).toMatchObject({ - OPENCODE_EXPERIMENTAL_DISABLE_FILEWATCHER: '1' - }) - await firstRuntime.dispose() + OPENCODE_EXPERIMENTAL_DISABLE_FILEWATCHER: "1", + }); + await firstRuntime.dispose(); await expect( - readFile(join(firstSkillsRoot, 'shared-skill', 'SKILL.md'), 'utf8') - ).resolves.toContain('name: shared-skill') + readFile(join(firstSkillsRoot, "shared-skill", "SKILL.md"), "utf8"), + ).resolves.toContain("name: shared-skill"); await expect(secondRuntime.testConnection()).resolves.toMatchObject({ - available: true - }) - const secondEnvironment = second.spawnMock.mock.calls[0]?.[2]?.env + available: true, + }); + const secondEnvironment = second.spawnMock.mock.calls[0]?.[2]?.env; const secondConfig = JSON.parse( - secondEnvironment?.OPENCODE_CONFIG_CONTENT ?? '{}' - ) as { skills?: { paths?: string[] } } - expect(secondEnvironment?.OPENCODE_CONFIG_DIR).toBe( - firstConfigDirectory - ) - expect(secondConfig.skills?.paths?.[0]).toBe(firstSkillsRoot) + secondEnvironment?.OPENCODE_CONFIG_CONTENT ?? "{}", + ) as { skills?: { paths?: string[] } }; + expect(secondEnvironment?.OPENCODE_CONFIG_DIR).toBe(firstConfigDirectory); + expect(secondConfig.skills?.paths?.[0]).toBe(firstSkillsRoot); } finally { - await firstRuntime.dispose() - await secondRuntime.dispose() - await rm(sourceRoot, { recursive: true, force: true }) - await rm(sharedCacheRoot, { recursive: true, force: true }) + await firstRuntime.dispose(); + await secondRuntime.dispose(); + await rm(sourceRoot, { recursive: true, force: true }); + await rm(sharedCacheRoot, { recursive: true, force: true }); } - }) + }); - it('injects an independent model profile without persisting its key', async () => { - const child = fakeChild() - const { deps, spawnMock } = dependencies(child) + it("injects an independent model profile without persisting its key", async () => { + const child = fakeChild(); + const { deps, spawnMock } = dependencies(child); const runtime = new OpenCodeRuntime( options({ modelProfile: { - id: '00000000-0000-4000-8000-000000000011', - name: '独立模型', - baseUrl: 'https://model.example', - modelName: 'private-model', - apiKey: 'private-key', - protocol: 'anthropic-messages', - authentication: 'api-key', - supportsImageInput: true - } + id: "00000000-0000-4000-8000-000000000011", + name: "独立模型", + baseUrl: "https://model.example", + modelName: "private-model", + apiKey: "private-key", + protocol: "anthropic-messages", + authentication: "api-key", + supportsImageInput: true, + }, }), - deps - ) + deps, + ); await expect(runtime.testConnection()).resolves.toMatchObject({ - available: true - }) + available: true, + }); const spawnOptions = spawnMock.mock.calls[0]?.[2] as - | { env?: NodeJS.ProcessEnv } - | undefined + { env?: NodeJS.ProcessEnv } | undefined; const config = JSON.parse( - spawnOptions?.env?.OPENCODE_CONFIG_CONTENT ?? '{}' - ) as Record + spawnOptions?.env?.OPENCODE_CONFIG_CONTENT ?? "{}", + ) as Record; expect(config).toMatchObject({ - model: 'goodbuddy-anthropic/private-model', + model: "goodbuddy-anthropic/private-model", provider: { - 'goodbuddy-anthropic': { - npm: '@ai-sdk/anthropic', + "goodbuddy-anthropic": { + npm: "@ai-sdk/anthropic", options: { - apiKey: 'private-key', - baseURL: 'https://model.example/v1' + apiKey: "private-key", + baseURL: "https://model.example/v1", }, models: { - 'private-model': { + "private-model": { attachment: true, modalities: { - input: ['text', 'image'], - output: ['text'] + input: ["text", "image"], + output: ["text"], }, provider: { - npm: '@ai-sdk/anthropic' - } - } - } - } - } - }) - await runtime.dispose() - }) + npm: "@ai-sdk/anthropic", + }, + }, + }, + }, + }, + }); + await runtime.dispose(); + }); - it('isolates an explicit profile from unrelated inherited credentials', async () => { - const child = fakeChild() - const { deps, spawnMock } = dependencies(child) + it("isolates an explicit profile from unrelated inherited credentials", async () => { + const child = fakeChild(); + const { deps, spawnMock } = dependencies(child); const inheritedCredentials = { - ANTHROPIC_API_KEY: 'inherited-anthropic', - OPENAI_API_KEY: 'inherited-openai', - GOOGLE_GENERATIVE_AI_API_KEY: 'inherited-google', - GEMINI_API_KEY: 'inherited-gemini', - AWS_ACCESS_KEY_ID: 'inherited-aws-access', - AWS_SECRET_ACCESS_KEY: 'inherited-aws-secret', - AWS_SESSION_TOKEN: 'inherited-aws-session', - AWS_PROFILE: 'inherited-aws-profile', - OPENROUTER_API_KEY: 'inherited-openrouter' - } + ANTHROPIC_API_KEY: "inherited-anthropic", + OPENAI_API_KEY: "inherited-openai", + GOOGLE_GENERATIVE_AI_API_KEY: "inherited-google", + GEMINI_API_KEY: "inherited-gemini", + AWS_ACCESS_KEY_ID: "inherited-aws-access", + AWS_SECRET_ACCESS_KEY: "inherited-aws-secret", + AWS_SESSION_TOKEN: "inherited-aws-session", + AWS_PROFILE: "inherited-aws-profile", + OPENROUTER_API_KEY: "inherited-openrouter", + }; const previousEnvironment = Object.fromEntries( Object.keys(inheritedCredentials).map((name) => [ name, - process.env[name] - ]) - ) - Object.assign(process.env, inheritedCredentials) + process.env[name], + ]), + ); + Object.assign(process.env, inheritedCredentials); const runtime = new OpenCodeRuntime( options({ modelProfile: { - id: '00000000-0000-4000-8000-000000000014', - name: 'Explicit OpenAI profile', - baseUrl: 'https://model.example/v1', - modelName: 'private-model', - protocol: 'openai-responses', - authentication: 'api-key', - apiKey: 'selected-openai-key' - } + id: "00000000-0000-4000-8000-000000000014", + name: "Explicit OpenAI profile", + baseUrl: "https://model.example/v1", + modelName: "private-model", + protocol: "openai-responses", + authentication: "api-key", + apiKey: "selected-openai-key", + }, }), - deps - ) + deps, + ); try { await expect(runtime.testConnection()).resolves.toMatchObject({ - available: true - }) + available: true, + }); const environment = ( - spawnMock.mock.calls[0]?.[2] as - | { env?: NodeJS.ProcessEnv } - | undefined - )?.env - expect(environment?.OPENAI_API_KEY).toBe('selected-openai-key') + spawnMock.mock.calls[0]?.[2] as { env?: NodeJS.ProcessEnv } | undefined + )?.env; + expect(environment?.OPENAI_API_KEY).toBe("selected-openai-key"); for (const name of Object.keys(inheritedCredentials)) { - if (name !== 'OPENAI_API_KEY') { - expect(environment).not.toHaveProperty(name) + if (name !== "OPENAI_API_KEY") { + expect(environment).not.toHaveProperty(name); } } } finally { - await runtime.dispose() - for (const [name, value] of Object.entries( - previousEnvironment - )) { + await runtime.dispose(); + for (const [name, value] of Object.entries(previousEnvironment)) { if (value === undefined) { - delete process.env[name] + delete process.env[name]; } else { - process.env[name] = value + process.env[name] = value; } } } - }) + }); it.each([ { - label: 'Chat Completions', - protocol: 'openai-chat-completions' as const, - expectedPath: '/v1/chat/completions', - unexpectedPath: '/v1/responses' + label: "Chat Completions", + protocol: "openai-chat-completions" as const, + expectedPath: "/v1/chat/completions", + unexpectedPath: "/v1/responses", }, { - label: 'Responses', - protocol: 'openai-responses' as const, - expectedPath: '/v1/responses', - unexpectedPath: '/v1/chat/completions' - } + label: "Responses", + protocol: "openai-responses" as const, + expectedPath: "/v1/responses", + unexpectedPath: "/v1/chat/completions", + }, ])( - 'routes a custom-base $label profile through the bundled OpenCode provider', - async ({ - protocol, - expectedPath, - unexpectedPath - }) => { - const root = await mkdtemp( - join(tmpdir(), 'goodbuddy-opencode-routing-') - ) - const requestPaths: string[] = [] - const tenantHeaders: Array = [] - const requestBodies: Array> = [] + "routes a custom-base $label profile through the bundled OpenCode provider", + async ({ protocol, expectedPath, unexpectedPath }) => { + const root = await mkdtemp(join(tmpdir(), "goodbuddy-opencode-routing-")); + const requestPaths: string[] = []; + const tenantHeaders: Array = []; + const requestBodies: Array> = []; const server = createServer(async (request, response) => { - requestPaths.push(request.url ?? '') - tenantHeaders.push(request.headers['x-tenant-id']) - let body = '' + requestPaths.push(request.url ?? ""); + tenantHeaders.push(request.headers["x-tenant-id"]); + let body = ""; for await (const chunk of request) { - body += String(chunk) + body += String(chunk); } if (body) { - requestBodies.push( - JSON.parse(body) as Record - ) + requestBodies.push(JSON.parse(body) as Record); } response.writeHead(400, { - 'content-type': 'application/json' - }) + "content-type": "application/json", + }); response.end( JSON.stringify({ error: { - message: 'Intentional local routing probe' - } - }) - ) - }) + message: "Intentional local routing probe", + }, + }), + ); + }); await new Promise((resolveListen, reject) => { - server.once('error', reject) - server.listen(0, '127.0.0.1', () => resolveListen()) - }) - const address = server.address() - if (!address || typeof address === 'string') { - throw new Error('Failed to bind local routing probe') + server.once("error", reject); + server.listen(0, "127.0.0.1", () => resolveListen()); + }); + const address = server.address(); + if (!address || typeof address === "string") { + throw new Error("Failed to bind local routing probe"); } const isolatedEnvironment = { - APPDATA: join(root, 'appdata'), + APPDATA: join(root, "appdata"), HOME: root, - LOCALAPPDATA: join(root, 'localappdata'), - USERPROFILE: root - } as const + LOCALAPPDATA: join(root, "localappdata"), + USERPROFILE: root, + } as const; const previousEnvironment = Object.fromEntries( Object.keys(isolatedEnvironment).map((name) => [ name, - process.env[name] - ]) - ) - Object.assign(process.env, isolatedEnvironment) + process.env[name], + ]), + ); + Object.assign(process.env, isolatedEnvironment); const runtime = new OpenCodeRuntime( options({ binaryPath: join( process.cwd(), - 'node_modules', - 'opencode-ai', - 'bin', - 'opencode.exe' + "node_modules", + "opencode-ai", + "bin", + "opencode.exe", ), defaultWorkspace: root, modelProfile: { - id: '00000000-0000-4000-8000-000000000013', - name: 'Local endpoint probe', + id: "00000000-0000-4000-8000-000000000013", + name: "Local endpoint probe", baseUrl: `http://127.0.0.1:${address.port}/v1`, - modelName: 'probe-model', + modelName: "probe-model", protocol, - authentication: 'api-key', - apiKey: 'local-probe-key', + authentication: "api-key", + apiKey: "local-probe-key", requestHeaders: { - 'x-tenant-id': 'opencode-native' + "x-tenant-id": "opencode-native", }, requestBody: { - goodbuddy_unsupported_probe: 'must-not-send' - } - } + goodbuddy_unsupported_probe: "must-not-send", + }, + }, }), - { startupTimeoutMs: 20_000 } - ) - const controller = new AbortController() + { startupTimeoutMs: 20_000 }, + ); + const controller = new AbortController(); const timeout = setTimeout( - () => controller.abort(new Error('Routing probe timed out')), - 20_000 - ) + () => controller.abort(new Error("Routing probe timed out")), + 20_000, + ); try { - let failure = '' + let failure = ""; await (async () => { for await (const _event of runtime.run( { - requestId: - '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'routing-probe', - prompt: 'Reply with OK', - workMode: 'execute' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "routing-probe", + prompt: "Reply with OK", + workMode: "execute", }, - controller.signal + controller.signal, )) { // The local probe intentionally returns an upstream error. - void _event + void _event; } })().catch((error) => { - failure = - error instanceof Error ? error.message : String(error) - }) + failure = error instanceof Error ? error.message : String(error); + }); if (requestPaths.length === 0) { - throw new Error(`OpenCode routing probe failed: ${failure}`) + throw new Error(`OpenCode routing probe failed: ${failure}`); } - expect(requestPaths).toContain(expectedPath) - expect(requestPaths).not.toContain(unexpectedPath) - expect(tenantHeaders).toContain('opencode-native') + expect(requestPaths).toContain(expectedPath); + expect(requestPaths).not.toContain(unexpectedPath); + expect(tenantHeaders).toContain("opencode-native"); expect(requestBodies).not.toContainEqual( expect.objectContaining({ - goodbuddy_unsupported_probe: 'must-not-send' - }) - ) + goodbuddy_unsupported_probe: "must-not-send", + }), + ); } finally { - clearTimeout(timeout) - await runtime.dispose() - for (const [name, value] of Object.entries( - previousEnvironment - )) { + clearTimeout(timeout); + await runtime.dispose(); + for (const [name, value] of Object.entries(previousEnvironment)) { if (value === undefined) { - delete process.env[name] + delete process.env[name]; } else { - process.env[name] = value + process.env[name] = value; } } await new Promise((resolveClose, reject) => { - server.close((error) => - error ? reject(error) : resolveClose() - ) - }) - await rm(root, { recursive: true, force: true }) + server.close((error) => (error ? reject(error) : resolveClose())); + }); + await rm(root, { recursive: true, force: true }); } }, - 30_000 - ) + 30_000, + ); it.each([ { - protocol: 'openai-chat-completions' as const, - authentication: 'none' as const, - providerId: 'goodbuddy-openai-chat', - providerPackage: '@ai-sdk/openai-compatible' + protocol: "openai-chat-completions" as const, + authentication: "none" as const, + providerId: "goodbuddy-openai-chat", + providerPackage: "@ai-sdk/openai-compatible", }, { - protocol: 'openai-responses' as const, - authentication: 'api-key' as const, - providerId: 'goodbuddy-openai-responses', - providerPackage: '@ai-sdk/openai' - } + protocol: "openai-responses" as const, + authentication: "api-key" as const, + providerId: "goodbuddy-openai-responses", + providerPackage: "@ai-sdk/openai", + }, ])( - 'generates an explicit $protocol provider configuration', - async ({ - protocol, - authentication, - providerId, - providerPackage - }) => { - const child = fakeChild() - const { deps, spawnMock } = dependencies(child) + "generates an explicit $protocol provider configuration", + async ({ protocol, authentication, providerId, providerPackage }) => { + const child = fakeChild(); + const { deps, spawnMock } = dependencies(child); const runtime = new OpenCodeRuntime( options({ modelProfile: { - id: '00000000-0000-4000-8000-000000000012', - name: 'OpenAI 独立模型', - baseUrl: 'https://model.example/v1', - modelName: 'custom-model', + id: "00000000-0000-4000-8000-000000000012", + name: "OpenAI 独立模型", + baseUrl: "https://model.example/v1", + modelName: "custom-model", protocol, authentication, requestHeaders: { - 'x-tenant-id': 'opencode-tenant' + "x-tenant-id": "opencode-tenant", }, requestBody: { - temperature: 0.25 + temperature: 0.25, }, - ...(authentication === 'api-key' - ? { apiKey: 'private-key' } - : {}) - } + ...(authentication === "api-key" ? { apiKey: "private-key" } : {}), + }, }), - deps - ) + deps, + ); await expect(runtime.testConnection()).resolves.toMatchObject({ - available: true - }) + available: true, + }); const spawnOptions = spawnMock.mock.calls[0]?.[2] as - | { env?: NodeJS.ProcessEnv } - | undefined + { env?: NodeJS.ProcessEnv } | undefined; const config = JSON.parse( - spawnOptions?.env?.OPENCODE_CONFIG_CONTENT ?? '{}' + spawnOptions?.env?.OPENCODE_CONFIG_CONTENT ?? "{}", ) as { - model?: string + model?: string; provider?: Record< string, { - npm?: string - options?: Record - models?: Record< - string, - { provider?: { npm?: string } } - > + npm?: string; + options?: Record; + models?: Record; } - > - } - expect(config.model).toBe(`${providerId}/custom-model`) + >; + }; + expect(config.model).toBe(`${providerId}/custom-model`); expect(config.provider?.[providerId]).toMatchObject({ npm: providerPackage, options: { - baseURL: 'https://model.example/v1', + baseURL: "https://model.example/v1", headers: { - 'x-tenant-id': 'opencode-tenant' - } + "x-tenant-id": "opencode-tenant", + }, }, models: { - 'custom-model': { + "custom-model": { provider: { - npm: providerPackage - } - } - } - }) - if (authentication === 'api-key') { - expect( - config.provider?.[providerId]?.options?.apiKey - ).toBe('private-key') + npm: providerPackage, + }, + }, + }, + }); + if (authentication === "api-key") { + expect(config.provider?.[providerId]?.options?.apiKey).toBe( + "private-key", + ); } else { - expect( - config.provider?.[providerId]?.options - ).not.toHaveProperty('apiKey') + expect(config.provider?.[providerId]?.options).not.toHaveProperty( + "apiKey", + ); } - expect(JSON.stringify(config)).not.toContain('temperature') - await runtime.dispose() - } - ) + expect(JSON.stringify(config)).not.toContain("temperature"); + await runtime.dispose(); + }, + ); - it('isolates embedded server configuration from inherited env', async () => { - const child = fakeChild() - const { deps, spawnMock } = dependencies(child) + it("isolates embedded server configuration from inherited env", async () => { + const child = fakeChild(); + const { deps, spawnMock } = dependencies(child); const isolatedNames = [ - 'OPENCODE_CONFIG', - 'OPENCODE_CONFIG_CONTENT', - 'OPENCODE_CONFIG_DIR', - 'OPENCODE_SERVER_PASSWORD', - 'OPENCODE_SERVER_USERNAME' - ] as const + "OPENCODE_CONFIG", + "OPENCODE_CONFIG_CONTENT", + "OPENCODE_CONFIG_DIR", + "OPENCODE_SERVER_PASSWORD", + "OPENCODE_SERVER_USERNAME", + ] as const; const inherited = Object.fromEntries( - isolatedNames.map((name) => [name, process.env[name]]) - ) - const inheritedOtel = process.env.OTEL_EXPORTER_OTLP_ENDPOINT + isolatedNames.map((name) => [name, process.env[name]]), + ); + const inheritedOtel = process.env.OTEL_EXPORTER_OTLP_ENDPOINT; for (const name of isolatedNames) { - process.env[name] = 'must-not-be-inherited' + process.env[name] = "must-not-be-inherited"; } - process.env.OTEL_EXPORTER_OTLP_ENDPOINT = - 'https://telemetry.invalid' + process.env.OTEL_EXPORTER_OTLP_ENDPOINT = "https://telemetry.invalid"; try { - const runtime = new OpenCodeRuntime(options(), deps) + const runtime = new OpenCodeRuntime(options(), deps); await expect(runtime.testConnection()).resolves.toMatchObject({ - available: true - }) + available: true, + }); const spawnOptions = spawnMock.mock.calls[0]?.[2] as - | { env?: NodeJS.ProcessEnv } - | undefined - expect(spawnOptions?.env?.OPENCODE_CONFIG).toBeUndefined() - expect( - spawnOptions?.env?.OPENCODE_CONFIG_CONTENT - ).not.toBe('must-not-be-inherited') + { env?: NodeJS.ProcessEnv } | undefined; + expect(spawnOptions?.env?.OPENCODE_CONFIG).toBeUndefined(); + expect(spawnOptions?.env?.OPENCODE_CONFIG_CONTENT).not.toBe( + "must-not-be-inherited", + ); expect(spawnOptions?.env?.OPENCODE_CONFIG_DIR).not.toBe( - 'must-not-be-inherited' - ) - expect(spawnOptions?.env?.OPENCODE_SERVER_USERNAME).toBe( - 'goodbuddy' - ) - expect( - spawnOptions?.env?.OPENCODE_SERVER_PASSWORD - ).not.toBe('must-not-be-inherited') + "must-not-be-inherited", + ); + expect(spawnOptions?.env?.OPENCODE_SERVER_USERNAME).toBe("goodbuddy"); + expect(spawnOptions?.env?.OPENCODE_SERVER_PASSWORD).not.toBe( + "must-not-be-inherited", + ); expect(spawnOptions?.env).toMatchObject({ - DO_NOT_TRACK: '1', - OPENCODE_DISABLE_AUTOUPDATE: '1', - OPENCODE_DISABLE_CLAUDE_CODE_SKILLS: '1', - OPENCODE_DISABLE_EMBEDDED_WEB_UI: '1', - OPENCODE_DISABLE_EXTERNAL_SKILLS: '1', - OPENCODE_DISABLE_LSP_DOWNLOAD: '1', - OPENCODE_DISABLE_MODELS_FETCH: '1', - OPENCODE_DISABLE_PROJECT_CONFIG: '1', - OPENCODE_DISABLE_SHARE: '1', - OTEL_EXPORTER_OTLP_ENDPOINT: '', - OTEL_EXPORTER_OTLP_HEADERS: '', - OTEL_EXPORTER_OTLP_LOGS_ENDPOINT: '', - OTEL_EXPORTER_OTLP_METRICS_ENDPOINT: '', - OTEL_EXPORTER_OTLP_TRACES_ENDPOINT: '', - OTEL_SDK_DISABLED: 'true' - }) - await runtime.dispose() + DO_NOT_TRACK: "1", + OPENCODE_DISABLE_AUTOUPDATE: "1", + OPENCODE_DISABLE_CLAUDE_CODE_SKILLS: "1", + OPENCODE_DISABLE_EMBEDDED_WEB_UI: "1", + OPENCODE_DISABLE_EXTERNAL_SKILLS: "1", + OPENCODE_DISABLE_LSP_DOWNLOAD: "1", + OPENCODE_DISABLE_MODELS_FETCH: "1", + OPENCODE_DISABLE_PROJECT_CONFIG: "1", + OPENCODE_DISABLE_SHARE: "1", + OTEL_EXPORTER_OTLP_ENDPOINT: "", + OTEL_EXPORTER_OTLP_HEADERS: "", + OTEL_EXPORTER_OTLP_LOGS_ENDPOINT: "", + OTEL_EXPORTER_OTLP_METRICS_ENDPOINT: "", + OTEL_EXPORTER_OTLP_TRACES_ENDPOINT: "", + OTEL_SDK_DISABLED: "true", + }); + await runtime.dispose(); } finally { for (const name of isolatedNames) { - const value = inherited[name] + const value = inherited[name]; if (value === undefined) { - delete process.env[name] + delete process.env[name]; } else { - process.env[name] = value + process.env[name] = value; } } if (inheritedOtel === undefined) { - delete process.env.OTEL_EXPORTER_OTLP_ENDPOINT + delete process.env.OTEL_EXPORTER_OTLP_ENDPOINT; } else { - process.env.OTEL_EXPORTER_OTLP_ENDPOINT = inheritedOtel + process.env.OTEL_EXPORTER_OTLP_ENDPOINT = inheritedOtel; } } - }) + }); - it('polls the known loopback URL without waiting for stdout', async () => { - const child = fakeChild() + it("polls the known loopback URL without waiting for stdout", async () => { + const child = fakeChild(); const { deps, checkServerHealth, createClient } = dependencies(child, { - startupTimeoutMs: 1_000 - }) - checkServerHealth - .mockResolvedValueOnce(false) - .mockResolvedValueOnce(true) - const runtime = new OpenCodeRuntime(options(), deps) + startupTimeoutMs: 1_000, + }); + checkServerHealth.mockResolvedValueOnce(false).mockResolvedValueOnce(true); + const runtime = new OpenCodeRuntime(options(), deps); await expect(runtime.testConnection()).resolves.toMatchObject({ - available: true - }) - expect(checkServerHealth).toHaveBeenCalledTimes(2) - const [url, authorization, signal] = - checkServerHealth.mock.calls[0] ?? [] - expect(url).toMatch(/^http:\/\/127\.0\.0\.1:\d+$/) - expect(authorization).toMatch(/^Basic /) - expect(signal).toBeInstanceOf(AbortSignal) + available: true, + }); + expect(checkServerHealth).toHaveBeenCalledTimes(2); + const [url, authorization, signal] = checkServerHealth.mock.calls[0] ?? []; + expect(url).toMatch(/^http:\/\/127\.0\.0\.1:\d+$/); + expect(authorization).toMatch(/^Basic /); + expect(signal).toBeInstanceOf(AbortSignal); expect(createClient).toHaveBeenCalledWith({ baseUrl: url, directory: process.cwd(), - headers: { Authorization: authorization } - }) - }) + headers: { Authorization: authorization }, + }); + }); - it('allows ARM-class startup to become healthy after ten seconds', async () => { - vi.useFakeTimers() + it("keeps a healthy cached server and client after idle", async () => { + const child = fakeChild(); + const { deps, checkServerHealth, createClient, spawnMock } = dependencies(child); + const runtime = new OpenCodeRuntime(options(), deps); try { - const child = fakeChild() - let healthy = false - const checkServerHealth = vi.fn(async () => healthy) - const { deps } = dependencies(child, { - checkServerHealth - }) - const runtime = new OpenCodeRuntime(options(), deps) - let settled = false - const statusPromise = runtime.testConnection().finally(() => { - settled = true - }) - - await vi.advanceTimersByTimeAsync(10_100) - expect(settled).toBe(false) - - healthy = true - await vi.advanceTimersByTimeAsync(100) - await expect(statusPromise).resolves.toMatchObject({ - available: true - }) - expect(checkServerHealth).toHaveBeenCalled() - - await runtime.dispose() + await runtime.testConnection(); + await expect(runtime.testConnection()).resolves.toMatchObject({ available: true }); + expect(checkServerHealth).toHaveBeenCalledTimes(2); + expect(createClient).toHaveBeenCalledOnce(); + expect(spawnMock).toHaveBeenCalledOnce(); + expect(child.kill).not.toHaveBeenCalled(); } finally { - vi.useRealTimers() + await runtime.dispose(); } - }) + }); - it('times out, terminates the process tree, and does not expose stderr', async () => { - const child = fakeChild() - const secret = 'private-config-token' + it.each(["unhealthy", "rejected", "hung"])( + "replaces a %s cached server once for concurrent callers", + async (failure) => { + const child = fakeChild(); + const replacement = fakeChild(43); + const { deps, checkServerHealth, createClient, spawnMock } = dependencies(child); + spawnMock.mockReturnValueOnce(child).mockReturnValue(replacement); + const runtime = new OpenCodeRuntime(options(), deps); + try { + await runtime.testConnection(); + if (failure === "hung") { + checkServerHealth.mockImplementationOnce(() => new Promise(() => {})); + } else if (failure === "rejected") { + checkServerHealth.mockRejectedValueOnce(new Error("fetch failed")); + } else { + checkServerHealth.mockResolvedValueOnce(false); + } + const statuses = await Promise.all([ + runtime.testConnection(), + runtime.testConnection(), + ]); + expect(statuses.every((status) => status.available)).toBe(true); + expect(child.kill).toHaveBeenCalledOnce(); + expect(replacement.kill).not.toHaveBeenCalled(); + expect(spawnMock).toHaveBeenCalledTimes(2); + expect(createClient).toHaveBeenCalledTimes(2); + expect(checkServerHealth).toHaveBeenCalledTimes(3); + } finally { + await runtime.dispose(); + } + }, + ); + + it("creates a new session after replacing a stale server without replaying prompts", async () => { + const harness = runClient([ + { id: "idle", type: "session.idle", properties: { sessionID: "session-1" } }, + ]); + const child = fakeChild(); + const { deps, checkServerHealth, spawnMock } = dependencies(child, { + createClient: () => harness.client, + }); + spawnMock.mockReturnValueOnce(child).mockReturnValue(fakeChild(43)); + const runtime = new OpenCodeRuntime(options(), deps); + try { + await collectRun(runtime, "execute"); + checkServerHealth.mockResolvedValueOnce(false); + await collectRun(runtime, "execute"); + expect(harness.session.create).toHaveBeenCalledTimes(2); + expect(harness.session.update).not.toHaveBeenCalled(); + expect(harness.session.promptAsync).toHaveBeenCalledTimes(2); + } finally { + await runtime.dispose(); + } + }); + + it("isolates cancellation while concurrent requests wait for a cached health check", async () => { + const setup = runClient([ + { id: "idle", type: "session.idle", properties: { sessionID: "session-1" } }, + ]); + const child = fakeChild(); + const { deps, checkServerHealth, spawnMock } = dependencies(child, { + createClient: () => setup.client, + }); + const runtime = new OpenCodeRuntime(options(), deps); + let resolveHealth!: (healthy: boolean) => void; + const health = new Promise((resolve) => { resolveHealth = resolve; }); + const runs: Promise[] = []; + try { + await runtime.testConnection(); + const controller = new AbortController(); + checkServerHealth.mockReturnValueOnce(health); + const stream = runtime.run( + { requestId: "cancelled", conversationId: "conversation-cancelled", prompt: "test" }, + controller.signal, + ); + const cancelled = stream.next().catch((error: unknown) => error); + runs.push(cancelled); + await vi.waitFor(() => expect(checkServerHealth).toHaveBeenCalledTimes(2)); + const waiting = collectRun(runtime); + runs.push(waiting); + // Let the second request acquire its conversation lock and join the probe. + await new Promise((resolve) => setImmediate(resolve)); + controller.abort(new Error("cancelled")); + await expect(cancelled).resolves.toMatchObject({ message: "cancelled" }); + expect(checkServerHealth.mock.calls[1]![2].aborted).toBe(false); + // A third caller must still join the pending probe after A exits. + const third = runtime.testConnection(); + runs.push(third); + expect(checkServerHealth).toHaveBeenCalledTimes(2); + resolveHealth(true); + await expect(waiting).resolves.toContainEqual(expect.objectContaining({ type: "done" })); + await expect(third).resolves.toMatchObject({ available: true }); + expect(child.kill).not.toHaveBeenCalled(); + expect(spawnMock).toHaveBeenCalledOnce(); + expect(checkServerHealth).toHaveBeenCalledTimes(2); + } finally { + resolveHealth(true); + await Promise.allSettled(runs); + await runtime.dispose(); + } + }); + + it("allows ARM-class startup to become healthy after ten seconds", async () => { + vi.useFakeTimers(); + try { + const child = fakeChild(); + let healthy = false; + const checkServerHealth = vi.fn(async () => healthy); + const { deps } = dependencies(child, { + checkServerHealth, + }); + const runtime = new OpenCodeRuntime(options(), deps); + let settled = false; + const statusPromise = runtime.testConnection().finally(() => { + settled = true; + }); + + await vi.advanceTimersByTimeAsync(10_100); + expect(settled).toBe(false); + + healthy = true; + await vi.advanceTimersByTimeAsync(100); + await expect(statusPromise).resolves.toMatchObject({ + available: true, + }); + expect(checkServerHealth).toHaveBeenCalled(); + + await runtime.dispose(); + } finally { + vi.useRealTimers(); + } + }); + + it("times out, terminates the process tree, and does not expose stderr", async () => { + const child = fakeChild(); + const secret = "private-config-token"; const { deps } = dependencies(child, { startupTimeoutMs: 5, - checkServerHealth: vi.fn().mockResolvedValue(false) - }) - stderrOf(child).write(secret) - const runtime = new OpenCodeRuntime(options(), deps) + checkServerHealth: vi.fn().mockResolvedValue(false), + }); + stderrOf(child).write(secret); + const runtime = new OpenCodeRuntime(options(), deps); - const status = await runtime.testConnection() + const status = await runtime.testConnection(); expect(status).toMatchObject({ available: false, - detail: 'OpenCode Server 启动超时（30 秒）' - }) - expect(status.detail).not.toContain(secret) - expect(child.kill).toHaveBeenCalledWith('SIGTERM') - }) + detail: "OpenCode Server 启动超时（30 秒）", + }); + expect(status.detail).not.toContain(secret); + expect(child.kill).toHaveBeenCalledWith("SIGTERM"); + }); - it('reports early exit without leaking captured stderr', async () => { - const child = fakeChild() - const secret = 'OPENCODE_CONFIG=/secret/config.json' - let registrationRoot = '' + it("reports early exit without leaking captured stderr", async () => { + const child = fakeChild(); + const secret = "OPENCODE_CONFIG=/secret/config.json"; + let registrationRoot = ""; const { deps } = dependencies(child, { spawn: vi.fn( ( _command: string, _args: string[], - spawnOptions: { env?: NodeJS.ProcessEnv } + spawnOptions: { env?: NodeJS.ProcessEnv }, ) => { registrationRoot = resolve( - spawnOptions.env?.OPENCODE_CONFIG_DIR ?? '', - '..' - ) + spawnOptions.env?.OPENCODE_CONFIG_DIR ?? "", + "..", + ); queueMicrotask(() => { - stderrOf(child).write(secret) - closeChild(child, 9) - }) - return child - } - ) as unknown as typeof spawn - }) - const runtime = new OpenCodeRuntime(options(), deps) + stderrOf(child).write(secret); + closeChild(child, 9); + }); + return child; + }, + ) as unknown as typeof spawn, + }); + const runtime = new OpenCodeRuntime(options(), deps); - const status = await runtime.testConnection() + const status = await runtime.testConnection(); - expect(status.detail).toBe('OpenCode Server 启动前退出（code 9）') - expect(status.detail).not.toContain(secret) - expect(registrationRoot).toBeTruthy() - await expect(stat(registrationRoot)).rejects.toThrow() - }) + expect(status.detail).toBe("OpenCode Server 启动前退出（code 9）"); + expect(status.detail).not.toContain(secret); + expect(registrationRoot).toBeTruthy(); + await expect(stat(registrationRoot)).rejects.toThrow(); + }); - it('terminates startup when the request is aborted', async () => { - const child = fakeChild() + it("terminates startup when the request is aborted", async () => { + const child = fakeChild(); const { deps, spawnMock } = dependencies(child, { - checkServerHealth: vi.fn().mockResolvedValue(false) - }) - const runtime = new OpenCodeRuntime(options(), deps) - const controller = new AbortController() + checkServerHealth: vi.fn().mockResolvedValue(false), + }); + const runtime = new OpenCodeRuntime(options(), deps); + const controller = new AbortController(); const stream = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'test' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "test", }, - controller.signal - ) + controller.signal, + ); - const pending = stream.next() - await vi.waitFor(() => expect(spawnMock).toHaveBeenCalledOnce()) - controller.abort(new Error('sensitive abort reason')) + const pending = stream.next(); + await vi.waitFor(() => expect(spawnMock).toHaveBeenCalledOnce()); + controller.abort(new Error("sensitive abort reason")); - await expect(pending).rejects.toThrow('OpenCode Server 启动已取消') - expect(child.kill).toHaveBeenCalledWith('SIGTERM') - }) + await expect(pending).rejects.toThrow("OpenCode Server 启动已取消"); + expect(child.kill).toHaveBeenCalledWith("SIGTERM"); + }); - it('bounds session setup without imposing a full-run deadline', async () => { - const harness = runClient([]) - ;( + it("bounds session setup without imposing a full-run deadline", async () => { + const harness = runClient([]); + ( harness.session.create as unknown as ReturnType - ).mockImplementation( - () => new Promise(() => undefined) - ) + ).mockImplementation(() => new Promise(() => undefined)); const runtime = embeddedRuntime( harness.client, {}, - { controlRequestTimeoutMs: 5 } - ) + { controlRequestTimeoutMs: 5 }, + ); - await expect(collectRun(runtime)).rejects.toThrow( - 'OpenCode 创建会话超时' - ) - await runtime.dispose() - }) + await expect(collectRun(runtime)).rejects.toThrow("OpenCode 创建会话超时"); + await runtime.dispose(); + }); - it('keeps external baseUrl mode free of binary detection and spawning', async () => { - const child = fakeChild() - const { deps, spawnMock, detectBinary, createClient } = dependencies(child) + it("keeps external baseUrl mode free of binary detection and spawning", async () => { + const child = fakeChild(); + const { deps, spawnMock, detectBinary, createClient } = dependencies(child); const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', - embedded: false + baseUrl: "http://127.0.0.1:4096", + embedded: false, }), - deps - ) + deps, + ); await expect(runtime.getStatus()).resolves.toMatchObject({ available: true, - detail: '已连接 http://127.0.0.1:4096' - }) - expect(detectBinary).not.toHaveBeenCalled() - expect(spawnMock).not.toHaveBeenCalled() + detail: "已连接 http://127.0.0.1:4096", + }); + expect(detectBinary).not.toHaveBeenCalled(); + expect(spawnMock).not.toHaveBeenCalled(); expect(createClient).toHaveBeenCalledWith({ - baseUrl: 'http://127.0.0.1:4096', - directory: process.cwd() - }) - expect(runtime.requiresToolApproval).toBe(false) - }) + baseUrl: "http://127.0.0.1:4096", + directory: process.cwd(), + }); + expect(runtime.requiresToolApproval).toBe(false); + }); - it.each(['completed', 'failed', 'cancelled', 'consumer-return'])( - 'closes the event subscription when a run is %s', + it.each(["completed", "failed", "cancelled", "consumer-return"])( + "closes the event subscription when a run is %s", async (outcome) => { - const setup = runClient([]) - const controller = new AbortController() - let subscriptionSignal: AbortSignal | undefined - const subscribe = setup.event.subscribe as unknown as ReturnType - subscribe.mockImplementation(async (_input, options: { signal: AbortSignal }) => { - subscriptionSignal = options.signal - return { - stream: (async function* () { - if (outcome === 'cancelled') { - controller.abort(new DOMException('Cancelled', 'AbortError')) - return - } - if (outcome === 'failed') return - yield { - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - })() - } - }) + const setup = runClient([]); + const controller = new AbortController(); + let subscriptionSignal: AbortSignal | undefined; + const subscribe = setup.event.subscribe as unknown as ReturnType< + typeof vi.fn + >; + subscribe.mockImplementation( + async (_input, options: { signal: AbortSignal }) => { + subscriptionSignal = options.signal; + return { + stream: (async function* () { + if (outcome === "cancelled") { + controller.abort(new DOMException("Cancelled", "AbortError")); + return; + } + if (outcome === "failed") return; + yield { + type: "session.idle", + properties: { sessionID: "session-1" }, + }; + })(), + }; + }, + ); const runtime = new OpenCodeRuntime( - options({ embedded: false, baseUrl: 'http://127.0.0.1:4096' }), - { createClient: (() => setup.client) as typeof createOpencodeClient } - ) + options({ embedded: false, baseUrl: "http://127.0.0.1:4096" }), + { createClient: (() => setup.client) as typeof createOpencodeClient }, + ); const collect = async (): Promise => { - for await (const event of runtime.run({ - requestId: crypto.randomUUID(), - conversationId: crypto.randomUUID(), - prompt: 'test', - workMode: 'execute' - }, controller.signal)) { - if (outcome === 'consumer-return' && event.type === 'done') break + for await (const event of runtime.run( + { + requestId: crypto.randomUUID(), + conversationId: crypto.randomUUID(), + prompt: "test", + workMode: "execute", + }, + controller.signal, + )) { + if (outcome === "consumer-return" && event.type === "done") break; } - } + }; try { - if (outcome === 'failed') { - await expect(collect()).rejects.toThrow('事件流意外结束') - } else if (outcome === 'cancelled') { - await expect(collect()).rejects.toMatchObject({ name: 'AbortError' }) + if (outcome === "failed") { + await expect(collect()).rejects.toThrow("事件流意外结束"); + } else if (outcome === "cancelled") { + await expect(collect()).rejects.toMatchObject({ name: "AbortError" }); } else { - await collect() + await collect(); } - expect(subscriptionSignal?.aborted).toBe(true) - expect(controller.signal.aborted).toBe(outcome === 'cancelled') + expect(subscriptionSignal?.aborted).toBe(true); + expect(controller.signal.aborted).toBe(outcome === "cancelled"); } finally { - await runtime.dispose() + await runtime.dispose(); } - } - ) + }, + ); - it('serializes external runs that share one conversation session', async () => { - const child = fakeChild() - let releaseFirst!: () => void + it("serializes external runs that share one conversation session", async () => { + const child = fakeChild(); + let releaseFirst!: () => void; const firstGate = new Promise((resolve) => { - releaseFirst = resolve - }) - let subscriptionCount = 0 + releaseFirst = resolve; + }); + let subscriptionCount = 0; const promptAsync = vi.fn().mockResolvedValue({ data: true, - error: undefined - }) + error: undefined, + }); const client = { session: { create: vi.fn().mockResolvedValue({ - data: { id: 'session-1' }, - error: undefined + data: { id: "session-1" }, + error: undefined, }), update: vi.fn().mockResolvedValue({ - data: { id: 'session-1' }, - error: undefined + data: { id: "session-1" }, + error: undefined, }), promptAsync, abort: vi.fn().mockResolvedValue({ data: true, - error: undefined - }) + error: undefined, + }), }, event: { subscribe: vi.fn().mockImplementation(async () => { - subscriptionCount += 1 - const current = subscriptionCount + subscriptionCount += 1; + const current = subscriptionCount; return { stream: (async function* () { if (current === 1) { - await firstGate + await firstGate; } yield { - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - })() - } - }) + type: "session.idle", + properties: { sessionID: "session-1" }, + }; + })(), + }; + }), }, tool: { ids: vi.fn().mockResolvedValue({ data: [], - error: undefined - }) - } - } as unknown as ReturnType + error: undefined, + }), + }, + } as unknown as ReturnType; const { deps } = dependencies(child, { createClient: vi.fn( - () => client - ) as unknown as typeof createOpencodeClient - }) + () => client, + ) as unknown as typeof createOpencodeClient, + }); const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', - embedded: false + baseUrl: "http://127.0.0.1:4096", + embedded: false, }), - deps - ) + deps, + ); const request = { - requestId: '00000000-0000-4000-8000-000000000101', - conversationId: 'shared-conversation', - prompt: 'first', - workMode: 'execute' as const - } + requestId: "00000000-0000-4000-8000-000000000101", + conversationId: "shared-conversation", + prompt: "first", + workMode: "execute" as const, + }; const collect = async ( - stream: AsyncGenerator + stream: AsyncGenerator, ): Promise => { - const events: RuntimeEvent[] = [] + const events: RuntimeEvent[] = []; for await (const event of stream) { - events.push(event) + events.push(event); } - return events - } - const first = collect(runtime.run( - request, - new AbortController().signal - )) - await vi.waitFor(() => expect(promptAsync).toHaveBeenCalledTimes(1)) + return events; + }; + const first = collect(runtime.run(request, new AbortController().signal)); + await vi.waitFor(() => expect(promptAsync).toHaveBeenCalledTimes(1)); const second = collect( runtime.run( { ...request, - requestId: '00000000-0000-4000-8000-000000000102', - prompt: 'second' + requestId: "00000000-0000-4000-8000-000000000102", + prompt: "second", }, - new AbortController().signal - ) - ) - await Promise.resolve() - expect(promptAsync).toHaveBeenCalledTimes(1) + new AbortController().signal, + ), + ); + await Promise.resolve(); + expect(promptAsync).toHaveBeenCalledTimes(1); - releaseFirst() - await first - await second - expect(promptAsync).toHaveBeenCalledTimes(2) - await runtime.dispose() - }) + releaseFirst(); + await first; + await second; + expect(promptAsync).toHaveBeenCalledTimes(2); + await runtime.dispose(); + }); - it('loads assigned Skills before prompting', async () => { - const child = fakeChild() - const promptAsync = vi.fn().mockResolvedValue({ error: undefined }) + it("loads assigned Skills before prompting", async () => { + const child = fakeChild(); + const promptAsync = vi.fn().mockResolvedValue({ error: undefined }); const client = { session: { - create: vi.fn().mockResolvedValue({ data: { id: 'session-1' } }), + create: vi.fn().mockResolvedValue({ data: { id: "session-1" } }), promptAsync, - abort: vi.fn().mockResolvedValue(undefined) + abort: vi.fn().mockResolvedValue(undefined), }, event: { subscribe: vi.fn().mockResolvedValue({ stream: (async function* () { yield { - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - })() - }) + type: "session.idle", + properties: { sessionID: "session-1" }, + }; + })(), + }), }, tool: { ids: vi.fn().mockResolvedValue({ - data: ['read', 'write', 'goodbuddy-mcp'], - error: undefined - }) - } - } as unknown as ReturnType + data: ["read", "write", "goodbuddy-mcp"], + error: undefined, + }), + }, + } as unknown as ReturnType; const { deps } = dependencies(child, { createClient: vi.fn( - () => client - ) as unknown as typeof createOpencodeClient - }) + () => client, + ) as unknown as typeof createOpencodeClient, + }); const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', + baseUrl: "http://127.0.0.1:4096", embedded: false, - skillInstructions: '# 文档写作' + skillInstructions: "# 文档写作", }), - deps - ) + deps, + ); - const events = [] + const events = []; for await (const event of runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'test', - workMode: 'execute', + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "test", + workMode: "execute", images: [ { - name: 'screenshot.png', - mediaType: 'image/png', - data: 'aW1hZ2U=' - } - ] + name: "screenshot.png", + mediaType: "image/png", + data: "aW1hZ2U=", + }, + ], }, - new AbortController().signal + new AbortController().signal, )) { - events.push(event) + events.push(event); } expect(promptAsync).toHaveBeenCalledWith( expect.objectContaining({ - system: '# 文档写作', + system: "# 文档写作", parts: [ - { type: 'text', text: 'test' }, + { type: "text", text: "test" }, { - type: 'file', - mime: 'image/png', - filename: 'screenshot.png', - url: 'data:image/png;base64,aW1hZ2U=' - } - ] + type: "file", + mime: "image/png", + filename: "screenshot.png", + url: "data:image/png;base64,aW1hZ2U=", + }, + ], }), expect.objectContaining({ - signal: expect.any(AbortSignal) - }) - ) - expect(events.at(-1)).toMatchObject({ type: 'done' }) - await runtime.dispose() - }) + signal: expect.any(AbortSignal), + }), + ); + expect(events.at(-1)).toMatchObject({ type: "done" }); + await runtime.dispose(); + }); - it('rejects images when the explicit model connection disables image input', async () => { - const child = fakeChild() - const { deps, createClient } = dependencies(child) + it("rejects images when the explicit model connection disables image input", async () => { + const child = fakeChild(); + const { deps, createClient } = dependencies(child); const runtime = new OpenCodeRuntime( options({ modelProfile: { - id: '00000000-0000-4000-8000-000000000011', - name: '文本模型', - baseUrl: 'https://model.example', - modelName: 'text-model', - protocol: 'anthropic-messages', - authentication: 'none', - supportsImageInput: false - } + id: "00000000-0000-4000-8000-000000000011", + name: "文本模型", + baseUrl: "https://model.example", + modelName: "text-model", + protocol: "anthropic-messages", + authentication: "none", + supportsImageInput: false, + }, }), - deps - ) + deps, + ); const stream = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'describe', + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "describe", images: [ { - name: 'screenshot.png', - mediaType: 'image/png', - data: 'aW1hZ2U=' - } - ] + name: "screenshot.png", + mediaType: "image/png", + data: "aW1hZ2U=", + }, + ], }, - new AbortController().signal - ) + new AbortController().signal, + ); - await expect(stream.next()).rejects.toThrow( - '当前模型连接未启用图像输入' - ) - expect(createClient).not.toHaveBeenCalled() - }) -}) + await expect(stream.next()).rejects.toThrow("当前模型连接未启用图像输入"); + expect(createClient).not.toHaveBeenCalled(); + }); +}); -describe('OpenCodeRuntime embedded permission mediation', () => { - it('shares assigned custom MCP only with embedded Execute through a scoped loopback token', async () => { +describe("OpenCodeRuntime embedded permission mediation", () => { + it("shares assigned custom MCP only with embedded Execute through a scoped loopback token", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const gateway = { - getEndpoint: vi.fn(() => 'http://127.0.0.1:4567/mcp'), - grantCustomMcp: vi.fn(() => 'custom-capability'), + getEndpoint: vi.fn(() => "http://127.0.0.1:4567/mcp"), + grantCustomMcp: vi.fn(() => "custom-capability"), prepareCustomMcpTools: vi.fn(async () => [ { - name: 'mcp_12345678_abcdef01_private_tool', - inputSchema: { type: 'object' } - } + name: "mcp_12345678_abcdef01_private_tool", + inputSchema: { type: "object" }, + }, ]), - revoke: vi.fn() - } as unknown as KnowledgeMcpGateway + revoke: vi.fn(), + } as unknown as KnowledgeMcpGateway; const runtime = embeddedRuntime(setup.client, { knowledgeGateway: gateway, mcpServers: [ { - id: '00000000-0000-4000-8000-000000000092', - name: 'Private MCP', - description: '', + id: "00000000-0000-4000-8000-000000000092", + name: "Private MCP", + description: "", enabled: true, allowDynamicTools: false, - assignments: ['opencode'], + assignments: ["opencode"], secretConfigured: true, - secret: 'must-stay-in-main', - transport: 'http', - url: 'https://private.example/mcp' - } - ] - }) + secret: "must-stay-in-main", + transport: "http", + url: "https://private.example/mcp", + }, + ], + }); - await collectRun(runtime, 'execute') + await collectRun(runtime, "execute"); expect(gateway.grantCustomMcp).toHaveBeenCalledWith( - '3f496642-f47d-4e0a-8944-a32c77b0d6ef', + "3f496642-f47d-4e0a-8944-a32c77b0d6ef", expect.any(Array), - expect.any(AbortSignal) - ) + expect.any(AbortSignal), + ); expect(setup.client.mcp.add).toHaveBeenCalledWith( { directory: process.cwd(), name: expect.stringMatching(/^goodbuddy-custom-[a-f0-9]{20}$/u), config: { - type: 'remote', - url: 'http://127.0.0.1:4567/mcp', + type: "remote", + url: "http://127.0.0.1:4567/mcp", enabled: true, headers: { - Authorization: 'Bearer custom-capability' + Authorization: "Bearer custom-capability", }, - oauth: false - } + oauth: false, + }, }, - { signal: expect.any(AbortSignal) } - ) - expect(JSON.stringify( - (setup.client.mcp.add as unknown as ReturnType) - .mock.calls - )).not.toContain('must-stay-in-main') - expect(JSON.stringify( - (setup.client.mcp.add as unknown as ReturnType) - .mock.calls - )).not.toContain('private.example') - expect(gateway.revoke).toHaveBeenCalledWith('custom-capability') - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ); + expect( + JSON.stringify( + (setup.client.mcp.add as unknown as ReturnType).mock + .calls, + ), + ).not.toContain("must-stay-in-main"); + expect( + JSON.stringify( + (setup.client.mcp.add as unknown as ReturnType).mock + .calls, + ), + ).not.toContain("private.example"); + expect(gateway.revoke).toHaveBeenCalledWith("custom-capability"); + await runtime.dispose(); + }); - it.each([ - ['ask', true] as const, - ['execute', false] as const - ])( - 'does not share custom MCP with OpenCode in %s mode when embedded is %s', + it.each([["ask", true] as const, ["execute", false] as const])( + "does not share custom MCP with OpenCode in %s mode when embedded is %s", async (workMode, embedded) => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const gateway = { - getEndpoint: vi.fn(() => 'http://127.0.0.1:4567/mcp'), - grantCustomMcp: vi.fn(() => 'custom-capability'), + getEndpoint: vi.fn(() => "http://127.0.0.1:4567/mcp"), + grantCustomMcp: vi.fn(() => "custom-capability"), prepareCustomMcpTools: vi.fn(async () => []), - revoke: vi.fn() - } as unknown as KnowledgeMcpGateway + revoke: vi.fn(), + } as unknown as KnowledgeMcpGateway; const runtime = embedded ? embeddedRuntime(setup.client, { knowledgeGateway: gateway, mcpServers: [ { - id: '00000000-0000-4000-8000-000000000093', - name: 'Private MCP', - description: '', + id: "00000000-0000-4000-8000-000000000093", + name: "Private MCP", + description: "", enabled: true, allowDynamicTools: false, - assignments: ['opencode'], + assignments: ["opencode"], secretConfigured: false, - transport: 'stdio', - command: 'private-command', - args: [] - } - ] + transport: "stdio", + command: "private-command", + args: [], + }, + ], }) : new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', + baseUrl: "http://127.0.0.1:4096", embedded: false, knowledgeGateway: gateway, mcpServers: [ { - id: '00000000-0000-4000-8000-000000000093', - name: 'Private MCP', - description: '', + id: "00000000-0000-4000-8000-000000000093", + name: "Private MCP", + description: "", enabled: true, allowDynamicTools: false, - assignments: ['opencode'], + assignments: ["opencode"], secretConfigured: false, - transport: 'stdio', - command: 'private-command', - args: [] - } - ] + transport: "stdio", + command: "private-command", + args: [], + }, + ], }), dependencies(fakeChild(), { createClient: vi.fn( - () => setup.client - ) as unknown as typeof createOpencodeClient - }).deps - ) + () => setup.client, + ) as unknown as typeof createOpencodeClient, + }).deps, + ); - await collectRun(runtime, workMode) + await collectRun(runtime, workMode); - expect(gateway.grantCustomMcp).not.toHaveBeenCalled() - expect(setup.client.mcp.add).not.toHaveBeenCalled() - await runtime.dispose() - } - ) + expect(gateway.grantCustomMcp).not.toHaveBeenCalled(); + expect(setup.client.mcp.add).not.toHaveBeenCalled(); + await runtime.dispose(); + }, + ); - it('parses OpenCode questions and sends the selected answers back', async () => { + it("parses OpenCode questions and sends the selected answers back", async () => { const setup = runClient([ { - id: 'question-event', - type: 'question.asked', + id: "question-event", + type: "question.asked", properties: { - id: 'question-1', - sessionID: 'session-1', + id: "question-1", + sessionID: "session-1", questions: [ { - header: '实现方式', - question: '请选择实现方式', + header: "实现方式", + question: "请选择实现方式", options: [ { - label: '直接修改', - description: '立即更新现有实现' + label: "直接修改", + description: "立即更新现有实现", }, { - label: '先写测试', - description: '先增加回归测试' - } + label: "先写测试", + description: "先增加回归测试", + }, ], multiple: false, - custom: true - } + custom: true, + }, ], tool: { - messageID: 'message-1', - callID: 'call-question-1' - } - } + messageID: "message-1", + callID: "call-question-1", + }, + }, }, { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(setup.client) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(setup.client); const stream = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'test', - workMode: 'execute' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "test", + workMode: "execute", }, - new AbortController().signal - ) + new AbortController().signal, + ); await expect(stream.next()).resolves.toMatchObject({ - value: { type: 'status' } - }) - const questionEvent = await stream.next() + value: { type: "status" }, + }); + const questionEvent = await stream.next(); expect(questionEvent.value).toMatchObject({ - type: 'question', + type: "question", questionId: expect.stringMatching(/^opencode-[a-f0-9]{48}$/u), questions: [ { - header: '实现方式', - question: '请选择实现方式', + header: "实现方式", + question: "请选择实现方式", multiple: false, - custom: true - } - ] - }) + custom: true, + }, + ], + }); const questionId = - questionEvent.value?.type === 'question' + questionEvent.value?.type === "question" ? questionEvent.value.questionId - : '' - await runtime.respondToQuestion(questionId, [['先写测试']]) + : ""; + await runtime.respondToQuestion(questionId, [["先写测试"]]); expect(setup.questionReply).toHaveBeenCalledWith( { - requestID: 'question-1', + requestID: "question-1", directory: process.cwd(), - answers: [['先写测试']] + answers: [["先写测试"]], }, - { signal: expect.any(AbortSignal) } - ) + { signal: expect.any(AbortSignal) }, + ); await expect(stream.next()).resolves.toMatchObject({ - value: { type: 'done' } - }) - await runtime.dispose() - }) + value: { type: "done" }, + }); + await runtime.dispose(); + }); - it('namespaces identical upstream question IDs across concurrent external conversations', async () => { - const setup = runClient([]) - ;( + it("namespaces identical upstream question IDs across concurrent external conversations", async () => { + const setup = runClient([]); + ( setup.event.subscribe as unknown as ReturnType ).mockImplementation(async () => ({ stream: (async function* () { yield { - id: 'question-event', - type: 'question.asked', + id: "question-event", + type: "question.asked", properties: { - id: 'shared-question', - sessionID: 'session-1', + id: "shared-question", + sessionID: "session-1", questions: [ { - header: 'Choice', - question: 'Choose', + header: "Choice", + question: "Choose", options: [], multiple: false, - custom: true - } - ] - } - } + custom: true, + }, + ], + }, + }; yield { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - })() - })) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }; + })(), + })); const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', - embedded: false + baseUrl: "http://127.0.0.1:4096", + embedded: false, }), dependencies(fakeChild(), { createClient: vi.fn( - () => setup.client - ) as unknown as typeof createOpencodeClient - }).deps - ) + () => setup.client, + ) as unknown as typeof createOpencodeClient, + }).deps, + ); const first = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'first', - workMode: 'execute' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "first", + workMode: "execute", }, - new AbortController().signal - ) + new AbortController().signal, + ); const second = runtime.run( { - requestId: '4f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-2', - prompt: 'second', - workMode: 'execute' + requestId: "4f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-2", + prompt: "second", + workMode: "execute", }, - new AbortController().signal - ) + new AbortController().signal, + ); - await Promise.all([first.next(), second.next()]) + await Promise.all([first.next(), second.next()]); const [firstQuestion, secondQuestion] = await Promise.all([ first.next(), - second.next() - ]) + second.next(), + ]); const firstId = - firstQuestion.value?.type === 'question' + firstQuestion.value?.type === "question" ? firstQuestion.value.questionId - : '' + : ""; const secondId = - secondQuestion.value?.type === 'question' + secondQuestion.value?.type === "question" ? secondQuestion.value.questionId - : '' - expect(firstId).toMatch(/^opencode-[a-f0-9]{48}$/u) - expect(secondId).toMatch(/^opencode-[a-f0-9]{48}$/u) - expect(firstId).not.toBe(secondId) + : ""; + expect(firstId).toMatch(/^opencode-[a-f0-9]{48}$/u); + expect(secondId).toMatch(/^opencode-[a-f0-9]{48}$/u); + expect(firstId).not.toBe(secondId); await Promise.all([ - runtime.respondToQuestion(firstId, [['first answer']]), - runtime.respondToQuestion(secondId, [['second answer']]) - ]) - expect(setup.questionReply).toHaveBeenCalledTimes(2) + runtime.respondToQuestion(firstId, [["first answer"]]), + runtime.respondToQuestion(secondId, [["second answer"]]), + ]); + expect(setup.questionReply).toHaveBeenCalledTimes(2); expect(setup.questionReply).toHaveBeenNthCalledWith( 1, { - requestID: 'shared-question', + requestID: "shared-question", directory: process.cwd(), - answers: [['first answer']] + answers: [["first answer"]], }, - { signal: expect.any(AbortSignal) } - ) + { signal: expect.any(AbortSignal) }, + ); expect(setup.questionReply).toHaveBeenNthCalledWith( 2, { - requestID: 'shared-question', + requestID: "shared-question", directory: process.cwd(), - answers: [['second answer']] + answers: [["second answer"]], }, - { signal: expect.any(AbortSignal) } - ) - await Promise.all([first.next(), second.next()]) - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ); + await Promise.all([first.next(), second.next()]); + await runtime.dispose(); + }); - it('emits native Task calls as subagents instead of generic tools', async () => { + it("emits native Task calls as subagents instead of generic tools", async () => { const taskInput = { - subagent_type: 'explorer', - description: 'Review application architecture', - prompt: 'Inspect the complete source tree.' - } + subagent_type: "explorer", + description: "Review application architecture", + prompt: "Inspect the complete source tree.", + }; const setup = runClient([ { - id: 'task-running', - type: 'message.part.updated', + id: "task-running", + type: "message.part.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", part: { - id: 'part-task-1', - callID: 'call-task-1', - type: 'tool', - tool: 'task', + id: "part-task-1", + callID: "call-task-1", + type: "tool", + tool: "task", state: { - status: 'running', + status: "running", input: taskInput, - time: { start: 1 } - } - } - } + time: { start: 1 }, + }, + }, + }, }, { - id: 'task-completed', - type: 'message.part.updated', + id: "task-completed", + type: "message.part.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", part: { - id: 'part-task-1', - callID: 'call-task-1', - type: 'tool', - tool: 'task', + id: "part-task-1", + callID: "call-task-1", + type: "tool", + tool: "task", state: { - status: 'completed', + status: "completed", input: taskInput, - output: 'Architecture review complete.', - title: 'Review application architecture', + output: "Architecture review complete.", + title: "Review application architecture", metadata: {}, - time: { start: 1, end: 2 } - } - } - } + time: { start: 1, end: 2 }, + }, + }, + }, }, { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(setup.client) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(setup.client); - const events = await collectRun(runtime) - const subagents = events.filter( - (event) => event.type === 'subagent' - ) + const events = await collectRun(runtime); + const subagents = events.filter((event) => event.type === "subagent"); - expect(subagents).toHaveLength(2) + expect(subagents).toHaveLength(2); expect(subagents[0]).toMatchObject({ - expertName: 'explorer', - routingMode: 'native', - runtimeCallId: 'call-task-1', - state: 'running', - reason: 'Review application architecture' - }) + expertName: "explorer", + routingMode: "native", + runtimeCallId: "call-task-1", + state: "running", + reason: "Review application architecture", + }); expect(subagents[1]).toMatchObject({ childTaskId: - subagents[0]?.type === 'subagent' + subagents[0]?.type === "subagent" ? subagents[0].childTaskId : undefined, - state: 'completed', - output: 'Architecture review complete.' - }) - expect(events.some((event) => event.type === 'tool')).toBe(false) - await runtime.dispose() - }) + state: "completed", + output: "Architecture review complete.", + }); + expect(events.some((event) => event.type === "tool")).toBe(false); + await runtime.dispose(); + }); - it('keeps non-Task tools with Task-shaped input as generic tools', async () => { + it("keeps non-Task tools with Task-shaped input as generic tools", async () => { const setup = runClient([ { - id: 'custom-tool-completed', - type: 'message.part.updated', + id: "custom-tool-completed", + type: "message.part.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", part: { - id: 'part-custom-1', - callID: 'call-custom-1', - type: 'tool', - tool: 'custom_delegate', + id: "part-custom-1", + callID: "call-custom-1", + type: "tool", + tool: "custom_delegate", state: { - status: 'completed', + status: "completed", input: { - subagent_type: 'explorer', - prompt: 'Inspect the complete source tree.' + subagent_type: "explorer", + prompt: "Inspect the complete source tree.", }, - output: 'Custom tool result.', - title: 'Custom delegate', + output: "Custom tool result.", + title: "Custom delegate", metadata: {}, - time: { start: 1, end: 2 } - } - } - } + time: { start: 1, end: 2 }, + }, + }, + }, }, { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(setup.client) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(setup.client); - const events = await collectRun(runtime) + const events = await collectRun(runtime); expect(events).toContainEqual( expect.objectContaining({ - type: 'tool', - callId: 'call-custom-1', - name: 'custom_delegate', - state: 'completed' - }) - ) - expect(events.some((event) => event.type === 'subagent')).toBe(false) - await runtime.dispose() - }) + type: "tool", + callId: "call-custom-1", + name: "custom_delegate", + state: "completed", + }), + ); + expect(events.some((event) => event.type === "subagent")).toBe(false); + await runtime.dispose(); + }); - it('does not impose a default wall-clock deadline on OpenCode runs', async () => { - vi.useFakeTimers() + it("does not impose a default wall-clock deadline on OpenCode runs", async () => { + vi.useFakeTimers(); try { - let release!: () => void + let release!: () => void; const gate = new Promise((resolve) => { - release = resolve - }) - const setup = runClient([]) - ;( + release = resolve; + }); + const setup = runClient([]); + ( setup.event.subscribe as unknown as ReturnType ).mockResolvedValue({ stream: (async function* () { - await gate + await gate; yield { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - })() - }) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }; + })(), + }); const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', - embedded: false + baseUrl: "http://127.0.0.1:4096", + embedded: false, }), dependencies(fakeChild(), { createClient: vi.fn( - () => setup.client - ) as unknown as typeof createOpencodeClient - }).deps - ) + () => setup.client, + ) as unknown as typeof createOpencodeClient, + }).deps, + ); const stream = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'work for more than a day', - workMode: 'execute' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "work for more than a day", + workMode: "execute", }, - new AbortController().signal - ) + new AbortController().signal, + ); await expect(stream.next()).resolves.toMatchObject({ - value: { type: 'status' } - }) - const pending = stream.next() - await vi.advanceTimersByTimeAsync(25 * 60 * 60_000) - expect(setup.session.abort).not.toHaveBeenCalled() - release() + value: { type: "status" }, + }); + const pending = stream.next(); + await vi.advanceTimersByTimeAsync(25 * 60 * 60_000); + expect(setup.session.abort).not.toHaveBeenCalled(); + release(); await expect(pending).resolves.toMatchObject({ - value: { type: 'done' } - }) - await runtime.dispose() + value: { type: "done" }, + }); + await runtime.dispose(); } finally { - vi.useRealTimers() + vi.useRealTimers(); } - }) + }); - it('aborts an OpenCode run at its total execution deadline', async () => { - const setup = runClient([]) - ;( + it("aborts an OpenCode run at its total execution deadline", async () => { + const setup = runClient([]); + ( setup.event.subscribe as unknown as ReturnType ).mockImplementation( - async ( - _input: unknown, - options: { signal: AbortSignal } - ) => ({ + async (_input: unknown, options: { signal: AbortSignal }) => ({ stream: (async function* () { await new Promise((_resolve, reject) => { options.signal.addEventListener( - 'abort', + "abort", () => reject(options.signal.reason), - { once: true } - ) - }) + { once: true }, + ); + }); yield { - id: 'unreachable', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - })() - }) - ) + id: "unreachable", + type: "session.idle", + properties: { sessionID: "session-1" }, + }; + })(), + }), + ); const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', - embedded: false + baseUrl: "http://127.0.0.1:4096", + embedded: false, }), dependencies(fakeChild(), { createClient: vi.fn( - () => setup.client + () => setup.client, ) as unknown as typeof createOpencodeClient, - executionTimeoutMs: 20 - }).deps - ) + executionTimeoutMs: 20, + }).deps, + ); const stream = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'never finish', - workMode: 'execute' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "never finish", + workMode: "execute", }, - new AbortController().signal - ) + new AbortController().signal, + ); await expect(stream.next()).resolves.toMatchObject({ - value: { type: 'status' } - }) + value: { type: "status" }, + }); await expect(stream.next()).rejects.toThrow( - 'OpenCode 执行超过 20 毫秒总时限' - ) - expect(setup.session.abort).toHaveBeenCalled() - await runtime.dispose() - }) + "OpenCode 执行超过 20 毫秒总时限", + ); + expect(setup.session.abort).toHaveBeenCalled(); + await runtime.dispose(); + }); - it('applies the total deadline while agent discovery is stalled', async () => { - const setup = runClient([]) - const agents = vi.fn( - () => new Promise(() => undefined) - ) + it("applies the total deadline while agent discovery is stalled", async () => { + const setup = runClient([]); + const agents = vi.fn(() => new Promise(() => undefined)); Object.assign(setup.client, { - app: { agents } - }) - const child = fakeChild() + app: { agents }, + }); + const child = fakeChild(); const runtime = new OpenCodeRuntime( options({ - customization: { defaultAgent: 'build' } + customization: { defaultAgent: "build" }, }), dependencies(child, { createClient: vi.fn( - () => setup.client + () => setup.client, ) as unknown as typeof createOpencodeClient, - executionTimeoutMs: 20 - }).deps - ) + executionTimeoutMs: 20, + }).deps, + ); const stream = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'never reach the stream', - workMode: 'execute' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "never reach the stream", + workMode: "execute", }, - new AbortController().signal - ) + new AbortController().signal, + ); await expect(stream.next()).rejects.toThrow( - 'OpenCode 执行超过 20 毫秒总时限' - ) + "OpenCode 执行超过 20 毫秒总时限", + ); expect(agents).toHaveBeenCalledWith( { directory: process.cwd() }, - { signal: expect.any(AbortSignal) } - ) - expect(setup.event.subscribe).not.toHaveBeenCalled() - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ); + expect(setup.event.subscribe).not.toHaveBeenCalled(); + await runtime.dispose(); + }); - it('deletes a session created after timeout and does not reuse it', async () => { + it("deletes a session created after timeout and does not reuse it", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); let resolveStalledCreation!: (value: { - data: { id: string } - error: undefined - }) => void - ;( + data: { id: string }; + error: undefined; + }) => void; + ( setup.session.create as unknown as ReturnType ).mockImplementationOnce( () => new Promise((resolve) => { - resolveStalledCreation = resolve - }) - ) + resolveStalledCreation = resolve; + }), + ); const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', - embedded: false + baseUrl: "http://127.0.0.1:4096", + embedded: false, }), dependencies(fakeChild(), { createClient: vi.fn( - () => setup.client + () => setup.client, ) as unknown as typeof createOpencodeClient, - executionTimeoutMs: 20 - }).deps - ) + executionTimeoutMs: 20, + }).deps, + ); const first = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'stalled creation', - workMode: 'execute' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "stalled creation", + workMode: "execute", }, - new AbortController().signal - ) + new AbortController().signal, + ); await expect(first.next()).rejects.toThrow( - 'OpenCode 执行超过 20 毫秒总时限' - ) + "OpenCode 执行超过 20 毫秒总时限", + ); resolveStalledCreation({ - data: { id: 'stale-session' }, - error: undefined - }) + data: { id: "stale-session" }, + error: undefined, + }); await vi.waitFor(() => expect(setup.session.delete).toHaveBeenCalledWith( { - sessionID: 'stale-session', - directory: process.cwd() + sessionID: "stale-session", + directory: process.cwd(), }, - { signal: expect.any(AbortSignal) } - ) - ) + { signal: expect.any(AbortSignal) }, + ), + ); - const secondEvents = await collectRun(runtime) - expect(secondEvents.at(-1)).toMatchObject({ type: 'done' }) - expect(setup.session.create).toHaveBeenCalledTimes(2) - expect(setup.session.update).not.toHaveBeenCalled() - await runtime.dispose() - }) + const secondEvents = await collectRun(runtime); + expect(secondEvents.at(-1)).toMatchObject({ type: "done" }); + expect(setup.session.create).toHaveBeenCalledTimes(2); + expect(setup.session.update).not.toHaveBeenCalled(); + await runtime.dispose(); + }); - it('truncates oversized text display without stopping the run', async () => { + it("truncates oversized text display without stopping the run", async () => { const setup = runClient([ { - id: 'first-text', - type: 'message.part.delta', + id: "first-text", + type: "message.part.delta", properties: { - sessionID: 'session-1', - partID: 'part-1', - field: 'text', - delta: 'a'.repeat(600_000) - } + sessionID: "session-1", + partID: "part-1", + field: "text", + delta: "a".repeat(600_000), + }, }, { - id: 'second-text', - type: 'message.part.delta', + id: "second-text", + type: "message.part.delta", properties: { - sessionID: 'session-1', - partID: 'part-1', - field: 'text', - delta: 'b'.repeat(600_000) - } + sessionID: "session-1", + partID: "part-1", + field: "text", + delta: "b".repeat(600_000), + }, }, { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(setup.client) - const events = await collectRun(runtime, 'execute') + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(setup.client); + const events = await collectRun(runtime, "execute"); const text = events - .flatMap((event) => event.type === 'text' ? [event.delta] : []) - .join('') + .flatMap((event) => (event.type === "text" ? [event.delta] : [])) + .join(""); - expect(Buffer.byteLength(text)).toBe(1024 * 1024) + expect(Buffer.byteLength(text)).toBe(1024 * 1024); expect(events).toContainEqual({ - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - type: 'status', - message: - 'OpenCode 输出较长，已停止继续展示文本；任务仍在继续' - }) - expect(events.at(-1)).toMatchObject({ type: 'done' }) - expect(setup.session.abort).not.toHaveBeenCalled() - await runtime.dispose() - }) + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + type: "status", + message: "OpenCode 输出较长，已停止继续展示文本；任务仍在继续", + }); + expect(events.at(-1)).toMatchObject({ type: "done" }); + expect(setup.session.abort).not.toHaveBeenCalled(); + await runtime.dispose(); + }); - it('completes more than 100 distinct tool calls in one run', async () => { + it("completes more than 100 distinct tool calls in one run", async () => { const setup = runClient([ ...Array.from({ length: 101 }, (_, index) => - completedToolEvent(`call-${index + 1}`) + completedToolEvent(`call-${index + 1}`), ), { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(setup.client) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(setup.client); - const events = await collectRun(runtime, 'execute') + const events = await collectRun(runtime, "execute"); - expect( - events.filter((event) => event.type === 'tool') - ).toHaveLength(101) - expect(events.at(-1)).toMatchObject({ type: 'done' }) - await runtime.dispose() - }) + expect(events.filter((event) => event.type === "tool")).toHaveLength(101); + expect(events.at(-1)).toMatchObject({ type: "done" }); + await runtime.dispose(); + }); - it('adds only request-scoped built-in read tools for Ask and disconnects them', async () => { + it("adds only request-scoped built-in read tools for Ask and disconnects them", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const toolIds = setup.tool.ids as unknown as ReturnType + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const toolIds = setup.tool.ids as unknown as ReturnType; toolIds .mockResolvedValueOnce({ - data: ['read', 'write', 'bash'], - error: undefined + data: ["read", "write", "bash"], + error: undefined, }) .mockResolvedValueOnce({ - data: [ - 'read', - 'write', - 'bash', - 'goodbuddy_knowledge_search' - ], - error: undefined + data: ["read", "write", "bash", "goodbuddy_knowledge_search"], + error: undefined, }) .mockResolvedValue({ - data: [ - 'read', - 'write', - 'bash', - 'goodbuddy_knowledge_search' - ], - error: undefined - }) + data: ["read", "write", "bash", "goodbuddy_knowledge_search"], + error: undefined, + }); const gateway = { - getEndpoint: () => 'http://127.0.0.1:4567/mcp', - getAvailableToolNames: () => ['knowledge_search'] - } as unknown as KnowledgeMcpGateway - const child = fakeChild() + getEndpoint: () => "http://127.0.0.1:4567/mcp", + getAvailableToolNames: () => ["knowledge_search"], + } as unknown as KnowledgeMcpGateway; + const child = fakeChild(); const { deps } = dependencies(child, { createClient: vi.fn( - () => setup.client - ) as unknown as typeof createOpencodeClient - }) + () => setup.client, + ) as unknown as typeof createOpencodeClient, + }); const runtime = new OpenCodeRuntime( options({ knowledgeGateway: gateway }), - deps - ) + deps, + ); - const events = [] + const events = []; for await (const event of runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'search', - workMode: 'ask', - knowledgeCapabilityToken: 'secret-capability' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "search", + workMode: "ask", + knowledgeCapabilityToken: "secret-capability", }, - new AbortController().signal + new AbortController().signal, )) { - events.push(event) + events.push(event); } expect(setup.client.mcp.add).toHaveBeenCalledWith( @@ -2516,341 +2527,526 @@ describe('OpenCodeRuntime embedded permission mediation', () => { directory: process.cwd(), name: expect.stringMatching(/^goodbuddy-data-[a-f0-9]{20}$/u), config: { - type: 'remote', - url: 'http://127.0.0.1:4567/mcp', + type: "remote", + url: "http://127.0.0.1:4567/mcp", enabled: true, headers: { - Authorization: 'Bearer secret-capability' + Authorization: "Bearer secret-capability", }, - oauth: false - } + oauth: false, + }, }, - { signal: expect.any(AbortSignal) } - ) + { signal: expect.any(AbortSignal) }, + ); const knowledgeMcpName = ( - ( - setup.client.mcp.add as unknown as ReturnType - ).mock.calls[0]?.[0] as { name: string } - ).name - const knowledgeToolId = `${knowledgeMcpName}_knowledge_search` + (setup.client.mcp.add as unknown as ReturnType).mock + .calls[0]?.[0] as { name: string } + ).name; + const knowledgeToolId = `${knowledgeMcpName}_knowledge_search`; expect(setup.session.create).toHaveBeenCalledWith( expect.objectContaining({ permission: [ - { permission: '*', pattern: '*', action: 'deny' }, + { permission: "*", pattern: "*", action: "deny" }, { permission: knowledgeToolId, - pattern: '*', - action: 'allow' - } - ] + pattern: "*", + action: "allow", + }, + ], }), - { signal: expect.any(AbortSignal) } - ) + { signal: expect.any(AbortSignal) }, + ); expect(setup.session.promptAsync).toHaveBeenCalledWith( expect.objectContaining({ tools: { read: false, write: false, bash: false, - 'goodbuddy-data-*': false, - 'goodbuddy-custom-*': false, - [knowledgeToolId]: true - } + "goodbuddy-data-*": false, + "goodbuddy-custom-*": false, + [knowledgeToolId]: true, + }, }), - expect.anything() - ) + expect.anything(), + ); expect(setup.client.mcp.disconnect).toHaveBeenCalledWith( { name: expect.stringMatching(/^goodbuddy-data-/u), - directory: process.cwd() + directory: process.cwd(), }, - { signal: expect.any(AbortSignal) } - ) - expect(events.at(-1)).toMatchObject({ type: 'done' }) - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ); + expect(events.at(-1)).toMatchObject({ type: "done" }); + await runtime.dispose(); + }); - it('enables the deterministic MCP tool name when tool ids omit dynamic tools', async () => { + it("enables the deterministic MCP tool name when tool ids omit dynamic tools", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const baseline = { - data: ['read', 'write', 'bash'], - error: undefined - } - const toolIds = setup.tool.ids as unknown as ReturnType - toolIds.mockResolvedValue(baseline) - const child = fakeChild() + data: ["read", "write", "bash"], + error: undefined, + }; + const toolIds = setup.tool.ids as unknown as ReturnType; + toolIds.mockResolvedValue(baseline); + const child = fakeChild(); const { deps } = dependencies(child, { createClient: vi.fn( - () => setup.client - ) as unknown as typeof createOpencodeClient - }) + () => setup.client, + ) as unknown as typeof createOpencodeClient, + }); const runtime = new OpenCodeRuntime( options({ knowledgeGateway: { - getEndpoint: () => 'http://127.0.0.1:4567/mcp', - getAvailableToolNames: () => ['knowledge_search'] - } as unknown as KnowledgeMcpGateway + getEndpoint: () => "http://127.0.0.1:4567/mcp", + getAvailableToolNames: () => ["knowledge_search"], + } as unknown as KnowledgeMcpGateway, }), - deps - ) + deps, + ); for await (const _event of runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'search', - workMode: 'ask', - knowledgeCapabilityToken: 'secret-capability' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "search", + workMode: "ask", + knowledgeCapabilityToken: "secret-capability", }, - new AbortController().signal + new AbortController().signal, )) { - void _event + void _event; } const knowledgeMcpName = ( - ( - setup.client.mcp.add as unknown as ReturnType - ).mock.calls[0]?.[0] as { name: string } - ).name - const knowledgeToolId = `${knowledgeMcpName}_knowledge_search` - expect(toolIds).toHaveBeenCalledTimes(1) + (setup.client.mcp.add as unknown as ReturnType).mock + .calls[0]?.[0] as { name: string } + ).name; + const knowledgeToolId = `${knowledgeMcpName}_knowledge_search`; + expect(toolIds).toHaveBeenCalledTimes(1); expect(setup.session.promptAsync).toHaveBeenCalledWith( expect.objectContaining({ tools: expect.objectContaining({ read: false, write: false, bash: false, - [knowledgeToolId]: true - }) + [knowledgeToolId]: true, + }), }), - expect.anything() - ) - await runtime.dispose() - }) + expect.anything(), + ); + await runtime.dispose(); + }); - it('runs embedded conversations concurrently with request-scoped MCP registrations', async () => { + it("runs embedded conversations concurrently while serializing MCP registrations", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const toolIds = setup.tool.ids as unknown as ReturnType + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const toolIds = setup.tool.ids as unknown as ReturnType; const baseline = { - data: ['read', 'write'], - error: undefined - } + data: ["read", "write"], + error: undefined, + }; const withKnowledge = { - data: ['read', 'write', 'goodbuddy_knowledge_search'], - error: undefined - } + data: ["read", "write", "goodbuddy_knowledge_search"], + error: undefined, + }; for (const response of [ baseline, withKnowledge, withKnowledge, baseline, withKnowledge, - withKnowledge + withKnowledge, ]) { - toolIds.mockResolvedValueOnce(response) + toolIds.mockResolvedValueOnce(response); } - let resolveFirstAdd!: () => void + let resolveFirstAdd!: () => void; const firstAdd = new Promise((resolve) => { - resolveFirstAdd = resolve - }) - const mcpAdd = setup.client.mcp.add as unknown as ReturnType + resolveFirstAdd = resolve; + }); + const mcpAdd = setup.client.mcp.add as unknown as ReturnType; mcpAdd .mockImplementationOnce(async (input: { name: string }) => { - await firstAdd + await firstAdd; return { data: { - [input.name]: { status: 'connected' } + [input.name]: { status: "connected" }, }, - error: undefined - } + error: undefined, + }; }) .mockImplementation(async (input: { name: string }) => ({ data: { - [input.name]: { status: 'connected' } + [input.name]: { status: "connected" }, }, - error: undefined - })) - const child = fakeChild() + error: undefined, + })); + const child = fakeChild(); const { deps } = dependencies(child, { createClient: vi.fn( - () => setup.client - ) as unknown as typeof createOpencodeClient - }) + () => setup.client, + ) as unknown as typeof createOpencodeClient, + }); const runtime = new OpenCodeRuntime( options({ knowledgeGateway: { - getEndpoint: () => 'http://127.0.0.1:4567/mcp', - getAvailableToolNames: () => ['knowledge_search'] - } as unknown as KnowledgeMcpGateway + getEndpoint: () => "http://127.0.0.1:4567/mcp", + getAvailableToolNames: () => ["knowledge_search"], + } as unknown as KnowledgeMcpGateway, }), - deps - ) + deps, + ); const collect = async ( requestId: string, conversationId: string, - token: string + token: string, ): Promise => { for await (const _event of runtime.run( { requestId, conversationId, - prompt: 'search', - workMode: 'ask', - knowledgeCapabilityToken: token + prompt: "search", + workMode: "ask", + knowledgeCapabilityToken: token, }, - new AbortController().signal + new AbortController().signal, )) { - void _event + void _event; } - } + }; const first = collect( - '3f496642-f47d-4e0a-8944-a32c77b0d6e1', - 'conversation-one', - 'first-token' - ) - await vi.waitFor(() => expect(mcpAdd).toHaveBeenCalledTimes(1)) + "3f496642-f47d-4e0a-8944-a32c77b0d6e1", + "conversation-one", + "first-token", + ); + await vi.waitFor(() => expect(mcpAdd).toHaveBeenCalledTimes(1)); const second = collect( - '3f496642-f47d-4e0a-8944-a32c77b0d6e2', - 'conversation-two', - 'second-token' - ) + "3f496642-f47d-4e0a-8944-a32c77b0d6e2", + "conversation-two", + "second-token", + ); try { - await vi.waitFor( - () => expect(mcpAdd).toHaveBeenCalledTimes(2), - { timeout: 500 } - ) - await vi.waitFor( - () => expect(setup.session.promptAsync).toHaveBeenCalledOnce(), - { timeout: 500 } - ) + await new Promise((resolve) => setTimeout(resolve, 50)); + expect(mcpAdd).toHaveBeenCalledTimes(1); + expect(setup.session.promptAsync).not.toHaveBeenCalled(); } finally { - resolveFirstAdd() + resolveFirstAdd(); } - await Promise.all([first, second]) + await vi.waitFor(() => expect(mcpAdd).toHaveBeenCalledTimes(2)); + await Promise.all([first, second]); expect( mcpAdd.mock.calls.map( ([input]) => - (input as { - config: { headers: { Authorization: string } } - }).config.headers.Authorization - ) - ).toEqual(['Bearer first-token', 'Bearer second-token']) - expect(setup.client.mcp.disconnect).toHaveBeenCalledTimes(2) - await runtime.dispose() - }) + ( + input as { + config: { headers: { Authorization: string } }; + } + ).config.headers.Authorization, + ), + ).toEqual(["Bearer first-token", "Bearer second-token"]); + expect(setup.client.mcp.disconnect).toHaveBeenCalledTimes(2); + await runtime.dispose(); + }); - it('does not send a knowledge capability to external OpenCode', async () => { + it.each(["knowledge", "custom"])("releases the conversation when a queued %s MCP add is cancelled", async (kind) => { + const setup = runClient([ + { id: "idle", type: "session.idle", properties: { sessionID: "session-1" } }, + ]); + let releaseRegistration!: () => void; + const registration = new Promise((resolve) => { releaseRegistration = resolve; }); + const add = setup.client.mcp.add as unknown as ReturnType; + add.mockImplementationOnce(async (input: { name: string }) => { + await registration; + return { data: { [input.name]: { status: "connected" } } }; + }); + const revoke = vi.fn(); + const runtime = embeddedRuntime(setup.client, { + knowledgeGateway: { + getEndpoint: () => "http://127.0.0.1:4567/mcp", + getAvailableToolNames: () => ["knowledge_search"], + grantCustomMcp: () => "custom-token", + prepareCustomMcpTools: async () => [{ name: "custom_tool" }], + revoke, + } as unknown as KnowledgeMcpGateway, + mcpServers: [{ + id: "custom", + name: "Custom", + description: "", + transport: "http", + url: "https://example.com/mcp", + enabled: true, + allowDynamicTools: false, + assignments: ["opencode"], + secretConfigured: false, + }], + }); + const mutations = vi.spyOn(runtime as unknown as { mutateMcp: (...args: unknown[]) => Promise }, "mutateMcp"); + const controller = new AbortController(); + const collect = async (requestId: string, conversationId: string, signal: AbortSignal, mcp: boolean) => { + const events: RuntimeEvent[] = []; + for await (const event of runtime.run({ + requestId, conversationId, prompt: "test", + workMode: mcp && kind === "custom" ? "execute" : "ask", + ...(mcp && kind === "knowledge" ? { knowledgeCapabilityToken: requestId } : {}), + }, signal)) events.push(event); + return events; + }; + const runs: Promise[] = []; + try { + runs.push(collect("first", "conversation-first", new AbortController().signal, true)); + await vi.waitFor(() => expect(add).toHaveBeenCalledOnce()); + const cancelled = collect("cancelled", "conversation-second", controller.signal, true).catch((error: unknown) => error); + runs.push(cancelled); + await vi.waitFor(() => expect(mutations).toHaveBeenCalledTimes(2)); + controller.abort(new Error("cancelled queued add")); + let cancellationFinished = false; + void cancelled.then(() => { cancellationFinished = true; }); + await vi.waitFor(() => expect(cancellationFinished).toBe(true)); + await expect(cancelled).resolves.toMatchObject({ message: "cancelled queued add" }); + if (kind === "custom") expect(revoke).toHaveBeenCalledWith("custom-token"); + const next = collect("next", "conversation-second", new AbortController().signal, false); + runs.push(next); + await expect(next).resolves.toContainEqual(expect.objectContaining({ type: "done" })); + expect(add).toHaveBeenCalledOnce(); + expect(setup.client.mcp.disconnect).not.toHaveBeenCalled(); + releaseRegistration(); + await Promise.all(runs); + expect(setup.client.mcp.disconnect).toHaveBeenCalledExactlyOnceWith( + { name: add.mock.calls[0]![0].name, directory: process.cwd() }, + { signal: expect.any(AbortSignal) }, + ); + } finally { + releaseRegistration(); + await Promise.allSettled(runs); + await runtime.dispose(); + } + }); + + it("keeps cleanup queued behind blocked registration while sessions run concurrently", async () => { + const setup = runClient([]); + const releases: (() => void)[] = []; + const finished = Array.from({ length: 3 }, () => + new Promise((resolve) => releases.push(resolve)), + ); + let releaseRegistration!: () => void; + const registration = new Promise((resolve) => { + releaseRegistration = resolve; + }); + const create = setup.session.create as unknown as ReturnType; + let sessionCount = 0; + create.mockImplementation(async () => ({ + data: { id: `session-${++sessionCount}` }, + error: undefined, + })); + const subscribe = setup.event.subscribe as unknown as ReturnType< + typeof vi.fn + >; + let subscriptionCount = 0; + subscribe.mockImplementation(async () => { + const index = subscriptionCount++; + return { + stream: (async function* () { + await finished[index]; + yield { + id: `idle-${index}`, + type: "session.idle", + properties: { sessionID: `session-${index + 1}` }, + }; + })(), + }; + }); + const add = setup.client.mcp.add as unknown as ReturnType; + add.mockImplementation(async (input: { name: string }) => { + if (add.mock.calls.length === 3) await registration; + return { data: { [input.name]: { status: "connected" } } }; + }); + const disconnect = setup.client.mcp.disconnect as unknown as ReturnType< + typeof vi.fn + >; + const runtime = embeddedRuntime(setup.client, { + knowledgeGateway: { + getEndpoint: () => "http://127.0.0.1:4567/mcp", + getAvailableToolNames: () => ["knowledge_search"], + } as unknown as KnowledgeMcpGateway, + }); + const events: RuntimeEvent[] = []; + const collect = async (index: number): Promise => { + for await (const event of runtime.run( + { + requestId: `3f496642-f47d-4e0a-8944-a32c77b0d6e${index}`, + conversationId: `conversation-${index}`, + prompt: "search", + workMode: "ask", + knowledgeCapabilityToken: `token-${index}`, + }, + new AbortController().signal, + )) { + events.push(event); + } + }; + const runs: Promise[] = []; + try { + runs.push(collect(1)); + await vi.waitFor(() => + expect(setup.session.promptAsync).toHaveBeenCalledTimes(1), + ); + runs.push(collect(2)); + await vi.waitFor(() => + expect(setup.session.promptAsync).toHaveBeenCalledTimes(2), + ); + expect(events.filter((event) => event.type === "done")).toHaveLength(0); + expect(create).toHaveBeenCalledTimes(2); + expect(setup.session.promptAsync).toHaveBeenNthCalledWith( + 2, + expect.objectContaining({ sessionID: "session-2" }), + expect.anything(), + ); + + runs.push(collect(3)); + await vi.waitFor(() => expect(add).toHaveBeenCalledTimes(3)); + releases[0]!(); + await vi.waitFor(() => + expect(events.filter((event) => event.type === "done")).toHaveLength(1), + ); + // Exceed the disconnect budget while another registration owns the queue. + await new Promise((resolve) => setTimeout(resolve, 1_100)); + expect(disconnect).not.toHaveBeenCalled(); + expect(setup.session.promptAsync).toHaveBeenCalledTimes(2); + + releaseRegistration(); + await runs[0]; + expect(disconnect).toHaveBeenCalledWith( + { name: add.mock.calls[0]![0].name, directory: process.cwd() }, + { signal: expect.any(AbortSignal) }, + ); + expect(disconnect.mock.calls[0]![1].signal.aborted).toBe(false); + await vi.waitFor(() => + expect(setup.session.promptAsync).toHaveBeenCalledTimes(3), + ); + releases[1]!(); + releases[2]!(); + await Promise.all(runs); + expect(disconnect.mock.calls.map(([input]) => input.name)).toEqual( + add.mock.calls.map(([input]) => input.name), + ); + expect(events.filter((event) => event.type === "done")).toHaveLength(3); + } finally { + releaseRegistration(); + releases.forEach((release) => release()); + await Promise.allSettled(runs); + await runtime.dispose(); + } + }); + + it("does not send a knowledge capability to external OpenCode", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const runtime = new OpenCodeRuntime( options({ embedded: false, - baseUrl: 'http://127.0.0.1:4096', + baseUrl: "http://127.0.0.1:4096", knowledgeGateway: { - getEndpoint: () => 'http://127.0.0.1:4567/mcp', - getAvailableToolNames: () => ['knowledge_search'] - } as unknown as KnowledgeMcpGateway + getEndpoint: () => "http://127.0.0.1:4567/mcp", + getAvailableToolNames: () => ["knowledge_search"], + } as unknown as KnowledgeMcpGateway, }), { createClient: vi.fn( - () => setup.client - ) as unknown as typeof createOpencodeClient - } - ) + () => setup.client, + ) as unknown as typeof createOpencodeClient, + }, + ); for await (const _event of runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'search', - workMode: 'ask', - knowledgeCapabilityToken: 'must-not-leave-main' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "search", + workMode: "ask", + knowledgeCapabilityToken: "must-not-leave-main", }, - new AbortController().signal + new AbortController().signal, )) { - void _event + void _event; } - expect(setup.client.mcp.add).not.toHaveBeenCalled() + expect(setup.client.mcp.add).not.toHaveBeenCalled(); expect( JSON.stringify( - ( - setup.session.promptAsync as unknown as ReturnType - ).mock.calls - ) - ).not.toContain('must-not-leave-main') - await runtime.dispose() - }) + (setup.session.promptAsync as unknown as ReturnType).mock + .calls, + ), + ).not.toContain("must-not-leave-main"); + await runtime.dispose(); + }); - it('allows only registered native Skills in read-only modes', async () => { + it("allows only registered native Skills in read-only modes", async () => { const sourceRoot = await mkdtemp( - join(tmpdir(), 'goodbuddy-opencode-permission-skill-') - ) - const skillDirectory = join(sourceRoot, 'longdoc-docx') - await mkdir(skillDirectory) + join(tmpdir(), "goodbuddy-opencode-permission-skill-"), + ); + const skillDirectory = join(sourceRoot, "longdoc-docx"); + await mkdir(skillDirectory); await writeFile( - join(skillDirectory, 'SKILL.md'), + join(skillDirectory, "SKILL.md"), [ - '---', - 'name: longdoc-docx', - 'description: Build a DOCX', - '---', - '', - '# Long document' - ].join('\n'), - 'utf8' - ) + "---", + "name: longdoc-docx", + "description: Build a DOCX", + "---", + "", + "# Long document", + ].join("\n"), + "utf8", + ); const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const runtime = embeddedRuntime(setup.client, { - skillInstructions: '# Original path: C:\\private\\skills', + skillInstructions: "# Original path: C:\\private\\skills", skillPackages: [ { - id: 'longdoc-docx', - directory: skillDirectory - } - ] - }) + id: "longdoc-docx", + directory: skillDirectory, + }, + ], + }); try { - await collectRun(runtime, 'ask') + await collectRun(runtime, "ask"); expect(setup.session.create).toHaveBeenCalledWith( { - title: 'GoodBuddy 对话', + title: "GoodBuddy 对话", directory: process.cwd(), permission: [ - { permission: '*', pattern: '*', action: 'deny' }, - { permission: 'skill', pattern: '*', action: 'deny' }, + { permission: "*", pattern: "*", action: "deny" }, + { permission: "skill", pattern: "*", action: "deny" }, { - permission: 'skill', - pattern: 'longdoc-docx', - action: 'allow' - } - ] + permission: "skill", + pattern: "longdoc-docx", + action: "allow", + }, + ], }, - { signal: expect.any(AbortSignal) } - ) + { signal: expect.any(AbortSignal) }, + ); expect(setup.session.promptAsync).toHaveBeenCalledWith( expect.objectContaining({ system: undefined, @@ -2859,1162 +3055,1140 @@ describe('OpenCodeRuntime embedded permission mediation', () => { write: false, bash: false, task: false, - 'goodbuddy-data-*': false, - 'goodbuddy-custom-*': false, - skill: true - } + "goodbuddy-data-*": false, + "goodbuddy-custom-*": false, + skill: true, + }, }), - expect.anything() - ) + expect.anything(), + ); } finally { - await runtime.dispose() - await rm(sourceRoot, { recursive: true, force: true }) + await runtime.dispose(); + await rm(sourceRoot, { recursive: true, force: true }); } - }) + }); - it('configures Execute tools as allowed before prompting', async () => { - const { - client, - callOrder, - permissionReply, - session - } = runClient([ - permissionEvent({ sessionID: 'unrelated-session' }), + it("configures Execute tools as allowed before prompting", async () => { + const { client, callOrder, permissionReply, session } = runClient([ + permissionEvent({ sessionID: "unrelated-session" }), permissionEvent(), permissionEvent(), completedToolEvent(), { - id: 'event-reasoning-part', - type: 'message.part.updated', + id: "event-reasoning-part", + type: "message.part.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", part: { - id: 'part-reasoning', - sessionID: 'session-1', - messageID: 'message-1', - type: 'reasoning', - text: '', - time: { start: 1 } - } - } + id: "part-reasoning", + sessionID: "session-1", + messageID: "message-1", + type: "reasoning", + text: "", + time: { start: 1 }, + }, + }, }, { - id: 'event-reasoning', - type: 'message.part.delta', + id: "event-reasoning", + type: "message.part.delta", properties: { - sessionID: 'session-1', - messageID: 'message-1', - partID: 'part-reasoning', - field: 'text', - delta: 'reasoning output' - } + sessionID: "session-1", + messageID: "message-1", + partID: "part-reasoning", + field: "text", + delta: "reasoning output", + }, }, { - id: 'event-text', - type: 'message.part.delta', + id: "event-text", + type: "message.part.delta", properties: { - sessionID: 'session-1', - messageID: 'message-1', - partID: 'part-1', - field: 'text', - delta: 'approved output' - } + sessionID: "session-1", + messageID: "message-1", + partID: "part-1", + field: "text", + delta: "approved output", + }, }, { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) - const events = await collectRun(runtime, 'execute') + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); + const events = await collectRun(runtime, "execute"); - expect(callOrder).toEqual(['subscribe', 'prompt']) + expect(callOrder).toEqual(["subscribe", "prompt"]); expect(session.create).toHaveBeenCalledWith( { - title: 'GoodBuddy 对话', + title: "GoodBuddy 对话", directory: process.cwd(), permission: [ - { permission: '*', pattern: '*', action: 'allow' }, + { permission: "*", pattern: "*", action: "allow" }, { - permission: 'goodbuddy-data-*', - pattern: '*', - action: 'deny' + permission: "goodbuddy-data-*", + pattern: "*", + action: "deny", }, { - permission: 'goodbuddy-custom-*', - pattern: '*', - action: 'deny' - } - ] + permission: "goodbuddy-custom-*", + pattern: "*", + action: "deny", + }, + ], }, - { signal: expect.any(AbortSignal) } - ) + { signal: expect.any(AbortSignal) }, + ); expect(session.promptAsync).toHaveBeenCalledWith( expect.objectContaining({ tools: { - 'goodbuddy-data-*': false, - 'goodbuddy-custom-*': false - } + "goodbuddy-data-*": false, + "goodbuddy-custom-*": false, + }, }), - expect.anything() - ) - expect(permissionReply).toHaveBeenCalledOnce() + expect.anything(), + ); + expect(permissionReply).toHaveBeenCalledOnce(); expect(permissionReply).toHaveBeenCalledWith( { - requestID: 'permission-1', + requestID: "permission-1", directory: process.cwd(), - reply: 'once' + reply: "once", }, - { signal: expect.any(AbortSignal) } - ) + { signal: expect.any(AbortSignal) }, + ); expect(events).toContainEqual( expect.objectContaining({ - type: 'reasoning', - delta: 'reasoning output' - }) - ) + type: "reasoning", + delta: "reasoning output", + }), + ); expect(events).toContainEqual( expect.objectContaining({ - type: 'tool', - callId: 'call-1', - state: 'pending' - }) - ) + type: "tool", + callId: "call-1", + state: "pending", + }), + ); expect(events).toContainEqual( expect.objectContaining({ - type: 'tool', - callId: 'call-1', - state: 'completed', - input: - '{\n "command": "npm test",\n "token": "visible-token"\n}', - output: - 'Tests passed\nAuthorization: Bearer secret-token' - }) - ) + type: "tool", + callId: "call-1", + state: "completed", + input: '{\n "command": "npm test",\n "token": "visible-token"\n}', + output: "Tests passed\nAuthorization: Bearer secret-token", + }), + ); expect(events).toContainEqual( expect.objectContaining({ - type: 'text', - delta: 'approved output' - }) - ) + type: "text", + delta: "approved output", + }), + ); expect( events.filter( (event) => - event.type === 'reasoning' || - event.type === 'text' || - event.type === 'tool' - ) + event.type === "reasoning" || + event.type === "text" || + event.type === "tool", + ), ).toEqual([ expect.objectContaining({ - type: 'tool', - callId: 'call-1', - state: 'pending' + type: "tool", + callId: "call-1", + state: "pending", }), expect.objectContaining({ - type: 'tool', - callId: 'call-1', - state: 'completed' + type: "tool", + callId: "call-1", + state: "completed", }), expect.objectContaining({ - type: 'reasoning', - delta: 'reasoning output' + type: "reasoning", + delta: "reasoning output", }), expect.objectContaining({ - type: 'text', - delta: 'approved output' - }) - ]) - expect(events.at(-1)).toMatchObject({ type: 'done' }) - await runtime.dispose() - }) + type: "text", + delta: "approved output", + }), + ]); + expect(events.at(-1)).toMatchObject({ type: "done" }); + await runtime.dispose(); + }); - it('auto-allows bounded fallback permission requests without GoodBuddy approval', async () => { + it("auto-allows bounded fallback permission requests without GoodBuddy approval", async () => { const { client, permissionReply } = runClient([ permissionEvent(), permissionEvent({ - id: 'permission-2', - patterns: ['npm run lint'], - metadata: { command: 'npm run lint' }, - always: ['npm run lint'], + id: "permission-2", + patterns: ["npm run lint"], + metadata: { command: "npm run lint" }, + always: ["npm run lint"], tool: { - messageID: 'message-2', - callID: 'call-2' - } + messageID: "message-2", + callID: "call-2", + }, }), - completedToolEvent('call-1'), - completedToolEvent('call-2'), + completedToolEvent("call-1"), + completedToolEvent("call-2"), { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) - await collectRun(runtime, 'execute') + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); + await collectRun(runtime, "execute"); expect(permissionReply.mock.calls).toEqual([ [ { - requestID: 'permission-1', + requestID: "permission-1", directory: process.cwd(), - reply: 'once' + reply: "once", }, - { signal: expect.any(AbortSignal) } + { signal: expect.any(AbortSignal) }, ], [ { - requestID: 'permission-2', + requestID: "permission-2", directory: process.cwd(), - reply: 'once' + reply: "once", }, - { signal: expect.any(AbortSignal) } - ] - ]) - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ], + ]); + await runtime.dispose(); + }); - it('fails the run when a tool reports an error before session idle', async () => { + it("fails the run when a tool reports an error before session idle", async () => { const { client, session } = runClient([ { - id: 'event-tool-error', - type: 'message.part.updated', + id: "event-tool-error", + type: "message.part.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", part: { - id: 'part-1', - callID: 'call-1', - type: 'tool', - tool: 'write', + id: "part-1", + callID: "call-1", + type: "tool", + tool: "write", state: { - status: 'error', - error: - 'write failed Authorization: Bearer secret-token' - } - } - } + status: "error", + error: "write failed Authorization: Bearer secret-token", + }, + }, + }, }, { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); const stream = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'test', - workMode: 'execute' + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "test", + workMode: "execute", }, - new AbortController().signal - ) + new AbortController().signal, + ); await expect(stream.next()).resolves.toMatchObject({ - value: { type: 'status' } - }) + value: { type: "status" }, + }); await expect(stream.next()).resolves.toMatchObject({ value: { - type: 'tool', - callId: 'call-1', - state: 'failed', - error: - 'write failed Authorization: Bearer secret-token' - } - }) + type: "tool", + callId: "call-1", + state: "failed", + error: "write failed Authorization: Bearer secret-token", + }, + }); await expect(stream.next()).rejects.toThrow( - 'write failed Authorization: Bearer secret-token' - ) - expect(session.abort).toHaveBeenCalledOnce() - await runtime.dispose() - }) + "write failed Authorization: Bearer secret-token", + ); + expect(session.abort).toHaveBeenCalledOnce(); + await runtime.dispose(); + }); - it('keeps a completed response when an earlier tool attempt failed', async () => { + it("keeps a completed response when an earlier tool attempt failed", async () => { const { client, session } = runClient([ { - id: 'event-tool-error', - type: 'message.part.updated', + id: "event-tool-error", + type: "message.part.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", part: { - id: 'part-1', - callID: 'call-1', - type: 'tool', - tool: 'read', + id: "part-1", + callID: "call-1", + type: "tool", + tool: "read", state: { - status: 'error', - error: 'Cannot read binary file' - } - } - } + status: "error", + error: "Cannot read binary file", + }, + }, + }, }, - completedToolEvent('call-2', 'write'), + completedToolEvent("call-2", "write"), { - id: 'event-text', - type: 'message.part.delta', + id: "event-text", + type: "message.part.delta", properties: { - sessionID: 'session-1', - messageID: 'message-1', - partID: 'part-text', - field: 'text', - delta: 'PPT 已生成并保存。' - } + sessionID: "session-1", + messageID: "message-1", + partID: "part-text", + field: "text", + delta: "PPT 已生成并保存。", + }, }, { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) - const events = await collectRun(runtime, 'execute') + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); + const events = await collectRun(runtime, "execute"); expect( events.filter( - (event) => - event.type === 'tool' && event.callId === 'call-1' - ) + (event) => event.type === "tool" && event.callId === "call-1", + ), ).toEqual([ expect.objectContaining({ - state: 'failed', - error: 'Cannot read binary file' + state: "failed", + error: "Cannot read binary file", }), expect.objectContaining({ - state: 'recoverable', - error: 'Cannot read binary file' - }) - ]) + state: "recoverable", + error: "Cannot read binary file", + }), + ]); expect(events).toContainEqual( expect.objectContaining({ - type: 'text', - delta: 'PPT 已生成并保存。' - }) - ) - expect(events.at(-1)).toMatchObject({ type: 'done' }) - expect(session.abort).not.toHaveBeenCalled() - await runtime.dispose() - }) + type: "text", + delta: "PPT 已生成并保存。", + }), + ); + expect(events.at(-1)).toMatchObject({ type: "done" }); + expect(session.abort).not.toHaveBeenCalled(); + await runtime.dispose(); + }); - it('surfaces a rejected async prompt instead of reporting success', async () => { + it("surfaces a rejected async prompt instead of reporting success", async () => { const { client, session } = runClient([ { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); vi.mocked(session.promptAsync).mockResolvedValueOnce({ data: undefined, error: { data: { - message: - 'prompt rejected Authorization: Bearer secret-token' - } - } - } as never) - const runtime = embeddedRuntime(client) + message: "prompt rejected Authorization: Bearer secret-token", + }, + }, + } as never); + const runtime = embeddedRuntime(client); await expect(collectRun(runtime)).rejects.toThrow( - 'prompt rejected Authorization: Bearer secret-token' - ) - await runtime.dispose() - }) + "prompt rejected Authorization: Bearer secret-token", + ); + await runtime.dispose(); + }); - it('deletes an ephemeral OpenCode session when released', async () => { + it("deletes an ephemeral OpenCode session when released", async () => { const { client, session } = runClient([ { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); - await collectRun(runtime) - await runtime.releaseConversation('conversation-1') + await collectRun(runtime); + await runtime.releaseConversation("conversation-1"); expect(session.delete).toHaveBeenCalledWith( { - sessionID: 'session-1', - directory: process.cwd() + sessionID: "session-1", + directory: process.cwd(), }, - { signal: expect.any(AbortSignal) } - ) - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ); + await runtime.dispose(); + }); - it('ignores unrelated requests and rejects bounded malformed requests without prompting', async () => { + it("ignores unrelated requests and rejects bounded malformed requests without prompting", async () => { const { client, permissionReply } = runClient([ - permissionEvent({ sessionID: 'unrelated-session' }), + permissionEvent({ sessionID: "unrelated-session" }), permissionEvent({ - patterns: Array.from({ length: 33 }, () => '*') + patterns: Array.from({ length: 33 }, () => "*"), }), { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) - await collectRun(runtime, 'execute') + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); + await collectRun(runtime, "execute"); - expect(permissionReply).toHaveBeenCalledOnce() + expect(permissionReply).toHaveBeenCalledOnce(); expect(permissionReply).toHaveBeenCalledWith( { - requestID: 'permission-1', + requestID: "permission-1", directory: process.cwd(), - reply: 'reject' + reply: "reject", }, - { signal: expect.any(AbortSignal) } - ) - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ); + await runtime.dispose(); + }); - it('fails closed when the OpenCode permission reply fails', async () => { - const { client, permissionReply, session } = runClient([ - permissionEvent() - ]) + it("fails closed when the OpenCode permission reply fails", async () => { + const { client, permissionReply, session } = runClient([permissionEvent()]); permissionReply.mockResolvedValue({ data: false, - error: { message: 'secret server error' } - }) - const runtime = embeddedRuntime(client) + error: { message: "secret server error" }, + }); + const runtime = embeddedRuntime(client); - await expect( - collectRun(runtime, 'execute') - ).rejects.toThrow('OpenCode 权限回复失败') + await expect(collectRun(runtime, "execute")).rejects.toThrow( + "OpenCode 权限回复失败", + ); expect(session.abort).toHaveBeenCalledWith({ - sessionID: 'session-1', - directory: process.cwd() - }) - await runtime.dispose() - }) + sessionID: "session-1", + directory: process.cwd(), + }); + await runtime.dispose(); + }); - it('uses deny-all session rules and hard tool disable in Ask mode', async () => { - const { client, session, tool } = runClient([ - { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) + it("uses deny-all session rules and hard tool disable in Ask mode", async () => { + const { client, session, tool } = runClient([ + { + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); - await collectRun(runtime, 'ask') + await collectRun(runtime, "ask"); - expect(session.create).toHaveBeenCalledWith( - { - title: 'GoodBuddy 对话', - directory: process.cwd(), - permission: [ - { permission: '*', pattern: '*', action: 'deny' } - ] + expect(session.create).toHaveBeenCalledWith( + { + title: "GoodBuddy 对话", + directory: process.cwd(), + permission: [{ permission: "*", pattern: "*", action: "deny" }], + }, + { signal: expect.any(AbortSignal) }, + ); + expect(tool.ids).toHaveBeenCalledWith( + { directory: process.cwd() }, + { signal: expect.any(AbortSignal) }, + ); + expect(session.promptAsync).toHaveBeenCalledWith( + expect.objectContaining({ + tools: { + read: false, + write: false, + bash: false, + task: false, + "goodbuddy-data-*": false, + "goodbuddy-custom-*": false, }, - { signal: expect.any(AbortSignal) } - ) - expect(tool.ids).toHaveBeenCalledWith( - { directory: process.cwd() }, - { signal: expect.any(AbortSignal) } - ) - expect(session.promptAsync).toHaveBeenCalledWith( - expect.objectContaining({ - tools: { - read: false, - write: false, - bash: false, - task: false, - 'goodbuddy-data-*': false, - 'goodbuddy-custom-*': false - } - }), - expect.anything() - ) - await runtime.dispose() - }) + }), + expect.anything(), + ); + await runtime.dispose(); + }); - it('updates reused sessions when the work mode changes', async () => { + it("updates reused sessions when the work mode changes", async () => { const { client, session } = runClient([ { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); - await collectRun(runtime, 'execute') - await collectRun(runtime, 'ask') + await collectRun(runtime, "execute"); + await collectRun(runtime, "ask"); expect(session.update).toHaveBeenCalledWith( { - sessionID: 'session-1', + sessionID: "session-1", directory: process.cwd(), - permission: [ - { permission: '*', pattern: '*', action: 'deny' } - ] + permission: [{ permission: "*", pattern: "*", action: "deny" }], }, - { signal: expect.any(AbortSignal) } - ) - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ); + await runtime.dispose(); + }); - it('configures external Execute sessions without whole-run approval', async () => { + it("configures external Execute sessions without whole-run approval", async () => { const { client, session, permissionReply } = runClient([ permissionEvent(), { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', - embedded: false + baseUrl: "http://127.0.0.1:4096", + embedded: false, }), { createClient: vi.fn( - () => client - ) as unknown as typeof createOpencodeClient - } - ) - await collectRun(runtime, 'execute') + () => client, + ) as unknown as typeof createOpencodeClient, + }, + ); + await collectRun(runtime, "execute"); - expect(runtime.requiresToolApproval).toBe(false) + expect(runtime.requiresToolApproval).toBe(false); expect(session.create).toHaveBeenCalledWith( { - title: 'GoodBuddy 对话', + title: "GoodBuddy 对话", directory: process.cwd(), - permission: [ - { permission: '*', pattern: '*', action: 'allow' } - ] + permission: [{ permission: "*", pattern: "*", action: "allow" }], }, - { signal: expect.any(AbortSignal) } - ) - expect(permissionReply).not.toHaveBeenCalled() - await runtime.dispose() - }) -}) + { signal: expect.any(AbortSignal) }, + ); + expect(permissionReply).not.toHaveBeenCalled(); + await runtime.dispose(); + }); +}); -describe('OpenCodeRuntime native customization', () => { - it('maps bounded native inventory and filters GoodBuddy-owned capabilities', async () => { +describe("OpenCodeRuntime native customization", () => { + it("maps bounded native inventory and filters GoodBuddy-owned capabilities", async () => { const sourceRoot = await mkdtemp( - join(tmpdir(), 'goodbuddy-opencode-native-snapshot-') - ) - const assignedSkillDirectory = join( - sourceRoot, - 'assigned-skill' - ) - await mkdir(assignedSkillDirectory) + join(tmpdir(), "goodbuddy-opencode-native-snapshot-"), + ); + const assignedSkillDirectory = join(sourceRoot, "assigned-skill"); + await mkdir(assignedSkillDirectory); await writeFile( - join(assignedSkillDirectory, 'SKILL.md'), + join(assignedSkillDirectory, "SKILL.md"), [ - '---', - 'name: assigned-skill', - 'description: Assigned test skill', - '---', - '', - '# Assigned skill' - ].join('\n'), - 'utf8' - ) + "---", + "name: assigned-skill", + "description: Assigned test skill", + "---", + "", + "# Assigned skill", + ].join("\n"), + "utf8", + ); const client = { session: { list: vi.fn().mockResolvedValue({ data: [], - error: undefined - }) + error: undefined, + }), }, app: { agents: vi.fn().mockResolvedValue({ data: [ { - name: 'build', - description: 'Primary builder', - mode: 'primary', + name: "build", + description: "Primary builder", + mode: "primary", native: true, hidden: false, permission: [], - options: {} + options: {}, }, { - name: 'hidden', - mode: 'all', + name: "hidden", + mode: "all", hidden: true, permission: [], - options: {} - } - ] + options: {}, + }, + ], }), skills: vi.fn().mockResolvedValue({ data: [ { - name: 'native-skill', - description: 'Native skill', - location: 'C:\\private\\native', - content: 'must not be exposed' + name: "native-skill", + description: "Native skill", + location: "C:\\private\\native", + content: "must not be exposed", }, { - name: 'assigned-skill', - location: 'C:\\private\\assigned', - content: 'assigned content' - } - ] - }) + name: "assigned-skill", + location: "C:\\private\\assigned", + content: "assigned content", + }, + ], + }), }, tool: { ids: vi.fn().mockResolvedValue({ data: [ - 'apply_patch', - 'bash', - 'edit', - 'glob', - 'grep', - 'invalid', - 'question', - 'read', - 'skill', - 'task', - 'todowrite', - 'webfetch', - 'websearch', - 'write', - 'goodbuddy-data-123_search', - 'extension_tool' + "apply_patch", + "bash", + "edit", + "glob", + "grep", + "invalid", + "question", + "read", + "skill", + "task", + "todowrite", + "webfetch", + "websearch", + "write", + "goodbuddy-data-123_search", + "extension_tool", ], - error: undefined - }) + error: undefined, + }), }, command: { list: vi.fn().mockResolvedValue({ data: [ { - name: 'review', - description: 'Review changes', - source: 'command', - template: 'private command template', - hints: [] + name: "review", + description: "Review changes", + source: "command", + template: "private command template", + hints: [], }, { - name: 'mcp-prompt', - description: 'Prompt from MCP', - source: 'mcp', - template: 'Inspect $ARGUMENTS', - hints: [] + name: "mcp-prompt", + description: "Prompt from MCP", + source: "mcp", + template: "Inspect $ARGUMENTS", + hints: [], }, { - name: 'assigned-skill', - source: 'skill', - template: 'assigned skill template', - hints: [] + name: "assigned-skill", + source: "skill", + template: "assigned skill template", + hints: [], }, { - name: 'goodbuddy-data-123', - source: 'mcp', - template: 'temporary prompt', - hints: [] - } - ] - }) + name: "goodbuddy-data-123", + source: "mcp", + template: "temporary prompt", + hints: [], + }, + ], + }), }, lsp: { status: vi.fn().mockResolvedValue({ data: [ { - id: 'typescript', - name: 'TypeScript', - root: 'C:\\private\\workspace', - status: 'connected' - } - ] - }) + id: "typescript", + name: "TypeScript", + root: "C:\\private\\workspace", + status: "connected", + }, + ], + }), }, formatter: { status: vi.fn().mockResolvedValue({ data: [ { - name: 'prettier', + name: "prettier", enabled: true, - extensions: ['.ts', '.tsx'] - } - ] - }) + extensions: [".ts", ".tsx"], + }, + ], + }), }, mcp: { status: vi.fn().mockResolvedValue({ data: { - public: { status: 'failed', error: 'private failure' }, - 'goodbuddy-custom-123': { status: 'connected' } - } - }) + public: { status: "failed", error: "private failure" }, + "goodbuddy-custom-123": { status: "connected" }, + }, + }), }, experimental: { resource: { list: vi.fn().mockResolvedValue({ data: { - 'public-resource': { - name: 'Public resource', - uri: 'docs://public', - description: 'Reference', - mimeType: 'text/plain', - client: 'public' + "public-resource": { + name: "Public resource", + uri: "docs://public", + description: "Reference", + mimeType: "text/plain", + client: "public", }, temporary: { - name: 'Temporary resource', - uri: 'docs://temporary', - client: 'goodbuddy-data-123' - } - } - }) - } - } - } as unknown as ReturnType + name: "Temporary resource", + uri: "docs://temporary", + client: "goodbuddy-data-123", + }, + }, + }), + }, + }, + } as unknown as ReturnType; const runtime = embeddedRuntime(client, { skillPackages: [ { - id: 'assigned-skill', - directory: assignedSkillDirectory - } - ] - }) + id: "assigned-skill", + directory: assignedSkillDirectory, + }, + ], + }); - const snapshot = await runtime.getNativeSnapshot() + const snapshot = await runtime.getNativeSnapshot(); expect(snapshot).toMatchObject({ available: true, - inventoryStatus: 'available', - detail: 'OpenCode 原生能力已就绪', + inventoryStatus: "available", + detail: "OpenCode 原生能力已就绪", agents: [ { - id: 'build', - mode: 'primary', + id: "build", + mode: "primary", native: true, - hidden: false + hidden: false, }, { - id: 'hidden', - hidden: true - } + id: "hidden", + hidden: true, + }, ], toolsSupported: true, tools: expect.arrayContaining([ { - id: 'edit', - name: 'edit', - kind: 'write', - source: 'runtime', - ask: 'blocked', - execute: 'allowed' + id: "edit", + name: "edit", + kind: "write", + source: "runtime", + ask: "blocked", + execute: "allowed", }, { - id: 'read', - name: 'read', - kind: 'read', - source: 'runtime', - ask: 'blocked', - execute: 'allowed' + id: "read", + name: "read", + kind: "read", + source: "runtime", + ask: "blocked", + execute: "allowed", }, { - id: 'skill', - name: 'skill', - kind: 'agent', - source: 'runtime', - ask: 'conditional', - execute: 'allowed' + id: "skill", + name: "skill", + kind: "agent", + source: "runtime", + ask: "conditional", + execute: "allowed", }, { - id: 'extension_tool', - name: 'extension_tool', - kind: 'other', - source: 'unknown', - ask: 'blocked', - execute: 'allowed' - } + id: "extension_tool", + name: "extension_tool", + kind: "other", + source: "unknown", + ask: "blocked", + execute: "allowed", + }, ]), commands: [ { - id: 'review', - source: 'command' + id: "review", + source: "command", }, { - id: 'mcp-prompt', - source: 'mcp' - } + id: "mcp-prompt", + source: "mcp", + }, ], prompts: [ { - id: 'mcp-prompt', - prompt: 'Inspect $ARGUMENTS', - source: 'mcp' - } + id: "mcp-prompt", + prompt: "Inspect $ARGUMENTS", + source: "mcp", + }, ], lsp: [ { - id: 'typescript', - name: 'TypeScript', - status: 'connected' - } + id: "typescript", + name: "TypeScript", + status: "connected", + }, ], formatters: [ { - id: 'prettier', + id: "prettier", enabled: true, - extensions: ['.ts', '.tsx'] - } + extensions: [".ts", ".tsx"], + }, ], mcpServers: [ { - id: 'public', - status: 'failed' - } + id: "public", + status: "failed", + }, ], skills: [ { - id: 'native-skill', - description: 'Native skill' - } + id: "native-skill", + description: "Native skill", + }, ], resources: [ { - id: 'public-resource', - uri: 'docs://public', - server: 'public' - } + id: "public-resource", + uri: "docs://public", + server: "public", + }, ], resourcesSupported: true, context: { - strategy: 'native', - manualCompact: true - } - }) - const serialized = JSON.stringify(snapshot) - expect(serialized).not.toContain('private failure') - expect(serialized).not.toContain('private command template') - expect(serialized).not.toContain('must not be exposed') - expect(serialized).not.toContain('C:\\private') - expect(serialized).not.toContain('assigned-skill') - expect(serialized).not.toContain('goodbuddy-data-') - expect(serialized).not.toContain('goodbuddy-custom-') - expect(serialized).not.toContain('"invalid"') + strategy: "native", + manualCompact: true, + }, + }); + const serialized = JSON.stringify(snapshot); + expect(serialized).not.toContain("private failure"); + expect(serialized).not.toContain("private command template"); + expect(serialized).not.toContain("must not be exposed"); + expect(serialized).not.toContain("C:\\private"); + expect(serialized).not.toContain("assigned-skill"); + expect(serialized).not.toContain("goodbuddy-data-"); + expect(serialized).not.toContain("goodbuddy-custom-"); + expect(serialized).not.toContain('"invalid"'); vi.mocked(client.tool.ids).mockRejectedValueOnce( - new Error('tool inventory unavailable') - ) - const partialSnapshot = await runtime.getNativeSnapshot() + new Error("tool inventory unavailable"), + ); + const partialSnapshot = await runtime.getNativeSnapshot(); expect(partialSnapshot).toMatchObject({ available: true, - inventoryStatus: 'partial', + inventoryStatus: "partial", tools: [], - toolsSupported: false - }) - expect(partialSnapshot.detail).toContain('工具') - await runtime.dispose() - await rm(sourceRoot, { recursive: true, force: true }) - }) + toolsSupported: false, + }); + expect(partialSnapshot.detail).toContain("工具"); + await runtime.dispose(); + await rm(sourceRoot, { recursive: true, force: true }); + }); - it('reports external OpenCode connectivity without claiming readable native inventory', async () => { - const client = runClient([]).client + it("reports external OpenCode connectivity without claiming readable native inventory", async () => { + const client = runClient([]).client; const runtime = new OpenCodeRuntime( options({ - baseUrl: 'http://127.0.0.1:4096', - embedded: false + baseUrl: "http://127.0.0.1:4096", + embedded: false, }), { createClient: vi.fn( - () => client - ) as unknown as typeof createOpencodeClient - } - ) + () => client, + ) as unknown as typeof createOpencodeClient, + }, + ); await expect(runtime.getNativeSnapshot()).resolves.toMatchObject({ available: true, - inventoryStatus: 'connection-only', + inventoryStatus: "connection-only", tools: [], - toolsSupported: false - }) - expect(client.tool.ids).not.toHaveBeenCalled() - await runtime.dispose() - }) + toolsSupported: false, + }); + expect(client.tool.ids).not.toHaveBeenCalled(); + await runtime.dispose(); + }); - it('uses an explicit valid agent over the configured default', async () => { + it("uses an explicit valid agent over the configured default", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); Object.assign(setup.client, { app: { agents: vi.fn().mockResolvedValue({ data: [ { - name: 'build', - mode: 'primary', + name: "build", + mode: "primary", hidden: false, permission: [], - options: {} + options: {}, }, { - name: 'plan', - mode: 'all', + name: "plan", + mode: "all", hidden: false, permission: [], - options: {} - } - ] - }) - } - }) + options: {}, + }, + ], + }), + }, + }); const runtime = embeddedRuntime(setup.client, { - customization: { defaultAgent: 'build' } - }) + customization: { defaultAgent: "build" }, + }); for await (const _event of runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'test', - workMode: 'execute', + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "test", + workMode: "execute", runtimeControl: { - provider: 'opencode', - agent: 'plan' - } + provider: "opencode", + agent: "plan", + }, }, - new AbortController().signal + new AbortController().signal, )) { - void _event + void _event; } expect(setup.session.create).toHaveBeenCalledWith( - expect.objectContaining({ agent: 'plan' }), - { signal: expect.any(AbortSignal) } - ) + expect.objectContaining({ agent: "plan" }), + { signal: expect.any(AbortSignal) }, + ); expect(setup.session.promptAsync).toHaveBeenCalledWith( - expect.objectContaining({ agent: 'plan' }), - expect.anything() - ) - await runtime.dispose() - }) + expect.objectContaining({ agent: "plan" }), + expect.anything(), + ); + await runtime.dispose(); + }); - it('uses the configured default agent when no request override is present', async () => { + it("uses the configured default agent when no request override is present", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); Object.assign(setup.client, { app: { agents: vi.fn().mockResolvedValue({ data: [ { - name: 'build', - mode: 'primary', + name: "build", + mode: "primary", hidden: false, permission: [], - options: {} - } - ] - }) - } - }) + options: {}, + }, + ], + }), + }, + }); const runtime = embeddedRuntime(setup.client, { - customization: { defaultAgent: 'build' } - }) + customization: { defaultAgent: "build" }, + }); - await collectRun(runtime) + await collectRun(runtime); expect(setup.session.create).toHaveBeenCalledWith( - expect.objectContaining({ agent: 'build' }), - { signal: expect.any(AbortSignal) } - ) + expect.objectContaining({ agent: "build" }), + { signal: expect.any(AbortSignal) }, + ); expect(setup.session.promptAsync).toHaveBeenCalledWith( - expect.objectContaining({ agent: 'build' }), - expect.anything() - ) - await runtime.dispose() - }) + expect.objectContaining({ agent: "build" }), + expect.anything(), + ); + await runtime.dispose(); + }); - it('rejects a stale or hidden agent instead of falling back', async () => { - const setup = runClient([]) + it("rejects a stale or hidden agent instead of falling back", async () => { + const setup = runClient([]); Object.assign(setup.client, { app: { agents: vi.fn().mockResolvedValue({ data: [ { - name: 'hidden', - mode: 'primary', + name: "hidden", + mode: "primary", hidden: true, permission: [], - options: {} - } - ] - }) - } - }) - const runtime = embeddedRuntime(setup.client) + options: {}, + }, + ], + }), + }, + }); + const runtime = embeddedRuntime(setup.client); const stream = runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'test', - workMode: 'execute', + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "test", + workMode: "execute", runtimeControl: { - provider: 'opencode', - agent: 'hidden' - } + provider: "opencode", + agent: "hidden", + }, }, - new AbortController().signal - ) + new AbortController().signal, + ); await expect(stream.next()).rejects.toThrow( - 'OpenCode Agent 不存在、已隐藏或不可作为主 Agent：hidden' - ) - expect(setup.session.create).not.toHaveBeenCalled() - expect(setup.session.promptAsync).not.toHaveBeenCalled() - await runtime.dispose() - }) + "OpenCode Agent 不存在、已隐藏或不可作为主 Agent：hidden", + ); + expect(setup.session.create).not.toHaveBeenCalled(); + expect(setup.session.promptAsync).not.toHaveBeenCalled(); + await runtime.dispose(); + }); - it('executes validated native commands through the command API', async () => { + it("executes validated native commands through the command API", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const command = vi.fn().mockResolvedValue({ - data: { info: {}, parts: [] } - }) + data: { info: {}, parts: [] }, + }); Object.assign(setup.client, { app: { agents: vi.fn().mockResolvedValue({ data: [ { - name: 'build', - mode: 'primary', + name: "build", + mode: "primary", hidden: false, permission: [], - options: {} - } - ] - }) + options: {}, + }, + ], + }), }, command: { list: vi.fn().mockResolvedValue({ data: [ { - name: 'review', - source: 'command', - template: 'Review $ARGUMENTS', - hints: [] - } - ] - }) - } - }) - Object.assign(setup.client.session, { command }) - const runtime = embeddedRuntime(setup.client) + name: "review", + source: "command", + template: "Review $ARGUMENTS", + hints: [], + }, + ], + }), + }, + }); + Object.assign(setup.client.session, { command }); + const runtime = embeddedRuntime(setup.client); - const events = [] + const events = []; for await (const event of runtime.run( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - prompt: 'must not become slash text', - workMode: 'execute', + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + prompt: "must not become slash text", + workMode: "execute", runtimeControl: { - provider: 'opencode', - agent: 'build', + provider: "opencode", + agent: "build", command: { - name: 'review', - arguments: '--staged' - } - } + name: "review", + arguments: "--staged", + }, + }, }, - new AbortController().signal + new AbortController().signal, )) { - events.push(event) + events.push(event); } expect(command).toHaveBeenCalledWith( { - sessionID: 'session-1', + sessionID: "session-1", directory: process.cwd(), - command: 'review', - arguments: '--staged', - agent: 'build' + command: "review", + arguments: "--staged", + agent: "build", }, expect.objectContaining({ - signal: expect.any(AbortSignal) - }) - ) - expect(setup.session.promptAsync).not.toHaveBeenCalled() + signal: expect.any(AbortSignal), + }), + ); + expect(setup.session.promptAsync).not.toHaveBeenCalled(); expect(events.at(-1)).toMatchObject({ - type: 'done', - sessionId: 'session-1' - }) - await runtime.dispose() - }) + type: "done", + sessionId: "session-1", + }); + await runtime.dispose(); + }); - it('compacts a managed session through the supported native API', async () => { + it("compacts a managed session through the supported native API", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const context = vi.fn().mockResolvedValue({ data: { data: [ { - type: 'assistant', + type: "assistant", model: { - providerID: 'anthropic', - id: 'claude-sonnet' - } - } - ] - } - }) + providerID: "anthropic", + id: "claude-sonnet", + }, + }, + ], + }, + }); const summarize = vi.fn().mockResolvedValue({ data: true, - error: undefined - }) + error: undefined, + }); Object.assign(setup.client, { v2: { - session: { context } + session: { context }, }, - session: { ...setup.client.session, summarize } - }) - const runtime = embeddedRuntime(setup.client) - await collectRun(runtime) + session: { ...setup.client.session, summarize }, + }); + const runtime = embeddedRuntime(setup.client); + await collectRun(runtime); vi.mocked(setup.event.subscribe).mockResolvedValueOnce({ stream: (async function* () { yield { - type: 'message.updated', + type: "message.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", info: { - id: 'compaction-message', - sessionID: 'session-1', - role: 'assistant', + id: "compaction-message", + sessionID: "session-1", + role: "assistant", time: { created: 1, - completed: 2 + completed: 2, }, - parentID: 'compaction-parent', - modelID: 'claude-sonnet', - providerID: 'anthropic', - mode: 'compaction', - agent: 'build', + parentID: "compaction-parent", + modelID: "claude-sonnet", + providerID: "anthropic", + mode: "compaction", + agent: "build", path: { cwd: process.cwd(), - root: process.cwd() + root: process.cwd(), }, cost: 0, tokens: { @@ -4023,161 +4197,161 @@ describe('OpenCodeRuntime native customization', () => { reasoning: 0, cache: { read: 30, - write: 4 + write: 4, }, - total: 124 + total: 124, }, - finish: 'stop' - } - } - } + finish: "stop", + }, + }, + }; yield { - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - })() - } as never) - const signal = new AbortController().signal + type: "session.idle", + properties: { sessionID: "session-1" }, + }; + })(), + } as never); + const signal = new AbortController().signal; await expect( runtime.compactConversation( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - runtimeSelection: { provider: 'opencode' }, + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + runtimeSelection: { provider: "opencode" }, history: [], - historyMessageIds: [] + historyMessageIds: [], }, - signal - ) + signal, + ), ).resolves.toEqual({ result: { - provider: 'opencode', - strategy: 'native', + provider: "opencode", + strategy: "native", compacted: true, - detail: 'OpenCode 已完成原生上下文压缩' + detail: "OpenCode 已完成原生上下文压缩", }, usageEvents: [ { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - type: 'model-usage', - callId: 'compaction-message', - runtime: 'opencode', - provider: 'anthropic', - model: 'claude-sonnet', + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + type: "model-usage", + callId: "compaction-message", + runtime: "opencode", + provider: "anthropic", + model: "claude-sonnet", inputTokens: 100, outputTokens: 20, cacheReadTokens: 30, cacheWriteTokens: 4, - reportedTotalTokens: 124 - } - ] - }) + reportedTotalTokens: 124, + }, + ], + }); expect(context).toHaveBeenCalledWith( - { sessionID: 'session-1' }, - { signal: expect.any(AbortSignal) } - ) + { sessionID: "session-1" }, + { signal: expect.any(AbortSignal) }, + ); expect(summarize).toHaveBeenCalledWith( { - sessionID: 'session-1', + sessionID: "session-1", directory: process.cwd(), - providerID: 'anthropic', - modelID: 'claude-sonnet', - auto: false + providerID: "anthropic", + modelID: "claude-sonnet", + auto: false, }, - { signal: expect.any(AbortSignal) } - ) - await runtime.dispose() - }) + { signal: expect.any(AbortSignal) }, + ); + await runtime.dispose(); + }); - it('reports when a managed session has no model to compact with', async () => { + it("reports when a managed session has no model to compact with", async () => { const setup = runClient([ { - id: 'idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) + id: "idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); const context = vi.fn().mockResolvedValue({ - data: { data: [] } - }) - const summarize = vi.fn() + data: { data: [] }, + }); + const summarize = vi.fn(); Object.assign(setup.client, { v2: { - session: { context } + session: { context }, }, - session: { ...setup.client.session, summarize } - }) - const runtime = embeddedRuntime(setup.client) - await collectRun(runtime) + session: { ...setup.client.session, summarize }, + }); + const runtime = embeddedRuntime(setup.client); + await collectRun(runtime); await expect( runtime.compactConversation( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'conversation-1', - runtimeSelection: { provider: 'opencode' }, + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "conversation-1", + runtimeSelection: { provider: "opencode" }, history: [], - historyMessageIds: [] + historyMessageIds: [], }, - new AbortController().signal - ) + new AbortController().signal, + ), ).resolves.toEqual({ result: { - provider: 'opencode', - strategy: 'native', + provider: "opencode", + strategy: "native", compacted: false, - detail: '当前 OpenCode 会话尚无可用于压缩的模型记录' - } - }) - expect(summarize).not.toHaveBeenCalled() - await runtime.dispose() - }) + detail: "当前 OpenCode 会话尚无可用于压缩的模型记录", + }, + }); + expect(summarize).not.toHaveBeenCalled(); + await runtime.dispose(); + }); - it('reports when no managed OpenCode session can be compacted', async () => { - const runtime = new OpenCodeRuntime(options()) + it("reports when no managed OpenCode session can be compacted", async () => { + const runtime = new OpenCodeRuntime(options()); await expect( runtime.compactConversation( { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - conversationId: 'missing-conversation', - runtimeSelection: { provider: 'opencode' }, + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + conversationId: "missing-conversation", + runtimeSelection: { provider: "opencode" }, history: [], - historyMessageIds: [] + historyMessageIds: [], }, - new AbortController().signal - ) + new AbortController().signal, + ), ).resolves.toEqual({ result: { - provider: 'opencode', - strategy: 'native', + provider: "opencode", + strategy: "native", compacted: false, - detail: '当前 GoodBuddy 对话尚无可压缩的 OpenCode 会话' - } - }) - await runtime.dispose() - }) -}) + detail: "当前 GoodBuddy 对话尚无可压缩的 OpenCode 会话", + }, + }); + await runtime.dispose(); + }); +}); -describe('OpenCodeRuntime model usage', () => { - it('emits one provider-reported usage event for each terminal assistant message', async () => { +describe("OpenCodeRuntime model usage", () => { + it("emits one provider-reported usage event for each terminal assistant message", async () => { const assistantMessage = { - id: 'message-assistant-1', - sessionID: 'session-1', - role: 'assistant', + id: "message-assistant-1", + sessionID: "session-1", + role: "assistant", time: { created: 1, - completed: 2 + completed: 2, }, - parentID: 'message-user-1', - modelID: 'claude-sonnet-provider', - providerID: 'anthropic', - mode: 'build', - agent: 'build', + parentID: "message-user-1", + modelID: "claude-sonnet-provider", + providerID: "anthropic", + mode: "build", + agent: "build", path: { cwd: process.cwd(), - root: process.cwd() + root: process.cwd(), }, cost: 0.01, tokens: { @@ -4187,93 +4361,91 @@ describe('OpenCodeRuntime model usage', () => { reasoning: 3, cache: { read: 7, - write: 5 - } - } - } + write: 5, + }, + }, + }; const { client } = runClient([ { - id: 'event-incomplete', - type: 'message.updated', + id: "event-incomplete", + type: "message.updated", properties: { - sessionID: 'session-1', + sessionID: "session-1", info: { ...assistantMessage, - time: { created: 1 } - } - } + time: { created: 1 }, + }, + }, }, { - id: 'event-terminal', - type: 'message.updated', + id: "event-terminal", + type: "message.updated", properties: { - sessionID: 'session-1', - info: assistantMessage - } + sessionID: "session-1", + info: assistantMessage, + }, }, { - id: 'event-terminal-duplicate', - type: 'message.updated', + id: "event-terminal-duplicate", + type: "message.updated", properties: { - sessionID: 'session-1', - info: assistantMessage - } + sessionID: "session-1", + info: assistantMessage, + }, }, { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); - const events = await collectRun(runtime) + const events = await collectRun(runtime); - expect( - events.filter((event) => event.type === 'model-usage') - ).toEqual([ + expect(events.filter((event) => event.type === "model-usage")).toEqual([ { - requestId: '3f496642-f47d-4e0a-8944-a32c77b0d6ef', - type: 'model-usage', - callId: 'message-assistant-1', - runtime: 'opencode', - provider: 'anthropic', - model: 'claude-sonnet-provider', + requestId: "3f496642-f47d-4e0a-8944-a32c77b0d6ef", + type: "model-usage", + callId: "message-assistant-1", + runtime: "opencode", + provider: "anthropic", + model: "claude-sonnet-provider", inputTokens: 23, outputTokens: 11, cacheReadTokens: 7, cacheWriteTokens: 5, - reportedTotalTokens: 42 - } - ]) - expect(events.at(-2)).toMatchObject({ type: 'model-usage' }) - expect(events.at(-1)).toMatchObject({ type: 'done' }) - await runtime.dispose() - }) + reportedTotalTokens: 42, + }, + ]); + expect(events.at(-2)).toMatchObject({ type: "model-usage" }); + expect(events.at(-1)).toMatchObject({ type: "done" }); + await runtime.dispose(); + }); - it('ignores assistant usage from another session', async () => { + it("ignores assistant usage from another session", async () => { const { client } = runClient([ { - id: 'event-unrelated-usage', - type: 'message.updated', + id: "event-unrelated-usage", + type: "message.updated", properties: { - sessionID: 'session-2', + sessionID: "session-2", info: { - id: 'message-assistant-2', - sessionID: 'session-2', - role: 'assistant', + id: "message-assistant-2", + sessionID: "session-2", + role: "assistant", time: { created: 1, - completed: 2 + completed: 2, }, - parentID: 'message-user-2', - modelID: 'unrelated-model', - providerID: 'unrelated-provider', - mode: 'build', - agent: 'build', + parentID: "message-user-2", + modelID: "unrelated-model", + providerID: "unrelated-provider", + mode: "build", + agent: "build", path: { cwd: process.cwd(), - root: process.cwd() + root: process.cwd(), }, cost: 0, tokens: { @@ -4282,25 +4454,23 @@ describe('OpenCodeRuntime model usage', () => { reasoning: 0, cache: { read: 0, - write: 0 - } - } - } - } + write: 0, + }, + }, + }, + }, }, { - id: 'event-idle', - type: 'session.idle', - properties: { sessionID: 'session-1' } - } - ]) - const runtime = embeddedRuntime(client) + id: "event-idle", + type: "session.idle", + properties: { sessionID: "session-1" }, + }, + ]); + const runtime = embeddedRuntime(client); - const events = await collectRun(runtime) + const events = await collectRun(runtime); - expect( - events.filter((event) => event.type === 'model-usage') - ).toEqual([]) - await runtime.dispose() - }) -}) + expect(events.filter((event) => event.type === "model-usage")).toEqual([]); + await runtime.dispose(); + }); +}); diff --git a/src/main/agent/opencode-runtime.ts b/src/main/agent/opencode-runtime.ts index ed55113c..554d037a 100644 --- a/src/main/agent/opencode-runtime.ts +++ b/src/main/agent/opencode-runtime.ts @@ -4,10 +4,10 @@ import { type OpencodeClient, type PermissionRequest, type PermissionRuleset, - type QuestionRequest -} from '@opencode-ai/sdk/v2' -import spawn from 'cross-spawn' -import { createHash, randomBytes } from 'node:crypto' + type QuestionRequest, +} from "@opencode-ai/sdk/v2"; +import spawn from "cross-spawn"; +import { createHash, randomBytes } from "node:crypto"; import { cp, mkdir, @@ -16,222 +16,219 @@ import { rename, rm, stat, - writeFile -} from 'node:fs/promises' -import { tmpdir } from 'node:os' -import { dirname, join, resolve } from 'node:path' -import { parse as parseYaml, stringify as stringifyYaml } from 'yaml' + writeFile, +} from "node:fs/promises"; +import { tmpdir } from "node:os"; +import { dirname, join, resolve } from "node:path"; +import { parse as parseYaml, stringify as stringifyYaml } from "yaml"; import type { AgentQuestionAnswer, AgentRuntimeStatus, - SubagentEvent -} from '../../shared/contracts' + SubagentEvent, +} from "../../shared/contracts"; import { boundedRuntimeIdentifierSchema, runtimeNativeInventoryLimits, type RuntimeConversationCompactInput, type RuntimeCustomizationSettings, type RuntimeNativeSnapshot, - type RuntimeNativeTool -} from '../../shared/runtime-customization-contracts' -import { createAnthropicApiBaseUrl } from './anthropic-endpoint' -import { createOpenAIApiBaseUrl } from './openai-endpoint' + type RuntimeNativeTool, +} from "../../shared/runtime-customization-contracts"; +import { createAnthropicApiBaseUrl } from "./anthropic-endpoint"; +import { createOpenAIApiBaseUrl } from "./openai-endpoint"; import type { AgentExecutionRequest, AgentRuntime, RuntimeConversationCompactOutcome, RuntimeEvent, - RuntimeModelUsageEvent -} from './runtime' -import type { KnowledgeMcpGateway } from './knowledge-mcp-gateway' -import { detectRuntimeBinary } from './runtime-discovery' -import { getAvailableLoopbackPort } from './loopback-port' -import type { ResolvedModelProfile } from '../runtime-settings-store' + RuntimeModelUsageEvent, +} from "./runtime"; +import type { KnowledgeMcpGateway } from "./knowledge-mcp-gateway"; +import { detectRuntimeBinary } from "./runtime-discovery"; +import { getAvailableLoopbackPort } from "./loopback-port"; +import type { ResolvedModelProfile } from "../runtime-settings-store"; import { applyLaunchEnvironmentPath, buildExplicitProfileRuntimeEnvironment, buildRuntimeEnvironment, - runtimePrivacyEnvironment -} from './process-environment' -import type { LaunchEnvironmentProvider } from '../local-tool-environment/launch-environment-provider' -import { - boundedToolDetail, - safeToolErrorDetail -} from './approval-summary' + runtimePrivacyEnvironment, +} from "./process-environment"; +import type { LaunchEnvironmentProvider } from "../local-tool-environment/launch-environment-provider"; +import { boundedToolDetail, safeToolErrorDetail } from "./approval-summary"; import type { ResolvedMcpServer, - RuntimeSkillPackage -} from '../capabilities/capability-service' -import { stageRuntimeSkillPackages } from './runtime-skill-packages' + RuntimeSkillPackage, +} from "../capabilities/capability-service"; +import { stageRuntimeSkillPackages } from "./runtime-skill-packages"; import { requestProcessTreeTermination, - waitForProcessExit -} from './child-process-termination' -import { OpenCodeSubagentProgress, toOpenCodeSubagentEvent } from './opencode-subagent' -import { promptWithUntrustedConversationHistory } from './runtime-conversation-history' + waitForProcessExit, +} from "./child-process-termination"; +import { + OpenCodeSubagentProgress, + toOpenCodeSubagentEvent, +} from "./opencode-subagent"; +import { promptWithUntrustedConversationHistory } from "./runtime-conversation-history"; -const STARTUP_TIMEOUT_MS = 30_000 -const STARTUP_POLL_INTERVAL_MS = 100 -const STARTUP_PROBE_TIMEOUT_MS = 500 -const CONTROL_REQUEST_TIMEOUT_MS = 30_000 -const MAX_PERMISSION_NAME_LENGTH = 128 -const MAX_PERMISSION_PATTERNS = 32 -const MAX_PERMISSION_PATTERN_LENGTH = 1_024 -const MAX_PERMISSION_PATTERNS_BYTES = 8 * 1_024 -const MAX_PERMISSION_METADATA_BYTES = 8 * 1_024 -const MAX_EXECUTION_OUTPUT_BYTES = 1024 * 1024 -const MAX_QUESTION_REQUEST_BYTES = 32 * 1_024 -const MAX_QUESTIONS_PER_REQUEST = 4 -const MAX_QUESTION_OPTIONS = 20 -const MAX_NATIVE_AGENTS = runtimeNativeInventoryLimits.agents -const MAX_NATIVE_TOOLS = runtimeNativeInventoryLimits.tools -const MAX_NATIVE_COMMANDS = runtimeNativeInventoryLimits.commands -const MAX_NATIVE_LSP = runtimeNativeInventoryLimits.lsp -const MAX_NATIVE_FORMATTERS = runtimeNativeInventoryLimits.formatters -const MAX_NATIVE_MCP_SERVERS = - runtimeNativeInventoryLimits.mcpServers -const MAX_NATIVE_SKILLS = runtimeNativeInventoryLimits.skills -const MAX_NATIVE_RESOURCES = runtimeNativeInventoryLimits.resources -const COMPACTION_USAGE_EVENT_GRACE_MS = 1_000 -const EMBEDDED_SERVER_USERNAME = 'goodbuddy' -const OPENCODE_SKILL_NAME_PATTERN = /^[a-z0-9]+(?:-[a-z0-9]+)*$/u +const STARTUP_TIMEOUT_MS = 30_000; +const STARTUP_POLL_INTERVAL_MS = 100; +const STARTUP_PROBE_TIMEOUT_MS = 500; +const CONTROL_REQUEST_TIMEOUT_MS = 30_000; +const MAX_PERMISSION_NAME_LENGTH = 128; +const MAX_PERMISSION_PATTERNS = 32; +const MAX_PERMISSION_PATTERN_LENGTH = 1_024; +const MAX_PERMISSION_PATTERNS_BYTES = 8 * 1_024; +const MAX_PERMISSION_METADATA_BYTES = 8 * 1_024; +const MAX_EXECUTION_OUTPUT_BYTES = 1024 * 1024; +const MAX_QUESTION_REQUEST_BYTES = 32 * 1_024; +const MAX_QUESTIONS_PER_REQUEST = 4; +const MAX_QUESTION_OPTIONS = 20; +const MAX_NATIVE_AGENTS = runtimeNativeInventoryLimits.agents; +const MAX_NATIVE_TOOLS = runtimeNativeInventoryLimits.tools; +const MAX_NATIVE_COMMANDS = runtimeNativeInventoryLimits.commands; +const MAX_NATIVE_LSP = runtimeNativeInventoryLimits.lsp; +const MAX_NATIVE_FORMATTERS = runtimeNativeInventoryLimits.formatters; +const MAX_NATIVE_MCP_SERVERS = runtimeNativeInventoryLimits.mcpServers; +const MAX_NATIVE_SKILLS = runtimeNativeInventoryLimits.skills; +const MAX_NATIVE_RESOURCES = runtimeNativeInventoryLimits.resources; +const COMPACTION_USAGE_EVENT_GRACE_MS = 1_000; +const EMBEDDED_SERVER_USERNAME = "goodbuddy"; +const OPENCODE_SKILL_NAME_PATTERN = /^[a-z0-9]+(?:-[a-z0-9]+)*$/u; const TEMPORARY_MCP_PREFIXES = [ - 'goodbuddy-data-', - 'goodbuddy-custom-' -] as const + "goodbuddy-data-", + "goodbuddy-custom-", +] as const; const temporaryMcpToolOverrides = Object.fromEntries( - TEMPORARY_MCP_PREFIXES.map((prefix) => [`${prefix}*`, false]) -) -const temporaryMcpDenyRules: PermissionRuleset = - TEMPORARY_MCP_PREFIXES.map((prefix) => ({ + TEMPORARY_MCP_PREFIXES.map((prefix) => [`${prefix}*`, false]), +); +const temporaryMcpDenyRules: PermissionRuleset = TEMPORARY_MCP_PREFIXES.map( + (prefix) => ({ permission: `${prefix}*`, - pattern: '*', - action: 'deny' - })) -const OPENCODE_INTERNAL_TOOL_IDS = new Set(['invalid']) + pattern: "*", + action: "deny", + }), +); +const OPENCODE_INTERNAL_TOOL_IDS = new Set(["invalid"]); const OPENCODE_BUILTIN_TOOL_KINDS: Readonly< - Partial> + Partial> > = { - apply_patch: 'write', - bash: 'shell', - edit: 'write', - glob: 'read', - grep: 'read', - question: 'interaction', - read: 'read', - skill: 'agent', - task: 'agent', - todowrite: 'agent', - webfetch: 'network', - websearch: 'network', - write: 'write' -} + apply_patch: "write", + bash: "shell", + edit: "write", + glob: "read", + grep: "read", + question: "interaction", + read: "read", + skill: "agent", + task: "agent", + todowrite: "agent", + webfetch: "network", + websearch: "network", + write: "write", +}; -type SpawnedProcess = ReturnType +type SpawnedProcess = ReturnType; type OpenCodeProviderConfig = { - model: string + model: string; provider: Record< string, { - name: string - npm: string + name: string; + npm: string; options: { - apiKey?: string - baseURL: string - headers?: Record - } + apiKey?: string; + baseURL: string; + headers?: Record; + }; models: Record< string, { - name: string - attachment: boolean + name: string; + attachment: boolean; modalities: { - input: Array<'text' | 'image'> - output: ['text'] - } + input: Array<"text" | "image">; + output: ["text"]; + }; provider: { - npm: string - } + npm: string; + }; } - > + >; } - > -} + >; +}; type OpenCodeProviderDescriptor = { - id: string - npm: string - baseURL: string -} + id: string; + npm: string; + baseURL: string; +}; type OpenCodeServer = { - url: string - authorization: string - close: () => Promise -} + url: string; + authorization: string; + close: () => Promise; +}; type OpenCodeSkillRegistration = { - root: string - configDirectory: string - skillsRoot: string -} + root: string; + configDirectory: string; + skillsRoot: string; +}; -const sharedSkillPreparations = new Map>() -const sharedConfigPreparations = new Map>() +const sharedSkillPreparations = new Map>(); +const sharedConfigPreparations = new Map>(); const executePermissionRules: PermissionRuleset = [ - { permission: '*', pattern: '*', action: 'allow' } -] + { permission: "*", pattern: "*", action: "allow" }, +]; const readOnlyPermissionRules: PermissionRuleset = [ - { permission: '*', pattern: '*', action: 'deny' } -] + { permission: "*", pattern: "*", action: "deny" }, +]; function resolveOpenCodeProvider( - profile: ResolvedModelProfile + profile: ResolvedModelProfile, ): OpenCodeProviderDescriptor { - if (profile.protocol === 'openai-images-generations') { - throw new Error( - 'OpenCode 独立模型连接不支持图像生成协议' - ) + if (profile.protocol === "openai-images-generations") { + throw new Error("OpenCode 独立模型连接不支持图像生成协议"); } - return profile.protocol === 'anthropic-messages' + return profile.protocol === "anthropic-messages" ? { - id: 'goodbuddy-anthropic', - npm: '@ai-sdk/anthropic', - baseURL: createAnthropicApiBaseUrl(profile.baseUrl) + id: "goodbuddy-anthropic", + npm: "@ai-sdk/anthropic", + baseURL: createAnthropicApiBaseUrl(profile.baseUrl), } - : profile.protocol === 'openai-chat-completions' + : profile.protocol === "openai-chat-completions" ? { - id: 'goodbuddy-openai-chat', - npm: '@ai-sdk/openai-compatible', - baseURL: createOpenAIApiBaseUrl(profile.baseUrl) + id: "goodbuddy-openai-chat", + npm: "@ai-sdk/openai-compatible", + baseURL: createOpenAIApiBaseUrl(profile.baseUrl), } : { - id: 'goodbuddy-openai-responses', - npm: '@ai-sdk/openai', - baseURL: createOpenAIApiBaseUrl(profile.baseUrl) - } + id: "goodbuddy-openai-responses", + npm: "@ai-sdk/openai", + baseURL: createOpenAIApiBaseUrl(profile.baseUrl), + }; } function createOpenCodeProviderConfig( - profile: ResolvedModelProfile + profile: ResolvedModelProfile, ): OpenCodeProviderConfig { - const provider = resolveOpenCodeProvider(profile) + const provider = resolveOpenCodeProvider(profile); const options: { - apiKey?: string - baseURL: string - headers?: Record + apiKey?: string; + baseURL: string; + headers?: Record; } = { baseURL: provider.baseURL, - ...(profile.requestHeaders && - Object.keys(profile.requestHeaders).length > 0 + ...(profile.requestHeaders && Object.keys(profile.requestHeaders).length > 0 ? { headers: profile.requestHeaders } - : {}) - } - if (profile.authentication === 'api-key' && profile.apiKey) { - options.apiKey = profile.apiKey + : {}), + }; + if (profile.authentication === "api-key" && profile.apiKey) { + options.apiKey = profile.apiKey; } return { model: `${provider.id}/${profile.modelName}`, @@ -245,73 +242,70 @@ function createOpenCodeProviderConfig( name: profile.name, attachment: profile.supportsImageInput === true, modalities: { - input: profile.supportsImageInput === true - ? ['text', 'image'] - : ['text'], - output: ['text'] + input: + profile.supportsImageInput === true + ? ["text", "image"] + : ["text"], + output: ["text"], }, provider: { - npm: provider.npm - } - } - } - } - } - } + npm: provider.npm, + }, + }, + }, + }, + }, + }; } function isRecord(value: unknown): value is Record { - return ( - typeof value === 'object' && - value !== null && - !Array.isArray(value) - ) + return typeof value === "object" && value !== null && !Array.isArray(value); } function opencodeErrorMessage(value: unknown, fallback: string): string { - return safeToolErrorDetail(value, 1_000) ?? fallback + return safeToolErrorDetail(value, 1_000) ?? fallback; } function byteLengthWithin(value: string, maximum: number): boolean { - return Buffer.byteLength(value) <= maximum + return Buffer.byteLength(value) <= maximum; } function createPublicQuestionId( requestId: string, sessionId: string, - upstreamQuestionId: string + upstreamQuestionId: string, ): string { - return `opencode-${createHash('sha256') + return `opencode-${createHash("sha256") .update(`${requestId}\0${sessionId}\0${upstreamQuestionId}`) - .digest('hex') - .slice(0, 48)}` + .digest("hex") + .slice(0, 48)}`; } function executionDeadlineLabel(milliseconds: number): string { return milliseconds % 60_000 === 0 ? `${milliseconds / 60_000} 分钟` - : `${milliseconds} 毫秒` + : `${milliseconds} 毫秒`; } function awaitWithAbort( operation: Promise, - signal: AbortSignal + signal: AbortSignal, ): Promise { - signal.throwIfAborted() + signal.throwIfAborted(); return new Promise((resolveOperation, rejectOperation) => { - const abort = (): void => rejectOperation(signal.reason) - signal.addEventListener('abort', abort, { once: true }) + const abort = (): void => rejectOperation(signal.reason); + signal.addEventListener("abort", abort, { once: true }); operation.then( (value) => { - signal.removeEventListener('abort', abort) - resolveOperation(value) + signal.removeEventListener("abort", abort); + resolveOperation(value); }, (error: unknown) => { - signal.removeEventListener('abort', abort) - rejectOperation(error) - } - ) - }) + signal.removeEventListener("abort", abort); + rejectOperation(error); + }, + ); + }); } function areBoundedPatterns(value: unknown): value is string[] { @@ -320,30 +314,26 @@ function areBoundedPatterns(value: unknown): value is string[] { value.length <= MAX_PERMISSION_PATTERNS && value.every( (pattern) => - typeof pattern === 'string' && - pattern.length <= MAX_PERMISSION_PATTERN_LENGTH + typeof pattern === "string" && + pattern.length <= MAX_PERMISSION_PATTERN_LENGTH, ) && - byteLengthWithin( - value.join('\0'), - MAX_PERMISSION_PATTERNS_BYTES - ) - ) + byteLengthWithin(value.join("\0"), MAX_PERMISSION_PATTERNS_BYTES) + ); } function parsePermissionRequest( properties: unknown, - sessionId: string + sessionId: string, ): PermissionRequest | undefined { if (!isRecord(properties) || properties.sessionID !== sessionId) { - return undefined + return undefined; } - const { id, permission, patterns, metadata, always, tool } = - properties + const { id, permission, patterns, metadata, always, tool } = properties; if ( - typeof id !== 'string' || + typeof id !== "string" || id.length === 0 || id.length > MAX_PERMISSION_NAME_LENGTH || - typeof permission !== 'string' || + typeof permission !== "string" || permission.length === 0 || permission.length > MAX_PERMISSION_NAME_LENGTH || !areBoundedPatterns(patterns) || @@ -351,42 +341,37 @@ function parsePermissionRequest( !areBoundedPatterns(always) || (tool !== undefined && (!isRecord(tool) || - typeof tool.messageID !== 'string' || + typeof tool.messageID !== "string" || tool.messageID.length === 0 || tool.messageID.length > 256 || - typeof tool.callID !== 'string' || + typeof tool.callID !== "string" || tool.callID.length === 0 || tool.callID.length > 256)) ) { - throw new Error('OpenCode 权限请求格式无效') + throw new Error("OpenCode 权限请求格式无效"); } - let serializedMetadata: string + let serializedMetadata: string; try { - serializedMetadata = JSON.stringify(metadata) + serializedMetadata = JSON.stringify(metadata); } catch { - throw new Error('OpenCode 权限请求元数据无效') + throw new Error("OpenCode 权限请求元数据无效"); } - if ( - !byteLengthWithin( - serializedMetadata, - MAX_PERMISSION_METADATA_BYTES - ) - ) { - throw new Error('OpenCode 权限请求元数据超过安全限制') + if (!byteLengthWithin(serializedMetadata, MAX_PERMISSION_METADATA_BYTES)) { + throw new Error("OpenCode 权限请求元数据超过安全限制"); } - return properties as PermissionRequest + return properties as PermissionRequest; } function parseQuestionRequest( properties: unknown, - sessionId: string + sessionId: string, ): QuestionRequest | undefined { if (!isRecord(properties) || properties.sessionID !== sessionId) { - return undefined + return undefined; } - const { id, questions, tool } = properties + const { id, questions, tool } = properties; if ( - typeof id !== 'string' || + typeof id !== "string" || id.length === 0 || id.length > MAX_PERMISSION_NAME_LENGTH || !Array.isArray(questions) || @@ -395,10 +380,10 @@ function parseQuestionRequest( !questions.every( (question) => isRecord(question) && - typeof question.question === 'string' && + typeof question.question === "string" && question.question.trim().length > 0 && question.question.length <= 2_000 && - typeof question.header === 'string' && + typeof question.header === "string" && question.header.trim().length > 0 && question.header.length <= 120 && Array.isArray(question.options) && @@ -406,49 +391,48 @@ function parseQuestionRequest( question.options.every( (option) => isRecord(option) && - typeof option.label === 'string' && + typeof option.label === "string" && option.label.trim().length > 0 && option.label.length <= 200 && - typeof option.description === 'string' && - option.description.length <= 1_000 + typeof option.description === "string" && + option.description.length <= 1_000, ) && (question.multiple === undefined || - typeof question.multiple === 'boolean') && - (question.custom === undefined || - typeof question.custom === 'boolean') + typeof question.multiple === "boolean") && + (question.custom === undefined || typeof question.custom === "boolean"), ) || (tool !== undefined && (!isRecord(tool) || - typeof tool.messageID !== 'string' || + typeof tool.messageID !== "string" || tool.messageID.length === 0 || tool.messageID.length > 256 || - typeof tool.callID !== 'string' || + typeof tool.callID !== "string" || tool.callID.length === 0 || tool.callID.length > 256)) ) { - throw new Error('OpenCode 提问请求格式无效') + throw new Error("OpenCode 提问请求格式无效"); } - let serialized: string + let serialized: string; try { - serialized = JSON.stringify(properties) + serialized = JSON.stringify(properties); } catch { - throw new Error('OpenCode 提问请求无法序列化') + throw new Error("OpenCode 提问请求无法序列化"); } if (!byteLengthWithin(serialized, MAX_QUESTION_REQUEST_BYTES)) { - throw new Error('OpenCode 提问请求超过安全限制') + throw new Error("OpenCode 提问请求超过安全限制"); } - return properties as QuestionRequest + return properties as QuestionRequest; } function isSafeTokenCount(value: number): boolean { - return Number.isSafeInteger(value) && value >= 0 + return Number.isSafeInteger(value) && value >= 0; } function createUsageEvent( requestId: string, - message: AssistantMessage + message: AssistantMessage, ): RuntimeModelUsageEvent | undefined { - const { tokens } = message + const { tokens } = message; if ( message.time.completed === undefined || !isSafeTokenCount(tokens.input) || @@ -457,14 +441,14 @@ function createUsageEvent( !isSafeTokenCount(tokens.cache.write) || (tokens.total !== undefined && !isSafeTokenCount(tokens.total)) ) { - return undefined + return undefined; } return { requestId, - type: 'model-usage', + type: "model-usage", callId: message.id.slice(0, 256), - runtime: 'opencode', + runtime: "opencode", provider: message.providerID.slice(0, 100), model: message.modelID.slice(0, 500), inputTokens: tokens.input, @@ -473,112 +457,110 @@ function createUsageEvent( cacheWriteTokens: tokens.cache.write, ...(tokens.total === undefined ? {} - : { reportedTotalTokens: tokens.total }) - } + : { reportedTotalTokens: tokens.total }), + }; } export type OpenCodeRuntimeDependencies = { - spawn: typeof spawn + spawn: typeof spawn; detectBinary: ( - runtime: 'opencode', + runtime: "opencode", configuredPath: string, - bundledPath?: string - ) => Promise<{ path?: string; detail: string }> + bundledPath?: string, + ) => Promise<{ path?: string; detail: string }>; validateBinary: ( - runtime: 'opencode', + runtime: "opencode", configuredPath: string, - bundledPath?: string - ) => Promise<{ path?: string; detail: string }> - createClient: typeof createOpencodeClient + bundledPath?: string, + ) => Promise<{ path?: string; detail: string }>; + createClient: typeof createOpencodeClient; checkServerHealth: ( url: string, authorization: string, - signal: AbortSignal - ) => Promise - platform: NodeJS.Platform - startupTimeoutMs: number - controlRequestTimeoutMs: number - executionTimeoutMs?: number -} + signal: AbortSignal, + ) => Promise; + platform: NodeJS.Platform; + startupTimeoutMs: number; + controlRequestTimeoutMs: number; + executionTimeoutMs?: number; +}; export type OpenCodeRuntimeOptions = { - baseUrl?: string - embedded: boolean - binaryPath: string - bundledBinaryPath?: string - bundledConfigPath?: string - configPath: string - defaultWorkspace: string - modelProfile?: ResolvedModelProfile - skillInstructions?: string - skillPackages?: RuntimeSkillPackage[] - sharedCacheRoot?: string - knowledgeGateway?: KnowledgeMcpGateway - mcpServers?: ResolvedMcpServer[] - customization?: RuntimeCustomizationSettings['opencode'] - launchEnvironmentProvider?: LaunchEnvironmentProvider -} + baseUrl?: string; + embedded: boolean; + binaryPath: string; + bundledBinaryPath?: string; + bundledConfigPath?: string; + configPath: string; + defaultWorkspace: string; + modelProfile?: ResolvedModelProfile; + skillInstructions?: string; + skillPackages?: RuntimeSkillPackage[]; + sharedCacheRoot?: string; + knowledgeGateway?: KnowledgeMcpGateway; + mcpServers?: ResolvedMcpServer[]; + customization?: RuntimeCustomizationSettings["opencode"]; + launchEnvironmentProvider?: LaunchEnvironmentProvider; +}; function boundedNativeIdentifier(value: unknown): string | undefined { - const parsed = boundedRuntimeIdentifierSchema.safeParse(value) - return parsed.success ? parsed.data : undefined + const parsed = boundedRuntimeIdentifierSchema.safeParse(value); + return parsed.success ? parsed.data : undefined; } function boundedNativeText( value: unknown, - maximum: number + maximum: number, ): string | undefined { - if (typeof value !== 'string') { - return undefined + if (typeof value !== "string") { + return undefined; } - const normalized = value.trim() - return normalized ? normalized.slice(0, maximum) : undefined + const normalized = value.trim(); + return normalized ? normalized.slice(0, maximum) : undefined; } function isTemporaryMcpName(value: string): boolean { - return TEMPORARY_MCP_PREFIXES.some((prefix) => - value.startsWith(prefix) - ) + return TEMPORARY_MCP_PREFIXES.some((prefix) => value.startsWith(prefix)); } function isTemporaryMcpInventoryItem( ...values: Array ): boolean { return values.some( - (value) => value !== undefined && isTemporaryMcpName(value) - ) + (value) => value !== undefined && isTemporaryMcpName(value), + ); } function mapOpenCodeNativeTools( - values: readonly string[] + values: readonly string[], ): RuntimeNativeTool[] { - const seen = new Set() - const tools: RuntimeNativeTool[] = [] + const seen = new Set(); + const tools: RuntimeNativeTool[] = []; for (const value of values) { if (tools.length >= MAX_NATIVE_TOOLS) { - break + break; } - const id = boundedNativeIdentifier(value) + const id = boundedNativeIdentifier(value); if ( !id || seen.has(id) || OPENCODE_INTERNAL_TOOL_IDS.has(id) || isTemporaryMcpName(id) ) { - continue + continue; } - seen.add(id) - const builtinKind = OPENCODE_BUILTIN_TOOL_KINDS[id] + seen.add(id); + const builtinKind = OPENCODE_BUILTIN_TOOL_KINDS[id]; tools.push({ id, name: id.slice(0, 200), - kind: builtinKind ?? 'other', - source: builtinKind ? 'runtime' : 'unknown', - ask: id === 'skill' ? 'conditional' : 'blocked', - execute: 'allowed' - }) + kind: builtinKind ?? "other", + source: builtinKind ? "runtime" : "unknown", + ask: id === "skill" ? "conditional" : "blocked", + execute: "allowed", + }); } - return tools + return tools; } function isLocalPathLikeResourceUri(value: string): boolean { @@ -586,183 +568,175 @@ function isLocalPathLikeResourceUri(value: string): boolean { /^file:/iu.test(value) || /^[a-z]:[\\/]/iu.test(value) || /^(?:[\\/]{1,2}|\.\.?[\\/]|~[\\/])/u.test(value) - ) + ); } function boundedResourceUri(value: unknown): string | undefined { - const bounded = boundedNativeText(value, 2_048) + const bounded = boundedNativeText(value, 2_048); if ( !bounded || isLocalPathLikeResourceUri(bounded) || /^data:/iu.test(bounded) ) { - return undefined + return undefined; } try { - const url = new URL(bounded) - if (url.protocol === 'file:' || url.protocol === 'data:') { - return undefined + const url = new URL(bounded); + if (url.protocol === "file:" || url.protocol === "data:") { + return undefined; } - url.username = '' - url.password = '' - url.search = '' - url.hash = '' - return url.toString().slice(0, 2_048) + url.username = ""; + url.password = ""; + url.search = ""; + url.hash = ""; + return url.toString().slice(0, 2_048); } catch { - return bounded.includes('?') || bounded.includes('#') - ? undefined - : bounded + return bounded.includes("?") || bounded.includes("#") ? undefined : bounded; } } function createSkillPermissionRules( - skillIds: readonly string[] + skillIds: readonly string[], ): PermissionRuleset { if (skillIds.length === 0) { - return [] + return []; } return Object.entries(createSkillPermissionConfig(skillIds)).map( ([pattern, action]) => ({ - permission: 'skill', + permission: "skill", pattern, - action - }) - ) + action, + }), + ); } function createSkillPermissionConfig( - skillIds: readonly string[] -): Record { + skillIds: readonly string[], +): Record { return Object.fromEntries([ - ['*', 'deny' as const], - ...skillIds.map((skillId) => [skillId, 'allow' as const]) - ]) + ["*", "deny" as const], + ...skillIds.map((skillId) => [skillId, "allow" as const]), + ]); } function createOpenCodeSkillConfig( registration: OpenCodeSkillRegistration, - skillIds: readonly string[] + skillIds: readonly string[], ): { - skills: { paths: string[]; urls: never[] } + skills: { paths: string[]; urls: never[] }; permission: { - skill: Record - } + skill: Record; + }; } { return { skills: { paths: [registration.skillsRoot], - urls: [] + urls: [], }, permission: { - skill: createSkillPermissionConfig(skillIds) - } - } + skill: createSkillPermissionConfig(skillIds), + }, + }; } async function normalizeOpenCodeSkillManifest( skillDirectory: string, - skillId: string + skillId: string, ): Promise { - const manifestPath = join(skillDirectory, 'SKILL.md') - const content = await readFile(manifestPath, 'utf8') - const match = - /^---\r?\n([\s\S]*?)\r?\n---\r?\n([\s\S]+)$/u.exec(content) + const manifestPath = join(skillDirectory, "SKILL.md"); + const content = await readFile(manifestPath, "utf8"); + const match = /^---\r?\n([\s\S]*?)\r?\n---\r?\n([\s\S]+)$/u.exec(content); if (!match?.[1] || !match[2]?.trim()) { - throw new Error('OpenCode Skill 清单格式无效') + throw new Error("OpenCode Skill 清单格式无效"); } - const metadata = parseYaml(match[1]) + const metadata = parseYaml(match[1]); if ( - typeof metadata !== 'object' || + typeof metadata !== "object" || metadata === null || Array.isArray(metadata) ) { - throw new Error('OpenCode Skill 清单元数据无效') + throw new Error("OpenCode Skill 清单元数据无效"); } const normalizedMetadata: Record = { ...metadata, - name: skillId - } - delete normalizedMetadata.id + name: skillId, + }; + delete normalizedMetadata.id; await writeFile( manifestPath, - [ - '---', - stringifyYaml(normalizedMetadata).trimEnd(), - '---', - match[2] - ].join('\n'), - 'utf8' - ) + ["---", stringifyYaml(normalizedMetadata).trimEnd(), "---", match[2]].join( + "\n", + ), + "utf8", + ); } async function defaultDetectBinary( - runtime: 'opencode', + runtime: "opencode", configuredPath: string, - bundledPath?: string + bundledPath?: string, ): Promise<{ path?: string; detail: string }> { return detectRuntimeBinary({ binaryPath: configuredPath, bundledPath, - validation: 'path-only', + validation: "path-only", binaryNames: [runtime], - label: 'OpenCode CLI' - }) + label: "OpenCode CLI", + }); } async function defaultValidateBinary( - runtime: 'opencode', - configuredPath: string + runtime: "opencode", + configuredPath: string, ): Promise<{ path?: string; detail: string }> { return detectRuntimeBinary({ binaryPath: configuredPath, - validation: 'canonical-file', + validation: "canonical-file", allowAutomaticDiscovery: false, binaryNames: [runtime], - label: 'OpenCode CLI' - }) + label: "OpenCode CLI", + }); } async function defaultCheckServerHealth( url: string, authorization: string, - signal: AbortSignal + signal: AbortSignal, ): Promise { const client = createOpencodeClient({ baseUrl: url, - headers: { Authorization: authorization } - }) - const response = await client.global.health({ signal }) - return response.error === undefined && response.data?.healthy === true + headers: { Authorization: authorization }, + }); + const response = await client.global.health({ signal }); + return response.error === undefined && response.data?.healthy === true; } export class OpenCodeRuntime implements AgentRuntime { - readonly runtimeId = 'opencode' - readonly requiresToolApproval = false - readonly supportsToolExecution = true - private client?: OpencodeClient - private clientInitialization?: Promise - private server?: OpenCodeServer - private startingChild?: SpawnedProcess - private readonly sessions = new Map() - private readonly sessionInitializations = new Map< - string, - Promise - >() + readonly runtimeId = "opencode"; + readonly requiresToolApproval = false; + readonly supportsToolExecution = true; + private client?: OpencodeClient; + private clientInitialization?: Promise; + private server?: OpenCodeServer; + private startingChild?: SpawnedProcess; + private readonly sessions = new Map(); + private readonly sessionInitializations = new Map>(); private readonly pendingQuestions = new Map< string, { - client: OpencodeClient - directory: string - questionCount: number - upstreamQuestionId: string + client: OpencodeClient; + directory: string; + questionCount: number; + upstreamQuestionId: string; } - >() - private readonly conversationRunTails = new Map>() - private readonly dependencies: OpenCodeRuntimeDependencies + >(); + private readonly conversationRunTails = new Map>(); + private mcpMutationTail: Promise = Promise.resolve(); + private readonly dependencies: OpenCodeRuntimeDependencies; constructor( private readonly options: OpenCodeRuntimeOptions, - dependencies: Partial = {} + dependencies: Partial = {}, ) { this.dependencies = { spawn, @@ -773,468 +747,461 @@ export class OpenCodeRuntime implements AgentRuntime { platform: process.platform, startupTimeoutMs: STARTUP_TIMEOUT_MS, controlRequestTimeoutMs: CONTROL_REQUEST_TIMEOUT_MS, - ...dependencies - } + ...dependencies, + }; } private usesEmbeddedPermissionMediation(): boolean { - return this.options.embedded && !this.options.baseUrl + return this.options.embedded && !this.options.baseUrl; } private supportsNativeCustomization(): boolean { - return this.usesEmbeddedPermissionMediation() + return this.usesEmbeddedPermissionMediation(); } get supportsScopedDataTools(): boolean { - return this.usesEmbeddedPermissionMediation() + return this.usesEmbeddedPermissionMediation(); } private async acquireConversationRun( conversationId: string, - signal: AbortSignal + signal: AbortSignal, ): Promise<() => void> { - signal.throwIfAborted() + signal.throwIfAborted(); const previous = - this.conversationRunTails.get(conversationId) ?? Promise.resolve() - let releaseGate!: () => void + this.conversationRunTails.get(conversationId) ?? Promise.resolve(); + let releaseGate!: () => void; const gate = new Promise((resolve) => { - releaseGate = resolve - }) + releaseGate = resolve; + }); const tail = previous.then( () => gate, - () => gate - ) - this.conversationRunTails.set(conversationId, tail) - let abort!: () => void + () => gate, + ); + this.conversationRunTails.set(conversationId, tail); + let abort!: () => void; const aborted = new Promise((_resolve, reject) => { - abort = () => reject(signal.reason) - }) - signal.addEventListener('abort', abort, { once: true }) + abort = () => reject(signal.reason); + }); + signal.addEventListener("abort", abort, { once: true }); try { - await Promise.race([previous, aborted]) - signal.throwIfAborted() + await Promise.race([previous, aborted]); + signal.throwIfAborted(); return () => { - releaseGate() + releaseGate(); if (this.conversationRunTails.get(conversationId) === tail) { - this.conversationRunTails.delete(conversationId) + this.conversationRunTails.delete(conversationId); } - } + }; } catch (error) { - releaseGate() + releaseGate(); if (this.conversationRunTails.get(conversationId) === tail) { - this.conversationRunTails.delete(conversationId) + this.conversationRunTails.delete(conversationId); } - throw error + throw error; } finally { - signal.removeEventListener('abort', abort) + signal.removeEventListener("abort", abort); } } private terminate(child: SpawnedProcess): void { requestProcessTreeTermination(child, { platform: this.dependencies.platform, - spawn: this.dependencies.spawn - }) + spawn: this.dependencies.spawn, + }); } private waitForExit(child: SpawnedProcess): Promise { - return waitForProcessExit(child) + return waitForProcessExit(child); } private getNativeSkillIds(): string[] { if (!this.usesEmbeddedPermissionMediation()) { - return [] + return []; } - const ids = (this.options.skillPackages ?? []).map( - (skill) => skill.id - ) + const ids = (this.options.skillPackages ?? []).map((skill) => skill.id); if ( new Set(ids).size !== ids.length || - ids.some( - (id) => - id.length > 64 || !OPENCODE_SKILL_NAME_PATTERN.test(id) - ) + ids.some((id) => id.length > 64 || !OPENCODE_SKILL_NAME_PATTERN.test(id)) ) { - throw new Error('OpenCode Skill 注册信息无效') + throw new Error("OpenCode Skill 注册信息无效"); } - return ids + return ids; } private async normalizeSkillPackages(skillsRoot: string): Promise { for (const skill of this.options.skillPackages ?? []) { await normalizeOpenCodeSkillManifest( join(skillsRoot, skill.id), - skill.id - ) + skill.id, + ); + } + } + + private async mutateMcp( + signal: AbortSignal, + operation: () => Promise, + ): Promise { + signal.throwIfAborted(); + const previous = this.mcpMutationTail; + let release!: () => void; + const current = new Promise((resolve) => { + release = resolve; + }); + this.mcpMutationTail = previous.then( + () => current, + () => current, + ); + try { + await awaitWithAbort(previous, signal); + signal.throwIfAborted(); + return await operation(); + } finally { + release(); } } private async prepareSharedSkills(sharedRoot: string): Promise { - const packages = this.options.skillPackages ?? [] - const fingerprint = createHash('sha256') + const packages = this.options.skillPackages ?? []; + const fingerprint = createHash("sha256") .update( packages .map((skill) => ({ id: skill.id, - digest: skill.digest ?? resolve(skill.directory) + digest: skill.digest ?? resolve(skill.directory), })) .sort((left, right) => left.id.localeCompare(right.id)) .map((skill) => `${skill.id}\0${skill.digest}`) - .join('\0') + .join("\0"), ) - .digest('hex') - const cacheRoot = join(sharedRoot, 'skills') - const target = join(cacheRoot, fingerprint) - const existing = sharedSkillPreparations.get(target) + .digest("hex"); + const cacheRoot = join(sharedRoot, "skills"); + const target = join(cacheRoot, fingerprint); + const existing = sharedSkillPreparations.get(target); if (existing) { - return existing + return existing; } const preparation = (async () => { try { if ((await stat(target)).isDirectory()) { - return target + return target; } } catch { // The immutable Skill snapshot has not been prepared yet. } - await mkdir(cacheRoot, { recursive: true, mode: 0o700 }) - const stagingRoot = await mkdtemp(join(cacheRoot, '.staging-')) + await mkdir(cacheRoot, { recursive: true, mode: 0o700 }); + const stagingRoot = await mkdtemp(join(cacheRoot, ".staging-")); try { const staged = await stageRuntimeSkillPackages( stagingRoot, packages, - 'OpenCode' - ) - await this.normalizeSkillPackages(staged) + "OpenCode", + ); + await this.normalizeSkillPackages(staged); try { - await rename(staged, target) + await rename(staged, target); } catch (error) { try { if ((await stat(target)).isDirectory()) { - return target + return target; } } catch { // Preserve the original rename failure below. } - throw error + throw error; } - return target + return target; } finally { - await rm(stagingRoot, { recursive: true, force: true }) + await rm(stagingRoot, { recursive: true, force: true }); } - })() - sharedSkillPreparations.set(target, preparation) + })(); + sharedSkillPreparations.set(target, preparation); try { - return await preparation + return await preparation; } catch (error) { - sharedSkillPreparations.delete(target) - throw error + sharedSkillPreparations.delete(target); + throw error; } } private async createSkillRegistration(): Promise { - const root = await mkdtemp(join(tmpdir(), 'goodbuddy-opencode-')) + const root = await mkdtemp(join(tmpdir(), "goodbuddy-opencode-")); try { - const sharedRoot = this.options.sharedCacheRoot?.trim() + const sharedRoot = this.options.sharedCacheRoot?.trim(); if (sharedRoot) { - const configDirectory = join( - resolve(sharedRoot), - 'config', - 'opencode' - ) - await this.prepareSharedConfig(configDirectory) - const skillsRoot = await this.prepareSharedSkills( - resolve(sharedRoot) - ) - return { root, configDirectory, skillsRoot } + const configDirectory = join(resolve(sharedRoot), "config", "opencode"); + await this.prepareSharedConfig(configDirectory); + const skillsRoot = await this.prepareSharedSkills(resolve(sharedRoot)); + return { root, configDirectory, skillsRoot }; } - const configDirectory = join(root, 'config') + const configDirectory = join(root, "config"); const skillsRoot = await stageRuntimeSkillPackages( configDirectory, this.options.skillPackages ?? [], - 'OpenCode' - ) - await this.normalizeSkillPackages(skillsRoot) - return { root, configDirectory, skillsRoot } + "OpenCode", + ); + await this.normalizeSkillPackages(skillsRoot); + return { root, configDirectory, skillsRoot }; } catch (error) { - await rm(root, { recursive: true, force: true }) - throw error + await rm(root, { recursive: true, force: true }); + throw error; } } - private async prepareSharedConfig( - configDirectory: string - ): Promise { - const bundledConfigPath = this.options.bundledConfigPath?.trim() + private async prepareSharedConfig(configDirectory: string): Promise { + const bundledConfigPath = this.options.bundledConfigPath?.trim(); if (!bundledConfigPath) { - await mkdir(configDirectory, { recursive: true, mode: 0o700 }) - return + await mkdir(configDirectory, { recursive: true, mode: 0o700 }); + return; } - const target = resolve(configDirectory) - const existing = sharedConfigPreparations.get(target) + const target = resolve(configDirectory); + const existing = sharedConfigPreparations.get(target); if (existing) { - return existing + return existing; } const preparation = (async () => { - const source = resolve(bundledConfigPath) - const markerName = '.goodbuddy-ready.json' - const sourceMarker = await readFile(join(source, markerName), 'utf8') + const source = resolve(bundledConfigPath); + const markerName = ".goodbuddy-ready.json"; + const sourceMarker = await readFile(join(source, markerName), "utf8"); try { if ( - (await readFile(join(target, markerName), 'utf8')) === - sourceMarker + (await readFile(join(target, markerName), "utf8")) === sourceMarker ) { - return + return; } } catch { // Install the bundled dependency tree below. } - await mkdir(dirname(target), { recursive: true, mode: 0o700 }) - const staging = `${target}.staging-${process.pid}-${randomBytes(6).toString('hex')}` - await rm(staging, { recursive: true, force: true }) + await mkdir(dirname(target), { recursive: true, mode: 0o700 }); + const staging = `${target}.staging-${process.pid}-${randomBytes(6).toString("hex")}`; + await rm(staging, { recursive: true, force: true }); try { await cp(source, staging, { recursive: true, errorOnExist: true, - force: false - }) - await rm(target, { recursive: true, force: true }) - await rename(staging, target) + force: false, + }); + await rm(target, { recursive: true, force: true }); + await rename(staging, target); } finally { - await rm(staging, { recursive: true, force: true }) + await rm(staging, { recursive: true, force: true }); } - })() - sharedConfigPreparations.set(target, preparation) + })(); + sharedConfigPreparations.set(target, preparation); try { - await preparation + await preparation; } catch (error) { - sharedConfigPreparations.delete(target) - throw error + sharedConfigPreparations.delete(target); + throw error; } } - private async inspectEmbeddedConfiguration( - validatePath = false - ): Promise<{ - path?: string - detail: string + private async inspectEmbeddedConfiguration(validatePath = false): Promise<{ + path?: string; + detail: string; }> { if ( - this.options.modelProfile?.authentication === 'api-key' && + this.options.modelProfile?.authentication === "api-key" && !this.options.modelProfile.apiKey ) { - throw new Error('OpenCode 独立模型连接尚未配置 API Key') + throw new Error("OpenCode 独立模型连接尚未配置 API Key"); } const detection = await this.dependencies.detectBinary( - 'opencode', + "opencode", this.options.binaryPath, - this.options.bundledBinaryPath - ) + this.options.bundledBinaryPath, + ); if (!validatePath || !detection.path) { - return detection + return detection; } - return this.dependencies.validateBinary( - 'opencode', - detection.path - ) + return this.dependencies.validateBinary("opencode", detection.path); } private async launchEmbedded(signal?: AbortSignal): Promise { if (signal?.aborted) { - throw new Error('OpenCode Server 启动已取消') + throw new Error("OpenCode Server 启动已取消"); } - const detection = await this.inspectEmbeddedConfiguration() - const binaryPath = detection.path + const detection = await this.inspectEmbeddedConfiguration(); + const binaryPath = detection.path; if (!binaryPath) { - throw new Error(detection.detail) + throw new Error(detection.detail); } if (signal?.aborted) { - throw new Error('OpenCode Server 启动已取消') + throw new Error("OpenCode Server 启动已取消"); } - const port = await getAvailableLoopbackPort() + const port = await getAvailableLoopbackPort(); if (signal?.aborted) { - throw new Error('OpenCode Server 启动已取消') + throw new Error("OpenCode Server 启动已取消"); } - const skillIds = this.getNativeSkillIds() - const registration = await this.createSkillRegistration() + const skillIds = this.getNativeSkillIds(); + const registration = await this.createSkillRegistration(); try { if (signal?.aborted) { - throw new Error('OpenCode Server 启动已取消') + throw new Error("OpenCode Server 启动已取消"); } - const profile = this.options.modelProfile + const profile = this.options.modelProfile; let env = profile ? buildExplicitProfileRuntimeEnvironment( runtimePrivacyEnvironment, - profile.authentication === 'api-key' && profile.apiKey + profile.authentication === "api-key" && profile.apiKey ? { name: - profile.protocol === 'anthropic-messages' - ? 'ANTHROPIC_API_KEY' - : 'OPENAI_API_KEY', - value: profile.apiKey + profile.protocol === "anthropic-messages" + ? "ANTHROPIC_API_KEY" + : "OPENAI_API_KEY", + value: profile.apiKey, } - : undefined + : undefined, ) - : buildRuntimeEnvironment(runtimePrivacyEnvironment) + : buildRuntimeEnvironment(runtimePrivacyEnvironment); env = applyLaunchEnvironmentPath( env, - this.options.launchEnvironmentProvider - ) - delete env.OPENCODE_CONFIG - delete env.OPENCODE_CONFIG_CONTENT - delete env.OPENCODE_CONFIG_DIR - delete env.OPENCODE_SERVER_PASSWORD - delete env.OPENCODE_SERVER_USERNAME - const serverPassword = randomBytes(32).toString('base64url') + this.options.launchEnvironmentProvider, + ); + delete env.OPENCODE_CONFIG; + delete env.OPENCODE_CONFIG_CONTENT; + delete env.OPENCODE_CONFIG_DIR; + delete env.OPENCODE_SERVER_PASSWORD; + delete env.OPENCODE_SERVER_USERNAME; + const serverPassword = randomBytes(32).toString("base64url"); const authorization = `Basic ${Buffer.from( - `${EMBEDDED_SERVER_USERNAME}:${serverPassword}` - ).toString('base64')}` - env.OPENCODE_SERVER_USERNAME = EMBEDDED_SERVER_USERNAME - env.OPENCODE_SERVER_PASSWORD = serverPassword - env.OPENCODE_CONFIG_DIR = registration.configDirectory - env.OPENCODE_DISABLE_AUTOUPDATE = '1' - env.OPENCODE_DISABLE_CLAUDE_CODE_SKILLS = '1' - env.OPENCODE_DISABLE_EMBEDDED_WEB_UI = '1' - env.OPENCODE_DISABLE_EXTERNAL_SKILLS = '1' - env.OPENCODE_EXPERIMENTAL_DISABLE_FILEWATCHER = '1' - env.OPENCODE_DISABLE_LSP_DOWNLOAD = '1' - env.OPENCODE_DISABLE_MODELS_FETCH = '1' - env.OPENCODE_DISABLE_PROJECT_CONFIG = '1' - env.OPENCODE_DISABLE_SHARE = '1' + `${EMBEDDED_SERVER_USERNAME}:${serverPassword}`, + ).toString("base64")}`; + env.OPENCODE_SERVER_USERNAME = EMBEDDED_SERVER_USERNAME; + env.OPENCODE_SERVER_PASSWORD = serverPassword; + env.OPENCODE_CONFIG_DIR = registration.configDirectory; + env.OPENCODE_DISABLE_AUTOUPDATE = "1"; + env.OPENCODE_DISABLE_CLAUDE_CODE_SKILLS = "1"; + env.OPENCODE_DISABLE_EMBEDDED_WEB_UI = "1"; + env.OPENCODE_DISABLE_EXTERNAL_SKILLS = "1"; + env.OPENCODE_EXPERIMENTAL_DISABLE_FILEWATCHER = "1"; + env.OPENCODE_DISABLE_LSP_DOWNLOAD = "1"; + env.OPENCODE_DISABLE_MODELS_FETCH = "1"; + env.OPENCODE_DISABLE_PROJECT_CONFIG = "1"; + env.OPENCODE_DISABLE_SHARE = "1"; env.XDG_CACHE_HOME = this.options.sharedCacheRoot?.trim() - ? join(resolve(this.options.sharedCacheRoot), 'cache') - : join(registration.root, 'xdg-cache') + ? join(resolve(this.options.sharedCacheRoot), "cache") + : join(registration.root, "xdg-cache"); env.XDG_CONFIG_HOME = this.options.sharedCacheRoot?.trim() ? dirname(registration.configDirectory) - : join(registration.root, 'xdg-config') - env.XDG_DATA_HOME = join(registration.root, 'xdg-data') - env.XDG_STATE_HOME = join(registration.root, 'xdg-state') - const skillConfig = createOpenCodeSkillConfig( - registration, - skillIds - ) + : join(registration.root, "xdg-config"); + env.XDG_DATA_HOME = join(registration.root, "xdg-data"); + env.XDG_STATE_HOME = join(registration.root, "xdg-state"); + const skillConfig = createOpenCodeSkillConfig(registration, skillIds); env.OPENCODE_CONFIG_CONTENT = JSON.stringify( profile ? { ...createOpenCodeProviderConfig(profile), - ...skillConfig + ...skillConfig, } - : skillConfig - ) + : skillConfig, + ); if (!profile && this.options.configPath.trim()) { - env.OPENCODE_CONFIG = resolve(this.options.configPath) + env.OPENCODE_CONFIG = resolve(this.options.configPath); } - const serverArgs = [ - 'serve', - '--hostname=127.0.0.1', - `--port=${port}` - ] + const serverArgs = ["serve", "--hostname=127.0.0.1", `--port=${port}`]; return await new Promise((resolveServer, reject) => { - const child = this.dependencies.spawn( - binaryPath, - serverArgs, - { - cwd: this.options.defaultWorkspace, - env, - shell: false, - stdio: ['ignore', 'pipe', 'pipe'], - windowsHide: true - } - ) - this.startingChild = child - const { stdout, stderr } = child - let settled = false - let pollTimer: NodeJS.Timeout | undefined - const probeController = new AbortController() + const child = this.dependencies.spawn(binaryPath, serverArgs, { + cwd: this.options.defaultWorkspace, + env, + shell: false, + stdio: ["ignore", "pipe", "pipe"], + windowsHide: true, + }); + this.startingChild = child; + const { stdout, stderr } = child; + let settled = false; + let pollTimer: NodeJS.Timeout | undefined; + const probeController = new AbortController(); const probeSignal = signal ? AbortSignal.any([signal, probeController.signal]) - : probeController.signal - const url = `http://127.0.0.1:${port}` + : probeController.signal; + const url = `http://127.0.0.1:${port}`; const cleanupStartupListeners = (): void => { - clearTimeout(timeout) + clearTimeout(timeout); if (pollTimer) { - clearTimeout(pollTimer) + clearTimeout(pollTimer); } - probeController.abort() - signal?.removeEventListener('abort', abort) - child.removeListener('error', onError) - child.removeListener('close', onClose) - } + probeController.abort(); + signal?.removeEventListener("abort", abort); + child.removeListener("error", onError); + child.removeListener("close", onClose); + }; const fail = (message: string): void => { if (settled) { - return + return; } - settled = true - cleanupStartupListeners() + settled = true; + cleanupStartupListeners(); const clearStartingChild = (): void => { if (this.startingChild === child) { - this.startingChild = undefined + this.startingChild = undefined; } - } - const exited = this.waitForExit(child) - this.terminate(child) + }; + const exited = this.waitForExit(child); + this.terminate(child); void exited.finally(() => { if (child.exitCode !== null) { - clearStartingChild() + clearStartingChild(); } - reject(new Error(message.slice(0, 1_000))) - }) - } + reject(new Error(message.slice(0, 1_000))); + }); + }; const succeed = (url: string): void => { if (settled) { - return + return; } - settled = true - cleanupStartupListeners() + settled = true; + cleanupStartupListeners(); if (this.startingChild === child) { - this.startingChild = undefined + this.startingChild = undefined; } - stdout?.resume() - stderr?.resume() + stdout?.resume(); + stderr?.resume(); resolveServer({ url, authorization, close: async () => { try { - const exited = this.waitForExit(child) - this.terminate(child) - await exited + const exited = this.waitForExit(child); + this.terminate(child); + await exited; } finally { await rm(registration.root, { recursive: true, - force: true - }) + force: true, + }); } - } - }) - } + }, + }); + }; const onError = (): void => { - fail('OpenCode Server 启动失败') - } + fail("OpenCode Server 启动失败"); + }; const onClose = (code: number | null): void => { - fail(`OpenCode Server 启动前退出（code ${code ?? 'unknown'}）`) - } + fail(`OpenCode Server 启动前退出（code ${code ?? "unknown"}）`); + }; const abort = (): void => { - fail('OpenCode Server 启动已取消') - } + fail("OpenCode Server 启动已取消"); + }; const probe = async (): Promise => { try { const attemptSignal = AbortSignal.any([ probeSignal, - AbortSignal.timeout(STARTUP_PROBE_TIMEOUT_MS) - ]) + AbortSignal.timeout(STARTUP_PROBE_TIMEOUT_MS), + ]); if ( await this.dependencies.checkServerHealth( url, authorization, - attemptSignal + attemptSignal, ) ) { - succeed(url) - return + succeed(url); + return; } } catch { // The server is not ready yet. @@ -1242,113 +1209,140 @@ export class OpenCodeRuntime implements AgentRuntime { if (!settled) { pollTimer = setTimeout( () => void probe(), - STARTUP_POLL_INTERVAL_MS - ) + STARTUP_POLL_INTERVAL_MS, + ); } - } + }; const timeout = setTimeout(() => { - fail('OpenCode Server 启动超时（30 秒）') - }, this.dependencies.startupTimeoutMs) + fail("OpenCode Server 启动超时（30 秒）"); + }, this.dependencies.startupTimeoutMs); if (!stdout || !stderr) { - fail('OpenCode Server 管道初始化失败') - return + fail("OpenCode Server 管道初始化失败"); + return; } - stdout.resume() - stderr.resume() - child.once('error', onError) - child.once('close', onClose) - signal?.addEventListener('abort', abort, { once: true }) + stdout.resume(); + stderr.resume(); + child.once("error", onError); + child.once("close", onClose); + signal?.addEventListener("abort", abort, { once: true }); if (signal?.aborted) { - abort() - return + abort(); + return; } - void probe() - }) + void probe(); + }); } catch (error) { await rm(registration.root, { recursive: true, - force: true - }).catch(() => undefined) - throw error + force: true, + }).catch(() => undefined); + throw error; } } private async getClient(signal?: AbortSignal): Promise { - if (this.client) { - return this.client - } - const existingInitialization = this.clientInitialization - this.clientInitialization ??= this.initializeClient(signal) - try { - return signal && existingInitialization - ? await awaitWithAbort(this.clientInitialization, signal) - : await this.clientInitialization - } catch (error) { - this.clientInitialization = undefined - throw error + signal?.throwIfAborted(); + if (this.client && !this.server) { + return this.client; } + const existingInitialization = this.clientInitialization; + const cached = Boolean(this.client && this.server); + // Cached health checks are shared, but each caller owns only its wait. + const initialization = this.clientInitialization ??= this.initializeClient( + cached ? undefined : signal, + ).finally(() => { + if (this.clientInitialization === initialization) { + this.clientInitialization = undefined; + } + }); + return signal && (cached || existingInitialization) + ? await awaitWithAbort(initialization, signal) + : await initialization; } private async controlRequest( label: string, operation: (signal: AbortSignal) => Promise, callerSignal?: AbortSignal, - onAbortedResult?: (value: T) => void | Promise + onAbortedResult?: (value: T) => void | Promise, ): Promise { - const timeoutController = new AbortController() - const timeout = setTimeout(() => { - timeoutController.abort( - new Error(`OpenCode ${label}超时`) - ) - }, Math.max(1, this.dependencies.controlRequestTimeoutMs)) - timeout.unref?.() + const timeoutController = new AbortController(); + const timeout = setTimeout( + () => { + timeoutController.abort(new Error(`OpenCode ${label}超时`)); + }, + Math.max(1, this.dependencies.controlRequestTimeoutMs), + ); + timeout.unref?.(); const controlSignal = callerSignal ? AbortSignal.any([callerSignal, timeoutController.signal]) - : timeoutController.signal + : timeoutController.signal; try { - controlSignal.throwIfAborted() - const request = operation(controlSignal) + controlSignal.throwIfAborted(); + const request = operation(controlSignal); if (onAbortedResult) { void request .then(async (value) => { if (controlSignal.aborted) { - await onAbortedResult(value) + await onAbortedResult(value); } }) - .catch(() => undefined) + .catch(() => undefined); } - return await awaitWithAbort( - request, - controlSignal - ) + return await awaitWithAbort(request, controlSignal); } catch (error) { - if ( - timeoutController.signal.aborted && - !callerSignal?.aborted - ) { - throw timeoutController.signal.reason + if (timeoutController.signal.aborted && !callerSignal?.aborted) { + throw timeoutController.signal.reason; } - throw error + throw error; } finally { - clearTimeout(timeout) + clearTimeout(timeout); } } private async initializeClient( - signal?: AbortSignal + signal?: AbortSignal, ): Promise { - let baseUrl = this.options.baseUrl + if (this.client && this.server) { + const probeSignal = AbortSignal.any([ + ...(signal ? [signal] : []), + AbortSignal.timeout(STARTUP_PROBE_TIMEOUT_MS), + ]); + let healthy = false; + try { + healthy = await awaitWithAbort( + this.dependencies.checkServerHealth( + this.server.url, + this.server.authorization, + probeSignal, + ), + probeSignal, + ); + } catch { + // A dead or unresponsive cached server must not poison later requests. + } + signal?.throwIfAborted(); + if (healthy) return this.client; + const server = this.server; + this.server = undefined; + this.client = undefined; + this.sessions.clear(); + this.sessionInitializations.clear(); + this.pendingQuestions.clear(); + await server.close(); + } + let baseUrl = this.options.baseUrl; if (baseUrl && this.options.modelProfile) { - throw new Error('OpenCode 独立模型连接仅支持由 GoodBuddy 启动的本机服务') + throw new Error("OpenCode 独立模型连接仅支持由 GoodBuddy 启动的本机服务"); } if (!baseUrl && this.options.embedded) { - this.server = await this.launchEmbedded(signal) - baseUrl = this.server.url + this.server = await this.launchEmbedded(signal); + baseUrl = this.server.url; } if (!baseUrl) { - throw new Error('未配置 OpenCode Server') + throw new Error("未配置 OpenCode Server"); } this.client = this.dependencies.createClient({ @@ -1357,119 +1351,112 @@ export class OpenCodeRuntime implements AgentRuntime { ...(this.server ? { headers: { - Authorization: this.server.authorization - } + Authorization: this.server.authorization, + }, } - : {}) - }) - return this.client + : {}), + }); + return this.client; } async getStatus(): Promise { if (this.usesEmbeddedPermissionMediation() && !this.client) { try { - const detection = - await this.inspectEmbeddedConfiguration(true) + const detection = await this.inspectEmbeddedConfiguration(true); return { - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: Boolean(detection.path), supportsToolExecution: this.supportsToolExecution, detail: detection.path - ? 'OpenCode 已配置，将在首次使用时启动' - : detection.detail - } + ? "OpenCode 已配置，将在首次使用时启动" + : detection.detail, + }; } catch (error) { return { - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: false, supportsToolExecution: this.supportsToolExecution, - detail: - error instanceof Error ? error.message : 'OpenCode 不可用' - } + detail: error instanceof Error ? error.message : "OpenCode 不可用", + }; } } - return this.checkConnection() + return this.checkConnection(); } async testConnection(): Promise { - return this.checkConnection() + return this.checkConnection(); } private async checkConnection(): Promise { try { - const client = await this.getClient() - const response = await this.controlRequest( - '连接检查', - (signal) => - client.session.list( - { directory: this.options.defaultWorkspace }, - { signal } - ) - ) + const client = await this.getClient(); + const response = await this.controlRequest("连接检查", (signal) => + client.session.list( + { directory: this.options.defaultWorkspace }, + { signal }, + ), + ); if (response.error) { - throw new Error('OpenCode Server 返回错误') + throw new Error("OpenCode Server 返回错误"); } return { - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: true, supportsToolExecution: this.supportsToolExecution, detail: this.server - ? '由 GoodBuddy 以当前用户权限管理本机 OpenCode 进程' - : `已连接 ${this.options.baseUrl}` - } + ? "由 GoodBuddy 以当前用户权限管理本机 OpenCode 进程" + : `已连接 ${this.options.baseUrl}`, + }; } catch (error) { return { - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: false, supportsToolExecution: this.supportsToolExecution, - detail: error instanceof Error ? error.message : 'OpenCode 不可用' - } + detail: error instanceof Error ? error.message : "OpenCode 不可用", + }; } } private async discoverAgents( client: OpencodeClient, - signal?: AbortSignal + signal?: AbortSignal, ): Promise< Array<{ - id: string - name: string - description?: string - mode: 'primary' | 'subagent' | 'all' - native: boolean - hidden: boolean + id: string; + name: string; + description?: string; + mode: "primary" | "subagent" | "all"; + native: boolean; + hidden: boolean; }> > { const response = await this.controlRequest( - '读取 Agent 清单', + "读取 Agent 清单", (controlSignal) => client.app.agents( { - directory: this.options.defaultWorkspace + directory: this.options.defaultWorkspace, }, - { signal: controlSignal } + { signal: controlSignal }, ), - signal - ) + signal, + ); if (response.error || !response.data) { - throw new Error('OpenCode Agent 清单不可用') + throw new Error("OpenCode Agent 清单不可用"); } return response.data .flatMap((agent) => { - const id = boundedNativeIdentifier(agent.name) + const id = boundedNativeIdentifier(agent.name); if (!id) { - return [] + return []; } - const description = boundedNativeText( - agent.description, - 2_000 - ) + const description = boundedNativeText(agent.description, 2_000); return [ { id, @@ -1477,59 +1464,54 @@ export class OpenCodeRuntime implements AgentRuntime { ...(description ? { description } : {}), mode: agent.mode, native: agent.native === true, - hidden: agent.hidden === true - } - ] + hidden: agent.hidden === true, + }, + ]; }) - .slice(0, MAX_NATIVE_AGENTS) + .slice(0, MAX_NATIVE_AGENTS); } private async resolveSelectedAgent( client: OpencodeClient, request: AgentExecutionRequest, - signal: AbortSignal + signal: AbortSignal, ): Promise { const control = - request.runtimeControl?.provider === 'opencode' + request.runtimeControl?.provider === "opencode" ? request.runtimeControl - : undefined - const selected = - control?.agent ?? this.options.customization?.defaultAgent + : undefined; + const selected = control?.agent ?? this.options.customization?.defaultAgent; if (!selected) { - return undefined + return undefined; } if (!this.supportsNativeCustomization()) { - throw new Error( - '外部 OpenCode Server 不支持由 GoodBuddy 选择 Agent' - ) + throw new Error("外部 OpenCode Server 不支持由 GoodBuddy 选择 Agent"); } - const agents = await this.discoverAgents(client, signal) + const agents = await this.discoverAgents(client, signal); if ( !agents.some( (agent) => agent.id === selected && !agent.hidden && - (agent.mode === 'primary' || agent.mode === 'all') + (agent.mode === "primary" || agent.mode === "all"), ) ) { throw new Error( - `OpenCode Agent 不存在、已隐藏或不可作为主 Agent：${selected.slice(0, 128)}` - ) + `OpenCode Agent 不存在、已隐藏或不可作为主 Agent：${selected.slice(0, 128)}`, + ); } - return selected + return selected; } async getNativeSnapshot(): Promise { - const controlled = this.supportsNativeCustomization() + const controlled = this.supportsNativeCustomization(); const empty: RuntimeNativeSnapshot = { - provider: 'opencode', + provider: "opencode", available: false, - inventoryStatus: controlled - ? 'unavailable' - : 'connection-only', + inventoryStatus: controlled ? "unavailable" : "connection-only", detail: controlled - ? 'OpenCode 原生能力不可用' - : '外部 OpenCode Server 仅支持连接状态；GoodBuddy 不支持读取或控制其原生自定义能力', + ? "OpenCode 原生能力不可用" + : "外部 OpenCode Server 仅支持连接状态；GoodBuddy 不支持读取或控制其原生自定义能力", agents: [], tools: [], toolsSupported: controlled, @@ -1543,55 +1525,53 @@ export class OpenCodeRuntime implements AgentRuntime { resources: [], resourcesSupported: false, context: { - strategy: controlled ? 'native' : 'unsupported', + strategy: controlled ? "native" : "unsupported", manualCompact: controlled, detail: controlled - ? '由 OpenCode 原生上下文与手动 Compact 管理' - : '外部 OpenCode Server 不支持由 GoodBuddy 管理上下文' - } - } + ? "由 OpenCode 原生上下文与手动 Compact 管理" + : "外部 OpenCode Server 不支持由 GoodBuddy 管理上下文", + }, + }; if (!controlled) { try { - const client = await this.getClient() - const response = await this.controlRequest( - '连接检查', - (signal) => - client.session.list( - { - directory: this.options.defaultWorkspace, - limit: 1 - }, - { signal } - ) - ) + const client = await this.getClient(); + const response = await this.controlRequest("连接检查", (signal) => + client.session.list( + { + directory: this.options.defaultWorkspace, + limit: 1, + }, + { signal }, + ), + ); if (response.error || !response.data) { - throw new Error('connection check failed') + throw new Error("connection check failed"); } - return { ...empty, available: true } + return { ...empty, available: true }; } catch { return { ...empty, - inventoryStatus: 'unavailable', + inventoryStatus: "unavailable", detail: - '外部 OpenCode Server 无法连接，且不支持由 GoodBuddy 读取或控制原生自定义能力' - } + "外部 OpenCode Server 无法连接，且不支持由 GoodBuddy 读取或控制原生自定义能力", + }; } } - let client: OpencodeClient + let client: OpencodeClient; try { - client = await this.getClient() + client = await this.getClient(); } catch { return { ...empty, - detail: 'OpenCode Server 无法连接' - } + detail: "OpenCode Server 无法连接", + }; } - const directory = this.options.defaultWorkspace + const directory = this.options.defaultWorkspace; const assignedSkillIds = new Set( - (this.options.skillPackages ?? []).map((skill) => skill.id) - ) + (this.options.skillPackages ?? []).map((skill) => skill.id), + ); const [ availabilityResult, agentsResult, @@ -1601,162 +1581,153 @@ export class OpenCodeRuntime implements AgentRuntime { formattersResult, mcpResult, skillsResult, - resourcesResult + resourcesResult, ] = await Promise.allSettled([ - this.controlRequest('连接检查', (signal) => + this.controlRequest("连接检查", (signal) => client.session.list( { directory, - limit: 1 + limit: 1, }, - { signal } - ) + { signal }, + ), ), this.discoverAgents(client), - this.controlRequest('读取工具清单', (signal) => - client.tool.ids({ directory }, { signal }) + this.controlRequest("读取工具清单", (signal) => + client.tool.ids({ directory }, { signal }), ), - this.controlRequest('读取命令清单', (signal) => - client.command.list({ directory }, { signal }) + this.controlRequest("读取命令清单", (signal) => + client.command.list({ directory }, { signal }), ), - this.controlRequest('读取 LSP 状态', (signal) => - client.lsp.status({ directory }, { signal }) + this.controlRequest("读取 LSP 状态", (signal) => + client.lsp.status({ directory }, { signal }), ), - this.controlRequest('读取格式化器状态', (signal) => - client.formatter.status({ directory }, { signal }) + this.controlRequest("读取格式化器状态", (signal) => + client.formatter.status({ directory }, { signal }), ), - this.controlRequest('读取 MCP 状态', (signal) => - client.mcp.status({ directory }, { signal }) + this.controlRequest("读取 MCP 状态", (signal) => + client.mcp.status({ directory }, { signal }), ), - this.controlRequest('读取技能清单', (signal) => - client.app.skills({ directory }, { signal }) + this.controlRequest("读取技能清单", (signal) => + client.app.skills({ directory }, { signal }), ), - this.controlRequest('读取资源清单', (signal) => - client.experimental.resource.list( - { directory }, - { signal } - ) - ) - ]) + this.controlRequest("读取资源清单", (signal) => + client.experimental.resource.list({ directory }, { signal }), + ), + ]); if ( - availabilityResult.status === 'rejected' || + availabilityResult.status === "rejected" || availabilityResult.value.error || !availabilityResult.value.data ) { return { ...empty, - detail: 'OpenCode Server 无法连接' - } + detail: "OpenCode Server 无法连接", + }; } - const unavailable: string[] = [] + const unavailable: string[] = []; const agents = - agentsResult.status === 'fulfilled' + agentsResult.status === "fulfilled" ? agentsResult.value - : (unavailable.push('Agent'), []) + : (unavailable.push("Agent"), []); const toolIds = - toolsResult.status === 'fulfilled' && + toolsResult.status === "fulfilled" && !toolsResult.value.error && toolsResult.value.data ? toolsResult.value.data - : (unavailable.push('工具'), []) - const tools = mapOpenCodeNativeTools(toolIds) + : (unavailable.push("工具"), []); + const tools = mapOpenCodeNativeTools(toolIds); const commandData = - commandsResult.status === 'fulfilled' && + commandsResult.status === "fulfilled" && !commandsResult.value.error && commandsResult.value.data ? commandsResult.value.data - : (unavailable.push('命令'), []) + : (unavailable.push("命令"), []); const nativeCommandData = commandData .filter((command) => { - const name = boundedNativeIdentifier(command.name) + const name = boundedNativeIdentifier(command.name); return ( name !== undefined && !isTemporaryMcpInventoryItem(name) && !assignedSkillIds.has(name) - ) + ); }) - .slice(0, MAX_NATIVE_COMMANDS) + .slice(0, MAX_NATIVE_COMMANDS); const commands = nativeCommandData.flatMap((command) => { - const id = boundedNativeIdentifier(command.name) + const id = boundedNativeIdentifier(command.name); if (!id) { - return [] + return []; } - const description = boundedNativeText( - command.description, - 2_000 - ) - const agent = boundedNativeIdentifier(command.agent) + const description = boundedNativeText(command.description, 2_000); + const agent = boundedNativeIdentifier(command.agent); return [ { id, name: id.slice(0, 200), ...(description ? { description } : {}), - source: command.source ?? ('command' as const), - ...(agent ? { agent } : {}) - } - ] - }) + source: command.source ?? ("command" as const), + ...(agent ? { agent } : {}), + }, + ]; + }); const prompts = nativeCommandData.flatMap((command) => { - if (command.source !== 'mcp') { - return [] + if (command.source !== "mcp") { + return []; } - const id = boundedNativeIdentifier(command.name) - const prompt = boundedNativeText(command.template, 20_000) + const id = boundedNativeIdentifier(command.name); + const prompt = boundedNativeText(command.template, 20_000); if (!id || !prompt) { - return [] + return []; } - const description = boundedNativeText( - command.description, - 2_000 - ) + const description = boundedNativeText(command.description, 2_000); return [ { id, name: id.slice(0, 200), ...(description ? { description } : {}), prompt, - source: 'mcp' as const - } - ] - }) + source: "mcp" as const, + }, + ]; + }); const lspData = - lspResult.status === 'fulfilled' && + lspResult.status === "fulfilled" && !lspResult.value.error && lspResult.value.data ? lspResult.value.data - : (unavailable.push('LSP'), []) + : (unavailable.push("LSP"), []); const lsp = lspData .flatMap((server) => { - const id = boundedNativeIdentifier(server.id) - const name = boundedNativeText(server.name, 200) + const id = boundedNativeIdentifier(server.id); + const name = boundedNativeText(server.name, 200); if (!id || !name) { - return [] + return []; } return [ { id, name, - status: server.status - } - ] + status: server.status, + }, + ]; }) - .slice(0, MAX_NATIVE_LSP) + .slice(0, MAX_NATIVE_LSP); const formatterData = - formattersResult.status === 'fulfilled' && + formattersResult.status === "fulfilled" && !formattersResult.value.error && formattersResult.value.data ? formattersResult.value.data - : (unavailable.push('Formatter'), []) + : (unavailable.push("Formatter"), []); const formatters = formatterData .flatMap((formatter) => { - const id = boundedNativeIdentifier(formatter.name) + const id = boundedNativeIdentifier(formatter.name); if (!id) { - return [] + return []; } return [ { @@ -1765,97 +1736,91 @@ export class OpenCodeRuntime implements AgentRuntime { enabled: formatter.enabled, extensions: formatter.extensions .flatMap((extension) => { - const value = boundedNativeText(extension, 32) - return value ? [value] : [] + const value = boundedNativeText(extension, 32); + return value ? [value] : []; }) - .slice(0, 100) - } - ] + .slice(0, 100), + }, + ]; }) - .slice(0, MAX_NATIVE_FORMATTERS) + .slice(0, MAX_NATIVE_FORMATTERS); const mcpData = - mcpResult.status === 'fulfilled' && + mcpResult.status === "fulfilled" && !mcpResult.value.error && mcpResult.value.data ? mcpResult.value.data - : (unavailable.push('MCP'), {}) + : (unavailable.push("MCP"), {}); const mcpServers = Object.entries(mcpData) .flatMap(([rawName, server]) => { - const id = boundedNativeIdentifier(rawName) + const id = boundedNativeIdentifier(rawName); if (!id || isTemporaryMcpName(id)) { - return [] + return []; } const status = - server.status === 'needs_auth' - ? ('needs-auth' as const) - : server.status === 'needs_client_registration' - ? ('unsupported' as const) - : server.status + server.status === "needs_auth" + ? ("needs-auth" as const) + : server.status === "needs_client_registration" + ? ("unsupported" as const) + : server.status; return [ { id, name: id.slice(0, 200), - status - } - ] + status, + }, + ]; }) - .slice(0, MAX_NATIVE_MCP_SERVERS) + .slice(0, MAX_NATIVE_MCP_SERVERS); const skillData = - skillsResult.status === 'fulfilled' && + skillsResult.status === "fulfilled" && !skillsResult.value.error && skillsResult.value.data ? skillsResult.value.data - : (unavailable.push('Skill'), []) + : (unavailable.push("Skill"), []); const skills = skillData .flatMap((skill) => { - const id = boundedNativeIdentifier(skill.name) + const id = boundedNativeIdentifier(skill.name); if (!id || assignedSkillIds.has(id)) { - return [] + return []; } - const description = boundedNativeText( - skill.description, - 2_000 - ) + const description = boundedNativeText(skill.description, 2_000); return [ { id, name: id.slice(0, 200), ...(description ? { description } : {}), - source: 'unknown' as const - } - ] + source: "unknown" as const, + }, + ]; }) - .slice(0, MAX_NATIVE_SKILLS) + .slice(0, MAX_NATIVE_SKILLS); const resourceData = - resourcesResult.status === 'fulfilled' && + resourcesResult.status === "fulfilled" && !resourcesResult.value.error && resourcesResult.value.data ? resourcesResult.value.data - : (unavailable.push('资源'), {}) + : (unavailable.push("资源"), {}); const resources = Object.entries(resourceData) .flatMap(([rawId, resource]) => { const id = boundedNativeIdentifier(rawId) ?? - boundedNativeIdentifier(resource.name) - const name = boundedNativeText(resource.name, 200) - const uri = boundedResourceUri(resource.uri) - const server = boundedNativeText(resource.client, 200) + boundedNativeIdentifier(resource.name); + const name = boundedNativeText(resource.name, 200); + const uri = boundedResourceUri(resource.uri); + const server = boundedNativeText(resource.client, 200); if ( !id || !name || !uri || isTemporaryMcpInventoryItem(id, name, server) ) { - return [] + return []; } - const description = boundedNativeText( - resource.description, - 2_000 - ) - const mimeType = boundedNativeText(resource.mimeType, 200) + const description = boundedNativeText(resource.description, 2_000); + const mimeType = boundedNativeText(resource.mimeType, 200); return [ { id, @@ -1863,28 +1828,27 @@ export class OpenCodeRuntime implements AgentRuntime { uri, ...(description ? { description } : {}), ...(mimeType ? { mimeType } : {}), - ...(server ? { server } : {}) - } - ] + ...(server ? { server } : {}), + }, + ]; }) - .slice(0, MAX_NATIVE_RESOURCES) + .slice(0, MAX_NATIVE_RESOURCES); return { - provider: 'opencode', + provider: "opencode", available: true, - inventoryStatus: - unavailable.length === 0 ? 'available' : 'partial', + inventoryStatus: unavailable.length === 0 ? "available" : "partial", detail: unavailable.length === 0 - ? 'OpenCode 原生能力已就绪' - : `OpenCode 已连接；部分原生清单暂不可用：${unavailable.join('、')}`.slice( + ? "OpenCode 原生能力已就绪" + : `OpenCode 已连接；部分原生清单暂不可用：${unavailable.join("、")}`.slice( 0, - 1_000 + 1_000, ), agents, tools, toolsSupported: - toolsResult.status === 'fulfilled' && + toolsResult.status === "fulfilled" && !toolsResult.value.error && toolsResult.value.data !== undefined, commands, @@ -1896,15 +1860,15 @@ export class OpenCodeRuntime implements AgentRuntime { prompts, resources, resourcesSupported: - resourcesResult.status === 'fulfilled' && + resourcesResult.status === "fulfilled" && !resourcesResult.value.error && resourcesResult.value.data !== undefined, context: { - strategy: 'native', + strategy: "native", manualCompact: true, - detail: '由 OpenCode 原生上下文与手动 Compact 管理' - } - } + detail: "由 OpenCode 原生上下文与手动 Compact 管理", + }, + }; } private async getSessionId( @@ -1913,102 +1877,95 @@ export class OpenCodeRuntime implements AgentRuntime { directory: string, signal: AbortSignal, agent?: string, - permission?: PermissionRuleset + permission?: PermissionRuleset, ): Promise<{ id: string; created: boolean }> { - const current = this.sessions.get(request.conversationId) + const current = this.sessions.get(request.conversationId); if (current) { - return { id: current, created: false } + return { id: current, created: false }; } - const pending = this.sessionInitializations.get( - request.conversationId - ) + const pending = this.sessionInitializations.get(request.conversationId); if (pending) { return { id: await awaitWithAbort(pending, signal), - created: false - } + created: false, + }; } const creation: Promise = this.controlRequest( - '创建会话', + "创建会话", (controlSignal) => client.session.create( { - title: 'GoodBuddy 对话', + title: "GoodBuddy 对话", directory, ...(agent ? { agent } : {}), - ...(permission ? { permission } : {}) + ...(permission ? { permission } : {}), }, - { signal: controlSignal } + { signal: controlSignal }, ), signal, async (response) => { - const sessionId = response.data?.id + const sessionId = response.data?.id; if (sessionId) { await client.session .delete( { sessionID: sessionId, - directory + directory, }, - { signal: AbortSignal.timeout(1_000) } + { signal: AbortSignal.timeout(1_000) }, ) - .catch(() => undefined) + .catch(() => undefined); } + }, + ).then((response) => { + if (!response.data) { + throw new Error("OpenCode 会话创建失败"); } - ) - .then((response) => { - if (!response.data) { - throw new Error('OpenCode 会话创建失败') + const sessionId = response.data.id; + const stillCurrent = + this.sessionInitializations.get(request.conversationId) === creation; + if (signal.aborted || !stillCurrent) { + if (this.sessions.get(request.conversationId) !== sessionId) { + void client.session + .delete( + { + sessionID: sessionId, + directory, + }, + { signal: AbortSignal.timeout(1_000) }, + ) + .catch(() => undefined); } - const sessionId = response.data.id - const stillCurrent = - this.sessionInitializations.get(request.conversationId) === - creation - if (signal.aborted || !stillCurrent) { - if (this.sessions.get(request.conversationId) !== sessionId) { - void client.session - .delete( - { - sessionID: sessionId, - directory - }, - { signal: AbortSignal.timeout(1_000) } - ) - .catch(() => undefined) - } - if (signal.aborted) { - throw signal.reason - } - throw new Error('OpenCode 会话初始化已失效') + if (signal.aborted) { + throw signal.reason; } - this.sessions.set(request.conversationId, sessionId) - return sessionId - }) - this.sessionInitializations.set(request.conversationId, creation) + throw new Error("OpenCode 会话初始化已失效"); + } + this.sessions.set(request.conversationId, sessionId); + return sessionId; + }); + this.sessionInitializations.set(request.conversationId, creation); try { return { id: await awaitWithAbort(creation, signal), - created: true - } + created: true, + }; } finally { if ( - this.sessionInitializations.get(request.conversationId) === - creation + this.sessionInitializations.get(request.conversationId) === creation ) { - this.sessionInitializations.delete(request.conversationId) + this.sessionInitializations.delete(request.conversationId); } } } async *run( request: AgentExecutionRequest, - signal: AbortSignal + signal: AbortSignal, ): AsyncGenerator { - const configuredTimeout = this.dependencies.executionTimeoutMs + const configuredTimeout = this.dependencies.executionTimeoutMs; const deadline = - configuredTimeout === undefined - ? undefined - : new AbortController() + configuredTimeout === undefined ? undefined : new AbortController(); const deadlineTimer = configuredTimeout === undefined || deadline === undefined ? undefined @@ -2017,229 +1974,225 @@ export class OpenCodeRuntime implements AgentRuntime { deadline.abort( new Error( `OpenCode 执行超过 ${executionDeadlineLabel( - configuredTimeout - )}总时限` - ) + configuredTimeout, + )}总时限`, + ), ), - configuredTimeout - ) - deadlineTimer?.unref?.() + configuredTimeout, + ); + deadlineTimer?.unref?.(); const executionSignal = deadline ? AbortSignal.any([signal, deadline.signal]) - : signal - let releaseConversation: (() => void) | undefined + : signal; + let releaseConversation: (() => void) | undefined; try { releaseConversation = await this.acquireConversationRun( request.conversationId, - executionSignal - ) - yield* this.runUnlocked(request, executionSignal) + executionSignal, + ); + yield* this.runUnlocked(request, executionSignal); } catch (error) { if (deadline?.signal.aborted && !signal.aborted) { - throw deadline.signal.reason + throw deadline.signal.reason; } - throw error + throw error; } finally { if (deadlineTimer) { - clearTimeout(deadlineTimer) + clearTimeout(deadlineTimer); } - releaseConversation?.() + releaseConversation?.(); } } private async *runUnlocked( request: AgentExecutionRequest, - signal: AbortSignal + signal: AbortSignal, ): AsyncGenerator { - signal.throwIfAborted() + signal.throwIfAborted(); if ( request.images?.length && this.options.modelProfile && this.options.modelProfile.supportsImageInput !== true ) { - throw new Error('当前模型连接未启用图像输入') + throw new Error("当前模型连接未启用图像输入"); } - const client = await this.getClient(signal) - const directory = this.options.defaultWorkspace + const client = await this.getClient(signal); + const directory = this.options.defaultWorkspace; const runtimeControl = - request.runtimeControl?.provider === 'opencode' + request.runtimeControl?.provider === "opencode" ? request.runtimeControl - : undefined - if ( - runtimeControl?.command && - !this.supportsNativeCustomization() - ) { - throw new Error( - '外部 OpenCode Server 不支持由 GoodBuddy 执行原生命令' - ) + : undefined; + if (runtimeControl?.command && !this.supportsNativeCustomization()) { + throw new Error("外部 OpenCode Server 不支持由 GoodBuddy 执行原生命令"); } const selectedAgent = await this.resolveSelectedAgent( client, request, - signal - ) + signal, + ); let selectedCommand: | { - name: string - arguments: string + name: string; + arguments: string; } - | undefined + | undefined; if (runtimeControl?.command) { const commandResponse = await this.controlRequest( - '验证原生命令', + "验证原生命令", (controlSignal) => - client.command.list( - { directory }, - { signal: controlSignal } - ), - signal - ) + client.command.list({ directory }, { signal: controlSignal }), + signal, + ); if (commandResponse.error || !commandResponse.data) { - throw new Error('OpenCode 无法验证原生命令') + throw new Error("OpenCode 无法验证原生命令"); } const assignedSkillIds = new Set( - (this.options.skillPackages ?? []).map((skill) => skill.id) - ) - const command = commandResponse.data.find( - (candidate) => { - const name = boundedNativeIdentifier(candidate.name) - return ( - name !== undefined && - name === runtimeControl.command?.name && - !isTemporaryMcpName(name) && - !assignedSkillIds.has(name) - ) - } - ) + (this.options.skillPackages ?? []).map((skill) => skill.id), + ); + const command = commandResponse.data.find((candidate) => { + const name = boundedNativeIdentifier(candidate.name); + return ( + name !== undefined && + name === runtimeControl.command?.name && + !isTemporaryMcpName(name) && + !assignedSkillIds.has(name) + ); + }); if (!command) { throw new Error( - `OpenCode 原生命令不存在或已失效：${runtimeControl.command.name.slice(0, 128)}` - ) + `OpenCode 原生命令不存在或已失效：${runtimeControl.command.name.slice(0, 128)}`, + ); } - selectedCommand = runtimeControl.command + selectedCommand = runtimeControl.command; } - const nativeSkillIds = this.getNativeSkillIds() + const nativeSkillIds = this.getNativeSkillIds(); const nativeSkillPermissionRules = - createSkillPermissionRules(nativeSkillIds) - let knowledgeMcpName: string | undefined - let knowledgeToolIds: string[] = [] - let customMcpName: string | undefined - let customMcpToken: string | undefined - const subscriptionController = new AbortController() + createSkillPermissionRules(nativeSkillIds); + let knowledgeMcpName: string | undefined; + let knowledgeToolIds: string[] = []; + let customMcpName: string | undefined; + let customMcpToken: string | undefined; + const attemptedMcpNames: string[] = []; + const subscriptionController = new AbortController(); try { if ( request.knowledgeCapabilityToken && this.usesEmbeddedPermissionMediation() && this.options.knowledgeGateway?.getEndpoint() ) { - const knowledgeEndpoint = - this.options.knowledgeGateway.getEndpoint()! - knowledgeMcpName = `goodbuddy-data-${createHash('sha256') + const knowledgeEndpoint = this.options.knowledgeGateway.getEndpoint()!; + knowledgeMcpName = `goodbuddy-data-${createHash("sha256") .update(`${request.conversationId}\0${request.requestId}`) - .digest('hex') - .slice(0, 20)}` - const added = await this.controlRequest( - '连接内置只读工具', - (controlSignal) => - client.mcp.add( - { - directory, - name: knowledgeMcpName, - config: { - type: 'remote', - url: knowledgeEndpoint, - enabled: true, - headers: { - Authorization: `Bearer ${request.knowledgeCapabilityToken}` + .digest("hex") + .slice(0, 20)}`; + const added = await this.mutateMcp(signal, () => + this.controlRequest( + "连接内置只读工具", + (controlSignal) => { + attemptedMcpNames.push(knowledgeMcpName!); + return client.mcp.add( + { + directory, + name: knowledgeMcpName, + config: { + type: "remote", + url: knowledgeEndpoint, + enabled: true, + headers: { + Authorization: `Bearer ${request.knowledgeCapabilityToken}`, + }, + oauth: false, + }, }, - oauth: false - } + { signal: controlSignal }, + ); }, - { signal: controlSignal } + signal, ), - signal - ) + ); if (added.error || !added.data) { - throw new Error('OpenCode 内置只读工具连接失败') - } - const addedStatus = added.data[knowledgeMcpName] - if (!addedStatus || addedStatus.status !== 'connected') { throw new Error( - `OpenCode 内置只读工具连接失败（${addedStatus?.status ?? 'unknown'}）` - ) + opencodeErrorMessage(added.error, "OpenCode 内置只读工具连接失败"), + ); + } + const addedStatus = added.data[knowledgeMcpName]; + if (!addedStatus || addedStatus.status !== "connected") { + throw new Error( + `OpenCode 内置只读工具连接失败（${addedStatus?.status ?? "unknown"}）`, + ); } // OpenCode 1.18.x does not include dynamically added MCP tools in // experimental/tool/ids. Its model tool namespace is deterministic: // "_". - knowledgeToolIds = - this.options.knowledgeGateway - .getAvailableToolNames(request.knowledgeCapabilityToken) - .map((toolName) => `${knowledgeMcpName}_${toolName}`) + knowledgeToolIds = this.options.knowledgeGateway + .getAvailableToolNames(request.knowledgeCapabilityToken) + .map((toolName) => `${knowledgeMcpName}_${toolName}`); } if ( - request.workMode === 'execute' && + request.workMode === "execute" && this.usesEmbeddedPermissionMediation() && this.options.knowledgeGateway?.getEndpoint() && this.options.mcpServers?.length ) { - const customMcpEndpoint = - this.options.knowledgeGateway.getEndpoint()! + const customMcpEndpoint = this.options.knowledgeGateway.getEndpoint()!; customMcpToken = this.options.knowledgeGateway.grantCustomMcp( request.requestId, this.options.mcpServers, - signal - ) + signal, + ); if (customMcpToken) { const tools = await this.options.knowledgeGateway.prepareCustomMcpTools( customMcpToken, - signal - ) - customMcpName = `goodbuddy-custom-${createHash('sha256') + signal, + ); + customMcpName = `goodbuddy-custom-${createHash("sha256") .update(`${request.conversationId}\0${request.requestId}`) - .digest('hex') - .slice(0, 20)}` - const added = await this.controlRequest( - '连接自定义 MCP 工具', - (controlSignal) => - client.mcp.add( - { - directory, - name: customMcpName, - config: { - type: 'remote', - url: customMcpEndpoint, - enabled: true, - headers: { - Authorization: `Bearer ${customMcpToken}` + .digest("hex") + .slice(0, 20)}`; + const added = await this.mutateMcp(signal, () => + this.controlRequest( + "连接自定义 MCP 工具", + (controlSignal) => { + attemptedMcpNames.push(customMcpName!); + return client.mcp.add( + { + directory, + name: customMcpName, + config: { + type: "remote", + url: customMcpEndpoint, + enabled: true, + headers: { + Authorization: `Bearer ${customMcpToken}`, + }, + oauth: false, + }, }, - oauth: false - } + { signal: controlSignal }, + ); }, - { signal: controlSignal } + signal, ), - signal - ) - const addedStatus = added.data?.[customMcpName] + ); + const addedStatus = added.data?.[customMcpName]; if ( added.error || !added.data || !addedStatus || - addedStatus.status !== 'connected' + addedStatus.status !== "connected" ) { throw new Error( - `OpenCode 自定义 MCP 连接失败（${addedStatus?.status ?? 'unknown'}）` - ) + `OpenCode 自定义 MCP 连接失败（${addedStatus?.status ?? "unknown"}）`, + ); } knowledgeToolIds.push( - ...tools.map( - (tool) => `${customMcpName}_${tool.name}` - ) - ) + ...tools.map((tool) => `${customMcpName}_${tool.name}`), + ); } } const permission = - request.workMode === 'execute' + request.workMode === "execute" ? [ ...executePermissionRules, ...(this.usesEmbeddedPermissionMediation() @@ -2248,9 +2201,9 @@ export class OpenCodeRuntime implements AgentRuntime { ...nativeSkillPermissionRules, ...knowledgeToolIds.map((toolId) => ({ permission: toolId, - pattern: '*', - action: 'allow' as const - })) + pattern: "*", + action: "allow" as const, + })), ] : knowledgeToolIds.length > 0 ? [ @@ -2258,45 +2211,37 @@ export class OpenCodeRuntime implements AgentRuntime { ...nativeSkillPermissionRules, ...knowledgeToolIds.map((toolId) => ({ permission: toolId, - pattern: '*', - action: 'allow' as const - })) + pattern: "*", + action: "allow" as const, + })), ] - : [ - ...readOnlyPermissionRules, - ...nativeSkillPermissionRules - ] - let toolOverrides: Record | undefined - if (request.workMode !== 'execute') { + : [...readOnlyPermissionRules, ...nativeSkillPermissionRules]; + let toolOverrides: Record | undefined; + if (request.workMode !== "execute") { const tools = await this.controlRequest( - '读取工具清单', + "读取工具清单", (controlSignal) => - client.tool.ids( - { directory }, - { signal: controlSignal } - ), - signal - ) + client.tool.ids({ directory }, { signal: controlSignal }), + signal, + ); if (tools.error || !tools.data) { - throw new Error('OpenCode 无法确认工具已禁用，已阻止只读请求') + throw new Error("OpenCode 无法确认工具已禁用，已阻止只读请求"); } toolOverrides = { - ...Object.fromEntries( - tools.data.map((toolId) => [toolId, false]) - ), + ...Object.fromEntries(tools.data.map((toolId) => [toolId, false])), ...temporaryMcpToolOverrides, ...Object.fromEntries( - knowledgeToolIds.map((toolId) => [toolId, true]) + knowledgeToolIds.map((toolId) => [toolId, true]), ), - ...(nativeSkillIds.length > 0 ? { skill: true } : {}) - } + ...(nativeSkillIds.length > 0 ? { skill: true } : {}), + }; } else if (this.usesEmbeddedPermissionMediation()) { toolOverrides = { ...temporaryMcpToolOverrides, ...Object.fromEntries( - knowledgeToolIds.map((toolId) => [toolId, true]) - ) - } + knowledgeToolIds.map((toolId) => [toolId, true]), + ), + }; } const session = await this.getSessionId( client, @@ -2304,802 +2249,764 @@ export class OpenCodeRuntime implements AgentRuntime { directory, signal, selectedAgent, - permission - ) - const sessionId = session.id + permission, + ); + const sessionId = session.id; if (!session.created) { const update = await this.controlRequest( - '更新会话权限', + "更新会话权限", (controlSignal) => client.session.update( - { - sessionID: sessionId, - directory, - permission - }, - { signal: controlSignal } - ), - signal - ) + { + sessionID: sessionId, + directory, + permission, + }, + { signal: controlSignal }, + ), + signal, + ); if (update.error || !update.data) { - throw new Error('OpenCode 会话权限配置失败') + throw new Error("OpenCode 会话权限配置失败"); } } - yield { - requestId: request.requestId, - type: 'status', - message: 'OpenCode 正在处理请求' - } + yield { + requestId: request.requestId, + type: "status", + message: "OpenCode 正在处理请求", + }; - const subscription = await this.controlRequest( - '订阅事件流', - (controlSignal) => - client.event.subscribe( - { directory }, - { signal: controlSignal } - ), - AbortSignal.any([signal, subscriptionController.signal]) - ) + const subscription = await this.controlRequest( + "订阅事件流", + (controlSignal) => + client.event.subscribe({ directory }, { signal: controlSignal }), + AbortSignal.any([signal, subscriptionController.signal]), + ); - const abortSession = (): void => { - void client.session.abort({ - sessionID: sessionId, - directory - }).catch(() => undefined) - } - signal.addEventListener('abort', abortSession, { once: true }) + const abortSession = (): void => { + void client.session + .abort({ + sessionID: sessionId, + directory, + }) + .catch(() => undefined); + }; + signal.addEventListener("abort", abortSession, { once: true }); - const toolStates = new Map< - string, - { - name: string - state: 'pending' | 'running' | 'completed' | 'failed' - input?: string - output?: string - error?: string - subagent?: SubagentEvent - } - >() - const reasoningPartIds = new Set() - const subagentProgress = new OpenCodeSubagentProgress(request.requestId, sessionId) - const reportedQuestionIds = new Map() - let aggregateOutputBytes = 0 - let outputTruncated = false - const retainOutputDelta = ( - delta: string - ): { delta: string; truncated: boolean } => { - if (outputTruncated) { - return { delta: '', truncated: false } - } - const bytes = Buffer.from(delta) - const remaining = - MAX_EXECUTION_OUTPUT_BYTES - aggregateOutputBytes - if (bytes.byteLength <= remaining) { - aggregateOutputBytes += bytes.byteLength - return { delta, truncated: false } - } - outputTruncated = true - aggregateOutputBytes = MAX_EXECUTION_OUTPUT_BYTES - return { - delta: bytes - .subarray(0, Math.max(0, remaining)) - .toString('utf8') - .replace(/\uFFFD+$/u, ''), - truncated: true - } - } - let hasResponseTextAfterFailure = false - try { - const promptText = promptWithUntrustedConversationHistory( - request, - session.created - ) - const imageParts = (request.images ?? []).map((image) => ({ - type: 'file' as const, - mime: image.mediaType, - filename: image.name, - url: `data:${image.mediaType};base64,${image.data}` - })) - const prompt = selectedCommand - ? client.session.command( - { - sessionID: sessionId, - directory, - command: selectedCommand.name, - arguments: selectedCommand.arguments, - ...(selectedAgent ? { agent: selectedAgent } : {}), - ...(this.options.modelProfile - ? { - model: `${resolveOpenCodeProvider( - this.options.modelProfile - ).id}/${this.options.modelProfile.modelName}` - } - : {}), - ...(imageParts.length > 0 - ? { parts: imageParts } - : {}) - }, - { signal } - ) - : this.controlRequest( - '提交请求', - (controlSignal) => - client.session.promptAsync( - { - sessionID: sessionId, - directory, - model: this.options.modelProfile - ? { - providerID: resolveOpenCodeProvider( - this.options.modelProfile - ).id, - modelID: this.options.modelProfile.modelName - } - : undefined, - ...(selectedAgent - ? { agent: selectedAgent } - : {}), - system: - nativeSkillIds.length > 0 - ? undefined - : this.options.skillInstructions || undefined, - ...(toolOverrides - ? { tools: toolOverrides } - : {}), - parts: [ - { type: 'text' as const, text: promptText }, - ...imageParts - ] - }, - { signal: controlSignal } - ), - signal - ) - prompt.catch(() => undefined) - - const repliedPermissionIds = new Set() - const reportedMessageIds = new Set() - for await (const event of subscription.stream) { - const childProgress = subagentProgress.update(event) - if ( - childProgress && - 'sessionID' in event.properties && - event.properties.sessionID !== sessionId - ) { - const task = childProgress.runtimeCallId - ? toolStates.get(childProgress.runtimeCallId) - : undefined - if (task) task.subagent = childProgress - yield childProgress + const toolStates = new Map< + string, + { + name: string; + state: "pending" | "running" | "completed" | "failed"; + input?: string; + output?: string; + error?: string; + subagent?: SubagentEvent; } - if ( - event.type === 'message.updated' && - event.properties.sessionID === sessionId && - event.properties.info.sessionID === sessionId && - event.properties.info.role === 'assistant' && - !reportedMessageIds.has(event.properties.info.id) - ) { - const usage = createUsageEvent( - request.requestId, - event.properties.info - ) - if (usage) { - reportedMessageIds.add(event.properties.info.id) - yield usage + >(); + const reasoningPartIds = new Set(); + const subagentProgress = new OpenCodeSubagentProgress( + request.requestId, + sessionId, + ); + const reportedQuestionIds = new Map(); + let aggregateOutputBytes = 0; + let outputTruncated = false; + const retainOutputDelta = ( + delta: string, + ): { delta: string; truncated: boolean } => { + if (outputTruncated) { + return { delta: "", truncated: false }; + } + const bytes = Buffer.from(delta); + const remaining = MAX_EXECUTION_OUTPUT_BYTES - aggregateOutputBytes; + if (bytes.byteLength <= remaining) { + aggregateOutputBytes += bytes.byteLength; + return { delta, truncated: false }; + } + outputTruncated = true; + aggregateOutputBytes = MAX_EXECUTION_OUTPUT_BYTES; + return { + delta: bytes + .subarray(0, Math.max(0, remaining)) + .toString("utf8") + .replace(/\uFFFD+$/u, ""), + truncated: true, + }; + }; + let hasResponseTextAfterFailure = false; + try { + const promptText = promptWithUntrustedConversationHistory( + request, + session.created, + ); + const imageParts = (request.images ?? []).map((image) => ({ + type: "file" as const, + mime: image.mediaType, + filename: image.name, + url: `data:${image.mediaType};base64,${image.data}`, + })); + const prompt = selectedCommand + ? client.session.command( + { + sessionID: sessionId, + directory, + command: selectedCommand.name, + arguments: selectedCommand.arguments, + ...(selectedAgent ? { agent: selectedAgent } : {}), + ...(this.options.modelProfile + ? { + model: `${ + resolveOpenCodeProvider(this.options.modelProfile).id + }/${this.options.modelProfile.modelName}`, + } + : {}), + ...(imageParts.length > 0 ? { parts: imageParts } : {}), + }, + { signal }, + ) + : this.controlRequest( + "提交请求", + (controlSignal) => + client.session.promptAsync( + { + sessionID: sessionId, + directory, + model: this.options.modelProfile + ? { + providerID: resolveOpenCodeProvider( + this.options.modelProfile, + ).id, + modelID: this.options.modelProfile.modelName, + } + : undefined, + ...(selectedAgent ? { agent: selectedAgent } : {}), + system: + nativeSkillIds.length > 0 + ? undefined + : this.options.skillInstructions || undefined, + ...(toolOverrides ? { tools: toolOverrides } : {}), + parts: [ + { type: "text" as const, text: promptText }, + ...imageParts, + ], + }, + { signal: controlSignal }, + ), + signal, + ); + prompt.catch(() => undefined); + + const repliedPermissionIds = new Set(); + const reportedMessageIds = new Set(); + for await (const event of subscription.stream) { + const childProgress = subagentProgress.update(event); + if ( + childProgress && + "sessionID" in event.properties && + event.properties.sessionID !== sessionId + ) { + const task = childProgress.runtimeCallId + ? toolStates.get(childProgress.runtimeCallId) + : undefined; + if (task) task.subagent = childProgress; + yield childProgress; } - } - - if ( - event.type === 'message.part.delta' && - event.properties.sessionID === sessionId && - event.properties.delta - ) { - const reasoning = - reasoningPartIds.has(event.properties.partID) || - [ - 'reasoning', - 'reasoning_content', - 'reasoning_details', - 'thinking' - ].includes(event.properties.field) - if (reasoning || event.properties.field === 'text') { - if ( - !reasoning && - /\S/u.test(event.properties.delta) && - [...toolStates.values()].some( - (tool) => tool.state === 'failed' - ) - ) { - hasResponseTextAfterFailure = true + if ( + event.type === "message.updated" && + event.properties.sessionID === sessionId && + event.properties.info.sessionID === sessionId && + event.properties.info.role === "assistant" && + !reportedMessageIds.has(event.properties.info.id) + ) { + const usage = createUsageEvent( + request.requestId, + event.properties.info, + ); + if (usage) { + reportedMessageIds.add(event.properties.info.id); + yield usage; } - const retained = retainOutputDelta(event.properties.delta) + } + + if ( + event.type === "message.part.delta" && + event.properties.sessionID === sessionId && + event.properties.delta + ) { + const reasoning = + reasoningPartIds.has(event.properties.partID) || + [ + "reasoning", + "reasoning_content", + "reasoning_details", + "thinking", + ].includes(event.properties.field); + if (reasoning || event.properties.field === "text") { + if ( + !reasoning && + /\S/u.test(event.properties.delta) && + [...toolStates.values()].some((tool) => tool.state === "failed") + ) { + hasResponseTextAfterFailure = true; + } + const retained = retainOutputDelta(event.properties.delta); + if (retained.delta) { + yield { + requestId: request.requestId, + type: reasoning ? "reasoning" : "text", + delta: retained.delta, + }; + } + if (retained.truncated) { + yield { + requestId: request.requestId, + type: "status", + message: + "OpenCode 输出较长，已停止继续展示文本；任务仍在继续", + }; + } + } + } + + if ( + event.type === "message.part.updated" && + event.properties.sessionID === sessionId + ) { + const { part } = event.properties; + if (part.type === "reasoning") { + reasoningPartIds.add(part.id); + } else if (part.type === "tool") { + const callId = part.callID || part.id; + if (!callId || callId.length > 256) { + throw new Error("OpenCode 工具调用 ID 格式无效"); + } + const toolName = part.tool.slice(0, 200); + const state = + part.state.status === "error" ? "failed" : part.state.status; + if (state === "failed") { + hasResponseTextAfterFailure = false; + } + const error = + part.state.status === "error" + ? safeToolErrorDetail(part.state.error) + : undefined; + const input = isRecord(part.state.input) + ? boundedToolDetail(part.state.input, 4_000) + : undefined; + const output = + part.state.status === "completed" && + typeof part.state.output === "string" + ? part.state.output.slice(0, 16_000) + : undefined; + const subagent = + toolName.trim().toLowerCase() === "task" + ? toOpenCodeSubagentEvent({ + requestId: request.requestId, + callId, + state: part.state.status, + input: part.state.input, + output, + error, + }) + : undefined; + toolStates.set(callId, { + name: toolName, + state, + ...(input ? { input } : {}), + ...(output ? { output } : {}), + ...(error ? { error } : {}), + ...(subagent ? { subagent } : {}), + }); + if (subagent) { + const retained = subagentProgress.retain(subagent); + toolStates.get(callId)!.subagent = retained; + yield retained; + } else { + yield { + requestId: request.requestId, + type: "tool", + callId, + name: toolName, + state, + summary: `OpenCode 工具：${toolName}`, + ...(input ? { input } : {}), + ...(output ? { output } : {}), + ...(error ? { error } : {}), + }; + } + } + } + + if ( + event.type === "session.next.reasoning.delta" && + event.properties.sessionID === sessionId && + event.properties.delta + ) { + const retained = retainOutputDelta(event.properties.delta); if (retained.delta) { yield { requestId: request.requestId, - type: reasoning ? 'reasoning' : 'text', - delta: retained.delta - } + type: "reasoning", + delta: retained.delta, + }; } if (retained.truncated) { yield { requestId: request.requestId, - type: 'status', - message: - 'OpenCode 输出较长，已停止继续展示文本；任务仍在继续' - } + type: "status", + message: "OpenCode 输出较长，已停止继续展示文本；任务仍在继续", + }; } } + + if ( + event.type === "question.asked" && + event.properties.sessionID === sessionId + ) { + const questionRequest = parseQuestionRequest( + event.properties, + sessionId, + ); + if ( + questionRequest && + !reportedQuestionIds.has(questionRequest.id) + ) { + const publicQuestionId = createPublicQuestionId( + request.requestId, + sessionId, + questionRequest.id, + ); + if (this.pendingQuestions.has(publicQuestionId)) { + throw new Error("OpenCode 提问公开 ID 与另一活动请求冲突"); + } + reportedQuestionIds.set(questionRequest.id, publicQuestionId); + this.pendingQuestions.set(publicQuestionId, { + client, + directory, + questionCount: questionRequest.questions.length, + upstreamQuestionId: questionRequest.id, + }); + yield { + requestId: request.requestId, + type: "question", + questionId: publicQuestionId, + questions: questionRequest.questions.map((question) => ({ + header: question.header, + question: question.question, + options: question.options.map((option) => ({ + label: option.label, + description: option.description, + })), + multiple: question.multiple ?? false, + custom: question.custom ?? true, + })), + }; + } + } + + if ( + (event.type === "question.replied" || + event.type === "question.rejected") && + event.properties.sessionID === sessionId + ) { + const publicQuestionId = reportedQuestionIds.get( + event.properties.requestID, + ); + if (publicQuestionId) { + this.pendingQuestions.delete(publicQuestionId); + } + } + + if ( + this.usesEmbeddedPermissionMediation() && + event.type === "permission.asked" + ) { + const properties = event.properties as unknown; + if ( + isRecord(properties) && + typeof properties.sessionID === "string" && + properties.sessionID !== sessionId + ) { + continue; + } + + let permissionRequest: PermissionRequest; + try { + const parsed = parsePermissionRequest(properties, sessionId); + if (!parsed) { + throw new Error("OpenCode 权限请求格式无效"); + } + permissionRequest = parsed; + } catch (error) { + if ( + isRecord(properties) && + typeof properties.id === "string" && + properties.id.length > 0 && + properties.id.length <= MAX_PERMISSION_NAME_LENGTH && + !repliedPermissionIds.has(properties.id) + ) { + const permissionId = properties.id; + repliedPermissionIds.add(properties.id); + const rejection = await this.controlRequest( + "回复权限请求", + (controlSignal) => + client.permission.reply( + { + requestID: permissionId, + directory, + reply: "reject", + }, + { signal: controlSignal }, + ), + signal, + ); + if (rejection.error || rejection.data !== true) { + throw new Error("OpenCode 权限拒绝回复失败", { + cause: error, + }); + } + continue; + } + throw error; + } + + if (repliedPermissionIds.has(permissionRequest.id)) { + continue; + } + repliedPermissionIds.add(permissionRequest.id); + + const callId = + permissionRequest.tool?.callID ?? permissionRequest.id; + const toolName = permissionRequest.permission.slice(0, 200); + toolStates.set(callId, { name: toolName, state: "pending" }); + yield { + requestId: request.requestId, + type: "tool", + callId, + name: toolName, + state: "pending", + summary: `OpenCode 工具：${toolName}`, + }; + const allowKnowledge = + request.workMode === "ask" && + knowledgeToolIds.includes(permissionRequest.permission); + const response = await this.controlRequest( + "回复权限请求", + (controlSignal) => + client.permission.reply( + { + requestID: permissionRequest.id, + directory, + reply: + request.workMode === "execute" || allowKnowledge + ? "once" + : "reject", + }, + { signal: controlSignal }, + ), + signal, + ); + if (response.error || response.data !== true) { + throw new Error("OpenCode 权限回复失败"); + } + } + + if ( + event.type === "session.error" && + event.properties.sessionID === sessionId + ) { + const error = event.properties.error; + throw new Error(opencodeErrorMessage(error, "OpenCode 执行失败")); + } + + if ( + event.type === "session.idle" && + event.properties.sessionID === sessionId + ) { + const promptResult = await prompt; + if (promptResult.error) { + throw new Error( + opencodeErrorMessage( + promptResult.error, + "OpenCode 提交请求失败", + ), + ); + } + const incompleteTool = [...toolStates.entries()].find( + ([, tool]) => + tool.state === "pending" || tool.state === "running", + ); + if (incompleteTool) { + const [callId] = incompleteTool; + throw new Error(`OpenCode 工具未完成（${callId.slice(0, 128)}）`); + } + const failedTools = [...toolStates.entries()].filter( + ([, tool]) => tool.state === "failed", + ); + if (failedTools.length > 0 && !hasResponseTextAfterFailure) { + const [callId, tool] = failedTools[0]!; + throw new Error( + `OpenCode 工具执行失败（${callId.slice(0, 128)}）${tool.error ? `：${tool.error}` : ""}`, + ); + } + for (const [callId, tool] of failedTools) { + if (tool.subagent) { + continue; + } + yield { + requestId: request.requestId, + type: "tool", + callId, + name: tool.name, + state: "recoverable", + summary: `OpenCode 已在后续响应中处理工具失败：${tool.name}`, + ...(tool.input ? { input: tool.input } : {}), + ...(tool.output ? { output: tool.output } : {}), + ...(tool.error ? { error: tool.error } : {}), + }; + } + yield { + requestId: request.requestId, + type: "done", + sessionId, + }; + return; + } } - if ( - event.type === 'message.part.updated' && - event.properties.sessionID === sessionId - ) { - const { part } = event.properties - if (part.type === 'reasoning') { - reasoningPartIds.add(part.id) - } else if (part.type === 'tool') { - const callId = part.callID || part.id - if (!callId || callId.length > 256) { - throw new Error('OpenCode 工具调用 ID 格式无效') - } - const toolName = part.tool.slice(0, 200) - const state = - part.state.status === 'error' ? 'failed' : part.state.status - if (state === 'failed') { - hasResponseTextAfterFailure = false - } - const error = - part.state.status === 'error' - ? safeToolErrorDetail(part.state.error) - : undefined - const input = isRecord(part.state.input) - ? boundedToolDetail(part.state.input, 4_000) - : undefined - const output = - part.state.status === 'completed' && - typeof part.state.output === 'string' - ? part.state.output.slice(0, 16_000) - : undefined - const subagent = - toolName.trim().toLowerCase() === 'task' - ? toOpenCodeSubagentEvent({ - requestId: request.requestId, - callId, - state: part.state.status, - input: part.state.input, - output, - error - }) - : undefined - toolStates.set(callId, { - name: toolName, - state, - ...(input ? { input } : {}), - ...(output ? { output } : {}), - ...(error ? { error } : {}), - ...(subagent ? { subagent } : {}) - }) - if (subagent) { - const retained = subagentProgress.retain(subagent) - toolStates.get(callId)!.subagent = retained - yield retained + const promptResult = await prompt; + if (promptResult.error) { + throw new Error( + opencodeErrorMessage(promptResult.error, "OpenCode 提交请求失败"), + ); + } + signal.throwIfAborted(); + throw new Error("OpenCode 事件流意外结束"); + } catch (error) { + abortSession(); + for (const [callId, tool] of toolStates) { + if (tool.state === "pending" || tool.state === "running") { + if (tool.subagent) { + yield { + ...tool.subagent, + state: signal.aborted ? "cancelled" : "failed", + ...(signal.aborted + ? { reason: tool.subagent.reason ?? "父请求已取消" } + : { + error: + tool.subagent.error ?? + "OpenCode 子 Agent 未返回完成状态", + }), + }; } else { yield { requestId: request.requestId, - type: 'tool', + type: "tool", callId, - name: toolName, - state, - summary: `OpenCode 工具：${toolName}`, - ...(input ? { input } : {}), - ...(output ? { output } : {}), - ...(error ? { error } : {}) - } + name: tool.name, + state: "failed", + summary: `OpenCode 工具：${tool.name}`, + ...(tool.input ? { input: tool.input } : {}), + ...(tool.output ? { output: tool.output } : {}), + ...(tool.error ? { error: tool.error } : {}), + }; } } } - - if ( - event.type === 'session.next.reasoning.delta' && - event.properties.sessionID === sessionId && - event.properties.delta - ) { - const retained = retainOutputDelta(event.properties.delta) - if (retained.delta) { - yield { - requestId: request.requestId, - type: 'reasoning', - delta: retained.delta - } - } - if (retained.truncated) { - yield { - requestId: request.requestId, - type: 'status', - message: - 'OpenCode 输出较长，已停止继续展示文本；任务仍在继续' - } - } - } - - if ( - event.type === 'question.asked' && - event.properties.sessionID === sessionId - ) { - const questionRequest = parseQuestionRequest( - event.properties, - sessionId - ) - if ( - questionRequest && - !reportedQuestionIds.has(questionRequest.id) - ) { - const publicQuestionId = createPublicQuestionId( - request.requestId, - sessionId, - questionRequest.id - ) - if (this.pendingQuestions.has(publicQuestionId)) { - throw new Error('OpenCode 提问公开 ID 与另一活动请求冲突') - } - reportedQuestionIds.set( - questionRequest.id, - publicQuestionId - ) - this.pendingQuestions.set(publicQuestionId, { - client, - directory, - questionCount: questionRequest.questions.length, - upstreamQuestionId: questionRequest.id - }) - yield { - requestId: request.requestId, - type: 'question', - questionId: publicQuestionId, - questions: questionRequest.questions.map((question) => ({ - header: question.header, - question: question.question, - options: question.options.map((option) => ({ - label: option.label, - description: option.description - })), - multiple: question.multiple ?? false, - custom: question.custom ?? true - })) - } - } - } - - if ( - (event.type === 'question.replied' || - event.type === 'question.rejected') && - event.properties.sessionID === sessionId - ) { - const publicQuestionId = reportedQuestionIds.get( - event.properties.requestID - ) - if (publicQuestionId) { - this.pendingQuestions.delete(publicQuestionId) - } - } - - if ( - this.usesEmbeddedPermissionMediation() && - event.type === 'permission.asked' - ) { - const properties = event.properties as unknown - if ( - isRecord(properties) && - typeof properties.sessionID === 'string' && - properties.sessionID !== sessionId - ) { - continue - } - - let permissionRequest: PermissionRequest - try { - const parsed = parsePermissionRequest(properties, sessionId) - if (!parsed) { - throw new Error('OpenCode 权限请求格式无效') - } - permissionRequest = parsed - } catch (error) { - if ( - isRecord(properties) && - typeof properties.id === 'string' && - properties.id.length > 0 && - properties.id.length <= MAX_PERMISSION_NAME_LENGTH && - !repliedPermissionIds.has(properties.id) - ) { - const permissionId = properties.id - repliedPermissionIds.add(properties.id) - const rejection = await this.controlRequest( - '回复权限请求', - (controlSignal) => - client.permission.reply( - { - requestID: permissionId, - directory, - reply: 'reject' - }, - { signal: controlSignal } - ), - signal - ) - if (rejection.error || rejection.data !== true) { - throw new Error('OpenCode 权限拒绝回复失败', { - cause: error - }) - } - continue - } - throw error - } - - if (repliedPermissionIds.has(permissionRequest.id)) { - continue - } - repliedPermissionIds.add(permissionRequest.id) - - const callId = ( - permissionRequest.tool?.callID ?? permissionRequest.id - ) - const toolName = permissionRequest.permission.slice(0, 200) - toolStates.set(callId, { name: toolName, state: 'pending' }) - yield { - requestId: request.requestId, - type: 'tool', - callId, - name: toolName, - state: 'pending', - summary: `OpenCode 工具：${toolName}` - } - const allowKnowledge = - request.workMode === 'ask' && - knowledgeToolIds.includes(permissionRequest.permission) - const response = await this.controlRequest( - '回复权限请求', - (controlSignal) => - client.permission.reply( - { - requestID: permissionRequest.id, - directory, - reply: - request.workMode === 'execute' || - allowKnowledge - ? 'once' - : 'reject' - }, - { signal: controlSignal } - ), - signal - ) - if (response.error || response.data !== true) { - throw new Error('OpenCode 权限回复失败') - } - } - - if ( - event.type === 'session.error' && - event.properties.sessionID === sessionId - ) { - const error = event.properties.error - throw new Error( - opencodeErrorMessage(error, 'OpenCode 执行失败') - ) - } - - if ( - event.type === 'session.idle' && - event.properties.sessionID === sessionId - ) { - const promptResult = await prompt - if (promptResult.error) { - throw new Error( - opencodeErrorMessage( - promptResult.error, - 'OpenCode 提交请求失败' - ) - ) - } - const incompleteTool = [...toolStates.entries()].find( - ([, tool]) => - tool.state === 'pending' || tool.state === 'running' - ) - if (incompleteTool) { - const [callId] = incompleteTool - throw new Error( - `OpenCode 工具未完成（${callId.slice(0, 128)}）` - ) - } - const failedTools = [...toolStates.entries()].filter( - ([, tool]) => tool.state === 'failed' - ) - if ( - failedTools.length > 0 && - !hasResponseTextAfterFailure - ) { - const [callId, tool] = failedTools[0]! - throw new Error( - `OpenCode 工具执行失败（${callId.slice(0, 128)}）${tool.error ? `：${tool.error}` : ''}` - ) - } - for (const [callId, tool] of failedTools) { - if (tool.subagent) { - continue - } - yield { - requestId: request.requestId, - type: 'tool', - callId, - name: tool.name, - state: 'recoverable', - summary: `OpenCode 已在后续响应中处理工具失败：${tool.name}`, - ...(tool.input ? { input: tool.input } : {}), - ...(tool.output ? { output: tool.output } : {}), - ...(tool.error ? { error: tool.error } : {}) - } - } - yield { - requestId: request.requestId, - type: 'done', - sessionId - } - return + throw error; + } finally { + signal.removeEventListener("abort", abortSession); + for (const questionId of reportedQuestionIds.values()) { + this.pendingQuestions.delete(questionId); } } - - const promptResult = await prompt - if (promptResult.error) { - throw new Error( - opencodeErrorMessage( - promptResult.error, - 'OpenCode 提交请求失败' - ) - ) - } - signal.throwIfAborted() - throw new Error('OpenCode 事件流意外结束') - } catch (error) { - abortSession() - for (const [callId, tool] of toolStates) { - if (tool.state === 'pending' || tool.state === 'running') { - if (tool.subagent) { - yield { - ...tool.subagent, - state: signal.aborted ? 'cancelled' : 'failed', - ...(signal.aborted - ? { reason: tool.subagent.reason ?? '父请求已取消' } - : { - error: - tool.subagent.error ?? - 'OpenCode 子 Agent 未返回完成状态' - }) - } - } else { - yield { - requestId: request.requestId, - type: 'tool', - callId, - name: tool.name, - state: 'failed', - summary: `OpenCode 工具：${tool.name}`, - ...(tool.input ? { input: tool.input } : {}), - ...(tool.output ? { output: tool.output } : {}), - ...(tool.error ? { error: tool.error } : {}) - } - } - } - } - throw error } finally { - signal.removeEventListener('abort', abortSession) - for (const questionId of reportedQuestionIds.values()) { - this.pendingQuestions.delete(questionId) - } - } - } finally { - subscriptionController.abort() - const cleanupSignal = AbortSignal.timeout(1_000) - if (knowledgeMcpName) { - await awaitWithAbort( - client.mcp.disconnect( - { name: knowledgeMcpName, directory }, - { signal: cleanupSignal } - ), - cleanupSignal - ).catch(() => undefined) - } - if (customMcpName) { - await awaitWithAbort( - client.mcp.disconnect( - { name: customMcpName, directory }, - { signal: cleanupSignal } - ), - cleanupSignal - ).catch(() => undefined) + subscriptionController.abort(); + for (const name of attemptedMcpNames) { + await this.mutateMcp(new AbortController().signal, () => { + // Queue wait must not consume the disconnect timeout. + const cleanupSignal = AbortSignal.timeout(1_000); + return awaitWithAbort( + client.mcp.disconnect( + { name, directory }, + { signal: cleanupSignal }, + ), + cleanupSignal, + ); + }).catch(() => undefined); } if (customMcpToken) { - this.options.knowledgeGateway?.revoke(customMcpToken) + this.options.knowledgeGateway?.revoke(customMcpToken); } } } async respondToQuestion( questionId: string, - answers?: AgentQuestionAnswer[] + answers?: AgentQuestionAnswer[], ): Promise { - const pending = this.pendingQuestions.get(questionId) + const pending = this.pendingQuestions.get(questionId); if (!pending) { - throw new Error('OpenCode 提问已失效或不存在') + throw new Error("OpenCode 提问已失效或不存在"); } const response = answers ? answers.length === pending.questionCount - ? await this.controlRequest( - '提交提问回答', - (signal) => - pending.client.question.reply( - { - requestID: pending.upstreamQuestionId, - directory: pending.directory, - answers - }, - { signal } - ) - ) - : undefined - : await this.controlRequest( - '取消提问', - (signal) => - pending.client.question.reject( + ? await this.controlRequest("提交提问回答", (signal) => + pending.client.question.reply( { requestID: pending.upstreamQuestionId, - directory: pending.directory + directory: pending.directory, + answers, }, - { signal } - ) - ) + { signal }, + ), + ) + : undefined + : await this.controlRequest("取消提问", (signal) => + pending.client.question.reject( + { + requestID: pending.upstreamQuestionId, + directory: pending.directory, + }, + { signal }, + ), + ); if (!response) { - throw new Error('OpenCode 提问回答数量不匹配') + throw new Error("OpenCode 提问回答数量不匹配"); } if (response.error || response.data !== true) { throw new Error( - answers ? 'OpenCode 提交回答失败' : 'OpenCode 取消提问失败' - ) + answers ? "OpenCode 提交回答失败" : "OpenCode 取消提问失败", + ); } - this.pendingQuestions.delete(questionId) + this.pendingQuestions.delete(questionId); } async compactConversation( request: RuntimeConversationCompactInput, - signal: AbortSignal + signal: AbortSignal, ): Promise { - signal.throwIfAborted() + signal.throwIfAborted(); if (!this.supportsNativeCustomization()) { throw new Error( - '外部 OpenCode Server 不支持由 GoodBuddy 执行原生 Compact' - ) + "外部 OpenCode Server 不支持由 GoodBuddy 执行原生 Compact", + ); } - const configuredTimeout = this.dependencies.executionTimeoutMs + const configuredTimeout = this.dependencies.executionTimeoutMs; const deadline = - configuredTimeout === undefined - ? undefined - : new AbortController() + configuredTimeout === undefined ? undefined : new AbortController(); const deadlineTimer = configuredTimeout === undefined || deadline === undefined ? undefined : setTimeout( - () => - deadline.abort( - new Error('OpenCode 原生 Compact 执行超时') - ), - configuredTimeout - ) - deadlineTimer?.unref?.() + () => deadline.abort(new Error("OpenCode 原生 Compact 执行超时")), + configuredTimeout, + ); + deadlineTimer?.unref?.(); const executionSignal = deadline ? AbortSignal.any([signal, deadline.signal]) - : signal - let releaseConversation: (() => void) | undefined + : signal; + let releaseConversation: (() => void) | undefined; try { releaseConversation = await this.acquireConversationRun( request.conversationId, - executionSignal - ) - const sessionId = this.sessions.get(request.conversationId) - if (!sessionId) { + executionSignal, + ); + const client = this.sessions.has(request.conversationId) + ? await this.getClient(executionSignal) + : undefined; + const sessionId = this.sessions.get(request.conversationId); + if (!sessionId || !client) { return { result: { - provider: 'opencode', - strategy: 'native', + provider: "opencode", + strategy: "native", compacted: false, - detail: '当前 GoodBuddy 对话尚无可压缩的 OpenCode 会话' - } - } + detail: "当前 GoodBuddy 对话尚无可压缩的 OpenCode 会话", + }, + }; } - const client = await this.getClient(executionSignal) const context = await this.controlRequest( - '读取原生上下文', + "读取原生上下文", (controlSignal) => client.v2.session.context( { sessionID: sessionId }, - { signal: controlSignal } + { signal: controlSignal }, ), - executionSignal - ) + executionSignal, + ); if (context.error || !context.data) { - throw new Error('OpenCode 原生上下文不可用，无法执行 Compact') + throw new Error("OpenCode 原生上下文不可用，无法执行 Compact"); } const latestAssistant = [...context.data.data] .reverse() - .find((message) => message.type === 'assistant') + .find((message) => message.type === "assistant"); const configuredModel = this.options.modelProfile ? { - providerID: resolveOpenCodeProvider( - this.options.modelProfile - ).id, - modelID: this.options.modelProfile.modelName + providerID: resolveOpenCodeProvider(this.options.modelProfile).id, + modelID: this.options.modelProfile.modelName, } - : latestAssistant?.type === 'assistant' + : latestAssistant?.type === "assistant" ? { providerID: latestAssistant.model.providerID, - modelID: latestAssistant.model.id + modelID: latestAssistant.model.id, } - : undefined + : undefined; if (!configuredModel) { return { result: { - provider: 'opencode', - strategy: 'native', + provider: "opencode", + strategy: "native", compacted: false, - detail: '当前 OpenCode 会话尚无可用于压缩的模型记录' - } - } + detail: "当前 OpenCode 会话尚无可用于压缩的模型记录", + }, + }; } - executionSignal.throwIfAborted() - const subscriptionController = new AbortController() + executionSignal.throwIfAborted(); + const subscriptionController = new AbortController(); const subscriptionSignal = AbortSignal.any([ executionSignal, - subscriptionController.signal - ]) + subscriptionController.signal, + ]); const subscription = await this.controlRequest( - '订阅 Compact 事件流', + "订阅 Compact 事件流", (controlSignal) => client.event.subscribe( { directory: this.options.defaultWorkspace }, - { signal: controlSignal } + { signal: controlSignal }, ), - subscriptionSignal - ) - const usageEvents: RuntimeModelUsageEvent[] = [] - const reportedMessageIds = new Set() + subscriptionSignal, + ); + const usageEvents: RuntimeModelUsageEvent[] = []; + const reportedMessageIds = new Set(); const usageCapture = (async () => { for await (const event of subscription.stream) { if ( - event.type === 'message.updated' && + event.type === "message.updated" && event.properties.sessionID === sessionId && event.properties.info.sessionID === sessionId && - event.properties.info.role === 'assistant' && + event.properties.info.role === "assistant" && !reportedMessageIds.has(event.properties.info.id) ) { const usage = createUsageEvent( request.requestId, - event.properties.info - ) + event.properties.info, + ); if (usage) { - reportedMessageIds.add(event.properties.info.id) - usageEvents.push(usage) + reportedMessageIds.add(event.properties.info.id); + usageEvents.push(usage); } } if ( - event.type === 'session.idle' && + event.type === "session.idle" && event.properties.sessionID === sessionId ) { - return + return; } } - })() + })(); try { const compact = await client.session.summarize( { @@ -3107,96 +3014,90 @@ export class OpenCodeRuntime implements AgentRuntime { directory: this.options.defaultWorkspace, providerID: configuredModel.providerID, modelID: configuredModel.modelID, - auto: false + auto: false, }, - { signal: executionSignal } - ) + { signal: executionSignal }, + ); if (compact.error || compact.data !== true) { throw new Error( - opencodeErrorMessage( - compact.error, - 'OpenCode 原生 Compact 失败' - ) - ) + opencodeErrorMessage(compact.error, "OpenCode 原生 Compact 失败"), + ); } - let graceTimer: ReturnType | undefined + let graceTimer: ReturnType | undefined; try { await Promise.race([ usageCapture, new Promise((resolveGrace) => { graceTimer = setTimeout( resolveGrace, - COMPACTION_USAGE_EVENT_GRACE_MS - ) - graceTimer.unref?.() - }) - ]) + COMPACTION_USAGE_EVENT_GRACE_MS, + ); + graceTimer.unref?.(); + }), + ]); } finally { if (graceTimer) { - clearTimeout(graceTimer) + clearTimeout(graceTimer); } } return { result: { - provider: 'opencode', - strategy: 'native', + provider: "opencode", + strategy: "native", compacted: true, - detail: 'OpenCode 已完成原生上下文压缩' + detail: "OpenCode 已完成原生上下文压缩", }, - ...(usageEvents.length > 0 ? { usageEvents } : {}) - } + ...(usageEvents.length > 0 ? { usageEvents } : {}), + }; } finally { - subscriptionController.abort() - await usageCapture.catch(() => undefined) + subscriptionController.abort(); + await usageCapture.catch(() => undefined); } } catch (error) { if (deadline?.signal.aborted && !signal.aborted) { - throw deadline.signal.reason + throw deadline.signal.reason; } - throw error + throw error; } finally { if (deadlineTimer) { - clearTimeout(deadlineTimer) + clearTimeout(deadlineTimer); } - releaseConversation?.() + releaseConversation?.(); } } async dispose(): Promise { - this.pendingQuestions.clear() - const startingChild = this.startingChild - this.startingChild = undefined + this.pendingQuestions.clear(); + const startingChild = this.startingChild; + this.startingChild = undefined; if (startingChild) { - this.terminate(startingChild) - await this.waitForExit(startingChild) + this.terminate(startingChild); + await this.waitForExit(startingChild); } - const server = this.server - this.server = undefined - this.client = undefined - this.clientInitialization = undefined - this.sessions.clear() - this.sessionInitializations.clear() - this.conversationRunTails.clear() - await server?.close() + const server = this.server; + this.server = undefined; + this.client = undefined; + this.clientInitialization = undefined; + this.sessions.clear(); + this.sessionInitializations.clear(); + this.conversationRunTails.clear(); + await server?.close(); } async releaseConversation(conversationId: string): Promise { - const sessionId = this.sessions.get(conversationId) - this.sessions.delete(conversationId) + const sessionId = this.sessions.get(conversationId); + this.sessions.delete(conversationId); if (!sessionId || !this.client) { - return + return; } - await this.controlRequest( - '释放会话', - (signal) => - this.client!.session.delete( - { - sessionID: sessionId, - directory: this.options.defaultWorkspace - }, - { signal } - ) - ) - .catch(() => undefined) + await this.controlRequest("释放会话", (signal) => + this.client!.session.delete( + { + sessionID: sessionId, + directory: this.options.defaultWorkspace, + }, + { signal }, + ), + ).catch(() => undefined); } } diff --git a/src/renderer/src/App.test.tsx b/src/renderer/src/App.test.tsx index 2ad1c575..20f4a3b8 100644 --- a/src/renderer/src/App.test.tsx +++ b/src/renderer/src/App.test.tsx @@ -5,203 +5,184 @@ import { render, screen, within, - waitFor -} from '@testing-library/react' -import { StrictMode } from 'react' -import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest' + waitFor, +} from "@testing-library/react"; +import { StrictMode } from "react"; +import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; import type { AgentEvent, BrowserLiveState, ContextAttachment, ConversationQueueDispatch, - DesktopApi -} from '../../shared/contracts' + DesktopApi, +} from "../../shared/contracts"; import { defaultLocalToolEnvironmentSettings, - type ApplicationSettings -} from '../../shared/application-settings-contracts' -import type { GlobalShortcutSettingsSnapshot } from '../../shared/shortcut' -import type { - TerminalSnapshot -} from '../../shared/terminal-contracts' -import { - loadBrandingPreferences, - saveBrandingPreferences -} from './branding' + type ApplicationSettings, +} from "../../shared/application-settings-contracts"; +import type { GlobalShortcutSettingsSnapshot } from "../../shared/shortcut"; +import type { TerminalSnapshot } from "../../shared/terminal-contracts"; +import { loadBrandingPreferences, saveBrandingPreferences } from "./branding"; import type { AssistantProject, AssistantSchedule, AssistantTask, - ConversationSnapshot -} from '../../shared/assistant-contracts' + ConversationSnapshot, +} from "../../shared/assistant-contracts"; import { builtInDefaultProjectSeedDescription, - builtInDefaultProjectSeedName -} from '../../shared/assistant-contracts' -import { agentRuntimeSelectionKey } from '../../shared/runtime-selection-contracts' + builtInDefaultProjectSeedName, +} from "../../shared/assistant-contracts"; +import { agentRuntimeSelectionKey } from "../../shared/runtime-selection-contracts"; const speechRecognitionMocks = vi.hoisted(() => ({ - startPcmRecording: vi.fn() -})) + startPcmRecording: vi.fn(), +})); -const messageRenderProbe = vi.hoisted(() => vi.fn()) +const messageRenderProbe = vi.hoisted(() => vi.fn()); -vi.mock('./MarkdownRenderer', async (importOriginal) => { - const original = await importOriginal() +vi.mock("./MarkdownRenderer", async (importOriginal) => { + const original = await importOriginal(); return { ...original, MarkdownRenderer: (props: { children: string; renderHtml?: boolean }) => { - messageRenderProbe(props.children) - return - } - } -}) + messageRenderProbe(props.children); + return ; + }, + }; +}); const lazyRouteMocks = vi.hoisted(() => { - let pending: Promise | undefined - let releasePending: (() => void) | undefined + let pending: Promise | undefined; + let releasePending: (() => void) | undefined; return { suspendKnowledgeRoute(): void { pending = new Promise((resolve) => { - releasePending = resolve - }) + releasePending = resolve; + }); }, releaseKnowledgeRoute(): void { - releasePending?.() - releasePending = undefined - pending = undefined + releasePending?.(); + releasePending = undefined; + pending = undefined; }, async waitForKnowledgeRoute(): Promise { - await pending - } - } -}) + await pending; + }, + }; +}); const routeModuleLoads = vi.hoisted(() => ({ activity: 0, heartbeat: 0, knowledge: 0, magicNotes: 0, - settings: 0 -})) + settings: 0, +})); -vi.mock('./speech-recognition', async (importOriginal) => ({ - ...(await importOriginal< - typeof import('./speech-recognition') - >()), - startPcmRecording: speechRecognitionMocks.startPcmRecording -})) +vi.mock("./speech-recognition", async (importOriginal) => ({ + ...(await importOriginal()), + startPcmRecording: speechRecognitionMocks.startPcmRecording, +})); -vi.mock('./KnowledgeWorkspace', async (importOriginal) => { - await lazyRouteMocks.waitForKnowledgeRoute() - routeModuleLoads.knowledge += 1 - return importOriginal() -}) +vi.mock("./KnowledgeWorkspace", async (importOriginal) => { + await lazyRouteMocks.waitForKnowledgeRoute(); + routeModuleLoads.knowledge += 1; + return importOriginal(); +}); -vi.mock('./HeartbeatCenter', async (importOriginal) => { - routeModuleLoads.heartbeat += 1 - return importOriginal() -}) +vi.mock("./HeartbeatCenter", async (importOriginal) => { + routeModuleLoads.heartbeat += 1; + return importOriginal(); +}); -vi.mock('./MagicNotesWorkspace', async (importOriginal) => { - routeModuleLoads.magicNotes += 1 - return importOriginal() -}) +vi.mock("./MagicNotesWorkspace", async (importOriginal) => { + routeModuleLoads.magicNotes += 1; + return importOriginal(); +}); -vi.mock('./SettingsPanel', async (importOriginal) => { - routeModuleLoads.settings += 1 - return importOriginal() -}) +vi.mock("./SettingsPanel", async (importOriginal) => { + routeModuleLoads.settings += 1; + return importOriginal(); +}); -vi.mock('./ActivityPanel', async (importOriginal) => { - routeModuleLoads.activity += 1 - return importOriginal() -}) +vi.mock("./ActivityPanel", async (importOriginal) => { + routeModuleLoads.activity += 1; + return importOriginal(); +}); -vi.mock('./TerminalPanel', () => ({ +vi.mock("./TerminalPanel", () => ({ TerminalPanel: ({ title }: { title?: string }) => ( -
- ) -})) +
+ ), +})); -import App from './App' -import { ACTIVITY_STORAGE_KEY } from './activity-store' -import { changeUiLocale } from './i18n' -import { UiLocaleProvider } from './i18n/UiLocaleProvider' +import App from "./App"; +import { ACTIVITY_STORAGE_KEY } from "./activity-store"; +import { changeUiLocale } from "./i18n"; +import { UiLocaleProvider } from "./i18n/UiLocaleProvider"; -let agentListener: ((event: AgentEvent) => void) | undefined -let browserListener: ((state: BrowserLiveState) => void) | undefined +let agentListener: ((event: AgentEvent) => void) | undefined; +let browserListener: ((state: BrowserLiveState) => void) | undefined; let fileSelectionProgressListener: - | Parameters< - DesktopApi['context']['onFileSelectionProgress'] - >[0] - | undefined -let newConversationListener: (() => void) | undefined -let magicTodoStatusChangedListener: (() => void) | undefined -let maximizedChangedListener: ((maximized: boolean) => void) | undefined -let beforeQuitListener: (() => Promise) | undefined + Parameters[0] | undefined; +let newConversationListener: (() => void) | undefined; +let magicTodoStatusChangedListener: (() => void) | undefined; +let maximizedChangedListener: ((maximized: boolean) => void) | undefined; +let beforeQuitListener: (() => Promise) | undefined; let conversationQueueDispatchListener: - | ((dispatch: ConversationQueueDispatch) => void) - | undefined + ((dispatch: ConversationQueueDispatch) => void) | undefined; let conversationQueueChangeListener: - | Parameters< - DesktopApi['conversationQueue']['onChanged'] - >[0] - | undefined + Parameters[0] | undefined; let remoteProjectSaveProgressListener: - | Parameters< - DesktopApi['projects']['remote']['onSaveProgress'] - >[0] - | undefined + Parameters[0] | undefined; let remoteProjectRecoveryProgressListener: - | Parameters< - DesktopApi['projects']['remote']['onRecoveryProgress'] - >[0] - | undefined -const removeMaximizedChangedListener = vi.fn() -const removeProjectRecoveryListener = vi.fn() -const run = vi.fn() -const modelProfileId = '00000000-0000-4000-8000-000000000001' -const projectId = '00000000-0000-4000-8000-000000000101' + | Parameters[0] + | undefined; +const removeMaximizedChangedListener = vi.fn(); +const removeProjectRecoveryListener = vi.fn(); +const run = vi.fn(); +const modelProfileId = "00000000-0000-4000-8000-000000000001"; +const projectId = "00000000-0000-4000-8000-000000000101"; const project = { id: projectId, name: builtInDefaultProjectSeedName, description: builtInDefaultProjectSeedDescription, - rootPath: 'C:\\Users\\test', + rootPath: "C:\\Users\\test", executionSpace: { - kind: 'local' as const, - rootPath: 'C:\\Users\\test' + kind: "local" as const, + rootPath: "C:\\Users\\test", }, - defaultWorkMode: 'ask' as const, - kind: 'user' as const, + defaultWorkMode: "ask" as const, + kind: "user" as const, builtInDefault: true, - status: 'active' as const, - createdAt: '2026-07-31T00:00:00.000Z', - updatedAt: '2026-07-31T00:00:00.000Z' -} + status: "active" as const, + createdAt: "2026-07-31T00:00:00.000Z", + updatedAt: "2026-07-31T00:00:00.000Z", +}; const terminalSnapshot: TerminalSnapshot = { - sessionId: '00000000-0000-4000-8000-000000000901', - target: { type: 'local' }, - targetLabel: '本机', - title: '终端 1', - state: 'running', - shell: 'PowerShell 7', - workingDirectory: 'C:\\Users\\test', + sessionId: "00000000-0000-4000-8000-000000000901", + target: { type: "local" }, + targetLabel: "本机", + title: "终端 1", + state: "running", + shell: "PowerShell 7", + workingDirectory: "C:\\Users\\test", size: { cols: 80, rows: 24 }, lastSequence: 0, exit: null, - error: null -} + error: null, +}; const api: DesktopApi = { app: { getInfo: vi.fn(async () => ({ - name: 'GoodBuddy', - version: '0.1.0', - platform: 'win32', - arch: 'x64', - shortcut: 'Ctrl+Shift+Space' + name: "GoodBuddy", + version: "0.1.0", + platform: "win32", + arch: "x64", + shortcut: "Ctrl+Shift+Space", })), show: vi.fn(async () => {}), hide: vi.fn(async () => {}), @@ -210,59 +191,59 @@ const api: DesktopApi = { close: vi.fn(async () => {}), isMaximized: vi.fn(async () => false), onMaximizedChanged: vi.fn((listener) => { - maximizedChangedListener = listener - return removeMaximizedChangedListener + maximizedChangedListener = listener; + return removeMaximizedChangedListener; }), onBeforeQuit: vi.fn((listener) => { - beforeQuitListener = listener + beforeQuitListener = listener; return () => { - beforeQuitListener = undefined - } + beforeQuitListener = undefined; + }; }), clearLocalData: vi.fn(async () => {}), onNewConversation: vi.fn((listener) => { - newConversationListener = listener + newConversationListener = listener; return () => { - newConversationListener = undefined - } + newConversationListener = undefined; + }; }), - onOpenSettings: vi.fn(() => () => {}) + onOpenSettings: vi.fn(() => () => {}), }, clipboard: { - readText: vi.fn(async () => ''), - writeText: vi.fn(async () => {}) + readText: vi.fn(async () => ""), + writeText: vi.fn(async () => {}), }, feedback: { - submit: vi.fn(async () => ({ + submit: vi.fn(async () => ({ ok: true, - reference: 'GOODBUDDY-000001', - duplicate: false - })) + reference: "GOODBUDDY-000001", + duplicate: false, + })), }, agent: { - getStatus: vi.fn(async () => ({ - id: 'model' as const, - label: 'sonnet-5', + getStatus: vi.fn(async () => ({ + id: "model" as const, + label: "sonnet-5", available: true, supportsToolExecution: true, - detail: 'Ready' + detail: "Ready", })), run, cancel: vi.fn(async () => {}), respondApproval: vi.fn(async () => {}), respondQuestion: vi.fn(async () => {}), compactConversation: vi.fn(async () => ({ - provider: 'continue' as const, - strategy: 'goodbuddy-summary' as const, + provider: "continue" as const, + strategy: "goodbuddy-summary" as const, compacted: false, - detail: 'No context to compact' + detail: "No context to compact", })), onEvent: vi.fn((listener) => { - agentListener = listener + agentListener = listener; return () => { - agentListener = undefined - } - }) + agentListener = undefined; + }; + }), }, browser: { navigate: vi.fn(async () => {}), @@ -272,84 +253,83 @@ const api: DesktopApi = { interact: vi.fn(async () => {}), stop: vi.fn(async () => {}), onState: vi.fn((listener) => { - browserListener = listener + browserListener = listener; return () => { - browserListener = undefined - } - }) + browserListener = undefined; + }; + }), }, terminal: { create: vi.fn(async (request) => ({ ...terminalSnapshot, target: request.target, - size: { cols: request.cols, rows: request.rows } + size: { cols: request.cols, rows: request.rows }, })), write: vi.fn(async () => undefined), resize: vi.fn(async () => undefined), close: vi.fn(async () => ({ ...terminalSnapshot, - state: 'exited' as const, - exit: { exitCode: 0, signal: null } + state: "exited" as const, + exit: { exitCode: 0, signal: null }, })), getSnapshot: vi.fn(async () => terminalSnapshot), ack: vi.fn(async () => undefined), - onEvent: vi.fn(() => () => {}) + onEvent: vi.fn(() => () => {}), }, speech: { - transcribe: vi.fn(async () => ({ text: '本地语音结果' })), - cancel: vi.fn(async () => true) + transcribe: vi.fn(async () => ({ text: "本地语音结果" })), + cancel: vi.fn(async () => true), }, settings: { - getRuntime: vi.fn(async () => ({ - provider: 'auto', - modelBaseUrl: 'https://bigtoken.ai', - modelName: 'sonnet-5', - modelProtocol: 'anthropic-messages', - modelAuthentication: 'api-key', - imageGenerationQuality: 'auto', - opencodeBaseUrl: '', + getRuntime: vi.fn(async () => ({ + provider: "auto", + modelBaseUrl: "https://bigtoken.ai", + modelName: "sonnet-5", + modelProtocol: "anthropic-messages", + modelAuthentication: "api-key", + imageGenerationQuality: "auto", + opencodeBaseUrl: "", opencodeEmbedded: false, - opencodeBinaryPath: '', - opencodeConfigPath: '', - continueBinaryPath: '', - continueConfigPath: '', - continueMode: 'chat', + opencodeBinaryPath: "", + opencodeConfigPath: "", + continueBinaryPath: "", + continueConfigPath: "", + continueMode: "chat", subagentSmartRoutingEnabled: false, knowledgeEmbeddingEnabled: false, - knowledgeEmbeddingBaseUrl: - 'http://127.0.0.1:11434/v1/embeddings', - knowledgeEmbeddingModel: 'nomic-embed-text', + knowledgeEmbeddingBaseUrl: "http://127.0.0.1:11434/v1/embeddings", + knowledgeEmbeddingModel: "nomic-embed-text", knowledgeEmbeddingApiKeyConfigured: false, - knowledgeEmbeddingCredentialSource: 'none', - workspacePath: 'C:\\Users\\test', + knowledgeEmbeddingCredentialSource: "none", + workspacePath: "C:\\Users\\test", apiKeyConfigured: false, - credentialSource: 'none', + credentialSource: "none", modelProfiles: [ { id: modelProfileId, - name: '默认模型', - baseUrl: 'https://bigtoken.ai', - modelName: 'sonnet-5', - protocol: 'anthropic-messages', - authentication: 'api-key', - imageGenerationQuality: 'auto', + name: "默认模型", + baseUrl: "https://bigtoken.ai", + modelName: "sonnet-5", + protocol: "anthropic-messages", + authentication: "api-key", + imageGenerationQuality: "auto", apiKeyConfigured: false, - credentialSource: 'none' - } + credentialSource: "none", + }, ], defaultModelProfileId: modelProfileId, opencodeModelSource: { - kind: 'profile', - profileId: modelProfileId + kind: "profile", + profileId: modelProfileId, }, continueModelSource: { - kind: 'profile', - profileId: modelProfileId + kind: "profile", + profileId: modelProfileId, }, secureStorageAvailable: true, - toolApproval: 'always' + toolApproval: "always", })), - updateRuntime: vi.fn( + updateRuntime: vi.fn( async (input) => ({ provider: input.provider, modelBaseUrl: input.modelBaseUrl, @@ -364,127 +344,119 @@ const api: DesktopApi = { continueBinaryPath: input.continueBinaryPath, continueConfigPath: input.continueConfigPath, continueMode: input.continueMode, - subagentSmartRoutingEnabled: - input.subagentSmartRoutingEnabled ?? false, + subagentSmartRoutingEnabled: input.subagentSmartRoutingEnabled ?? false, knowledgeEmbeddingEnabled: input.knowledgeEmbeddingEnabled, knowledgeEmbeddingBaseUrl: input.knowledgeEmbeddingBaseUrl, knowledgeEmbeddingModel: input.knowledgeEmbeddingModel, knowledgeEmbeddingApiKeyConfigured: - input.knowledgeEmbeddingApiKey?.action === 'replace', + input.knowledgeEmbeddingApiKey?.action === "replace", knowledgeEmbeddingCredentialSource: - input.knowledgeEmbeddingApiKey?.action === 'replace' - ? 'encrypted' - : 'none', + input.knowledgeEmbeddingApiKey?.action === "replace" + ? "encrypted" + : "none", workspacePath: input.workspacePath, - apiKeyConfigured: input.apiKey.action === 'replace', + apiKeyConfigured: input.apiKey.action === "replace", credentialSource: - input.apiKey.action === 'replace' ? 'encrypted' : 'none', + input.apiKey.action === "replace" ? "encrypted" : "none", modelProfiles: ( input.modelProfiles ?? [ { id: modelProfileId, - name: '默认模型', + name: "默认模型", baseUrl: input.modelBaseUrl, modelName: input.modelName, protocol: input.modelProtocol, authentication: input.modelAuthentication, - imageGenerationQuality: - input.imageGenerationQuality, - apiKey: input.apiKey - } + imageGenerationQuality: input.imageGenerationQuality, + apiKey: input.apiKey, + }, ] ).map(({ apiKey, ...profile }) => ({ ...profile, - apiKeyConfigured: apiKey.action === 'replace', + apiKeyConfigured: apiKey.action === "replace", credentialSource: - apiKey.action === 'replace' - ? ('encrypted' as const) - : ('none' as const) + apiKey.action === "replace" + ? ("encrypted" as const) + : ("none" as const), })), - defaultModelProfileId: - input.defaultModelProfileId ?? modelProfileId, - opencodeModelSource: - input.opencodeModelSource ?? { kind: 'platform' }, - continueModelSource: - input.continueModelSource ?? { kind: 'platform' }, + defaultModelProfileId: input.defaultModelProfileId ?? modelProfileId, + opencodeModelSource: input.opencodeModelSource ?? { kind: "platform" }, + continueModelSource: input.continueModelSource ?? { kind: "platform" }, secureStorageAvailable: true, - toolApproval: input.toolApproval - }) + toolApproval: input.toolApproval, + }), ), selectWorkspace: vi.fn(async () => undefined), - detectAgentRuntimes: vi.fn< - DesktopApi['settings']['detectAgentRuntimes'] - >(async () => ({ - opencode: { - available: false, - detail: '未检测到 OpenCode' - }, - continue: { - available: false, - detail: '未检测到 Continue' - }, - deepseekHarness: { - available: true, - path: 'bundled://deepseek-harness', - detail: 'Bundled Harness Adapter ready' - } - })), + detectAgentRuntimes: vi.fn( + async () => ({ + opencode: { + available: false, + detail: "未检测到 OpenCode", + }, + continue: { + available: false, + detail: "未检测到 Continue", + }, + deepseekHarness: { + available: true, + path: "bundled://deepseek-harness", + detail: "Bundled Harness Adapter ready", + }, + }), + ), selectRuntimeFile: vi.fn(async () => undefined), openRuntimeConfig: vi.fn(async () => {}), - testModelConnection: vi.fn< - DesktopApi['settings']['testModelConnection'] - >(async () => ({ - id: 'model', - label: 'sonnet-5', - available: true, - supportsToolExecution: true, - detail: 'Ready' - })), - testRuntime: vi.fn( + testModelConnection: vi.fn( async () => ({ - id: 'model', - label: 'sonnet-5', + id: "model", + label: "sonnet-5", available: true, supportsToolExecution: true, - detail: 'Ready' - }) - ) + detail: "Ready", + }), + ), + testRuntime: vi.fn(async () => ({ + id: "model", + label: "sonnet-5", + available: true, + supportsToolExecution: true, + detail: "Ready", + })), }, projects: { list: vi.fn(async () => [project]), create: vi.fn(async (input) => ({ ...project, ...input, - id: crypto.randomUUID() + id: crypto.randomUUID(), })), update: vi.fn(async (_projectId, input) => ({ ...project, ...input, - id: _projectId + id: _projectId, })), setArchived: vi.fn(async () => {}), delete: vi.fn(async () => {}), remote: { - save: vi.fn(), - cancelCurrent: - vi.fn(), + save: vi.fn(), + cancelCurrent: vi.fn(), onSaveProgress: vi.fn((listener) => { - remoteProjectSaveProgressListener = listener + remoteProjectSaveProgressListener = listener; return () => { if (remoteProjectSaveProgressListener === listener) { - remoteProjectSaveProgressListener = undefined + remoteProjectSaveProgressListener = undefined; } - } + }; }), getRecoverySnapshot: vi.fn(async () => ({ - recoveries: [] + recoveries: [], })), retryRecovery: vi.fn(), onRecoveryProgress: vi.fn((listener) => { - remoteProjectRecoveryProgressListener = listener - return removeProjectRecoveryListener - }) - } + remoteProjectRecoveryProgressListener = listener; + return removeProjectRecoveryListener; + }), + }, }, conversations: { list: vi.fn(async () => []), @@ -494,14 +466,14 @@ const api: DesktopApi = { id: crypto.randomUUID(), branch: { sourceConversationId: input.sourceConversationId, - sourceTitle: '新对话' + sourceTitle: "新对话", }, title: input.title, updatedAt: Date.now(), - messages: [] + messages: [], })), deleteLocal: vi.fn(async () => true), - onChanged: vi.fn(() => () => undefined) + onChanged: vi.fn(() => () => undefined), }, conversationQueue: { list: vi.fn(async () => []), @@ -509,69 +481,69 @@ const api: DesktopApi = { const item = { id: crypto.randomUUID(), conversationId: input.conversationId, - source: 'user' as const, + source: "user" as const, label: input.prompt, - createdAt: '2026-07-31T00:00:00.000Z' - } + createdAt: "2026-07-31T00:00:00.000Z", + }; queueMicrotask(() => - conversationQueueDispatchListener?.({ item, input }) - ) - return item + conversationQueueDispatchListener?.({ item, input }), + ); + return item; }), remove: vi.fn(async () => {}), interruptAndRun: vi.fn(async () => {}), releaseUser: vi.fn(async () => {}), ready: vi.fn(async () => {}), onChanged: vi.fn((listener) => { - conversationQueueChangeListener = listener + conversationQueueChangeListener = listener; return () => { - conversationQueueChangeListener = undefined - } + conversationQueueChangeListener = undefined; + }; }), onDispatch: vi.fn((listener) => { - conversationQueueDispatchListener = listener + conversationQueueDispatchListener = listener; return () => { - conversationQueueDispatchListener = undefined - } - }) + conversationQueueDispatchListener = undefined; + }; + }), }, workspace: { getFileDiff: vi.fn(), getChanges: vi.fn(async () => ({ - rootPath: 'C:\\Workspace', + rootPath: "C:\\Workspace", available: true, - status: '', - patch: '', + status: "", + patch: "", files: [], - truncated: false + truncated: false, })), listDirectory: vi.fn(async (path: string) => ({ path, entries: [], - truncated: false + truncated: false, })), readFile: vi.fn(async (path: string) => ({ path, - name: path.split('/').at(-1) ?? path, - content: '', - mimeType: 'text/plain' as const, + name: path.split("/").at(-1) ?? path, + content: "", + mimeType: "text/plain" as const, size: 0, offsetBytes: 0, nextOffsetBytes: 0, - truncated: false + truncated: false, })), - openPath: vi.fn(async () => {}) + openPath: vi.fn(async () => {}), }, tasks: { list: vi.fn(async () => []), - setStatus: vi.fn(async () => {}) + setStatus: vi.fn(async () => {}), }, activityHistory: { get: vi.fn(async () => ({ records: [], - legacyHistoryMayBeIncomplete: false + legacyHistoryMayBeIncomplete: false, })), - replace: vi.fn(async () => {}) + replace: vi.fn(async () => {}), }, usage: { getTokenSummary: vi.fn(async () => ({ @@ -581,17 +553,17 @@ const api: DesktopApi = { output: 0, cacheRead: 0, cacheWrite: 0, - totalTokens: 0 + totalTokens: 0, }, - records: [] - })) + records: [], + })), }, artifacts: { list: vi.fn(async () => []), get: vi.fn(async () => { - throw new Error('Artifact not found') + throw new Error("Artifact not found"); }), - importFiles: vi.fn(async () => []) + importFiles: vi.fn(async () => []), }, memory: { list: vi.fn(async () => []), @@ -600,12 +572,12 @@ const api: DesktopApi = { id: crypto.randomUUID(), confidence: 1, salience: 1, - status: 'confirmed' as const, - createdAt: '2026-07-31T00:00:00.000Z', - updatedAt: '2026-07-31T00:00:00.000Z' + status: "confirmed" as const, + createdAt: "2026-07-31T00:00:00.000Z", + updatedAt: "2026-07-31T00:00:00.000Z", })), setStatus: vi.fn(async () => {}), - remove: vi.fn(async () => {}) + remove: vi.fn(async () => {}), }, schedules: { list: vi.fn(async () => []), @@ -615,42 +587,42 @@ const api: DesktopApi = { taskId: crypto.randomUUID(), conversationId: input.conversationId ?? crypto.randomUUID(), enabled: true, - createdAt: '2026-07-31T00:00:00.000Z', - updatedAt: '2026-07-31T00:00:00.000Z' + createdAt: "2026-07-31T00:00:00.000Z", + updatedAt: "2026-07-31T00:00:00.000Z", })), setEnabled: vi.fn(async () => {}), remove: vi.fn(async () => {}), - runNow: vi.fn(async () => {}) + runNow: vi.fn(async () => {}), }, heartbeats: { list: vi.fn(async () => []), create: vi.fn(async (input) => ({ ...input, id: crypto.randomUUID(), - nextRunAt: '2026-08-01T09:00:00.000Z', - createdAt: '2026-08-01T00:00:00.000Z', - updatedAt: '2026-08-01T00:00:00.000Z' + nextRunAt: "2026-08-01T09:00:00.000Z", + createdAt: "2026-08-01T00:00:00.000Z", + updatedAt: "2026-08-01T00:00:00.000Z", })), update: vi.fn(async (heartbeatId, input) => ({ ...input, id: heartbeatId, - nextRunAt: '2026-08-01T09:00:00.000Z', - createdAt: '2026-08-01T00:00:00.000Z', - updatedAt: '2026-08-01T00:00:00.000Z' + nextRunAt: "2026-08-01T09:00:00.000Z", + createdAt: "2026-08-01T00:00:00.000Z", + updatedAt: "2026-08-01T00:00:00.000Z", })), setPaused: vi.fn(async () => {}), remove: vi.fn(async () => {}), runNow: vi.fn(async (heartbeatId) => ({ id: crypto.randomUUID(), configId: heartbeatId, - trigger: 'manual' as const, - scheduledFor: '2026-08-01T00:00:00.000Z', - status: 'completed' as const, + trigger: "manual" as const, + scheduledFor: "2026-08-01T00:00:00.000Z", + status: "completed" as const, attemptCount: 1, - createdAt: '2026-08-01T00:00:00.000Z', - updatedAt: '2026-08-01T00:00:00.000Z' + createdAt: "2026-08-01T00:00:00.000Z", + updatedAt: "2026-08-01T00:00:00.000Z", })), - history: vi.fn(async () => ({ runs: [], entries: [] })) + history: vi.fn(async () => ({ runs: [], entries: [] })), }, experts: { list: vi.fn(async () => []), @@ -658,87 +630,87 @@ const api: DesktopApi = { ...input, id: crypto.randomUUID(), enabled: true, - createdAt: '2026-07-31T00:00:00.000Z', - updatedAt: '2026-07-31T00:00:00.000Z' + createdAt: "2026-07-31T00:00:00.000Z", + updatedAt: "2026-07-31T00:00:00.000Z", })), update: vi.fn(async (expertId, input) => ({ ...input, id: expertId, enabled: true, - createdAt: '2026-07-31T00:00:00.000Z', - updatedAt: '2026-08-04T00:00:00.000Z' + createdAt: "2026-07-31T00:00:00.000Z", + updatedAt: "2026-08-04T00:00:00.000Z", })), - remove: vi.fn(async () => {}) + remove: vi.fn(async () => {}), }, capabilities: { getSnapshot: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), importSkill: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), removeSkill: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), setSkillEnabled: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), setSkillAssignments: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), setBuiltinMcpServerEnabled: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), setBuiltinMcpServerAssignments: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), saveMcpServer: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), removeMcpServer: vi.fn(async () => ({ skills: [], - mcpServers: [] + mcpServers: [], })), testMcpServer: vi.fn(async () => ({ dynamicToolsSupported: false, toolCount: 0, - tools: [] - })) + tools: [], + })), }, runtimeExtensions: { getSnapshot: vi.fn(async () => ({ marketplaceEnabled: false, catalog: [], - installed: [] + installed: [], })), apply: vi.fn(async () => ({ marketplaceEnabled: false, catalog: [], - installed: [] - })) + installed: [], + })), }, runtimeCustomization: { getSettings: vi.fn(async () => ({ opencode: {}, - continue: { presets: [] } + continue: { presets: [] }, })), updateSettings: vi.fn(async (settings) => settings), getNativeSnapshot: vi.fn(async (input) => ({ provider: input.provider, available: true, - inventoryStatus: 'available' as const, - detail: 'Ready', + inventoryStatus: "available" as const, + detail: "Ready", agents: [], tools: [], - toolsSupported: input.provider !== 'continue', + toolsSupported: input.provider !== "continue", commands: [], lsp: [], formatters: [], @@ -749,77 +721,77 @@ const api: DesktopApi = { resources: [], resourcesSupported: false, context: { - strategy: 'unsupported' as const, + strategy: "unsupported" as const, manualCompact: false, - detail: 'Unsupported' - } - })) + detail: "Unsupported", + }, + })), }, context: { selectFiles: vi.fn(async () => []), onFileSelectionProgress: vi.fn((listener) => { - fileSelectionProgressListener = listener + fileSelectionProgressListener = listener; return () => { - fileSelectionProgressListener = undefined - } + fileSelectionProgressListener = undefined; + }; }), addPastedImage: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), captureScreen: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), listWindows: vi.fn(async () => []), captureWindow: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), readClipboard: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), - remove: vi.fn(async () => {}) + remove: vi.fn(async () => {}), }, magicNotes: { list: vi.fn(async () => ({ notes: [] })), get: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), create: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), update: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), remove: vi.fn(async () => {}), createEntry: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), updateEntry: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), removeEntry: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), analyze: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), analyzeDraft: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), listTodos: vi.fn(async () => ({ todos: [] })), getTodoStatus: vi.fn(async () => ({ incompleteCount: 0 })), updateTodo: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), analyzeTodo: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), onAnalysisEvent: vi.fn(() => vi.fn()), onTodoStatusChanged: vi.fn((listener) => { - magicTodoStatusChangedListener = listener + magicTodoStatusChangedListener = listener; return () => { - magicTodoStatusChangedListener = undefined - } - }) + magicTodoStatusChangedListener = undefined; + }; + }), }, knowledge: { getSnapshot: vi.fn(async () => ({ @@ -828,14 +800,14 @@ const api: DesktopApi = { documents: [], graphNodes: [], graphRelations: [], - evidence: [] + evidence: [], })), createLibrary: vi.fn(async (input) => ({ ...input, id: crypto.randomUUID(), sourceCount: 0, documentCount: 0, - indexedDocumentCount: 0 + indexedDocumentCount: 0, })), updateLibrary: vi.fn(async () => {}), deleteLibrary: vi.fn(async () => {}), @@ -863,7 +835,7 @@ const api: DesktopApi = { contextMaxCharacters: 16_000, adjacentChunkCount: 0, localRerankEnabled: false, - rerankMode: 'none' + rerankMode: "none", }, diagnostics: { requestedChannels: [], @@ -875,28 +847,28 @@ const api: DesktopApi = { filteredByThresholdCount: 0, filteredByBudgetCount: 0, rerank: { - requested: 'none' as const, - used: 'none' as const, - status: 'skipped' as const, + requested: "none" as const, + used: "none" as const, + status: "skipped" as const, candidateCount: 0, - durationMs: 0 - } + durationMs: 0, + }, }, results: [], context: { characterCount: 0, truncated: false, - groups: [] - } + groups: [], + }, })), updateSettings: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), listChunks: vi.fn(async () => ({ items: [], page: 1, pageSize: 50, - totalItems: 0 + totalItems: 0, })), updateChunk: vi.fn(async () => {}), deleteChunk: vi.fn(async () => {}), @@ -906,31 +878,31 @@ const api: DesktopApi = { documents: [], graphNodes: [], graphRelations: [], - evidence: [] + evidence: [], })), openDocumentSource: vi.fn(async () => {}), rebuildLibrary: vi.fn(async () => ({ rebuilt: 0, - failed: 0 + failed: 0, })), cancelRebuild: vi.fn(async () => true), getEmbeddingIndex: vi.fn(async (knowledgeBaseId: string) => ({ knowledgeBaseId, enabled: false, coverage: { total: 0, indexed: 0, missing: 0, error: 0 }, - indexStatus: { job: null } + indexStatus: { job: null }, })), rebuildEmbeddingIndex: vi.fn(async (knowledgeBaseId: string) => ({ knowledgeBaseId, enabled: false, coverage: { total: 0, indexed: 0, missing: 0, error: 0 }, - indexStatus: { job: null } + indexStatus: { job: null }, })), cancelEmbeddingIndex: vi.fn(async () => false), cancelTask: vi.fn(async () => false), retryTask: vi.fn(async () => {}), getReferenceContext: vi.fn(async () => { - throw new Error('not used') + throw new Error("not used"); }), openReferenceSource: vi.fn(async () => {}), createEntity: vi.fn(async () => {}), @@ -940,753 +912,707 @@ const api: DesktopApi = { mergeEntities: vi.fn(async () => {}), createRelation: vi.fn(async () => {}), updateRelation: vi.fn(async () => {}), - deleteRelation: vi.fn(async () => {}) - } -} + deleteRelation: vi.fn(async () => {}), + }, +}; function composerMenuTrigger( - label: '专家角色' | '工作模式' + label: "专家角色" | "工作模式", ): HTMLButtonElement { - return screen.getByRole('button', { - name: new RegExp(`^${label}：`, 'u') - }) + return screen.getByRole("button", { + name: new RegExp(`^${label}：`, "u"), + }); } -function openComposerMenu( - label: '专家角色' | '工作模式' -): HTMLElement { - fireEvent.click(composerMenuTrigger(label)) - return screen.getByRole('menu', { name: label }) +function openComposerMenu(label: "专家角色" | "工作模式"): HTMLElement { + fireEvent.click(composerMenuTrigger(label)); + return screen.getByRole("menu", { name: label }); } function selectComposerOption( - label: '专家角色' | '工作模式', - optionLabel: string + label: "专家角色" | "工作模式", + optionLabel: string, ): void { - const menu = openComposerMenu(label) + const menu = openComposerMenu(label); const option = within(menu) - .getByText(optionLabel, { selector: 'span' }) - .closest('button') + .getByText(optionLabel, { selector: "span" }) + .closest("button"); if (!option) { - throw new Error(`Missing ${label} option: ${optionLabel}`) + throw new Error(`Missing ${label} option: ${optionLabel}`); } - fireEvent.click(option) + fireEvent.click(option); } function selectProjectOption(projectName: string): void { - fireEvent.click(screen.getByRole('button', { name: '当前项目' })) - const menu = screen.getByRole('menu', { name: '当前项目' }) + fireEvent.click(screen.getByRole("button", { name: "当前项目" })); + const menu = screen.getByRole("menu", { name: "当前项目" }); const option = within(menu) - .getByText(projectName, { selector: 'b' }) - .closest('[role="menuitemradio"]') + .getByText(projectName, { selector: "b" }) + .closest('[role="menuitemradio"]'); if (!option) { - throw new Error(`Missing project option: ${projectName}`) + throw new Error(`Missing project option: ${projectName}`); } - fireEvent.click(option) + fireEvent.click(option); } function deferred(): { - promise: Promise - resolve: (value: T) => void - reject: (reason: unknown) => void + promise: Promise; + resolve: (value: T) => void; + reject: (reason: unknown) => void; } { - let resolve!: (value: T) => void - let reject!: (reason: unknown) => void + let resolve!: (value: T) => void; + let reject!: (reason: unknown) => void; const promise = new Promise((resolvePromise, rejectPromise) => { - resolve = resolvePromise - reject = rejectPromise - }) - return { promise, resolve, reject } + resolve = resolvePromise; + reject = rejectPromise; + }); + return { promise, resolve, reject }; } function createBrowserState( conversationId: string, - overrides: Partial = {} + overrides: Partial = {}, ): BrowserLiveState { return { conversationId, - status: 'ready', + status: "ready", sessionActive: true, isLoading: false, canGoBack: false, updatedAt: Date.now(), - ...overrides - } + ...overrides, + }; } function installRemoteProjectsSetting(enabled: boolean): { - updateSettings: NonNullable['updateSettings'] + updateSettings: NonNullable["updateSettings"]; } { let settings: ApplicationSettings = { checkUpdatesOnStartup: false, - updateSource: 'github', - modelDownloadSource: 'modelscope', + updateSource: "github", + modelDownloadSource: "modelscope", localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: enabled, magicNotesEnabled: false, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate', - magicNoteCommentFormat: 'combined' - } + magicNoteCommentMode: "immediate", + magicNoteCommentFormat: "combined", + }; const updateSettings = vi.fn< - NonNullable['updateSettings'] + NonNullable["updateSettings"] >(async (input) => { - settings = { ...settings, ...input } - return { ...settings } - }) + settings = { ...settings, ...input }; + return { ...settings }; + }); api.updates = { getSettings: vi.fn(async () => ({ ...settings })), updateSettings, check: vi.fn(), openReleasePage: vi.fn(), - onResult: vi.fn(() => () => {}) - } - return { updateSettings } + onResult: vi.fn(() => () => {}), + }; + return { updateSettings }; } -describe('App', () => { +describe("App", () => { beforeEach(() => { - localStorage.clear() - delete document.documentElement.dataset.theme - document.documentElement.style.colorScheme = '' - vi.clearAllMocks() - magicTodoStatusChangedListener = undefined + localStorage.clear(); + delete document.documentElement.dataset.theme; + document.documentElement.style.colorScheme = ""; + vi.clearAllMocks(); + magicTodoStatusChangedListener = undefined; vi.mocked(api.magicNotes.getTodoStatus) .mockReset() - .mockResolvedValue({ incompleteCount: 0 }) + .mockResolvedValue({ incompleteCount: 0 }); vi.mocked(api.magicNotes.onTodoStatusChanged) .mockReset() .mockImplementation((listener) => { - magicTodoStatusChangedListener = listener + magicTodoStatusChangedListener = listener; return () => { - magicTodoStatusChangedListener = undefined - } - }) - vi.mocked(api.conversations.list).mockReset().mockResolvedValue([]) - vi.mocked(api.conversations.replace) - .mockReset() - .mockResolvedValue() - vi.mocked(api.conversations.saveLocal) - .mockReset() - .mockResolvedValue() + magicTodoStatusChangedListener = undefined; + }; + }); + vi.mocked(api.conversations.list).mockReset().mockResolvedValue([]); + vi.mocked(api.conversations.replace).mockReset().mockResolvedValue(); + vi.mocked(api.conversations.saveLocal).mockReset().mockResolvedValue(); vi.mocked(api.conversations.branchLocal) .mockReset() .mockImplementation(async (input) => ({ id: crypto.randomUUID(), branch: { sourceConversationId: input.sourceConversationId, - sourceTitle: '新对话' + sourceTitle: "新对话", }, title: input.title, updatedAt: Date.now(), - messages: [] - })) + messages: [], + })); vi.mocked(api.conversations.deleteLocal) .mockReset() - .mockResolvedValue(true) + .mockResolvedValue(true); vi.mocked(api.conversations.onChanged) .mockReset() - .mockReturnValue(() => undefined) + .mockReturnValue(() => undefined); vi.mocked(api.projects.remote.cancelCurrent) .mockReset() - .mockResolvedValue() - remoteProjectSaveProgressListener = undefined + .mockResolvedValue(); + remoteProjectSaveProgressListener = undefined; vi.mocked(api.projects.remote.onSaveProgress) .mockReset() .mockImplementation((listener) => { - remoteProjectSaveProgressListener = listener + remoteProjectSaveProgressListener = listener; return () => { if (remoteProjectSaveProgressListener === listener) { - remoteProjectSaveProgressListener = undefined + remoteProjectSaveProgressListener = undefined; } - } - }) - remoteProjectRecoveryProgressListener = undefined - removeProjectRecoveryListener.mockReset() + }; + }); + remoteProjectRecoveryProgressListener = undefined; + removeProjectRecoveryListener.mockReset(); vi.mocked(api.projects.remote.getRecoverySnapshot) .mockReset() - .mockResolvedValue({ recoveries: [] }) - vi.mocked(api.projects.remote.retryRecovery).mockReset() + .mockResolvedValue({ recoveries: [] }); + vi.mocked(api.projects.remote.retryRecovery).mockReset(); vi.mocked(api.projects.remote.onRecoveryProgress) .mockReset() .mockImplementation((listener) => { - remoteProjectRecoveryProgressListener = listener - return removeProjectRecoveryListener - }) - conversationQueueChangeListener = undefined - conversationQueueDispatchListener = undefined - vi.mocked(api.conversationQueue.list) - .mockReset() - .mockResolvedValue([]) + remoteProjectRecoveryProgressListener = listener; + return removeProjectRecoveryListener; + }); + conversationQueueChangeListener = undefined; + conversationQueueDispatchListener = undefined; + vi.mocked(api.conversationQueue.list).mockReset().mockResolvedValue([]); vi.mocked(api.conversationQueue.enqueueUser) .mockReset() .mockImplementation(async (input) => { const item = { id: crypto.randomUUID(), conversationId: input.conversationId, - source: 'user' as const, + source: "user" as const, label: input.prompt, - createdAt: '2026-07-31T00:00:00.000Z' - } + createdAt: "2026-07-31T00:00:00.000Z", + }; queueMicrotask(() => - conversationQueueDispatchListener?.({ item, input }) - ) - return item - }) - vi.mocked(api.conversationQueue.remove) - .mockReset() - .mockResolvedValue() + conversationQueueDispatchListener?.({ item, input }), + ); + return item; + }); + vi.mocked(api.conversationQueue.remove).mockReset().mockResolvedValue(); vi.mocked(api.conversationQueue.interruptAndRun) .mockReset() - .mockResolvedValue() + .mockResolvedValue(); vi.mocked(api.conversationQueue.releaseUser) .mockReset() - .mockResolvedValue() - vi.mocked(api.conversationQueue.ready) - .mockReset() - .mockResolvedValue() + .mockResolvedValue(); + vi.mocked(api.conversationQueue.ready).mockReset().mockResolvedValue(); vi.mocked(api.conversationQueue.onChanged) .mockReset() .mockImplementation((listener) => { - conversationQueueChangeListener = listener + conversationQueueChangeListener = listener; return () => { - conversationQueueChangeListener = undefined - } - }) + conversationQueueChangeListener = undefined; + }; + }); vi.mocked(api.conversationQueue.onDispatch) .mockReset() .mockImplementation((listener) => { - conversationQueueDispatchListener = listener + conversationQueueDispatchListener = listener; return () => { - conversationQueueDispatchListener = undefined - } - }) - vi.mocked(api.tasks.list).mockReset().mockResolvedValue([]) - vi.mocked(api.activityHistory.get) - .mockReset() - .mockResolvedValue({ - records: [], - legacyHistoryMayBeIncomplete: false - }) - vi.mocked(api.activityHistory.replace) - .mockReset() - .mockResolvedValue() - vi.mocked(api.schedules.list).mockReset().mockResolvedValue([]) - api.channels = undefined - api.updates = undefined - newConversationListener = undefined - beforeQuitListener = undefined - browserListener = undefined - fileSelectionProgressListener = undefined - maximizedChangedListener = undefined + conversationQueueDispatchListener = undefined; + }; + }); + vi.mocked(api.tasks.list).mockReset().mockResolvedValue([]); + vi.mocked(api.activityHistory.get).mockReset().mockResolvedValue({ + records: [], + legacyHistoryMayBeIncomplete: false, + }); + vi.mocked(api.activityHistory.replace).mockReset().mockResolvedValue(); + vi.mocked(api.schedules.list).mockReset().mockResolvedValue([]); + api.channels = undefined; + api.updates = undefined; + newConversationListener = undefined; + beforeQuitListener = undefined; + browserListener = undefined; + fileSelectionProgressListener = undefined; + maximizedChangedListener = undefined; speechRecognitionMocks.startPcmRecording.mockResolvedValue({ result: Promise.resolve({ audio: new Float32Array([0, 0.25, -0.25]).buffer, - sampleRate: 16_000 + sampleRate: 16_000, }), stop: vi.fn(), - cancel: vi.fn() - }) - Object.defineProperty(navigator, 'mediaDevices', { + cancel: vi.fn(), + }); + Object.defineProperty(navigator, "mediaDevices", { configurable: true, - value: { getUserMedia: vi.fn() } - }) - Object.defineProperty(window, 'AudioContext', { + value: { getUserMedia: vi.fn() }, + }); + Object.defineProperty(window, "AudioContext", { configurable: true, - value: class AudioContextMock {} - }) + value: class AudioContextMock {}, + }); vi.mocked(api.speech!.transcribe).mockResolvedValue({ - text: '本地语音结果' - }) - vi.mocked(api.speech!.cancel).mockResolvedValue(true) + text: "本地语音结果", + }); + vi.mocked(api.speech!.cancel).mockResolvedValue(true); vi.mocked(api.agent.getStatus).mockResolvedValue({ - id: 'model', - label: 'sonnet-5', + id: "model", + label: "sonnet-5", available: true, supportsToolExecution: true, - detail: 'Ready' - }) + detail: "Ready", + }); vi.stubGlobal( - 'requestIdleCallback', - vi.fn(() => 1) - ) - vi.stubGlobal('cancelIdleCallback', vi.fn()) - Object.defineProperty(window, 'goodbuddy', { + "requestIdleCallback", + vi.fn(() => 1), + ); + vi.stubGlobal("cancelIdleCallback", vi.fn()); + Object.defineProperty(window, "goodbuddy", { configurable: true, - value: api - }) - }) + value: api, + }); + }); afterEach(() => { - cleanup() - vi.unstubAllGlobals() - vi.restoreAllMocks() - }) + cleanup(); + vi.unstubAllGlobals(); + vi.restoreAllMocks(); + }); - it('provides custom minimize, maximize, and close controls', async () => { - const { unmount } = render() + it("provides custom minimize, maximize, and close controls", async () => { + const { unmount } = render(); - fireEvent.click(screen.getByLabelText('最小化窗口')) - fireEvent.click(screen.getByLabelText('最大化窗口')) - fireEvent.click(screen.getByLabelText('关闭窗口')) + fireEvent.click(screen.getByLabelText("最小化窗口")); + fireEvent.click(screen.getByLabelText("最大化窗口")); + fireEvent.click(screen.getByLabelText("关闭窗口")); await waitFor(() => { - expect(api.app.minimize).toHaveBeenCalledOnce() - expect(api.app.toggleMaximize).toHaveBeenCalledOnce() - expect(api.app.close).toHaveBeenCalledOnce() - }) - act(() => maximizedChangedListener?.(true)) - expect(await screen.findByLabelText('还原窗口')).toBeInTheDocument() - act(() => maximizedChangedListener?.(false)) - expect(screen.getByLabelText('最大化窗口')).toBeInTheDocument() + expect(api.app.minimize).toHaveBeenCalledOnce(); + expect(api.app.toggleMaximize).toHaveBeenCalledOnce(); + expect(api.app.close).toHaveBeenCalledOnce(); + }); + act(() => maximizedChangedListener?.(true)); + expect(await screen.findByLabelText("还原窗口")).toBeInTheDocument(); + act(() => maximizedChangedListener?.(false)); + expect(screen.getByLabelText("最大化窗口")).toBeInTheDocument(); - unmount() - expect(removeMaximizedChangedListener).toHaveBeenCalledOnce() - }) + unmount(); + expect(removeMaximizedChangedListener).toHaveBeenCalledOnce(); + }); - it('renders the core app shell in English', async () => { - await changeUiLocale('en-US') + it("renders the core app shell in English", async () => { + await changeUiLocale("en-US"); try { - render() + render(); expect( - await screen.findByText('New conversation', { - selector: '.new-chat span' - }) - ).toBeInTheDocument() + await screen.findByText("New conversation", { + selector: ".new-chat span", + }), + ).toBeInTheDocument(); expect( - screen.getByRole('navigation', { - name: 'Main navigation' - }) - ).toBeInTheDocument() + screen.getByRole("navigation", { + name: "Main navigation", + }), + ).toBeInTheDocument(); + expect(screen.getByRole("button", { name: "Chat" })).toBeInTheDocument(); + expect(screen.queryByText("Desktop workspace")).not.toBeInTheDocument(); + expect(screen.queryByText("GOODBUDDY WORKSPACE")).not.toBeInTheDocument(); expect( - screen.getByRole('button', { name: 'Chat' }) - ).toBeInTheDocument() - expect(screen.queryByText('Desktop workspace')).not.toBeInTheDocument() - expect(screen.queryByText('GOODBUDDY WORKSPACE')).not.toBeInTheDocument() - expect( - screen.getByRole('heading', { + screen.getByRole("heading", { level: 1, - name: 'Conversation' - }) - ).toBeInTheDocument() + name: "Conversation", + }), + ).toBeInTheDocument(); expect( - screen.getByRole('heading', { + screen.getByRole("heading", { level: 2, - name: 'What would you like to accomplish today?' - }) - ).toBeInTheDocument() + name: "What would you like to accomplish today?", + }), + ).toBeInTheDocument(); expect( - screen.getByRole('button', { name: 'Current project' }) - ).toHaveTextContent('Default project') - expect(screen.getByText('Project: Default project')).toHaveClass( - 'scope-badge' - ) + screen.getByRole("button", { name: "Current project" }), + ).toHaveTextContent("Default project"); + expect(screen.getByText("Project: Default project")).toHaveClass( + "scope-badge", + ); expect( screen.getByText( - /Hi, I’m GoodBuddy\. Ask me a question, add local files/u - ) - ).toBeInTheDocument() - expect( - screen.getByLabelText('Message GoodBuddy') - ).toHaveAttribute( - 'placeholder', - 'Message GoodBuddy…\nEnter to send, Shift+Enter for a new line, Ctrl+V to paste an image or text' - ) - expect( - screen.getByLabelText('Message GoodBuddy') - ).toHaveAttribute( - 'title', - 'Enter to send, Shift+Enter for a new line, Ctrl+V to paste an image or text' - ) + /Hi, I’m GoodBuddy\. Ask me a question, add local files/u, + ), + ).toBeInTheDocument(); + expect(screen.getByLabelText("Message GoodBuddy")).toHaveAttribute( + "placeholder", + "Message GoodBuddy…\nEnter to send, Shift+Enter for a new line, Ctrl+V to paste an image or text", + ); + expect(screen.getByLabelText("Message GoodBuddy")).toHaveAttribute( + "title", + "Enter to send, Shift+Enter for a new line, Ctrl+V to paste an image or text", + ); } finally { - cleanup() - await changeUiLocale('zh-CN') + cleanup(); + await changeUiLocale("zh-CN"); } - }) + }); - it('localizes untouched project labels but keeps settings and destructive values raw', async () => { + it("localizes untouched project labels but keeps settings and destructive values raw", async () => { const secondProject = { ...project, - id: '00000000-0000-4000-8000-000000000102', - name: 'Second project', - description: 'Another workspace', - rootPath: 'C:\\Second', + id: "00000000-0000-4000-8000-000000000102", + name: "Second project", + description: "Another workspace", + rootPath: "C:\\Second", executionSpace: { - kind: 'local' as const, - rootPath: 'C:\\Second' - } - } + kind: "local" as const, + rootPath: "C:\\Second", + }, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - secondProject - ]) + secondProject, + ]); vi.mocked(api.tasks.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000901', + id: "00000000-0000-4000-8000-000000000901", projectId, - title: 'Seed project task', - instructions: 'Verify the project label', - origin: 'schedule', - status: 'queued', - workMode: 'ask', - createdAt: '2026-07-31T01:00:00.000Z' - } - ]) - await changeUiLocale('en-US') + title: "Seed project task", + instructions: "Verify the project label", + origin: "schedule", + status: "queued", + workMode: "ask", + createdAt: "2026-07-31T01:00:00.000Z", + }, + ]); + await changeUiLocale("en-US"); try { - render() + render(); - const projectTrigger = await screen.findByRole('button', { - name: 'Current project' - }) - expect(projectTrigger).toHaveTextContent('Default project') - fireEvent.click(projectTrigger) + const projectTrigger = await screen.findByRole("button", { + name: "Current project", + }); + expect(projectTrigger).toHaveTextContent("Default project"); + fireEvent.click(projectTrigger); expect( - within( - screen.getByRole('menu', { name: 'Current project' }) - ).getByText('Default project', { selector: 'b' }) - ).toBeInTheDocument() - fireEvent.click(projectTrigger) + within(screen.getByRole("menu", { name: "Current project" })).getByText( + "Default project", + { selector: "b" }, + ), + ).toBeInTheDocument(); + fireEvent.click(projectTrigger); + + fireEvent.click(screen.getByLabelText("Toggle assistant workspace")); + fireEvent.click(await screen.findByRole("tab", { name: "Task center" })); + const assistantSidebar = screen.getByLabelText("Assistant workspace"); + expect( + within(assistantSidebar).getByText("Project: Default project"), + ).toBeInTheDocument(); + const taskIndexHeading = within(assistantSidebar).getByRole("heading", { + name: "Task index", + }); + const newTaskButton = within(assistantSidebar).getByRole("button", { + name: "New task", + }); + expect(taskIndexHeading.parentElement).toContainElement(newTaskButton); + fireEvent.click(newTaskButton); + const taskDialog = screen.getByRole("dialog", { + name: "New custom task", + }); + expect( + within(taskDialog).getByText("Default project"), + ).toBeInTheDocument(); + expect( + within(taskDialog).getByRole("button", { + name: "Current conversation", + }), + ).toHaveAttribute("aria-pressed", "true"); + fireEvent.click( + within(taskDialog).getByRole("button", { + name: "Close new custom task", + }), + ); + fireEvent.click(screen.getByLabelText("Toggle assistant workspace")); + + fireEvent.click(screen.getByLabelText("Project settings")); + const settingsDialog = screen.getByRole("dialog", { + name: "Project settings", + }); + expect(within(settingsDialog).getByLabelText("Name")).toHaveValue( + builtInDefaultProjectSeedName, + ); + expect(within(settingsDialog).getByLabelText("Description")).toHaveValue( + builtInDefaultProjectSeedDescription, + ); fireEvent.click( - screen.getByLabelText('Toggle assistant workspace') - ) - fireEvent.click( - await screen.findByRole('tab', { name: 'Task center' }) - ) - const assistantSidebar = screen.getByLabelText( - 'Assistant workspace' - ) - expect( - within(assistantSidebar).getByText('Project: Default project') - ).toBeInTheDocument() - const taskIndexHeading = within(assistantSidebar).getByRole( - 'heading', - { name: 'Task index' } - ) - const newTaskButton = within(assistantSidebar).getByRole( - 'button', - { name: 'New task' } - ) - expect(taskIndexHeading.parentElement).toContainElement( - newTaskButton - ) - fireEvent.click(newTaskButton) - const taskDialog = screen.getByRole('dialog', { - name: 'New custom task' - }) - expect( - within(taskDialog).getByText('Default project') - ).toBeInTheDocument() - expect( - within(taskDialog).getByRole('button', { - name: 'Current conversation' - }) - ).toHaveAttribute('aria-pressed', 'true') - fireEvent.click( - within(taskDialog).getByRole('button', { - name: 'Close new custom task' - }) - ) - fireEvent.click( - screen.getByLabelText('Toggle assistant workspace') - ) - - fireEvent.click(screen.getByLabelText('Project settings')) - const settingsDialog = screen.getByRole('dialog', { - name: 'Project settings' - }) - expect( - within(settingsDialog).getByLabelText('Name') - ).toHaveValue(builtInDefaultProjectSeedName) - expect( - within(settingsDialog).getByLabelText('Description') - ).toHaveValue(builtInDefaultProjectSeedDescription) - - fireEvent.click( - within(settingsDialog).getByRole('button', { - name: 'Delete project' - }) - ) + within(settingsDialog).getByRole("button", { + name: "Delete project", + }), + ); const confirmation = within(settingsDialog).getByLabelText( - `Enter “${builtInDefaultProjectSeedName}” to confirm deletion` - ) - const deleteButton = within(settingsDialog).getByRole('button', { - name: 'Permanently delete project' - }) + `Enter “${builtInDefaultProjectSeedName}” to confirm deletion`, + ); + const deleteButton = within(settingsDialog).getByRole("button", { + name: "Permanently delete project", + }); fireEvent.change(confirmation, { - target: { value: 'Default project' } - }) - expect(deleteButton).toBeDisabled() + target: { value: "Default project" }, + }); + expect(deleteButton).toBeDisabled(); fireEvent.change(confirmation, { - target: { value: builtInDefaultProjectSeedName } - }) - expect(deleteButton).toBeEnabled() + target: { value: builtInDefaultProjectSeedName }, + }); + expect(deleteButton).toBeEnabled(); fireEvent.click( - within(settingsDialog).getByRole('button', { - name: 'Cancel deletion' - }) - ) + within(settingsDialog).getByRole("button", { + name: "Cancel deletion", + }), + ); fireEvent.click( - within(settingsDialog).getByRole('button', { - name: 'Save project' - }) - ) + within(settingsDialog).getByRole("button", { + name: "Save project", + }), + ); await waitFor(() => expect(api.projects.update).toHaveBeenCalledWith( projectId, expect.objectContaining({ name: builtInDefaultProjectSeedName, - description: builtInDefaultProjectSeedDescription - }) - ) - ) + description: builtInDefaultProjectSeedDescription, + }), + ), + ); expect( - screen.getByRole('button', { name: 'Current project' }) - ).toHaveTextContent('Default project') - expect(api.projects.delete).not.toHaveBeenCalled() + screen.getByRole("button", { name: "Current project" }), + ).toHaveTextContent("Default project"); + expect(api.projects.delete).not.toHaveBeenCalled(); } finally { - cleanup() - await changeUiLocale('zh-CN') + cleanup(); + await changeUiLocale("zh-CN"); } - }) + }); - it('renders localized workspace branding in Chinese', async () => { - render() + it("renders localized workspace branding in Chinese", async () => { + render(); - await screen.findByRole('heading', { - name: '今天想一起完成什么？' - }) - const runtimeStatus = document.querySelector( - '.runtime-status' - ) - expect(runtimeStatus).not.toBeNull() + await screen.findByRole("heading", { + name: "今天想一起完成什么？", + }); + const runtimeStatus = + document.querySelector(".runtime-status"); + expect(runtimeStatus).not.toBeNull(); + expect(within(runtimeStatus!).queryByText("就绪")).not.toBeInTheDocument(); + expect(runtimeStatus).toHaveClass("runtime-status--ready"); expect( - within(runtimeStatus!).queryByText('就绪') - ).not.toBeInTheDocument() - expect(runtimeStatus).toHaveClass('runtime-status--ready') - expect( - runtimeStatus?.querySelector('.runtime-status__dot') - ).toBeInTheDocument() + runtimeStatus?.querySelector(".runtime-status__dot"), + ).toBeInTheDocument(); expect(runtimeStatus).toHaveAttribute( - 'aria-describedby', - 'topbar-runtime-detail' - ) - expect(screen.getByText('Ready', { selector: '.sr-only' })) - .toHaveAttribute('id', 'topbar-runtime-detail') + "aria-describedby", + "topbar-runtime-detail", + ); + expect(screen.getByText("Ready", { selector: ".sr-only" })).toHaveAttribute( + "id", + "topbar-runtime-detail", + ); expect( screen.getByText( - /你好，我是 GoodBuddy。你可以直接向我提问、添加本地文件/u - ) - ).toBeInTheDocument() - expect(screen.getByText('智能工作台')).toBeInTheDocument() - expect(screen.queryByText('GOODBUDDY 工作台')).not.toBeInTheDocument() + /你好，我是 GoodBuddy。你可以直接向我提问、添加本地文件/u, + ), + ).toBeInTheDocument(); + expect(screen.getByText("智能工作台")).toBeInTheDocument(); + expect(screen.queryByText("GOODBUDDY 工作台")).not.toBeInTheDocument(); expect( - document.querySelector('.composer__runtime-toolbar') - ).not.toBeInTheDocument() - }) + document.querySelector(".composer__runtime-toolbar"), + ).not.toBeInTheDocument(); + }); - it('uses a compact color indicator while connecting to a Runtime', () => { - render() - const connectingStatus = document.querySelector( - '.runtime-status' - ) - expect(connectingStatus).toHaveClass('runtime-status--connecting') - expect(within(connectingStatus!).queryByText('连接中')) - .not.toBeInTheDocument() - }) + it("uses a compact color indicator while connecting to a Runtime", () => { + render(); + const connectingStatus = + document.querySelector(".runtime-status"); + expect(connectingStatus).toHaveClass("runtime-status--connecting"); + expect( + within(connectingStatus!).queryByText("连接中"), + ).not.toBeInTheDocument(); + }); - it('loads customized branding into the primary sidebar', async () => { + it("loads customized branding into the primary sidebar", async () => { expect( saveBrandingPreferences({ - name: '研发伙伴', - subtitleZhCN: '团队工作台', - subtitleEnUS: 'Team Workspace', - logoDataUrl: 'data:image/png;base64,iVBORw0KGgo=' - }) - ).toBe(true) + name: "研发伙伴", + subtitleZhCN: "团队工作台", + subtitleEnUS: "Team Workspace", + logoDataUrl: "data:image/png;base64,iVBORw0KGgo=", + }), + ).toBe(true); - render() + render(); - await screen.findByRole('heading', { - name: '今天想一起完成什么？' - }) - const brand = document.querySelector('.brand') - expect(brand).not.toBeNull() - expect(within(brand as HTMLElement).getByText('研发伙伴')) - .toBeInTheDocument() - expect(within(brand as HTMLElement).getByText('团队工作台')) - .toBeInTheDocument() - expect(brand?.querySelector('img')).toHaveAttribute( - 'src', - 'data:image/png;base64,iVBORw0KGgo=' - ) - }) + await screen.findByRole("heading", { + name: "今天想一起完成什么？", + }); + const brand = document.querySelector(".brand"); + expect(brand).not.toBeNull(); + expect( + within(brand as HTMLElement).getByText("研发伙伴"), + ).toBeInTheDocument(); + expect( + within(brand as HTMLElement).getByText("团队工作台"), + ).toBeInTheDocument(); + expect(brand?.querySelector("img")).toHaveAttribute( + "src", + "data:image/png;base64,iVBORw0KGgo=", + ); + }); - it('renders the unified user and Scheduled Task queue above the Composer', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000951' + it("renders the unified user and Scheduled Task queue above the Composer", async () => { + const conversationId = "00000000-0000-4000-8000-000000000951"; vi.mocked(api.conversations.list).mockResolvedValue([ { id: conversationId, projectId, - title: '排队对话', + title: "排队对话", updatedAt: Date.now(), - messages: [] - } - ]) + messages: [], + }, + ]); vi.mocked(api.conversationQueue.list).mockResolvedValue([ { - id: '00000000-0000-4000-8000-000000000952', + id: "00000000-0000-4000-8000-000000000952", conversationId, - source: 'schedule', - label: '每日汇总', - createdAt: '2026-08-20T09:00:00.000Z' + source: "schedule", + label: "每日汇总", + createdAt: "2026-08-20T09:00:00.000Z", }, { - id: '00000000-0000-4000-8000-000000000953', + id: "00000000-0000-4000-8000-000000000953", conversationId, - source: 'user', - label: '补充说明', - createdAt: '2026-08-20T09:01:00.000Z' - } - ]) + source: "user", + label: "补充说明", + createdAt: "2026-08-20T09:01:00.000Z", + }, + ]); - render() + render(); - const queue = await screen.findByRole('region', { - name: '对话待发送队列' - }) - expect(queue.parentElement).toHaveClass('composer-wrap') - expect(queue.closest('.composer')).toBeNull() - expect(queue.nextElementSibling).toHaveClass('composer') - expect(within(queue).queryByText('待发送（2）')).not.toBeInTheDocument() - expect(within(queue).getByText('每日汇总')).toBeInTheDocument() - expect(within(queue).getByText('补充说明')).toBeInTheDocument() + const queue = await screen.findByRole("region", { + name: "对话待发送队列", + }); + expect(queue.parentElement).toHaveClass("composer-wrap"); + expect(queue.closest(".composer")).toBeNull(); + expect(queue.nextElementSibling).toHaveClass("composer"); + expect(within(queue).queryByText("待发送（2）")).not.toBeInTheDocument(); + expect(within(queue).getByText("每日汇总")).toBeInTheDocument(); + expect(within(queue).getByText("补充说明")).toBeInTheDocument(); fireEvent.click( - within(queue).getByRole('button', { - name: '立即运行“每日汇总”' - }) - ) + within(queue).getByRole("button", { + name: "立即运行“每日汇总”", + }), + ); await waitFor(() => - expect( - api.conversationQueue.interruptAndRun - ).toHaveBeenCalledWith( - '00000000-0000-4000-8000-000000000952' - ) - ) + expect(api.conversationQueue.interruptAndRun).toHaveBeenCalledWith( + "00000000-0000-4000-8000-000000000952", + ), + ); fireEvent.click( - within(queue).getByRole('button', { - name: '从待发送队列删除“补充说明”' - }) - ) + within(queue).getByRole("button", { + name: "从待发送队列删除“补充说明”", + }), + ); await waitFor(() => expect(api.conversationQueue.remove).toHaveBeenCalledWith( - '00000000-0000-4000-8000-000000000953' - ) - ) - }) + "00000000-0000-4000-8000-000000000953", + ), + ); + }); - it('preserves scoped queue updates that arrive during another refresh', async () => { - const firstConversationId = - '00000000-0000-4000-8000-000000000954' - const secondConversationId = - '00000000-0000-4000-8000-000000000955' + it("preserves scoped queue updates that arrive during another refresh", async () => { + const firstConversationId = "00000000-0000-4000-8000-000000000954"; + const secondConversationId = "00000000-0000-4000-8000-000000000955"; vi.mocked(api.conversations.list).mockResolvedValue([ { id: firstConversationId, projectId, - title: '第一条排队对话', + title: "第一条排队对话", updatedAt: Date.now(), - messages: [] + messages: [], }, { id: secondConversationId, projectId, - title: '第二条排队对话', + title: "第二条排队对话", updatedAt: Date.now() - 1, - messages: [] - } - ]) - const firstRefresh = deferred< - Awaited< - ReturnType - > - >() + messages: [], + }, + ]); + const firstRefresh = + deferred>>(); vi.mocked(api.conversationQueue.list).mockImplementation( async (conversationId) => { if (conversationId === firstConversationId) { - return firstRefresh.promise + return firstRefresh.promise; } - return [] - } - ) + return []; + }, + ); - render() - await waitFor(() => - expect(conversationQueueChangeListener).toBeDefined() - ) + render(); + await waitFor(() => expect(conversationQueueChangeListener).toBeDefined()); act(() => { - conversationQueueChangeListener?.(firstConversationId) - }) + conversationQueueChangeListener?.(firstConversationId); + }); await waitFor(() => expect(api.conversationQueue.list).toHaveBeenCalledWith( - firstConversationId - ) - ) + firstConversationId, + ), + ); await act(async () => { - conversationQueueChangeListener?.(secondConversationId) - await new Promise((resolve) => window.setTimeout(resolve, 0)) - }) + conversationQueueChangeListener?.(secondConversationId); + await new Promise((resolve) => window.setTimeout(resolve, 0)); + }); act(() => { firstRefresh.resolve([ { - id: '00000000-0000-4000-8000-000000000956', + id: "00000000-0000-4000-8000-000000000956", conversationId: firstConversationId, - source: 'user', - label: '不能被后续刷新丢弃', - createdAt: '2026-08-20T09:03:00.000Z' - } - ]) - }) + source: "user", + label: "不能被后续刷新丢弃", + createdAt: "2026-08-20T09:03:00.000Z", + }, + ]); + }); - expect( - await screen.findByText('不能被后续刷新丢弃') - ).toBeInTheDocument() + expect(await screen.findByText("不能被后续刷新丢弃")).toBeInTheDocument(); await waitFor(() => expect(api.conversationQueue.list).toHaveBeenCalledWith( - secondConversationId - ) - ) - }) + secondConversationId, + ), + ); + }); - it('discovers product Tasks through their Conversation without exposing Runs', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000821' - const taskId = '00000000-0000-4000-8000-000000000822' - const scheduleId = - '00000000-0000-4000-8000-000000000823' + it("discovers product Tasks through their Conversation without exposing Runs", async () => { + const conversationId = "00000000-0000-4000-8000-000000000821"; + const taskId = "00000000-0000-4000-8000-000000000822"; + const scheduleId = "00000000-0000-4000-8000-000000000823"; const conversation: ConversationSnapshot = { id: conversationId, projectId, - title: '产品发布讨论', + title: "产品发布讨论", updatedAt: Date.now(), - messages: [] - } + messages: [], + }; const task: AssistantTask = { id: taskId, projectId, conversationId, scheduleId, - title: '每周发布总结', - instructions: '总结本周发布进度', - origin: 'schedule', - status: 'idle', - createdAt: '2026-08-19T00:00:00.000Z' - } + title: "每周发布总结", + instructions: "总结本周发布进度", + origin: "schedule", + status: "idle", + createdAt: "2026-08-19T00:00:00.000Z", + }; const schedule: AssistantSchedule = { id: scheduleId, projectId, @@ -1694,96 +1620,95 @@ describe('App', () => { conversationId, title: task.title, prompt: task.instructions, - workMode: 'execute', - recurrence: 'weekly', - nextRunAt: '2026-08-21T09:00:00.000Z', + workMode: "execute", + recurrence: "weekly", + nextRunAt: "2026-08-21T09:00:00.000Z", enabled: true, createdAt: task.createdAt, - updatedAt: task.createdAt - } - vi.mocked(api.conversations.list).mockResolvedValue([conversation]) - vi.mocked(api.tasks.list).mockResolvedValue([task]) - vi.mocked(api.schedules.list).mockResolvedValue([schedule]) + updatedAt: task.createdAt, + }; + vi.mocked(api.conversations.list).mockResolvedValue([conversation]); + vi.mocked(api.tasks.list).mockResolvedValue([task]); + vi.mocked(api.schedules.list).mockResolvedValue([schedule]); - render() + render(); const toggle = await screen.findByLabelText( - '展开或折叠“产品发布讨论”中的 1 个任务' - ) + "展开或折叠“产品发布讨论”中的 1 个任务", + ); const conversationButton = - toggle.parentElement?.querySelector('.conversation-item') - expect(toggle.nextElementSibling).toBe(conversationButton) + toggle.parentElement?.querySelector(".conversation-item"); + expect(toggle.nextElementSibling).toBe(conversationButton); expect( - conversationButton?.querySelector('.conversation-item__title') - ).toHaveAttribute('title', '产品发布讨论') - expect(screen.queryByText('任务 1')).not.toBeInTheDocument() - fireEvent.click(toggle) - const taskTitle = await screen.findByText('每周发布总结', { - selector: '.conversation-task-child__title' - }) - const taskChild = taskTitle.closest('button') - expect(taskChild).not.toBeNull() + conversationButton?.querySelector(".conversation-item__title"), + ).toHaveAttribute("title", "产品发布讨论"); + expect(screen.queryByText("任务 1")).not.toBeInTheDocument(); + fireEvent.click(toggle); + const taskTitle = await screen.findByText("每周发布总结", { + selector: ".conversation-task-child__title", + }); + const taskChild = taskTitle.closest("button"); + expect(taskChild).not.toBeNull(); + expect(taskChild?.querySelector(".task-status-dot")).toHaveClass( + "task-status-dot--idle", + ); expect( - taskChild?.querySelector('.task-status-dot') - ).toHaveClass('task-status-dot--idle') - expect(taskChild?.querySelector('.lucide-clock-fading')) - .not.toBeInTheDocument() + taskChild?.querySelector(".lucide-clock-fading"), + ).not.toBeInTheDocument(); expect( - taskChild?.querySelector('.conversation-task-child__meta') - ).toHaveTextContent('Execute · 每周 · 空闲') - expect(screen.queryByText('Run')).not.toBeInTheDocument() - fireEvent.click(taskChild!) - const taskRegion = await screen.findByRole('region', { - name: '当前会话的任务' - }) - expect(within(taskRegion).getByText('Execute')).toBeInTheDocument() - }) + taskChild?.querySelector(".conversation-task-child__meta"), + ).toHaveTextContent("Execute · 每周 · 空闲"); + expect(screen.queryByText("Run")).not.toBeInTheDocument(); + fireEvent.click(taskChild!); + const taskRegion = await screen.findByRole("region", { + name: "当前会话的任务", + }); + expect(within(taskRegion).getByText("Execute")).toBeInTheDocument(); + }); - it('keeps completed Conversation Task metadata compact and accessible', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000824' - const scheduleId = - '00000000-0000-4000-8000-000000000828' + it("keeps completed Conversation Task metadata compact and accessible", async () => { + const conversationId = "00000000-0000-4000-8000-000000000824"; + const scheduleId = "00000000-0000-4000-8000-000000000828"; const conversation: ConversationSnapshot = { id: conversationId, projectId, - title: '任务状态会话', + title: "任务状态会话", updatedAt: Date.now(), - messages: [] - } + messages: [], + }; const tasks: AssistantTask[] = [ { - id: '00000000-0000-4000-8000-000000000825', + id: "00000000-0000-4000-8000-000000000825", projectId, conversationId, scheduleId, - title: '归档任务', - instructions: '检查完成状态', - origin: 'schedule', - status: 'completed', - createdAt: '2026-08-19T03:00:00.000Z' + title: "归档任务", + instructions: "检查完成状态", + origin: "schedule", + status: "completed", + createdAt: "2026-08-19T03:00:00.000Z", }, { - id: '00000000-0000-4000-8000-000000000826', + id: "00000000-0000-4000-8000-000000000826", projectId, conversationId, - title: '运行中任务', - instructions: '检查运行状态', - origin: 'schedule', - status: 'running', - createdAt: '2026-08-19T02:00:00.000Z' + title: "运行中任务", + instructions: "检查运行状态", + origin: "schedule", + status: "running", + createdAt: "2026-08-19T02:00:00.000Z", }, { - id: '00000000-0000-4000-8000-000000000827', + id: "00000000-0000-4000-8000-000000000827", projectId, conversationId, - title: '失败任务', - instructions: '检查失败状态', - origin: 'schedule', - status: 'failed', - createdAt: '2026-08-19T01:00:00.000Z' - } - ] + title: "失败任务", + instructions: "检查失败状态", + origin: "schedule", + status: "failed", + createdAt: "2026-08-19T01:00:00.000Z", + }, + ]; const schedule: AssistantSchedule = { id: scheduleId, projectId, @@ -1791,970 +1716,909 @@ describe('App', () => { conversationId, title: tasks[0]!.title, prompt: tasks[0]!.instructions, - workMode: 'ask', - recurrence: 'once', - nextRunAt: '2026-08-19T03:00:00.000Z', + workMode: "ask", + recurrence: "once", + nextRunAt: "2026-08-19T03:00:00.000Z", enabled: false, createdAt: tasks[0]!.createdAt, - updatedAt: tasks[0]!.createdAt - } - vi.mocked(api.conversations.list).mockResolvedValue([ - conversation - ]) - vi.mocked(api.tasks.list).mockResolvedValue(tasks) - vi.mocked(api.schedules.list).mockResolvedValue([schedule]) + updatedAt: tasks[0]!.createdAt, + }; + vi.mocked(api.conversations.list).mockResolvedValue([conversation]); + vi.mocked(api.tasks.list).mockResolvedValue(tasks); + vi.mocked(api.schedules.list).mockResolvedValue([schedule]); - render() + render(); fireEvent.click( - await screen.findByLabelText( - '展开或折叠“任务状态会话”中的 3 个任务' - ) - ) - const completedTitle = await screen.findByText('归档任务', { - selector: '.conversation-task-child__title' - }) - const completedTask = completedTitle.closest('button') + await screen.findByLabelText("展开或折叠“任务状态会话”中的 3 个任务"), + ); + const completedTitle = await screen.findByText("归档任务", { + selector: ".conversation-task-child__title", + }); + const completedTask = completedTitle.closest("button"); const completedStatus = completedTask?.querySelector( - '.conversation-task-child__completed-status' - ) + ".conversation-task-child__completed-status", + ); expect(completedStatus).toHaveClass( - 'task-status-dot', - 'task-status-dot--completed' - ) - expect(completedStatus).toHaveAttribute('title', '已完成') + "task-status-dot", + "task-status-dot--completed", + ); + expect(completedStatus).toHaveAttribute("title", "已完成"); + expect(completedStatus?.querySelector(".lucide-check")).toBeInTheDocument(); + expect(completedTask?.firstElementChild).toBe(completedTitle); + expect(completedTask?.lastElementChild).toBe(completedStatus); + expect(completedTask).toHaveAccessibleName(/已完成/u); expect( - completedStatus?.querySelector('.lucide-check') - ).toBeInTheDocument() - expect(completedTask?.firstElementChild).toBe(completedTitle) - expect(completedTask?.lastElementChild).toBe(completedStatus) - expect(completedTask).toHaveAccessibleName(/已完成/u) + completedTask?.querySelector(".conversation-task-child__meta"), + ).toHaveTextContent("Ask · 仅一次"); expect( - completedTask?.querySelector('.conversation-task-child__meta') - ).toHaveTextContent('Ask · 仅一次') - expect( - completedTask?.querySelector('.conversation-task-child__meta') - ).not.toHaveTextContent('已完成') + completedTask?.querySelector(".conversation-task-child__meta"), + ).not.toHaveTextContent("已完成"); for (const [title, status, label] of [ - ['运行中任务', 'running', '运行中'], - ['失败任务', 'failed', '失败'] + ["运行中任务", "running", "运行中"], + ["失败任务", "failed", "失败"], ] as const) { const taskTitle = await screen.findByText(title, { - selector: '.conversation-task-child__title' - }) - const taskChild = taskTitle.closest('button') - expect( - taskChild?.querySelector('.task-status-dot') - ).toHaveClass(`task-status-dot--${status}`) - expect(taskChild?.firstElementChild).toBe(taskTitle) + selector: ".conversation-task-child__title", + }); + const taskChild = taskTitle.closest("button"); + expect(taskChild?.querySelector(".task-status-dot")).toHaveClass( + `task-status-dot--${status}`, + ); + expect(taskChild?.firstElementChild).toBe(taskTitle); expect(taskChild?.lastElementChild).toBe( - taskChild?.querySelector('.task-status-dot') - ) + taskChild?.querySelector(".task-status-dot"), + ); expect( - taskChild?.querySelector('.conversation-task-child__meta') - ).toHaveTextContent(label) + taskChild?.querySelector(".conversation-task-child__meta"), + ).toHaveTextContent(label); } expect( - document.querySelector('.conversation-task-child__icon') - ).not.toBeInTheDocument() - }) + document.querySelector(".conversation-task-child__icon"), + ).not.toBeInTheDocument(); + }); - it('routes scheduled Task approvals to the associated Conversation', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000831' - const taskId = '00000000-0000-4000-8000-000000000832' + it("routes scheduled Task approvals to the associated Conversation", async () => { + const conversationId = "00000000-0000-4000-8000-000000000831"; + const taskId = "00000000-0000-4000-8000-000000000832"; vi.mocked(api.conversations.list).mockResolvedValue([ { id: conversationId, projectId, - title: '发布审批会话', + title: "发布审批会话", updatedAt: Date.now(), - messages: [] - } - ]) + messages: [], + }, + ]); vi.mocked(api.tasks.list).mockResolvedValue([ { id: taskId, projectId, conversationId, - scheduleId: '00000000-0000-4000-8000-000000000833', - title: '发布任务', - instructions: '更新发布文件', - origin: 'schedule', - status: 'running', - createdAt: '2026-08-19T00:00:00.000Z' - } - ]) - render() - await screen.findAllByText('发布审批会话') - await waitFor(() => expect(api.tasks.list).toHaveBeenCalled()) + scheduleId: "00000000-0000-4000-8000-000000000833", + title: "发布任务", + instructions: "更新发布文件", + origin: "schedule", + status: "running", + createdAt: "2026-08-19T00:00:00.000Z", + }, + ]); + render(); + await screen.findAllByText("发布审批会话"); + await waitFor(() => expect(api.tasks.list).toHaveBeenCalled()); act(() => { agentListener?.({ requestId: taskId, - type: 'approval', - approvalId: '00000000-0000-4000-8000-000000000834', - title: '请求写入工作区', - description: '更新发布文件', - toolName: 'write_file', - argumentSummary: 'release.md', - allowPermanent: false - }) - }) + type: "approval", + approvalId: "00000000-0000-4000-8000-000000000834", + title: "请求写入工作区", + description: "更新发布文件", + toolName: "write_file", + argumentSummary: "release.md", + allowPermanent: false, + }); + }); - expect(await screen.findAllByText('请求写入工作区')) - .not.toHaveLength(0) - expect(screen.getByText('仅此次')).toBeInTheDocument() - expect(screen.getByLabelText('任务结果：发布任务')) - .toBeInTheDocument() - }) + expect(await screen.findAllByText("请求写入工作区")).not.toHaveLength(0); + expect(screen.getByText("仅此次")).toBeInTheDocument(); + expect(screen.getByLabelText("任务结果：发布任务")).toBeInTheDocument(); + }); - it('idle-preloads only the small Heartbeat route at startup', async () => { - render() + it("idle-preloads only the small Heartbeat route at startup", async () => { + render(); expect(window.requestIdleCallback).toHaveBeenCalledWith( expect.any(Function), - { timeout: 2000 } - ) - const idleCallback = vi.mocked(window.requestIdleCallback) - .mock.calls[0]?.[0] + { timeout: 2000 }, + ); + const idleCallback = vi.mocked(window.requestIdleCallback).mock + .calls[0]?.[0]; await act(async () => { idleCallback?.({ didTimeout: false, - timeRemaining: () => 50 - }) - }) - await waitFor(() => expect(routeModuleLoads.heartbeat).toBe(1)) + timeRemaining: () => 50, + }); + }); + await waitFor(() => expect(routeModuleLoads.heartbeat).toBe(1)); expect(routeModuleLoads).toMatchObject({ activity: 0, knowledge: 0, magicNotes: 0, - settings: 0 - }) - }) + settings: 0, + }); + }); - it('saves branding from Appearance and updates the sidebar immediately', async () => { - render() + it("saves branding from Appearance and updates the sidebar immediately", async () => { + render(); fireEvent.click( - await screen.findByRole('button', { - name: /本地工作区/u - }) - ) - await screen.findByRole( - 'heading', - { name: '设置中心' }, - { timeout: 3000 } - ) - fireEvent.click( - screen.getByRole('tab', { name: '外观' }) - ) - fireEvent.change(screen.getByLabelText('品牌名称'), { - target: { value: '研发伙伴' } - }) - fireEvent.change(screen.getByLabelText('中文副标题'), { - target: { value: '团队工作台' } - }) - fireEvent.click( - screen.getByRole('button', { name: '保存品牌设置' }) - ) + await screen.findByRole("button", { + name: /本地工作区/u, + }), + ); + await screen.findByRole("heading", { name: "设置中心" }, { timeout: 3000 }); + fireEvent.click(screen.getByRole("tab", { name: "外观" })); + fireEvent.change(screen.getByLabelText("品牌名称"), { + target: { value: "研发伙伴" }, + }); + fireEvent.change(screen.getByLabelText("中文副标题"), { + target: { value: "团队工作台" }, + }); + fireEvent.click(screen.getByRole("button", { name: "保存品牌设置" })); - expect(await screen.findByText('品牌设置已保存')) - .toBeInTheDocument() - const brand = document.querySelector('.brand') - expect(brand).not.toBeNull() - expect(within(brand as HTMLElement).getByText('研发伙伴')) - .toBeInTheDocument() - expect(within(brand as HTMLElement).getByText('团队工作台')) - .toBeInTheDocument() + expect(await screen.findByText("品牌设置已保存")).toBeInTheDocument(); + const brand = document.querySelector(".brand"); + expect(brand).not.toBeNull(); + expect( + within(brand as HTMLElement).getByText("研发伙伴"), + ).toBeInTheDocument(); + expect( + within(brand as HTMLElement).getByText("团队工作台"), + ).toBeInTheDocument(); expect(loadBrandingPreferences()).toMatchObject({ - name: '研发伙伴', - subtitleZhCN: '团队工作台' - }) - }) + name: "研发伙伴", + subtitleZhCN: "团队工作台", + }); + }); - it('preloads a heavy route once across repeated pointer and focus intent', async () => { - render() - const activity = await screen.findByRole('button', { - name: '运行记录' - }) + it("preloads a heavy route once across repeated pointer and focus intent", async () => { + render(); + const activity = await screen.findByRole("button", { + name: "运行记录", + }); - fireEvent.pointerEnter(activity) - fireEvent.focus(activity) - fireEvent.pointerEnter(activity) + fireEvent.pointerEnter(activity); + fireEvent.focus(activity); + fireEvent.pointerEnter(activity); - await waitFor(() => expect(routeModuleLoads.activity).toBe(1)) - }) + await waitFor(() => expect(routeModuleLoads.activity).toBe(1)); + }); - it('waits for project bootstrap before project-scoped startup loads', async () => { - const projects = deferred<(typeof project)[]>() - vi.mocked(api.projects.list).mockImplementationOnce( - () => projects.promise - ) + it("waits for project bootstrap before project-scoped startup loads", async () => { + const projects = deferred<(typeof project)[]>(); + vi.mocked(api.projects.list).mockImplementationOnce(() => projects.promise); - render() + render(); - expect(api.projects.list).toHaveBeenCalledOnce() - expect(api.memory.list).not.toHaveBeenCalled() - expect(api.schedules.list).not.toHaveBeenCalled() - expect(api.heartbeats.list).not.toHaveBeenCalled() + expect(api.projects.list).toHaveBeenCalledOnce(); + expect(api.memory.list).not.toHaveBeenCalled(); + expect(api.schedules.list).not.toHaveBeenCalled(); + expect(api.heartbeats.list).not.toHaveBeenCalled(); - await act(async () => projects.resolve([project])) + await act(async () => projects.resolve([project])); await waitFor(() => { - expect(api.memory.list).toHaveBeenCalledTimes(2) - expect(api.memory.list).toHaveBeenCalledWith(projectId) - expect(api.memory.list).toHaveBeenCalledWith() - expect(api.schedules.list).toHaveBeenCalledOnce() - expect(api.schedules.list).toHaveBeenCalledWith() - expect(api.heartbeats.list).toHaveBeenCalledOnce() - }) - }) + expect(api.memory.list).toHaveBeenCalledTimes(2); + expect(api.memory.list).toHaveBeenCalledWith(projectId); + expect(api.memory.list).toHaveBeenCalledWith(); + expect(api.schedules.list).toHaveBeenCalledOnce(); + expect(api.schedules.list).toHaveBeenCalledWith(); + expect(api.heartbeats.list).toHaveBeenCalledOnce(); + }); + }); - it('shows an accessible fallback while a lazy route loads', async () => { - lazyRouteMocks.suspendKnowledgeRoute() + it("shows an accessible fallback while a lazy route loads", async () => { + lazyRouteMocks.suspendKnowledgeRoute(); try { - render() + render(); - fireEvent.click( - await screen.findByRole('button', { name: '知识库' }) - ) + fireEvent.click(await screen.findByRole("button", { name: "知识库" })); - const loading = screen.getByRole('status', { - name: '正在加载页面…' - }) - expect(loading).toHaveAttribute('aria-live', 'polite') - expect(loading).toHaveAttribute('aria-busy', 'true') - await act(async () => lazyRouteMocks.releaseKnowledgeRoute()) + const loading = screen.getByRole("status", { + name: "正在加载页面…", + }); + expect(loading).toHaveAttribute("aria-live", "polite"); + expect(loading).toHaveAttribute("aria-busy", "true"); + await act(async () => lazyRouteMocks.releaseKnowledgeRoute()); expect( await screen.findByRole( - 'heading', + "heading", { level: 1, - name: '知识库' + name: "知识库", }, - { timeout: 3000 } - ) - ).toBeInTheDocument() + { timeout: 3000 }, + ), + ).toBeInTheDocument(); expect( - screen.queryByRole('status', { name: '正在加载页面…' }) - ).not.toBeInTheDocument() + screen.queryByRole("status", { name: "正在加载页面…" }), + ).not.toBeInTheDocument(); } finally { - lazyRouteMocks.releaseKnowledgeRoute() + lazyRouteMocks.releaseKnowledgeRoute(); } - }) + }); - it('keeps a recently visited workspace page mounted', async () => { - render() - fireEvent.click( - await screen.findByRole('button', { name: '知识库' }) - ) - const heading = await screen.findByRole('heading', { + it("keeps a recently visited workspace page mounted", async () => { + render(); + fireEvent.click(await screen.findByRole("button", { name: "知识库" })); + const heading = await screen.findByRole("heading", { level: 1, - name: '知识库' - }) - const route = heading.closest('[data-route="knowledge"]') - expect(route).not.toHaveAttribute('hidden') + name: "知识库", + }); + const route = heading.closest('[data-route="knowledge"]'); + expect(route).not.toHaveAttribute("hidden"); - fireEvent.click(screen.getByRole('button', { name: '对话' })) - expect(heading).toBeInTheDocument() - expect(route).toHaveAttribute('hidden') + fireEvent.click(screen.getByRole("button", { name: "对话" })); + expect(heading).toBeInTheDocument(); + expect(route).toHaveAttribute("hidden"); - fireEvent.click(screen.getByRole('button', { name: '知识库' })) + fireEvent.click(screen.getByRole("button", { name: "知识库" })); expect( - await screen.findByRole('heading', { + await screen.findByRole("heading", { level: 1, - name: '知识库' - }) - ).toBe(heading) - expect(route).not.toHaveAttribute('hidden') - }) + name: "知识库", + }), + ).toBe(heading); + expect(route).not.toHaveAttribute("hidden"); + }); - it('enforces the workspace KeepAlive cap during rapid visits', async () => { - let now = 2_000_000 - vi.spyOn(Date, 'now').mockImplementation(() => ++now) - render() + it("enforces the workspace KeepAlive cap during rapid visits", async () => { + let now = 2_000_000; + vi.spyOn(Date, "now").mockImplementation(() => ++now); + render(); - fireEvent.click( - await screen.findByRole('button', { name: '知识库' }) - ) - await screen.findByRole('heading', { name: '知识库' }) - fireEvent.click(screen.getByRole('button', { name: '智能心跳' })) - await screen.findByRole('heading', { name: '智能心跳' }) - fireEvent.click(screen.getByRole('button', { name: '运行记录' })) - await screen.findByRole('heading', { name: '运行记录' }) - fireEvent.click(screen.getByRole('button', { name: '对话' })) - fireEvent.click( - screen.getByRole('button', { name: /本地工作区/u }) - ) - await screen.findByRole('heading', { name: '设置中心' }) + fireEvent.click(await screen.findByRole("button", { name: "知识库" })); + await screen.findByRole("heading", { name: "知识库" }); + fireEvent.click(screen.getByRole("button", { name: "智能心跳" })); + await screen.findByRole("heading", { name: "智能心跳" }); + fireEvent.click(screen.getByRole("button", { name: "运行记录" })); + await screen.findByRole("heading", { name: "运行记录" }); + fireEvent.click(screen.getByRole("button", { name: "对话" })); + fireEvent.click(screen.getByRole("button", { name: /本地工作区/u })); + await screen.findByRole("heading", { name: "设置中心" }); + expect(document.querySelectorAll(".workspace-route-cache")).toHaveLength(4); expect( - document.querySelectorAll('.workspace-route-cache') - ).toHaveLength(4) - expect( - document.querySelector('[data-route="knowledge"]') - ).not.toBeInTheDocument() - }) + document.querySelector('[data-route="knowledge"]'), + ).not.toBeInTheDocument(); + }); - it('preserves title, message, and project filtering with deferred search', async () => { + it("preserves title, message, and project filtering with deferred search", async () => { vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000411', + id: "00000000-0000-4000-8000-000000000411", projectId, - title: '标题里的 Alpha', + title: "标题里的 Alpha", updatedAt: 1_775_000_000_003, messages: [ { - id: '00000000-0000-4000-8000-000000000412', - role: 'assistant', - content: '普通正文', + id: "00000000-0000-4000-8000-000000000412", + role: "assistant", + content: "普通正文", createdAt: 1_775_000_000_003, - state: 'complete' - } - ] + state: "complete", + }, + ], }, { - id: '00000000-0000-4000-8000-000000000413', + id: "00000000-0000-4000-8000-000000000413", projectId, - title: '正文命中的会话', + title: "正文命中的会话", updatedAt: 1_775_000_000_002, messages: [ { - id: '00000000-0000-4000-8000-000000000414', - role: 'user', - content: '这里包含 Beta Needle', + id: "00000000-0000-4000-8000-000000000414", + role: "user", + content: "这里包含 Beta Needle", createdAt: 1_775_000_000_002, - state: 'complete' - } - ] + state: "complete", + }, + ], }, { - id: '00000000-0000-4000-8000-000000000415', - projectId: '00000000-0000-4000-8000-000000000999', - title: '其他项目里的 Alpha', + id: "00000000-0000-4000-8000-000000000415", + projectId: "00000000-0000-4000-8000-000000000999", + title: "其他项目里的 Alpha", updatedAt: 1_775_000_000_001, messages: [ { - id: '00000000-0000-4000-8000-000000000416', - role: 'user', - content: 'Beta Needle', + id: "00000000-0000-4000-8000-000000000416", + role: "user", + content: "Beta Needle", createdAt: 1_775_000_000_001, - state: 'complete' - } - ] - } - ]) - const { container } = render() - const search = await screen.findByLabelText('搜索对话') + state: "complete", + }, + ], + }, + ]); + const { container } = render(); + const search = await screen.findByLabelText("搜索对话"); const conversationList = - container.querySelector('.conversation-list') + container.querySelector(".conversation-list"); if (!conversationList) { - throw new Error('Missing conversation list') + throw new Error("Missing conversation list"); } expect( - await within(conversationList).findByText('标题里的 Alpha') - ).toBeInTheDocument() + await within(conversationList).findByText("标题里的 Alpha"), + ).toBeInTheDocument(); expect( - within(conversationList).queryByText('其他项目里的 Alpha') - ).not.toBeInTheDocument() + within(conversationList).queryByText("其他项目里的 Alpha"), + ).not.toBeInTheDocument(); - fireEvent.change(search, { target: { value: 'beta needle' } }) - expect(search).toHaveValue('beta needle') + fireEvent.change(search, { target: { value: "beta needle" } }); + expect(search).toHaveValue("beta needle"); await waitFor(() => { expect( - within(conversationList).getByText('正文命中的会话') - ).toBeInTheDocument() + within(conversationList).getByText("正文命中的会话"), + ).toBeInTheDocument(); expect( - within(conversationList).queryByText('标题里的 Alpha') - ).not.toBeInTheDocument() - }) + within(conversationList).queryByText("标题里的 Alpha"), + ).not.toBeInTheDocument(); + }); - fireEvent.change(search, { target: { value: 'ALPHA' } }) + fireEvent.change(search, { target: { value: "ALPHA" } }); await waitFor(() => { expect( - within(conversationList).getByText('标题里的 Alpha') - ).toBeInTheDocument() + within(conversationList).getByText("标题里的 Alpha"), + ).toBeInTheDocument(); expect( - within(conversationList).queryByText('正文命中的会话') - ).not.toBeInTheDocument() + within(conversationList).queryByText("正文命中的会话"), + ).not.toBeInTheDocument(); expect( - within(conversationList).queryByText('其他项目里的 Alpha') - ).not.toBeInTheDocument() - }) + within(conversationList).queryByText("其他项目里的 Alpha"), + ).not.toBeInTheDocument(); + }); - fireEvent.change(search, { target: { value: 'missing topic' } }) + fireEvent.change(search, { target: { value: "missing topic" } }); expect( - await within(conversationList).findByText('没有匹配的对话') - ).toBeInTheDocument() + await within(conversationList).findByText("没有匹配的对话"), + ).toBeInTheDocument(); expect( within(conversationList).getByText( - '请修改搜索关键词，或清除搜索后查看全部会话。' - ) - ).toBeInTheDocument() + "请修改搜索关键词，或清除搜索后查看全部会话。", + ), + ).toBeInTheDocument(); fireEvent.click( - within(conversationList).getByRole('button', { - name: '清除搜索' - }) - ) - expect(search).toHaveValue('') + within(conversationList).getByRole("button", { + name: "清除搜索", + }), + ); + expect(search).toHaveValue(""); expect( - await within(conversationList).findByText('标题里的 Alpha') - ).toBeInTheDocument() - }) + await within(conversationList).findByText("标题里的 Alpha"), + ).toBeInTheDocument(); + }); - it('keeps Settings open when the interface language changes', async () => { + it("keeps Settings open when the interface language changes", async () => { api.updates = { getSettings: vi.fn(async () => ({ checkUpdatesOnStartup: false, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: false, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), updateSettings: vi.fn(), check: vi.fn(), openReleasePage: vi.fn(), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; try { render( - - ) + , + ); fireEvent.click( - await screen.findByRole('button', { - name: /本地工作区/u - }) - ) + await screen.findByRole("button", { + name: /本地工作区/u, + }), + ); await screen.findByRole( - 'heading', - { name: '设置中心' }, - { timeout: 3000 } - ) - fireEvent.click( - screen.getByRole('tab', { name: '外观' }) - ) - const projectsList = vi.mocked(api.projects.list) - const expertsList = vi.mocked(api.experts.list) - const tasksList = vi.mocked(api.tasks.list) + "heading", + { name: "设置中心" }, + { timeout: 3000 }, + ); + fireEvent.click(screen.getByRole("tab", { name: "外观" })); + const projectsList = vi.mocked(api.projects.list); + const expertsList = vi.mocked(api.experts.list); + const tasksList = vi.mocked(api.tasks.list); await waitFor(() => { - expect(projectsList).toHaveBeenCalled() - expect(expertsList).toHaveBeenCalled() - expect(tasksList).toHaveBeenCalled() - }) + expect(projectsList).toHaveBeenCalled(); + expect(expertsList).toHaveBeenCalled(); + expect(tasksList).toHaveBeenCalled(); + }); const loadCounts = { projects: projectsList.mock.calls.length, experts: expertsList.mock.calls.length, - tasks: tasksList.mock.calls.length - } + tasks: tasksList.mock.calls.length, + }; fireEvent.click( - screen.getByRole('radio', { - name: /^English/u - }) - ) + screen.getByRole("radio", { + name: /^English/u, + }), + ); expect( - await screen.findByRole('region', { - name: 'Settings' - }) - ).toBeInTheDocument() + await screen.findByRole("region", { + name: "Settings", + }), + ).toBeInTheDocument(); expect( - screen.getByRole('heading', { + screen.getByRole("heading", { level: 1, - name: 'Settings' - }) - ).toBeInTheDocument() - expect(api.updates.getSettings).toHaveBeenCalledOnce() + name: "Settings", + }), + ).toBeInTheDocument(); + expect(api.updates.getSettings).toHaveBeenCalledOnce(); await act(async () => { - await new Promise((resolve) => setTimeout(resolve, 0)) - }) - expect(projectsList).toHaveBeenCalledTimes(loadCounts.projects) - expect(expertsList).toHaveBeenCalledTimes(loadCounts.experts) - expect(tasksList).toHaveBeenCalledTimes(loadCounts.tasks) + await new Promise((resolve) => setTimeout(resolve, 0)); + }); + expect(projectsList).toHaveBeenCalledTimes(loadCounts.projects); + expect(expertsList).toHaveBeenCalledTimes(loadCounts.experts); + expect(tasksList).toHaveBeenCalledTimes(loadCounts.tasks); } finally { - delete api.updates - cleanup() - await changeUiLocale('zh-CN') + delete api.updates; + cleanup(); + await changeUiLocale("zh-CN"); } - }) + }); - it('checks for updates silently on startup and only reports a new version', async () => { + it("checks for updates silently on startup and only reports a new version", async () => { const check = vi.fn(async () => ({ updateAvailable: true, - currentVersion: '0.8.4', - latestVersion: '0.9.0', - releaseUrl: - 'https://github.com/mesalogo/goodbuddy/releases/tag/v0.9.0', + currentVersion: "0.8.4", + latestVersion: "0.9.0", + releaseUrl: "https://github.com/mesalogo/goodbuddy/releases/tag/v0.9.0", target: { - platform: 'windows' as const, - arch: 'x64' as const, - formats: ['nsis', 'portable'], + platform: "windows" as const, + arch: "x64" as const, + formats: ["nsis", "portable"], files: [ { - name: 'GoodBuddy-0.9.0-windows-x64-portable.zip', + name: "GoodBuddy-0.9.0-windows-x64-portable.zip", size: 1, - sha256: 'a'.repeat(64) - } - ] - } - })) + sha256: "a".repeat(64), + }, + ], + }, + })); api.updates = { getSettings: vi.fn(async () => ({ checkUpdatesOnStartup: true, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: true, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), updateSettings: vi.fn(async () => ({ checkUpdatesOnStartup: true, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: true, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), check, openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; try { - render() - await waitFor(() => expect(check).toHaveBeenCalledOnce()) + render(); + await waitFor(() => expect(check).toHaveBeenCalledOnce()); expect( - await screen.findByText( - '发现 GoodBuddy 0.9.0，可在“关于与更新”中查看' - ) - ).toBeInTheDocument() - expect(api.updates.onResult).not.toHaveBeenCalled() + await screen.findByText("发现 GoodBuddy 0.9.0，可在“关于与更新”中查看"), + ).toBeInTheDocument(); + expect(api.updates.onResult).not.toHaveBeenCalled(); } finally { - delete api.updates + delete api.updates; } - }) + }); - it('shows and acknowledges pending release notes on startup', async () => { - const acknowledge = vi.fn(async () => {}) + it("shows and acknowledges pending release notes on startup", async () => { + const acknowledge = vi.fn(async () => {}); api.releaseNotes = { getPending: vi.fn(async () => ({ - currentVersion: '0.8.18', + currentVersion: "0.8.18", releases: [ { - version: '0.8.18', - releasedAt: '2026-08-11', + version: "0.8.18", + releasedAt: "2026-08-11", notes: { - 'zh-CN': { - highlights: ['多 Runtime 工作流更加连贯'], - features: ['新增版本更新说明'], - fixes: ['修复重复显示'], - notices: ['Ask 模式保持只读'] + "zh-CN": { + highlights: ["多 Runtime 工作流更加连贯"], + features: ["新增版本更新说明"], + fixes: ["修复重复显示"], + notices: ["Ask 模式保持只读"], }, - 'en-US': { - highlights: ['Multi-Runtime workflows are more cohesive'], - features: ['Added release notes'], - fixes: ['Fixed repeated display'], - notices: ['Ask remains read-only'] - } - } - } - ] + "en-US": { + highlights: ["Multi-Runtime workflows are more cohesive"], + features: ["Added release notes"], + fixes: ["Fixed repeated display"], + notices: ["Ask remains read-only"], + }, + }, + }, + ], })), - acknowledge - } + acknowledge, + }; try { - render() + render(); expect( - await screen.findByRole('dialog', { - name: 'GoodBuddy 0.8.18 更新内容' - }) - ).toBeInTheDocument() - fireEvent.click(screen.getByRole('button', { name: '开始使用' })) + await screen.findByRole("dialog", { + name: "GoodBuddy 0.8.18 更新内容", + }), + ).toBeInTheDocument(); + fireEvent.click(screen.getByRole("button", { name: "开始使用" })); + await waitFor(() => expect(acknowledge).toHaveBeenCalledWith("0.8.18")); await waitFor(() => - expect(acknowledge).toHaveBeenCalledWith('0.8.18') - ) - await waitFor(() => - expect(screen.queryByRole('dialog')).not.toBeInTheDocument() - ) + expect(screen.queryByRole("dialog")).not.toBeInTheDocument(), + ); } finally { - delete api.releaseNotes + delete api.releaseNotes; } - }) + }); - it('keeps current-version checks quiet and reports startup check failures', async () => { + it("keeps current-version checks quiet and reports startup check failures", async () => { const currentResult = { updateAvailable: false, - currentVersion: '0.8.4', - latestVersion: '0.8.4', - releaseUrl: - 'https://github.com/mesalogo/goodbuddy/releases/tag/v0.8.4', + currentVersion: "0.8.4", + latestVersion: "0.8.4", + releaseUrl: "https://github.com/mesalogo/goodbuddy/releases/tag/v0.8.4", target: { - platform: 'windows' as const, - arch: 'x64' as const, - formats: ['nsis', 'portable'], - files: [] - } - } + platform: "windows" as const, + arch: "x64" as const, + formats: ["nsis", "portable"], + files: [], + }, + }; const check = vi .fn() .mockResolvedValueOnce(currentResult) .mockRejectedValueOnce( new Error( - "Error invoking remote method 'application:update:check': TypeError: fetch failed" - ) - ) + "Error invoking remote method 'application:update:check': TypeError: fetch failed", + ), + ); api.updates = { getSettings: vi.fn(async () => ({ checkUpdatesOnStartup: true, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: true, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), updateSettings: vi.fn(async () => ({ checkUpdatesOnStartup: true, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: true, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), check, openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; try { - const first = render() - await waitFor(() => expect(check).toHaveBeenCalledTimes(1)) + const first = render(); + await waitFor(() => expect(check).toHaveBeenCalledTimes(1)); expect( - screen.queryByText(/发现 GoodBuddy|版本检查失败/u) - ).not.toBeInTheDocument() + screen.queryByText(/发现 GoodBuddy|版本检查失败/u), + ).not.toBeInTheDocument(); - first.unmount() - render() - await waitFor(() => expect(check).toHaveBeenCalledTimes(2)) - expect( - await screen.findByRole('alert') - ).toHaveTextContent( - '启动更新检查失败（GitHub）：无法连接更新源，请检查网络或代理后重试' - ) + first.unmount(); + render(); + await waitFor(() => expect(check).toHaveBeenCalledTimes(2)); + expect(await screen.findByRole("alert")).toHaveTextContent( + "启动更新检查失败（GitHub）：无法连接更新源，请检查网络或代理后重试", + ); } finally { - delete api.updates + delete api.updates; } - }) + }); - it('shows a recoverable recent-conversation load failure', async () => { + it("shows a recoverable recent-conversation load failure", async () => { vi.mocked(api.projects.list) - .mockRejectedValueOnce(new Error('database busy')) - .mockResolvedValueOnce([project]) + .mockRejectedValueOnce(new Error("database busy")) + .mockResolvedValueOnce([project]); - render() + render(); - const loadError = await screen.findByRole('alert') - expect(loadError).toHaveTextContent('最近会话加载失败') - expect(loadError).toHaveTextContent('database busy') + const loadError = await screen.findByRole("alert"); + expect(loadError).toHaveTextContent("最近会话加载失败"); + expect(loadError).toHaveTextContent("database busy"); fireEvent.click( - within(loadError).getByRole('button', { - name: '重新读取会话' - }) - ) + within(loadError).getByRole("button", { + name: "重新读取会话", + }), + ); + await waitFor(() => expect(api.projects.list).toHaveBeenCalledTimes(2)); await waitFor(() => - expect(api.projects.list).toHaveBeenCalledTimes(2) - ) - await waitFor(() => - expect(screen.queryByText('最近会话加载失败')) - .not.toBeInTheDocument() - ) - }) + expect(screen.queryByText("最近会话加载失败")).not.toBeInTheDocument(), + ); + }); - it('keeps rendering when an older preload has no browser bridge', async () => { - Object.defineProperty(window, 'goodbuddy', { + it("keeps rendering when an older preload has no browser bridge", async () => { + Object.defineProperty(window, "goodbuddy", { configurable: true, value: { ...api, - browser: undefined - } - }) + browser: undefined, + }, + }); - render() + render(); expect( - await screen.findByLabelText('向 GoodBuddy 提问') - ).toBeInTheDocument() - }) + await screen.findByLabelText("向 GoodBuddy 提问"), + ).toBeInTheDocument(); + }); - it('uses local transcription when Electron has no Web Speech API', async () => { - Object.defineProperty(window, 'SpeechRecognition', { + it("uses local transcription when Electron has no Web Speech API", async () => { + Object.defineProperty(window, "SpeechRecognition", { configurable: true, - value: undefined - }) - Object.defineProperty(window, 'webkitSpeechRecognition', { + value: undefined, + }); + Object.defineProperty(window, "webkitSpeechRecognition", { configurable: true, - value: undefined - }) + value: undefined, + }); - render() - fireEvent.click(await screen.findByLabelText('语音输入')) + render(); + fireEvent.click(await screen.findByLabelText("语音输入")); await waitFor(() => expect(api.speech?.transcribe).toHaveBeenCalledWith( expect.objectContaining({ sampleRate: 16_000, - audio: expect.any(ArrayBuffer) - }) - ) - ) - expect(await screen.findByDisplayValue('本地语音结果')).toBeInTheDocument() - expect( - screen.getByText('快捷唤起：', { exact: false }) - ).toHaveTextContent('快捷唤起：Ctrl+Shift+Space') - expect( - screen.queryByText(/CommandOrControl/) - ).not.toBeInTheDocument() - }) + audio: expect.any(ArrayBuffer), + }), + ), + ); + expect(await screen.findByDisplayValue("本地语音结果")).toBeInTheDocument(); + expect(screen.getByText("快捷唤起：", { exact: false })).toHaveTextContent( + "快捷唤起：Ctrl+Shift+Space", + ); + expect(screen.queryByText(/CommandOrControl/)).not.toBeInTheDocument(); + }); - it('measures composer height once per input and resizes after sending', async () => { - render() - const input = await screen.findByLabelText('向 GoodBuddy 提问') + it("measures composer height once per input and resizes after sending", async () => { + render(); + const input = await screen.findByLabelText("向 GoodBuddy 提问"); const measureHeight = vi.fn(() => - (input as HTMLTextAreaElement).value ? 300 : 0 - ) - Object.defineProperty(input, 'scrollHeight', { + (input as HTMLTextAreaElement).value ? 300 : 0, + ); + Object.defineProperty(input, "scrollHeight", { configurable: true, - get: measureHeight - }) + get: measureHeight, + }); - fireEvent.input(input, { target: { value: 'First line\nSecond line' } }) - expect(measureHeight).toHaveBeenCalledTimes(1) - expect(input).toHaveStyle({ height: '220px' }) + fireEvent.input(input, { target: { value: "First line\nSecond line" } }); + expect(measureHeight).toHaveBeenCalledTimes(1); + expect(input).toHaveStyle({ height: "220px" }); - measureHeight.mockClear() - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(input).toHaveValue('')) - expect(measureHeight).toHaveBeenCalledTimes(1) - expect(input).toHaveStyle({ height: '72px' }) - }) + measureHeight.mockClear(); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(input).toHaveValue("")); + expect(measureHeight).toHaveBeenCalledTimes(1); + expect(input).toHaveStyle({ height: "72px" }); + }); - it('shows a red recording state until microphone capture stops', async () => { + it("shows a red recording state until microphone capture stops", async () => { let resolveRecording!: (value: { - audio: ArrayBuffer - sampleRate: 16_000 - }) => void - const stop = vi.fn() + audio: ArrayBuffer; + sampleRate: 16_000; + }) => void; + const stop = vi.fn(); speechRecognitionMocks.startPcmRecording.mockResolvedValueOnce({ result: new Promise((resolve) => { - resolveRecording = resolve + resolveRecording = resolve; }), stop, - cancel: vi.fn() - }) + cancel: vi.fn(), + }); - render() - const input = await screen.findByLabelText('向 GoodBuddy 提问') - expect(input).toHaveAttribute('rows', '3') - expect(input).toHaveStyle({ height: '72px' }) + render(); + const input = await screen.findByLabelText("向 GoodBuddy 提问"); + expect(input).toHaveAttribute("rows", "3"); + expect(input).toHaveStyle({ height: "72px" }); - fireEvent.click(screen.getByLabelText('语音输入')) - const recordingButton = await screen.findByRole('button', { - name: '停止录音' - }) - expect(recordingButton).toHaveAttribute('data-state', 'recording') - expect(recordingButton).toHaveAttribute('aria-pressed', 'true') - expect(recordingButton).toHaveClass( - 'composer__voice-button--recording' - ) + fireEvent.click(screen.getByLabelText("语音输入")); + const recordingButton = await screen.findByRole("button", { + name: "停止录音", + }); + expect(recordingButton).toHaveAttribute("data-state", "recording"); + expect(recordingButton).toHaveAttribute("aria-pressed", "true"); + expect(recordingButton).toHaveClass("composer__voice-button--recording"); - fireEvent.click(recordingButton) - expect(stop).toHaveBeenCalledOnce() + fireEvent.click(recordingButton); + expect(stop).toHaveBeenCalledOnce(); resolveRecording({ audio: new Float32Array([0, 0.25, -0.25]).buffer, - sampleRate: 16_000 - }) - expect( - await screen.findByDisplayValue('本地语音结果') - ).toBeInTheDocument() - }) + sampleRate: 16_000, + }); + expect(await screen.findByDisplayValue("本地语音结果")).toBeInTheDocument(); + }); - it('keeps conversation actions in the conversation list', async () => { - const { container } = render() - const topbar = container.querySelector('.topbar') + it("keeps conversation actions in the conversation list", async () => { + const { container } = render(); + const topbar = container.querySelector(".topbar"); const conversationList = - container.querySelector('.conversation-list') - expect(topbar).not.toBeNull() - expect(conversationList).not.toBeNull() + container.querySelector(".conversation-list"); + expect(topbar).not.toBeNull(); + expect(conversationList).not.toBeNull(); if (!topbar || !conversationList) { - return + return; } expect( - within(topbar).queryByRole('button', { - name: /^专家角色：/u - }) - ).not.toBeInTheDocument() - expect(composerMenuTrigger('专家角色').closest('.composer')).not.toBeNull() + within(topbar).queryByRole("button", { + name: /^专家角色：/u, + }), + ).not.toBeInTheDocument(); + expect(composerMenuTrigger("专家角色").closest(".composer")).not.toBeNull(); - const themeToggle = within(topbar).getByRole('button', { - name: '切换深色主题' - }) - fireEvent.click(themeToggle) + const themeToggle = within(topbar).getByRole("button", { + name: "切换深色主题", + }); + fireEvent.click(themeToggle); await waitFor(() => - expect(document.documentElement.dataset.theme).toBe('dark') - ) + expect(document.documentElement.dataset.theme).toBe("dark"), + ); expect( - screen.queryByRole('menuitem', { name: '重命名会话' }) - ).not.toBeInTheDocument() + screen.queryByRole("menuitem", { name: "重命名会话" }), + ).not.toBeInTheDocument(); expect( - within(topbar).getByRole('button', { - name: '切换浅色主题' - }) - ).toBe(themeToggle) - expect(screen.queryByRole('menu')).not.toBeInTheDocument() + within(topbar).getByRole("button", { + name: "切换浅色主题", + }), + ).toBe(themeToggle); + expect(screen.queryByRole("menu")).not.toBeInTheDocument(); - const conversationMenuTrigger = within( - conversationList - ).getByLabelText('更多会话操作 新对话') - fireEvent.click(conversationMenuTrigger) - const renameButton = within(conversationList).getByRole('button', { - name: '重命名会话' - }) - expect(renameButton).toBeVisible() + const conversationMenuTrigger = + within(conversationList).getByLabelText("更多会话操作 新对话"); + fireEvent.click(conversationMenuTrigger); + const renameButton = within(conversationList).getByRole("button", { + name: "重命名会话", + }); + expect(renameButton).toBeVisible(); expect( - within(conversationList).getByRole('button', { - name: '复制完整会话' - }) - ).toBeVisible() + within(conversationList).getByRole("button", { + name: "复制完整会话", + }), + ).toBeVisible(); expect( - within(conversationList).getByRole('button', { - name: '导出 Markdown' - }) - ).toBeVisible() + within(conversationList).getByRole("button", { + name: "导出 Markdown", + }), + ).toBeVisible(); - fireEvent.click(renameButton) - const renameInput = within(conversationList).getByLabelText( - '重命名会话 新对话' - ) + fireEvent.click(renameButton); + const renameInput = + within(conversationList).getByLabelText("重命名会话 新对话"); fireEvent.change(renameInput, { - target: { value: '重命名后的会话' } - }) - fireEvent.submit(renameInput.closest('form')!) + target: { value: "重命名后的会话" }, + }); + fireEvent.submit(renameInput.closest("form")!); expect( - within(conversationList).getByText('重命名后的会话') - ).toBeInTheDocument() - await waitFor(() => expect(conversationMenuTrigger).toHaveFocus()) + within(conversationList).getByText("重命名后的会话"), + ).toBeInTheDocument(); + await waitFor(() => expect(conversationMenuTrigger).toHaveFocus()); - fireEvent.click(screen.getByRole('button', { name: '知识库' })) + fireEvent.click(screen.getByRole("button", { name: "知识库" })); fireEvent.click( - within(conversationList).getByLabelText( - '更多会话操作 重命名后的会话' - ) - ) + within(conversationList).getByLabelText("更多会话操作 重命名后的会话"), + ); fireEvent.click( - within(conversationList).getByRole('button', { - name: '复制完整会话' - }) - ) - await waitFor(() => - expect(api.clipboard.writeText).toHaveBeenCalledOnce() - ) - expect(await screen.findByRole('status')).toBeVisible() - }) + within(conversationList).getByRole("button", { + name: "复制完整会话", + }), + ); + await waitFor(() => expect(api.clipboard.writeText).toHaveBeenCalledOnce()); + expect(await screen.findByRole("status")).toBeVisible(); + }); - it('copies a completed assistant reply through the desktop bridge', async () => { - render() + it("copies a completed assistant reply through the desktop bridge", async () => { + render(); - fireEvent.click( - screen.getByRole('button', { name: '复制此回复' }) - ) + fireEvent.click(screen.getByRole("button", { name: "复制此回复" })); await waitFor(() => expect(api.clipboard.writeText).toHaveBeenCalledWith( - expect.stringContaining('你好，我是 GoodBuddy') - ) - ) - expect( - await screen.findByText('回复已复制到剪贴板') - ).toBeVisible() - }) + expect.stringContaining("你好，我是 GoodBuddy"), + ), + ); + expect(await screen.findByText("回复已复制到剪贴板")).toBeVisible(); + }); - it('reports a desktop clipboard write failure', async () => { + it("reports a desktop clipboard write failure", async () => { vi.mocked(api.clipboard.writeText).mockRejectedValueOnce( - new Error('clipboard unavailable') - ) - render() + new Error("clipboard unavailable"), + ); + render(); - fireEvent.click( - screen.getByRole('button', { name: '复制此回复' }) - ) + fireEvent.click(screen.getByRole("button", { name: "复制此回复" })); - expect( - await screen.findByRole('alert') - ).toHaveTextContent('无法访问剪贴板，请检查系统权限') - }) + expect(await screen.findByRole("alert")).toHaveTextContent( + "无法访问剪贴板，请检查系统权限", + ); + }); - it('flushes and opens an independent conversation branch with provenance badges', async () => { - const sourceConversationId = - '00000000-0000-4000-8000-000000000421' - const branchConversationId = - '00000000-0000-4000-8000-000000000422' + it("flushes and opens an independent conversation branch with provenance badges", async () => { + const sourceConversationId = "00000000-0000-4000-8000-000000000421"; + const branchConversationId = "00000000-0000-4000-8000-000000000422"; const sourceConversation: ConversationSnapshot = { id: sourceConversationId, projectId, - title: '方案讨论', + title: "方案讨论", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000423', - role: 'user', - content: '探索不同发布方案', + id: "00000000-0000-4000-8000-000000000423", + role: "user", + content: "探索不同发布方案", createdAt: 1_775_000_000_000, - state: 'complete' + state: "complete", }, { - id: '00000000-0000-4000-8000-000000000424', - role: 'assistant', - content: '可以比较风险、成本和回滚路径。', + id: "00000000-0000-4000-8000-000000000424", + role: "assistant", + content: "可以比较风险、成本和回滚路径。", createdAt: 1_775_000_001_000, - state: 'complete' - } - ] - } + state: "complete", + }, + ], + }; vi.mocked(api.conversations.list).mockResolvedValueOnce([ - sourceConversation - ]) + sourceConversation, + ]); vi.mocked(api.conversations.branchLocal).mockImplementationOnce( async (input) => ({ ...sourceConversation, id: branchConversationId, branch: { sourceConversationId, - sourceTitle: '方案讨论更新' + sourceTitle: "方案讨论更新", }, contextMetrics: undefined, contextCompressionState: undefined, @@ -2762,712 +2626,649 @@ describe('App', () => { updatedAt: 1_775_000_002_000, messages: sourceConversation.messages.map((message, index) => ({ ...message, - id: `00000000-0000-4000-8000-${String( - 430 + index - ).padStart(12, '0')}` - })) - }) - ) - const { container } = render() - expect( - await screen.findByText('探索不同发布方案') - ).toBeInTheDocument() + id: `00000000-0000-4000-8000-${String(430 + index).padStart( + 12, + "0", + )}`, + })), + }), + ); + const { container } = render(); + expect(await screen.findByText("探索不同发布方案")).toBeInTheDocument(); - fireEvent.click( - await screen.findByLabelText('更多会话操作 方案讨论') - ) - fireEvent.click( - screen.getByRole('button', { name: '重命名会话' }) - ) - const renameInput = screen.getByLabelText('重命名会话 方案讨论') + fireEvent.click(await screen.findByLabelText("更多会话操作 方案讨论")); + fireEvent.click(screen.getByRole("button", { name: "重命名会话" })); + const renameInput = screen.getByLabelText("重命名会话 方案讨论"); fireEvent.change(renameInput, { - target: { value: '方案讨论更新' } - }) - fireEvent.submit(renameInput.closest('form')!) - expect( - await screen.findAllByText('方案讨论更新') - ).toHaveLength(2) - vi.mocked(api.conversations.saveLocal).mockClear() + target: { value: "方案讨论更新" }, + }); + fireEvent.submit(renameInput.closest("form")!); + expect(await screen.findAllByText("方案讨论更新")).toHaveLength(2); + vi.mocked(api.conversations.saveLocal).mockClear(); - fireEvent.click( - screen.getByLabelText('更多会话操作 方案讨论更新') - ) - const branchButton = screen.getByRole('button', { - name: '在新会话中继续' - }) - await waitFor(() => expect(branchButton).toBeEnabled()) - fireEvent.click(branchButton) + fireEvent.click(screen.getByLabelText("更多会话操作 方案讨论更新")); + const branchButton = screen.getByRole("button", { + name: "在新会话中继续", + }); + await waitFor(() => expect(branchButton).toBeEnabled()); + fireEvent.click(branchButton); await waitFor(() => expect(api.conversations.saveLocal).toHaveBeenCalledWith([ expect.objectContaining({ header: expect.objectContaining({ id: sourceConversationId, - title: '方案讨论更新' - }) - }) - ]) - ) + title: "方案讨论更新", + }), + }), + ]), + ); await waitFor(() => expect(api.conversations.branchLocal).toHaveBeenCalledWith({ sourceConversationId, - title: '方案讨论更新 · 分支' - }) - ) + title: "方案讨论更新 · 分支", + }), + ); expect( - vi.mocked(api.conversations.saveLocal).mock.invocationCallOrder.at(-1) + vi.mocked(api.conversations.saveLocal).mock.invocationCallOrder.at(-1), ).toBeLessThan( - vi.mocked(api.conversations.branchLocal).mock.invocationCallOrder[0]! - ) + vi.mocked(api.conversations.branchLocal).mock.invocationCallOrder[0]!, + ); expect( - await screen.findAllByLabelText( - '分支会话，来源：方案讨论更新' - ) - ).toHaveLength(2) - const sidebar = container.querySelector('.conversation-list') + await screen.findAllByLabelText("分支会话，来源：方案讨论更新"), + ).toHaveLength(2); + const sidebar = container.querySelector(".conversation-list"); const badge = within(sidebar as HTMLElement).getByLabelText( - '分支会话，来源：方案讨论更新' - ) - expect(badge).toHaveClass('conversation-branch-badge') - expect(badge.querySelector('.lucide-git-fork')).toBeInTheDocument() - expect( - container.querySelector('.conversation-title') - ).toHaveTextContent('方案讨论更新 · 分支') + "分支会话，来源：方案讨论更新", + ); + expect(badge).toHaveClass("conversation-branch-badge"); + expect(badge.querySelector(".lucide-git-fork")).toBeInTheDocument(); + expect(container.querySelector(".conversation-title")).toHaveTextContent( + "方案讨论更新 · 分支", + ); await waitFor(() => - expect(screen.getByLabelText('向 GoodBuddy 提问')).toHaveFocus() - ) + expect(screen.getByLabelText("向 GoodBuddy 提问")).toHaveFocus(), + ); expect( - screen.getAllByText('可以比较风险、成本和回滚路径。').length - ).toBeGreaterThan(0) - expect( - await screen.findByText('已创建并打开分支会话') - ).toBeInTheDocument() - }) + screen.getAllByText("可以比较风险、成本和回滚路径。").length, + ).toBeGreaterThan(0); + expect(await screen.findByText("已创建并打开分支会话")).toBeInTheDocument(); + }); - it('keeps the source conversation open when branch persistence fails', async () => { - const sourceConversationId = - '00000000-0000-4000-8000-000000000441' + it("keeps the source conversation open when branch persistence fails", async () => { + const sourceConversationId = "00000000-0000-4000-8000-000000000441"; vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: sourceConversationId, projectId, - title: '保留的来源会话', + title: "保留的来源会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000442', - role: 'user', - content: '不能丢失的来源内容', + id: "00000000-0000-4000-8000-000000000442", + role: "user", + content: "不能丢失的来源内容", createdAt: 1_775_000_000_000, - state: 'complete' - } - ] - } - ]) + state: "complete", + }, + ], + }, + ]); vi.mocked(api.conversations.branchLocal).mockRejectedValueOnce( - new Error('分支数据写入失败') - ) - const { container } = render() - expect( - await screen.findByText('不能丢失的来源内容') - ).toBeInTheDocument() + new Error("分支数据写入失败"), + ); + const { container } = render(); + expect(await screen.findByText("不能丢失的来源内容")).toBeInTheDocument(); - const actionTrigger = await screen.findByLabelText( - '更多会话操作 保留的来源会话' - ) - fireEvent.click(actionTrigger) - const branchButton = screen.getByRole('button', { - name: '在新会话中继续' - }) - await waitFor(() => expect(branchButton).toBeEnabled()) - fireEvent.click(branchButton) + const actionTrigger = + await screen.findByLabelText("更多会话操作 保留的来源会话"); + fireEvent.click(actionTrigger); + const branchButton = screen.getByRole("button", { + name: "在新会话中继续", + }); + await waitFor(() => expect(branchButton).toBeEnabled()); + fireEvent.click(branchButton); + expect(await screen.findByText("分支数据写入失败")).toBeInTheDocument(); + expect(container.querySelector(".conversation-title")).toHaveTextContent( + "保留的来源会话", + ); + expect(screen.getByText("不能丢失的来源内容")).toBeInTheDocument(); expect( - await screen.findByText('分支数据写入失败') - ).toBeInTheDocument() - expect( - container.querySelector('.conversation-title') - ).toHaveTextContent('保留的来源会话') - expect( - screen.getByText('不能丢失的来源内容') - ).toBeInTheDocument() - expect( - screen.queryByLabelText(/^分支会话，来源：/u) - ).not.toBeInTheDocument() - await waitFor(() => expect(actionTrigger).toHaveFocus()) - }) + screen.queryByLabelText(/^分支会话，来源：/u), + ).not.toBeInTheDocument(); + await waitFor(() => expect(actionTrigger).toHaveFocus()); + }); - it('explains why a queued conversation cannot be branched', async () => { - const sourceConversationId = - '00000000-0000-4000-8000-000000000451' + it("explains why a queued conversation cannot be branched", async () => { + const sourceConversationId = "00000000-0000-4000-8000-000000000451"; vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: sourceConversationId, projectId, - title: '等待队列的会话', + title: "等待队列的会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000452', - role: 'user', - content: '还有待发送内容', + id: "00000000-0000-4000-8000-000000000452", + role: "user", + content: "还有待发送内容", createdAt: 1_775_000_000_000, - state: 'complete' - } - ] - } - ]) + state: "complete", + }, + ], + }, + ]); vi.mocked(api.conversationQueue.list).mockResolvedValue([ { - id: '00000000-0000-4000-8000-000000000453', + id: "00000000-0000-4000-8000-000000000453", conversationId: sourceConversationId, - source: 'user', - label: '待发送', - createdAt: '2026-08-20T08:00:00.000Z' - } - ]) - render() - expect( - await screen.findByText('还有待发送内容') - ).toBeInTheDocument() + source: "user", + label: "待发送", + createdAt: "2026-08-20T08:00:00.000Z", + }, + ]); + render(); + expect(await screen.findByText("还有待发送内容")).toBeInTheDocument(); fireEvent.click( - await screen.findByLabelText('更多会话操作 等待队列的会话') - ) - const branchButton = screen.getByRole('button', { - name: '在新会话中继续' - }) + await screen.findByLabelText("更多会话操作 等待队列的会话"), + ); + const branchButton = screen.getByRole("button", { + name: "在新会话中继续", + }); await waitFor(() => - expect(branchButton).toHaveAttribute('aria-disabled', 'true') - ) + expect(branchButton).toHaveAttribute("aria-disabled", "true"), + ); expect(branchButton).toHaveAccessibleDescription( - '会话运行或待发送内容处理完后才能创建分支' - ) - fireEvent.click(branchButton) + "会话运行或待发送内容处理完后才能创建分支", + ); + fireEvent.click(branchButton); await waitFor(() => expect( - screen.getAllByText( - '会话运行或待发送内容处理完后才能创建分支' - ) - ).toHaveLength(2) - ) - expect(api.conversations.branchLocal).not.toHaveBeenCalled() - }) + screen.getAllByText("会话运行或待发送内容处理完后才能创建分支"), + ).toHaveLength(2), + ); + expect(api.conversations.branchLocal).not.toHaveBeenCalled(); + }); - it('offers a floating control when more messages remain below', async () => { + it("offers a floating control when more messages remain below", async () => { vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000401', + id: "00000000-0000-4000-8000-000000000401", projectId, - title: '长会话', + title: "长会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000402', - role: 'assistant', - content: '较早的会话内容', + id: "00000000-0000-4000-8000-000000000402", + role: "assistant", + content: "较早的会话内容", createdAt: 1_775_000_000_000, - state: 'complete' - } - ] - } - ]) - const { container } = render() + state: "complete", + }, + ], + }, + ]); + const { container } = render(); - expect(await screen.findByText('较早的会话内容')).toBeInTheDocument() - const chat = container.querySelector('.chat') + expect(await screen.findByText("较早的会话内容")).toBeInTheDocument(); + const chat = container.querySelector(".chat"); if (!chat) { - throw new Error('Missing chat scroll container') + throw new Error("Missing chat scroll container"); } Object.defineProperties(chat, { clientHeight: { configurable: true, value: 400 }, scrollHeight: { configurable: true, value: 1_200 }, - scrollTop: { configurable: true, writable: true, value: 100 } - }) - const scrollTo = vi.fn() - chat.scrollTo = scrollTo + scrollTop: { configurable: true, writable: true, value: 100 }, + }); + const scrollTo = vi.fn(); + chat.scrollTo = scrollTo; - fireEvent.scroll(chat) - const scrollButton = screen.getByRole('button', { - name: '到底部' - }) - expect(scrollButton).toHaveAttribute( - 'aria-controls', - 'chat-message-list' - ) - expect(scrollButton).toHaveAttribute('title', '到底部') - expect(scrollButton).toHaveTextContent('') + fireEvent.scroll(chat); + const scrollButton = screen.getByRole("button", { + name: "到底部", + }); + expect(scrollButton).toHaveAttribute("aria-controls", "chat-message-list"); + expect(scrollButton).toHaveAttribute("title", "到底部"); + expect(scrollButton).toHaveTextContent(""); - fireEvent.click(scrollButton) + fireEvent.click(scrollButton); expect(scrollTo).toHaveBeenLastCalledWith({ top: 1_200, - behavior: 'smooth' - }) + behavior: "smooth", + }); - chat.scrollTop = 750 - fireEvent.scroll(chat) + chat.scrollTop = 750; + fireEvent.scroll(chat); expect( - screen.queryByRole('button', { name: '到底部' }) - ).not.toBeInTheDocument() - }) + screen.queryByRole("button", { name: "到底部" }), + ).not.toBeInTheDocument(); + }); - it('keeps a newly opened persisted conversation at the bottom while its content settles', async () => { - let resizeCallback: ResizeObserverCallback | undefined - const disconnect = vi.fn() + it("keeps a newly opened persisted conversation at the bottom while its content settles", async () => { + let resizeCallback: ResizeObserverCallback | undefined; + const disconnect = vi.fn(); vi.stubGlobal( - 'ResizeObserver', + "ResizeObserver", class { constructor(callback: ResizeObserverCallback) { - resizeCallback = callback + resizeCallback = callback; } observe(): void {} unobserve(): void {} disconnect(): void { - disconnect() + disconnect(); } - } - ) + }, + ); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000417', + id: "00000000-0000-4000-8000-000000000417", projectId, - title: '已有会话', + title: "已有会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000418', - role: 'user', - content: '最后一个用户问题', + id: "00000000-0000-4000-8000-000000000418", + role: "user", + content: "最后一个用户问题", createdAt: 1_775_000_000_000, - state: 'complete' + state: "complete", }, { - id: '00000000-0000-4000-8000-000000000419', - role: 'assistant', - content: '加载后会继续完成布局的回复', + id: "00000000-0000-4000-8000-000000000419", + role: "assistant", + content: "加载后会继续完成布局的回复", createdAt: 1_775_000_000_001, - state: 'complete' - } - ] - } - ]) - const { container } = render() + state: "complete", + }, + ], + }, + ]); + const { container } = render(); expect( - await screen.findByText('加载后会继续完成布局的回复') - ).toBeInTheDocument() - const chat = container.querySelector('.chat') + await screen.findByText("加载后会继续完成布局的回复"), + ).toBeInTheDocument(); + const chat = container.querySelector(".chat"); if (!chat || !resizeCallback) { - throw new Error('Missing chat resize observer') + throw new Error("Missing chat resize observer"); } - let scrollHeight = 1_200 + let scrollHeight = 1_200; Object.defineProperties(chat, { clientHeight: { configurable: true, value: 400 }, scrollHeight: { configurable: true, - get: () => scrollHeight + get: () => scrollHeight, }, - scrollTop: { configurable: true, writable: true, value: 800 } - }) - const scrollTo = vi.fn() - chat.scrollTo = scrollTo + scrollTop: { configurable: true, writable: true, value: 800 }, + }); + const scrollTo = vi.fn(); + chat.scrollTo = scrollTo; - scrollHeight = 1_800 - act(() => resizeCallback?.([], {} as ResizeObserver)) + scrollHeight = 1_800; + act(() => resizeCallback?.([], {} as ResizeObserver)); expect(scrollTo).toHaveBeenLastCalledWith({ top: 1_800, - behavior: 'auto' - }) + behavior: "auto", + }); - chat.scrollTop = 100 - fireEvent.scroll(chat) - scrollTo.mockClear() - scrollHeight = 2_200 - act(() => resizeCallback?.([], {} as ResizeObserver)) - expect(scrollTo).not.toHaveBeenCalled() - }) + chat.scrollTop = 100; + fireEvent.scroll(chat); + scrollTo.mockClear(); + scrollHeight = 2_200; + act(() => resizeCallback?.([], {} as ResizeObserver)); + expect(scrollTo).not.toHaveBeenCalled(); + }); - it('renders the latest 80 messages and preserves scroll when revealing earlier messages', async () => { + it("renders the latest 80 messages and preserves scroll when revealing earlier messages", async () => { vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000421', + id: "00000000-0000-4000-8000-000000000421", projectId, - title: '超长会话', + title: "超长会话", updatedAt: 1_775_000_000_000, messages: Array.from({ length: 161 }, (_, index) => ({ - id: `00000000-0000-4000-8000-${String(index).padStart(12, '0')}`, - role: index % 2 === 0 ? ('user' as const) : ('assistant' as const), - content: `历史消息 ${String(index).padStart(3, '0')}`, + id: `00000000-0000-4000-8000-${String(index).padStart(12, "0")}`, + role: index % 2 === 0 ? ("user" as const) : ("assistant" as const), + content: `历史消息 ${String(index).padStart(3, "0")}`, createdAt: 1_775_000_000_000 + index, - state: 'complete' as const - })) - } - ]) - const { container } = render() + state: "complete" as const, + })), + }, + ]); + const { container } = render(); - expect(await screen.findByText('历史消息 160')).toBeInTheDocument() - expect(screen.queryByText('历史消息 080')).not.toBeInTheDocument() - expect(container.querySelectorAll('.message')).toHaveLength(80) - const chat = container.querySelector('.chat') + expect(await screen.findByText("历史消息 160")).toBeInTheDocument(); + expect(screen.queryByText("历史消息 080")).not.toBeInTheDocument(); + expect(container.querySelectorAll(".message")).toHaveLength(80); + const chat = container.querySelector(".chat"); if (!chat) { - throw new Error('Missing chat scroll container') + throw new Error("Missing chat scroll container"); } Object.defineProperties(chat, { clientHeight: { configurable: true, value: 400 }, scrollHeight: { configurable: true, - get: () => container.querySelectorAll('.message').length * 10 + get: () => container.querySelectorAll(".message").length * 10, }, scrollTop: { configurable: true, writable: true, - value: 125 - } - }) + value: 125, + }, + }); fireEvent.click( - screen.getByRole('button', { - name: '加载更早的消息（还剩 81 条）' - }) - ) + screen.getByRole("button", { + name: "加载更早的消息（还剩 81 条）", + }), + ); - expect(container.querySelectorAll('.message')).toHaveLength(160) - expect(screen.getByText('历史消息 001')).toBeInTheDocument() - expect(screen.queryByText('历史消息 000')).not.toBeInTheDocument() - expect(chat.scrollTop).toBe(925) + expect(container.querySelectorAll(".message")).toHaveLength(160); + expect(screen.getByText("历史消息 001")).toBeInTheDocument(); + expect(screen.queryByText("历史消息 000")).not.toBeInTheDocument(); + expect(chat.scrollTop).toBe(925); fireEvent.click( - screen.getByRole('button', { - name: '加载更早的消息（还剩 1 条）' - }) - ) - expect(container.querySelectorAll('.message')).toHaveLength(161) - expect(screen.getByText('历史消息 000')).toBeInTheDocument() + screen.getByRole("button", { + name: "加载更早的消息（还剩 1 条）", + }), + ); + expect(container.querySelectorAll(".message")).toHaveLength(161); + expect(screen.getByText("历史消息 000")).toBeInTheDocument(); expect( - screen.queryByRole('button', { name: /加载更早的消息/u }) - ).not.toBeInTheDocument() - expect( - screen.getByText('历史消息 000').closest('article') - ).toHaveFocus() - }) + screen.queryByRole("button", { name: /加载更早的消息/u }), + ).not.toBeInTheDocument(); + expect(screen.getByText("历史消息 000").closest("article")).toHaveFocus(); + }); - it('keeps the reader position while a response continues below', async () => { - render() - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '继续生成较长回复' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + it("keeps the reader position while a response continues below", async () => { + render(); + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "继续生成较长回复" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } await act( () => - new Promise((resolve) => - requestAnimationFrame(() => resolve()) - ) - ) + new Promise((resolve) => requestAnimationFrame(() => resolve())), + ); - const chat = document.querySelector('.chat') + const chat = document.querySelector(".chat"); if (!chat) { - throw new Error('Missing chat scroll container') + throw new Error("Missing chat scroll container"); } Object.defineProperties(chat, { clientHeight: { configurable: true, value: 400 }, scrollHeight: { configurable: true, value: 1_200 }, - scrollTop: { configurable: true, writable: true, value: 100 } - }) - const scrollTo = vi.fn() - chat.scrollTo = scrollTo - fireEvent.scroll(chat) + scrollTop: { configurable: true, writable: true, value: 100 }, + }); + const scrollTo = vi.fn(); + chat.scrollTo = scrollTo; + fireEvent.scroll(chat); act(() => { agentListener?.({ requestId: request.requestId, - type: 'text', - delta: '新增的回复内容' - }) - }) - expect(await screen.findByText('新增的回复内容')).toBeInTheDocument() + type: "text", + delta: "新增的回复内容", + }); + }); + expect(await screen.findByText("新增的回复内容")).toBeInTheDocument(); await act( () => - new Promise((resolve) => - requestAnimationFrame(() => resolve()) - ) - ) + new Promise((resolve) => requestAnimationFrame(() => resolve())), + ); - expect(scrollTo).not.toHaveBeenCalled() - expect( - screen.getByRole('button', { name: '到底部' }) - ).toBeInTheDocument() - }) + expect(scrollTo).not.toHaveBeenCalled(); + expect(screen.getByRole("button", { name: "到底部" })).toBeInTheDocument(); + }); - it('restores the reader position after activity continues on another page', async () => { - const { container } = render() - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '离开页面后继续生成' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + it("restores the reader position after activity continues on another page", async () => { + const { container } = render(); + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "离开页面后继续生成" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } const conversationList = - container.querySelector('.conversation-list') + container.querySelector(".conversation-list"); if (!conversationList) { - throw new Error('Missing conversation list') + throw new Error("Missing conversation list"); } - const activeIndicator = within(conversationList).getByRole('status', { - name: '会话正在活动' - }) - expect(activeIndicator).toHaveClass( - 'conversation-activity-indicator' - ) - expect(activeIndicator.closest('.conversation-row')).toHaveTextContent( - '离开页面后继续生成' - ) + const activeIndicator = within(conversationList).getByRole("status", { + name: "会话正在活动", + }); + expect(activeIndicator).toHaveClass("conversation-activity-indicator"); + expect(activeIndicator.closest(".conversation-row")).toHaveTextContent( + "离开页面后继续生成", + ); await act( () => - new Promise((resolve) => - requestAnimationFrame(() => resolve()) - ) - ) + new Promise((resolve) => requestAnimationFrame(() => resolve())), + ); - const chat = container.querySelector('.chat') + const chat = container.querySelector(".chat"); if (!chat) { - throw new Error('Missing chat scroll container') + throw new Error("Missing chat scroll container"); } Object.defineProperties(chat, { clientHeight: { configurable: true, value: 400 }, scrollHeight: { configurable: true, value: 1_200 }, - scrollTop: { configurable: true, writable: true, value: 175 } - }) - fireEvent.scroll(chat) + scrollTop: { configurable: true, writable: true, value: 175 }, + }); + fireEvent.scroll(chat); - fireEvent.click(screen.getByRole('button', { name: '知识库' })) - expect(chat).toBeInTheDocument() - expect(chat.closest('[data-route="chat"]')).toHaveAttribute('hidden') + fireEvent.click(screen.getByRole("button", { name: "知识库" })); + expect(chat).toBeInTheDocument(); + expect(chat.closest('[data-route="chat"]')).toHaveAttribute("hidden"); expect( - within(conversationList).getByRole('status', { - name: '会话正在活动' - }) - ).toBeInTheDocument() + within(conversationList).getByRole("status", { + name: "会话正在活动", + }), + ).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'text', - delta: '后台新增的回复内容' - }) - }) + type: "text", + delta: "后台新增的回复内容", + }); + }); act(() => { agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); await waitFor(() => expect( - within(conversationList).queryByRole('status', { - name: '会话正在活动' - }) - ).not.toBeInTheDocument() - ) + within(conversationList).queryByRole("status", { + name: "会话正在活动", + }), + ).not.toBeInTheDocument(), + ); - fireEvent.click(screen.getByRole('button', { name: '对话' })) - expect(await screen.findByText('后台新增的回复内容')).toBeInTheDocument() - const restoredChat = container.querySelector('.chat') - expect(restoredChat).toBe(chat) - expect(restoredChat?.scrollTop).toBe(175) - expect( - screen.getByRole('button', { name: '到底部' }) - ).toBeInTheDocument() - }) + fireEvent.click(screen.getByRole("button", { name: "对话" })); + expect(await screen.findByText("后台新增的回复内容")).toBeInTheDocument(); + const restoredChat = container.querySelector(".chat"); + expect(restoredChat).toBe(chat); + expect(restoredChat?.scrollTop).toBe(175); + expect(screen.getByRole("button", { name: "到底部" })).toBeInTheDocument(); + }); - it('keeps each conversation history window and reader position', async () => { - const firstConversationId = - '00000000-0000-4000-8000-000000000461' - const secondConversationId = - '00000000-0000-4000-8000-000000000462' + it("keeps each conversation history window and reader position", async () => { + const firstConversationId = "00000000-0000-4000-8000-000000000461"; + const secondConversationId = "00000000-0000-4000-8000-000000000462"; const draftAttachment = { - id: '00000000-0000-4000-8000-000000000463', - name: '第一段草稿附件.md', + id: "00000000-0000-4000-8000-000000000463", + name: "第一段草稿附件.md", size: 1_024, - preview: '会话级草稿附件', - kind: 'text' as const - } - vi.mocked(api.context.selectFiles).mockResolvedValueOnce([ - draftAttachment - ]) + preview: "会话级草稿附件", + kind: "text" as const, + }; + vi.mocked(api.context.selectFiles).mockResolvedValueOnce([draftAttachment]); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: firstConversationId, projectId, - title: '第一段长会话', + title: "第一段长会话", updatedAt: 1_775_000_000_002, messages: Array.from({ length: 161 }, (_, index) => ({ - id: `00000000-0000-4000-8100-${String(index).padStart(12, '0')}`, - role: index % 2 === 0 ? ('user' as const) : ('assistant' as const), - content: `第一段历史 ${String(index).padStart(3, '0')}`, - reasoning: index === 159 ? '需要保留展开状态' : undefined, + id: `00000000-0000-4000-8100-${String(index).padStart(12, "0")}`, + role: index % 2 === 0 ? ("user" as const) : ("assistant" as const), + content: `第一段历史 ${String(index).padStart(3, "0")}`, + reasoning: index === 159 ? "需要保留展开状态" : undefined, createdAt: 1_775_000_000_000 + index, - state: 'complete' as const - })) + state: "complete" as const, + })), }, { id: secondConversationId, projectId, - title: '第二段会话', + title: "第二段会话", updatedAt: 1_775_000_000_001, messages: [ { - id: '00000000-0000-4000-8200-000000000001', - role: 'assistant', - content: '第二段会话内容', + id: "00000000-0000-4000-8200-000000000001", + role: "assistant", + content: "第二段会话内容", createdAt: 1_775_000_000_001, - state: 'complete' - } - ] - } - ]) - const { container } = render() + state: "complete", + }, + ], + }, + ]); + const { container } = render(); - expect(await screen.findByText('第一段历史 160')).toBeInTheDocument() + expect(await screen.findByText("第一段历史 160")).toBeInTheDocument(); fireEvent.click( - screen.getByRole('button', { - name: '加载更早的消息（还剩 81 条）' - }) - ) + screen.getByRole("button", { + name: "加载更早的消息（还剩 81 条）", + }), + ); const firstPane = container.querySelector( - `[data-conversation-id="${firstConversationId}"]` - ) - expect(firstPane?.querySelectorAll('.message')).toHaveLength(160) - const firstChat = firstPane?.querySelector('.chat') + `[data-conversation-id="${firstConversationId}"]`, + ); + expect(firstPane?.querySelectorAll(".message")).toHaveLength(160); + const firstChat = firstPane?.querySelector(".chat"); if (!firstChat) { - throw new Error('Missing first chat scroll container') + throw new Error("Missing first chat scroll container"); } - const reasoningDetails = firstPane?.querySelector( - '.message-reasoning' - ) + const reasoningDetails = + firstPane?.querySelector(".message-reasoning"); if (!reasoningDetails) { - throw new Error('Missing reasoning details') + throw new Error("Missing reasoning details"); } - reasoningDetails.open = true + reasoningDetails.open = true; Object.defineProperties(firstChat, { clientHeight: { configurable: true, value: 400 }, scrollHeight: { configurable: true, value: 1_600 }, - scrollTop: { configurable: true, writable: true, value: 225 } - }) - fireEvent.scroll(firstChat) - messageRenderProbe.mockClear() - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '第一段会话草稿' } - }) - expect(messageRenderProbe).not.toHaveBeenCalled() - expect(firstPane?.querySelectorAll('.message')).toHaveLength(160) - fireEvent.click(screen.getByLabelText('添加附件')) - expect( - await screen.findByText(draftAttachment.name) - ).toBeInTheDocument() + scrollTop: { configurable: true, writable: true, value: 225 }, + }); + fireEvent.scroll(firstChat); + messageRenderProbe.mockClear(); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "第一段会话草稿" }, + }); + expect(messageRenderProbe).not.toHaveBeenCalled(); + expect(firstPane?.querySelectorAll(".message")).toHaveLength(160); + fireEvent.click(screen.getByLabelText("添加附件")); + expect(await screen.findByText(draftAttachment.name)).toBeInTheDocument(); - fireEvent.click( - screen.getByText('第二段会话').closest('button')! - ) - expect(await screen.findByText('第二段会话内容')).toBeInTheDocument() - expect(screen.getByLabelText('向 GoodBuddy 提问')).toHaveValue('') - expect( - screen.queryByText(draftAttachment.name) - ).not.toBeInTheDocument() - messageRenderProbe.mockClear() - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '第二段会话草稿' } - }) - expect(messageRenderProbe).not.toHaveBeenCalled() - expect(firstPane?.querySelectorAll('.message')).toHaveLength(160) - fireEvent.click( - screen.getByText('第一段长会话').closest('button')! - ) + fireEvent.click(screen.getByText("第二段会话").closest("button")!); + expect(await screen.findByText("第二段会话内容")).toBeInTheDocument(); + expect(screen.getByLabelText("向 GoodBuddy 提问")).toHaveValue(""); + expect(screen.queryByText(draftAttachment.name)).not.toBeInTheDocument(); + messageRenderProbe.mockClear(); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "第二段会话草稿" }, + }); + expect(messageRenderProbe).not.toHaveBeenCalled(); + expect(firstPane?.querySelectorAll(".message")).toHaveLength(160); + fireEvent.click(screen.getByText("第一段长会话").closest("button")!); - expect(await screen.findByText('第一段历史 001')).toBeInTheDocument() + expect(await screen.findByText("第一段历史 001")).toBeInTheDocument(); const restoredFirstPane = container.querySelector( - `[data-conversation-id="${firstConversationId}"]` - ) - expect(restoredFirstPane).toBe(firstPane) - expect(restoredFirstPane?.querySelectorAll('.message')).toHaveLength( - 160 - ) - expect(restoredFirstPane?.querySelector('.chat')).toBe(firstChat) - expect(firstChat.scrollTop).toBe(225) - expect(reasoningDetails).toHaveAttribute('open') - expect(screen.getByLabelText('向 GoodBuddy 提问')).toHaveValue( - '第一段会话草稿' - ) - expect(screen.getByText(draftAttachment.name)).toBeInTheDocument() - }) + `[data-conversation-id="${firstConversationId}"]`, + ); + expect(restoredFirstPane).toBe(firstPane); + expect(restoredFirstPane?.querySelectorAll(".message")).toHaveLength(160); + expect(restoredFirstPane?.querySelector(".chat")).toBe(firstChat); + expect(firstChat.scrollTop).toBe(225); + expect(reasoningDetails).toHaveAttribute("open"); + expect(screen.getByLabelText("向 GoodBuddy 提问")).toHaveValue( + "第一段会话草稿", + ); + expect(screen.getByText(draftAttachment.name)).toBeInTheDocument(); + }); - it('requires an accessible confirmation before permanently deleting a conversation', async () => { - render() - const menuTrigger = screen.getByLabelText( - '更多会话操作 新对话' - ) - fireEvent.click(menuTrigger) - const deleteTrigger = screen.getByRole('button', { - name: '删除对话 新对话' - }) - fireEvent.click(deleteTrigger) + it("requires an accessible confirmation before permanently deleting a conversation", async () => { + render(); + const menuTrigger = screen.getByLabelText("更多会话操作 新对话"); + fireEvent.click(menuTrigger); + const deleteTrigger = screen.getByRole("button", { + name: "删除对话 新对话", + }); + fireEvent.click(deleteTrigger); - const dialog = screen.getByRole('alertdialog', { - name: '确认永久删除对话 新对话' - }) - expect(dialog).toHaveTextContent('将永久删除此会话的全部内容') + const dialog = screen.getByRole("alertdialog", { + name: "确认永久删除对话 新对话", + }); + expect(dialog).toHaveTextContent("将永久删除此会话的全部内容"); expect(dialog).toHaveTextContent( - '如果此会话有正在运行的任务，也会同时停止' - ) - expect(dialog).toHaveTextContent('此操作不可恢复') - const cancel = screen.getByRole('button', { - name: '取消删除对话 新对话' - }) - expect(cancel).toHaveFocus() + "如果此会话有正在运行的任务，也会同时停止", + ); + expect(dialog).toHaveTextContent("此操作不可恢复"); + const cancel = screen.getByRole("button", { + name: "取消删除对话 新对话", + }); + expect(cancel).toHaveFocus(); - expect(fireEvent.keyDown(dialog, { key: 'Tab' })).toBe(true) - expect(cancel).toHaveFocus() - fireEvent.keyDown(dialog, { key: 'Escape' }) + expect(fireEvent.keyDown(dialog, { key: "Tab" })).toBe(true); + expect(cancel).toHaveFocus(); + fireEvent.keyDown(dialog, { key: "Escape" }); await waitFor(() => expect( - screen.getByRole('button', { name: '删除对话 新对话' }) - ).toHaveFocus() - ) + screen.getByRole("button", { name: "删除对话 新对话" }), + ).toHaveFocus(), + ); expect( - screen.queryByRole('alertdialog', { - name: '确认永久删除对话 新对话' - }) - ).not.toBeInTheDocument() + screen.queryByRole("alertdialog", { + name: "确认永久删除对话 新对话", + }), + ).not.toBeInTheDocument(); + fireEvent.click(screen.getByRole("button", { name: "删除对话 新对话" })); fireEvent.click( - screen.getByRole('button', { name: '删除对话 新对话' }) - ) - fireEvent.click( - screen.getByRole('button', { - name: '确认永久删除对话 新对话' - }) - ) + screen.getByRole("button", { + name: "确认永久删除对话 新对话", + }), + ); await waitFor(() => expect( - screen.queryByRole('alertdialog', { - name: '确认永久删除对话 新对话' - }) - ).not.toBeInTheDocument() - ) + screen.queryByRole("alertdialog", { + name: "确认永久删除对话 新对话", + }), + ).not.toBeInTheDocument(), + ); expect(api.conversations.deleteLocal).toHaveBeenCalledWith( - expect.any(String) - ) - }) + expect.any(String), + ); + }); - it('keeps a local conversation visible when deleting its persisted record fails', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000431' - const title = '删除失败时保留的本地会话' + it("keeps a local conversation visible when deleting its persisted record fails", async () => { + const conversationId = "00000000-0000-4000-8000-000000000431"; + const title = "删除失败时保留的本地会话"; vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: conversationId, @@ -3476,1066 +3277,991 @@ describe('App', () => { updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000432', - role: 'assistant', - content: '需要保留的消息', + id: "00000000-0000-4000-8000-000000000432", + role: "assistant", + content: "需要保留的消息", createdAt: 1_775_000_000_000, - state: 'complete' - } - ] - } - ]) + state: "complete", + }, + ], + }, + ]); vi.mocked(api.conversations.deleteLocal).mockRejectedValueOnce( - new Error('delete failed') - ) - render() + new Error("delete failed"), + ); + render(); + fireEvent.click(await screen.findByLabelText(`更多会话操作 ${title}`)); + fireEvent.click(screen.getByRole("button", { name: `删除对话 ${title}` })); fireEvent.click( - await screen.findByLabelText(`更多会话操作 ${title}`) - ) - fireEvent.click( - screen.getByRole('button', { name: `删除对话 ${title}` }) - ) - fireEvent.click( - screen.getByRole('button', { - name: `确认永久删除对话 ${title}` - }) - ) + screen.getByRole("button", { + name: `确认永久删除对话 ${title}`, + }), + ); await waitFor(() => expect(api.conversations.deleteLocal).toHaveBeenCalledWith( - conversationId - ) - ) + conversationId, + ), + ); expect( - await screen.findByText('删除本地会话失败，已保留当前对话') - ).toBeInTheDocument() - expect(screen.getByText('需要保留的消息')).toBeInTheDocument() - const dialog = screen.getByRole('alertdialog', { - name: `确认永久删除对话 ${title}` - }) + await screen.findByText("删除本地会话失败，已保留当前对话"), + ).toBeInTheDocument(); + expect(screen.getByText("需要保留的消息")).toBeInTheDocument(); + const dialog = screen.getByRole("alertdialog", { + name: `确认永久删除对话 ${title}`, + }); + expect(dialog).toBeInTheDocument(); + fireEvent.keyDown(dialog, { key: "Escape" }); expect( - dialog - ).toBeInTheDocument() - fireEvent.keyDown(dialog, { key: 'Escape' }) - expect( - await screen.findByLabelText(`更多会话操作 ${title}`) - ).toBeInTheDocument() - }) + await screen.findByLabelText(`更多会话操作 ${title}`), + ).toBeInTheDocument(); + }); - it('keeps a conversation when cancelling its active task fails', async () => { + it("keeps a conversation when cancelling its active task fails", async () => { vi.mocked(api.agent.cancel).mockRejectedValueOnce( - new Error('cancel failed') - ) - render() - const title = '取消失败时保留会话' + new Error("cancel failed"), + ); + render(); + const title = "取消失败时保留会话"; - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: title } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: title }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); - fireEvent.click( - await screen.findByLabelText(`更多会话操作 ${title}`) - ) - fireEvent.click( - screen.getByRole('button', { name: `删除对话 ${title}` }) - ) - const confirm = screen.getByRole('button', { - name: `确认永久删除对话 ${title}` - }) - fireEvent.click(confirm) + fireEvent.click(await screen.findByLabelText(`更多会话操作 ${title}`)); + fireEvent.click(screen.getByRole("button", { name: `删除对话 ${title}` })); + const confirm = screen.getByRole("button", { + name: `确认永久删除对话 ${title}`, + }); + fireEvent.click(confirm); - await waitFor(() => - expect(api.agent.cancel).toHaveBeenCalledOnce() - ) + await waitFor(() => expect(api.agent.cancel).toHaveBeenCalledOnce()); expect( - await screen.findByText('停止会话中的运行任务失败，尚未删除对话') - ).toBeInTheDocument() + await screen.findByText("停止会话中的运行任务失败，尚未删除对话"), + ).toBeInTheDocument(); expect( - screen.getByRole('alertdialog', { - name: `确认永久删除对话 ${title}` - }) - ).toBeInTheDocument() - await waitFor(() => expect(confirm).toBeEnabled()) + screen.getByRole("alertdialog", { + name: `确认永久删除对话 ${title}`, + }), + ).toBeInTheDocument(); + await waitFor(() => expect(confirm).toBeEnabled()); - vi.mocked(api.agent.cancel).mockResolvedValueOnce() - fireEvent.click(confirm) + vi.mocked(api.agent.cancel).mockResolvedValueOnce(); + fireEvent.click(confirm); await waitFor(() => expect( - screen.queryByRole('alertdialog', { - name: `确认永久删除对话 ${title}` - }) - ).not.toBeInTheDocument() - ) - }) + screen.queryByRole("alertdialog", { + name: `确认永久删除对话 ${title}`, + }), + ).not.toBeInTheDocument(), + ); + }); - it('renders streamed reasoning, text, and tools in event order', async () => { - render() + it("renders streamed reasoning, text, and tools in event order", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '帮我分析项目' } - }) - await waitFor(() => expect(screen.getByLabelText('发送')).toBeEnabled()) - fireEvent.click(screen.getByLabelText('发送')) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "帮我分析项目" }, + }); + await waitFor(() => expect(screen.getByLabelText("发送")).toBeEnabled()); + fireEvent.click(screen.getByLabelText("发送")); - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] - expect(request?.prompt).toBe('帮我分析项目') + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; + expect(request?.prompt).toBe("帮我分析项目"); expect(request?.runtimeSelection).toEqual({ - provider: 'model', - profileId: modelProfileId - }) + provider: "model", + profileId: modelProfileId, + }); expect( screen - .getByText('正在连接 Agent Runtime') - .querySelector('.message__status-dot') - ).toHaveClass('message__status-dot--active') + .getByText("正在连接 Agent Runtime") + .querySelector(".message__status-dot"), + ).toHaveClass("message__status-dot--active"); const userMessage = screen - .getAllByText('帮我分析项目') - .map((element) => element.closest('article')) - .find((element) => element?.classList.contains('message--user')) - expect(userMessage).toHaveClass('message--user') + .getAllByText("帮我分析项目") + .map((element) => element.closest("article")) + .find((element) => element?.classList.contains("message--user")); + expect(userMessage).toHaveClass("message--user"); act(() => { if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } agentListener?.({ requestId: request.requestId, - type: 'text', - delta: '这是' - }) + type: "text", + delta: "这是", + }); agentListener?.({ requestId: request.requestId, - type: 'text', - delta: '回答内容' - }) + type: "text", + delta: "回答内容", + }); agentListener?.({ requestId: request.requestId, - type: 'reasoning', - delta: '先检查项目结构' - }) - }) + type: "reasoning", + delta: "先检查项目结构", + }); + }); - expect(await screen.findByText('这是回答内容')).toBeInTheDocument() - const streamingReasoning = screen - .getByText('正在推理') - .closest('details') - expect(streamingReasoning).toHaveAttribute('open') - expect(screen.getByText('先检查项目结构')).toBeInTheDocument() - const reasoningContent = - streamingReasoning!.querySelector( - '.message-reasoning__content' - ) + expect(await screen.findByText("这是回答内容")).toBeInTheDocument(); + const streamingReasoning = screen.getByText("正在推理").closest("details"); + expect(streamingReasoning).toHaveAttribute("open"); + expect(screen.getByText("先检查项目结构")).toBeInTheDocument(); + const reasoningContent = streamingReasoning!.querySelector( + ".message-reasoning__content", + ); if (!reasoningContent) { - throw new Error('Missing reasoning content') + throw new Error("Missing reasoning content"); } - const reasoningScrollTo = vi.fn() - reasoningContent.scrollTo = reasoningScrollTo - Object.defineProperty(reasoningContent, 'scrollHeight', { + const reasoningScrollTo = vi.fn(); + reasoningContent.scrollTo = reasoningScrollTo; + Object.defineProperty(reasoningContent, "scrollHeight", { configurable: true, - value: 640 - }) + value: 640, + }); act(() => { if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } agentListener?.({ requestId: request.requestId, - type: 'reasoning', - delta: '，再确认依赖' - }) - }) + type: "reasoning", + delta: "，再确认依赖", + }); + }); expect(reasoningScrollTo).toHaveBeenLastCalledWith({ top: 640, - behavior: 'auto' - }) - expect(screen.getByText('先检查项目结构，再确认依赖')).toBeInTheDocument() + behavior: "auto", + }); + expect(screen.getByText("先检查项目结构，再确认依赖")).toBeInTheDocument(); act(() => { if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } agentListener?.({ requestId: request.requestId, - type: 'tool', - callId: 'call-1', - name: 'read', - state: 'running', - summary: 'OpenCode 工具：read', - input: '{"path":"README.md"}' - }) - agentListener?.({ - requestId: request.requestId, - type: 'tool', - callId: 'call-2', - name: 'grep', - state: 'completed', - summary: 'OpenCode 工具：grep', - input: '{"pattern":"runtime"}', - output: 'src/main/agent/runtime.ts' - }) - agentListener?.({ - requestId: request.requestId, - type: 'reasoning', - delta: '再检查关键文件' - }) - agentListener?.({ - requestId: request.requestId, - type: 'text', - delta: '最终结论' - }) - agentListener?.({ - requestId: request.requestId, - type: 'tool', - callId: 'call-1', - name: 'read', - state: 'completed', - summary: 'OpenCode 工具：read', + type: "tool", + callId: "call-1", + name: "read", + state: "running", + summary: "OpenCode 工具：read", input: '{"path":"README.md"}', - output: - 'README contents\nAuthorization: Bearer visible-token' - }) - }) + }); + agentListener?.({ + requestId: request.requestId, + type: "tool", + callId: "call-2", + name: "grep", + state: "completed", + summary: "OpenCode 工具：grep", + input: '{"pattern":"runtime"}', + output: "src/main/agent/runtime.ts", + }); + agentListener?.({ + requestId: request.requestId, + type: "reasoning", + delta: "再检查关键文件", + }); + agentListener?.({ + requestId: request.requestId, + type: "text", + delta: "最终结论", + }); + agentListener?.({ + requestId: request.requestId, + type: "tool", + callId: "call-1", + name: "read", + state: "completed", + summary: "OpenCode 工具：read", + input: '{"path":"README.md"}', + output: "README contents\nAuthorization: Bearer visible-token", + }); + }); - const assistantArticle = screen - .getByText('最终结论') - .closest('article') + const assistantArticle = screen.getByText("最终结论").closest("article"); const orderedBlocks = [ - ...assistantArticle!.querySelectorAll('.message-blocks > *') - ].map((element) => element.textContent) + ...assistantArticle!.querySelectorAll(".message-blocks > *"), + ].map((element) => element.textContent); expect(orderedBlocks).toEqual([ - expect.stringContaining('这是回答内容'), - expect.stringContaining('先检查项目结构'), - expect.stringContaining('OpenCode 工具：read'), - expect.stringContaining('再检查关键文件'), - expect.stringContaining('最终结论') - ]) + expect.stringContaining("这是回答内容"), + expect.stringContaining("先检查项目结构"), + expect.stringContaining("OpenCode 工具：read"), + expect.stringContaining("再检查关键文件"), + expect.stringContaining("最终结论"), + ]); + expect(screen.getAllByText("OpenCode 工具：read")).toHaveLength(1); expect( - screen.getAllByText('OpenCode 工具：read') - ).toHaveLength(1) - expect( - screen.getByRole('region', { name: '工具执行，共 2 项' }) - ).toBeInTheDocument() - const readTool = screen.getByText('read').closest('details') + screen.getByRole("region", { name: "工具执行，共 2 项" }), + ).toBeInTheDocument(); + const readTool = screen.getByText("read").closest("details"); const rawToolOutput = - 'README contents\nAuthorization: Bearer visible-token' - expect(readTool).not.toHaveAttribute('open') + "README contents\nAuthorization: Bearer visible-token"; + expect(readTool).not.toHaveAttribute("open"); expect( - screen.getByText( - (_, element) => element?.textContent === rawToolOutput - ) - ).not.toBeVisible() - fireEvent.click(screen.getByText('read').closest('summary')!) - expect(readTool).toHaveAttribute('open') - expect(within(readTool!).getByText('调用参数')).toBeVisible() + screen.getByText((_, element) => element?.textContent === rawToolOutput), + ).not.toBeVisible(); + fireEvent.click(screen.getByText("read").closest("summary")!); + expect(readTool).toHaveAttribute("open"); + expect(within(readTool!).getByText("调用参数")).toBeVisible(); expect( within(readTool!).getByText( - (_, element) => element?.textContent === rawToolOutput - ) - ).toBeVisible() - const activeReasoning = screen.getAllByText('正在推理') - expect(activeReasoning).toHaveLength(2) + (_, element) => element?.textContent === rawToolOutput, + ), + ).toBeVisible(); + const activeReasoning = screen.getAllByText("正在推理"); + expect(activeReasoning).toHaveLength(2); for (const reasoning of activeReasoning) { - expect(reasoning.closest('details')).toHaveAttribute('open') + expect(reasoning.closest("details")).toHaveAttribute("open"); } act(() => { if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); - const completedReasoning = await screen.findAllByText('推理过程') - expect(completedReasoning).toHaveLength(2) + const completedReasoning = await screen.findAllByText("推理过程"); + expect(completedReasoning).toHaveLength(2); for (const reasoning of completedReasoning) { - expect(reasoning.closest('details')).not.toHaveAttribute('open') + expect(reasoning.closest("details")).not.toHaveAttribute("open"); } - expect(screen.getByText('项目：默认项目')).toHaveClass('scope-badge') - }) + expect(screen.getByText("项目：默认项目")).toHaveClass("scope-badge"); + }); - it('keeps every activity detail in a long tool chain', async () => { - render() + it("keeps every activity detail in a long tool chain", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '执行长工具链' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "执行长工具链" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { for (let index = 0; index < 501; index += 1) { agentListener?.({ requestId: request.requestId, - type: 'tool', + type: "tool", callId: `call-${index + 1}`, - name: index === 500 ? `read${'长'.repeat(300)}` : 'read', - state: 'completed', + name: index === 500 ? `read${"长".repeat(300)}` : "read", + state: "completed", summary: index === 500 - ? `OpenCode 工具：read 501 ${'详情'.repeat(2_100)}` - : `OpenCode 工具：read ${index + 1}` - }) + ? `OpenCode 工具：read 501 ${"详情".repeat(2_100)}` + : `OpenCode 工具：read ${index + 1}`, + }); } - }) + }); expect( - screen.getByRole('region', { name: '工具执行，共 501 项' }) - ).toBeInTheDocument() + screen.getByRole("region", { name: "工具执行，共 501 项" }), + ).toBeInTheDocument(); expect( screen.queryByText( - '活动记录过多，后续详情未在本地保留；Runtime 任务仍继续执行' - ) - ).not.toBeInTheDocument() + "活动记录过多，后续详情未在本地保留；Runtime 任务仍继续执行", + ), + ).not.toBeInTheDocument(); await waitFor( () => { const saved = - vi.mocked(api.activityHistory.replace).mock.calls.at(-1)?.[0] ?? - [] - expect( - saved.filter((record) => record.kind === 'tool') - ).toHaveLength(501) - const lastTool = saved.find((record) => record.callId === 'call-501') - expect(lastTool?.detail.length).toBeGreaterThan(4_000) - expect(lastTool?.title).toHaveLength(240) + vi.mocked(api.activityHistory.replace).mock.calls.at(-1)?.[0] ?? []; + expect(saved.filter((record) => record.kind === "tool")).toHaveLength( + 501, + ); + const lastTool = saved.find((record) => record.callId === "call-501"); + expect(lastTool?.detail.length).toBeGreaterThan(4_000); + expect(lastTool?.title).toHaveLength(240); }, - { timeout: 3_000 } - ) - }) + { timeout: 3_000 }, + ); + }); - it('keeps adjacent streamed text blocks exact in StrictMode', async () => { + it("keeps adjacent streamed text blocks exact in StrictMode", async () => { render( - - ) + , + ); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '读取验收标记' } - }) - await waitFor(() => expect(screen.getByLabelText('发送')).toBeEnabled()) - fireEvent.click(screen.getByLabelText('发送')) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "读取验收标记" }, + }); + await waitFor(() => expect(screen.getByLabelText("发送")).toBeEnabled()); + fireEvent.click(screen.getByLabelText("发送")); - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'text', - delta: 'GO' - }) + type: "text", + delta: "GO", + }); agentListener?.({ requestId: request.requestId, - type: 'text', - delta: 'ODBUDDY_REMOTE_TOOL_4C6B9D8A' - }) - }) + type: "text", + delta: "ODBUDDY_REMOTE_TOOL_4C6B9D8A", + }); + }); - const expected = 'GOODBUDDY_REMOTE_TOOL_4C6B9D8A' - const text = await screen.findByText(expected) - const assistantArticle = text.closest('article') - expect(assistantArticle).not.toBeNull() + const expected = "GOODBUDDY_REMOTE_TOOL_4C6B9D8A"; + const text = await screen.findByText(expected); + const assistantArticle = text.closest("article"); + expect(assistantArticle).not.toBeNull(); expect( - assistantArticle!.querySelectorAll( - '.message-blocks > .message__content' - ) - ).toHaveLength(1) + assistantArticle!.querySelectorAll(".message-blocks > .message__content"), + ).toHaveLength(1); expect( - assistantArticle!.querySelector('.message-blocks') - ).toHaveTextContent(expected) + assistantArticle!.querySelector(".message-blocks"), + ).toHaveTextContent(expected); act(() => { agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) - }) + type: "done", + }); + }); + }); - it('persists only the changed conversation and streamed assistant message', async () => { - const activeConversationId = - '00000000-0000-4000-8000-000000000441' - const activeMessageId = - '00000000-0000-4000-8000-000000000442' - const unrelatedConversationId = - '00000000-0000-4000-8000-000000000443' - const unrelatedMessageId = - '00000000-0000-4000-8000-000000000444' + it("persists only the changed conversation and streamed assistant message", async () => { + const activeConversationId = "00000000-0000-4000-8000-000000000441"; + const activeMessageId = "00000000-0000-4000-8000-000000000442"; + const unrelatedConversationId = "00000000-0000-4000-8000-000000000443"; + const unrelatedMessageId = "00000000-0000-4000-8000-000000000444"; vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: activeConversationId, projectId, - title: '增量持久化会话', + title: "增量持久化会话", updatedAt: 1_775_000_000_002, messages: [ { id: activeMessageId, - role: 'assistant', - content: '原有消息', + role: "assistant", + content: "原有消息", createdAt: 1_775_000_000_002, - state: 'complete' - } - ] + state: "complete", + }, + ], }, { id: unrelatedConversationId, projectId, - title: '无关会话', + title: "无关会话", updatedAt: 1_775_000_000_001, messages: [ { id: unrelatedMessageId, - role: 'assistant', - content: '不应重复保存', + role: "assistant", + content: "不应重复保存", createdAt: 1_775_000_000_001, - state: 'complete' - } - ] - } - ]) - render() - expect(await screen.findByText('原有消息')).toBeInTheDocument() + state: "complete", + }, + ], + }, + ]); + render(); + expect(await screen.findByText("原有消息")).toBeInTheDocument(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '只更新当前会话' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "只更新当前会话" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } - await waitFor(() => - expect(api.conversations.saveLocal).toHaveBeenCalled() - ) - vi.mocked(api.conversations.saveLocal).mockClear() + await waitFor(() => expect(api.conversations.saveLocal).toHaveBeenCalled()); + vi.mocked(api.conversations.saveLocal).mockClear(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'text', - delta: '流式增量' - }) - }) + type: "text", + delta: "流式增量", + }); + }); await waitFor( () => { const batch = vi .mocked(api.conversations.saveLocal) - .mock.calls.at(-1)?.[0] - expect(batch).toHaveLength(1) - expect(batch?.[0]?.header.id).toBe(activeConversationId) + .mock.calls.at(-1)?.[0]; + expect(batch).toHaveLength(1); + expect(batch?.[0]?.header.id).toBe(activeConversationId); expect(batch?.[0]?.messages).toEqual([ expect.objectContaining({ - role: 'assistant', - content: '流式增量', - state: 'streaming' - }) - ]) + role: "assistant", + content: "流式增量", + state: "streaming", + }), + ]); expect( - batch?.some( - (entry) => entry.header.id === unrelatedConversationId - ) - ).toBe(false) + batch?.some((entry) => entry.header.id === unrelatedConversationId), + ).toBe(false); expect( - batch?.flatMap((entry) => entry.messages).some( - (message) => - message.id === activeMessageId || - message.id === unrelatedMessageId || - message.role === 'user' - ) - ).toBe(false) + batch + ?.flatMap((entry) => entry.messages) + .some( + (message) => + message.id === activeMessageId || + message.id === unrelatedMessageId || + message.role === "user", + ), + ).toBe(false); }, - { timeout: 2_000 } - ) - }) + { timeout: 2_000 }, + ); + }); - it('flushes pending local conversation changes before quit', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000445' + it("flushes pending local conversation changes before quit", async () => { + const conversationId = "00000000-0000-4000-8000-000000000445"; vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: conversationId, projectId, - title: '退出持久化会话', + title: "退出持久化会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000446', - role: 'assistant', - content: '已有内容', + id: "00000000-0000-4000-8000-000000000446", + role: "assistant", + content: "已有内容", createdAt: 1_775_000_000_000, - state: 'complete' - } - ] - } - ]) - render() - expect(await screen.findByText('已有内容')).toBeInTheDocument() - vi.mocked(api.conversations.saveLocal).mockClear() + state: "complete", + }, + ], + }, + ]); + render(); + expect(await screen.findByText("已有内容")).toBeInTheDocument(); + vi.mocked(api.conversations.saveLocal).mockClear(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '退出前必须保存' } - }) - fireEvent.click(screen.getByLabelText('发送')) - expect( - await screen.findByText('退出前必须保存') - ).toBeInTheDocument() + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "退出前必须保存" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + expect(await screen.findByText("退出前必须保存")).toBeInTheDocument(); if (!beforeQuitListener) { - throw new Error('Missing before-quit persistence listener') + throw new Error("Missing before-quit persistence listener"); } - await act(async () => beforeQuitListener?.()) + await act(async () => beforeQuitListener?.()); expect(api.conversations.saveLocal).toHaveBeenCalledWith([ expect.objectContaining({ header: expect.objectContaining({ id: conversationId }), messages: expect.arrayContaining([ expect.objectContaining({ - role: 'user', - content: '退出前必须保存', - state: 'complete' - }) - ]) - }) - ]) - }) + role: "user", + content: "退出前必须保存", + state: "complete", + }), + ]), + }), + ]); + }); - it('replaces the direct-model thinking status with real reasoning', async () => { - render() + it("replaces the direct-model thinking status with real reasoning", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '分析这个问题' } - }) - const send = await screen.findByLabelText('发送') - await waitFor(() => expect(send).toBeEnabled()) - fireEvent.click(send) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "分析这个问题" }, + }); + const send = await screen.findByLabelText("发送"); + await waitFor(() => expect(send).toBeEnabled()); + fireEvent.click(send); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'status', - message: 'deepseek-v4-flash 正在思考' - }) - }) - expect( - screen.getByText('deepseek-v4-flash 正在思考') - ).toBeInTheDocument() + type: "status", + message: "deepseek-v4-flash 正在思考", + }); + }); + expect(screen.getByText("deepseek-v4-flash 正在思考")).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'reasoning', - delta: '正在分析真实推理内容' - }) - }) + type: "reasoning", + delta: "正在分析真实推理内容", + }); + }); expect( - screen.queryByText('deepseek-v4-flash 正在思考') - ).not.toBeInTheDocument() - expect(screen.getByText('正在推理').closest('details')).toHaveAttribute( - 'open' - ) - expect(screen.getByText('正在分析真实推理内容')).toBeVisible() - }) + screen.queryByText("deepseek-v4-flash 正在思考"), + ).not.toBeInTheDocument(); + expect(screen.getByText("正在推理").closest("details")).toHaveAttribute( + "open", + ); + expect(screen.getByText("正在分析真实推理内容")).toBeVisible(); + }); - it('updates context usage after model responses and keeps compression status', async () => { - const settings = await api.settings.getRuntime() + it("updates context usage after model responses and keeps compression status", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'model', + provider: "model", modelProfiles: settings.modelProfiles.map((profile) => ({ ...profile, - contextWindowTokens: 32_000 + contextWindowTokens: 32_000, })), contextCompression: { enabled: true, triggerTokens: 200_000, recentRawTokens: 32_000, - modelSource: { kind: 'current' }, - summaryPrompt: 'Preserve important facts.' - } - }) - render() + modelSource: { kind: "current" }, + summaryPrompt: "Preserve important facts.", + }, + }); + render(); - await screen.findByLabelText('向 GoodBuddy 提问') + await screen.findByLabelText("向 GoodBuddy 提问"); expect( - screen.queryByRole('progressbar', { - name: '当前上下文使用量' - }) - ).not.toBeInTheDocument() + screen.queryByRole("progressbar", { + name: "当前上下文使用量", + }), + ).not.toBeInTheDocument(); + expect(screen.queryByText("只读问答，不修改文件")).not.toBeInTheDocument(); expect( - screen.queryByText('只读问答，不修改文件') - ).not.toBeInTheDocument() - expect( - await screen.findByText('快捷唤起：', { exact: false }) - ).toHaveTextContent('快捷唤起：Ctrl+Shift+Space') + await screen.findByText("快捷唤起：", { exact: false }), + ).toHaveTextContent("快捷唤起：Ctrl+Shift+Space"); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '中'.repeat(1_000) } - }) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "中".repeat(1_000) }, + }); expect( - screen.queryByRole('progressbar', { - name: '当前上下文使用量' - }) - ).not.toBeInTheDocument() - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + screen.queryByRole("progressbar", { + name: "当前上下文使用量", + }), + ).not.toBeInTheDocument(); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-metrics', + type: "context-metrics", contextTokens: 22_000, effectiveTriggerTokens: 20_000, contextWindowTokens: 32_000, compressionEnabled: true, - source: 'provider' - }) - }) + source: "provider", + }); + }); expect( - screen.getByText('本次调用 22.0K / 32.0K · 69%') - ).toBeInTheDocument() + screen.getByText("本次调用 22.0K / 32.0K · 69%"), + ).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-compression', - scope: 'conversation', - state: 'started', + type: "context-compression", + scope: "conversation", + state: "started", estimatedBeforeTokens: 22_000, effectiveTriggerTokens: 20_000, contextWindowTokens: 32_000, recentRawTokens: 32_000, - coveredMessageCount: 2 - }) - }) - expect( - screen.getByText('正在压缩较早对话…') - ).toBeInTheDocument() + coveredMessageCount: 2, + }); + }); + expect(screen.getByText("正在压缩较早对话…")).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'status', - message: 'sonnet-5 正在思考' - }) - }) - expect( - screen.getByText('正在压缩较早对话…') - ).toBeInTheDocument() - expect(screen.getByText('sonnet-5 正在思考')).toBeInTheDocument() + type: "status", + message: "sonnet-5 正在思考", + }); + }); + expect(screen.getByText("正在压缩较早对话…")).toBeInTheDocument(); + expect(screen.getByText("sonnet-5 正在思考")).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-compression', - scope: 'conversation', - state: 'completed', + type: "context-compression", + scope: "conversation", + state: "completed", estimatedBeforeTokens: 22_000, estimatedAfterTokens: 9_000, effectiveTriggerTokens: 20_000, contextWindowTokens: 32_000, recentRawTokens: 32_000, coveredMessageCount: 2, - summaryTokens: 1_000 - }) - }) + summaryTokens: 1_000, + }); + }); expect( - screen.getByText( - '已压缩较早对话（估算） · ≈22.0K → ≈9.0K' - ) - ).toBeInTheDocument() + screen.getByText("已压缩较早对话（估算） · ≈22.0K → ≈9.0K"), + ).toBeInTheDocument(); expect( - screen.getByText( - '压缩后对话估算 ≈9.0K / 32.0K · 28%' - ) - ).toBeInTheDocument() + screen.getByText("压缩后对话估算 ≈9.0K / 32.0K · 28%"), + ).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-compression', - scope: 'agent-run', - state: 'started', + type: "context-compression", + scope: "agent-run", + state: "started", estimatedBeforeTokens: 24_000, effectiveTriggerTokens: 20_000, contextWindowTokens: 32_000, recentRawTokens: 4_000, - compressionCount: 1 - }) - }) - expect( - screen.getByText('正在整理 Agent 执行上下文…') - ).toBeInTheDocument() + compressionCount: 1, + }); + }); + expect(screen.getByText("正在整理 Agent 执行上下文…")).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-compression', - scope: 'agent-run', - state: 'completed', + type: "context-compression", + scope: "agent-run", + state: "completed", estimatedBeforeTokens: 24_000, estimatedAfterTokens: 11_000, effectiveTriggerTokens: 20_000, contextWindowTokens: 32_000, recentRawTokens: 4_000, - compressionCount: 2 - }) - }) + compressionCount: 2, + }); + }); expect( screen.getByText( - 'Agent 执行期间已压缩上下文 2 次（估算） · ≈24.0K → ≈11.0K' - ) - ).toBeInTheDocument() + "Agent 执行期间已压缩上下文 2 次（估算） · ≈24.0K → ≈11.0K", + ), + ).toBeInTheDocument(); expect( - screen.getByText( - '已压缩较早对话（估算） · ≈22.0K → ≈9.0K' - ) - ).toBeInTheDocument() + screen.getByText("已压缩较早对话（估算） · ≈22.0K → ≈9.0K"), + ).toBeInTheDocument(); expect( - screen.getByText( - '压缩后对话估算 ≈9.0K / 32.0K · 28%' - ) - ).toBeInTheDocument() + screen.getByText("压缩后对话估算 ≈9.0K / 32.0K · 28%"), + ).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-compression', - scope: 'conversation', - state: 'completed', + type: "context-compression", + scope: "conversation", + state: "completed", estimatedBeforeTokens: 24_000, estimatedAfterTokens: 8_500, effectiveTriggerTokens: 20_000, contextWindowTokens: 32_000, recentRawTokens: 4_000, - coveredMessageCount: 4 - }) - }) + coveredMessageCount: 4, + }); + }); expect( screen.getByText( - 'Agent 执行期间已压缩上下文 2 次（估算） · ≈24.0K → ≈11.0K' - ) - ).toBeInTheDocument() + "Agent 执行期间已压缩上下文 2 次（估算） · ≈24.0K → ≈11.0K", + ), + ).toBeInTheDocument(); expect( - screen.getByText( - '已压缩较早对话（估算） · ≈24.0K → ≈8.5K' - ) - ).toBeInTheDocument() + screen.getByText("已压缩较早对话（估算） · ≈24.0K → ≈8.5K"), + ).toBeInTheDocument(); expect( - screen.getByText( - '压缩后对话估算 ≈8.5K / 32.0K · 27%' - ) - ).toBeInTheDocument() + screen.getByText("压缩后对话估算 ≈8.5K / 32.0K · 27%"), + ).toBeInTheDocument(); expect( Array.from( - document.querySelectorAll( - '.context-compression-event__label' - ) - ).map((element) => element.textContent) + document.querySelectorAll(".context-compression-event__label"), + ).map((element) => element.textContent), ).toEqual([ - 'Agent 执行期间已压缩上下文 2 次（估算） · ≈24.0K → ≈11.0K', - '已压缩较早对话（估算） · ≈24.0K → ≈8.5K' - ]) + "Agent 执行期间已压缩上下文 2 次（估算） · ≈24.0K → ≈11.0K", + "已压缩较早对话（估算） · ≈24.0K → ≈8.5K", + ]); act(() => { agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); expect( screen.getByText( - 'Agent 执行期间已压缩上下文 2 次（估算） · ≈24.0K → ≈11.0K' - ) - ).toBeInTheDocument() - }) + "Agent 执行期间已压缩上下文 2 次（估算） · ≈24.0K → ≈11.0K", + ), + ).toBeInTheDocument(); + }); - it('does not present the compression threshold as a context-window percentage', async () => { - const settings = await api.settings.getRuntime() + it("does not present the compression threshold as a context-window percentage", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'model', + provider: "model", modelProfiles: settings.modelProfiles.map((profile) => ({ ...profile, - contextWindowTokens: undefined + contextWindowTokens: undefined, })), contextCompression: { enabled: true, triggerTokens: 12_000, recentRawTokens: 4_000, - modelSource: { kind: 'current' }, - summaryPrompt: 'Preserve important facts.' - } - }) - render() + modelSource: { kind: "current" }, + summaryPrompt: "Preserve important facts.", + }, + }); + render(); - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '检查上下文显示' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "检查上下文显示" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-metrics', + type: "context-metrics", contextTokens: 22_000, effectiveTriggerTokens: 20_000, compressionEnabled: true, - source: 'provider' - }) - }) + source: "provider", + }); + }); expect( - screen.getByText('本次调用 22.0K · 压缩线 12.0K') - ).toBeInTheDocument() + screen.getByText("本次调用 22.0K · 压缩线 12.0K"), + ).toBeInTheDocument(); expect( - screen.queryByRole('progressbar', { - name: '当前上下文使用量' - }) - ).not.toBeInTheDocument() - expect(screen.queryByText('110%')).not.toBeInTheDocument() + screen.queryByRole("progressbar", { + name: "当前上下文使用量", + }), + ).not.toBeInTheDocument(); + expect(screen.queryByText("110%")).not.toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-compression', - scope: 'conversation', - state: 'completed', + type: "context-compression", + scope: "conversation", + state: "completed", estimatedBeforeTokens: 22_000, estimatedAfterTokens: 9_400, effectiveTriggerTokens: 20_000, recentRawTokens: 4_000, - coveredMessageCount: 2 - }) - }) + coveredMessageCount: 2, + }); + }); expect( - screen.getByText( - '压缩后对话估算 ≈9.4K · 压缩线 12.0K' - ) - ).toBeInTheDocument() + screen.getByText("压缩后对话估算 ≈9.4K · 压缩线 12.0K"), + ).toBeInTheDocument(); expect( - screen.queryByText('本次调用 22.0K · 压缩线 12.0K') - ).not.toBeInTheDocument() - }) + screen.queryByText("本次调用 22.0K · 压缩线 12.0K"), + ).not.toBeInTheDocument(); + }); - it('updates the composer compression line immediately after settings change', async () => { - const settings = await api.settings.getRuntime() + it("updates the composer compression line immediately after settings change", async () => { + const settings = await api.settings.getRuntime(); const initialSettings = { ...settings, - provider: 'model' as const, + provider: "model" as const, modelProfiles: settings.modelProfiles.map((profile) => ({ ...profile, - contextWindowTokens: undefined + contextWindowTokens: undefined, })), contextCompression: { enabled: true, triggerTokens: 20_000, recentRawTokens: 4_000, - modelSource: { kind: 'current' as const }, - summaryPrompt: 'Preserve important facts.' - } - } + modelSource: { kind: "current" as const }, + summaryPrompt: "Preserve important facts.", + }, + }; vi.mocked(api.settings.getRuntime) .mockResolvedValueOnce(initialSettings) - .mockResolvedValueOnce(initialSettings) + .mockResolvedValueOnce(initialSettings); vi.mocked(api.settings.updateRuntime).mockImplementationOnce( async (input) => ({ ...initialSettings, contextCompression: - input.contextCompression ?? - initialSettings.contextCompression - }) - ) - render() + input.contextCompression ?? initialSettings.contextCompression, + }), + ); + render(); - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '检查压缩线设置刷新' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "检查压缩线设置刷新" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-metrics', + type: "context-metrics", contextTokens: 9_000, effectiveTriggerTokens: 20_000, compressionEnabled: true, - source: 'provider' - }) - }) + source: "provider", + }); + }); expect( - screen.getByText('本次调用 9.0K · 压缩线 20.0K') - ).toBeInTheDocument() + screen.getByText("本次调用 9.0K · 压缩线 20.0K"), + ).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); - fireEvent.click( - screen.getByRole('button', { name: /本地工作区/u }) - ) - await screen.findByRole('heading', { name: '设置中心' }) - fireEvent.click( - screen.getByRole('tab', { name: '上下文控制' }) - ) - const trigger = await screen.findByLabelText('压缩触发阈值') - fireEvent.change(trigger, { target: { value: '12' } }) - fireEvent.click(screen.getByRole('button', { name: '保存设置' })) + fireEvent.click(screen.getByRole("button", { name: /本地工作区/u })); + await screen.findByRole("heading", { name: "设置中心" }); + fireEvent.click(screen.getByRole("tab", { name: "上下文控制" })); + const trigger = await screen.findByLabelText("压缩触发阈值"); + fireEvent.change(trigger, { target: { value: "12" } }); + fireEvent.click(screen.getByRole("button", { name: "保存设置" })); await waitFor(() => expect(api.settings.updateRuntime).toHaveBeenCalledWith( expect.objectContaining({ contextCompression: expect.objectContaining({ - triggerTokens: 12_000 - }) - }) - ) - ) + triggerTokens: 12_000, + }), + }), + ), + ); - fireEvent.click(screen.getByRole('button', { name: '对话' })) + fireEvent.click(screen.getByRole("button", { name: "对话" })); expect( - await screen.findByText('本次调用 9.0K · 压缩线 12.0K') - ).toBeInTheDocument() - }) + await screen.findByText("本次调用 9.0K · 压缩线 12.0K"), + ).toBeInTheDocument(); + }); - it('guards sidebar navigation away from dirty Settings drafts', async () => { - render() + it("guards sidebar navigation away from dirty Settings drafts", async () => { + render(); - fireEvent.click(await screen.findByText('本地工作区')) - await screen.findByRole('heading', { name: '设置中心' }) - fireEvent.change(await screen.findByLabelText('默认工作区目录'), { - target: { value: 'C:\\Unsaved from App' } - }) - const knowledgeNavigation = screen.getByRole('button', { - name: '知识库' - }) - fireEvent.click(knowledgeNavigation) + fireEvent.click(await screen.findByText("本地工作区")); + await screen.findByRole("heading", { name: "设置中心" }); + fireEvent.change(await screen.findByLabelText("默认工作区目录"), { + target: { value: "C:\\Unsaved from App" }, + }); + const knowledgeNavigation = screen.getByRole("button", { + name: "知识库", + }); + fireEvent.click(knowledgeNavigation); + expect(screen.getByRole("heading", { name: "设置中心" })).toBeVisible(); + expect(screen.getByRole("alert")).toHaveTextContent("当前设置有未保存更改"); + fireEvent.click(screen.getByRole("button", { name: "放弃更改并关闭" })); expect( - screen.getByRole('heading', { name: '设置中心' }) - ).toBeVisible() - expect(screen.getByRole('alert')).toHaveTextContent( - '当前设置有未保存更改' - ) - fireEvent.click( - screen.getByRole('button', { name: '放弃更改并关闭' }) - ) - expect( - await screen.findByRole('heading', { name: '知识库' }) - ).toBeVisible() - await waitFor(() => expect(knowledgeNavigation).toHaveFocus()) - }) + await screen.findByRole("heading", { name: "知识库" }), + ).toBeVisible(); + await waitFor(() => expect(knowledgeNavigation).toHaveFocus()); + }); - it('restores focus outside Settings after closing the page', async () => { - render() + it("restores focus outside Settings after closing the page", async () => { + render(); - await screen.findByLabelText('向 GoodBuddy 提问') - const settingsTrigger = screen.getByRole('button', { - name: /本地工作区/u - }) - fireEvent.click(settingsTrigger) - await screen.findByRole('heading', { name: '设置中心' }) + await screen.findByLabelText("向 GoodBuddy 提问"); + const settingsTrigger = screen.getByRole("button", { + name: /本地工作区/u, + }); + fireEvent.click(settingsTrigger); + await screen.findByRole("heading", { name: "设置中心" }); - fireEvent.click( - screen.getByRole('button', { name: '关闭设置' }) - ) + fireEvent.click(screen.getByRole("button", { name: "关闭设置" })); - await waitFor(() => expect(settingsTrigger).toHaveFocus()) - }) + await waitFor(() => expect(settingsTrigger).toHaveFocus()); + }); - it('focuses the Composer after closing automatically opened Settings', async () => { + it("focuses the Composer after closing automatically opened Settings", async () => { vi.mocked(api.agent.getStatus).mockResolvedValueOnce({ - id: 'setup', - label: '未配置', + id: "setup", + label: "未配置", available: false, supportsToolExecution: false, - detail: '尚未配置 Runtime' - }) - render() + detail: "尚未配置 Runtime", + }); + render(); - await screen.findByRole('heading', { name: '设置中心' }) - const composer = screen.getByLabelText('向 GoodBuddy 提问') - fireEvent.click( - screen.getByRole('button', { name: '关闭设置' }) - ) + await screen.findByRole("heading", { name: "设置中心" }); + const composer = screen.getByLabelText("向 GoodBuddy 提问"); + fireEvent.click(screen.getByRole("button", { name: "关闭设置" })); - await waitFor(() => expect(composer).toHaveFocus()) - }) + await waitFor(() => expect(composer).toHaveFocus()); + }); - it('updates and removes the composer shortcut hint immediately after saving Settings', async () => { + it("updates and removes the composer shortcut hint immediately after saving Settings", async () => { let applicationSettings: ApplicationSettings = { checkUpdatesOnStartup: false, - updateSource: 'github', - modelDownloadSource: 'modelscope', + updateSource: "github", + modelDownloadSource: "modelscope", localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: false, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate', - magicNoteCommentFormat: 'combined' - } + magicNoteCommentMode: "immediate", + magicNoteCommentFormat: "combined", + }; let shortcutSnapshot: GlobalShortcutSettingsSnapshot = { settings: { enabled: true, - accelerator: 'CommandOrControl+Shift+Space' + accelerator: "CommandOrControl+Shift+Space", }, defaultSettings: { enabled: true, - accelerator: 'CommandOrControl+Shift+Space' + accelerator: "CommandOrControl+Shift+Space", }, - platform: 'win32', - displayAccelerator: 'Ctrl+Shift+Space', + platform: "win32", + displayAccelerator: "Ctrl+Shift+Space", registered: true, - registeredAccelerator: 'CommandOrControl+Shift+Space', - status: 'registered' - } + registeredAccelerator: "CommandOrControl+Shift+Space", + status: "registered", + }; api.updates = { getSettings: vi.fn(async () => ({ ...applicationSettings })), updateSettings: vi.fn(async (input) => { - applicationSettings = { ...applicationSettings, ...input } - return { ...applicationSettings } + applicationSettings = { ...applicationSettings, ...input }; + return { ...applicationSettings }; }), check: vi.fn(), openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; api.shortcuts = { getSettings: vi.fn(async () => shortcutSnapshot), updateSettings: vi.fn(async (input) => { @@ -4543,861 +4269,795 @@ describe('App', () => { ...shortcutSnapshot, settings: input, displayAccelerator: - input.accelerator === 'Control+Alt+K' - ? 'Ctrl+Alt+K' - : 'Ctrl+Shift+Space', + input.accelerator === "Control+Alt+K" + ? "Ctrl+Alt+K" + : "Ctrl+Shift+Space", registered: input.enabled, - registeredAccelerator: input.enabled - ? input.accelerator - : undefined, - status: input.enabled ? 'registered' : 'disabled' - } - return { ok: true as const, snapshot: shortcutSnapshot } - }) - } + registeredAccelerator: input.enabled ? input.accelerator : undefined, + status: input.enabled ? "registered" : "disabled", + }; + return { ok: true as const, snapshot: shortcutSnapshot }; + }), + }; try { - render() - expect( - await screen.findByText('Ctrl+Shift+Space') - ).toBeInTheDocument() - fireEvent.click(await screen.findByText('本地工作区')) - await screen.findByRole('heading', { name: '设置中心' }) - fireEvent.click(screen.getByRole('tab', { name: '平台功能' })) - const shortcutInput = await screen.findByLabelText('快捷键') + render(); + expect(await screen.findByText("Ctrl+Shift+Space")).toBeInTheDocument(); + fireEvent.click(await screen.findByText("本地工作区")); + await screen.findByRole("heading", { name: "设置中心" }); + fireEvent.click(screen.getByRole("tab", { name: "平台功能" })); + const shortcutInput = await screen.findByLabelText("快捷键"); fireEvent.change(shortcutInput, { - target: { value: 'Control+Alt+K' } - }) - fireEvent.click( - screen.getByRole('button', { name: '保存快捷键' }) - ) + target: { value: "Control+Alt+K" }, + }); + fireEvent.click(screen.getByRole("button", { name: "保存快捷键" })); await waitFor(() => expect(api.shortcuts?.updateSettings).toHaveBeenCalledWith({ enabled: true, - accelerator: 'Control+Alt+K' - }) - ) - fireEvent.click(screen.getByRole('button', { name: '对话' })) - expect(await screen.findByText('Ctrl+Alt+K')).toBeInTheDocument() + accelerator: "Control+Alt+K", + }), + ); + fireEvent.click(screen.getByRole("button", { name: "对话" })); + expect(await screen.findByText("Ctrl+Alt+K")).toBeInTheDocument(); - fireEvent.click(await screen.findByText('本地工作区')) - await screen.findByRole('heading', { name: '设置中心' }) - fireEvent.click(screen.getByRole('tab', { name: '平台功能' })) - const shortcutSwitch = await screen.findByRole('switch', { - name: '启用全局快捷键' - }) - fireEvent.click(shortcutSwitch) - fireEvent.click( - screen.getByRole('button', { name: '保存快捷键' }) - ) + fireEvent.click(await screen.findByText("本地工作区")); + await screen.findByRole("heading", { name: "设置中心" }); + fireEvent.click(screen.getByRole("tab", { name: "平台功能" })); + const shortcutSwitch = await screen.findByRole("switch", { + name: "启用全局快捷键", + }); + fireEvent.click(shortcutSwitch); + fireEvent.click(screen.getByRole("button", { name: "保存快捷键" })); await waitFor(() => expect(api.shortcuts?.updateSettings).toHaveBeenCalledWith({ enabled: false, - accelerator: 'Control+Alt+K' - }) - ) - fireEvent.click(screen.getByRole('button', { name: '对话' })) - await screen.findByLabelText('向 GoodBuddy 提问') - expect( - screen.queryByText('Ctrl+Alt+K') - ).not.toBeInTheDocument() + accelerator: "Control+Alt+K", + }), + ); + fireEvent.click(screen.getByRole("button", { name: "对话" })); + await screen.findByLabelText("向 GoodBuddy 提问"); + expect(screen.queryByText("Ctrl+Alt+K")).not.toBeInTheDocument(); } finally { - delete api.shortcuts - delete api.updates + delete api.shortcuts; + delete api.updates; } - }) + }); - it('restores persisted context usage and compression state after restart', async () => { - const settings = await api.settings.getRuntime() - const profile = settings.modelProfiles[0]! - const conversationId = - '00000000-0000-4000-8000-000000000451' + it("restores persisted context usage and compression state after restart", async () => { + const settings = await api.settings.getRuntime(); + const profile = settings.modelProfiles[0]!; + const conversationId = "00000000-0000-4000-8000-000000000451"; const compressionState = { - coveredHistoryDigest: 'a'.repeat(64), + coveredHistoryDigest: "a".repeat(64), coveredMessageCount: 2, - coveredFromMessageId: - '00000000-0000-4000-8000-000000000452', - coveredThroughMessageId: - '00000000-0000-4000-8000-000000000453', - summary: 'Persisted conversation summary' - } + coveredFromMessageId: "00000000-0000-4000-8000-000000000452", + coveredThroughMessageId: "00000000-0000-4000-8000-000000000453", + summary: "Persisted conversation summary", + }; vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'model', + provider: "model", defaultModelProfileId: profile.id, modelProfiles: settings.modelProfiles.map((candidate) => ({ ...candidate, - contextWindowTokens: undefined + contextWindowTokens: undefined, })), contextCompression: { enabled: true, triggerTokens: 12_000, recentRawTokens: 4_000, - modelSource: { kind: 'current' }, - summaryPrompt: 'Preserve important facts.' - } - }) + modelSource: { kind: "current" }, + summaryPrompt: "Preserve important facts.", + }, + }); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: conversationId, projectId, runtimeSelection: { - provider: 'model', - profileId: profile.id + provider: "model", + profileId: profile.id, }, contextMetrics: { runtimeSelectionKey: `model:${profile.id}`, contextTokens: 9_000, - source: 'estimated', - basis: 'conversation' + source: "estimated", + basis: "conversation", }, contextCompressionState: compressionState, - title: '已压缩会话', + title: "已压缩会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000452', - role: 'user', - content: '此前问题', + id: "00000000-0000-4000-8000-000000000452", + role: "user", + content: "此前问题", createdAt: 1_775_000_000_000, - state: 'complete' + state: "complete", }, { - id: '00000000-0000-4000-8000-000000000453', - role: 'assistant', - content: '此前回答', + id: "00000000-0000-4000-8000-000000000453", + role: "assistant", + content: "此前回答", createdAt: 1_775_000_000_001, - state: 'complete', + state: "complete", contextCompression: { - state: 'completed', - scope: 'conversation', + state: "completed", + scope: "conversation", estimatedBeforeTokens: 22_000, - estimatedAfterTokens: 9_000 - } - } - ] - } - ]) - render() + estimatedAfterTokens: 9_000, + }, + }, + ], + }, + ]); + render(); expect( - await screen.findByText( - '压缩后对话估算 ≈9.0K · 压缩线 12.0K' - ) - ).toBeInTheDocument() + await screen.findByText("压缩后对话估算 ≈9.0K · 压缩线 12.0K"), + ).toBeInTheDocument(); expect( - screen.getByText( - '已压缩较早对话（估算） · ≈22.0K → ≈9.0K' - ) - ).toBeInTheDocument() + screen.getByText("已压缩较早对话（估算） · ≈22.0K → ≈9.0K"), + ).toBeInTheDocument(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '继续工作' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "继续工作" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); expect(run.mock.calls[0]?.[0].contextCompressionState).toEqual( - compressionState - ) + compressionState, + ); expect(run.mock.calls[0]?.[0]).toMatchObject({ historyMessageIds: [ - '00000000-0000-4000-8000-000000000452', - '00000000-0000-4000-8000-000000000453' + "00000000-0000-4000-8000-000000000452", + "00000000-0000-4000-8000-000000000453", ], currentUserMessageId: expect.any(String), - currentAssistantMessageId: expect.any(String) - }) - }) + currentAssistantMessageId: expect.any(String), + }); + }); - it('keeps a tool failure in details and hides retry after continuing', async () => { - render() + it("keeps a tool failure in details and hides retry after continuing", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '读取演示文稿' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "读取演示文稿" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } const toolError = - 'Cannot read binary file: D:\\workspace\\presentation.pptx' - const runtimeError = `OpenCode 工具执行失败（call-1）：${toolError}` + "Cannot read binary file: D:\\workspace\\presentation.pptx"; + const runtimeError = `OpenCode 工具执行失败（call-1）：${toolError}`; act(() => { agentListener?.({ requestId: request.requestId, - type: 'tool', - callId: 'call-1', - name: 'read', - state: 'failed', - summary: 'OpenCode 工具：read', + type: "tool", + callId: "call-1", + name: "read", + state: "failed", + summary: "OpenCode 工具：read", input: '{"path":"D:\\\\workspace\\\\presentation.pptx"}', - error: toolError - }) + error: toolError, + }); agentListener?.({ requestId: request.requestId, - type: 'error', - status: 'failed', - message: runtimeError - }) - }) + type: "error", + status: "failed", + message: runtimeError, + }); + }); - expect(screen.getByText(toolError)).toBeInTheDocument() - expect(screen.queryByText(runtimeError)).not.toBeInTheDocument() + expect(screen.getByText(toolError)).toBeInTheDocument(); + expect(screen.queryByText(runtimeError)).not.toBeInTheDocument(); expect( - screen.getByRole('button', { name: '重新编辑并发送' }) - ).toBeInTheDocument() + screen.getByRole("button", { name: "重新编辑并发送" }), + ).toBeInTheDocument(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '继续处理' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledTimes(2)) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "继续处理" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledTimes(2)); expect( - screen.queryByRole('button', { name: '重新编辑并发送' }) - ).not.toBeInTheDocument() - }) + screen.queryByRole("button", { name: "重新编辑并发送" }), + ).not.toBeInTheDocument(); + }); - it('submits knowledge scope without eager search or prompt injection and merges runtime references', async () => { - const libraryId = '11111111-1111-4111-8111-111111111111' + it("submits knowledge scope without eager search or prompt injection and merges runtime references", async () => { + const libraryId = "11111111-1111-4111-8111-111111111111"; vi.mocked(api.knowledge.getSnapshot).mockResolvedValueOnce({ libraries: [ { id: libraryId, - name: '产品知识', - description: '', - storageMode: 'managed', + name: "产品知识", + description: "", + storageMode: "managed", graphEnabled: false, - graphStrategy: 'rules', + graphStrategy: "rules", sourceCount: 1, documentCount: 1, - indexedDocumentCount: 1 - } + indexedDocumentCount: 1, + }, ], sources: [], documents: [], graphNodes: [], graphRelations: [], - evidence: [] - }) - render() - const knowledgeScopeTrigger = await screen.findByRole('button', { - name: '选择知识库，本次已启用 0 个' - }) - expect(knowledgeScopeTrigger).toHaveAttribute( - 'aria-haspopup', - 'dialog' - ) - fireEvent.click(knowledgeScopeTrigger) - expect(knowledgeScopeTrigger).toHaveAttribute( - 'aria-expanded', - 'true' - ) + evidence: [], + }); + render(); + const knowledgeScopeTrigger = await screen.findByRole("button", { + name: "选择知识库，本次已启用 0 个", + }); + expect(knowledgeScopeTrigger).toHaveAttribute("aria-haspopup", "dialog"); + fireEvent.click(knowledgeScopeTrigger); + expect(knowledgeScopeTrigger).toHaveAttribute("aria-expanded", "true"); expect( - screen.getByRole('dialog', { - name: '本次对话检索范围' - }) - ).toBeInTheDocument() + screen.getByRole("dialog", { + name: "本次对话检索范围", + }), + ).toBeInTheDocument(); await waitFor(() => - expect( - screen.getByRole('checkbox', { name: /产品知识/u }) - ).toHaveFocus() - ) - fireEvent.keyDown(document.activeElement!, { key: 'Escape' }) - expect(knowledgeScopeTrigger).toHaveAttribute( - 'aria-expanded', - 'false' - ) - expect(knowledgeScopeTrigger).toHaveFocus() + expect(screen.getByRole("checkbox", { name: /产品知识/u })).toHaveFocus(), + ); + fireEvent.keyDown(document.activeElement!, { key: "Escape" }); + expect(knowledgeScopeTrigger).toHaveAttribute("aria-expanded", "false"); + expect(knowledgeScopeTrigger).toHaveFocus(); - fireEvent.click(knowledgeScopeTrigger) - const scopeCheckbox = screen.getByRole('checkbox', { - name: /产品知识/u - }) - await waitFor(() => expect(scopeCheckbox).toHaveFocus()) - fireEvent.focusOut(scopeCheckbox, { relatedTarget: null }) - expect(knowledgeScopeTrigger).toHaveAttribute('aria-expanded', 'true') - fireEvent.click(screen.getByText('产品知识', { selector: 'span' })) - expect(scopeCheckbox).toBeChecked() - fireEvent.click(scopeCheckbox.closest('label')!) - expect(scopeCheckbox).not.toBeChecked() - fireEvent.click(scopeCheckbox.closest('label')!.querySelector('small')!) - expect(scopeCheckbox).toBeChecked() - const composerInput = screen.getByLabelText('向 GoodBuddy 提问') + fireEvent.click(knowledgeScopeTrigger); + const scopeCheckbox = screen.getByRole("checkbox", { + name: /产品知识/u, + }); + await waitFor(() => expect(scopeCheckbox).toHaveFocus()); + fireEvent.focusOut(scopeCheckbox, { relatedTarget: null }); + expect(knowledgeScopeTrigger).toHaveAttribute("aria-expanded", "true"); + fireEvent.click(screen.getByText("产品知识", { selector: "span" })); + expect(scopeCheckbox).toBeChecked(); + fireEvent.click(scopeCheckbox.closest("label")!); + expect(scopeCheckbox).not.toBeChecked(); + fireEvent.click(scopeCheckbox.closest("label")!.querySelector("small")!); + expect(scopeCheckbox).toBeChecked(); + const composerInput = screen.getByLabelText("向 GoodBuddy 提问"); fireEvent.focusOut(scopeCheckbox, { - relatedTarget: composerInput - }) - composerInput.focus() - expect(knowledgeScopeTrigger).toHaveAttribute( - 'aria-expanded', - 'false' - ) - expect(composerInput).toHaveFocus() + relatedTarget: composerInput, + }); + composerInput.focus(); + expect(knowledgeScopeTrigger).toHaveAttribute("aria-expanded", "false"); + expect(composerInput).toHaveFocus(); fireEvent.change(composerInput, { - target: { value: '发布流程是什么？' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + target: { value: "发布流程是什么？" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; expect(request).toMatchObject({ - prompt: '发布流程是什么？', + prompt: "发布流程是什么？", knowledgeLibraryIds: [libraryId], - knowledgeRetrievalMode: 'auto' - }) - expect(api.knowledge.search).not.toHaveBeenCalled() - expect( - screen.queryByText(/查看 \d+ 条证据引用/u) - ).not.toBeInTheDocument() + knowledgeRetrievalMode: "auto", + }); + expect(api.knowledge.search).not.toHaveBeenCalled(); + expect(screen.queryByText(/查看 \d+ 条证据引用/u)).not.toBeInTheDocument(); act(() => { if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } for (let batch = 0; batch < 5; batch += 1) { agentListener?.({ requestId: request.requestId, - type: 'source-references', + type: "source-references", references: Array.from({ length: 25 }, (_, index) => ({ libraryId, - libraryName: '产品知识', + libraryName: "产品知识", documentId: crypto.randomUUID(), chunkId: crypto.randomUUID(), documentName: `发布手册 ${batch}-${index}`, sourceName: `release-${batch}-${index}.md`, locator: `第 ${batch}-${index} 节`, snippet: `证据 ${batch}-${index}`, - rank: index + 1 - })) - }) + rank: index + 1, + })), + }); } agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) - expect( - await screen.findByText('查看 20 条证据引用') - ).toBeInTheDocument() + type: "done", + }); + }); + expect(await screen.findByText("查看 20 条证据引用")).toBeInTheDocument(); vi.mocked(api.knowledge.getReferenceContext).mockResolvedValueOnce({ knowledgeBaseId: libraryId, - documentId: 'document-context', - chunkId: 'chunk-context', - documentTitle: '发布手册', - sourceDisplayName: 'release.md', - locator: '第 4 节', - matchedContent: '命中分块内容', - contextContent: '上文\n\n命中分块内容\n\n下文', - contextChunkIds: ['chunk-context'], - truncated: false - }) - fireEvent.click(screen.getByText('查看 20 条证据引用')) - fireEvent.click( - screen.getAllByRole('button', { name: '查看上下文' })[0]! - ) + documentId: "document-context", + chunkId: "chunk-context", + documentTitle: "发布手册", + sourceDisplayName: "release.md", + locator: "第 4 节", + matchedContent: "命中分块内容", + contextContent: "上文\n\n命中分块内容\n\n下文", + contextChunkIds: ["chunk-context"], + truncated: false, + }); + fireEvent.click(screen.getByText("查看 20 条证据引用")); + fireEvent.click(screen.getAllByRole("button", { name: "查看上下文" })[0]!); expect( - await screen.findByRole('dialog', { name: '引用上下文' }) - ).toBeInTheDocument() + await screen.findByRole("dialog", { name: "引用上下文" }), + ).toBeInTheDocument(); expect(api.knowledge.getReferenceContext).toHaveBeenCalledWith( expect.objectContaining({ - knowledgeBaseId: libraryId - }) - ) + knowledgeBaseId: libraryId, + }), + ); fireEvent.click( - screen.getAllByRole('button', { - name: '关闭引用上下文' - })[0]! - ) + screen.getAllByRole("button", { + name: "关闭引用上下文", + })[0]!, + ); await waitFor( () => { const persistedMessages = vi .mocked(api.conversations.saveLocal) .mock.calls.flatMap(([batch]) => - batch.flatMap((conversation) => conversation.messages) - ) + batch.flatMap((conversation) => conversation.messages), + ); const persisted = persistedMessages - .filter((message) => message.role === 'assistant') + .filter((message) => message.role === "assistant") .slice() .reverse() - .find( - (message) => message.sourceReferences?.length === 20 - ) - expect(persisted?.sourceReferences).toHaveLength(20) - expect(persisted?.sources).toBeUndefined() + .find((message) => message.sourceReferences?.length === 20); + expect(persisted?.sourceReferences).toHaveLength(20); + expect(persisted?.sources).toBeUndefined(); }, - { timeout: 2_000 } - ) - }) + { timeout: 2_000 }, + ); + }); - it('restores an independent knowledge scope for each conversation', async () => { - const libraryId = '11111111-1111-4111-8111-111111111111' - const firstConversationId = - '22222222-2222-4222-8222-222222222222' - const secondConversationId = - '33333333-3333-4333-8333-333333333333' + it("restores an independent knowledge scope for each conversation", async () => { + const libraryId = "11111111-1111-4111-8111-111111111111"; + const firstConversationId = "22222222-2222-4222-8222-222222222222"; + const secondConversationId = "33333333-3333-4333-8333-333333333333"; vi.mocked(api.knowledge.getSnapshot).mockResolvedValueOnce({ libraries: [ { id: libraryId, - name: '产品知识', - description: '', - storageMode: 'managed', + name: "产品知识", + description: "", + storageMode: "managed", graphEnabled: false, - graphStrategy: 'rules', + graphStrategy: "rules", sourceCount: 1, documentCount: 1, - indexedDocumentCount: 1 - } + indexedDocumentCount: 1, + }, ], sources: [], documents: [], graphNodes: [], graphRelations: [], - evidence: [] - }) + evidence: [], + }); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: firstConversationId, projectId, knowledgeLibraryIds: [libraryId], - knowledgeRetrievalMode: 'always', - title: '带知识范围', + knowledgeRetrievalMode: "always", + title: "带知识范围", updatedAt: 200, - messages: [] + messages: [], }, { id: secondConversationId, projectId, knowledgeLibraryIds: [], - knowledgeRetrievalMode: 'auto', - title: '空知识范围', + knowledgeRetrievalMode: "auto", + title: "空知识范围", updatedAt: 100, - messages: [] - } - ]) + messages: [], + }, + ]); - render() + render(); expect( - await screen.findByRole('button', { - name: '选择知识库，本次已启用 1 个' - }) - ).toBeInTheDocument() - fireEvent.click( - screen.getByRole('button', { name: /^空知识范围/u }) - ) + await screen.findByRole("button", { + name: "选择知识库，本次已启用 1 个", + }), + ).toBeInTheDocument(); + fireEvent.click(screen.getByRole("button", { name: /^空知识范围/u })); expect( - screen.getByRole('button', { - name: '选择知识库，本次已启用 0 个' - }) - ).toBeInTheDocument() - fireEvent.click( - screen.getByRole('button', { name: /^带知识范围/u }) - ) + screen.getByRole("button", { + name: "选择知识库，本次已启用 0 个", + }), + ).toBeInTheDocument(); + fireEvent.click(screen.getByRole("button", { name: /^带知识范围/u })); expect( - screen.getByRole('button', { - name: '选择知识库，本次已启用 1 个' - }) - ).toBeInTheDocument() - }) + screen.getByRole("button", { + name: "选择知识库，本次已启用 1 个", + }), + ).toBeInTheDocument(); + }); - it('persists and submits always-retrieve mode for the active conversation', async () => { - const libraryId = '11111111-1111-4111-8111-111111111111' + it("persists and submits always-retrieve mode for the active conversation", async () => { + const libraryId = "11111111-1111-4111-8111-111111111111"; vi.mocked(api.knowledge.getSnapshot).mockResolvedValueOnce({ libraries: [ { id: libraryId, - name: '产品知识', - description: '', - storageMode: 'managed', + name: "产品知识", + description: "", + storageMode: "managed", graphEnabled: false, - graphStrategy: 'rules', + graphStrategy: "rules", sourceCount: 1, documentCount: 1, - indexedDocumentCount: 1 - } + indexedDocumentCount: 1, + }, ], sources: [], documents: [], graphNodes: [], graphRelations: [], - evidence: [] - }) - render() - const knowledgeScope = await screen.findByRole('button', { - name: '选择知识库，本次已启用 0 个' - }) - fireEvent.click(knowledgeScope) + evidence: [], + }); + render(); + const knowledgeScope = await screen.findByRole("button", { + name: "选择知识库，本次已启用 0 个", + }); + fireEvent.click(knowledgeScope); + fireEvent.click(screen.getByRole("checkbox", { name: /产品知识/u })); + const retrievalMode = screen.getByRole("group", { + name: "知识检索方式", + }); fireEvent.click( - screen.getByRole('checkbox', { name: /产品知识/u }) - ) - const retrievalMode = screen.getByRole('group', { - name: '知识检索方式' - }) - fireEvent.click( - within(retrievalMode).getByRole('button', { - name: '每次先检索' - }) - ) + within(retrievalMode).getByRole("button", { + name: "每次先检索", + }), + ); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '必须查询发布流程' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "必须查询发布流程" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); expect(run.mock.calls[0]?.[0]).toMatchObject({ - prompt: '必须查询发布流程', + prompt: "必须查询发布流程", knowledgeLibraryIds: [libraryId], - knowledgeRetrievalMode: 'always' - }) - const request = run.mock.calls[0]?.[0] + knowledgeRetrievalMode: "always", + }); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'knowledge-retrieval', - mode: 'always', - state: 'degraded', + type: "knowledge-retrieval", + mode: "always", + state: "degraded", libraryCount: 1, resultCount: 2, durationMs: 18, - usedChannels: ['fts', 'cjk'], - warnings: ['向量模型未配置'] - }) - }) + usedChannels: ["fts", "cjk"], + warnings: ["向量模型未配置"], + }); + }); + expect(await screen.findByText("知识检索已降级")).toBeInTheDocument(); expect( - await screen.findByText('知识检索已降级') - ).toBeInTheDocument() - expect( - screen.getByText(/已检索 1 个知识库，获得 2 条结果/u) - ).toBeInTheDocument() - expect(screen.getByText('向量模型未配置')).toBeInTheDocument() + screen.getByText(/已检索 1 个知识库，获得 2 条结果/u), + ).toBeInTheDocument(); + expect(screen.getByText("向量模型未配置")).toBeInTheDocument(); await waitFor(() => { const snapshots = vi .mocked(api.conversations.saveLocal) - .mock.calls.flatMap(([batch]) => batch) + .mock.calls.flatMap(([batch]) => batch); expect( snapshots?.some( (conversation) => - conversation.header.knowledgeRetrievalMode === 'always' - ) - ).toBe(true) - }) - }) + conversation.header.knowledgeRetrievalMode === "always", + ), + ).toBe(true); + }); + }); - it('keeps a running response visible when cancellation fails', async () => { + it("keeps a running response visible when cancellation fails", async () => { vi.mocked(api.agent.cancel).mockRejectedValueOnce( - new Error('cancel failed') - ) - render() + new Error("cancel failed"), + ); + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '开始一个长任务' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "开始一个长任务" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); - fireEvent.click(await screen.findByLabelText('停止生成')) + fireEvent.click(await screen.findByLabelText("停止生成")); - await waitFor(() => - expect(api.agent.cancel).toHaveBeenCalledOnce() - ) + await waitFor(() => expect(api.agent.cancel).toHaveBeenCalledOnce()); expect( - await screen.findByText(/停止生成失败，请重试/u) - ).toBeInTheDocument() - expect(screen.getByLabelText('停止生成')).toBeInTheDocument() - vi.useFakeTimers() + await screen.findByText(/停止生成失败，请重试/u), + ).toBeInTheDocument(); + expect(screen.getByLabelText("停止生成")).toBeInTheDocument(); + vi.useFakeTimers(); try { - act(() => vi.advanceTimersByTime(10_000)) - expect( - screen.getByText(/停止生成失败，请重试/u) - ).toBeInTheDocument() + act(() => vi.advanceTimersByTime(10_000)); + expect(screen.getByText(/停止生成失败，请重试/u)).toBeInTheDocument(); fireEvent.click( - screen.getByRole('button', { - name: '关闭通知' - }) - ) + screen.getByRole("button", { + name: "关闭通知", + }), + ); expect( - screen.queryByText(/停止生成失败，请重试/u) - ).not.toBeInTheDocument() + screen.queryByText(/停止生成失败，请重试/u), + ).not.toBeInTheDocument(); } finally { - vi.useRealTimers() + vi.useRealTimers(); } - }) + }); - it('locks agent context controls while a response is running', async () => { - render() + it("locks agent context controls while a response is running", async () => { + render(); - const expertButton = composerMenuTrigger('专家角色') - const modeButton = composerMenuTrigger('工作模式') - const runtimeButton = await screen.findByRole('button', { - name: /sonnet-5/u - }) - expect(expertButton).toBeEnabled() - expect(modeButton).toBeEnabled() - expect(runtimeButton).toBeEnabled() + const expertButton = composerMenuTrigger("专家角色"); + const modeButton = composerMenuTrigger("工作模式"); + const runtimeButton = await screen.findByRole("button", { + name: /sonnet-5/u, + }); + expect(expertButton).toBeEnabled(); + expect(modeButton).toBeEnabled(); + expect(runtimeButton).toBeEnabled(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '检查运行上下文锁定' } - }) - openComposerMenu('专家角色') - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "检查运行上下文锁定" }, + }); + openComposerMenu("专家角色"); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); expect( - screen.queryByRole('menu', { name: '专家角色' }) - ).not.toBeInTheDocument() - expect(expertButton).toBeDisabled() - expect(modeButton).toBeDisabled() - expect(runtimeButton).toBeDisabled() + screen.queryByRole("menu", { name: "专家角色" }), + ).not.toBeInTheDocument(); + expect(expertButton).toBeDisabled(); + expect(modeButton).toBeDisabled(); + expect(runtimeButton).toBeDisabled(); - const request = run.mock.calls[0]?.[0] + const request = run.mock.calls[0]?.[0]; act(() => { if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); - await waitFor(() => expect(expertButton).toBeEnabled()) - expect(modeButton).toBeEnabled() - expect(runtimeButton).toBeEnabled() - }) + await waitFor(() => expect(expertButton).toBeEnabled()); + expect(modeButton).toBeEnabled(); + expect(runtimeButton).toBeEnabled(); + }); - it('queues another ordinary message while the Conversation is running', async () => { - render() + it("queues another ordinary message while the Conversation is running", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '第一条长消息' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "第一条长消息" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); const secondItem = { - id: '00000000-0000-4000-8000-000000000941', + id: "00000000-0000-4000-8000-000000000941", conversationId: run.mock.calls[0]![0].conversationId, - source: 'user' as const, - label: '第二条排队消息', - createdAt: '2026-08-20T09:01:00.000Z' - } - vi.mocked(api.conversationQueue.enqueueUser) - .mockResolvedValueOnce(secondItem) - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: secondItem.label } - }) - fireEvent.click( - await screen.findByLabelText('加入待发送队列') - ) + source: "user" as const, + label: "第二条排队消息", + createdAt: "2026-08-20T09:01:00.000Z", + }; + vi.mocked(api.conversationQueue.enqueueUser).mockResolvedValueOnce( + secondItem, + ); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: secondItem.label }, + }); + fireEvent.click(await screen.findByLabelText("加入待发送队列")); await waitFor(() => - expect(api.conversationQueue.enqueueUser).toHaveBeenCalledTimes(2) - ) - expect(run).toHaveBeenCalledTimes(1) + expect(api.conversationQueue.enqueueUser).toHaveBeenCalledTimes(2), + ); + expect(run).toHaveBeenCalledTimes(1); expect( - screen.queryByText(/当前对话已有任务正在运行/u) - ).not.toBeInTheDocument() + screen.queryByText(/当前对话已有任务正在运行/u), + ).not.toBeInTheDocument(); - const firstRequest = run.mock.calls[0]![0] + const firstRequest = run.mock.calls[0]![0]; act(() => { agentListener?.({ requestId: firstRequest.requestId, - type: 'done' - }) - }) - const secondInput = vi.mocked( - api.conversationQueue.enqueueUser - ).mock.calls[1]![0] + type: "done", + }); + }); + const secondInput = vi.mocked(api.conversationQueue.enqueueUser).mock + .calls[1]![0]; act(() => { conversationQueueDispatchListener?.({ item: secondItem, - input: secondInput - }) - }) + input: secondInput, + }); + }); - await waitFor(() => expect(run).toHaveBeenCalledTimes(2)) + await waitFor(() => expect(run).toHaveBeenCalledTimes(2)); expect(run.mock.calls[1]![0]).toMatchObject({ queueItemId: secondItem.id, - prompt: secondItem.label - }) - }) + prompt: secondItem.label, + }); + }); - it('restores a queued message when Agent preflight rejects it', async () => { + it("restores a queued message when Agent preflight rejects it", async () => { run.mockRejectedValueOnce( new Error( - "Error invoking remote method 'agent:run': Error: Runtime 暂不可用" - ) - ) - render() + "Error invoking remote method 'agent:run': Error: Runtime 暂不可用", + ), + ); + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '需要稍后重试的消息' } - }) - fireEvent.click(await screen.findByLabelText('发送')) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "需要稍后重试的消息" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); await waitFor(() => - expect(api.conversationQueue.releaseUser).toHaveBeenCalledOnce() - ) + expect(api.conversationQueue.releaseUser).toHaveBeenCalledOnce(), + ); + expect(await screen.findByText("Runtime 暂不可用")).toBeInTheDocument(); expect( - await screen.findByText('Runtime 暂不可用') - ).toBeInTheDocument() - expect( - screen.queryByText(/Error invoking remote method/u) - ).not.toBeInTheDocument() - expect( - screen.queryByText('需要稍后重试的消息') - ).not.toBeInTheDocument() - expect( - screen.queryByLabelText('停止生成') - ).not.toBeInTheDocument() - }) + screen.queryByText(/Error invoking remote method/u), + ).not.toBeInTheDocument(); + expect(screen.queryByText("需要稍后重试的消息")).not.toBeInTheDocument(); + expect(screen.queryByLabelText("停止生成")).not.toBeInTheDocument(); + }); - it('persists the claimed queue identity with a dispatched user message', async () => { - render() + it("persists the claimed queue identity with a dispatched user message", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '只发送一次' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "只发送一次" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request?.queueItemId) { - throw new Error('Missing queued request') + throw new Error("Missing queued request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'error', - status: 'failed', - message: 'Remote Runtime failed' - }) - }) + type: "error", + status: "failed", + message: "Remote Runtime failed", + }); + }); await waitFor(() => expect( - vi.mocked(api.conversations.saveLocal).mock.calls.some( - ([batch]) => + vi + .mocked(api.conversations.saveLocal) + .mock.calls.some(([batch]) => batch.some((conversation) => conversation.messages.some( (message) => - message.role === 'user' && - message.content === '只发送一次' && - message.queueItemId === request.queueItemId - ) - ) - ) - ).toBe(true) - ) - }) + message.role === "user" && + message.content === "只发送一次" && + message.queueItemId === request.queueItemId, + ), + ), + ), + ).toBe(true), + ); + }); - it('keeps sent documents and images in conversation history', async () => { + it("keeps sent documents and images in conversation history", async () => { const documentAttachment = { - id: '00000000-0000-4000-8000-000000000301', - name: '需求说明.md', + id: "00000000-0000-4000-8000-000000000301", + name: "需求说明.md", size: 2_048, - preview: '需要保留在用户消息中的文档', - kind: 'text' as const - } + preview: "需要保留在用户消息中的文档", + kind: "text" as const, + }; const imageAttachment = { - id: '00000000-0000-4000-8000-000000000302', - name: '页面截图.png', + id: "00000000-0000-4000-8000-000000000302", + name: "页面截图.png", size: 4_096, - preview: '1280 × 720', - kind: 'image' as const, - thumbnailUrl: - 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAAB', - contentUrl: - 'data:image/jpeg;base64,/9j/4AAQSkZJRgABAQEASABIAAD/2Q==' - } + preview: "1280 × 720", + kind: "image" as const, + thumbnailUrl: "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAAB", + contentUrl: "data:image/jpeg;base64,/9j/4AAQSkZJRgABAQEASABIAAD/2Q==", + }; vi.mocked(api.context.selectFiles).mockResolvedValueOnce([ documentAttachment, - imageAttachment - ]) - render() + imageAttachment, + ]); + render(); - fireEvent.click(await screen.findByLabelText('添加附件')) + fireEvent.click(await screen.findByLabelText("添加附件")); const composer = screen - .getByLabelText('向 GoodBuddy 提问') - .closest('.composer') - expect(composer).not.toBeNull() + .getByLabelText("向 GoodBuddy 提问") + .closest(".composer"); + expect(composer).not.toBeNull(); if (!composer) { - return + return; } expect( - await within(composer).findByText('需求说明.md') - ).toBeInTheDocument() - expect( - within(composer).getByText('页面截图.png') - ).toBeInTheDocument() - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '分析这些附件' } - }) - fireEvent.click(screen.getByLabelText('发送')) + await within(composer).findByText("需求说明.md"), + ).toBeInTheDocument(); + expect(within(composer).getByText("页面截图.png")).toBeInTheDocument(); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "分析这些附件" }, + }); + fireEvent.click(screen.getByLabelText("发送")); - await waitFor(() => expect(run).toHaveBeenCalledOnce()) + await waitFor(() => expect(run).toHaveBeenCalledOnce()); expect(run.mock.calls[0]?.[0].contextIds).toEqual([ documentAttachment.id, - imageAttachment.id - ]) + imageAttachment.id, + ]); const userArticle = screen - .getAllByText('分析这些附件') - .map((element) => element.closest('article')) - .find((element) => element?.classList.contains('message--user')) - expect(userArticle).not.toBeNull() + .getAllByText("分析这些附件") + .map((element) => element.closest("article")) + .find((element) => element?.classList.contains("message--user")); + expect(userArticle).not.toBeNull(); if (!userArticle) { - return + return; } const anchorClick = vi - .spyOn(HTMLAnchorElement.prototype, 'click') - .mockImplementation(() => {}) - expect(within(userArticle).getByText('需求说明.md')).toBeInTheDocument() - expect(within(userArticle).getByText('2 KB')).toBeInTheDocument() + .spyOn(HTMLAnchorElement.prototype, "click") + .mockImplementation(() => {}); + expect(within(userArticle).getByText("需求说明.md")).toBeInTheDocument(); + expect(within(userArticle).getByText("2 KB")).toBeInTheDocument(); expect( - within(userArticle).getByRole('img', { name: '页面截图.png' }) - ).toHaveAttribute('src', imageAttachment.contentUrl) - const viewerTrigger = within(userArticle).getByRole('button', { - name: '查看图片 页面截图.png' - }) - fireEvent.click(viewerTrigger) - const imageDialog = await screen.findByRole('dialog', { - name: '页面截图.png' - }) + within(userArticle).getByRole("img", { name: "页面截图.png" }), + ).toHaveAttribute("src", imageAttachment.contentUrl); + const viewerTrigger = within(userArticle).getByRole("button", { + name: "查看图片 页面截图.png", + }); + fireEvent.click(viewerTrigger); + const imageDialog = await screen.findByRole("dialog", { + name: "页面截图.png", + }); expect( - within(imageDialog).getByRole('img', { name: '页面截图.png' }) - ).toHaveAttribute('src', imageAttachment.contentUrl) - const closeViewer = within(imageDialog).getByRole('button', { - name: '关闭图片查看器' - }) - expect(closeViewer).toHaveFocus() - expect(document.querySelector('main')?.inert).toBe(true) - fireEvent.keyDown(closeViewer, { key: 'Tab' }) + within(imageDialog).getByRole("img", { name: "页面截图.png" }), + ).toHaveAttribute("src", imageAttachment.contentUrl); + const closeViewer = within(imageDialog).getByRole("button", { + name: "关闭图片查看器", + }); + expect(closeViewer).toHaveFocus(); + expect(document.querySelector("main")?.inert).toBe(true); + fireEvent.keyDown(closeViewer, { key: "Tab" }); expect( - within(imageDialog).getByRole('button', { - name: '下载图片' - }) - ).toHaveFocus() - fireEvent.keyDown(imageDialog, { key: 'Escape' }) - await waitFor(() => - expect(viewerTrigger).toHaveFocus() - ) + within(imageDialog).getByRole("button", { + name: "下载图片", + }), + ).toHaveFocus(); + fireEvent.keyDown(imageDialog, { key: "Escape" }); + await waitFor(() => expect(viewerTrigger).toHaveFocus()); expect( - screen.queryByRole('dialog', { name: '页面截图.png' }) - ).not.toBeInTheDocument() - expect(document.querySelector('main')?.inert).toBe(false) + screen.queryByRole("dialog", { name: "页面截图.png" }), + ).not.toBeInTheDocument(); + expect(document.querySelector("main")?.inert).toBe(false); fireEvent.click( - within(userArticle).getByRole('button', { - name: '下载图片 页面截图.png' - }) - ) - expect(anchorClick).toHaveBeenCalledOnce() + within(userArticle).getByRole("button", { + name: "下载图片 页面截图.png", + }), + ); + expect(anchorClick).toHaveBeenCalledOnce(); await waitFor( () => expect(api.conversations.saveLocal).toHaveBeenCalledWith( @@ -5405,608 +5065,594 @@ describe('App', () => { expect.objectContaining({ messages: expect.arrayContaining([ expect.objectContaining({ - role: 'user', - attachments: [ - documentAttachment, - imageAttachment - ] - }) - ]) - }) - ]) + role: "user", + attachments: [documentAttachment, imageAttachment], + }), + ]), + }), + ]), ), - { timeout: 2_000 } - ) - }) + { timeout: 2_000 }, + ); + }); - it('shows attachment parsing progress and prevents duplicate selection', async () => { + it("shows attachment parsing progress and prevents duplicate selection", async () => { const attachment = { - id: '00000000-0000-4000-8000-000000000309', - name: '扫描材料.pdf', + id: "00000000-0000-4000-8000-000000000309", + name: "扫描材料.pdf", size: 8_705_692, - preview: '解析后的文档', - kind: 'text' as const - } + preview: "解析后的文档", + kind: "text" as const, + }; let resolveSelection: - | ((attachments: ContextAttachment[]) => void) - | undefined + ((attachments: ContextAttachment[]) => void) | undefined; vi.mocked(api.context.selectFiles).mockImplementationOnce( () => new Promise((resolve) => { - resolveSelection = resolve - }) - ) - render() + resolveSelection = resolve; + }), + ); + render(); - const addButton = await screen.findByLabelText('添加附件') - fireEvent.click(addButton) + const addButton = await screen.findByLabelText("添加附件"); + fireEvent.click(addButton); - expect(addButton).toBeDisabled() + expect(addButton).toBeDisabled(); expect( - screen.getByRole('progressbar', { - name: '附件读取与解析进度' - }) - ).toBeInTheDocument() - expect(screen.getByText('正在选择附件…')).toBeInTheDocument() + screen.getByRole("progressbar", { + name: "附件读取与解析进度", + }), + ).toBeInTheDocument(); + expect(screen.getByText("正在选择附件…")).toBeInTheDocument(); act(() => { fileSelectionProgressListener?.({ - phase: 'parsing', - fileName: '扫描材料.pdf', + phase: "parsing", + fileName: "扫描材料.pdf", fileNumber: 1, - fileCount: 1 - }) - }) - expect(screen.getByText('正在解析 扫描材料.pdf')).toBeInTheDocument() - expect(screen.getByText('第 1 / 1 个文件')).toBeInTheDocument() - fireEvent.click(addButton) - expect(api.context.selectFiles).toHaveBeenCalledOnce() + fileCount: 1, + }); + }); + expect(screen.getByText("正在解析 扫描材料.pdf")).toBeInTheDocument(); + expect(screen.getByText("第 1 / 1 个文件")).toBeInTheDocument(); + fireEvent.click(addButton); + expect(api.context.selectFiles).toHaveBeenCalledOnce(); - act(() => resolveSelection?.([attachment])) - expect(await screen.findByText('扫描材料.pdf')).toBeInTheDocument() + act(() => resolveSelection?.([attachment])); + expect(await screen.findByText("扫描材料.pdf")).toBeInTheDocument(); await waitFor(() => { - expect(addButton).toBeEnabled() + expect(addButton).toBeEnabled(); expect( - screen.queryByRole('progressbar', { - name: '附件读取与解析进度' - }) - ).not.toBeInTheDocument() - }) - }) + screen.queryByRole("progressbar", { + name: "附件读取与解析进度", + }), + ).not.toBeInTheDocument(); + }); + }); - it('associates attachment errors with the Composer controls', async () => { + it("associates attachment errors with the Composer controls", async () => { vi.mocked(api.context.selectFiles).mockRejectedValueOnce( - new Error('文件解析失败，请重新选择') - ) - render() + new Error("文件解析失败，请重新选择"), + ); + render(); - const addButton = await screen.findByLabelText('添加附件') - const input = screen.getByLabelText('向 GoodBuddy 提问') - fireEvent.click(addButton) + const addButton = await screen.findByLabelText("添加附件"); + const input = screen.getByLabelText("向 GoodBuddy 提问"); + fireEvent.click(addButton); - const error = await screen.findByRole('alert', { - name: '文件解析失败，请重新选择' - }) - expect(error).toHaveAttribute('id', 'composer-context-error') - expect(input).toHaveAttribute('aria-invalid', 'true') - expect(input).toHaveAttribute( - 'aria-describedby', - 'composer-context-error' - ) - expect(addButton).toHaveAttribute('aria-invalid', 'true') + const error = await screen.findByRole("alert", { + name: "文件解析失败，请重新选择", + }); + expect(error).toHaveAttribute("id", "composer-context-error"); + expect(input).toHaveAttribute("aria-invalid", "true"); + expect(input).toHaveAttribute("aria-describedby", "composer-context-error"); + expect(addButton).toHaveAttribute("aria-invalid", "true"); expect(addButton).toHaveAttribute( - 'aria-describedby', - 'composer-context-error' - ) - }) + "aria-describedby", + "composer-context-error", + ); + }); - it('sends and renders five selected images together', async () => { + it("sends and renders five selected images together", async () => { const imageAttachments = Array.from({ length: 5 }, (_, index) => ({ id: `00000000-0000-4000-8000-00000000031${index}`, name: `参考图-${index + 1}.png`, size: 4_096, - preview: '640 × 480', - kind: 'image' as const, - thumbnailUrl: - 'data:image/jpeg;base64,/9j/4AAQSkZJRgABAQEASABIAAD/2Q==', - contentUrl: - 'data:image/jpeg;base64,/9j/4AAQSkZJRgABAQEASABIAAD/2Q==' - })) - vi.mocked(api.context.selectFiles).mockResolvedValueOnce(imageAttachments) - render() + preview: "640 × 480", + kind: "image" as const, + thumbnailUrl: "data:image/jpeg;base64,/9j/4AAQSkZJRgABAQEASABIAAD/2Q==", + contentUrl: "data:image/jpeg;base64,/9j/4AAQSkZJRgABAQEASABIAAD/2Q==", + })); + vi.mocked(api.context.selectFiles).mockResolvedValueOnce(imageAttachments); + render(); - fireEvent.click(await screen.findByLabelText('添加附件')) + fireEvent.click(await screen.findByLabelText("添加附件")); const composer = screen - .getByLabelText('向 GoodBuddy 提问') - .closest('.composer') - expect(composer).not.toBeNull() + .getByLabelText("向 GoodBuddy 提问") + .closest(".composer"); + expect(composer).not.toBeNull(); if (!composer) { - return + return; } await waitFor(() => - expect( - within(composer).getAllByText(/^参考图-\d\.png$/u) - ).toHaveLength(5) - ) - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '比较这五张图片' } - }) - fireEvent.click(screen.getByLabelText('发送')) + expect(within(composer).getAllByText(/^参考图-\d\.png$/u)).toHaveLength( + 5, + ), + ); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "比较这五张图片" }, + }); + fireEvent.click(screen.getByLabelText("发送")); - await waitFor(() => expect(run).toHaveBeenCalledOnce()) + await waitFor(() => expect(run).toHaveBeenCalledOnce()); expect(run.mock.calls[0]?.[0].contextIds).toEqual( - imageAttachments.map((attachment) => attachment.id) - ) + imageAttachments.map((attachment) => attachment.id), + ); const userArticle = screen - .getAllByText('比较这五张图片') - .map((element) => element.closest('article')) - .find((element) => element?.classList.contains('message--user')) - expect(userArticle).not.toBeNull() + .getAllByText("比较这五张图片") + .map((element) => element.closest("article")) + .find((element) => element?.classList.contains("message--user")); + expect(userArticle).not.toBeNull(); if (!userArticle) { - return + return; } - expect(within(userArticle).getAllByRole('img')).toHaveLength(5) - expect(within(userArticle).getByLabelText('消息附件')).toHaveClass( - 'message-attachments' - ) - }) + expect(within(userArticle).getAllByRole("img")).toHaveLength(5); + expect(within(userArticle).getByLabelText("消息附件")).toHaveClass( + "message-attachments", + ); + }); - it('accepts pasted images without intercepting pasted text', async () => { + it("accepts pasted images without intercepting pasted text", async () => { vi.mocked(api.context.addPastedImage).mockResolvedValueOnce({ - id: '00000000-0000-4000-8000-000000000303', - name: '粘贴图片.jpg', + id: "00000000-0000-4000-8000-000000000303", + name: "粘贴图片.jpg", size: 120_000, - preview: '1280 × 800', - kind: 'image', - thumbnailUrl: - 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAAB' - }) + preview: "1280 × 800", + kind: "image", + thumbnailUrl: "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAAB", + }); const pastedImage = new File( [Uint8Array.from([0x89, 0x50, 0x4e, 0x47])], - 'pasted.png', - { type: 'image/png' } - ) - render() + "pasted.png", + { type: "image/png" }, + ); + render(); - const input = await screen.findByLabelText('向 GoodBuddy 提问') + const input = await screen.findByLabelText("向 GoodBuddy 提问"); expect( fireEvent.paste(input, { clipboardData: { items: [ { getAsFile: () => null, - kind: 'string', - type: 'text/plain' - } - ] - } - }) - ).toBe(true) - expect(api.context.addPastedImage).not.toHaveBeenCalled() + kind: "string", + type: "text/plain", + }, + ], + }, + }), + ).toBe(true); + expect(api.context.addPastedImage).not.toHaveBeenCalled(); fireEvent.paste(input, { clipboardData: { items: [ { getAsFile: () => pastedImage, - kind: 'file', - type: 'image/png' - } - ] - } - }) + kind: "file", + type: "image/png", + }, + ], + }, + }); await waitFor(() => expect(api.context.addPastedImage).toHaveBeenCalledWith({ data: Uint8Array.from([0x89, 0x50, 0x4e, 0x47]), - mimeType: 'image/png' - }) - ) - expect(api.context.addPastedImage).toHaveBeenCalledTimes(1) - expect(api.context.readClipboard).not.toHaveBeenCalled() - const composer = input.closest('.composer') - expect(composer).not.toBeNull() + mimeType: "image/png", + }), + ); + expect(api.context.addPastedImage).toHaveBeenCalledTimes(1); + expect(api.context.readClipboard).not.toHaveBeenCalled(); + const composer = input.closest(".composer"); + expect(composer).not.toBeNull(); if (!composer) { - return + return; } expect( - await within(composer).findByText('粘贴图片.jpg') - ).toBeInTheDocument() + await within(composer).findByText("粘贴图片.jpg"), + ).toBeInTheDocument(); expect( - within(composer).queryByRole('button', { - name: '截取当前屏幕' - }) - ).not.toBeInTheDocument() + within(composer).queryByRole("button", { + name: "截取当前屏幕", + }), + ).not.toBeInTheDocument(); expect( - within(composer).queryByRole('button', { - name: '捕获应用窗口' - }) - ).not.toBeInTheDocument() + within(composer).queryByRole("button", { + name: "捕获应用窗口", + }), + ).not.toBeInTheDocument(); expect( - within(composer).queryByRole('button', { - name: '读取剪贴板' - }) - ).not.toBeInTheDocument() - }) + within(composer).queryByRole("button", { + name: "读取剪贴板", + }), + ).not.toBeInTheDocument(); + }); - it('keeps a draft in chat when Enter is pressed while the runtime loads', async () => { - vi.mocked(api.agent.getStatus).mockReturnValue( - new Promise(() => {}) - ) - render() + it("keeps a draft in chat when Enter is pressed while the runtime loads", async () => { + vi.mocked(api.agent.getStatus).mockReturnValue(new Promise(() => {})); + render(); - const composer = screen.getByLabelText('向 GoodBuddy 提问') + const composer = screen.getByLabelText("向 GoodBuddy 提问"); fireEvent.change(composer, { - target: { value: '等待 Runtime' } - }) - fireEvent.keyDown(composer, { key: 'Enter' }) + target: { value: "等待 Runtime" }, + }); + fireEvent.keyDown(composer, { key: "Enter" }); - expect(composer).toHaveValue('等待 Runtime') + expect(composer).toHaveValue("等待 Runtime"); expect( - screen.queryByRole('heading', { name: '设置中心' }) - ).not.toBeInTheDocument() + screen.queryByRole("heading", { name: "设置中心" }), + ).not.toBeInTheDocument(); expect( - await screen.findByText('Agent Runtime 正在加载，请稍后重试') - ).toBeInTheDocument() - expect(run).not.toHaveBeenCalled() - }) + await screen.findByText("Agent Runtime 正在加载，请稍后重试"), + ).toBeInTheDocument(); + expect(run).not.toHaveBeenCalled(); + }); - it('keeps a new-conversation draft in chat when the runtime is unavailable', async () => { + it("keeps a new-conversation draft in chat when the runtime is unavailable", async () => { vi.mocked(api.agent.getStatus).mockResolvedValue({ - id: 'setup', - label: '需要配置模型', + id: "setup", + label: "需要配置模型", available: false, supportsToolExecution: false, - detail: '请配置模型' - }) - render() + detail: "请配置模型", + }); + render(); expect( - await screen.findByRole('heading', { name: '设置中心' }) - ).toBeInTheDocument() - const newConversation = screen.getByRole('button', { - name: /新建对话/u - }) - fireEvent.click(newConversation) + await screen.findByRole("heading", { name: "设置中心" }), + ).toBeInTheDocument(); + const newConversation = screen.getByRole("button", { + name: /新建对话/u, + }); + fireEvent.click(newConversation); - const composer = screen.getByLabelText('向 GoodBuddy 提问') - await waitFor(() => expect(composer).toHaveFocus()) + const composer = screen.getByLabelText("向 GoodBuddy 提问"); + await waitFor(() => expect(composer).toHaveFocus()); fireEvent.change(composer, { - target: { value: '保留这条草稿' } - }) - fireEvent.keyDown(composer, { key: 'Enter' }) + target: { value: "保留这条草稿" }, + }); + fireEvent.keyDown(composer, { key: "Enter" }); - expect(composer).toHaveValue('保留这条草稿') + expect(composer).toHaveValue("保留这条草稿"); expect( - screen.queryByRole('heading', { name: '设置中心' }) - ).not.toBeInTheDocument() - expect(screen.getByLabelText('发送')).toBeDisabled() - expect(run).not.toHaveBeenCalled() - }) + screen.queryByRole("heading", { name: "设置中心" }), + ).not.toBeInTheDocument(); + expect(screen.getByLabelText("发送")).toBeDisabled(); + expect(run).not.toHaveBeenCalled(); + }); - it('opens chat and focuses the composer for tray conversations', async () => { - render() + it("opens chat and focuses the composer for tray conversations", async () => { + render(); - fireEvent.click(await screen.findByText('本地工作区')) + fireEvent.click(await screen.findByText("本地工作区")); expect( - await screen.findByRole('heading', { name: '设置中心' }) - ).toBeInTheDocument() + await screen.findByRole("heading", { name: "设置中心" }), + ).toBeInTheDocument(); - act(() => newConversationListener?.()) + act(() => newConversationListener?.()); - const composer = await screen.findByLabelText('向 GoodBuddy 提问') - await waitFor(() => expect(composer).toHaveFocus()) - }) + const composer = await screen.findByLabelText("向 GoodBuddy 提问"); + await waitFor(() => expect(composer).toHaveFocus()); + }); - it('reuses the active empty conversation and preserves its draft', async () => { - render() + it("reuses the active empty conversation and preserves its draft", async () => { + render(); - const composer = await screen.findByLabelText('向 GoodBuddy 提问') + const composer = await screen.findByLabelText("向 GoodBuddy 提问"); fireEvent.change(composer, { - target: { value: '尚未发送的草稿' } - }) - const newConversation = screen.getByRole('button', { - name: /新建对话/u - }) - fireEvent.click(newConversation) - fireEvent.click(newConversation) + target: { value: "尚未发送的草稿" }, + }); + const newConversation = screen.getByRole("button", { + name: /新建对话/u, + }); + fireEvent.click(newConversation); + fireEvent.click(newConversation); - expect(composer).toHaveValue('尚未发送的草稿') + expect(composer).toHaveValue("尚未发送的草稿"); expect( - screen.getAllByRole('button', { - name: '更多会话操作 新对话' - }) - ).toHaveLength(1) - }) + screen.getAllByRole("button", { + name: "更多会话操作 新对话", + }), + ).toHaveLength(1); + }); - it('coalesces batched new-conversation requests after a used conversation', async () => { - render() + it("coalesces batched new-conversation requests after a used conversation", async () => { + render(); - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '已有内容' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const requestId = run.mock.calls[0]?.[0].requestId + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "已有内容" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const requestId = run.mock.calls[0]?.[0].requestId; act(() => { if (requestId) { - agentListener?.({ requestId, type: 'done' }) + agentListener?.({ requestId, type: "done" }); } - newConversationListener?.() - newConversationListener?.() - }) + newConversationListener?.(); + newConversationListener?.(); + }); await waitFor(() => expect( - screen.getAllByRole('button', { - name: /^更多会话操作/u - }) - ).toHaveLength(2) - ) - }) + screen.getAllByRole("button", { + name: /^更多会话操作/u, + }), + ).toHaveLength(2), + ); + }); - it('opens a workspace Markdown file in the right-side preview', async () => { + it("opens a workspace Markdown file in the right-side preview", async () => { vi.mocked(api.workspace.listDirectory).mockResolvedValue({ - path: '', + path: "", entries: [ { - name: 'README.md', - path: 'README.md', - type: 'file' - } + name: "README.md", + path: "README.md", + type: "file", + }, ], - truncated: false - }) + truncated: false, + }); vi.mocked(api.workspace.readFile).mockResolvedValue({ - path: 'README.md', - name: 'README.md', - content: '# 工作区说明', - mimeType: 'text/markdown', + path: "README.md", + name: "README.md", + content: "# 工作区说明", + mimeType: "text/markdown", size: 19, offsetBytes: 0, nextOffsetBytes: 19, - truncated: false - }) - render() + truncated: false, + }); + render(); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - fireEvent.click(await screen.findByRole('tab', { name: '工作区' })) - fireEvent.click( - await screen.findByRole('button', { name: 'README.md' }) - ) + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + fireEvent.click(await screen.findByRole("tab", { name: "工作区" })); + fireEvent.click(await screen.findByRole("button", { name: "README.md" })); expect( - await screen.findByRole('heading', { name: '工作区说明' }) - ).toBeInTheDocument() - expect( - screen.getByRole('tab', { name: '工作区' }) - ).toHaveAttribute('aria-selected', 'true') - expect( - screen.queryByRole('tab', { name: '预览' }) - ).not.toBeInTheDocument() + await screen.findByRole("heading", { name: "工作区说明" }), + ).toBeInTheDocument(); + expect(screen.getByRole("tab", { name: "工作区" })).toHaveAttribute( + "aria-selected", + "true", + ); + expect(screen.queryByRole("tab", { name: "预览" })).not.toBeInTheDocument(); expect(api.workspace.readFile).toHaveBeenCalledWith( projectId, - 'README.md', - 0 - ) - fireEvent.click( - screen.getByRole('button', { name: '返回工作区' }) - ) + "README.md", + 0, + ); + fireEvent.click(screen.getByRole("button", { name: "返回工作区" })); expect( - await screen.findByRole('button', { name: 'README.md' }) - ).toBeInTheDocument() - }) + await screen.findByRole("button", { name: "README.md" }), + ).toBeInTheDocument(); + }); - it('opens multiple current-project terminals without a second target picker', async () => { - render() + it("opens multiple current-project terminals without a second target picker", async () => { + render(); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - const add = await screen.findByRole('button', { - name: '打开工作栏应用' - }) + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + const add = await screen.findByRole("button", { + name: "打开工作栏应用", + }); for (let count = 1; count <= 2; count += 1) { - fireEvent.click(add) + fireEvent.click(add); fireEvent.click( - await screen.findByRole('button', { - name: /打开当前项目的用户终端/u - }) - ) + await screen.findByRole("button", { + name: /打开当前项目的用户终端/u, + }), + ); expect( - await screen.findByRole('tab', { - name: `终端 ${count}` - }) - ).toHaveAttribute('aria-selected', 'true') + await screen.findByRole("tab", { + name: `终端 ${count}`, + }), + ).toHaveAttribute("aria-selected", "true"); } expect( - screen.getByRole('button', { name: '打开工作栏应用' }) - ).toBeVisible() - expect( - screen.getAllByRole('tab', { name: /^终端 \d+$/u }) - ).toHaveLength(2) - }) + screen.getByRole("button", { name: "打开工作栏应用" }), + ).toBeVisible(); + expect(screen.getAllByRole("tab", { name: /^终端 \d+$/u })).toHaveLength(2); + }); - it('opens workspace entries from their row actions', async () => { + it("opens workspace entries from their row actions", async () => { vi.mocked(api.workspace.listDirectory).mockResolvedValue({ - path: '', + path: "", entries: [ - { name: 'docs', path: 'docs', type: 'directory' }, - { name: 'README.md', path: 'README.md', type: 'file' } + { name: "docs", path: "docs", type: "directory" }, + { name: "README.md", path: "README.md", type: "file" }, ], - truncated: false - }) - render() + truncated: false, + }); + render(); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - fireEvent.click(await screen.findByRole('tab', { name: '工作区' })) + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + fireEvent.click(await screen.findByRole("tab", { name: "工作区" })); fireEvent.click( - await screen.findByRole('button', { - name: '在系统资源管理器中打开文件夹 docs' - }) - ) + await screen.findByRole("button", { + name: "在系统资源管理器中打开文件夹 docs", + }), + ); fireEvent.click( - screen.getByRole('button', { - name: '使用默认应用打开文件 README.md' - }) - ) + screen.getByRole("button", { + name: "使用默认应用打开文件 README.md", + }), + ); await waitFor(() => expect(api.workspace.openPath).toHaveBeenCalledWith( projectId, - 'docs', - 'directory' - ) - ) + "docs", + "directory", + ), + ); expect(api.workspace.openPath).toHaveBeenCalledWith( projectId, - 'README.md', - 'file' - ) - }) + "README.md", + "file", + ); + }); - it('refreshes generated workspace files when a run completes', async () => { - vi.mocked(api.workspace.listDirectory).mockImplementation(async (_projectId, path) => ({ - path, - entries: vi.mocked(api.workspace.getChanges).mock.calls.length >= 2 - ? [{ name: 'tree-only.txt', path: 'tree-only.txt', type: 'file' as const }] - : [], - truncated: false - })) + it("refreshes generated workspace files when a run completes", async () => { + vi.mocked(api.workspace.listDirectory).mockImplementation( + async (_projectId, path) => ({ + path, + entries: + vi.mocked(api.workspace.getChanges).mock.calls.length >= 2 + ? [ + { + name: "tree-only.txt", + path: "tree-only.txt", + type: "file" as const, + }, + ] + : [], + truncated: false, + }), + ); vi.mocked(api.workspace.getChanges) .mockResolvedValueOnce({ rootPath: project.rootPath, available: true, - status: '', - patch: '', + status: "", + patch: "", files: [], - truncated: false + truncated: false, }) .mockResolvedValueOnce({ rootPath: project.rootPath, available: true, - status: '?? generated.md', - patch: '', - files: [{ path: 'generated.md', status: '??' }], - truncated: false - }) - render() + status: "?? generated.md", + patch: "", + files: [{ path: "generated.md", status: "??" }], + truncated: false, + }); + render(); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - fireEvent.click(await screen.findByRole('tab', { name: '工作区' })) + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + fireEvent.click(await screen.findByRole("tab", { name: "工作区" })); await waitFor(() => - expect(api.workspace.getChanges).toHaveBeenCalledOnce() - ) - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '生成文件' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const requestId = run.mock.calls[0]?.[0].requestId + expect(api.workspace.getChanges).toHaveBeenCalledOnce(), + ); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "生成文件" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const requestId = run.mock.calls[0]?.[0].requestId; act(() => { if (requestId) { - agentListener?.({ requestId, type: 'done' }) + agentListener?.({ requestId, type: "done" }); } - }) + }); - expect(await screen.findByText('generated.md')).toBeInTheDocument() - expect(await screen.findByText('tree-only.txt')).toBeInTheDocument() - }) + expect(await screen.findByText("generated.md")).toBeInTheDocument(); + expect(await screen.findByText("tree-only.txt")).toBeInTheDocument(); + }); - it('ignores stale Git changes after switching projects', async () => { + it("ignores stale Git changes after switching projects", async () => { const secondProject = { ...project, - id: '00000000-0000-4000-8000-000000000102', - name: '第二项目', - rootPath: 'C:\\Second', + id: "00000000-0000-4000-8000-000000000102", + name: "第二项目", + rootPath: "C:\\Second", executionSpace: { - kind: 'local' as const, - rootPath: 'C:\\Second' - } - } + kind: "local" as const, + rootPath: "C:\\Second", + }, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - secondProject - ]) + secondProject, + ]); let resolveFirst: - | ((value: Awaited>) => void) - | undefined + | (( + value: Awaited>, + ) => void) + | undefined; let resolveSecond: - | ((value: Awaited>) => void) - | undefined + | (( + value: Awaited>, + ) => void) + | undefined; vi.mocked(api.workspace.getChanges) .mockImplementationOnce( () => new Promise((resolve) => { - resolveFirst = resolve - }) + resolveFirst = resolve; + }), ) .mockImplementationOnce( () => new Promise((resolve) => { - resolveSecond = resolve - }) - ) - render() + resolveSecond = resolve; + }), + ); + render(); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - fireEvent.click(await screen.findByRole('tab', { name: '工作区' })) + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + fireEvent.click(await screen.findByRole("tab", { name: "工作区" })); await waitFor(() => - expect(api.workspace.getChanges).toHaveBeenCalledWith(projectId) - ) - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - selectProjectOption(secondProject.name) + expect(api.workspace.getChanges).toHaveBeenCalledWith(projectId), + ); + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + selectProjectOption(secondProject.name); await waitFor(() => - expect(api.workspace.getChanges).toHaveBeenCalledWith( - secondProject.id - ) - ) + expect(api.workspace.getChanges).toHaveBeenCalledWith(secondProject.id), + ); resolveSecond?.({ rootPath: secondProject.rootPath, available: true, - status: '?? second.md', - patch: '', - files: [{ path: 'second.md', status: '??' }], - truncated: false - }) - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - expect(await screen.findByText('second.md')).toBeInTheDocument() + status: "?? second.md", + patch: "", + files: [{ path: "second.md", status: "??" }], + truncated: false, + }); + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + expect(await screen.findByText("second.md")).toBeInTheDocument(); resolveFirst?.({ rootPath: project.rootPath, available: true, - status: '?? stale.md', - patch: '', - files: [{ path: 'stale.md', status: '??' }], - truncated: false - }) + status: "?? stale.md", + patch: "", + files: [{ path: "stale.md", status: "??" }], + truncated: false, + }); await waitFor(() => - expect(screen.queryByText('stale.md')).not.toBeInTheDocument() - ) - expect(screen.getByText('second.md')).toBeInTheDocument() - }) + expect(screen.queryByText("stale.md")).not.toBeInTheDocument(), + ); + expect(screen.getByText("second.md")).toBeInTheDocument(); + }); - it('applies and persists a dark appearance from Settings', async () => { - render() + it("applies and persists a dark appearance from Settings", async () => { + render(); - fireEvent.click(await screen.findByText('本地工作区')) - await screen.findByRole( - 'heading', - { name: '设置中心' }, - { timeout: 3000 } - ) - fireEvent.click(screen.getByRole('tab', { name: '外观' })) - fireEvent.click(screen.getByRole('radio', { name: /暗色/u })) + fireEvent.click(await screen.findByText("本地工作区")); + await screen.findByRole("heading", { name: "设置中心" }, { timeout: 3000 }); + fireEvent.click(screen.getByRole("tab", { name: "外观" })); + fireEvent.click(screen.getByRole("radio", { name: /暗色/u })); await waitFor(() => - expect(document.documentElement.dataset.theme).toBe('dark') - ) - expect(document.documentElement.style.colorScheme).toBe('dark') - expect(localStorage.getItem('goodbuddy.appearance-theme')).toBe( - 'dark' - ) - }) + expect(document.documentElement.dataset.theme).toBe("dark"), + ); + expect(document.documentElement.style.colorScheme).toBe("dark"); + expect(localStorage.getItem("goodbuddy.appearance-theme")).toBe("dark"); + }); - it('loads token usage in activity and refreshes it when a run finishes', async () => { + it("loads token usage in activity and refreshes it when a run finishes", async () => { vi.mocked(api.usage.getTokenSummary).mockResolvedValueOnce({ totals: { callCount: 1, @@ -6014,57 +5660,53 @@ describe('App', () => { output: 20, cacheRead: 0, cacheWrite: 0, - totalTokens: 120 + totalTokens: 120, }, records: [ { - requestId: 'usage-request-1', + requestId: "usage-request-1", projectId, projectName: project.name, - conversationId: 'usage-conversation-1', - conversationTitle: '用量会话', - runtime: 'model', - provider: 'anthropic', - model: 'sonnet-5', + conversationId: "usage-conversation-1", + conversationTitle: "用量会话", + runtime: "model", + provider: "anthropic", + model: "sonnet-5", callCount: 1, input: 100, output: 20, cacheRead: 0, cacheWrite: 0, - totalTokens: 120 - } - ] - }) + totalTokens: 120, + }, + ], + }); - render() - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '统计用量' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + render(); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "统计用量" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } - fireEvent.click(screen.getByText('运行记录')) - fireEvent.click( - await screen.findByRole('tab', { name: '用量统计' }) - ) - const stats = await screen.findByLabelText('Token 用量统计') + fireEvent.click(screen.getByText("运行记录")); + fireEvent.click(await screen.findByRole("tab", { name: "用量统计" })); + const stats = await screen.findByLabelText("Token 用量统计"); expect( - screen.getByRole('heading', { level: 1, name: '运行记录' }) - ).toBeInTheDocument() + screen.getByRole("heading", { level: 1, name: "运行记录" }), + ).toBeInTheDocument(); expect( - screen.queryByRole('button', { name: /^专家角色：/u }) - ).not.toBeInTheDocument() - expect( - screen.queryByLabelText('切换助手工作栏') - ).toBeInTheDocument() + screen.queryByRole("button", { name: /^专家角色：/u }), + ).not.toBeInTheDocument(); + expect(screen.queryByLabelText("切换助手工作栏")).toBeInTheDocument(); await waitFor(() => - expect(api.usage.getTokenSummary).toHaveBeenCalledOnce() - ) - expect(within(stats).getByText('120')).toBeInTheDocument() + expect(api.usage.getTokenSummary).toHaveBeenCalledOnce(), + ); + expect(within(stats).getByText("120")).toBeInTheDocument(); vi.mocked(api.usage.getTokenSummary).mockResolvedValueOnce({ totals: { @@ -6073,7 +5715,7 @@ describe('App', () => { output: 45, cacheRead: 10, cacheWrite: 5, - totalTokens: 360 + totalTokens: 360, }, records: [ { @@ -6081,1794 +5723,1684 @@ describe('App', () => { projectId, projectName: project.name, conversationId: request.conversationId, - conversationTitle: '用量会话', - runtime: 'model', - provider: 'anthropic', - model: 'sonnet-5', + conversationTitle: "用量会话", + runtime: "model", + provider: "anthropic", + model: "sonnet-5", callCount: 2, input: 300, output: 45, cacheRead: 10, cacheWrite: 5, - totalTokens: 360 - } - ] - }) + totalTokens: 360, + }, + ], + }); act(() => { agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); await waitFor(() => - expect(api.usage.getTokenSummary).toHaveBeenCalledTimes(2) - ) - expect(within(stats).getByText('345')).toBeInTheDocument() - }) + expect(api.usage.getTokenSummary).toHaveBeenCalledTimes(2), + ); + expect(within(stats).getByText("345")).toBeInTheDocument(); + }); - it('offers only Ask and Execute in visible work mode controls', async () => { - render() + it("offers only Ask and Execute in visible work mode controls", async () => { + render(); - await screen.findByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - const modeMenu = openComposerMenu('工作模式') + await screen.findByRole("button", { + name: "工作模式：Ask · 只读问答", + }); + const modeMenu = openComposerMenu("工作模式"); expect( within(modeMenu) - .getAllByRole('menuitemradio') - .map((option) => option.querySelector('span')?.textContent) - ).toEqual(['Ask · 只读问答', 'Execute · 完全权限']) + .getAllByRole("menuitemradio") + .map((option) => option.querySelector("span")?.textContent), + ).toEqual(["Ask · 只读问答", "Execute · 完全权限"]); - fireEvent.click(screen.getByLabelText('新建项目')) - const dialog = screen.getByRole('dialog', { name: '新建项目' }) + fireEvent.click(screen.getByLabelText("新建项目")); + const dialog = screen.getByRole("dialog", { name: "新建项目" }); const defaultMode = within(dialog) - .getAllByRole('group', { name: '默认模式' }) - .find((candidate) => - candidate.classList.contains('segmented-control') - )! + .getAllByRole("group", { name: "默认模式" }) + .find((candidate) => candidate.classList.contains("segmented-control"))!; expect( within(defaultMode) - .getAllByRole('button') - .map((button) => button.textContent) - ).toEqual(['Ask · 只读问答', 'Execute · 完全权限']) - expect(screen.queryByRole('option', { name: /Plan/u })).toBeNull() - }) + .getAllByRole("button") + .map((button) => button.textContent), + ).toEqual(["Ask · 只读问答", "Execute · 完全权限"]); + expect(screen.queryByRole("option", { name: /Plan/u })).toBeNull(); + }); - it('matches expert and work mode keyboard menus to the model picker', async () => { - render() + it("matches expert and work mode keyboard menus to the model picker", async () => { + render(); - const expertTrigger = await screen.findByRole('button', { - name: '专家角色：通用助手' - }) - expect(expertTrigger).toHaveClass('model-button') - fireEvent.keyDown(expertTrigger, { key: 'ArrowDown' }) + const expertTrigger = await screen.findByRole("button", { + name: "专家角色：通用助手", + }); + expect(expertTrigger).toHaveClass("model-button"); + fireEvent.keyDown(expertTrigger, { key: "ArrowDown" }); - const expertMenu = screen.getByRole('menu', { - name: '专家角色' - }) - expect(expertMenu).toHaveClass('runtime-picker__menu') - const generalExpert = within(expertMenu).getByRole( - 'menuitemradio', - { name: /^通用助手/u } - ) - const expertTeam = within(expertMenu).getByRole( - 'menuitemradio', - { name: /^专家团队（并行）/u } - ) - await waitFor(() => expect(generalExpert).toHaveFocus()) - fireEvent.keyDown(generalExpert, { key: 'ArrowDown' }) - expect(expertTeam).toHaveFocus() - fireEvent.keyDown(expertTeam, { key: 'Escape' }) - expect(expertTrigger).toHaveFocus() + const expertMenu = screen.getByRole("menu", { + name: "专家角色", + }); + expect(expertMenu).toHaveClass("runtime-picker__menu"); + const generalExpert = within(expertMenu).getByRole("menuitemradio", { + name: /^通用助手/u, + }); + const expertTeam = within(expertMenu).getByRole("menuitemradio", { + name: /^专家团队（并行）/u, + }); + await waitFor(() => expect(generalExpert).toHaveFocus()); + fireEvent.keyDown(generalExpert, { key: "ArrowDown" }); + expect(expertTeam).toHaveFocus(); + fireEvent.keyDown(expertTeam, { key: "Escape" }); + expect(expertTrigger).toHaveFocus(); expect( - screen.queryByRole('menu', { name: '专家角色' }) - ).not.toBeInTheDocument() + screen.queryByRole("menu", { name: "专家角色" }), + ).not.toBeInTheDocument(); - const modeTrigger = composerMenuTrigger('工作模式') - fireEvent.click(modeTrigger) - const modeMenu = screen.getByRole('menu', { name: '工作模式' }) - expect(modeMenu).toHaveClass('runtime-picker__menu') - fireEvent.pointerDown(screen.getByLabelText('向 GoodBuddy 提问')) + const modeTrigger = composerMenuTrigger("工作模式"); + fireEvent.click(modeTrigger); + const modeMenu = screen.getByRole("menu", { name: "工作模式" }); + expect(modeMenu).toHaveClass("runtime-picker__menu"); + fireEvent.pointerDown(screen.getByLabelText("向 GoodBuddy 提问")); expect( - screen.queryByRole('menu', { name: '工作模式' }) - ).not.toBeInTheDocument() - }) + screen.queryByRole("menu", { name: "工作模式" }), + ).not.toBeInTheDocument(); + }); - it('groups composer tools and exposes clear control descriptions', async () => { - render() + it("groups composer tools and exposes clear control descriptions", async () => { + render(); - const composer = (await screen.findByLabelText( - '向 GoodBuddy 提问' - )).closest('.composer') - expect(composer).not.toBeNull() + const composer = ( + await screen.findByLabelText("向 GoodBuddy 提问") + ).closest(".composer"); + expect(composer).not.toBeNull(); - const contentTools = within(composer!).getByRole('group', { - name: '添加内容' - }) + const contentTools = within(composer!).getByRole("group", { + name: "添加内容", + }); expect( - within(contentTools).getByRole('button', { name: '添加附件' }) - ).toHaveAttribute('title', '添加附件') + within(contentTools).getByRole("button", { name: "添加附件" }), + ).toHaveAttribute("title", "添加附件"); expect( - within(contentTools).getByRole('button', { name: '语音输入' }) - ).toHaveAttribute('title', '语音转文字，转写后可编辑再发送') + within(contentTools).getByRole("button", { name: "语音输入" }), + ).toHaveAttribute("title", "语音转文字，转写后可编辑再发送"); - const conversationSettings = within(composer!).getByRole( - 'group', - { name: '对话设置' } - ) + const conversationSettings = within(composer!).getByRole("group", { + name: "对话设置", + }); expect( - within(conversationSettings).getByRole('button', { - name: '专家角色：通用助手' - }) - ).toBeInTheDocument() + within(conversationSettings).getByRole("button", { + name: "专家角色：通用助手", + }), + ).toBeInTheDocument(); expect( - within(conversationSettings).getByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - ).toBeInTheDocument() + within(conversationSettings).getByRole("button", { + name: "工作模式：Ask · 只读问答", + }), + ).toBeInTheDocument(); expect( - within(conversationSettings).getByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - ).toHaveTextContent(/^Ask$/u) - expect(screen.getByLabelText('向 GoodBuddy 提问')).toHaveAttribute( - 'placeholder', - '给 GoodBuddy 发消息…\nEnter 发送，Shift+Enter 换行，Ctrl+V 粘贴图片或文本' - ) + within(conversationSettings).getByRole("button", { + name: "工作模式：Ask · 只读问答", + }), + ).toHaveTextContent(/^Ask$/u); + expect(screen.getByLabelText("向 GoodBuddy 提问")).toHaveAttribute( + "placeholder", + "给 GoodBuddy 发消息…\nEnter 发送，Shift+Enter 换行，Ctrl+V 粘贴图片或文本", + ); expect( - within(conversationSettings).getByRole('button', { - name: /默认模型/u - }) - ).toHaveAttribute( - 'title', - expect.stringContaining('Runtime 和模型') - ) - }) + within(conversationSettings).getByRole("button", { + name: /默认模型/u, + }), + ).toHaveAttribute("title", expect.stringContaining("Runtime 和模型")); + }); - it('starts in the first local project and persists manual selection', async () => { + it("starts in the first local project and persists manual selection", async () => { const secondProject = { ...project, - id: '00000000-0000-4000-8000-000000000102', - name: '第二项目', - rootPath: 'C:\\Second', + id: "00000000-0000-4000-8000-000000000102", + name: "第二项目", + rootPath: "C:\\Second", executionSpace: { - kind: 'local' as const, - rootPath: 'C:\\Second' + kind: "local" as const, + rootPath: "C:\\Second", }, - defaultWorkMode: 'execute' as const - } + defaultWorkMode: "execute" as const, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - secondProject - ]) - localStorage.setItem( - 'goodbuddy.active-project.v1', - secondProject.id - ) + secondProject, + ]); + localStorage.setItem("goodbuddy.active-project.v1", secondProject.id); - render() + render(); expect( - await screen.findByRole('button', { name: '当前项目' }) - ).toHaveTextContent(project.name) + await screen.findByRole("button", { name: "当前项目" }), + ).toHaveTextContent(project.name); expect( - screen.getByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - ).toBeEnabled() + screen.getByRole("button", { + name: "工作模式：Ask · 只读问答", + }), + ).toBeEnabled(); - selectProjectOption(secondProject.name) + selectProjectOption(secondProject.name); await waitFor(() => - expect( - localStorage.getItem('goodbuddy.active-project.v1') - ).toBe(secondProject.id) - ) - }) + expect(localStorage.getItem("goodbuddy.active-project.v1")).toBe( + secondProject.id, + ), + ); + }); - it('persists a new local conversation before queueing its first message', async () => { + it("persists a new local conversation before queueing its first message", async () => { const secondProject = { ...project, - id: '00000000-0000-4000-8000-000000000105', - name: '新本地项目', - rootPath: 'C:\\NewLocal', + id: "00000000-0000-4000-8000-000000000105", + name: "新本地项目", + rootPath: "C:\\NewLocal", executionSpace: { - kind: 'local' as const, - rootPath: 'C:\\NewLocal' - } - } - const persistence = deferred() + kind: "local" as const, + rootPath: "C:\\NewLocal", + }, + }; + const persistence = deferred(); vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - secondProject - ]) - vi.mocked(api.conversations.saveLocal).mockImplementation( - async (batch) => { - if ( - batch.some( - (conversation) => - conversation.header.projectId === secondProject.id - ) - ) { - await persistence.promise - } + secondProject, + ]); + vi.mocked(api.conversations.saveLocal).mockImplementation(async (batch) => { + if ( + batch.some( + (conversation) => conversation.header.projectId === secondProject.id, + ) + ) { + await persistence.promise; } - ) - render() + }); + render(); expect( - await screen.findByRole('button', { name: '当前项目' }) - ).toHaveTextContent(project.name) + await screen.findByRole("button", { name: "当前项目" }), + ).toHaveTextContent(project.name); - selectProjectOption(secondProject.name) + selectProjectOption(secondProject.name); await waitFor(() => expect( - screen.getByRole('button', { name: '当前项目' }) - ).toHaveTextContent(secondProject.name) - ) - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '首条消息必须先保存会话' } - }) - await waitFor(() => - expect(screen.getByLabelText('发送')).toBeEnabled() - ) - fireEvent.click(screen.getByLabelText('发送')) + screen.getByRole("button", { name: "当前项目" }), + ).toHaveTextContent(secondProject.name), + ); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "首条消息必须先保存会话" }, + }); + await waitFor(() => expect(screen.getByLabelText("发送")).toBeEnabled()); + fireEvent.click(screen.getByLabelText("发送")); await waitFor(() => expect(api.conversations.saveLocal).toHaveBeenCalledWith([ expect.objectContaining({ header: expect.objectContaining({ - projectId: secondProject.id - }) - }) - ]) - ) - expect(api.conversationQueue.enqueueUser).not.toHaveBeenCalled() + projectId: secondProject.id, + }), + }), + ]), + ); + expect(api.conversationQueue.enqueueUser).not.toHaveBeenCalled(); - persistence.resolve(undefined) + persistence.resolve(undefined); await waitFor(() => - expect(api.conversationQueue.enqueueUser).toHaveBeenCalledOnce() - ) + expect(api.conversationQueue.enqueueUser).toHaveBeenCalledOnce(), + ); const projectSaveIndex = vi .mocked(api.conversations.saveLocal) .mock.calls.findIndex(([batch]) => batch.some( - (conversation) => - conversation.header.projectId === secondProject.id - ) - ) - expect(projectSaveIndex).toBeGreaterThanOrEqual(0) + (conversation) => conversation.header.projectId === secondProject.id, + ), + ); + expect(projectSaveIndex).toBeGreaterThanOrEqual(0); expect( vi.mocked(api.conversations.saveLocal).mock.invocationCallOrder[ projectSaveIndex - ] + ], ).toBeLessThan( - vi.mocked( - api.conversationQueue.enqueueUser - ).mock.invocationCallOrder[0]! - ) - }) + vi.mocked(api.conversationQueue.enqueueUser).mock.invocationCallOrder[0]!, + ); + }); - it('starts in the first local project instead of restoring an SSH project', async () => { + it("starts in the first local project instead of restoring an SSH project", async () => { const remoteProject = { ...project, - id: '00000000-0000-4000-8000-000000000103', - name: '远程项目', - rootPath: '/srv/project', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000103", + name: "远程项目", + rootPath: "/srv/project", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000203', - remoteRootPath: '/srv/project' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000203", + remoteRootPath: "/srv/project", }, - builtInDefault: false - } + builtInDefault: false, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ remoteProject, - project - ]) - localStorage.setItem( - 'goodbuddy.active-project.v1', - remoteProject.id - ) + project, + ]); + localStorage.setItem("goodbuddy.active-project.v1", remoteProject.id); - render() + render(); expect( - await screen.findByRole('button', { name: '当前项目' }) - ).toHaveTextContent(project.name) + await screen.findByRole("button", { name: "当前项目" }), + ).toHaveTextContent(project.name); await waitFor(() => - expect( - localStorage.getItem('goodbuddy.active-project.v1') - ).toBe(project.id) - ) - }) + expect(localStorage.getItem("goodbuddy.active-project.v1")).toBe( + project.id, + ), + ); + }); - it('switches immediately to a managed SSH project and shows only supported OpenCode choices', async () => { - installRemoteProjectsSetting(true) + it("switches immediately to a managed SSH project and shows only supported OpenCode choices", async () => { + installRemoteProjectsSetting(true); const remoteProject = { ...project, - id: '00000000-0000-4000-8000-000000000119', - name: '仅 OpenCode 远程项目', - rootPath: '/srv/project', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000119", + name: "仅 OpenCode 远程项目", + rootPath: "/srv/project", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000219', - remoteRootPath: '/srv/project' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000219", + remoteRootPath: "/srv/project", }, - builtInDefault: false - } + builtInDefault: false, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - remoteProject - ]) + remoteProject, + ]); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000319', + id: "00000000-0000-4000-8000-000000000319", projectId: remoteProject.id, runtimeSelection: { - provider: 'model', - profileId: modelProfileId + provider: "model", + profileId: modelProfileId, }, - title: '旧直连模型会话', + title: "旧直连模型会话", updatedAt: 1_775_000_000_000, - messages: [] + messages: [], }, { - id: '00000000-0000-4000-8000-000000000419', + id: "00000000-0000-4000-8000-000000000419", projectId: remoteProject.id, runtimeSelection: { - provider: 'continue', - profileId: modelProfileId + provider: "continue", + profileId: modelProfileId, }, - title: '旧 Continue 会话', + title: "旧 Continue 会话", updatedAt: 1_774_000_000_000, - messages: [] - } - ]) + messages: [], + }, + ]); - render() - await screen.findByRole('button', { name: '当前项目' }) - selectProjectOption(remoteProject.name) + render(); + await screen.findByRole("button", { name: "当前项目" }); + selectProjectOption(remoteProject.name); await waitFor(() => expect( - screen.getByRole('button', { name: '当前项目' }) - ).toHaveTextContent(remoteProject.name) - ) - expect( - screen.queryByRole('progressbar') - ).not.toBeInTheDocument() + screen.getByRole("button", { name: "当前项目" }), + ).toHaveTextContent(remoteProject.name), + ); + expect(screen.queryByRole("progressbar")).not.toBeInTheDocument(); - const runtimeButton = await screen.findByRole('button', { - name: /OpenCode · 默认模型/u - }) - fireEvent.click(runtimeButton) - const runtimeMenu = screen.getByRole('menu', { - name: 'Runtime 和模型' - }) + const runtimeButton = await screen.findByRole("button", { + name: /OpenCode · 默认模型/u, + }); + fireEvent.click(runtimeButton); + const runtimeMenu = screen.getByRole("menu", { + name: "Runtime 和模型", + }); expect( - within(runtimeMenu).getByRole('menuitemradio', { - name: /^OpenCode · 默认模型.*sonnet-5$/u - }) - ).toBeInTheDocument() + within(runtimeMenu).getByRole("menuitemradio", { + name: /^OpenCode · 默认模型.*sonnet-5$/u, + }), + ).toBeInTheDocument(); + expect(within(runtimeMenu).getAllByRole("menuitemradio")).toHaveLength(1); + expect(within(runtimeMenu).queryByText("直连模型")).not.toBeInTheDocument(); expect( - within(runtimeMenu).getAllByRole('menuitemradio') - ).toHaveLength(1) + within(runtimeMenu).queryByText("Continue Runtime"), + ).not.toBeInTheDocument(); expect( - within(runtimeMenu).queryByText('直连模型') - ).not.toBeInTheDocument() + within(runtimeMenu).queryByText(/DeepSeek Harness/u), + ).not.toBeInTheDocument(); expect( - within(runtimeMenu).queryByText('Continue Runtime') - ).not.toBeInTheDocument() - expect( - within(runtimeMenu).queryByText(/DeepSeek Harness/u) - ).not.toBeInTheDocument() - expect( - within(runtimeMenu).getByRole('menuitem', { - name: '管理 Runtime 和模型连接' - }) - ).toBeInTheDocument() + within(runtimeMenu).getByRole("menuitem", { + name: "管理 Runtime 和模型连接", + }), + ).toBeInTheDocument(); - fireEvent.keyDown(document, { key: 'Escape' }) - fireEvent.click( - screen.getByText('旧 Continue 会话').closest('button')! - ) + fireEvent.keyDown(document, { key: "Escape" }); + fireEvent.click(screen.getByText("旧 Continue 会话").closest("button")!); expect( - await screen.findByRole('button', { - name: /OpenCode · 默认模型/u - }) - ).toBeInTheDocument() - }) + await screen.findByRole("button", { + name: /OpenCode · 默认模型/u, + }), + ).toBeInTheDocument(); + }); - it('keeps managed SSH histories visible while independently blocking recovery projects', async () => { - installRemoteProjectsSetting(true) + it("keeps managed SSH histories visible while independently blocking recovery projects", async () => { + installRemoteProjectsSetting(true); const first = { ...project, - id: '00000000-0000-4000-8000-000000000131', - name: '恢复项目一', - rootPath: '/srv/one', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000131", + name: "恢复项目一", + rootPath: "/srv/one", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000231', - remoteRootPath: '/srv/one' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000231", + remoteRootPath: "/srv/one", }, - builtInDefault: false - } + builtInDefault: false, + }; const second = { ...first, - id: '00000000-0000-4000-8000-000000000132', - name: '恢复项目二', - rootPath: '/srv/two', + id: "00000000-0000-4000-8000-000000000132", + name: "恢复项目二", + rootPath: "/srv/two", executionSpace: { ...first.executionSpace, - remoteRootPath: '/srv/two' - } - } - const firstConversationId = - '00000000-0000-4000-8000-000000000331' - const secondConversationId = - '00000000-0000-4000-8000-000000000332' - const firstRequestId = - '00000000-0000-4000-8000-000000000431' + remoteRootPath: "/srv/two", + }, + }; + const firstConversationId = "00000000-0000-4000-8000-000000000331"; + const secondConversationId = "00000000-0000-4000-8000-000000000332"; + const firstRequestId = "00000000-0000-4000-8000-000000000431"; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, first, - second - ]) + second, + ]); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: firstConversationId, projectId: first.id, - runtimeSelection: { provider: 'opencode' }, - title: '恢复前的会话', + runtimeSelection: { provider: "opencode" }, + title: "恢复前的会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000531', - role: 'assistant', - content: '恢复期间仍可查看的历史', + id: "00000000-0000-4000-8000-000000000531", + role: "assistant", + content: "恢复期间仍可查看的历史", createdAt: 1_775_000_000_000, - state: 'complete' - } - ] + state: "complete", + }, + ], }, { id: secondConversationId, projectId: second.id, - runtimeSelection: { provider: 'opencode' }, - title: '第二个恢复会话', + runtimeSelection: { provider: "opencode" }, + title: "第二个恢复会话", updatedAt: 1_774_000_000_000, - messages: [] - } - ]) - vi.mocked( - api.projects.remote.getRecoverySnapshot - ).mockResolvedValueOnce({ + messages: [], + }, + ]); + vi.mocked(api.projects.remote.getRecoverySnapshot).mockResolvedValueOnce({ recoveries: [ { projectId: first.id, requestId: firstRequestId, - stage: 'cursor', - current: '125' + stage: "cursor", + current: "125", }, { projectId: second.id, - requestId: - '00000000-0000-4000-8000-000000000432', - stage: 'agent' - } - ] - }) + requestId: "00000000-0000-4000-8000-000000000432", + stage: "agent", + }, + ], + }); - render() - await screen.findByRole('button', { name: '当前项目' }) + render(); + await screen.findByRole("button", { name: "当前项目" }); await waitFor(() => - expect( - api.projects.remote.getRecoverySnapshot - ).toHaveBeenCalledOnce() - ) - vi.mocked(api.conversationQueue.ready).mockClear() + expect(api.projects.remote.getRecoverySnapshot).toHaveBeenCalledOnce(), + ); + vi.mocked(api.conversationQueue.ready).mockClear(); - selectProjectOption(first.name) + selectProjectOption(first.name); + expect(screen.getByRole("button", { name: "当前项目" })).toHaveTextContent( + first.name, + ); expect( - screen.getByRole('button', { name: '当前项目' }) - ).toHaveTextContent(first.name) + await screen.findByText("恢复期间仍可查看的历史"), + ).toBeInTheDocument(); expect( - await screen.findByText('恢复期间仍可查看的历史') - ).toBeInTheDocument() - expect( - screen.getByText('正在恢复对话，已同步到事件 125') - ).toBeInTheDocument() + screen.getByText("正在恢复对话，已同步到事件 125"), + ).toBeInTheDocument(); expect(api.conversationQueue.ready).not.toHaveBeenCalledWith( - firstConversationId - ) + firstConversationId, + ); - const firstComposer = screen.getByLabelText('向 GoodBuddy 提问') + const firstComposer = screen.getByLabelText("向 GoodBuddy 提问"); fireEvent.change(firstComposer, { - target: { value: '恢复完成后发送' } - }) - fireEvent.keyDown(firstComposer, { key: 'Enter' }) - expect(firstComposer).toHaveValue('恢复完成后发送') - expect(api.conversationQueue.enqueueUser).not.toHaveBeenCalled() - expect(screen.getByLabelText('发送')).toBeDisabled() + target: { value: "恢复完成后发送" }, + }); + fireEvent.keyDown(firstComposer, { key: "Enter" }); + expect(firstComposer).toHaveValue("恢复完成后发送"); + expect(api.conversationQueue.enqueueUser).not.toHaveBeenCalled(); + expect(screen.getByLabelText("发送")).toBeDisabled(); - fireEvent.click(screen.getByRole('button', { name: '当前项目' })) - const menu = screen.getByRole('menu', { name: '当前项目' }) - expect( - within(menu).getByText('正在恢复远端 Agent…') - ).toBeInTheDocument() + fireEvent.click(screen.getByRole("button", { name: "当前项目" })); + const menu = screen.getByRole("menu", { name: "当前项目" }); + expect(within(menu).getByText("正在恢复远端 Agent…")).toBeInTheDocument(); fireEvent.click( - within(menu).getByText(project.name, { selector: 'b' }) - .closest('[role="menuitemradio"]')! - ) - const localComposer = screen.getByLabelText('向 GoodBuddy 提问') + within(menu) + .getByText(project.name, { selector: "b" }) + .closest('[role="menuitemradio"]')!, + ); + const localComposer = screen.getByLabelText("向 GoodBuddy 提问"); fireEvent.change(localComposer, { - target: { value: '本地项目不受影响' } - }) + target: { value: "本地项目不受影响" }, + }); + await waitFor(() => expect(screen.getByLabelText("发送")).toBeEnabled()); + fireEvent.keyDown(localComposer, { key: "Enter" }); await waitFor(() => - expect(screen.getByLabelText('发送')).toBeEnabled() - ) - fireEvent.keyDown(localComposer, { key: 'Enter' }) - await waitFor(() => - expect(api.conversationQueue.enqueueUser).toHaveBeenCalledOnce() - ) + expect(api.conversationQueue.enqueueUser).toHaveBeenCalledOnce(), + ); - selectProjectOption(first.name) + selectProjectOption(first.name); act(() => { remoteProjectRecoveryProgressListener?.({ projectId: first.id, - requestId: - '00000000-0000-4000-8000-000000000499', - stage: 'completed' - }) - }) - expect(screen.getByLabelText('发送')).toBeDisabled() - expect(firstComposer).toHaveValue('恢复完成后发送') + requestId: "00000000-0000-4000-8000-000000000499", + stage: "completed", + }); + }); + expect(screen.getByLabelText("发送")).toBeDisabled(); + expect(firstComposer).toHaveValue("恢复完成后发送"); act(() => { remoteProjectRecoveryProgressListener?.({ projectId: first.id, requestId: firstRequestId, - stage: 'completed' - }) - }) + stage: "completed", + }); + }); await waitFor(() => - expect(screen.getByText('恢复完成')).toBeInTheDocument() - ) - expect(screen.getByLabelText('发送')).toBeEnabled() + expect(screen.getByText("恢复完成")).toBeInTheDocument(), + ); + expect(screen.getByLabelText("发送")).toBeEnabled(); await waitFor(() => expect(api.conversationQueue.ready).toHaveBeenCalledWith( - firstConversationId - ) - ) + firstConversationId, + ), + ); expect(api.conversationQueue.ready).not.toHaveBeenCalledWith( - secondConversationId - ) - }) + secondConversationId, + ); + }); - describe('persisted terminal state convergence', () => { - let conversationsChangedListener: (() => void) | undefined + describe("persisted terminal state convergence", () => { + let conversationsChangedListener: (() => void) | undefined; const startStreamingRun = async (): Promise<{ - requestId: string - conversationId: string - assistantMessageId: string - userMessageId: string + requestId: string; + conversationId: string; + assistantMessageId: string; + userMessageId: string; }> => { - conversationsChangedListener = undefined - vi.mocked(api.conversations.onChanged).mockImplementation( - (listener) => { - conversationsChangedListener = listener - return () => { - if (conversationsChangedListener === listener) { - conversationsChangedListener = undefined - } + conversationsChangedListener = undefined; + vi.mocked(api.conversations.onChanged).mockImplementation((listener) => { + conversationsChangedListener = listener; + return () => { + if (conversationsChangedListener === listener) { + conversationsChangedListener = undefined; } - } - ) - render() - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '断网期间继续执行' } - }) - await waitFor(() => - expect(screen.getByLabelText('发送')).toBeEnabled() - ) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + }; + }); + render(); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "断网期间继续执行" }, + }); + await waitFor(() => expect(screen.getByLabelText("发送")).toBeEnabled()); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if ( !request?.currentAssistantMessageId || !request.currentUserMessageId ) { - throw new Error('Missing request') + throw new Error("Missing request"); } expect( screen - .getByText('正在连接 Agent Runtime') - .querySelector('.message__status-dot') - ).toHaveClass('message__status-dot--active') - expect(screen.getByLabelText('加入待发送队列')).toBeInTheDocument() - vi.mocked(api.conversationQueue.ready).mockClear() + .getByText("正在连接 Agent Runtime") + .querySelector(".message__status-dot"), + ).toHaveClass("message__status-dot--active"); + expect(screen.getByLabelText("加入待发送队列")).toBeInTheDocument(); + vi.mocked(api.conversationQueue.ready).mockClear(); return { requestId: request.requestId, conversationId: request.conversationId, assistantMessageId: request.currentAssistantMessageId, - userMessageId: request.currentUserMessageId - } - } + userMessageId: request.currentUserMessageId, + }; + }; const persistedSnapshot = ( input: { - conversationId: string - userMessageId: string - assistantMessageId: string + conversationId: string; + userMessageId: string; + assistantMessageId: string; }, assistant: { - state: 'streaming' | 'complete' | 'error' - content: string - status?: string - } + state: "streaming" | "complete" | "error"; + content: string; + status?: string; + }, ): ConversationSnapshot => ({ id: input.conversationId, projectId, - runtimeSelection: { provider: 'model', profileId: modelProfileId }, - title: '断网期间继续执行', + runtimeSelection: { provider: "model", profileId: modelProfileId }, + title: "断网期间继续执行", updatedAt: 1_000, messages: [ { id: input.userMessageId, - role: 'user', - content: '断网期间继续执行', + role: "user", + content: "断网期间继续执行", createdAt: 900, - state: 'complete' + state: "complete", }, { id: input.assistantMessageId, - role: 'assistant', + role: "assistant", content: assistant.content, createdAt: 950, state: assistant.state, - status: assistant.status - } - ] - }) + status: assistant.status, + }, + ], + }); - it('converges a streaming message onto the persisted completed state after reconnecting', async () => { - const started = await startStreamingRun() - expect(conversationsChangedListener).toBeDefined() + it("converges a streaming message onto the persisted completed state after reconnecting", async () => { + const started = await startStreamingRun(); + expect(conversationsChangedListener).toBeDefined(); vi.mocked(api.conversations.list).mockResolvedValue([ persistedSnapshot(started, { - state: 'complete', - content: '远端在断线期间已经完成的回复', - status: '任务已完成' - }) - ]) + state: "complete", + content: "远端在断线期间已经完成的回复", + status: "任务已完成", + }), + ]); act(() => { - conversationsChangedListener?.() - }) + conversationsChangedListener?.(); + }); expect( - await screen.findByText('远端在断线期间已经完成的回复') - ).toBeInTheDocument() + await screen.findByText("远端在断线期间已经完成的回复"), + ).toBeInTheDocument(); await waitFor(() => expect( - screen.queryByText('正在连接 Agent Runtime') - ).not.toBeInTheDocument() - ) - expect( - document.querySelector('.message__status-dot--active') - ).toBeNull() - expect(screen.getByLabelText('发送')).toBeInTheDocument() + screen.queryByText("正在连接 Agent Runtime"), + ).not.toBeInTheDocument(), + ); + expect(document.querySelector(".message__status-dot--active")).toBeNull(); + expect(screen.getByLabelText("发送")).toBeInTheDocument(); await waitFor(() => expect(api.conversationQueue.ready).toHaveBeenCalledWith( - started.conversationId - ) - ) + started.conversationId, + ), + ); // A late terminal event for the settled run must be ignored rather // than resurrecting the message or double-releasing the queue. - vi.mocked(api.conversationQueue.ready).mockClear() + vi.mocked(api.conversationQueue.ready).mockClear(); act(() => { agentListener?.({ requestId: started.requestId, - type: 'error', - status: 'failed', - message: '迟到的连接中断错误' - }) - }) + type: "error", + status: "failed", + message: "迟到的连接中断错误", + }); + }); + expect(screen.queryByText("迟到的连接中断错误")).not.toBeInTheDocument(); expect( - screen.queryByText('迟到的连接中断错误') - ).not.toBeInTheDocument() - expect( - screen.getByText('远端在断线期间已经完成的回复') - ).toBeInTheDocument() - expect(api.conversationQueue.ready).not.toHaveBeenCalled() - }) + screen.getByText("远端在断线期间已经完成的回复"), + ).toBeInTheDocument(); + expect(api.conversationQueue.ready).not.toHaveBeenCalled(); + }); - it('converges a streaming message onto the persisted failed state after reconnecting', async () => { - const started = await startStreamingRun() + it("converges a streaming message onto the persisted failed state after reconnecting", async () => { + const started = await startStreamingRun(); vi.mocked(api.conversations.list).mockResolvedValue([ persistedSnapshot(started, { - state: 'error', - content: '', - status: '远端 Agent 执行失败' - }) - ]) + state: "error", + content: "", + status: "远端 Agent 执行失败", + }), + ]); act(() => { - conversationsChangedListener?.() - }) + conversationsChangedListener?.(); + }); expect( - await screen.findByText('远端 Agent 执行失败') - ).toBeInTheDocument() + await screen.findByText("远端 Agent 执行失败"), + ).toBeInTheDocument(); expect( - screen.queryByText('正在连接 Agent Runtime') - ).not.toBeInTheDocument() - expect( - document.querySelector('.message__status-dot--active') - ).toBeNull() + screen.queryByText("正在连接 Agent Runtime"), + ).not.toBeInTheDocument(); + expect(document.querySelector(".message__status-dot--active")).toBeNull(); await waitFor(() => - expect(screen.getByLabelText('发送')).toBeInTheDocument() - ) + expect(screen.getByLabelText("发送")).toBeInTheDocument(), + ); await waitFor(() => expect(api.conversationQueue.ready).toHaveBeenCalledWith( - started.conversationId - ) - ) - }) + started.conversationId, + ), + ); + }); - it('keeps the run in flight while the persisted message is still streaming and refreshes on focus', async () => { - const started = await startStreamingRun() + it("keeps the run in flight while the persisted message is still streaming and refreshes on focus", async () => { + const started = await startStreamingRun(); vi.mocked(api.conversations.list).mockResolvedValue([ persistedSnapshot(started, { - state: 'streaming', - content: '', - status: '远端 Agent 正在处理请求' - }) - ]) + state: "streaming", + content: "", + status: "远端 Agent 正在处理请求", + }), + ]); act(() => { - conversationsChangedListener?.() - }) + conversationsChangedListener?.(); + }); await waitFor(() => - expect(api.conversations.list).toHaveBeenCalledTimes(2) - ) + expect(api.conversations.list).toHaveBeenCalledTimes(2), + ); // The local streaming copy still wins over a non-terminal snapshot. - expect( - screen.getByText('正在连接 Agent Runtime') - ).toBeInTheDocument() - expect(screen.getByLabelText('加入待发送队列')).toBeInTheDocument() - expect(api.conversationQueue.ready).not.toHaveBeenCalled() + expect(screen.getByText("正在连接 Agent Runtime")).toBeInTheDocument(); + expect(screen.getByLabelText("加入待发送队列")).toBeInTheDocument(); + expect(api.conversationQueue.ready).not.toHaveBeenCalled(); // Regaining focus re-reads the store and converges once the remote // terminal state has landed, even without another change event. vi.mocked(api.conversations.list).mockResolvedValue([ persistedSnapshot(started, { - state: 'complete', - content: '聚焦后读到的最终回复', - status: '任务已完成' - }) - ]) + state: "complete", + content: "聚焦后读到的最终回复", + status: "任务已完成", + }), + ]); act(() => { - window.dispatchEvent(new Event('focus')) - }) + window.dispatchEvent(new Event("focus")); + }); expect( - await screen.findByText('聚焦后读到的最终回复') - ).toBeInTheDocument() + await screen.findByText("聚焦后读到的最终回复"), + ).toBeInTheDocument(); expect( - screen.queryByText('正在连接 Agent Runtime') - ).not.toBeInTheDocument() + screen.queryByText("正在连接 Agent Runtime"), + ).not.toBeInTheDocument(); await waitFor(() => expect(api.conversationQueue.ready).toHaveBeenCalledWith( - started.conversationId - ) - ) + started.conversationId, + ), + ); // The live done event now targets an already-settled run: no-op. act(() => { - agentListener?.({ requestId: started.requestId, type: 'done' }) - }) - expect( - screen.getByText('聚焦后读到的最终回复') - ).toBeInTheDocument() - }) + agentListener?.({ requestId: started.requestId, type: "done" }); + }); + expect(screen.getByText("聚焦后读到的最终回复")).toBeInTheDocument(); + }); - it('keeps streaming through window switches while the store still reports the run', async () => { - const started = await startStreamingRun() + it("keeps streaming through window switches while the store still reports the run", async () => { + const started = await startStreamingRun(); act(() => { agentListener?.({ requestId: started.requestId, - type: 'text', - delta: '第一段回答' - }) - }) - expect(await screen.findByText('第一段回答')).toBeInTheDocument() + type: "text", + delta: "第一段回答", + }); + }); + expect(await screen.findByText("第一段回答")).toBeInTheDocument(); // The renderer keeps saving the streaming message, so a refresh reads // it back as streaming. That must not be reported as an interruption. vi.mocked(api.conversations.list).mockResolvedValue([ persistedSnapshot(started, { - state: 'streaming', - content: '第一段回答' - }) - ]) + state: "streaming", + content: "第一段回答", + }), + ]); act(() => { - window.dispatchEvent( - new Event('visibilitychange', { bubbles: true }) - ) - window.dispatchEvent(new Event('focus')) - }) + window.dispatchEvent(new Event("visibilitychange", { bubbles: true })); + window.dispatchEvent(new Event("focus")); + }); await waitFor(() => - expect(api.conversations.list).toHaveBeenCalledTimes(2) - ) + expect(api.conversations.list).toHaveBeenCalledTimes(2), + ); expect( - screen.queryByText('上次运行意外中断，可以重新发送问题') - ).not.toBeInTheDocument() + screen.queryByText("上次运行意外中断，可以重新发送问题"), + ).not.toBeInTheDocument(); expect( - screen.queryByRole('button', { name: '重新编辑并发送' }) - ).not.toBeInTheDocument() - expect(api.conversationQueue.ready).not.toHaveBeenCalled() + screen.queryByRole("button", { name: "重新编辑并发送" }), + ).not.toBeInTheDocument(); + expect(api.conversationQueue.ready).not.toHaveBeenCalled(); // The run is still live, so later deltas and the terminal event must // still reach the message. act(() => { agentListener?.({ requestId: started.requestId, - type: 'text', - delta: '，第二段回答' - }) + type: "text", + delta: "，第二段回答", + }); agentListener?.({ requestId: started.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); expect( - await screen.findByText('第一段回答，第二段回答') - ).toBeInTheDocument() + await screen.findByText("第一段回答，第二段回答"), + ).toBeInTheDocument(); await waitFor(() => expect(api.conversationQueue.ready).toHaveBeenCalledWith( - started.conversationId - ) - ) - expect(screen.getByLabelText('发送')).toBeInTheDocument() - }) - }) + started.conversationId, + ), + ); + expect(screen.getByLabelText("发送")).toBeInTheDocument(); + }); + }); - it('retries a failed recovery with a new request ID and cleans up its listener', async () => { - installRemoteProjectsSetting(true) + it("retries a failed recovery with a new request ID and cleans up its listener", async () => { + installRemoteProjectsSetting(true); const remoteProject = { ...project, - id: '00000000-0000-4000-8000-000000000133', - name: '失败的恢复项目', - rootPath: '/srv/failed', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000133", + name: "失败的恢复项目", + rootPath: "/srv/failed", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000233', - remoteRootPath: '/srv/failed' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000233", + remoteRootPath: "/srv/failed", }, - builtInDefault: false - } - const oldRequestId = - '00000000-0000-4000-8000-000000000433' - const newRequestId = - '00000000-0000-4000-8000-000000000434' + builtInDefault: false, + }; + const oldRequestId = "00000000-0000-4000-8000-000000000433"; + const newRequestId = "00000000-0000-4000-8000-000000000434"; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - remoteProject - ]) - vi.mocked( - api.projects.remote.getRecoverySnapshot - ).mockResolvedValueOnce({ + remoteProject, + ]); + vi.mocked(api.projects.remote.getRecoverySnapshot).mockResolvedValueOnce({ recoveries: [ { projectId: remoteProject.id, requestId: oldRequestId, - stage: 'failed', - message: '无法连接 Host', - retryable: true - } - ] - }) + stage: "failed", + message: "无法连接 Host", + retryable: true, + }, + ], + }); vi.mocked(api.projects.remote.retryRecovery).mockResolvedValueOnce({ projectId: remoteProject.id, requestId: newRequestId, - stage: 'network' - }) + stage: "network", + }); - const { unmount } = render() - await screen.findByRole('button', { name: '当前项目' }) - selectProjectOption(remoteProject.name) - expect( - await screen.findByRole('alert') - ).toHaveTextContent('恢复失败：无法连接 Host') + const { unmount } = render(); + await screen.findByRole("button", { name: "当前项目" }); + selectProjectOption(remoteProject.name); + expect(await screen.findByRole("alert")).toHaveTextContent( + "恢复失败：无法连接 Host", + ); fireEvent.click( - screen.getByRole('button', { - name: `重试恢复项目 ${remoteProject.name}` - }) - ) + screen.getByRole("button", { + name: `重试恢复项目 ${remoteProject.name}`, + }), + ); await waitFor(() => - expect( - api.projects.remote.retryRecovery - ).toHaveBeenCalledWith(remoteProject.id) - ) - expect( - await screen.findByText('正在恢复网络连接…') - ).toBeInTheDocument() + expect(api.projects.remote.retryRecovery).toHaveBeenCalledWith( + remoteProject.id, + ), + ); + expect(await screen.findByText("正在恢复网络连接…")).toBeInTheDocument(); act(() => { remoteProjectRecoveryProgressListener?.({ projectId: remoteProject.id, requestId: oldRequestId, - stage: 'completed' - }) - }) - expect(screen.getByText('正在恢复网络连接…')).toBeInTheDocument() - expect(screen.getByLabelText('发送')).toBeDisabled() + stage: "completed", + }); + }); + expect(screen.getByText("正在恢复网络连接…")).toBeInTheDocument(); + expect(screen.getByLabelText("发送")).toBeDisabled(); act(() => { remoteProjectRecoveryProgressListener?.({ projectId: remoteProject.id, requestId: newRequestId, - stage: 'completed' - }) - }) - expect( - await screen.findByText('恢复完成') - ).toBeInTheDocument() + stage: "completed", + }); + }); + expect(await screen.findByText("恢复完成")).toBeInTheDocument(); act(() => { remoteProjectRecoveryProgressListener?.({ projectId: remoteProject.id, requestId: newRequestId, - stage: 'network' - }) - }) - expect(screen.getByText('恢复完成')).toBeInTheDocument() + stage: "network", + }); + }); + expect(screen.getByText("恢复完成")).toBeInTheDocument(); - unmount() - expect(removeProjectRecoveryListener).toHaveBeenCalledOnce() - }) + unmount(); + expect(removeProjectRecoveryListener).toHaveBeenCalledOnce(); + }); - it('switches between projects on one Host without a connection phase', async () => { - installRemoteProjectsSetting(true) + it("switches between projects on one Host without a connection phase", async () => { + installRemoteProjectsSetting(true); const first = { ...project, - id: '00000000-0000-4000-8000-000000000126', - name: 'Host 项目一', - rootPath: '/srv/one', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000126", + name: "Host 项目一", + rootPath: "/srv/one", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000226', - remoteRootPath: '/srv/one' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000226", + remoteRootPath: "/srv/one", }, - builtInDefault: false - } + builtInDefault: false, + }; const second = { ...first, - id: '00000000-0000-4000-8000-000000000127', - name: 'Host 项目二', - rootPath: '/srv/two', + id: "00000000-0000-4000-8000-000000000127", + name: "Host 项目二", + rootPath: "/srv/two", executionSpace: { ...first.executionSpace, - remoteRootPath: '/srv/two' - } - } + remoteRootPath: "/srv/two", + }, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, first, - second - ]) + second, + ]); - render() - await screen.findByRole('button', { name: '当前项目' }) - vi.mocked( - api.runtimeCustomization.getNativeSnapshot - ).mockClear() + render(); + await screen.findByRole("button", { name: "当前项目" }); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockClear(); for (const remoteProject of [first, second, first]) { - selectProjectOption(remoteProject.name) + selectProjectOption(remoteProject.name); expect( - screen.getByRole('button', { name: '当前项目' }) - ).toHaveTextContent(remoteProject.name) - expect( - screen.queryByRole('progressbar') - ).not.toBeInTheDocument() + screen.getByRole("button", { name: "当前项目" }), + ).toHaveTextContent(remoteProject.name); + expect(screen.queryByRole("progressbar")).not.toBeInTheDocument(); } - expect( - api.runtimeCustomization.getNativeSnapshot - ).not.toHaveBeenCalled() - }) + expect(api.runtimeCustomization.getNativeSnapshot).not.toHaveBeenCalled(); + }); - it('keeps saved remote projects hidden while the preview is disabled', async () => { - installRemoteProjectsSetting(false) + it("keeps saved remote projects hidden while the preview is disabled", async () => { + installRemoteProjectsSetting(false); const remoteProject = { ...project, - id: '00000000-0000-4000-8000-000000000116', - name: '已保存远程项目', - rootPath: '/srv/project', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000116", + name: "已保存远程项目", + rootPath: "/srv/project", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000216', - remoteRootPath: '/srv/project' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000216", + remoteRootPath: "/srv/project", }, - builtInDefault: false - } + builtInDefault: false, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - remoteProject - ]) + remoteProject, + ]); - render() + render(); - const trigger = await screen.findByRole('button', { - name: '当前项目' - }) - fireEvent.click(trigger) - const menu = screen.getByRole('menu', { name: '当前项目' }) + const trigger = await screen.findByRole("button", { + name: "当前项目", + }); + fireEvent.click(trigger); + const menu = screen.getByRole("menu", { name: "当前项目" }); expect( - within(menu).queryByText(remoteProject.name) - ).not.toBeInTheDocument() - }) + within(menu).queryByText(remoteProject.name), + ).not.toBeInTheDocument(); + }); - it('falls back to a local project when the remote preview is disabled', async () => { - const { updateSettings } = installRemoteProjectsSetting(true) + it("falls back to a local project when the remote preview is disabled", async () => { + const { updateSettings } = installRemoteProjectsSetting(true); const remoteProject = { ...project, - id: '00000000-0000-4000-8000-000000000117', - name: '预览远程项目', - rootPath: '/srv/project', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000117", + name: "预览远程项目", + rootPath: "/srv/project", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000217', - remoteRootPath: '/srv/project' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000217", + remoteRootPath: "/srv/project", }, - builtInDefault: false - } + builtInDefault: false, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - remoteProject - ]) + remoteProject, + ]); - render() - await screen.findByRole('button', { name: '当前项目' }) - selectProjectOption(remoteProject.name) + render(); + await screen.findByRole("button", { name: "当前项目" }); + selectProjectOption(remoteProject.name); await waitFor(() => expect( - screen.getByRole('button', { name: '当前项目' }) - ).toHaveTextContent(remoteProject.name) - ) + screen.getByRole("button", { name: "当前项目" }), + ).toHaveTextContent(remoteProject.name), + ); - fireEvent.click(screen.getByText('本地工作区')) + fireEvent.click(screen.getByText("本地工作区")); + fireEvent.click(await screen.findByRole("tab", { name: "平台功能" })); fireEvent.click( - await screen.findByRole('tab', { name: '平台功能' }) - ) - fireEvent.click( - await screen.findByRole('tab', { - name: '远程项目（技术预览）' - }) - ) - const remoteProjectsSwitch = await screen.findByRole('switch', { - name: '远程项目（技术预览）' - }) - expect(remoteProjectsSwitch).toBeChecked() - fireEvent.click(remoteProjectsSwitch) + await screen.findByRole("tab", { + name: "远程项目（技术预览）", + }), + ); + const remoteProjectsSwitch = await screen.findByRole("switch", { + name: "远程项目（技术预览）", + }); + expect(remoteProjectsSwitch).toBeChecked(); + fireEvent.click(remoteProjectsSwitch); await waitFor(() => expect(updateSettings).toHaveBeenCalledWith({ - remoteProjectsEnabled: false - }) - ) + remoteProjectsEnabled: false, + }), + ); await waitFor(() => expect( - screen.getByRole('button', { name: '当前项目' }) - ).toHaveTextContent(project.name) - ) - fireEvent.click( - screen.getByRole('button', { name: '当前项目' }) - ) - expect( - screen.queryByText(remoteProject.name) - ).not.toBeInTheDocument() - expect(api.projects.setArchived).not.toHaveBeenCalled() - expect(api.projects.delete).not.toHaveBeenCalled() - }) + screen.getByRole("button", { name: "当前项目" }), + ).toHaveTextContent(project.name), + ); + fireEvent.click(screen.getByRole("button", { name: "当前项目" })); + expect(screen.queryByText(remoteProject.name)).not.toBeInTheDocument(); + expect(api.projects.setArchived).not.toHaveBeenCalled(); + expect(api.projects.delete).not.toHaveBeenCalled(); + }); - it('deletes an unreachable non-active remote project without connecting', async () => { - installRemoteProjectsSetting(true) + it("deletes an unreachable non-active remote project without connecting", async () => { + installRemoteProjectsSetting(true); const remoteProject = { ...project, - id: '00000000-0000-4000-8000-000000000125', - name: '目录已丢失的远程项目', - rootPath: '/srv/missing', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000125", + name: "目录已丢失的远程项目", + rootPath: "/srv/missing", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000225', - remoteRootPath: '/srv/missing' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000225", + remoteRootPath: "/srv/missing", }, - builtInDefault: false - } + builtInDefault: false, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - remoteProject - ]) - render() - await screen.findByRole('button', { name: '当前项目' }) + remoteProject, + ]); + render(); + await screen.findByRole("button", { name: "当前项目" }); - fireEvent.click(screen.getByLabelText('当前项目')) + fireEvent.click(screen.getByLabelText("当前项目")); fireEvent.click( - screen.getByRole('menuitem', { - name: `管理项目 ${remoteProject.name}` - }) - ) - const dialog = screen.getByRole('dialog', { name: '项目设置' }) - fireEvent.click( - within(dialog).getByRole('button', { name: '删除项目' }) - ) + screen.getByRole("menuitem", { + name: `管理项目 ${remoteProject.name}`, + }), + ); + const dialog = screen.getByRole("dialog", { name: "项目设置" }); + fireEvent.click(within(dialog).getByRole("button", { name: "删除项目" })); fireEvent.change( - within(dialog).getByLabelText( - `输入“${remoteProject.name}”确认删除` - ), - { target: { value: remoteProject.name } } - ) + within(dialog).getByLabelText(`输入“${remoteProject.name}”确认删除`), + { target: { value: remoteProject.name } }, + ); fireEvent.click( - within(dialog).getByRole('button', { - name: '永久删除项目' - }) - ) + within(dialog).getByRole("button", { + name: "永久删除项目", + }), + ); await waitFor(() => expect(api.projects.delete).toHaveBeenCalledWith( remoteProject.id, - remoteProject.name - ) - ) - expect( - screen.getByRole('button', { name: '当前项目' }) - ).toHaveTextContent(project.name) - }) + remoteProject.name, + ), + ); + expect(screen.getByRole("button", { name: "当前项目" })).toHaveTextContent( + project.name, + ); + }); - it('shows grouped project details and keeps the rich menu keyboard accessible', async () => { + it("shows grouped project details and keeps the rich menu keyboard accessible", async () => { const channelProject = { ...project, - id: '00000000-0000-4000-8000-000000000201', - name: '微信 ClawBot', - kind: 'channel' as const, - channel: 'weixin' as const - } + id: "00000000-0000-4000-8000-000000000201", + name: "微信 ClawBot", + kind: "channel" as const, + channel: "weixin" as const, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - channelProject - ]) - render() + channelProject, + ]); + render(); - const trigger = await screen.findByRole('button', { - name: '当前项目' - }) - fireEvent.keyDown(trigger, { key: 'ArrowDown' }) + const trigger = await screen.findByRole("button", { + name: "当前项目", + }); + fireEvent.keyDown(trigger, { key: "ArrowDown" }); - const menu = screen.getByRole('menu', { name: '当前项目' }) + const menu = screen.getByRole("menu", { name: "当前项目" }); expect( - within(menu).getByRole('group', { name: '本地项目' }) - ).toHaveTextContent('本地目录 · C:\\Users\\test') + within(menu).getByRole("group", { name: "本地项目" }), + ).toHaveTextContent("本地目录 · C:\\Users\\test"); expect( - within(menu).getByRole('group', { name: '远程通道' }) - ).toHaveTextContent( - '微信 ClawBot · 远程通道 · C:\\Users\\test' - ) + within(menu).getByRole("group", { name: "远程通道" }), + ).toHaveTextContent("微信 ClawBot · 远程通道 · C:\\Users\\test"); const selectedProject = within(menu) - .getByText(project.name, { selector: 'b' }) - .closest('[role="menuitemradio"]') - const remoteProjectSettings = within(menu).getByRole( - 'menuitem', - { name: `管理项目 ${channelProject.name}` } - ) - expect(selectedProject).toHaveAttribute('aria-checked', 'true') - await waitFor(() => expect(selectedProject).toHaveFocus()) + .getByText(project.name, { selector: "b" }) + .closest('[role="menuitemradio"]'); + const remoteProjectSettings = within(menu).getByRole("menuitem", { + name: `管理项目 ${channelProject.name}`, + }); + expect(selectedProject).toHaveAttribute("aria-checked", "true"); + await waitFor(() => expect(selectedProject).toHaveFocus()); - fireEvent.keyDown(selectedProject!, { key: 'End' }) - expect(remoteProjectSettings).toHaveFocus() - fireEvent.keyDown(remoteProjectSettings, { key: 'Escape' }) - expect(trigger).toHaveFocus() + fireEvent.keyDown(selectedProject!, { key: "End" }); + expect(remoteProjectSettings).toHaveFocus(); + fireEvent.keyDown(remoteProjectSettings, { key: "Escape" }); + expect(trigger).toHaveFocus(); expect( - screen.queryByRole('menu', { name: '当前项目' }) - ).not.toBeInTheDocument() - }) + screen.queryByRole("menu", { name: "当前项目" }), + ).not.toBeInTheDocument(); + }); - it('keeps channel projects empty until a client message creates a remote conversation', async () => { + it("keeps channel projects empty until a client message creates a remote conversation", async () => { const channelProject = { ...project, - id: '00000000-0000-4000-8000-000000000201', - name: '微信 ClawBot', - kind: 'channel' as const, - channel: 'weixin' as const, + id: "00000000-0000-4000-8000-000000000201", + name: "微信 ClawBot", + kind: "channel" as const, + channel: "weixin" as const, runtimeSelection: { - provider: 'model' as const, - profileId: modelProfileId - } - } + provider: "model" as const, + profileId: modelProfileId, + }, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - channelProject - ]) - render() + channelProject, + ]); + render(); - await screen.findByRole('button', { name: '当前项目' }) - selectProjectOption(channelProject.name) + await screen.findByRole("button", { name: "当前项目" }); + selectProjectOption(channelProject.name); expect( - screen.queryByRole('button', { name: /新建对话/u }) - ).not.toBeInTheDocument() - expect(screen.queryByText('Ctrl N')).not.toBeInTheDocument() - expect( - screen.getAllByText('尚无远程会话').length - ).toBeGreaterThan(0) + screen.queryByRole("button", { name: /新建对话/u }), + ).not.toBeInTheDocument(); + expect(screen.queryByText("Ctrl N")).not.toBeInTheDocument(); + expect(screen.getAllByText("尚无远程会话").length).toBeGreaterThan(0); expect( screen.getByText( - '请先连接微信 ClawBot，远程用户发送第一条消息后，会话会自动出现在这里。' - ) - ).toBeInTheDocument() + "请先连接微信 ClawBot，远程用户发送第一条消息后，会话会自动出现在这里。", + ), + ).toBeInTheDocument(); fireEvent.keyDown(document, { - key: 'n', - ctrlKey: true - }) + key: "n", + ctrlKey: true, + }); expect( - await screen.findByText( - '通道项目的会话由客户端收到新消息后自动创建' - ) - ).toBeInTheDocument() + await screen.findByText("通道项目的会话由客户端收到新消息后自动创建"), + ).toBeInTheDocument(); await act( () => new Promise((resolve) => { - setTimeout(resolve, 550) - }) - ) + setTimeout(resolve, 550); + }), + ); const savedChannelHeaders = vi .mocked(api.conversations.saveLocal) - .mock.calls.flatMap(([batch]) => batch.map((entry) => entry.header)) + .mock.calls.flatMap(([batch]) => batch.map((entry) => entry.header)); expect( savedChannelHeaders.some( - (header) => header.projectId === channelProject.id - ) - ).toBe(false) - expect(screen.getAllByText('尚无远程会话').length).toBeGreaterThan(0) + (header) => header.projectId === channelProject.id, + ), + ).toBe(false); + expect(screen.getAllByText("尚无远程会话").length).toBeGreaterThan(0); - fireEvent.click(screen.getByRole('button', { name: '打开设置' })) + fireEvent.click(screen.getByRole("button", { name: "打开设置" })); - const channelSettingsTab = await screen.findByRole('tab', { - name: '消息通道' - }) - expect(channelSettingsTab).toHaveAttribute('aria-selected', 'true') - }) + const channelSettingsTab = await screen.findByRole("tab", { + name: "消息通道", + }); + expect(channelSettingsTab).toHaveAttribute("aria-selected", "true"); + }); - it('keeps channel project settings synchronized between both entry points', async () => { + it("keeps channel project settings synchronized between both entry points", async () => { let channelProjects: AssistantProject[] = [ - ['weixin', '微信 ClawBot'], - ['wecom', '企业微信'], - ['dingtalk', '钉钉'] + ["weixin", "微信 ClawBot"], + ["wecom", "企业微信"], + ["dingtalk", "钉钉"], ].map(([channel, name], index) => ({ ...project, id: `00000000-0000-4000-8000-00000000020${index + 1}`, name: name!, description: `${name}通道项目`, - kind: 'channel' as const, - channel: channel as 'weixin' | 'wecom' | 'dingtalk', + kind: "channel" as const, + channel: channel as "weixin" | "wecom" | "dingtalk", runtimeSelection: { - provider: 'model' as const, - profileId: modelProfileId - } - })) + provider: "model" as const, + profileId: modelProfileId, + }, + })); vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - ...channelProjects - ]) + ...channelProjects, + ]); vi.mocked(api.projects.update).mockImplementation( async (projectToUpdate, input) => { const existing = channelProjects.find( - (candidate) => candidate.id === projectToUpdate - ) + (candidate) => candidate.id === projectToUpdate, + ); if (!existing) { return { ...project, ...input, - id: projectToUpdate - } + id: projectToUpdate, + }; } const updated = { ...existing, ...input, - updatedAt: '2026-08-14T00:00:00.000Z' - } + updatedAt: "2026-08-14T00:00:00.000Z", + }; channelProjects = channelProjects.map((candidate) => - candidate.id === updated.id ? updated : candidate - ) - return updated - } - ) + candidate.id === updated.id ? updated : candidate, + ); + return updated; + }, + ); api.channels = { getSnapshot: vi.fn(async () => ({ weixin: { enabled: false, bindingConfigured: false, - source: 'none' as const, - status: { state: 'disabled' as const } + source: "none" as const, + status: { state: "disabled" as const }, }, wecom: { enabled: false, - botId: '', + botId: "", secretConfigured: false, - source: 'none' as const, + source: "none" as const, readOnly: false, allowedSenderIds: [], allowGroupMessages: false, - status: { state: 'disabled' as const } + status: { state: "disabled" as const }, }, dingtalk: { enabled: false, - clientId: '', + clientId: "", secretConfigured: false, - source: 'none' as const, + source: "none" as const, readOnly: false, allowedSenderIds: [], allowGroupMessages: false, - status: { state: 'disabled' as const } - } + status: { state: "disabled" as const }, + }, })), apply: vi.fn(async () => { - throw new Error('No channel connection settings changed') + throw new Error("No channel connection settings changed"); }), testConnection: vi.fn(async (channel) => ({ channel, - ok: true + ok: true, })), getWeixinBinding: vi.fn(async () => ({ - status: 'stopped' as const + status: "stopped" as const, })), startWeixinBinding: vi.fn(async () => ({ - status: 'starting' as const + status: "starting" as const, })), submitWeixinVerification: vi.fn(async () => ({ - status: 'scanned' as const + status: "scanned" as const, })), disconnectWeixin: vi.fn(async () => ({ - status: 'stopped' as const + status: "stopped" as const, })), onWeixinBindingChanged: vi.fn(() => () => undefined), - onRemoteActivity: vi.fn(() => () => undefined) - } + onRemoteActivity: vi.fn(() => () => undefined), + }; - render() - const weixinProject = channelProjects[0]! - await screen.findByRole('button', { name: '当前项目' }) - selectProjectOption(weixinProject.name) - fireEvent.click( - await screen.findByRole('button', { name: '打开设置' }) - ) + render(); + const weixinProject = channelProjects[0]!; + await screen.findByRole("button", { name: "当前项目" }); + selectProjectOption(weixinProject.name); + fireEvent.click(await screen.findByRole("button", { name: "打开设置" })); + expect(await screen.findByLabelText("微信 ClawBot 项目说明")).toHaveValue( + "微信 ClawBot通道项目", + ); + + fireEvent.click(screen.getByLabelText("项目设置")); + let dialog = screen.getByRole("dialog", { name: "项目设置" }); + const dialogDescription = + within(dialog).getByLabelText("微信 ClawBot 项目说明"); + expect(dialogDescription).toHaveFocus(); + const dialogBackend = + within(dialog).getByLabelText("微信 ClawBot 消息处理后端"); expect( - await screen.findByLabelText('微信 ClawBot 项目说明') - ).toHaveValue('微信 ClawBot通道项目') - - fireEvent.click(screen.getByLabelText('项目设置')) - let dialog = screen.getByRole('dialog', { name: '项目设置' }) - const dialogDescription = within(dialog).getByLabelText( - '微信 ClawBot 项目说明' - ) - expect(dialogDescription).toHaveFocus() - const dialogBackend = within(dialog).getByLabelText( - '微信 ClawBot 消息处理后端' - ) - expect( - within(dialogBackend).getByRole('option', { - name: 'DeepSeek Harness（预览 · OpenAI 兼容）' - }) - ).toBeInTheDocument() + within(dialogBackend).getByRole("option", { + name: "DeepSeek Harness（预览 · OpenAI 兼容）", + }), + ).toBeInTheDocument(); + fireEvent.change(dialogDescription, { target: { value: "从左上角更新" } }); fireEvent.change( - dialogDescription, - { target: { value: '从左上角更新' } } - ) - fireEvent.change( - within(dialog).getByLabelText('微信 ClawBot 默认工作目录'), - { target: { value: 'C:\\FromSwitcher' } } - ) - fireEvent.change( - dialogBackend, - { - target: { - value: agentRuntimeSelectionKey({ - provider: 'opencode' - }) - } - } - ) + within(dialog).getByLabelText("微信 ClawBot 默认工作目录"), + { target: { value: "C:\\FromSwitcher" } }, + ); + fireEvent.change(dialogBackend, { + target: { + value: agentRuntimeSelectionKey({ + provider: "opencode", + }), + }, + }); fireEvent.click( within( - within(dialog).getByRole('group', { - name: '微信 ClawBot 默认模式' - }) - ).getByRole('button', { name: '执行' }) - ) - fireEvent.click( - within(dialog).getByRole('button', { name: '保存项目' }) - ) + within(dialog).getByRole("group", { + name: "微信 ClawBot 默认模式", + }), + ).getByRole("button", { name: "执行" }), + ); + fireEvent.click(within(dialog).getByRole("button", { name: "保存项目" })); await waitFor(() => { - expect( - screen.getByLabelText('微信 ClawBot 项目说明') - ).toHaveValue('从左上角更新') - expect( - screen.getByLabelText('微信 ClawBot 默认工作目录') - ).toHaveValue('C:\\FromSwitcher') - expect( - screen.getByLabelText('微信 ClawBot 消息处理后端') - ).toHaveValue( - agentRuntimeSelectionKey({ provider: 'opencode' }) - ) - }) + expect(screen.getByLabelText("微信 ClawBot 项目说明")).toHaveValue( + "从左上角更新", + ); + expect(screen.getByLabelText("微信 ClawBot 默认工作目录")).toHaveValue( + "C:\\FromSwitcher", + ); + expect(screen.getByLabelText("微信 ClawBot 消息处理后端")).toHaveValue( + agentRuntimeSelectionKey({ provider: "opencode" }), + ); + }); - fireEvent.change( - screen.getByLabelText('微信 ClawBot 项目说明'), - { target: { value: '从消息通道更新' } } - ) - fireEvent.change( - screen.getByLabelText('微信 ClawBot 默认工作目录'), - { target: { value: 'C:\\FromChannels' } } - ) - fireEvent.change( - screen.getByLabelText('微信 ClawBot 消息处理后端'), - { - target: { - value: agentRuntimeSelectionKey({ - provider: 'continue' - }) - } - } - ) + fireEvent.change(screen.getByLabelText("微信 ClawBot 项目说明"), { + target: { value: "从消息通道更新" }, + }); + fireEvent.change(screen.getByLabelText("微信 ClawBot 默认工作目录"), { + target: { value: "C:\\FromChannels" }, + }); + fireEvent.change(screen.getByLabelText("微信 ClawBot 消息处理后端"), { + target: { + value: agentRuntimeSelectionKey({ + provider: "continue", + }), + }, + }); fireEvent.click( within( - screen.getByRole('group', { - name: '微信 ClawBot 默认模式' - }) - ).getByRole('button', { name: '对话' }) - ) - fireEvent.click( - screen.getByRole('button', { name: '保存通道设置' }) - ) + screen.getByRole("group", { + name: "微信 ClawBot 默认模式", + }), + ).getByRole("button", { name: "对话" }), + ); + fireEvent.click(screen.getByRole("button", { name: "保存通道设置" })); await waitFor(() => expect(api.projects.update).toHaveBeenCalledWith( weixinProject.id, expect.objectContaining({ - description: '从消息通道更新', - rootPath: 'C:\\FromChannels', - defaultWorkMode: 'ask', - runtimeSelection: { provider: 'continue' } - }) - ) - ) + description: "从消息通道更新", + rootPath: "C:\\FromChannels", + defaultWorkMode: "ask", + runtimeSelection: { provider: "continue" }, + }), + ), + ); - fireEvent.click(screen.getByLabelText('项目设置')) - dialog = screen.getByRole('dialog', { name: '项目设置' }) + fireEvent.click(screen.getByLabelText("项目设置")); + dialog = screen.getByRole("dialog", { name: "项目设置" }); + expect(within(dialog).getByLabelText("微信 ClawBot 项目说明")).toHaveValue( + "从消息通道更新", + ); expect( - within(dialog).getByLabelText('微信 ClawBot 项目说明') - ).toHaveValue('从消息通道更新') + within(dialog).getByLabelText("微信 ClawBot 默认工作目录"), + ).toHaveValue("C:\\FromChannels"); expect( - within(dialog).getByLabelText('微信 ClawBot 默认工作目录') - ).toHaveValue('C:\\FromChannels') - expect( - within(dialog).getByLabelText('微信 ClawBot 消息处理后端') - ).toHaveValue( - agentRuntimeSelectionKey({ provider: 'continue' }) - ) + within(dialog).getByLabelText("微信 ClawBot 消息处理后端"), + ).toHaveValue(agentRuntimeSelectionKey({ provider: "continue" })); expect( within( - within(dialog).getByRole('group', { - name: '微信 ClawBot 默认模式' - }) - ).getByRole('button', { name: '对话' }) - ).toHaveAttribute('aria-pressed', 'true') - }) + within(dialog).getByRole("group", { + name: "微信 ClawBot 默认模式", + }), + ).getByRole("button", { name: "对话" }), + ).toHaveAttribute("aria-pressed", "true"); + }); - it('shows client-created remote conversations without obsolete approval copy', async () => { + it("shows client-created remote conversations without obsolete approval copy", async () => { const channelProject = { ...project, - id: '00000000-0000-4000-8000-000000000201', - name: '微信 ClawBot', - kind: 'channel' as const, - channel: 'weixin' as const, + id: "00000000-0000-4000-8000-000000000201", + name: "微信 ClawBot", + kind: "channel" as const, + channel: "weixin" as const, runtimeSelection: { - provider: 'model' as const, - profileId: modelProfileId - } - } + provider: "model" as const, + profileId: modelProfileId, + }, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - channelProject - ]) + channelProject, + ]); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000301', + id: "00000000-0000-4000-8000-000000000301", projectId: channelProject.id, runtimeSelection: channelProject.runtimeSelection, remote: { - channel: 'weixin', - accountDisplay: '发送者 ****0001', - conversationType: 'direct' + channel: "weixin", + accountDisplay: "发送者 ****0001", + conversationType: "direct", }, - title: '微信 ClawBot · ****0001', + title: "微信 ClawBot · ****0001", updatedAt: 1_775_000_000_000, - messages: [] - } - ]) - render() + messages: [], + }, + ]); + render(); - await screen.findByRole('button', { name: '当前项目' }) - selectProjectOption(channelProject.name) + await screen.findByRole("button", { name: "当前项目" }); + selectProjectOption(channelProject.name); expect( - screen.getAllByRole('button', { - name: /微信 ClawBot · \*{4}0001/u - }).length - ).toBeGreaterThan(0) + screen.getAllByRole("button", { + name: /微信 ClawBot · \*{4}0001/u, + }).length, + ).toBeGreaterThan(0); expect( screen.getByText( - '请在 微信 ClawBot 客户端继续发送消息。本窗口用于查看历史、任务与执行结果。' - ) - ).toBeInTheDocument() - expect(screen.queryByText(/审批执行/u)).not.toBeInTheDocument() - }) + "请在 微信 ClawBot 客户端继续发送消息。本窗口用于查看历史、任务与执行结果。", + ), + ).toBeInTheDocument(); + expect(screen.queryByText(/审批执行/u)).not.toBeInTheDocument(); + }); - it('falls back when the last active project is no longer available', async () => { + it("falls back when the last active project is no longer available", async () => { localStorage.setItem( - 'goodbuddy.active-project.v1', - '00000000-0000-4000-8000-000000000999' - ) + "goodbuddy.active-project.v1", + "00000000-0000-4000-8000-000000000999", + ); - render() + render(); expect( - await screen.findByRole('button', { name: '当前项目' }) - ).toHaveTextContent(project.name) + await screen.findByRole("button", { name: "当前项目" }), + ).toHaveTextContent(project.name); await waitFor(() => - expect( - localStorage.getItem('goodbuddy.active-project.v1') - ).toBe(project.id) - ) - }) + expect(localStorage.getItem("goodbuddy.active-project.v1")).toBe( + project.id, + ), + ); + }); it.each([ - ['opencode', 'OpenCode'], - ['continue', 'Continue CLI'] - ] as const)( - 'lets %s select Ask or Execute', - async (runtimeId, label) => { + ["opencode", "OpenCode"], + ["continue", "Continue CLI"], + ] as const)("lets %s select Ask or Execute", async (runtimeId, label) => { vi.mocked(api.agent.getStatus).mockResolvedValue({ id: runtimeId, label, available: true, supportsToolExecution: true, - detail: 'Ready' - }) - render() + detail: "Ready", + }); + render(); - const mode = await screen.findByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - expect(mode).toBeEnabled() - expect(mode.closest('.composer')).not.toBeNull() + const mode = await screen.findByRole("button", { + name: "工作模式：Ask · 只读问答", + }); + expect(mode).toBeEnabled(); + expect(mode.closest(".composer")).not.toBeNull(); expect( - await screen.findByText('快捷唤起：', { exact: false }) - ).toBeInTheDocument() - selectComposerOption('工作模式', 'Execute · 完全权限') - expect(mode).toHaveAccessibleName( - '工作模式：Execute · 完全权限' - ) - expect(mode).toHaveTextContent(/^Execute$/u) + await screen.findByText("快捷唤起：", { exact: false }), + ).toBeInTheDocument(); + selectComposerOption("工作模式", "Execute · 完全权限"); + expect(mode).toHaveAccessibleName("工作模式：Execute · 完全权限"); + expect(mode).toHaveTextContent(/^Execute$/u); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '执行任务' } - }) - fireEvent.click(await screen.findByLabelText('发送')) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "执行任务" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); await waitFor(() => expect(run).toHaveBeenCalledWith( expect.objectContaining({ - prompt: '执行任务', - workMode: 'execute' - }) - ) - ) - } - ) + prompt: "执行任务", + workMode: "execute", + }), + ), + ); + }); - it('submits native OpenCode Agent and Command controls', async () => { - const settings = await api.settings.getRuntime() + it("submits native OpenCode Agent and Command controls", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'opencode', + provider: "opencode", opencodeEmbedded: true, - opencodeModelSource: { kind: 'platform' } - }) + opencodeModelSource: { kind: "platform" }, + }); vi.mocked(api.agent.getStatus).mockResolvedValueOnce({ - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: true, supportsToolExecution: true, - detail: 'Ready' - }) - vi.mocked( - api.runtimeCustomization.getSettings - ).mockResolvedValueOnce({ + detail: "Ready", + }); + vi.mocked(api.runtimeCustomization.getSettings).mockResolvedValueOnce({ opencode: {}, - continue: { presets: [] } - }) - vi.mocked( - api.runtimeCustomization.getNativeSnapshot - ).mockResolvedValueOnce({ - provider: 'opencode', - available: true, - inventoryStatus: 'available', - detail: 'Ready', - agents: [ - { - id: 'planner', - name: 'Planner', - description: 'Plan before editing', - mode: 'primary', - native: true, - hidden: false - } - ], - tools: [], - toolsSupported: true, - commands: [ - { - id: 'review', - name: 'review', - description: 'Review a target', - source: 'command' - } - ], - lsp: [], - formatters: [], - mcpServers: [], - skills: [], - rules: [], - prompts: [], - resources: [], - resourcesSupported: true, - context: { - strategy: 'native', - manualCompact: true, - detail: 'OpenCode native context' - } - }) + continue: { presets: [] }, + }); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockResolvedValueOnce( + { + provider: "opencode", + available: true, + inventoryStatus: "available", + detail: "Ready", + agents: [ + { + id: "planner", + name: "Planner", + description: "Plan before editing", + mode: "primary", + native: true, + hidden: false, + }, + ], + tools: [], + toolsSupported: true, + commands: [ + { + id: "review", + name: "review", + description: "Review a target", + source: "command", + }, + ], + lsp: [], + formatters: [], + mcpServers: [], + skills: [], + rules: [], + prompts: [], + resources: [], + resourcesSupported: true, + context: { + strategy: "native", + manualCompact: true, + detail: "OpenCode native context", + }, + }, + ); - render() + render(); - const agentPicker = await screen.findByRole('button', { - name: /OpenCode Runtime Agent/u - }) - const runtimeToolbar = screen.getByRole('group', { - name: 'OpenCode 专属功能' - }) - const universalSettings = screen.getByRole('group', { - name: '对话设置' - }) - expect(runtimeToolbar).toHaveClass('composer__runtime-toolbar') - expect(runtimeToolbar).toContainElement(agentPicker) - expect(universalSettings).not.toContainElement(agentPicker) - expect(universalSettings.closest('.composer__toolbar')).toHaveClass( - 'composer__toolbar--with-runtime-controls' - ) - fireEvent.click(agentPicker) + const agentPicker = await screen.findByRole("button", { + name: /OpenCode Runtime Agent/u, + }); + const runtimeToolbar = screen.getByRole("group", { + name: "OpenCode 专属功能", + }); + const universalSettings = screen.getByRole("group", { + name: "对话设置", + }); + expect(runtimeToolbar).toHaveClass("composer__runtime-toolbar"); + expect(runtimeToolbar).toContainElement(agentPicker); + expect(universalSettings).not.toContainElement(agentPicker); + expect(universalSettings.closest(".composer__toolbar")).toHaveClass( + "composer__toolbar--with-runtime-controls", + ); + fireEvent.click(agentPicker); fireEvent.click( within( - screen.getByRole('menu', { - name: 'OpenCode Runtime Agent' - }) - ).getByRole('menuitemradio', { name: /Planner/u }) - ) - const actionPicker = screen.getByRole('button', { - name: /Runtime 快捷操作/u - }) - expect(runtimeToolbar).toContainElement(actionPicker) - fireEvent.click(actionPicker) + screen.getByRole("menu", { + name: "OpenCode Runtime Agent", + }), + ).getByRole("menuitemradio", { name: /Planner/u }), + ); + const actionPicker = screen.getByRole("button", { + name: /Runtime 快捷操作/u, + }); + expect(runtimeToolbar).toContainElement(actionPicker); + fireEvent.click(actionPicker); fireEvent.click( within( - screen.getByRole('menu', { - name: 'Runtime 快捷操作' - }) - ).getByRole('menuitemradio', { name: /\/review/u }) - ) - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: 'src/main' } - }) - fireEvent.click(screen.getByLabelText('发送')) + screen.getByRole("menu", { + name: "Runtime 快捷操作", + }), + ).getByRole("menuitemradio", { name: /\/review/u }), + ); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "src/main" }, + }); + fireEvent.click(screen.getByLabelText("发送")); await waitFor(() => expect(run).toHaveBeenCalledWith( expect.objectContaining({ - prompt: '/review src/main', + prompt: "/review src/main", runtimeControl: { - provider: 'opencode', - agent: 'planner', + provider: "opencode", + agent: "planner", command: { - name: 'review', - arguments: 'src/main' - } - } - }) - ) - ) - }) + name: "review", + arguments: "src/main", + }, + }, + }), + ), + ); + }); - it('keeps native OpenCode controls mounted across conversation refreshes', async () => { - const conversationId = '00000000-0000-4000-8000-000000000761' - const settings = await api.settings.getRuntime() + it.each(["rejected", "unavailable", "partial"] as const)("retries %s native OpenCode controls and keeps them mounted across conversation refreshes", async (failure) => { + const conversationId = "00000000-0000-4000-8000-000000000761"; + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'opencode', + provider: "opencode", opencodeEmbedded: true, - opencodeModelSource: { kind: 'platform' } - }) + opencodeModelSource: { kind: "platform" }, + }); vi.mocked(api.agent.getStatus).mockResolvedValue({ - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: true, supportsToolExecution: true, - detail: 'Ready' - }) - vi.mocked( - api.runtimeCustomization.getSettings - ).mockResolvedValueOnce({ + detail: "Ready", + }); + vi.mocked(api.runtimeCustomization.getSettings).mockResolvedValueOnce({ opencode: {}, - continue: { presets: [] } - }) - vi.mocked( - api.runtimeCustomization.getNativeSnapshot - ).mockResolvedValueOnce({ - provider: 'opencode', - available: true, - inventoryStatus: 'available', - detail: 'Ready', - agents: [ - { - id: 'planner', - name: 'Planner', - description: 'Plan before editing', - mode: 'primary', - native: true, - hidden: false - } - ], - tools: [], - toolsSupported: true, - commands: [], - lsp: [], - formatters: [], - mcpServers: [], - skills: [], - rules: [], - prompts: [], - resources: [], - resourcesSupported: true, - context: { - strategy: 'native', - manualCompact: true, - detail: 'OpenCode native context' - } - }) + continue: { presets: [] }, + }); + const emptySnapshot = await api.runtimeCustomization.getNativeSnapshot({ provider: "opencode" }); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockClear(); + if (failure === "rejected") { + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockRejectedValueOnce( + new Error("temporary inventory failure"), + ); + } else { + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockResolvedValueOnce({ + ...emptySnapshot, + available: failure === "partial", + inventoryStatus: failure, + }); + } + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockResolvedValueOnce({ + provider: "opencode", + available: true, + inventoryStatus: "available", + detail: "Ready", + agents: [ + { + id: "planner", + name: "Planner", + description: "Plan before editing", + mode: "primary", + native: true, + hidden: false, + }, + ], + tools: [], + toolsSupported: true, + commands: [], + lsp: [], + formatters: [], + mcpServers: [], + skills: [], + rules: [], + prompts: [], + resources: [], + resourcesSupported: true, + context: { + strategy: "native", + manualCompact: true, + detail: "OpenCode native context", + }, + }); // Every read returns freshly parsed rows, so the conversation and its // nested runtime selection change identity on each refresh even when the // selection itself is unchanged. The store keeps the persisted timestamp @@ -7877,933 +7409,1048 @@ describe('App', () => { { id: conversationId, projectId, - runtimeSelection: { provider: 'opencode' }, - title: 'OpenCode 会话', + runtimeSelection: { provider: "opencode" }, + title: "OpenCode 会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000762', - role: 'assistant', - content: '已就绪', + id: "00000000-0000-4000-8000-000000000762", + role: "assistant", + content: "已就绪", createdAt: 1_775_000_000_000, - state: 'complete' - } - ] - } - ]) - let conversationsChangedListener: (() => void) | undefined - vi.mocked(api.conversations.onChanged).mockImplementation( - (listener) => { - conversationsChangedListener = listener - return () => { - if (conversationsChangedListener === listener) { - conversationsChangedListener = undefined - } + state: "complete", + }, + ], + }, + ]); + let conversationsChangedListener: (() => void) | undefined; + vi.mocked(api.conversations.onChanged).mockImplementation((listener) => { + conversationsChangedListener = listener; + return () => { + if (conversationsChangedListener === listener) { + conversationsChangedListener = undefined; } - } - ) + }; + }); - render() + render(); - const runtimeToolbar = await screen.findByRole('group', { - name: 'OpenCode 专属功能' - }) - expect(await screen.findByText('已就绪')).toBeInTheDocument() + const runtimeToolbar = await screen.findByRole("group", { + name: "OpenCode 专属功能", + }); + expect(await screen.findByText("已就绪")).toBeInTheDocument(); await waitFor(() => - expect( - api.runtimeCustomization.getNativeSnapshot - ).toHaveBeenCalled() - ) - vi.mocked( - api.runtimeCustomization.getNativeSnapshot - ).mockClear() + expect(api.runtimeCustomization.getNativeSnapshot).toHaveBeenCalledTimes( + 2, + ), + ); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockClear(); // The resolved runtime scope is unchanged, so the runtime controls must // not unmount and refetch, which visibly shifted the conversation // layout on every window switch. - const refreshesBefore = vi.mocked(api.conversations.list).mock.calls - .length + const refreshesBefore = vi.mocked(api.conversations.list).mock.calls.length; act(() => { - conversationsChangedListener?.() - window.dispatchEvent(new Event('focus')) - }) + conversationsChangedListener?.(); + window.dispatchEvent(new Event("focus")); + }); await waitFor(() => expect( - vi.mocked(api.conversations.list).mock.calls.length - ).toBeGreaterThan(refreshesBefore) - ) + vi.mocked(api.conversations.list).mock.calls.length, + ).toBeGreaterThan(refreshesBefore), + ); // Let the merged conversation render before asserting that the runtime // scope stayed stable. await act(async () => { - await Promise.resolve() - }) + await Promise.resolve(); + }); + expect(screen.getByRole("group", { name: "OpenCode 专属功能" })).toBe( + runtimeToolbar, + ); expect( - screen.getByRole('group', { name: 'OpenCode 专属功能' }) - ).toBe(runtimeToolbar) - expect( - screen.getByRole('button', { name: /OpenCode Runtime Agent/u }) - ).toBeInTheDocument() - expect( - api.runtimeCustomization.getNativeSnapshot - ).not.toHaveBeenCalled() - }) + screen.getByRole("button", { name: /OpenCode Runtime Agent/u }), + ).toBeInTheDocument(); + expect(api.runtimeCustomization.getNativeSnapshot).not.toHaveBeenCalled(); + }); - it('fills editable Continue Prompts and submits the selected preset', async () => { - const presetId = '00000000-0000-4000-8000-000000000721' - const promptId = '00000000-0000-4000-8000-000000000722' - const settings = await api.settings.getRuntime() + it("shows native agents and commands after an unavailable inventory retries to available partial data", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'continue', - continueModelSource: { kind: 'platform' } - }) + provider: "opencode", + opencodeEmbedded: true, + opencodeModelSource: { kind: "platform" }, + }); + vi.mocked(api.agent.getStatus).mockResolvedValue({ + id: "opencode", label: "OpenCode", available: true, + supportsToolExecution: true, detail: "Ready", + }); + const empty = await api.runtimeCustomization.getNativeSnapshot({ provider: "opencode" }); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockClear(); + vi.mocked(api.runtimeCustomization.getNativeSnapshot) + .mockResolvedValueOnce({ + ...empty, + available: false, + inventoryStatus: "unavailable", + }) + .mockResolvedValueOnce({ + ...empty, + available: true, + inventoryStatus: "partial", + agents: [{ id: "planner", name: "Planner", mode: "primary", native: true, hidden: false }], + commands: [{ id: "review", name: "review", source: "command" }], + }); + + render(); + + fireEvent.click(await screen.findByRole("button", { name: /OpenCode Runtime Agent/u })); + expect(screen.getByRole("menuitemradio", { name: /Planner/u })).toBeVisible(); + fireEvent.keyDown(screen.getByRole("menu", { name: "OpenCode Runtime Agent" }), { key: "Escape" }); + fireEvent.click(screen.getByRole("button", { name: /Runtime 快捷操作/u })); + expect(screen.getByRole("menuitemradio", { name: /review/u })).toBeVisible(); + expect(api.runtimeCustomization.getNativeSnapshot).toHaveBeenCalledTimes(2); + }); + + it.each(["unavailable", "partial"] as const)("retains cached native controls through %s project revalidation with a bounded retry", async (inventoryStatus) => { + const settings = await api.settings.getRuntime(); + vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ + ...settings, + provider: "opencode", + opencodeEmbedded: true, + opencodeModelSource: { kind: "platform" }, + }); + vi.mocked(api.agent.getStatus).mockResolvedValue({ + id: "opencode", label: "OpenCode", available: true, + supportsToolExecution: true, detail: "Ready", + }); + const firstProject = { ...project, runtimeSelection: { provider: "opencode" as const } }; + const secondProject = { ...firstProject, id: "00000000-0000-4000-8000-000000000179", name: "Inventory project B" }; + vi.mocked(api.projects.list).mockResolvedValueOnce([firstProject, secondProject]); + const empty = await api.runtimeCustomization.getNativeSnapshot({ provider: "opencode" }); + const healthy = { + ...empty, + agents: [{ id: "planner", name: "Planner", mode: "primary" as const, native: true, hidden: false }], + }; + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockClear(); + vi.mocked(api.runtimeCustomization.getNativeSnapshot) + .mockResolvedValueOnce(healthy) + .mockResolvedValueOnce(healthy); + render(); + await screen.findByRole("button", { name: /OpenCode Runtime Agent/u }); + selectProjectOption(secondProject.name); + await waitFor(() => expect(api.runtimeCustomization.getNativeSnapshot).toHaveBeenLastCalledWith( + expect.objectContaining({ projectId: secondProject.id }), + )); + const pending = deferred(); + const failed = { ...empty, available: inventoryStatus === "partial", inventoryStatus }; + vi.mocked(api.runtimeCustomization.getNativeSnapshot) + .mockReturnValueOnce(pending.promise) + .mockResolvedValueOnce(failed); + const callsBefore = vi.mocked(api.runtimeCustomization.getNativeSnapshot).mock.calls.length; + selectProjectOption(firstProject.name); + const toolbar = await screen.findByRole("group", { name: "OpenCode 专属功能" }); + await act(async () => { + pending.resolve(failed); + await pending.promise; + }); + await waitFor(() => expect(api.runtimeCustomization.getNativeSnapshot).toHaveBeenCalledTimes(callsBefore + 2)); + expect(screen.getByRole("group", { name: "OpenCode 专属功能" })).toBe(toolbar); + fireEvent.click(screen.getByRole("button", { name: /OpenCode Runtime Agent/u })); + expect(screen.getByRole("menuitemradio", { name: /Planner/u })).toBeInTheDocument(); + fireEvent.keyDown(screen.getByRole("menu", { name: "OpenCode Runtime Agent" }), { key: "Escape" }); + + // A successful empty inventory is authoritative, not a transient failure. + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockResolvedValueOnce(healthy); + selectProjectOption(secondProject.name); + await waitFor(() => expect(api.runtimeCustomization.getNativeSnapshot).toHaveBeenCalledTimes(callsBefore + 3)); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockResolvedValueOnce(empty); + selectProjectOption(firstProject.name); + await waitFor(() => expect(screen.queryByRole("group", { name: "OpenCode 专属功能" })).not.toBeInTheDocument()); + expect(api.runtimeCustomization.getNativeSnapshot).toHaveBeenCalledTimes(callsBefore + 4); + }); + + it("resets selected native Agent and Command when switching same-scope conversations", async () => { + const settings = await api.settings.getRuntime(); + vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ + ...settings, + provider: "opencode", + opencodeEmbedded: true, + opencodeModelSource: { kind: "platform" }, + }); + vi.mocked(api.agent.getStatus).mockResolvedValue({ + id: "opencode", label: "OpenCode", available: true, + supportsToolExecution: true, detail: "Ready", + }); + const snapshot = await api.runtimeCustomization.getNativeSnapshot({ provider: "opencode" }); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockClear(); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockResolvedValueOnce({ + ...snapshot, + agents: [{ id: "planner", name: "Planner", mode: "primary", native: true, hidden: false }], + commands: [{ id: "review", name: "review", source: "command" }], + }); + const rows = [1, 2].map((index) => ({ + id: `00000000-0000-4000-8000-00000000077${index}`, + projectId, + runtimeSelection: { provider: "opencode" as const }, + title: `Native conversation ${index}`, + updatedAt: 1_775_000_000_000 - index, + messages: [], + })); + vi.mocked(api.conversations.list).mockResolvedValueOnce(rows); + render(); + + const toolbar = await screen.findByRole("group", { name: "OpenCode 专属功能" }); + fireEvent.click(screen.getByRole("button", { name: /OpenCode Runtime Agent/u })); + fireEvent.click(screen.getByRole("menuitemradio", { name: /Planner/u })); + fireEvent.click(screen.getByRole("button", { name: /Runtime 快捷操作/u })); + fireEvent.click(screen.getByRole("menuitemradio", { name: /\/review/u })); + const inventoryCalls = vi.mocked(api.runtimeCustomization.getNativeSnapshot).mock.calls.length; + fireEvent.click(screen.getByText(rows[1]!.title).closest("button")!); + expect(screen.getByRole("group", { name: "OpenCode 专属功能" })).toBe(toolbar); + expect(api.runtimeCustomization.getNativeSnapshot).toHaveBeenCalledTimes(inventoryCalls); + await waitFor(() => expect(screen.getByRole("button", { name: /OpenCode Runtime Agent/u })).not.toHaveTextContent("Planner")); + expect(screen.getByRole("button", { name: /Runtime 快捷操作/u })).not.toHaveTextContent("/review"); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { target: { value: "plain request" } }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + expect(run.mock.calls[0]?.[0]).toMatchObject({ conversationId: rows[1]!.id, prompt: "plain request" }); + expect(run.mock.calls[0]?.[0].runtimeControl).toBeUndefined(); + }); + + it("fills editable Continue Prompts and submits the selected preset", async () => { + const presetId = "00000000-0000-4000-8000-000000000721"; + const promptId = "00000000-0000-4000-8000-000000000722"; + const settings = await api.settings.getRuntime(); + vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ + ...settings, + provider: "continue", + continueModelSource: { kind: "platform" }, + }); vi.mocked(api.agent.getStatus).mockResolvedValueOnce({ - id: 'continue', - label: 'Continue', + id: "continue", + label: "Continue", available: true, supportsToolExecution: true, - detail: 'Ready' - }) - vi.mocked( - api.runtimeCustomization.getSettings - ).mockResolvedValueOnce({ + detail: "Ready", + }); + vi.mocked(api.runtimeCustomization.getSettings).mockResolvedValueOnce({ opencode: {}, continue: { defaultPresetId: presetId, presets: [ { id: presetId, - name: '代码审查', + name: "代码审查", rules: [], prompts: [ { id: promptId, - name: '审查草稿', - description: '检查当前草稿', - prompt: '请审查当前草稿。' - } - ] - } - ] - } - }) - vi.mocked( - api.runtimeCustomization.getNativeSnapshot - ).mockResolvedValueOnce({ - provider: 'continue', - available: true, - inventoryStatus: 'available', - detail: 'Ready', - agents: [], - tools: [], - toolsSupported: false, - commands: [], - lsp: [], - formatters: [], - mcpServers: [], - skills: [], - rules: [], - prompts: [ - { - id: promptId, - name: '原生同 ID Prompt', - prompt: '不应填入此原生 Prompt。', - source: 'configuration' - } - ], - resources: [], - resourcesSupported: false, - context: { - strategy: 'goodbuddy-summary', - manualCompact: true, - detail: 'GoodBuddy summary context' - } - }) + name: "审查草稿", + description: "检查当前草稿", + prompt: "请审查当前草稿。", + }, + ], + }, + ], + }, + }); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockResolvedValueOnce( + { + provider: "continue", + available: true, + inventoryStatus: "available", + detail: "Ready", + agents: [], + tools: [], + toolsSupported: false, + commands: [], + lsp: [], + formatters: [], + mcpServers: [], + skills: [], + rules: [], + prompts: [ + { + id: promptId, + name: "原生同 ID Prompt", + prompt: "不应填入此原生 Prompt。", + source: "configuration", + }, + ], + resources: [], + resourcesSupported: false, + context: { + strategy: "goodbuddy-summary", + manualCompact: true, + detail: "GoodBuddy summary context", + }, + }, + ); - render() + render(); - const presetPicker = await screen.findByRole('button', { - name: /Continue 配置预设.*使用设置默认预设/u - }) - const runtimeToolbar = screen.getByRole('group', { - name: 'Continue 专属功能' - }) - expect(runtimeToolbar).toHaveClass('composer__runtime-toolbar') - expect(runtimeToolbar).toContainElement(presetPicker) + const presetPicker = await screen.findByRole("button", { + name: /Continue 配置预设.*使用设置默认预设/u, + }); + const runtimeToolbar = screen.getByRole("group", { + name: "Continue 专属功能", + }); + expect(runtimeToolbar).toHaveClass("composer__runtime-toolbar"); + expect(runtimeToolbar).toContainElement(presetPicker); expect( - screen.getByRole('group', { name: '对话设置' }) - ).not.toContainElement(presetPicker) - fireEvent.click(presetPicker) + screen.getByRole("group", { name: "对话设置" }), + ).not.toContainElement(presetPicker); + fireEvent.click(presetPicker); fireEvent.click( within( - screen.getByRole('menu', { - name: 'Continue 配置预设' - }) - ).getByRole('menuitemradio', { name: /代码审查/u }) - ) - const actionPicker = screen.getByRole('button', { - name: /Runtime 快捷操作/u - }) - expect(runtimeToolbar).toContainElement(actionPicker) - fireEvent.click(actionPicker) + screen.getByRole("menu", { + name: "Continue 配置预设", + }), + ).getByRole("menuitemradio", { name: /代码审查/u }), + ); + const actionPicker = screen.getByRole("button", { + name: /Runtime 快捷操作/u, + }); + expect(runtimeToolbar).toContainElement(actionPicker); + fireEvent.click(actionPicker); fireEvent.click( within( - screen.getByRole('menu', { - name: 'Runtime 快捷操作' - }) - ).getByRole('menuitemradio', { name: /审查草稿/u }) - ) - const composer = screen.getByLabelText('向 GoodBuddy 提问') - expect(composer).toHaveValue('请审查当前草稿。') + screen.getByRole("menu", { + name: "Runtime 快捷操作", + }), + ).getByRole("menuitemradio", { name: /审查草稿/u }), + ); + const composer = screen.getByLabelText("向 GoodBuddy 提问"); + expect(composer).toHaveValue("请审查当前草稿。"); fireEvent.change(composer, { - target: { value: '请审查当前草稿，并优先检查权限。' } - }) - fireEvent.click(screen.getByLabelText('发送')) + target: { value: "请审查当前草稿，并优先检查权限。" }, + }); + fireEvent.click(screen.getByLabelText("发送")); await waitFor(() => expect(run).toHaveBeenCalledWith( expect.objectContaining({ - prompt: '请审查当前草稿，并优先检查权限。', + prompt: "请审查当前草稿，并优先检查权限。", runtimeControl: { - provider: 'continue', - presetId - } - }) - ) - ) - }) + provider: "continue", + presetId, + }, + }), + ), + ); + }); - it('manually compacts Continue context and persists the summary state', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000731' + it("manually compacts Continue context and persists the summary state", async () => { + const conversationId = "00000000-0000-4000-8000-000000000731"; const messages = [ { - id: '00000000-0000-4000-8000-000000000732', - role: 'user' as const, - content: '第一轮问题', + id: "00000000-0000-4000-8000-000000000732", + role: "user" as const, + content: "第一轮问题", createdAt: 1_775_000_000_000, - state: 'complete' as const + state: "complete" as const, }, { - id: '00000000-0000-4000-8000-000000000733', - role: 'assistant' as const, - content: '第一轮回答', + id: "00000000-0000-4000-8000-000000000733", + role: "assistant" as const, + content: "第一轮回答", createdAt: 1_775_000_000_001, - state: 'complete' as const - } - ] + state: "complete" as const, + }, + ]; const summaryState = { - coveredHistoryDigest: 'a'.repeat(64), + coveredHistoryDigest: "a".repeat(64), coveredMessageCount: 1, coveredFromMessageId: messages[0]!.id, coveredThroughMessageId: messages[0]!.id, - summary: '用户提出了第一轮问题。' - } - const settings = await api.settings.getRuntime() + summary: "用户提出了第一轮问题。", + }; + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, contextCompression: { enabled: true, triggerTokens: 20_000, recentRawTokens: 4_000, - modelSource: { kind: 'current' }, - summaryPrompt: 'Preserve important facts.' - } - }) + modelSource: { kind: "current" }, + summaryPrompt: "Preserve important facts.", + }, + }); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: conversationId, projectId, - runtimeSelection: { provider: 'continue' }, - title: 'Continue 长对话', + runtimeSelection: { provider: "continue" }, + title: "Continue 长对话", updatedAt: 1_775_000_000_001, - messages - } - ]) + messages, + }, + ]); vi.mocked(api.agent.getStatus).mockResolvedValueOnce({ - id: 'continue', - label: 'Continue', + id: "continue", + label: "Continue", available: true, supportsToolExecution: true, - detail: 'Ready' - }) - vi.mocked( - api.runtimeCustomization.getNativeSnapshot - ).mockResolvedValueOnce({ - provider: 'continue', - available: true, - inventoryStatus: 'available', - detail: 'Ready', - agents: [], - tools: [], - toolsSupported: false, - commands: [], - lsp: [], - formatters: [], - mcpServers: [], - skills: [], - rules: [], - prompts: [], - resources: [], - resourcesSupported: false, - context: { - strategy: 'goodbuddy-summary', - manualCompact: true, - detail: 'GoodBuddy summary context' - } - }) + detail: "Ready", + }); + vi.mocked(api.runtimeCustomization.getNativeSnapshot).mockResolvedValueOnce( + { + provider: "continue", + available: true, + inventoryStatus: "available", + detail: "Ready", + agents: [], + tools: [], + toolsSupported: false, + commands: [], + lsp: [], + formatters: [], + mcpServers: [], + skills: [], + rules: [], + prompts: [], + resources: [], + resourcesSupported: false, + context: { + strategy: "goodbuddy-summary", + manualCompact: true, + detail: "GoodBuddy summary context", + }, + }, + ); vi.mocked(api.agent.compactConversation).mockResolvedValueOnce({ - provider: 'continue', - strategy: 'goodbuddy-summary', + provider: "continue", + strategy: "goodbuddy-summary", compacted: true, - detail: '已压缩 Continue 对话历史', - contextCompressionState: summaryState - }) + detail: "已压缩 Continue 对话历史", + contextCompressionState: summaryState, + }); - render() + render(); - const compactContext = await screen.findByRole('button', { - name: '压缩上下文' - }) + const compactContext = await screen.findByRole("button", { + name: "压缩上下文", + }); expect(compactContext.parentElement).toHaveClass( - 'composer-meta', - 'composer-meta--with-context-compact' - ) - expect( - compactContext.parentElement?.firstElementChild - ).toBe(compactContext) - fireEvent.click(compactContext) + "composer-meta", + "composer-meta--with-context-compact", + ); + expect(compactContext.parentElement?.firstElementChild).toBe( + compactContext, + ); + fireEvent.click(compactContext); await waitFor(() => expect(api.agent.compactConversation).toHaveBeenCalledWith({ requestId: expect.any(String), conversationId, projectId, - runtimeSelection: { provider: 'continue' }, + runtimeSelection: { provider: "continue" }, history: messages.map(({ role, content }) => ({ role, - content + content, })), historyMessageIds: messages.map((message) => message.id), - contextCompressionState: undefined - }) - ) + contextCompressionState: undefined, + }), + ); expect( - await screen.findByText('已压缩 Continue 对话历史') - ).toBeInTheDocument() - expect( - await screen.findByText(/压缩后对话估算/u) - ).not.toHaveTextContent('压缩线') + await screen.findByText("已压缩 Continue 对话历史"), + ).toBeInTheDocument(); + expect(await screen.findByText(/压缩后对话估算/u)).not.toHaveTextContent( + "压缩线", + ); await waitFor(() => expect(api.conversations.saveLocal).toHaveBeenCalledWith([ expect.objectContaining({ header: expect.objectContaining({ contextCompressionState: summaryState, contextMetrics: expect.objectContaining({ - basis: 'conversation', - source: 'estimated' - }) - }) - }) - ]) - ) - }) + basis: "conversation", + source: "estimated", + }), + }), + }), + ]), + ); + }); - it('restores the direct-model mode after leaving an Agent Runtime', async () => { - const settings = await api.settings.getRuntime() + it("restores the direct-model mode after leaving an Agent Runtime", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'opencode', + provider: "opencode", opencodeEmbedded: true, - opencodeModelSource: { kind: 'platform' } - }) + opencodeModelSource: { kind: "platform" }, + }); vi.mocked(api.agent.getStatus) .mockResolvedValueOnce({ - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: true, supportsToolExecution: true, - detail: 'Ready' + detail: "Ready", }) .mockResolvedValueOnce({ - id: 'model', - label: 'sonnet-5', + id: "model", + label: "sonnet-5", available: true, supportsToolExecution: false, - detail: 'Ready' - }) - render() + detail: "Ready", + }); + render(); - const mode = await screen.findByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - expect(mode).toBeEnabled() - selectComposerOption('工作模式', 'Execute · 完全权限') - expect(mode).toHaveAccessibleName( - '工作模式：Execute · 完全权限' - ) + const mode = await screen.findByRole("button", { + name: "工作模式：Ask · 只读问答", + }); + expect(mode).toBeEnabled(); + selectComposerOption("工作模式", "Execute · 完全权限"); + expect(mode).toHaveAccessibleName("工作模式：Execute · 完全权限"); - fireEvent.click(await screen.findByRole('button', { name: /OpenCode/u })) + fireEvent.click(await screen.findByRole("button", { name: /OpenCode/u })); fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^默认模型.*sonnet-5$/u - }) - ) + screen.getByRole("menuitemradio", { + name: /^默认模型.*sonnet-5$/u, + }), + ); await waitFor(() => { - expect(mode).toHaveAccessibleName('工作模式：Ask · 只读问答') - expect(mode).toBeEnabled() - }) - }) + expect(mode).toHaveAccessibleName("工作模式：Ask · 只读问答"); + expect(mode).toBeEnabled(); + }); + }); - it('disables Execute for a runtime without tool support', async () => { + it("disables Execute for a runtime without tool support", async () => { vi.mocked(api.agent.getStatus).mockResolvedValue({ - id: 'model', - label: 'legacy-model', + id: "model", + label: "legacy-model", available: true, supportsToolExecution: false, - detail: 'Ready' - }) - render() + detail: "Ready", + }); + render(); - const mode = await screen.findByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - const modeMenu = openComposerMenu('工作模式') + const mode = await screen.findByRole("button", { + name: "工作模式：Ask · 只读问答", + }); + const modeMenu = openComposerMenu("工作模式"); expect( - within(modeMenu).getByRole('menuitemradio', { - name: /^Execute · 完全权限/u - }) - ).toBeDisabled() - expect(mode).toHaveAccessibleName('工作模式：Ask · 只读问答') - }) + within(modeMenu).getByRole("menuitemradio", { + name: /^Execute · 完全权限/u, + }), + ).toBeDisabled(); + expect(mode).toHaveAccessibleName("工作模式：Ask · 只读问答"); + }); - it('allows a direct model to submit Execute with GoodBuddy approvals', async () => { + it("allows a direct model to submit Execute with GoodBuddy approvals", async () => { vi.mocked(api.agent.getStatus).mockResolvedValue({ - id: 'model', - label: 'sonnet-5', + id: "model", + label: "sonnet-5", available: true, supportsToolExecution: true, - detail: 'Ready' - }) - render() + detail: "Ready", + }); + render(); - const mode = await screen.findByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - selectComposerOption('工作模式', 'Execute · 完全权限') - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '读取项目文件' } - }) - fireEvent.click(await screen.findByLabelText('发送')) + const mode = await screen.findByRole("button", { + name: "工作模式：Ask · 只读问答", + }); + selectComposerOption("工作模式", "Execute · 完全权限"); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "读取项目文件" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); await waitFor(() => expect(run).toHaveBeenCalledWith( expect.objectContaining({ - prompt: '读取项目文件', - workMode: 'execute' - }) - ) - ) - expect(mode).toBeDisabled() - }) + prompt: "读取项目文件", + workMode: "execute", + }), + ), + ); + expect(mode).toBeDisabled(); + }); - it('terminalizes tools and activity when a request is cancelled', async () => { + it("terminalizes tools and activity when a request is cancelled", async () => { vi.mocked(api.agent.getStatus).mockResolvedValue({ - id: 'opencode', - label: 'OpenCode', + id: "opencode", + label: "OpenCode", available: true, supportsToolExecution: true, - detail: 'Ready' - }) - render() - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '执行长任务' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + detail: "Ready", + }); + render(); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "执行长任务" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'tool', - callId: 'call-1', - name: 'bash', - state: 'running', - summary: 'OpenCode 工具：bash' - }) + type: "tool", + callId: "call-1", + name: "bash", + state: "running", + summary: "OpenCode 工具：bash", + }); agentListener?.({ requestId: request.requestId, - type: 'error', - status: 'cancelled', - message: '请求已取消' - }) - }) + type: "error", + status: "cancelled", + message: "请求已取消", + }); + }); - expect(await screen.findByText('已取消')).toBeInTheDocument() + expect(await screen.findByText("已取消")).toBeInTheDocument(); const cancelledStatus = screen - .getAllByText('请求已取消') - .find((element) => element.classList.contains('message__status')) - expect(cancelledStatus).toBeDefined() - const cancelledDot = cancelledStatus?.querySelector( - '.message__status-dot' - ) - expect(cancelledDot).toHaveClass('message__status-dot') - expect(cancelledDot).not.toHaveClass('message__status-dot--active') - fireEvent.click(screen.getByText('运行记录')) - expect((await screen.findAllByText('已取消')).length).toBeGreaterThan(0) - fireEvent.click( - screen.getByRole('button', { name: /^进行中 \d+$/u }) - ) + .getAllByText("请求已取消") + .find((element) => element.classList.contains("message__status")); + expect(cancelledStatus).toBeDefined(); + const cancelledDot = cancelledStatus?.querySelector(".message__status-dot"); + expect(cancelledDot).toHaveClass("message__status-dot"); + expect(cancelledDot).not.toHaveClass("message__status-dot--active"); + fireEvent.click(screen.getByText("运行记录")); + expect((await screen.findAllByText("已取消")).length).toBeGreaterThan(0); + fireEvent.click(screen.getByRole("button", { name: /^进行中 \d+$/u })); expect( - screen.getByText('当前没有等待中或正在运行的活动。') - ).toBeInTheDocument() - }) + screen.getByText("当前没有等待中或正在运行的活动。"), + ).toBeInTheDocument(); + }); - it('keeps persisted completed message statuses static', async () => { + it("keeps persisted completed message statuses static", async () => { vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000210', + id: "00000000-0000-4000-8000-000000000210", projectId, - title: '已完成会话', + title: "已完成会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000211', - role: 'assistant', - content: '任务结果', + id: "00000000-0000-4000-8000-000000000211", + role: "assistant", + content: "任务结果", createdAt: 1_775_000_000_000, - state: 'complete', - status: '任务已完成' - } - ] - } - ]) + state: "complete", + status: "任务已完成", + }, + ], + }, + ]); - render() + render(); - const completedStatus = await screen.findByText('任务已完成') + const completedStatus = await screen.findByText("任务已完成"); + expect(completedStatus.querySelector(".message__status-dot")).toHaveClass( + "message__status-dot", + ); expect( - completedStatus.querySelector('.message__status-dot') - ).toHaveClass('message__status-dot') - expect( - completedStatus.querySelector('.message__status-dot') - ).not.toHaveClass('message__status-dot--active') - }) + completedStatus.querySelector(".message__status-dot"), + ).not.toHaveClass("message__status-dot--active"); + }); - it('switches runtime profiles from the composer dropdown', async () => { - render() + it("switches runtime profiles from the composer dropdown", async () => { + render(); - const runtimeButton = await screen.findByRole('button', { - name: /sonnet-5/u - }) - fireEvent.click(runtimeButton) + const runtimeButton = await screen.findByRole("button", { + name: /sonnet-5/u, + }); + fireEvent.click(runtimeButton); expect( - await screen.findByRole('menu', { name: 'Runtime 和模型' }) - ).toBeInTheDocument() + await screen.findByRole("menu", { name: "Runtime 和模型" }), + ).toBeInTheDocument(); + expect(screen.queryByText("自动选择")).not.toBeInTheDocument(); expect( - screen.queryByText('自动选择') - ).not.toBeInTheDocument() + screen.getByRole("menuitemradio", { + name: /^默认模型.*sonnet-5$/u, + }), + ).toBeInTheDocument(); expect( - screen.getByRole('menuitemradio', { - name: /^默认模型.*sonnet-5$/u - }) - ).toBeInTheDocument() + screen.getByRole("menuitemradio", { + name: /^OpenCode · 默认模型.*sonnet-5$/u, + }), + ).toBeInTheDocument(); expect( - screen.getByRole('menuitemradio', { - name: /^OpenCode · 默认模型.*sonnet-5$/u - }) - ).toBeInTheDocument() + screen.getByRole("menuitemradio", { + name: /^Continue · 默认模型.*sonnet-5$/u, + }), + ).toBeInTheDocument(); expect( - screen.getByRole('menuitemradio', { - name: /^Continue · 默认模型.*sonnet-5$/u - }) - ).toBeInTheDocument() - expect( - screen.getByRole('menuitemradio', { - name: /^DeepSeek Harness · 自身配置/u - }) - ).toBeInTheDocument() + screen.getByRole("menuitemradio", { + name: /^DeepSeek Harness · 自身配置/u, + }), + ).toBeInTheDocument(); fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^默认模型.*sonnet-5$/u - }) - ) + screen.getByRole("menuitemradio", { + name: /^默认模型.*sonnet-5$/u, + }), + ); await waitFor(() => expect(api.agent.getStatus).toHaveBeenLastCalledWith({ - provider: 'model', - profileId: modelProfileId - }) - ) - expect(api.settings.updateRuntime).not.toHaveBeenCalled() + provider: "model", + profileId: modelProfileId, + }), + ); + expect(api.settings.updateRuntime).not.toHaveBeenCalled(); expect( - screen.queryByRole('heading', { name: '设置中心' }) - ).not.toBeInTheDocument() - }) + screen.queryByRole("heading", { name: "设置中心" }), + ).not.toBeInTheDocument(); + }); - it('shows Runtime switches globally without replacing composer guidance', async () => { - render() - const runtimeButton = await screen.findByRole('button', { - name: /sonnet-5/u - }) - vi.useFakeTimers() + it("shows Runtime switches globally without replacing composer guidance", async () => { + render(); + const runtimeButton = await screen.findByRole("button", { + name: /sonnet-5/u, + }); + vi.useFakeTimers(); try { - fireEvent.click(runtimeButton) + fireEvent.click(runtimeButton); fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^OpenCode · 默认模型.*sonnet-5$/u - }) - ) + screen.getByRole("menuitemradio", { + name: /^OpenCode · 默认模型.*sonnet-5$/u, + }), + ); await act(async () => { - await Promise.resolve() - }) + await Promise.resolve(); + }); - const notification = screen.getByRole('status') + const notification = screen.getByRole("status"); expect(notification).toHaveTextContent( - '当前对话已切换到 OpenCode · 默认模型' - ) + "当前对话已切换到 OpenCode · 默认模型", + ); expect( - screen.getByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - ).toBeInTheDocument() + screen.getByRole("button", { + name: "工作模式：Ask · 只读问答", + }), + ).toBeInTheDocument(); fireEvent.click( - screen.getByRole('button', { - name: /OpenCode · 默认模型/u - }) - ) + screen.getByRole("button", { + name: /OpenCode · 默认模型/u, + }), + ); fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^Continue · 默认模型.*sonnet-5$/u - }) - ) + screen.getByRole("menuitemradio", { + name: /^Continue · 默认模型.*sonnet-5$/u, + }), + ); await act(async () => { - await Promise.resolve() - }) - expect(screen.getAllByRole('status')).toHaveLength(1) - expect(screen.getByRole('status')).toHaveTextContent( - '当前对话已切换到 Continue · 默认模型' - ) + await Promise.resolve(); + }); + expect(screen.getAllByRole("status")).toHaveLength(1); + expect(screen.getByRole("status")).toHaveTextContent( + "当前对话已切换到 Continue · 默认模型", + ); - act(() => vi.advanceTimersByTime(4_500)) + act(() => vi.advanceTimersByTime(4_500)); expect( - screen.queryByText( - '当前对话已切换到 Continue · 默认模型' - ) - ).not.toBeInTheDocument() + screen.queryByText("当前对话已切换到 Continue · 默认模型"), + ).not.toBeInTheDocument(); expect( - screen.getByRole('button', { - name: '工作模式：Ask · 只读问答' - }) - ).toBeInTheDocument() + screen.getByRole("button", { + name: "工作模式：Ask · 只读问答", + }), + ).toBeInTheDocument(); } finally { - vi.useRealTimers() + vi.useRealTimers(); } - }) + }); - it('shows one configured choice per Agent Runtime in a flat keyboard menu', async () => { - const settings = await api.settings.getRuntime() - const secondProfileId = - '00000000-0000-4000-8000-000000000002' + it("shows one configured choice per Agent Runtime in a flat keyboard menu", async () => { + const settings = await api.settings.getRuntime(); + const secondProfileId = "00000000-0000-4000-8000-000000000002"; vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'model', + provider: "model", modelProfiles: [ ...settings.modelProfiles, { id: secondProfileId, - name: '第二模型', - baseUrl: 'http://127.0.0.1:11434/v1', - modelName: 'qwen3', - protocol: 'openai-chat-completions', - authentication: 'none', - imageGenerationQuality: 'auto', + name: "第二模型", + baseUrl: "http://127.0.0.1:11434/v1", + modelName: "qwen3", + protocol: "openai-chat-completions", + authentication: "none", + imageGenerationQuality: "auto", apiKeyConfigured: false, - credentialSource: 'none' - } - ] - }) - render() + credentialSource: "none", + }, + ], + }); + render(); - const runtimeButton = await screen.findByRole('button', { - name: /sonnet-5/u - }) - fireEvent.click(runtimeButton) - const runtimeMenu = screen.getByRole('menu', { - name: 'Runtime 和模型' - }) - const directModel = screen.getByRole('menuitemradio', { - name: /^默认模型.*sonnet-5$/u - }) - const secondDirectModel = screen.getByRole('menuitemradio', { - name: /^第二模型.*qwen3$/u - }) - const openCodeModel = screen.getByRole('menuitemradio', { - name: /^OpenCode · 默认模型.*sonnet-5$/u - }) - const continueModel = screen.getByRole('menuitemradio', { - name: /^Continue · 默认模型.*sonnet-5$/u - }) - const deepseekHarness = screen.getByRole('menuitemradio', { - name: /^DeepSeek Harness · 自身配置/u - }) - expect(directModel).toBeEnabled() - expect(secondDirectModel).toBeEnabled() - expect(openCodeModel).toBeEnabled() - expect(continueModel).toBeEnabled() - expect(deepseekHarness).toBeEnabled() - expect(screen.getAllByRole('menuitemradio')).toHaveLength(5) - expect(within(runtimeMenu).getAllByRole('separator')).toHaveLength(4) - expect(within(runtimeMenu).queryByRole('menu')).not.toBeInTheDocument() + const runtimeButton = await screen.findByRole("button", { + name: /sonnet-5/u, + }); + fireEvent.click(runtimeButton); + const runtimeMenu = screen.getByRole("menu", { + name: "Runtime 和模型", + }); + const directModel = screen.getByRole("menuitemradio", { + name: /^默认模型.*sonnet-5$/u, + }); + const secondDirectModel = screen.getByRole("menuitemradio", { + name: /^第二模型.*qwen3$/u, + }); + const openCodeModel = screen.getByRole("menuitemradio", { + name: /^OpenCode · 默认模型.*sonnet-5$/u, + }); + const continueModel = screen.getByRole("menuitemradio", { + name: /^Continue · 默认模型.*sonnet-5$/u, + }); + const deepseekHarness = screen.getByRole("menuitemradio", { + name: /^DeepSeek Harness · 自身配置/u, + }); + expect(directModel).toBeEnabled(); + expect(secondDirectModel).toBeEnabled(); + expect(openCodeModel).toBeEnabled(); + expect(continueModel).toBeEnabled(); + expect(deepseekHarness).toBeEnabled(); + expect(screen.getAllByRole("menuitemradio")).toHaveLength(5); + expect(within(runtimeMenu).getAllByRole("separator")).toHaveLength(4); + expect(within(runtimeMenu).queryByRole("menu")).not.toBeInTheDocument(); expect( - screen.queryByRole('menuitemradio', { - name: /^OpenCode · 第二模型/u - }) - ).not.toBeInTheDocument() + screen.queryByRole("menuitemradio", { + name: /^OpenCode · 第二模型/u, + }), + ).not.toBeInTheDocument(); expect( - screen.queryByRole('menuitemradio', { - name: /^Continue · 第二模型/u - }) - ).not.toBeInTheDocument() + screen.queryByRole("menuitemradio", { + name: /^Continue · 第二模型/u, + }), + ).not.toBeInTheDocument(); expect( - screen.queryByRole('menuitem', { name: /Agent Runtime/u }) - ).not.toBeInTheDocument() + screen.queryByRole("menuitem", { name: /Agent Runtime/u }), + ).not.toBeInTheDocument(); - await waitFor(() => expect(directModel).toHaveFocus()) - expect(directModel).toHaveAttribute('tabindex', '0') - expect(secondDirectModel).toHaveAttribute('tabindex', '-1') - fireEvent.keyDown(directModel, { key: 'ArrowDown' }) - expect(secondDirectModel).toHaveFocus() - expect(directModel).toHaveAttribute('tabindex', '-1') - expect(secondDirectModel).toHaveAttribute('tabindex', '0') - fireEvent.keyDown(secondDirectModel, { key: 'ArrowDown' }) - expect(openCodeModel).toHaveFocus() - fireEvent.keyDown(document.activeElement!, { key: 'Escape' }) - expect(runtimeButton).toHaveFocus() + await waitFor(() => expect(directModel).toHaveFocus()); + expect(directModel).toHaveAttribute("tabindex", "0"); + expect(secondDirectModel).toHaveAttribute("tabindex", "-1"); + fireEvent.keyDown(directModel, { key: "ArrowDown" }); + expect(secondDirectModel).toHaveFocus(); + expect(directModel).toHaveAttribute("tabindex", "-1"); + expect(secondDirectModel).toHaveAttribute("tabindex", "0"); + fireEvent.keyDown(secondDirectModel, { key: "ArrowDown" }); + expect(openCodeModel).toHaveFocus(); + fireEvent.keyDown(document.activeElement!, { key: "Escape" }); + expect(runtimeButton).toHaveFocus(); expect( - screen.queryByRole('menu', { name: 'Runtime 和模型' }) - ).not.toBeInTheDocument() + screen.queryByRole("menu", { name: "Runtime 和模型" }), + ).not.toBeInTheDocument(); - fireEvent.click(runtimeButton) + fireEvent.click(runtimeButton); fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^OpenCode · 默认模型.*sonnet-5$/u - }) - ) - expect(runtimeButton).toHaveFocus() + screen.getByRole("menuitemradio", { + name: /^OpenCode · 默认模型.*sonnet-5$/u, + }), + ); + expect(runtimeButton).toHaveFocus(); await waitFor(() => expect(api.agent.getStatus).toHaveBeenLastCalledWith({ - provider: 'opencode', - profileId: modelProfileId - }) - ) + provider: "opencode", + profileId: modelProfileId, + }), + ); - const selectedRuntimeButton = await screen.findByRole('button', { - name: /OpenCode · 默认模型/u - }) - fireEvent.click(selectedRuntimeButton) - const selectedOpenCodeModel = screen.getByRole('menuitemradio', { - name: /^OpenCode · 默认模型.*sonnet-5$/u - }) - await waitFor(() => expect(selectedOpenCodeModel).toHaveFocus()) - expect(selectedOpenCodeModel).toHaveAttribute('aria-checked', 'true') - expect(selectedOpenCodeModel).toHaveAttribute('tabindex', '0') + const selectedRuntimeButton = await screen.findByRole("button", { + name: /OpenCode · 默认模型/u, + }); + fireEvent.click(selectedRuntimeButton); + const selectedOpenCodeModel = screen.getByRole("menuitemradio", { + name: /^OpenCode · 默认模型.*sonnet-5$/u, + }); + await waitFor(() => expect(selectedOpenCodeModel).toHaveFocus()); + expect(selectedOpenCodeModel).toHaveAttribute("aria-checked", "true"); + expect(selectedOpenCodeModel).toHaveAttribute("tabindex", "0"); fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^Continue · 默认模型.*sonnet-5$/u - }) - ) + screen.getByRole("menuitemradio", { + name: /^Continue · 默认模型.*sonnet-5$/u, + }), + ); await waitFor(() => expect(api.agent.getStatus).toHaveBeenLastCalledWith({ - provider: 'continue', - profileId: modelProfileId - }) - ) - }) + provider: "continue", + profileId: modelProfileId, + }), + ); + }); - it('dismisses the Runtime menu on outside pointer and focus changes', async () => { - render() + it("dismisses the Runtime menu on outside pointer and focus changes", async () => { + render(); - const runtimeButton = await screen.findByRole('button', { - name: /sonnet-5/u - }) - const composer = screen.getByLabelText('向 GoodBuddy 提问') + const runtimeButton = await screen.findByRole("button", { + name: /sonnet-5/u, + }); + const composer = screen.getByLabelText("向 GoodBuddy 提问"); - fireEvent.click(runtimeButton) + fireEvent.click(runtimeButton); expect( - screen.getByRole('menu', { name: 'Runtime 和模型' }) - ).toBeInTheDocument() - fireEvent.pointerDown(composer) + screen.getByRole("menu", { name: "Runtime 和模型" }), + ).toBeInTheDocument(); + fireEvent.pointerDown(composer); expect( - screen.queryByRole('menu', { name: 'Runtime 和模型' }) - ).not.toBeInTheDocument() + screen.queryByRole("menu", { name: "Runtime 和模型" }), + ).not.toBeInTheDocument(); - fireEvent.click(runtimeButton) - const selectedModel = screen.getByRole('menuitemradio', { - name: /^默认模型.*sonnet-5$/u - }) - await waitFor(() => expect(selectedModel).toHaveFocus()) - fireEvent.keyDown(selectedModel, { key: 'Tab' }) - composer.focus() - expect(composer).toHaveFocus() + fireEvent.click(runtimeButton); + const selectedModel = screen.getByRole("menuitemradio", { + name: /^默认模型.*sonnet-5$/u, + }); + await waitFor(() => expect(selectedModel).toHaveFocus()); + fireEvent.keyDown(selectedModel, { key: "Tab" }); + composer.focus(); + expect(composer).toHaveFocus(); await waitFor(() => expect( - screen.queryByRole('menu', { name: 'Runtime 和模型' }) - ).not.toBeInTheDocument() - ) - }) + screen.queryByRole("menu", { name: "Runtime 和模型" }), + ).not.toBeInTheDocument(), + ); + }); - it('labels explicitly configured Runtime-owned model sources', async () => { - const settings = await api.settings.getRuntime() + it("labels explicitly configured Runtime-owned model sources", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'model', - opencodeModelSource: { kind: 'platform' }, - continueModelSource: { kind: 'platform' } - }) - render() + provider: "model", + opencodeModelSource: { kind: "platform" }, + continueModelSource: { kind: "platform" }, + }); + render(); - fireEvent.click( - await screen.findByRole('button', { name: /sonnet-5/u }) - ) + fireEvent.click(await screen.findByRole("button", { name: /sonnet-5/u })); expect( - screen.getByRole('menuitemradio', { - name: /^OpenCode · 自身配置.*使用 OpenCode 自身配置$/u - }) - ).toBeInTheDocument() - const continueChoice = screen.getByRole('menuitemradio', { - name: /^Continue · 自身配置.*使用 Continue 自身配置$/u - }) - fireEvent.click(continueChoice) + screen.getByRole("menuitemradio", { + name: /^OpenCode · 自身配置.*使用 OpenCode 自身配置$/u, + }), + ).toBeInTheDocument(); + const continueChoice = screen.getByRole("menuitemradio", { + name: /^Continue · 自身配置.*使用 Continue 自身配置$/u, + }); + fireEvent.click(continueChoice); await waitFor(() => expect(api.agent.getStatus).toHaveBeenLastCalledWith({ - provider: 'continue' - }) - ) - }) + provider: "continue", + }), + ); + }); - it('persists metadata-only retrieval and Runtime changes without rewriting messages', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000451' - const existingMessageId = - '00000000-0000-4000-8000-000000000452' - const libraryId = '11111111-1111-4111-8111-111111111111' - const secondProfileId = - '00000000-0000-4000-8000-000000000453' - const settings = await api.settings.getRuntime() + it("persists metadata-only retrieval and Runtime changes without rewriting messages", async () => { + const conversationId = "00000000-0000-4000-8000-000000000451"; + const existingMessageId = "00000000-0000-4000-8000-000000000452"; + const libraryId = "11111111-1111-4111-8111-111111111111"; + const secondProfileId = "00000000-0000-4000-8000-000000000453"; + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, modelProfiles: [ ...settings.modelProfiles, { id: secondProfileId, - name: '仅元数据模型', - baseUrl: 'http://127.0.0.1:11434/v1', - modelName: 'qwen3', - protocol: 'openai-chat-completions', - authentication: 'none', - imageGenerationQuality: 'auto', + name: "仅元数据模型", + baseUrl: "http://127.0.0.1:11434/v1", + modelName: "qwen3", + protocol: "openai-chat-completions", + authentication: "none", + imageGenerationQuality: "auto", apiKeyConfigured: false, - credentialSource: 'none' - } - ] - }) + credentialSource: "none", + }, + ], + }); vi.mocked(api.knowledge.getSnapshot).mockResolvedValueOnce({ libraries: [ { id: libraryId, - name: '产品知识', - description: '', - storageMode: 'managed', + name: "产品知识", + description: "", + storageMode: "managed", graphEnabled: false, - graphStrategy: 'rules', + graphStrategy: "rules", sourceCount: 1, documentCount: 1, - indexedDocumentCount: 1 - } + indexedDocumentCount: 1, + }, ], sources: [], documents: [], graphNodes: [], graphRelations: [], - evidence: [] - }) + evidence: [], + }); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: conversationId, projectId, runtimeSelection: { - provider: 'model', - profileId: modelProfileId + provider: "model", + profileId: modelProfileId, }, knowledgeLibraryIds: [libraryId], - knowledgeRetrievalMode: 'auto', - title: '元数据会话', + knowledgeRetrievalMode: "auto", + title: "元数据会话", updatedAt: 1_775_000_000_000, messages: [ { id: existingMessageId, - role: 'assistant', - content: '现有消息不应重写', + role: "assistant", + content: "现有消息不应重写", createdAt: 1_775_000_000_000, - state: 'complete' - } - ] - } - ]) - vi.mocked(api.agent.getStatus).mockImplementation( - async (selection) => ({ - id: 'model', - label: - selection?.provider === 'model' && - selection.profileId === secondProfileId - ? 'qwen3' - : 'sonnet-5', - available: true, - supportsToolExecution: true, - detail: 'Ready' - }) - ) - render() + state: "complete", + }, + ], + }, + ]); + vi.mocked(api.agent.getStatus).mockImplementation(async (selection) => ({ + id: "model", + label: + selection?.provider === "model" && + selection.profileId === secondProfileId + ? "qwen3" + : "sonnet-5", + available: true, + supportsToolExecution: true, + detail: "Ready", + })); + render(); - const knowledgeScope = await screen.findByRole('button', { - name: '选择知识库，本次已启用 1 个' - }) - fireEvent.click(knowledgeScope) + const knowledgeScope = await screen.findByRole("button", { + name: "选择知识库，本次已启用 1 个", + }); + fireEvent.click(knowledgeScope); fireEvent.click( - within( - screen.getByRole('group', { name: '知识检索方式' }) - ).getByRole('button', { name: '每次先检索' }) - ) + within(screen.getByRole("group", { name: "知识检索方式" })).getByRole( + "button", + { name: "每次先检索" }, + ), + ); await waitFor( () => expect(api.conversations.saveLocal).toHaveBeenCalledWith([ { header: expect.objectContaining({ id: conversationId, - knowledgeRetrievalMode: 'always' + knowledgeRetrievalMode: "always", }), - messages: [] - } + messages: [], + }, ]), - { timeout: 2_000 } - ) + { timeout: 2_000 }, + ); - vi.mocked(api.conversations.saveLocal).mockClear() + vi.mocked(api.conversations.saveLocal).mockClear(); + fireEvent.click(screen.getByRole("button", { name: /sonnet-5/u })); fireEvent.click( - screen.getByRole('button', { name: /sonnet-5/u }) - ) - fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^仅元数据模型.*qwen3$/u - }) - ) + screen.getByRole("menuitemradio", { + name: /^仅元数据模型.*qwen3$/u, + }), + ); await waitFor( () => expect(api.conversations.saveLocal).toHaveBeenCalledWith([ @@ -8811,364 +8458,347 @@ describe('App', () => { header: expect.objectContaining({ id: conversationId, runtimeSelection: { - provider: 'model', - profileId: secondProfileId + provider: "model", + profileId: secondProfileId, }, - knowledgeRetrievalMode: 'always' + knowledgeRetrievalMode: "always", }), - messages: [] - } + messages: [], + }, ]), - { timeout: 2_000 } - ) + { timeout: 2_000 }, + ); const savedMessages = vi .mocked(api.conversations.saveLocal) .mock.calls.flatMap(([batch]) => - batch.flatMap((entry) => entry.messages) - ) + batch.flatMap((entry) => entry.messages), + ); expect( - savedMessages.some((message) => message.id === existingMessageId) - ).toBe(false) - }) + savedMessages.some((message) => message.id === existingMessageId), + ).toBe(false); + }); - it('migrates legacy startup conversations with replace when SQLite has no local conversation', async () => { - const settings = await api.settings.getRuntime() + it("migrates legacy startup conversations with replace when SQLite has no local conversation", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, modelProfiles: settings.modelProfiles.map((profile) => ({ ...profile, - contextWindowTokens: undefined + contextWindowTokens: undefined, })), contextCompression: { enabled: true, triggerTokens: 12_000, recentRawTokens: 4_000, - modelSource: { kind: 'current' }, - summaryPrompt: 'Preserve important facts.' - } - }) + modelSource: { kind: "current" }, + summaryPrompt: "Preserve important facts.", + }, + }); const normalizedContextMetrics = { runtimeSelectionKey: `model:${settings.defaultModelProfileId}`, contextTokens: 9_000, - source: 'provider' as const, - basis: 'model-call' as const - } + source: "provider" as const, + basis: "model-call" as const, + }; const legacyConversation = { - id: '00000000-0000-4000-8000-000000000461', + id: "00000000-0000-4000-8000-000000000461", runtimeSelection: { - provider: 'model' as const, - profileId: settings.defaultModelProfileId + provider: "model" as const, + profileId: settings.defaultModelProfileId, }, contextMetrics: { ...normalizedContextMetrics, effectiveTriggerTokens: 20_000, contextWindowTokens: 32_000, - compressionEnabled: true + compressionEnabled: true, }, - title: '待迁移旧会话', + title: "待迁移旧会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000462', - role: 'assistant' as const, - content: '旧版浏览器存储消息', + id: "00000000-0000-4000-8000-000000000462", + role: "assistant" as const, + content: "旧版浏览器存储消息", createdAt: 1_775_000_000_000, - state: 'complete' as const - } - ] - } + state: "complete" as const, + }, + ], + }; localStorage.setItem( - 'goodbuddy.conversations.v1', - JSON.stringify([legacyConversation]) - ) - render() + "goodbuddy.conversations.v1", + JSON.stringify([legacyConversation]), + ); + render(); + expect(await screen.findByText("旧版浏览器存储消息")).toBeInTheDocument(); expect( - await screen.findByText('旧版浏览器存储消息') - ).toBeInTheDocument() - expect( - screen.getByText('本次调用 9.0K · 压缩线 12.0K') - ).toBeInTheDocument() + screen.getByText("本次调用 9.0K · 压缩线 12.0K"), + ).toBeInTheDocument(); await waitFor(() => expect(api.conversations.replace).toHaveBeenCalledWith([ expect.objectContaining({ ...legacyConversation, contextMetrics: normalizedContextMetrics, - projectId - }) - ]) - ) - expect(api.conversations.saveLocal).not.toHaveBeenCalled() - }) + projectId, + }), + ]), + ); + expect(api.conversations.saveLocal).not.toHaveBeenCalled(); + }); - it('does not let remote rows displace legacy local conversations during migration', async () => { + it("does not let remote rows displace legacy local conversations during migration", async () => { const legacyConversations = [0, 1].map((index) => ({ - id: `00000000-0000-4000-8000-${String(470 + index).padStart(12, '0')}`, + id: `00000000-0000-4000-8000-${String(470 + index).padStart(12, "0")}`, title: `待迁移本地会话 ${index}`, updatedAt: 1_775_000_000_000 + index, messages: [ { - id: `00000000-0000-4000-8001-${String(470 + index).padStart(12, '0')}`, - role: 'assistant' as const, + id: `00000000-0000-4000-8001-${String(470 + index).padStart(12, "0")}`, + role: "assistant" as const, content: `待迁移消息 ${index}`, createdAt: 1_775_000_000_000 + index, - state: 'complete' as const - } - ] - })) + state: "complete" as const, + }, + ], + })); localStorage.setItem( - 'goodbuddy.conversations.v1', - JSON.stringify(legacyConversations) - ) - const remoteProjectId = - '00000000-0000-4000-8000-000000000499' + "goodbuddy.conversations.v1", + JSON.stringify(legacyConversations), + ); + const remoteProjectId = "00000000-0000-4000-8000-000000000499"; vi.mocked(api.conversations.list).mockResolvedValueOnce( Array.from({ length: 100 }, (_, index) => ({ - id: `00000000-0000-4000-8002-${String(index).padStart(12, '0')}`, + id: `00000000-0000-4000-8002-${String(index).padStart(12, "0")}`, projectId: remoteProjectId, remote: { - channel: 'weixin' as const, + channel: "weixin" as const, accountDisplay: `远程联系人 ${index}`, - conversationType: 'direct' as const + conversationType: "direct" as const, }, title: `远程会话 ${index}`, updatedAt: 1_775_000_100_000 + index, - messages: [] - })) - ) - render() + messages: [], + })), + ); + render(); - await waitFor(() => - expect(api.conversations.saveLocal).toHaveBeenCalled() - ) - expect(api.conversations.replace).not.toHaveBeenCalled() + await waitFor(() => expect(api.conversations.saveLocal).toHaveBeenCalled()); + expect(api.conversations.replace).not.toHaveBeenCalled(); const migrated = - vi.mocked(api.conversations.saveLocal).mock.calls[0]?.[0] ?? [] + vi.mocked(api.conversations.saveLocal).mock.calls[0]?.[0] ?? []; expect(migrated.map((conversation) => conversation.header.id)).toEqual( expect.arrayContaining( - legacyConversations.map((conversation) => conversation.id) - ) - ) - }) + legacyConversations.map((conversation) => conversation.id), + ), + ); + }); - it('preserves a legacy Auto conversation without silently persisting a replacement', async () => { - const settings = await api.settings.getRuntime() + it("preserves a legacy Auto conversation without silently persisting a replacement", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - provider: 'auto', - opencodeBaseUrl: '', + provider: "auto", + opencodeBaseUrl: "", opencodeEmbedded: false, modelProfiles: settings.modelProfiles.map((profile) => ({ ...profile, - contextWindowTokens: 32_000 + contextWindowTokens: 32_000, })), contextCompression: { enabled: true, triggerTokens: 20_000, recentRawTokens: 4_000, - modelSource: { kind: 'current' }, - summaryPrompt: 'Preserve important facts.' - } - }) + modelSource: { kind: "current" }, + summaryPrompt: "Preserve important facts.", + }, + }); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000020', + id: "00000000-0000-4000-8000-000000000020", projectId, - runtimeSelection: { provider: 'auto' }, + runtimeSelection: { provider: "auto" }, contextMetrics: { - runtimeSelectionKey: 'auto:default', + runtimeSelectionKey: "auto:default", contextTokens: 9_000, - source: 'provider', - basis: 'model-call' + source: "provider", + basis: "model-call", }, - title: '旧自动对话', + title: "旧自动对话", updatedAt: 1, messages: [ { - id: '00000000-0000-4000-8000-000000000021', - role: 'assistant', - content: '旧消息', + id: "00000000-0000-4000-8000-000000000021", + role: "assistant", + content: "旧消息", createdAt: 1, - state: 'complete' - } - ] - } - ]) - render() + state: "complete", + }, + ], + }, + ]); + render(); expect( - await screen.findByRole('button', { name: /自动.*sonnet-5/u }) - ).toBeInTheDocument() - expect( - screen.queryByText(/本次调用 9\.0K/u) - ).not.toBeInTheDocument() - expect(screen.queryByText(/压缩线/u)).not.toBeInTheDocument() - expect(api.conversations.replace).not.toHaveBeenCalled() - expect(api.conversations.saveLocal).not.toHaveBeenCalled() + await screen.findByRole("button", { name: /自动.*sonnet-5/u }), + ).toBeInTheDocument(); + expect(screen.queryByText(/本次调用 9\.0K/u)).not.toBeInTheDocument(); + expect(screen.queryByText(/压缩线/u)).not.toBeInTheDocument(); + expect(api.conversations.replace).not.toHaveBeenCalled(); + expect(api.conversations.saveLocal).not.toHaveBeenCalled(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '刷新自动 Runtime 用量' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "刷新自动 Runtime 用量" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'context-metrics', + type: "context-metrics", contextTokens: 9_000, effectiveTriggerTokens: 20_000, compressionEnabled: false, - source: 'provider' - }) - }) - expect(screen.getByText(/本次调用 9\.0K/u)).toBeInTheDocument() - expect(screen.queryByText(/压缩线/u)).not.toBeInTheDocument() + source: "provider", + }); + }); + expect(screen.getByText(/本次调用 9\.0K/u)).toBeInTheDocument(); + expect(screen.queryByText(/压缩线/u)).not.toBeInTheDocument(); const currentRuntimeSelectionKey = - settings.opencodeModelSource.kind === 'profile' + settings.opencodeModelSource.kind === "profile" ? `opencode:${settings.opencodeModelSource.profileId}` - : 'opencode:platform' + : "opencode:platform"; await waitFor(() => expect(api.conversations.saveLocal).toHaveBeenCalledWith([ expect.objectContaining({ header: expect.objectContaining({ contextMetrics: expect.objectContaining({ - runtimeSelectionKey: currentRuntimeSelectionKey - }) - }) - }) - ]) - ) - }) + runtimeSelectionKey: currentRuntimeSelectionKey, + }), + }), + }), + ]), + ); + }); - it('keeps a removed model selection visible until the user replaces it', async () => { - const removedProfileId = - '00000000-0000-4000-8000-000000000099' + it("keeps a removed model selection visible until the user replaces it", async () => { + const removedProfileId = "00000000-0000-4000-8000-000000000099"; vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000022', + id: "00000000-0000-4000-8000-000000000022", projectId, runtimeSelection: { - provider: 'model', - profileId: removedProfileId + provider: "model", + profileId: removedProfileId, }, - title: '旧模型对话', + title: "旧模型对话", updatedAt: 1, messages: [ { - id: '00000000-0000-4000-8000-000000000023', - role: 'assistant', - content: '旧消息', + id: "00000000-0000-4000-8000-000000000023", + role: "assistant", + content: "旧消息", createdAt: 1, - state: 'complete' - } - ] - } - ]) - render() + state: "complete", + }, + ], + }, + ]); + render(); + expect(await screen.findByText("旧消息")).toBeInTheDocument(); expect( - await screen.findByText('旧消息') - ).toBeInTheDocument() - expect( - screen.getByRole('button', { name: /模型配置不可用/u }) - ).toBeInTheDocument() - expect(api.conversations.replace).not.toHaveBeenCalled() - expect(api.conversations.saveLocal).not.toHaveBeenCalled() - }) + screen.getByRole("button", { name: /模型配置不可用/u }), + ).toBeInTheDocument(); + expect(api.conversations.replace).not.toHaveBeenCalled(); + expect(api.conversations.saveLocal).not.toHaveBeenCalled(); + }); - it('keeps model Runtime selection scoped to its conversation', async () => { - const secondProfileId = - '00000000-0000-4000-8000-000000000002' - const settings = await api.settings.getRuntime() + it("keeps model Runtime selection scoped to its conversation", async () => { + const secondProfileId = "00000000-0000-4000-8000-000000000002"; + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, modelProfiles: [ ...settings.modelProfiles, { id: secondProfileId, - name: '第二模型', - baseUrl: 'http://127.0.0.1:11434/v1', - modelName: 'qwen3', - protocol: 'openai-chat-completions', - authentication: 'none', - imageGenerationQuality: 'auto', + name: "第二模型", + baseUrl: "http://127.0.0.1:11434/v1", + modelName: "qwen3", + protocol: "openai-chat-completions", + authentication: "none", + imageGenerationQuality: "auto", apiKeyConfigured: false, - credentialSource: 'none' - } - ] - }) - vi.mocked(api.agent.getStatus).mockImplementation( - async (selection) => ({ - id: 'model', - label: - selection?.provider === 'model' && - selection.profileId === secondProfileId - ? 'qwen3' - : 'sonnet-5', - available: true, - supportsToolExecution: true, - detail: 'Ready' - }) - ) - render() + credentialSource: "none", + }, + ], + }); + vi.mocked(api.agent.getStatus).mockImplementation(async (selection) => ({ + id: "model", + label: + selection?.provider === "model" && + selection.profileId === secondProfileId + ? "qwen3" + : "sonnet-5", + available: true, + supportsToolExecution: true, + detail: "Ready", + })); + render(); + fireEvent.click(await screen.findByRole("button", { name: /sonnet-5/u })); fireEvent.click( - await screen.findByRole('button', { name: /sonnet-5/u }) - ) - fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^第二模型.*qwen3$/u - }) - ) + screen.getByRole("menuitemradio", { + name: /^第二模型.*qwen3$/u, + }), + ); expect( - await screen.findByRole('button', { name: /第二模型/u }) - ).toBeInTheDocument() + await screen.findByRole("button", { name: /第二模型/u }), + ).toBeInTheDocument(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '第二模型对话' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "第二模型对话" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; expect(request?.runtimeSelection).toEqual({ - provider: 'model', - profileId: secondProfileId - }) + provider: "model", + profileId: secondProfileId, + }); if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); - fireEvent.click( - screen.getByRole('button', { name: /新建对话/u }) - ) + fireEvent.click(screen.getByRole("button", { name: /新建对话/u })); expect( - await screen.findByRole('button', { name: /默认模型/u }) - ).toBeInTheDocument() + await screen.findByRole("button", { name: /默认模型/u }), + ).toBeInTheDocument(); const previousConversation = screen - .getAllByText('第二模型对话') - .map((element) => element.closest('button')) - .find((button) => button?.classList.contains('conversation-item')) + .getAllByText("第二模型对话") + .map((element) => element.closest("button")) + .find((button) => button?.classList.contains("conversation-item")); if (!previousConversation) { - throw new Error('Missing previous conversation') + throw new Error("Missing previous conversation"); } - fireEvent.click(previousConversation) + fireEvent.click(previousConversation); await waitFor(() => expect( screen - .getAllByRole('button', { name: /第二模型/u }) - .find((button) => button.classList.contains('model-button')) - ).toBeInTheDocument() - ) + .getAllByRole("button", { name: /第二模型/u }) + .find((button) => button.classList.contains("model-button")), + ).toBeInTheDocument(), + ); await waitFor( () => @@ -9176,510 +8806,524 @@ describe('App', () => { expect.arrayContaining([ expect.objectContaining({ header: expect.objectContaining({ - title: '第二模型对话', + title: "第二模型对话", runtimeSelection: { - provider: 'model', - profileId: secondProfileId - } + provider: "model", + profileId: secondProfileId, + }, }), - messages: expect.any(Array) - }) - ]) + messages: expect.any(Array), + }), + ]), ), - { timeout: 2_000 } - ) - }) + { timeout: 2_000 }, + ); + }); - it('ignores a stale picker status after rapidly changing conversations', async () => { - const secondProfileId = - '00000000-0000-4000-8000-000000000002' - const thirdProfileId = - '00000000-0000-4000-8000-000000000003' - const settings = await api.settings.getRuntime() + it("ignores a stale picker status after rapidly changing conversations", async () => { + const secondProfileId = "00000000-0000-4000-8000-000000000002"; + const thirdProfileId = "00000000-0000-4000-8000-000000000003"; + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, modelProfiles: [ ...settings.modelProfiles, { id: secondProfileId, - name: '第二模型', - baseUrl: 'http://127.0.0.1:11434/v1', - modelName: 'qwen3', - protocol: 'openai-chat-completions', - authentication: 'none', - imageGenerationQuality: 'auto', + name: "第二模型", + baseUrl: "http://127.0.0.1:11434/v1", + modelName: "qwen3", + protocol: "openai-chat-completions", + authentication: "none", + imageGenerationQuality: "auto", apiKeyConfigured: false, - credentialSource: 'none' + credentialSource: "none", }, { id: thirdProfileId, - name: '第三模型', - baseUrl: 'http://127.0.0.1:11435/v1', - modelName: 'llama3', - protocol: 'openai-chat-completions', - authentication: 'none', - imageGenerationQuality: 'auto', + name: "第三模型", + baseUrl: "http://127.0.0.1:11435/v1", + modelName: "llama3", + protocol: "openai-chat-completions", + authentication: "none", + imageGenerationQuality: "auto", apiKeyConfigured: false, - credentialSource: 'none' - } - ] - }) + credentialSource: "none", + }, + ], + }); let resolveThird!: (status: { - id: 'model' - label: string - available: boolean - supportsToolExecution: boolean - detail: string - }) => void + id: "model"; + label: string; + available: boolean; + supportsToolExecution: boolean; + detail: string; + }) => void; const thirdStatus = new Promise<{ - id: 'model' - label: string - available: boolean - supportsToolExecution: boolean - detail: string + id: "model"; + label: string; + available: boolean; + supportsToolExecution: boolean; + detail: string; }>((resolve) => { - resolveThird = resolve - }) + resolveThird = resolve; + }); vi.mocked(api.agent.getStatus).mockImplementation(async (selection) => { if ( - selection?.provider === 'model' && + selection?.provider === "model" && selection.profileId === thirdProfileId ) { - return thirdStatus + return thirdStatus; } return { - id: 'model', + id: "model", label: - selection?.provider === 'model' && + selection?.provider === "model" && selection.profileId === secondProfileId - ? 'qwen3' - : 'sonnet-5', + ? "qwen3" + : "sonnet-5", available: true, supportsToolExecution: true, - detail: 'Ready' - } - }) - render() + detail: "Ready", + }; + }); + render(); + fireEvent.click(await screen.findByRole("button", { name: /sonnet-5/u })); fireEvent.click( - await screen.findByRole('button', { name: /sonnet-5/u }) - ) - fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^第二模型.*qwen3$/u - }) - ) - await screen.findByRole('button', { name: /第二模型/u }) - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '保留第二模型会话' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + screen.getByRole("menuitemradio", { + name: /^第二模型.*qwen3$/u, + }), + ); + await screen.findByRole("button", { name: /第二模型/u }); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "保留第二模型会话" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { - agentListener?.({ requestId: request.requestId, type: 'done' }) - }) + agentListener?.({ requestId: request.requestId, type: "done" }); + }); const secondModelButton = screen - .getAllByRole('button', { name: /第二模型/u }) - .find((button) => button.classList.contains('model-button')) + .getAllByRole("button", { name: /第二模型/u }) + .find((button) => button.classList.contains("model-button")); if (!secondModelButton) { - throw new Error('Missing second model picker') + throw new Error("Missing second model picker"); } - fireEvent.click(secondModelButton) + fireEvent.click(secondModelButton); fireEvent.click( - screen.getByRole('menuitemradio', { - name: /^第三模型.*llama3$/u - }) - ) + screen.getByRole("menuitemradio", { + name: /^第三模型.*llama3$/u, + }), + ); await waitFor(() => expect(api.agent.getStatus).toHaveBeenCalledWith({ - provider: 'model', - profileId: thirdProfileId - }) - ) - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '状态未完成时不能发送' } - }) - expect(screen.getByLabelText('发送')).toBeDisabled() - fireEvent.click(screen.getByRole('button', { name: /新建对话/u })) - await screen.findByRole('button', { name: /默认模型/u }) + provider: "model", + profileId: thirdProfileId, + }), + ); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "状态未完成时不能发送" }, + }); + expect(screen.getByLabelText("发送")).toBeDisabled(); + fireEvent.click(screen.getByRole("button", { name: /新建对话/u })); + await screen.findByRole("button", { name: /默认模型/u }); await act(async () => { resolveThird({ - id: 'model', - label: 'llama3', + id: "model", + label: "llama3", available: true, supportsToolExecution: true, - detail: 'Ready' - }) - await thirdStatus - }) + detail: "Ready", + }); + await thirdStatus; + }); expect( - screen.getByRole('button', { name: /默认模型/u }) - ).toBeInTheDocument() + screen.getByRole("button", { name: /默认模型/u }), + ).toBeInTheDocument(); expect( - screen.queryByRole('button', { name: /第三模型/u }) - ).not.toBeInTheDocument() - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '新对话仍可发送' } - }) - await waitFor(() => expect(screen.getByLabelText('发送')).toBeEnabled()) - }) + screen.queryByRole("button", { name: /第三模型/u }), + ).not.toBeInTheDocument(); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "新对话仍可发送" }, + }); + await waitFor(() => expect(screen.getByLabelText("发送")).toBeEnabled()); + }); - it('opens project creation as an unobscured dialog', async () => { - render() + it.each(["resolved", "rejected", "returned"] as const)("ignores a %s deferred picker switch after same-runtime conversation navigation", async (outcome) => { + const settings = await api.settings.getRuntime(); + const selection = { provider: "model" as const, profileId: settings.defaultModelProfileId! }; + const rows = [1, 2].map((index) => ({ + id: `00000000-0000-4000-8000-00000000078${index}`, + projectId, + runtimeSelection: selection, + title: `Picker conversation ${index}`, + updatedAt: 1_775_000_000_000 - index, + messages: [], + })); + vi.mocked(api.conversations.list).mockResolvedValueOnce(rows); + render(); + const picker = await screen.findByRole("button", { name: /sonnet-5/u }); + const pending = deferred>>(); + vi.mocked(api.agent.getStatus).mockReturnValueOnce(pending.promise); + fireEvent.click(picker); + fireEvent.click(screen.getByRole("menuitemradio", { name: /^OpenCode/u })); + fireEvent.click(screen.getByText(rows[1]!.title).closest("button")!); + if (outcome === "returned") { + fireEvent.click(screen.getByText(rows[0]!.title).closest("button")!); + } + const activeRow = rows[outcome === "returned" ? 0 : 1]!; + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { target: { value: "still usable" } }); + await waitFor(() => expect(screen.getByLabelText("发送")).toBeEnabled()); + await act(async () => { + if (outcome === "rejected") { + pending.reject(new Error("obsolete picker failure")); + } else { + pending.resolve({ id: "opencode", label: "OpenCode", available: true, supportsToolExecution: true, detail: "Ready" }); + } + await pending.promise.catch(() => undefined); + }); + expect(screen.queryByText("obsolete picker failure")).not.toBeInTheDocument(); + expect(screen.getByRole("button", { name: /sonnet-5/u })).toBeInTheDocument(); + expect(screen.getByLabelText("发送")).toBeEnabled(); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + expect(run.mock.calls[0]?.[0]).toMatchObject({ conversationId: activeRow.id, runtimeSelection: selection }); + }); - const newProjectButton = await screen.findByLabelText('新建项目') - fireEvent.click(newProjectButton) - let dialog = screen.getByRole('dialog', { name: '新建项目' }) - expect(dialog).toHaveClass('project-create-card') - expect(within(dialog).getByRole('button', { name: '创建' })) - .toBeDisabled() - expect(within(dialog).getByLabelText('名称')).toHaveFocus() + it("opens project creation as an unobscured dialog", async () => { + render(); - fireEvent.keyDown(document, { key: 'Escape' }) + const newProjectButton = await screen.findByLabelText("新建项目"); + fireEvent.click(newProjectButton); + let dialog = screen.getByRole("dialog", { name: "新建项目" }); + expect(dialog).toHaveClass("project-create-card"); + expect(within(dialog).getByRole("button", { name: "创建" })).toBeDisabled(); + expect(within(dialog).getByLabelText("名称")).toHaveFocus(); + + fireEvent.keyDown(document, { key: "Escape" }); expect( - screen.queryByRole('dialog', { name: '新建项目' }) - ).not.toBeInTheDocument() - expect(newProjectButton).toHaveFocus() + screen.queryByRole("dialog", { name: "新建项目" }), + ).not.toBeInTheDocument(); + expect(newProjectButton).toHaveFocus(); - fireEvent.click(newProjectButton) - dialog = screen.getByRole('dialog', { name: '新建项目' }) + fireEvent.click(newProjectButton); + dialog = screen.getByRole("dialog", { name: "新建项目" }); - fireEvent.change(within(dialog).getByLabelText('名称'), { - target: { value: '新项目' } - }) - fireEvent.click(within(dialog).getByRole('button', { name: '创建' })) + fireEvent.change(within(dialog).getByLabelText("名称"), { + target: { value: "新项目" }, + }); + fireEvent.click(within(dialog).getByRole("button", { name: "创建" })); await waitFor(() => expect(api.projects.create).toHaveBeenCalledWith( expect.objectContaining({ - name: '新项目', - rootPath: '' - }) - ) - ) - }) + name: "新项目", + rootPath: "", + }), + ), + ); + }); - it('keeps project input when creation fails', async () => { + it("keeps project input when creation fails", async () => { vi.mocked(api.projects.create).mockRejectedValueOnce( - new Error('项目目录不可用') - ) - render() + new Error("项目目录不可用"), + ); + render(); - fireEvent.click(await screen.findByLabelText('新建项目')) - const dialog = screen.getByRole('dialog', { name: '新建项目' }) - const nameInput = within(dialog).getByLabelText('名称') + fireEvent.click(await screen.findByLabelText("新建项目")); + const dialog = screen.getByRole("dialog", { name: "新建项目" }); + const nameInput = within(dialog).getByLabelText("名称"); fireEvent.change(nameInput, { - target: { value: '保留的项目名称' } - }) - fireEvent.click( - within(dialog).getByRole('button', { name: '创建' }) - ) + target: { value: "保留的项目名称" }, + }); + fireEvent.click(within(dialog).getByRole("button", { name: "创建" })); - expect(await within(dialog).findByRole('alert')).toHaveTextContent( - '项目目录不可用' - ) - expect(nameInput).toHaveValue('保留的项目名称') + expect(await within(dialog).findByRole("alert")).toHaveTextContent( + "项目目录不可用", + ); + expect(nameInput).toHaveValue("保留的项目名称"); expect( - screen.getByRole('dialog', { name: '新建项目' }) - ).toBeInTheDocument() - }) + screen.getByRole("dialog", { name: "新建项目" }), + ).toBeInTheDocument(); + }); - it('edits and safely deletes the current project from project settings', async () => { + it("edits and safely deletes the current project from project settings", async () => { const hiddenRemoteProject = { ...project, - id: '00000000-0000-4000-8000-000000000119', - name: '隐藏远程项目', - rootPath: '/srv/hidden', - runtimeSelection: { provider: 'opencode' as const }, + id: "00000000-0000-4000-8000-000000000119", + name: "隐藏远程项目", + rootPath: "/srv/hidden", + runtimeSelection: { provider: "opencode" as const }, executionSpace: { - kind: 'ssh' as const, - hostId: '00000000-0000-4000-8000-000000000219', - remoteRootPath: '/srv/hidden' + kind: "ssh" as const, + hostId: "00000000-0000-4000-8000-000000000219", + remoteRootPath: "/srv/hidden", }, - builtInDefault: false - } + builtInDefault: false, + }; const secondProject = { ...project, - id: '00000000-0000-4000-8000-000000000102', - name: '第二项目', - rootPath: 'C:\\Second', + id: "00000000-0000-4000-8000-000000000102", + name: "第二项目", + rootPath: "C:\\Second", executionSpace: { - kind: 'local' as const, - rootPath: 'C:\\Second' - } - } + kind: "local" as const, + rootPath: "C:\\Second", + }, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ hiddenRemoteProject, project, - secondProject - ]) - render() + secondProject, + ]); + render(); - fireEvent.click(await screen.findByLabelText('项目设置')) - let dialog = screen.getByRole('dialog', { name: '项目设置' }) - expect(within(dialog).getByLabelText('名称')).toHaveValue( - project.name - ) - expect(within(dialog).getByLabelText('根目录')).toHaveValue( - project.rootPath - ) - expect( - within(dialog).getByLabelText('新对话默认 Runtime') - ).toHaveValue( + fireEvent.click(await screen.findByLabelText("项目设置")); + let dialog = screen.getByRole("dialog", { name: "项目设置" }); + expect(within(dialog).getByLabelText("名称")).toHaveValue(project.name); + expect(within(dialog).getByLabelText("根目录")).toHaveValue( + project.rootPath, + ); + expect(within(dialog).getByLabelText("新对话默认 Runtime")).toHaveValue( agentRuntimeSelectionKey({ - provider: 'model', - profileId: modelProfileId - }) - ) - fireEvent.change(within(dialog).getByLabelText('说明'), { - target: { value: '更新后的说明' } - }) - fireEvent.change( - within(dialog).getByLabelText('新对话默认 Runtime'), - { - target: { - value: agentRuntimeSelectionKey({ - provider: 'continue', - profileId: modelProfileId - }) - } - } - ) - fireEvent.click( - within(dialog).getByRole('button', { name: '保存项目' }) - ) + provider: "model", + profileId: modelProfileId, + }), + ); + fireEvent.change(within(dialog).getByLabelText("说明"), { + target: { value: "更新后的说明" }, + }); + fireEvent.change(within(dialog).getByLabelText("新对话默认 Runtime"), { + target: { + value: agentRuntimeSelectionKey({ + provider: "continue", + profileId: modelProfileId, + }), + }, + }); + fireEvent.click(within(dialog).getByRole("button", { name: "保存项目" })); await waitFor(() => expect(api.projects.update).toHaveBeenCalledWith( project.id, expect.objectContaining({ - description: '更新后的说明', + description: "更新后的说明", rootPath: project.rootPath, runtimeSelection: { - provider: 'continue', - profileId: modelProfileId - } - }) - ) - ) + provider: "continue", + profileId: modelProfileId, + }, + }), + ), + ); - fireEvent.click(screen.getByLabelText('项目设置')) - dialog = screen.getByRole('dialog', { name: '项目设置' }) - expect(dialog).toHaveTextContent('不会删除磁盘上的项目目录或文件') - fireEvent.click( - within(dialog).getByRole('button', { name: '删除项目' }) - ) + fireEvent.click(screen.getByLabelText("项目设置")); + dialog = screen.getByRole("dialog", { name: "项目设置" }); + expect(dialog).toHaveTextContent("不会删除磁盘上的项目目录或文件"); + fireEvent.click(within(dialog).getByRole("button", { name: "删除项目" })); const confirmation = within(dialog).getByLabelText( - `输入“${project.name}”确认删除` - ) - const deleteButton = within(dialog).getByRole('button', { - name: '永久删除项目' - }) - expect(deleteButton).toBeDisabled() + `输入“${project.name}”确认删除`, + ); + const deleteButton = within(dialog).getByRole("button", { + name: "永久删除项目", + }); + expect(deleteButton).toBeDisabled(); fireEvent.change(confirmation, { - target: { value: project.name } - }) - expect(deleteButton).toBeEnabled() - fireEvent.click(deleteButton) + target: { value: project.name }, + }); + expect(deleteButton).toBeEnabled(); + fireEvent.click(deleteButton); await waitFor(() => expect(api.projects.delete).toHaveBeenCalledWith( project.id, - project.name - ) - ) - expect(screen.getByRole('button', { name: '当前项目' })).toHaveTextContent( - secondProject.name - ) - }) + project.name, + ), + ); + expect(screen.getByRole("button", { name: "当前项目" })).toHaveTextContent( + secondProject.name, + ); + }); - it('uses the project default Runtime for new conversations', async () => { + it("uses the project default Runtime for new conversations", async () => { vi.mocked(api.projects.list).mockResolvedValueOnce([ { ...project, runtimeSelection: { - provider: 'opencode', - profileId: modelProfileId - } - } - ]) + provider: "opencode", + profileId: modelProfileId, + }, + }, + ]); vi.mocked(api.conversations.list).mockResolvedValueOnce([ { - id: '00000000-0000-4000-8000-000000000220', + id: "00000000-0000-4000-8000-000000000220", projectId, runtimeSelection: { - provider: 'model', - profileId: modelProfileId + provider: "model", + profileId: modelProfileId, }, - title: '已有对话', + title: "已有对话", updatedAt: 1, - messages: [] - } - ]) - render() + messages: [], + }, + ]); + render(); - await screen.findAllByText('已有对话') - fireEvent.click( - screen.getByRole('button', { name: /新建对话/u }) - ) + await screen.findAllByText("已有对话"); + fireEvent.click(screen.getByRole("button", { name: /新建对话/u })); await waitFor(() => expect(api.agent.getStatus).toHaveBeenLastCalledWith({ - provider: 'opencode', - profileId: modelProfileId - }) - ) + provider: "opencode", + profileId: modelProfileId, + }), + ); expect( - screen.getByRole('button', { - name: /OpenCode · 默认模型/u - }) - ).toBeInTheDocument() - }) + screen.getByRole("button", { + name: /OpenCode · 默认模型/u, + }), + ).toBeInTheDocument(); + }); - it('uses a message icon for conversation navigation', async () => { - render() + it("uses a message icon for conversation navigation", async () => { + render(); - const conversationNavigation = await screen.findByRole('button', { - name: '对话' - }) + const conversationNavigation = await screen.findByRole("button", { + name: "对话", + }); expect( - conversationNavigation.querySelector('.lucide-message-square') - ).not.toBeNull() - expect( - conversationNavigation.querySelector('.lucide-history') - ).toBeNull() - }) + conversationNavigation.querySelector(".lucide-message-square"), + ).not.toBeNull(); + expect(conversationNavigation.querySelector(".lucide-history")).toBeNull(); + }); - it('marks an image model and renders its generated artifact', async () => { + it("marks an image model and renders its generated artifact", async () => { const anchorClick = vi - .spyOn(HTMLAnchorElement.prototype, 'click') - .mockImplementation(() => {}) + .spyOn(HTMLAnchorElement.prototype, "click") + .mockImplementation(() => {}); vi.mocked(api.agent.getStatus).mockResolvedValueOnce({ - id: 'model', - label: 'gpt-image-2', + id: "model", + label: "gpt-image-2", available: true, supportsToolExecution: false, - detail: 'OpenAI Images Generations', - capability: 'image-generation' - }) - render() + detail: "OpenAI Images Generations", + capability: "image-generation", + }); + render(); - expect((await screen.findAllByText('生图')).length).toBeGreaterThan(0) - expect(screen.getByLabelText('向 GoodBuddy 提问')).toHaveAttribute( - 'placeholder', - '描述你想生成的图片…\nEnter 发送，Shift+Enter 换行，Ctrl+V 粘贴图片或文本' - ) - await waitFor(() => - expect(api.artifacts.list).toHaveBeenCalled() - ) + expect((await screen.findAllByText("生图")).length).toBeGreaterThan(0); + expect(screen.getByLabelText("向 GoodBuddy 提问")).toHaveAttribute( + "placeholder", + "描述你想生成的图片…\nEnter 发送，Shift+Enter 换行，Ctrl+V 粘贴图片或文本", + ); + await waitFor(() => expect(api.artifacts.list).toHaveBeenCalled()); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '生成一只蓝色的猫' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "生成一只蓝色的猫" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } - const artifactId = '00000000-0000-4000-8000-000000000301' - vi.mocked(api.artifacts.get).mockResolvedValueOnce( - { - id: artifactId, - projectId, - taskId: request.requestId, - kind: 'image', - title: '生成一只蓝色的猫', - mimeType: 'image/png', - content: - 'data:image/png;base64,iVBORw0KGgoAAAAAAAAAAAAA', - byteSize: 42, - createdAt: '2026-08-01T00:00:00.000Z', - updatedAt: '2026-08-01T00:00:00.000Z' - } - ) + const artifactId = "00000000-0000-4000-8000-000000000301"; + vi.mocked(api.artifacts.get).mockResolvedValueOnce({ + id: artifactId, + projectId, + taskId: request.requestId, + kind: "image", + title: "生成一只蓝色的猫", + mimeType: "image/png", + content: "data:image/png;base64,iVBORw0KGgoAAAAAAAAAAAAA", + byteSize: 42, + createdAt: "2026-08-01T00:00:00.000Z", + updatedAt: "2026-08-01T00:00:00.000Z", + }); act(() => { agentListener?.({ requestId: request.requestId, - type: 'artifact', + type: "artifact", artifactId, - kind: 'image', - title: '生成一只蓝色的猫' - }) - }) + kind: "image", + title: "生成一只蓝色的猫", + }); + }); expect( - await screen.findByRole('img', { name: '生成一只蓝色的猫' }) - ).toHaveAttribute('src', expect.stringMatching(/^data:image\/png/u)) + await screen.findByRole("img", { name: "生成一只蓝色的猫" }), + ).toHaveAttribute("src", expect.stringMatching(/^data:image\/png/u)); fireEvent.click( - screen.getByRole('button', { - name: '下载图片 生成一只蓝色的猫' - }) - ) - expect(anchorClick).toHaveBeenCalledOnce() + screen.getByRole("button", { + name: "下载图片 生成一只蓝色的猫", + }), + ); + expect(anchorClick).toHaveBeenCalledOnce(); fireEvent.click( - screen.getByRole('button', { - name: '查看图片 生成一只蓝色的猫' - }) - ) - const imageDialog = await screen.findByRole('dialog', { - name: '生成一只蓝色的猫' - }) + screen.getByRole("button", { + name: "查看图片 生成一只蓝色的猫", + }), + ); + const imageDialog = await screen.findByRole("dialog", { + name: "生成一只蓝色的猫", + }); expect( - within(imageDialog).getByRole('img', { - name: '生成一只蓝色的猫' - }) - ).toHaveAttribute('src', expect.stringMatching(/^data:image\/png/u)) + within(imageDialog).getByRole("img", { + name: "生成一只蓝色的猫", + }), + ).toHaveAttribute("src", expect.stringMatching(/^data:image\/png/u)); fireEvent.click( - within(imageDialog).getByRole('button', { name: '下载图片' }) - ) - expect(anchorClick).toHaveBeenCalledTimes(2) - fireEvent.keyDown(imageDialog, { key: 'Escape' }) + within(imageDialog).getByRole("button", { name: "下载图片" }), + ); + expect(anchorClick).toHaveBeenCalledTimes(2); + fireEvent.keyDown(imageDialog, { key: "Escape" }); expect( - screen.queryByRole('dialog', { name: '生成一只蓝色的猫' }) - ).not.toBeInTheDocument() - anchorClick.mockRestore() - }) + screen.queryByRole("dialog", { name: "生成一只蓝色的猫" }), + ).not.toBeInTheDocument(); + anchorClick.mockRestore(); + }); - it('can dispatch a request to the parallel expert team', async () => { - render() + it("can dispatch a request to the parallel expert team", async () => { + render(); - selectComposerOption('专家角色', '专家团队（并行）') - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '制定发布计划' } - }) - fireEvent.click(await screen.findByLabelText('发送')) + selectComposerOption("专家角色", "专家团队（并行）"); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "制定发布计划" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); await waitFor(() => expect(run).toHaveBeenCalledWith( expect.objectContaining({ teamMode: true, expertId: undefined, - prompt: '制定发布计划' - }) - ) - ) - }) + prompt: "制定发布计划", + }), + ), + ); + }); - it('requests smart routing only when enabled without an explicit expert', async () => { - const settings = await api.settings.getRuntime() + it("requests smart routing only when enabled without an explicit expert", async () => { + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - subagentSmartRoutingEnabled: true - }) - render() + subagentSmartRoutingEnabled: true, + }); + render(); - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '分析发布风险' } - }) - fireEvent.click(screen.getByLabelText('发送')) + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "分析发布风险" }, + }); + fireEvent.click(screen.getByLabelText("发送")); await waitFor(() => expect(run).toHaveBeenCalledWith( @@ -9687,1342 +9331,1295 @@ describe('App', () => { smartRouting: true, expertId: undefined, teamMode: false, - workMode: 'ask' - }) - ) - ) - }) + workMode: "ask", + }), + ), + ); + }); - it('gives an explicitly selected expert priority over smart routing', async () => { - const expertId = '00000000-0000-4000-8000-000000000501' - const settings = await api.settings.getRuntime() + it("gives an explicitly selected expert priority over smart routing", async () => { + const expertId = "00000000-0000-4000-8000-000000000501"; + const settings = await api.settings.getRuntime(); vi.mocked(api.settings.getRuntime).mockResolvedValueOnce({ ...settings, - subagentSmartRoutingEnabled: true - }) + subagentSmartRoutingEnabled: true, + }); vi.mocked(api.experts.list).mockResolvedValueOnce([ { id: expertId, - name: '发布专家', - description: '检查发布风险', - systemInstructions: 'Review release risks.', - routingKeywords: ['发布', '风险'], + name: "发布专家", + description: "检查发布风险", + systemInstructions: "Review release risks.", + routingKeywords: ["发布", "风险"], enabled: true, - createdAt: '2026-08-01T00:00:00.000Z', - updatedAt: '2026-08-01T00:00:00.000Z' - } - ]) - render() + createdAt: "2026-08-01T00:00:00.000Z", + updatedAt: "2026-08-01T00:00:00.000Z", + }, + ]); + render(); - await waitFor(() => expect(api.experts.list).toHaveBeenCalled()) - const expertMenu = openComposerMenu('专家角色') + await waitFor(() => expect(api.experts.list).toHaveBeenCalled()); + const expertMenu = openComposerMenu("专家角色"); fireEvent.click( - (await within(expertMenu).findByText('发布专家', { - selector: 'span' - })) - .closest('button')! - ) - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '检查发布方案' } - }) - fireEvent.click(screen.getByLabelText('发送')) + ( + await within(expertMenu).findByText("发布专家", { + selector: "span", + }) + ).closest("button")!, + ); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "检查发布方案" }, + }); + fireEvent.click(screen.getByLabelText("发送")); await waitFor(() => expect(run).toHaveBeenCalledWith( expect.objectContaining({ expertId, smartRouting: undefined, - teamMode: false - }) - ) - ) - }) + teamMode: false, + }), + ), + ); + }); - it('shows Subagent states and records child expert activity', async () => { - render() - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '分析复杂问题' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + it("shows Subagent states and records child expert activity", async () => { + render(); + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "分析复杂问题" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } const events = [ { - childTaskId: '00000000-0000-4000-8000-000000000601', - expertId: '00000000-0000-4000-8000-000000000701', - expertName: '研究专家', - routingMode: 'smart' as const, - state: 'queued' as const + childTaskId: "00000000-0000-4000-8000-000000000601", + expertId: "00000000-0000-4000-8000-000000000701", + expertName: "研究专家", + routingMode: "smart" as const, + state: "queued" as const, }, { - childTaskId: '00000000-0000-4000-8000-000000000602', - expertId: '00000000-0000-4000-8000-000000000702', - expertName: '代码专家', - routingMode: 'manual' as const, - state: 'running' as const + childTaskId: "00000000-0000-4000-8000-000000000602", + expertId: "00000000-0000-4000-8000-000000000702", + expertName: "代码专家", + routingMode: "manual" as const, + state: "running" as const, }, { - childTaskId: '00000000-0000-4000-8000-000000000603', - expertId: '00000000-0000-4000-8000-000000000703', - expertName: '安全专家', - routingMode: 'smart' as const, - state: 'failed' as const, - error: '无法读取必要上下文' + childTaskId: "00000000-0000-4000-8000-000000000603", + expertId: "00000000-0000-4000-8000-000000000703", + expertName: "安全专家", + routingMode: "smart" as const, + state: "failed" as const, + error: "无法读取必要上下文", }, { - childTaskId: '00000000-0000-4000-8000-000000000604', - expertId: '00000000-0000-4000-8000-000000000704', - expertName: '第四位专家', - routingMode: 'smart' as const, - state: 'completed' as const - } - ] + childTaskId: "00000000-0000-4000-8000-000000000604", + expertId: "00000000-0000-4000-8000-000000000704", + expertName: "第四位专家", + routingMode: "smart" as const, + state: "completed" as const, + }, + ]; act(() => { for (const event of events) { agentListener?.({ requestId: request.requestId, - type: 'subagent', - ...event - }) + type: "subagent", + ...event, + }); } - }) + }); - const statusRegion = await screen.findByLabelText('子代理状态') - expect(within(statusRegion).getByText('研究专家')).toBeInTheDocument() - expect(within(statusRegion).getByText('等待中')).toBeInTheDocument() - expect(within(statusRegion).getByText('代码专家')).toBeInTheDocument() - expect(within(statusRegion).getByText('进行中')).toBeInTheDocument() - expect(within(statusRegion).getByText('安全专家')).toBeInTheDocument() - expect(within(statusRegion).getByText('失败')).toBeInTheDocument() - const subagentCards = within(statusRegion).getAllByRole('group') + const statusRegion = await screen.findByLabelText("子代理状态"); + expect(within(statusRegion).getByText("研究专家")).toBeInTheDocument(); + expect(within(statusRegion).getByText("等待中")).toBeInTheDocument(); + expect(within(statusRegion).getByText("代码专家")).toBeInTheDocument(); + expect(within(statusRegion).getByText("进行中")).toBeInTheDocument(); + expect(within(statusRegion).getByText("安全专家")).toBeInTheDocument(); + expect(within(statusRegion).getByText("失败")).toBeInTheDocument(); + const subagentCards = within(statusRegion).getAllByRole("group"); fireEvent.click( - within(subagentCards[2]!).getByText('安全专家', { - selector: 'strong' - }) - ) - fireEvent(subagentCards[2]!, new Event('toggle')) + within(subagentCards[2]!).getByText("安全专家", { + selector: "strong", + }), + ); + fireEvent(subagentCards[2]!, new Event("toggle")); expect( - within(statusRegion).getByText('无法读取必要上下文') - ).toBeInTheDocument() - expect( - within(statusRegion).getByText('第四位专家') - ).toBeInTheDocument() + within(statusRegion).getByText("无法读取必要上下文"), + ).toBeInTheDocument(); + expect(within(statusRegion).getByText("第四位专家")).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'subagent', + type: "subagent", ...events[0]!, - state: 'completed', - output: '研究专家的独立结论' - }) + state: "completed", + output: "研究专家的独立结论", + }); agentListener?.({ requestId: request.requestId, - type: 'subagent', + type: "subagent", ...events[1]!, - state: 'cancelled', - reason: '父任务已停止', - output: '代码专家的部分结论' - }) - }) - expect(within(statusRegion).getAllByText('已完成')).toHaveLength(2) - expect(within(statusRegion).getByText('已取消')).toBeInTheDocument() - expect(within(statusRegion).getByText('父任务已停止')).toBeInTheDocument() - for (const [index, expertName] of [ - '研究专家', - '代码专家' - ].entries()) { + state: "cancelled", + reason: "父任务已停止", + output: "代码专家的部分结论", + }); + }); + expect(within(statusRegion).getAllByText("已完成")).toHaveLength(2); + expect(within(statusRegion).getByText("已取消")).toBeInTheDocument(); + expect(within(statusRegion).getByText("父任务已停止")).toBeInTheDocument(); + for (const [index, expertName] of ["研究专家", "代码专家"].entries()) { fireEvent.click( within(subagentCards[index]!).getByText(expertName, { - selector: 'strong' - }) - ) - fireEvent(subagentCards[index]!, new Event('toggle')) + selector: "strong", + }), + ); + fireEvent(subagentCards[index]!, new Event("toggle")); } expect( - within(statusRegion).getByText('研究专家的独立结论') - ).toBeInTheDocument() + within(statusRegion).getByText("研究专家的独立结论"), + ).toBeInTheDocument(); expect( - within(statusRegion).getByText('代码专家的部分结论') - ).toBeInTheDocument() + within(statusRegion).getByText("代码专家的部分结论"), + ).toBeInTheDocument(); await waitFor(() => { const persistedMessages = vi .mocked(api.conversations.saveLocal) .mock.calls.flatMap(([batch]) => - batch.flatMap((conversation) => conversation.messages) - ) + batch.flatMap((conversation) => conversation.messages), + ); expect( persistedMessages.some( (message) => - message.subagents?.[0]?.output === - '研究专家的独立结论' && - message.subagents?.[1]?.output === '代码专家的部分结论' - ) - ).toBe(true) - }) + message.subagents?.[0]?.output === "研究专家的独立结论" && + message.subagents?.[1]?.output === "代码专家的部分结论", + ), + ).toBe(true); + }); - fireEvent.click(screen.getByText('运行记录')) - expect(await screen.findAllByText('子专家')).toHaveLength(4) - expect(screen.getAllByText(/智能路由/u).length).toBeGreaterThan(0) - expect(screen.getAllByText(/指定专家/u).length).toBeGreaterThan(0) - expect( - screen.getByText(/研究专家的独立结论/u) - ).toBeInTheDocument() - expect( - screen.getByText(/代码专家的部分结论/u) - ).toBeInTheDocument() - }) + fireEvent.click(screen.getByText("运行记录")); + expect(await screen.findAllByText("子专家")).toHaveLength(4); + expect(screen.getAllByText(/智能路由/u).length).toBeGreaterThan(0); + expect(screen.getAllByText(/指定专家/u).length).toBeGreaterThan(0); + expect(screen.getByText(/研究专家的独立结论/u)).toBeInTheDocument(); + expect(screen.getByText(/代码专家的部分结论/u)).toBeInTheDocument(); + }); - it('replaces an incremental OpenCode Task tool with a native subagent card', async () => { - render() - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '检查架构' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + it("replaces an incremental OpenCode Task tool with a native subagent card", async () => { + render(); + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "检查架构" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'tool', - callId: 'call-task-1', - name: 'Review application architecture', - state: 'running', - summary: '远端 Runtime 工具：Review application architecture' - }) + type: "tool", + callId: "call-task-1", + name: "Review application architecture", + state: "running", + summary: "远端 Runtime 工具：Review application architecture", + }); agentListener?.({ requestId: request.requestId, - type: 'reasoning', - delta: '等待子代理结果' - }) - }) - expect(await screen.findByText('等待子代理结果')).toBeInTheDocument() + type: "reasoning", + delta: "等待子代理结果", + }); + }); + expect(await screen.findByText("等待子代理结果")).toBeInTheDocument(); act(() => { agentListener?.({ requestId: request.requestId, - type: 'subagent', - childTaskId: '00000000-0000-4000-8000-000000000611', - expertId: '00000000-0000-4000-8000-000000000711', - expertName: 'explorer', - routingMode: 'native', - runtimeCallId: 'call-task-1', - state: 'running', - reason: 'Review application architecture' - }) + type: "subagent", + childTaskId: "00000000-0000-4000-8000-000000000611", + expertId: "00000000-0000-4000-8000-000000000711", + expertName: "explorer", + routingMode: "native", + runtimeCallId: "call-task-1", + state: "running", + reason: "Review application architecture", + }); agentListener?.({ requestId: request.requestId, - type: 'text', - delta: '父 Agent 最终回复' - }) - }) + type: "text", + delta: "父 Agent 最终回复", + }); + }); - const region = await screen.findByLabelText('子代理状态') - expect(within(region).getByText('explorer')).toBeInTheDocument() - expect(within(region).getByText('OpenCode 原生')) - .toBeInTheDocument() + const region = await screen.findByLabelText("子代理状态"); + expect(within(region).getByText("explorer")).toBeInTheDocument(); + expect(within(region).getByText("OpenCode 原生")).toBeInTheDocument(); expect( - screen.queryByRole('region', { - name: '工具执行，共 1 项' - }) - ).not.toBeInTheDocument() + screen.queryByRole("region", { + name: "工具执行，共 1 项", + }), + ).not.toBeInTheDocument(); const assistantArticle = screen - .getByText('父 Agent 最终回复') - .closest('article') + .getByText("父 Agent 最终回复") + .closest("article"); const orderedBlocks = [ - ...assistantArticle!.querySelectorAll('.message-blocks > *') - ].map((element) => element.textContent) + ...assistantArticle!.querySelectorAll(".message-blocks > *"), + ].map((element) => element.textContent); expect(orderedBlocks).toEqual([ - expect.stringContaining('Review application architecture'), - expect.stringContaining('等待子代理结果'), - expect.stringContaining('父 Agent 最终回复') - ]) - fireEvent.click(screen.getByText('运行记录')) - expect(await screen.findAllByText('子专家')).toHaveLength(1) + expect.stringContaining("Review application architecture"), + expect.stringContaining("等待子代理结果"), + expect.stringContaining("父 Agent 最终回复"), + ]); + fireEvent.click(screen.getByText("运行记录")); + expect(await screen.findAllByText("子专家")).toHaveLength(1); expect( - screen.queryByText( - '远端 Runtime 工具：Review application architecture' - ) - ).not.toBeInTheDocument() - }) + screen.queryByText("远端 Runtime 工具：Review application architecture"), + ).not.toBeInTheDocument(); + }); - it('records direct-model child runs without adding Assistant Tasks', async () => { - render() - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '修复并验证聚焦变更' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + it("records direct-model child runs without adding Assistant Tasks", async () => { + render(); + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "修复并验证聚焦变更" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } - const childRunId = - '00000000-0000-4000-8000-000000000613' + const childRunId = "00000000-0000-4000-8000-000000000613"; act(() => { agentListener?.({ requestId: request.requestId, - type: 'subagent', + type: "subagent", childTaskId: childRunId, actor: { - kind: 'direct-model', - label: '编程 Subagent' + kind: "direct-model", + label: "编程 Subagent", }, - routingMode: 'native', - workMode: 'execute', - state: 'completed', - reason: '修复并验证聚焦变更', - output: '验证通过' - }) - }) + routingMode: "native", + workMode: "execute", + state: "completed", + reason: "修复并验证聚焦变更", + output: "验证通过", + }); + }); - const region = await screen.findByLabelText('子代理状态') + const region = await screen.findByLabelText("子代理状态"); expect( - within(region).getByText('编程 Subagent', { - selector: 'strong' - }) - ).toBeInTheDocument() - expect(within(region).getByText('直连模型 · Execute')) - .toBeInTheDocument() + within(region).getByText("编程 Subagent", { + selector: "strong", + }), + ).toBeInTheDocument(); + expect(within(region).getByText("直连模型 · Execute")).toBeInTheDocument(); await waitFor(() => { const persistedActivities = vi .mocked(api.conversations.saveLocal) .mock.calls.flatMap(([batch]) => batch.flatMap((conversation) => - conversation.messages.flatMap( - (message) => message.subagents ?? [] - ) - ) - ) + conversation.messages.flatMap((message) => message.subagents ?? []), + ), + ); expect(persistedActivities).toContainEqual( expect.objectContaining({ childTaskId: childRunId, actor: { - kind: 'direct-model', - label: '编程 Subagent' + kind: "direct-model", + label: "编程 Subagent", }, - workMode: 'execute', - output: '验证通过' - }) - ) - }) + workMode: "execute", + output: "验证通过", + }), + ); + }); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) + fireEvent.click(screen.getByLabelText("切换助手工作栏")); expect( - await screen.findByText('明确创建的任务会显示在这里。') - ).toBeInTheDocument() - }) + await screen.findByText("明确创建的任务会显示在这里。"), + ).toBeInTheDocument(); + }); - it('fails an unterminated subagent when the parent reports done', async () => { - render() - fireEvent.change(await screen.findByLabelText('向 GoodBuddy 提问'), { - target: { value: '检查未完成子任务' } - }) - fireEvent.click(screen.getByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + it("fails an unterminated subagent when the parent reports done", async () => { + render(); + fireEvent.change(await screen.findByLabelText("向 GoodBuddy 提问"), { + target: { value: "检查未完成子任务" }, + }); + fireEvent.click(screen.getByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'subagent', - childTaskId: '00000000-0000-4000-8000-000000000612', - expertId: '00000000-0000-4000-8000-000000000712', - expertName: 'explorer', - routingMode: 'native', - runtimeCallId: 'call-task-2', - state: 'running', - reason: '检查未完成路径' - }) + type: "subagent", + childTaskId: "00000000-0000-4000-8000-000000000612", + expertId: "00000000-0000-4000-8000-000000000712", + expertName: "explorer", + routingMode: "native", + runtimeCallId: "call-task-2", + state: "running", + reason: "检查未完成路径", + }); agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) + type: "done", + }); + }); - const region = await screen.findByLabelText('子代理状态') - expect(within(region).getByText('失败')).toBeInTheDocument() - const subagentCard = within(region).getByRole('group') + const region = await screen.findByLabelText("子代理状态"); + expect(within(region).getByText("失败")).toBeInTheDocument(); + const subagentCard = within(region).getByRole("group"); fireEvent.click( - within(subagentCard).getByText('explorer', { - selector: 'strong' - }) - ) - fireEvent(subagentCard, new Event('toggle')) + within(subagentCard).getByText("explorer", { + selector: "strong", + }), + ); + fireEvent(subagentCard, new Event("toggle")); expect( - within(region).getByText( - '父请求已结束，但子 Agent 未报告完成状态。' - ) - ).toBeInTheDocument() + within(region).getByText("父请求已结束，但子 Agent 未报告完成状态。"), + ).toBeInTheDocument(); await waitFor(() => { const persistedMessages = vi .mocked(api.conversations.saveLocal) .mock.calls.flatMap(([batch]) => - batch.flatMap((conversation) => conversation.messages) - ) + batch.flatMap((conversation) => conversation.messages), + ); expect( persistedMessages.some( (message) => - message.subagents?.[0]?.state === 'failed' && + message.subagents?.[0]?.state === "failed" && message.subagents[0].error === - '父请求已结束，但子 Agent 未报告完成状态。' - ) - ).toBe(true) - }) - }) + "父请求已结束，但子 Agent 未报告完成状态。", + ), + ).toBe(true); + }); + }); - it('offers once, session, permanent, and deny for a tool call', async () => { - render() + it("offers once, session, permanent, and deny for a tool call", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '运行工具' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "运行工具" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'approval', + type: "approval", approvalId: crypto.randomUUID(), - title: 'Continue 请求调用 Bash', - description: '确认工具调用', - toolName: 'Bash', - argumentSummary: 'echo safe', - allowPermanent: true - }) - }) + title: "Continue 请求调用 Bash", + description: "确认工具调用", + toolName: "Bash", + argumentSummary: "echo safe", + allowPermanent: true, + }); + }); - expect(await screen.findByText('仅此次')).toBeInTheDocument() - expect(screen.getByText('此会话')).toBeInTheDocument() - expect(screen.getByText('永久允许')).toBeInTheDocument() - expect(screen.getAllByText('拒绝')).toHaveLength(2) - fireEvent.click(screen.getByText('此会话')) + expect(await screen.findByText("仅此次")).toBeInTheDocument(); + expect(screen.getByText("此会话")).toBeInTheDocument(); + expect(screen.getByText("永久允许")).toBeInTheDocument(); + expect(screen.getAllByText("拒绝")).toHaveLength(2); + fireEvent.click(screen.getByText("此会话")); await waitFor(() => expect(api.agent.respondApproval).toHaveBeenCalledWith( expect.any(String), - 'session' - ) - ) - }) + "session", + ), + ); + }); - it('renders and answers an OpenCode question request', async () => { - render() - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '需要确认的任务' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + it("renders and answers an OpenCode question request", async () => { + render(); + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "需要确认的任务" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'question', - questionId: 'question-1', + type: "question", + questionId: "question-1", questions: [ { - header: '实现方式', - question: '请选择实现方式', + header: "实现方式", + question: "请选择实现方式", options: [ { - label: '直接修改', - description: '立即更新现有实现' + label: "直接修改", + description: "立即更新现有实现", }, { - label: '先写测试', - description: '先增加回归测试' - } + label: "先写测试", + description: "先增加回归测试", + }, ], multiple: false, - custom: true - } - ] - }) - }) + custom: true, + }, + ], + }); + }); expect( - await screen.findByText('OpenCode 需要补充信息') - ).toBeInTheDocument() - fireEvent.click(screen.getByLabelText(/先写测试/u)) - fireEvent.click(screen.getByRole('button', { name: '提交回答' })) + await screen.findByText("OpenCode 需要补充信息"), + ).toBeInTheDocument(); + fireEvent.click(screen.getByLabelText(/先写测试/u)); + fireEvent.click(screen.getByRole("button", { name: "提交回答" })); await waitFor(() => - expect(api.agent.respondQuestion).toHaveBeenCalledWith( - 'question-1', - [['先写测试']] - ) - ) - expect( - screen.queryByText('OpenCode 需要补充信息') - ).not.toBeInTheDocument() - }) + expect(api.agent.respondQuestion).toHaveBeenCalledWith("question-1", [ + ["先写测试"], + ]), + ); + expect(screen.queryByText("OpenCode 需要补充信息")).not.toBeInTheDocument(); + }); - it('configures a runtime without reading an existing API key', async () => { - render() + it("configures a runtime without reading an existing API key", async () => { + render(); - fireEvent.click(await screen.findByText('本地工作区')) + fireEvent.click(await screen.findByText("本地工作区")); expect( - await screen.findByRole('heading', { - name: '设置中心' - }) - ).toBeInTheDocument() - expect(screen.queryByRole('dialog')).not.toBeInTheDocument() - expect(screen.getByRole('region', { name: '设置中心' })) - .toBeInTheDocument() + await screen.findByRole("heading", { + name: "设置中心", + }), + ).toBeInTheDocument(); + expect(screen.queryByRole("dialog")).not.toBeInTheDocument(); expect( - screen.getByRole('tab', { name: 'Agent Runtime' }) - ).toBeInTheDocument() + screen.getByRole("region", { name: "设置中心" }), + ).toBeInTheDocument(); expect( - screen.getByRole('tab', { name: '安全与数据' }) - ).toBeInTheDocument() - fireEvent.click(screen.getByRole('tab', { name: '模型连接' })) + screen.getByRole("tab", { name: "Agent Runtime" }), + ).toBeInTheDocument(); + expect(screen.getByRole("tab", { name: "安全与数据" })).toBeInTheDocument(); + fireEvent.click(screen.getByRole("tab", { name: "模型连接" })); - const apiKeyInput = screen.getByLabelText('API Key') - expect(apiKeyInput).toHaveValue('') + const apiKeyInput = screen.getByLabelText("API Key"); + expect(apiKeyInput).toHaveValue(""); fireEvent.change(apiKeyInput, { - target: { value: 'test-api-key' } - }) - fireEvent.click(screen.getByRole('button', { name: '保存设置' })) + target: { value: "test-api-key" }, + }); + fireEvent.click(screen.getByRole("button", { name: "保存设置" })); await waitFor(() => expect(api.settings.updateRuntime).toHaveBeenCalledWith( expect.objectContaining({ apiKey: { - action: 'replace', - value: 'test-api-key' - } - }) - ) - ) - await waitFor(() => expect(apiKeyInput).toHaveValue('')) - }) + action: "replace", + value: "test-api-key", + }, + }), + ), + ); + await waitFor(() => expect(apiKeyInput).toHaveValue("")); + }); - it('opens the global assistant sidebar and switches work tabs', async () => { - const { container } = render() + it("opens the global assistant sidebar and switches work tabs", async () => { + const { container } = render(); - const sidebar = screen.getByLabelText('助手工作栏') - expect(sidebar).not.toHaveClass('assistant-sidebar--open') - const assistantTrigger = - screen.getByLabelText('切换助手工作栏') + const sidebar = screen.getByLabelText("助手工作栏"); + expect(sidebar).not.toHaveClass("assistant-sidebar--open"); + const assistantTrigger = screen.getByLabelText("切换助手工作栏"); expect(assistantTrigger.parentElement).toBe( - container.querySelector('.app-content') - ) - expect(container.querySelector('.sidebar')?.parentElement).toBe( - container.querySelector('.app-shell') - ) - expect(container.querySelector('.app-frame')?.parentElement).toBe( - container.querySelector('.app-shell') - ) - expect(container.querySelector('.topbar')?.parentElement).toBe( - container.querySelector('.app-frame') - ) - expect(assistantTrigger).toHaveClass('assistant-sidebar-toggle') - expect(assistantTrigger).not.toHaveClass('icon-button--active') - expect(assistantTrigger).toHaveAttribute('aria-expanded', 'false') + container.querySelector(".app-content"), + ); + expect(container.querySelector(".sidebar")?.parentElement).toBe( + container.querySelector(".app-shell"), + ); + expect(container.querySelector(".app-frame")?.parentElement).toBe( + container.querySelector(".app-shell"), + ); + expect(container.querySelector(".topbar")?.parentElement).toBe( + container.querySelector(".app-frame"), + ); + expect(assistantTrigger).toHaveClass("assistant-sidebar-toggle"); + expect(assistantTrigger).not.toHaveClass("icon-button--active"); + expect(assistantTrigger).toHaveAttribute("aria-expanded", "false"); expect(assistantTrigger).toHaveAttribute( - 'aria-controls', - 'assistant-sidebar' - ) + "aria-controls", + "assistant-sidebar", + ); expect( - assistantTrigger.querySelector('.lucide-panel-right-open') - ).toBeInTheDocument() + assistantTrigger.querySelector(".lucide-panel-right-open"), + ).toBeInTheDocument(); expect( - assistantTrigger.querySelector('.lucide-panel-right-close') - ).not.toBeInTheDocument() + assistantTrigger.querySelector(".lucide-panel-right-close"), + ).not.toBeInTheDocument(); + expect(container.querySelector(".topbar")).toContainElement( + screen.getByLabelText("关闭窗口"), + ); + expect(container.querySelector(".topbar")).toContainElement( + container.querySelector(".conversation-title"), + ); + expect(container.querySelector(".topbar")).not.toContainElement( + assistantTrigger, + ); + expect(screen.queryByLabelText("关闭助手工作栏")).not.toBeInTheDocument(); + fireEvent.click(assistantTrigger); + expect(sidebar).toHaveClass("assistant-sidebar--open"); + expect(sidebar).toHaveAttribute("id", "assistant-sidebar"); + expect(sidebar).toHaveAttribute("role", "complementary"); + expect(sidebar).not.toHaveAttribute("aria-modal"); + expect(assistantTrigger).toHaveAttribute("aria-expanded", "true"); + expect(assistantTrigger).not.toHaveClass("icon-button--active"); expect( - container.querySelector('.topbar') - ).toContainElement(screen.getByLabelText('关闭窗口')) - expect(container.querySelector('.topbar')).toContainElement( - container.querySelector('.conversation-title') - ) + assistantTrigger.querySelector(".lucide-panel-right-close"), + ).toBeInTheDocument(); expect( - container.querySelector('.topbar') - ).not.toContainElement(assistantTrigger) + assistantTrigger.querySelector(".lucide-panel-right-open"), + ).not.toBeInTheDocument(); + const main = document.querySelector("main"); + expect(main).not.toHaveAttribute("inert"); expect( - screen.queryByLabelText('关闭助手工作栏') - ).not.toBeInTheDocument() - fireEvent.click(assistantTrigger) - expect(sidebar).toHaveClass('assistant-sidebar--open') - expect(sidebar).toHaveAttribute('id', 'assistant-sidebar') - expect(sidebar).toHaveAttribute('role', 'complementary') - expect(sidebar).not.toHaveAttribute('aria-modal') - expect(assistantTrigger).toHaveAttribute('aria-expanded', 'true') - expect(assistantTrigger).not.toHaveClass('icon-button--active') - expect( - assistantTrigger.querySelector('.lucide-panel-right-close') - ).toBeInTheDocument() - expect( - assistantTrigger.querySelector('.lucide-panel-right-open') - ).not.toBeInTheDocument() - const main = document.querySelector('main') - expect(main).not.toHaveAttribute('inert') - expect( - container.querySelector('.assistant-sidebar-backdrop') - ).not.toBeInTheDocument() + container.querySelector(".assistant-sidebar-backdrop"), + ).not.toBeInTheDocument(); + expect(screen.getByRole("tab", { name: "任务中心" })).toHaveAttribute( + "aria-selected", + "true", + ); + const taskIndexHeading = screen.getByRole("heading", { + name: "任务索引", + }); + const newTaskButton = screen.getByRole("button", { + name: "新建任务", + }); + expect(taskIndexHeading.parentElement).toContainElement(newTaskButton); expect( - screen.getByRole('tab', { name: '任务中心' }) - ).toHaveAttribute('aria-selected', 'true') - const taskIndexHeading = screen.getByRole('heading', { - name: '任务索引' - }) - const newTaskButton = screen.getByRole('button', { - name: '新建任务' - }) - expect(taskIndexHeading.parentElement).toContainElement( - newTaskButton - ) + screen.queryByRole("region", { name: "当前会话的任务" }), + ).not.toBeInTheDocument(); + expect(screen.queryByText("最近任务")).not.toBeInTheDocument(); expect( - screen.queryByRole('region', { name: '当前会话的任务' }) - ).not.toBeInTheDocument() - expect(screen.queryByText('最近任务')).not.toBeInTheDocument() + screen.queryByRole("tab", { name: "上下文" }), + ).not.toBeInTheDocument(); + fireEvent.click(screen.getByRole("tab", { name: "工作区" })); + expect(screen.getByRole("tab", { name: "工作区" })).toHaveAttribute( + "title", + "浏览项目文件、Git 变更与文件内容", + ); expect( - screen.queryByRole('tab', { name: '上下文' }) - ).not.toBeInTheDocument() - fireEvent.click(screen.getByRole('tab', { name: '工作区' })) + screen.queryByRole("heading", { name: "项目工作区" }), + ).not.toBeInTheDocument(); expect( - screen.getByRole('tab', { name: '工作区' }) - ).toHaveAttribute('title', '浏览项目文件、Git 变更与文件内容') + screen.getByRole("button", { name: "刷新工作区文件" }), + ).toBeInTheDocument(); expect( - screen.queryByRole('heading', { name: '项目工作区' }) - ).not.toBeInTheDocument() + screen.queryByText(/选择文件后在当前工作区内预览/), + ).not.toBeInTheDocument(); + fireEvent.click(screen.getByRole("tab", { name: "浏览器" })); + expect(screen.getByText("实时浏览器")).toBeInTheDocument(); + expect(screen.getByText(/Agent 打开网页后/)).toBeInTheDocument(); + fireEvent.click(screen.getByRole("tab", { name: "成果" })); expect( - screen.getByRole('button', { name: '刷新工作区文件' }) - ).toBeInTheDocument() + screen.getByRole("button", { name: "导入 PDF、图片或网页" }), + ).toBeInTheDocument(); expect( - screen.queryByText(/选择文件后在当前工作区内预览/) - ).not.toBeInTheDocument() - fireEvent.click(screen.getByRole('tab', { name: '浏览器' })) - expect(screen.getByText('实时浏览器')).toBeInTheDocument() + screen.getByRole("heading", { name: "生成与导入成果" }), + ).toBeInTheDocument(); + expect(screen.queryByRole("tab", { name: "预览" })).not.toBeInTheDocument(); + fireEvent.click(assistantTrigger); + expect(sidebar).not.toHaveClass("assistant-sidebar--open"); + expect(assistantTrigger).toHaveAttribute("aria-expanded", "false"); expect( - screen.getByText(/Agent 打开网页后/) - ).toBeInTheDocument() - fireEvent.click(screen.getByRole('tab', { name: '成果' })) - expect( - screen.getByRole('button', { name: '导入 PDF、图片或网页' }) - ).toBeInTheDocument() - expect( - screen.getByRole('heading', { name: '生成与导入成果' }) - ).toBeInTheDocument() - expect( - screen.queryByRole('tab', { name: '预览' }) - ).not.toBeInTheDocument() - fireEvent.click(assistantTrigger) - expect(sidebar).not.toHaveClass('assistant-sidebar--open') - expect(assistantTrigger).toHaveAttribute('aria-expanded', 'false') - expect( - assistantTrigger.querySelector('.lucide-panel-right-open') - ).toBeInTheDocument() - expect(main).not.toHaveAttribute('inert') - await waitFor(() => expect(assistantTrigger).toHaveFocus()) - }) + assistantTrigger.querySelector(".lucide-panel-right-open"), + ).toBeInTheDocument(); + expect(main).not.toHaveAttribute("inert"); + await waitFor(() => expect(assistantTrigger).toHaveFocus()); + }); - it('keeps completed chat replies out of the results sidebar', async () => { - render() + it("keeps completed chat replies out of the results sidebar", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '普通聊天问题' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor(() => expect(run).toHaveBeenCalledOnce()) - const request = run.mock.calls[0]?.[0] + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "普通聊天问题" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce()); + const request = run.mock.calls[0]?.[0]; if (!request) { - throw new Error('Missing request') + throw new Error("Missing request"); } act(() => { agentListener?.({ requestId: request.requestId, - type: 'text', - delta: '这是一条普通聊天回复' - }) + type: "text", + delta: "这是一条普通聊天回复", + }); agentListener?.({ requestId: request.requestId, - type: 'done' - }) - }) - expect( - await screen.findByText('这是一条普通聊天回复') - ).toBeInTheDocument() + type: "done", + }); + }); + expect(await screen.findByText("这是一条普通聊天回复")).toBeInTheDocument(); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - fireEvent.click(screen.getByRole('tab', { name: '成果' })) - const sidebar = screen.getByLabelText('助手工作栏') + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + fireEvent.click(screen.getByRole("tab", { name: "成果" })); + const sidebar = screen.getByLabelText("助手工作栏"); expect( - within(sidebar).queryByText('这是一条普通聊天回复') - ).not.toBeInTheDocument() + within(sidebar).queryByText("这是一条普通聊天回复"), + ).not.toBeInTheDocument(); expect( within(sidebar).getByText( - '生成的文件、图片、报告和手动导入内容会显示在这里。' - ) - ).toBeInTheDocument() - }) + "生成的文件、图片、报告和手动导入内容会显示在这里。", + ), + ).toBeInTheDocument(); + }); - it('shows the browser toolbar and starts an empty session by address', async () => { - render() + it("shows the browser toolbar and starts an empty session by address", async () => { + render(); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - fireEvent.click(screen.getByRole('tab', { name: '浏览器' })) + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + fireEvent.click(screen.getByRole("tab", { name: "浏览器" })); - const toolbar = screen.getByRole('form', { - name: '浏览器工具栏' - }) - const address = within(toolbar).getByRole('textbox', { - name: '浏览器地址' - }) - const go = within(toolbar).getByRole('button', { name: '前往' }) - expect(within(toolbar).getByRole('button', { name: '返回' })) - .toBeDisabled() - expect(within(toolbar).getByRole('button', { name: '刷新' })) - .toBeDisabled() - expect(within(toolbar).getByRole('button', { name: '交互' })) - .toBeDisabled() + const toolbar = screen.getByRole("form", { + name: "浏览器工具栏", + }); + const address = within(toolbar).getByRole("textbox", { + name: "浏览器地址", + }); + const go = within(toolbar).getByRole("button", { name: "前往" }); expect( - within(toolbar).getByRole('button', { name: '关闭浏览器' }) - ).toBeDisabled() - expect(go).toBeDisabled() - expect(screen.getByText(/输入地址并前往可打开页面/u)) - .toBeInTheDocument() + within(toolbar).getByRole("button", { name: "返回" }), + ).toBeDisabled(); + expect( + within(toolbar).getByRole("button", { name: "刷新" }), + ).toBeDisabled(); + expect( + within(toolbar).getByRole("button", { name: "交互" }), + ).toBeDisabled(); + expect( + within(toolbar).getByRole("button", { name: "关闭浏览器" }), + ).toBeDisabled(); + expect(go).toBeDisabled(); + expect(screen.getByText(/输入地址并前往可打开页面/u)).toBeInTheDocument(); fireEvent.change(address, { - target: { value: 'https://example.com/start' } - }) - expect(go).toBeEnabled() - fireEvent.click(go) + target: { value: "https://example.com/start" }, + }); + expect(go).toBeEnabled(); + fireEvent.click(go); await waitFor(() => expect(api.browser.navigate).toHaveBeenLastCalledWith( expect.any(String), - 'https://example.com/start' - ) - ) - expect(address).toHaveValue('https://example.com/start') + "https://example.com/start", + ), + ); + expect(address).toHaveValue("https://example.com/start"); fireEvent.change(address, { - target: { value: 'example.com/enter' } - }) - fireEvent.keyDown(address, { key: 'Enter' }) + target: { value: "example.com/enter" }, + }); + fireEvent.keyDown(address, { key: "Enter" }); await waitFor(() => expect(api.browser.navigate).toHaveBeenLastCalledWith( expect.any(String), - 'http://example.com/enter' - ) - ) - expect(address).toHaveValue('http://example.com/enter') + "http://example.com/enter", + ), + ); + expect(address).toHaveValue("http://example.com/enter"); fireEvent.change(address, { - target: { value: 'mailto:user@example.com' } - }) - fireEvent.keyDown(address, { key: 'Enter' }) + target: { value: "mailto:user@example.com" }, + }); + fireEvent.keyDown(address, { key: "Enter" }); await waitFor(() => expect(api.browser.navigate).toHaveBeenLastCalledWith( expect.any(String), - 'mailto:user@example.com' - ) - ) + "mailto:user@example.com", + ), + ); fireEvent.change(address, { - target: { value: 'localhost:3000/path' } - }) - fireEvent.keyDown(address, { key: 'Enter' }) + target: { value: "localhost:3000/path" }, + }); + fireEvent.keyDown(address, { key: "Enter" }); await waitFor(() => expect(api.browser.navigate).toHaveBeenLastCalledWith( expect.any(String), - 'http://localhost:3000/path' - ) - ) - expect(api.browser.navigate).toHaveBeenCalledTimes(4) - }) + "http://localhost:3000/path", + ), + ); + expect(api.browser.navigate).toHaveBeenCalledTimes(4); + }); - it('suppresses duplicate browser actions while keeping stop loading available', async () => { - const navigation = deferred() - vi.mocked(api.browser.navigate).mockReturnValueOnce( - navigation.promise - ) - render() + it("suppresses duplicate browser actions while keeping stop loading available", async () => { + const navigation = deferred(); + vi.mocked(api.browser.navigate).mockReturnValueOnce(navigation.promise); + render(); - fireEvent.click(screen.getByLabelText('切换助手工作栏')) - fireEvent.click(screen.getByRole('tab', { name: '浏览器' })) + fireEvent.click(screen.getByLabelText("切换助手工作栏")); + fireEvent.click(screen.getByRole("tab", { name: "浏览器" })); - const toolbar = screen.getByRole('form', { - name: '浏览器工具栏' - }) - const address = within(toolbar).getByRole('textbox', { - name: '浏览器地址' - }) - const go = within(toolbar).getByRole('button', { name: '前往' }) + const toolbar = screen.getByRole("form", { + name: "浏览器工具栏", + }); + const address = within(toolbar).getByRole("textbox", { + name: "浏览器地址", + }); + const go = within(toolbar).getByRole("button", { name: "前往" }); fireEvent.change(address, { - target: { value: 'https://example.com/pending' } - }) - fireEvent.click(go) - fireEvent.click(go) - fireEvent.keyDown(address, { key: 'Enter' }) + target: { value: "https://example.com/pending" }, + }); + fireEvent.click(go); + fireEvent.click(go); + fireEvent.keyDown(address, { key: "Enter" }); - expect(api.browser.navigate).toHaveBeenCalledOnce() + expect(api.browser.navigate).toHaveBeenCalledOnce(); const conversationId = - vi.mocked(api.browser.navigate).mock.calls[0]?.[0] ?? '' + vi.mocked(api.browser.navigate).mock.calls[0]?.[0] ?? ""; act(() => { - browserListener?.(createBrowserState(conversationId, { - status: 'loading', - isLoading: true, - url: 'https://example.com/pending', - updatedAt: Date.now() - })) - }) + browserListener?.( + createBrowserState(conversationId, { + status: "loading", + isLoading: true, + url: "https://example.com/pending", + updatedAt: Date.now(), + }), + ); + }); expect( - within(toolbar).getByRole('button', { name: '返回' }) - ).toBeDisabled() - expect(address).toBeDisabled() - expect(go).toBeDisabled() + within(toolbar).getByRole("button", { name: "返回" }), + ).toBeDisabled(); + expect(address).toBeDisabled(); + expect(go).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '交互' }) - ).toBeDisabled() + within(toolbar).getByRole("button", { name: "交互" }), + ).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '关闭浏览器' }) - ).toBeDisabled() - const stopLoading = within(toolbar).getByRole('button', { - name: '停止加载' - }) - expect(stopLoading).toBeEnabled() - fireEvent.click(stopLoading) - expect(api.browser.stopLoading).toHaveBeenCalledOnce() + within(toolbar).getByRole("button", { name: "关闭浏览器" }), + ).toBeDisabled(); + const stopLoading = within(toolbar).getByRole("button", { + name: "停止加载", + }); + expect(stopLoading).toBeEnabled(); + fireEvent.click(stopLoading); + expect(api.browser.stopLoading).toHaveBeenCalledOnce(); await act(async () => { - navigation.resolve() - await navigation.promise - }) - const back = within(toolbar).getByRole('button', { name: '返回' }) - const backNavigation = deferred() - vi.mocked(api.browser.back).mockReturnValueOnce( - backNavigation.promise - ) + navigation.resolve(); + await navigation.promise; + }); + const back = within(toolbar).getByRole("button", { name: "返回" }); + const backNavigation = deferred(); + vi.mocked(api.browser.back).mockReturnValueOnce(backNavigation.promise); act(() => { - browserListener?.(createBrowserState(conversationId, { - canGoBack: true, - url: 'https://example.com/pending', - updatedAt: Date.now() + 1 - })) - }) - expect(back).toBeEnabled() - fireEvent.click(back) - fireEvent.click(back) - expect(api.browser.back).toHaveBeenCalledOnce() + browserListener?.( + createBrowserState(conversationId, { + canGoBack: true, + url: "https://example.com/pending", + updatedAt: Date.now() + 1, + }), + ); + }); + expect(back).toBeEnabled(); + fireEvent.click(back); + fireEvent.click(back); + expect(api.browser.back).toHaveBeenCalledOnce(); await act(async () => { - backNavigation.resolve() - await backNavigation.promise - }) - }) + backNavigation.resolve(); + await backNavigation.promise; + }); + }); - it('controls the active live browser without overwriting its address draft', async () => { - render() + it("controls the active live browser without overwriting its address draft", async () => { + render(); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '打开示例网页' } - }) - fireEvent.click(await screen.findByLabelText('发送')) - await waitFor( - () => expect(run).toHaveBeenCalledOnce(), - { timeout: 3000 } - ) - const conversationId = run.mock.calls[0]?.[0].conversationId - expect(conversationId).toBeTruthy() + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "打开示例网页" }, + }); + fireEvent.click(await screen.findByLabelText("发送")); + await waitFor(() => expect(run).toHaveBeenCalledOnce(), { timeout: 3000 }); + const conversationId = run.mock.calls[0]?.[0].conversationId; + expect(conversationId).toBeTruthy(); act(() => { - browserListener?.(createBrowserState(conversationId ?? '', { - canGoBack: false, - url: 'https://example.com/', - frameDataUrl: 'data:image/jpeg;base64,/9j/2Q==', - updatedAt: Date.now() - })) - }) + browserListener?.( + createBrowserState(conversationId ?? "", { + canGoBack: false, + url: "https://example.com/", + frameDataUrl: "data:image/jpeg;base64,/9j/2Q==", + updatedAt: Date.now(), + }), + ); + }); - expect(screen.getByLabelText('助手工作栏')).toHaveClass( - 'assistant-sidebar--open' - ) - expect( - screen.getByRole('tab', { name: '浏览器' }) - ).toHaveAttribute('aria-selected', 'true') - expect( - screen.getByAltText('Agent 实时浏览器画面') - ).toHaveAttribute( - 'src', - 'data:image/jpeg;base64,/9j/2Q==' - ) - const toolbar = screen.getByRole('form', { - name: '浏览器工具栏' - }) - const address = within(toolbar).getByRole('textbox', { - name: '浏览器地址' - }) - const back = within(toolbar).getByRole('button', { name: '返回' }) - expect(address).toHaveValue('https://example.com/') - expect(back).toBeDisabled() + expect(screen.getByLabelText("助手工作栏")).toHaveClass( + "assistant-sidebar--open", + ); + expect(screen.getByRole("tab", { name: "浏览器" })).toHaveAttribute( + "aria-selected", + "true", + ); + expect(screen.getByAltText("Agent 实时浏览器画面")).toHaveAttribute( + "src", + "data:image/jpeg;base64,/9j/2Q==", + ); + const toolbar = screen.getByRole("form", { + name: "浏览器工具栏", + }); + const address = within(toolbar).getByRole("textbox", { + name: "浏览器地址", + }); + const back = within(toolbar).getByRole("button", { name: "返回" }); + expect(address).toHaveValue("https://example.com/"); + expect(back).toBeDisabled(); - fireEvent.focus(address) + fireEvent.focus(address); fireEvent.change(address, { - target: { value: 'https://typed.example/draft' } - }) + target: { value: "https://typed.example/draft" }, + }); act(() => { - browserListener?.(createBrowserState(conversationId ?? '', { - canGoBack: true, - url: 'https://example.com/redirected', - updatedAt: Date.now() + 1 - })) - }) - expect(address).toHaveValue('https://typed.example/draft') - expect( - screen.getByAltText('Agent 实时浏览器画面') - ).toHaveAttribute( - 'src', - 'data:image/jpeg;base64,/9j/2Q==' - ) + browserListener?.( + createBrowserState(conversationId ?? "", { + canGoBack: true, + url: "https://example.com/redirected", + updatedAt: Date.now() + 1, + }), + ); + }); + expect(address).toHaveValue("https://typed.example/draft"); + expect(screen.getByAltText("Agent 实时浏览器画面")).toHaveAttribute( + "src", + "data:image/jpeg;base64,/9j/2Q==", + ); - fireEvent.keyDown(address, { key: 'Escape' }) - expect(address).toHaveValue('https://example.com/redirected') - expect(address).not.toHaveFocus() - expect(back).toBeEnabled() - fireEvent.click(back) + fireEvent.keyDown(address, { key: "Escape" }); + expect(address).toHaveValue("https://example.com/redirected"); + expect(address).not.toHaveFocus(); + expect(back).toBeEnabled(); + fireEvent.click(back); await waitFor(() => - expect(api.browser.back).toHaveBeenCalledWith(conversationId) - ) + expect(api.browser.back).toHaveBeenCalledWith(conversationId), + ); - fireEvent.click( - within(toolbar).getByRole('button', { name: '刷新' }) - ) + fireEvent.click(within(toolbar).getByRole("button", { name: "刷新" })); await waitFor(() => - expect(api.browser.reload).toHaveBeenCalledWith(conversationId) - ) + expect(api.browser.reload).toHaveBeenCalledWith(conversationId), + ); fireEvent.change(address, { - target: { value: 'https://example.com/manual' } - }) - fireEvent.click( - within(toolbar).getByRole('button', { name: '前往' }) - ) + target: { value: "https://example.com/manual" }, + }); + fireEvent.click(within(toolbar).getByRole("button", { name: "前往" })); await waitFor(() => expect(api.browser.navigate).toHaveBeenCalledWith( conversationId, - 'https://example.com/manual' - ) - ) - expect(address).toHaveValue('https://example.com/redirected') + "https://example.com/manual", + ), + ); + expect(address).toHaveValue("https://example.com/redirected"); act(() => { - browserListener?.(createBrowserState(conversationId ?? '', { - canGoBack: true, - url: 'https://example.com/canonical/', - updatedAt: Date.now() + 2 - })) - }) - expect(address).toHaveValue('https://example.com/canonical/') - fireEvent.focus(address) + browserListener?.( + createBrowserState(conversationId ?? "", { + canGoBack: true, + url: "https://example.com/canonical/", + updatedAt: Date.now() + 2, + }), + ); + }); + expect(address).toHaveValue("https://example.com/canonical/"); + fireEvent.focus(address); fireEvent.change(address, { - target: { value: 'https://example.com/canonical' } - }) - fireEvent.click( - within(toolbar).getByRole('button', { name: '前往' }) - ) + target: { value: "https://example.com/canonical" }, + }); + fireEvent.click(within(toolbar).getByRole("button", { name: "前往" })); await waitFor(() => expect(api.browser.navigate).toHaveBeenCalledWith( conversationId, - 'https://example.com/canonical' - ) - ) - expect(address).toHaveValue('https://example.com/canonical/') - fireEvent.click( - within(toolbar).getByRole('button', { name: '交互' }) - ) + "https://example.com/canonical", + ), + ); + expect(address).toHaveValue("https://example.com/canonical/"); + fireEvent.click(within(toolbar).getByRole("button", { name: "交互" })); await waitFor(() => - expect(api.browser.interact).toHaveBeenCalledWith(conversationId) - ) + expect(api.browser.interact).toHaveBeenCalledWith(conversationId), + ); act(() => { - browserListener?.(createBrowserState(conversationId ?? '', { - status: 'loading', - isLoading: false, - canGoBack: true, - url: 'https://example.com/canonical/', - updatedAt: Date.now() + 3 - })) - }) - expect(back).toBeDisabled() + browserListener?.( + createBrowserState(conversationId ?? "", { + status: "loading", + isLoading: false, + canGoBack: true, + url: "https://example.com/canonical/", + updatedAt: Date.now() + 3, + }), + ); + }); + expect(back).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '刷新' }) - ).toBeDisabled() - expect(address).toBeDisabled() + within(toolbar).getByRole("button", { name: "刷新" }), + ).toBeDisabled(); + expect(address).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '交互' }) - ).toBeDisabled() + within(toolbar).getByRole("button", { name: "交互" }), + ).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '关闭浏览器' }) - ).toBeDisabled() + within(toolbar).getByRole("button", { name: "关闭浏览器" }), + ).toBeDisabled(); act(() => { - browserListener?.(createBrowserState(conversationId ?? '', { - status: 'acting', - canGoBack: true, - url: 'https://example.com/canonical/', - updatedAt: Date.now() + 4 - })) - }) - expect(back).toBeDisabled() + browserListener?.( + createBrowserState(conversationId ?? "", { + status: "acting", + canGoBack: true, + url: "https://example.com/canonical/", + updatedAt: Date.now() + 4, + }), + ); + }); + expect(back).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '刷新' }) - ).toBeDisabled() - expect(address).toBeDisabled() + within(toolbar).getByRole("button", { name: "刷新" }), + ).toBeDisabled(); + expect(address).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '前往' }) - ).toBeDisabled() + within(toolbar).getByRole("button", { name: "前往" }), + ).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '交互' }) - ).toBeDisabled() + within(toolbar).getByRole("button", { name: "交互" }), + ).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '关闭浏览器' }) - ).toBeDisabled() + within(toolbar).getByRole("button", { name: "关闭浏览器" }), + ).toBeDisabled(); act(() => { - browserListener?.(createBrowserState(conversationId ?? '', { - status: 'acting', - isLoading: true, - canGoBack: true, - url: 'https://example.com/canonical/', - updatedAt: Date.now() + 5 - })) - }) + browserListener?.( + createBrowserState(conversationId ?? "", { + status: "acting", + isLoading: true, + canGoBack: true, + url: "https://example.com/canonical/", + updatedAt: Date.now() + 5, + }), + ); + }); expect( - within(toolbar).queryByRole('button', { name: '刷新' }) - ).not.toBeInTheDocument() - const stopLoading = within(toolbar).getByRole('button', { - name: '停止加载' - }) - expect(stopLoading).toBeEnabled() + within(toolbar).queryByRole("button", { name: "刷新" }), + ).not.toBeInTheDocument(); + const stopLoading = within(toolbar).getByRole("button", { + name: "停止加载", + }); + expect(stopLoading).toBeEnabled(); expect( - within(toolbar).getByRole('button', { name: '关闭浏览器' }) - ).toBeDisabled() - fireEvent.click(stopLoading) + within(toolbar).getByRole("button", { name: "关闭浏览器" }), + ).toBeDisabled(); + fireEvent.click(stopLoading); await waitFor(() => - expect(api.browser.stopLoading).toHaveBeenCalledWith( - conversationId - ) - ) + expect(api.browser.stopLoading).toHaveBeenCalledWith(conversationId), + ); act(() => { - browserListener?.(createBrowserState(conversationId ?? '', { - status: 'interactive', - canGoBack: true, - url: 'https://example.com/canonical/', - updatedAt: Date.now() + 6 - })) - }) - expect(back).toBeDisabled() + browserListener?.( + createBrowserState(conversationId ?? "", { + status: "interactive", + canGoBack: true, + url: "https://example.com/canonical/", + updatedAt: Date.now() + 6, + }), + ); + }); + expect(back).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '刷新' }) - ).toBeDisabled() - expect(address).toBeDisabled() + within(toolbar).getByRole("button", { name: "刷新" }), + ).toBeDisabled(); + expect(address).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '交互' }) - ).toBeDisabled() + within(toolbar).getByRole("button", { name: "交互" }), + ).toBeDisabled(); expect( - within(toolbar).getByRole('button', { name: '关闭浏览器' }) - ).toBeDisabled() + within(toolbar).getByRole("button", { name: "关闭浏览器" }), + ).toBeDisabled(); act(() => { - browserListener?.(createBrowserState(conversationId ?? '', { - canGoBack: true, - url: 'https://example.com/canonical/', - updatedAt: Date.now() + 7 - })) - }) + browserListener?.( + createBrowserState(conversationId ?? "", { + canGoBack: true, + url: "https://example.com/canonical/", + updatedAt: Date.now() + 7, + }), + ); + }); fireEvent.click( - within(toolbar).getByRole('button', { name: '关闭浏览器' }) - ) + within(toolbar).getByRole("button", { name: "关闭浏览器" }), + ); await waitFor(() => - expect(api.browser.stop).toHaveBeenCalledWith(conversationId) - ) - }) + expect(api.browser.stop).toHaveBeenCalledWith(conversationId), + ); + }); - it('opens Smart Heartbeat as a first-class workspace', async () => { - render() + it("opens Smart Heartbeat as a first-class workspace", async () => { + render(); - fireEvent.click( - screen.getByRole('button', { name: '智能心跳' }) - ) + fireEvent.click(screen.getByRole("button", { name: "智能心跳" })); expect( - await screen.findByRole('heading', { name: '智能心跳' }) - ).toBeInTheDocument() + await screen.findByRole("heading", { name: "智能心跳" }), + ).toBeInTheDocument(); expect( - await screen.findByRole('tab', { name: '运行概览' }) - ).toBeInTheDocument() + await screen.findByRole("tab", { name: "运行概览" }), + ).toBeInTheDocument(); expect( - screen.getByRole('button', { name: '配置智能心跳' }) - ).toBeInTheDocument() - expect( - screen.queryByLabelText('切换助手工作栏') - ).toBeInTheDocument() - }) + screen.getByRole("button", { name: "配置智能心跳" }), + ).toBeInTheDocument(); + expect(screen.queryByLabelText("切换助手工作栏")).toBeInTheDocument(); + }); - it('excludes ignored heartbeat suggestions from the navigation badge', async () => { - const heartbeatId = '00000000-0000-4000-8000-000000000701' - const cancelledTaskId = '00000000-0000-4000-8000-000000000801' - const pendingTaskId = '00000000-0000-4000-8000-000000000802' + it("excludes ignored heartbeat suggestions from the navigation badge", async () => { + const heartbeatId = "00000000-0000-4000-8000-000000000701"; + const cancelledTaskId = "00000000-0000-4000-8000-000000000801"; + const pendingTaskId = "00000000-0000-4000-8000-000000000802"; vi.mocked(api.heartbeats.list).mockResolvedValue([ { id: heartbeatId, - scope: { kind: 'projects', projectIds: [projectId] }, - name: '每日回顾', - timezone: 'Asia/Shanghai', - recurrence: { type: 'daily', localTime: '09:00' }, + scope: { kind: "projects", projectIds: [projectId] }, + name: "每日回顾", + timezone: "Asia/Shanghai", + recurrence: { type: "daily", localTime: "09:00" }, enabled: true, lookbackHours: 24, retentionDays: 30, - nextRunAt: '2026-08-05T01:00:00.000Z', - createdAt: '2026-08-01T00:00:00.000Z', - updatedAt: '2026-08-01T00:00:00.000Z' - } - ]) + nextRunAt: "2026-08-05T01:00:00.000Z", + createdAt: "2026-08-01T00:00:00.000Z", + updatedAt: "2026-08-01T00:00:00.000Z", + }, + ]); vi.mocked(api.heartbeats.history).mockResolvedValue({ runs: [], entries: [ { - id: '00000000-0000-4000-8000-000000000901', + id: "00000000-0000-4000-8000-000000000901", configId: heartbeatId, - runId: '00000000-0000-4000-8000-000000000902', - scheduledFor: '2026-08-04T01:00:00.000Z', - summary: '建议处理两个后续行动。', + runId: "00000000-0000-4000-8000-000000000902", + scheduledFor: "2026-08-04T01:00:00.000Z", + summary: "建议处理两个后续行动。", highlights: [], proposedMemoryIds: [], followUpTaskIds: [cancelledTaskId, pendingTaskId], - createdAt: '2026-08-04T01:00:00.000Z' - } - ] - }) + createdAt: "2026-08-04T01:00:00.000Z", + }, + ], + }); vi.mocked(api.tasks.list).mockResolvedValue([ { id: cancelledTaskId, projectId, - title: '已忽略建议', - instructions: '无需继续处理。', - origin: 'assistant', - status: 'cancelled', - createdAt: '2026-08-04T01:00:00.000Z' + title: "已忽略建议", + instructions: "无需继续处理。", + origin: "assistant", + status: "cancelled", + createdAt: "2026-08-04T01:00:00.000Z", }, { id: pendingTaskId, projectId, - title: '待处理建议', - instructions: '继续处理此建议。', - origin: 'assistant', - status: 'paused', - createdAt: '2026-08-04T01:00:00.000Z' - } - ]) + title: "待处理建议", + instructions: "继续处理此建议。", + origin: "assistant", + status: "paused", + createdAt: "2026-08-04T01:00:00.000Z", + }, + ]); try { - render() + render(); expect( - await screen.findByLabelText('1 条待处理建议') - ).toBeInTheDocument() - expect( - screen.queryByLabelText('2 条待处理建议') - ).not.toBeInTheDocument() + await screen.findByLabelText("1 条待处理建议"), + ).toBeInTheDocument(); + expect(screen.queryByLabelText("2 条待处理建议")).not.toBeInTheDocument(); } finally { - cleanup() - vi.mocked(api.heartbeats.list).mockResolvedValue([]) + cleanup(); + vi.mocked(api.heartbeats.list).mockResolvedValue([]); vi.mocked(api.heartbeats.history).mockResolvedValue({ runs: [], - entries: [] - }) - vi.mocked(api.tasks.list).mockResolvedValue([]) + entries: [], + }); + vi.mocked(api.tasks.list).mockResolvedValue([]); } - }) + }); - it('shows retryable page-local knowledge errors without an empty-state flash', async () => { + it("shows retryable page-local knowledge errors without an empty-state flash", async () => { vi.mocked(api.knowledge.getSnapshot).mockRejectedValueOnce( - new Error('知识数据库暂时不可用') - ) - render() + new Error("知识数据库暂时不可用"), + ); + render(); - fireEvent.click(screen.getByRole('button', { name: '知识库' })) - expect( - await screen.findByText('知识库加载失败') - ).toBeInTheDocument() - expect(screen.getAllByText('知识数据库暂时不可用')).toHaveLength(1) - expect(screen.queryByText('建立第一个知识库')).not.toBeInTheDocument() + fireEvent.click(screen.getByRole("button", { name: "知识库" })); + expect(await screen.findByText("知识库加载失败")).toBeInTheDocument(); + expect(screen.getAllByText("知识数据库暂时不可用")).toHaveLength(1); + expect(screen.queryByText("建立第一个知识库")).not.toBeInTheDocument(); - fireEvent.click(screen.getByRole('button', { name: '重试' })) - expect( - await screen.findByText('建立第一个知识库') - ).toBeInTheDocument() - }) + fireEvent.click(screen.getByRole("button", { name: "重试" })); + expect(await screen.findByText("建立第一个知识库")).toBeInTheDocument(); + }); - it('retries the knowledge library whose selection failed', async () => { - const firstLibraryId = '11111111-1111-4111-8111-111111111111' - const secondLibraryId = '22222222-2222-4222-8222-222222222222' + it("retries the knowledge library whose selection failed", async () => { + const firstLibraryId = "11111111-1111-4111-8111-111111111111"; + const secondLibraryId = "22222222-2222-4222-8222-222222222222"; const libraries = [ { id: firstLibraryId, - name: '产品知识', - description: '', - storageMode: 'managed' as const, + name: "产品知识", + description: "", + storageMode: "managed" as const, graphEnabled: false, - graphStrategy: 'rules' as const, + graphStrategy: "rules" as const, sourceCount: 0, documentCount: 0, - indexedDocumentCount: 0 + indexedDocumentCount: 0, }, { id: secondLibraryId, - name: '工程知识', - description: '', - storageMode: 'managed' as const, + name: "工程知识", + description: "", + storageMode: "managed" as const, graphEnabled: false, - graphStrategy: 'rules' as const, + graphStrategy: "rules" as const, sourceCount: 0, documentCount: 0, - indexedDocumentCount: 0 - } - ] + indexedDocumentCount: 0, + }, + ]; const snapshot = { libraries, sources: [], documents: [], graphNodes: [], graphRelations: [], - evidence: [] - } + evidence: [], + }; vi.mocked(api.knowledge.getSnapshot) .mockResolvedValueOnce({ ...snapshot, - selectedLibraryId: firstLibraryId + selectedLibraryId: firstLibraryId, }) - .mockRejectedValueOnce(new Error('工程知识暂时不可用')) + .mockRejectedValueOnce(new Error("工程知识暂时不可用")) .mockResolvedValueOnce({ ...snapshot, - selectedLibraryId: secondLibraryId - }) - render() + selectedLibraryId: secondLibraryId, + }); + render(); - fireEvent.click(screen.getByRole('button', { name: '知识库' })) + fireEvent.click(screen.getByRole("button", { name: "知识库" })); fireEvent.click( - await screen.findByRole('button', { - name: /^工程知识/u - }, { timeout: 5_000 }) - ) - expect( - await screen.findByText('知识库刷新失败') - ).toBeInTheDocument() + await screen.findByRole( + "button", + { + name: /^工程知识/u, + }, + { timeout: 5_000 }, + ), + ); + expect(await screen.findByText("知识库刷新失败")).toBeInTheDocument(); - fireEvent.click(screen.getByRole('button', { name: '重试' })) + fireEvent.click(screen.getByRole("button", { name: "重试" })); await waitFor(() => expect(api.knowledge.getSnapshot).toHaveBeenLastCalledWith( - secondLibraryId - ) - ) + secondLibraryId, + ), + ); expect( - screen.getByRole('button', { - name: /^工程知识/u - }) - ).toHaveAttribute('aria-current', 'page') - }) + screen.getByRole("button", { + name: /^工程知识/u, + }), + ).toHaveAttribute("aria-current", "page"); + }); - it('shows retryable page-local heartbeat errors without first-time guidance', async () => { + it("shows retryable page-local heartbeat errors without first-time guidance", async () => { vi.mocked(api.heartbeats.list).mockRejectedValue( - new Error('心跳数据库暂时不可用') - ) - render() + new Error("心跳数据库暂时不可用"), + ); + render(); - fireEvent.click( - screen.getByRole('button', { name: '智能心跳' }) - ) + fireEvent.click(screen.getByRole("button", { name: "智能心跳" })); + expect(await screen.findByText("智能心跳加载失败")).toBeInTheDocument(); expect( - await screen.findByText('智能心跳加载失败') - ).toBeInTheDocument() - expect( - screen.queryByRole('button', { name: '配置智能心跳' }) - ).not.toBeInTheDocument() + screen.queryByRole("button", { name: "配置智能心跳" }), + ).not.toBeInTheDocument(); - const retry = await screen.findByRole('button', { name: '重试' }) - vi.mocked(api.heartbeats.list).mockResolvedValue([]) - fireEvent.click(retry) + const retry = await screen.findByRole("button", { name: "重试" }); + vi.mocked(api.heartbeats.list).mockResolvedValue([]); + fireEvent.click(retry); expect( - await screen.findByRole('button', { name: '配置智能心跳' }) - ).toBeInTheDocument() - }) + await screen.findByRole("button", { name: "配置智能心跳" }), + ).toBeInTheDocument(); + }); - it('keeps heartbeat plans independent of the active project', async () => { + it("keeps heartbeat plans independent of the active project", async () => { const secondProject = { ...project, - id: '00000000-0000-4000-8000-000000000102', - name: '第二项目', - rootPath: 'C:\\Second', + id: "00000000-0000-4000-8000-000000000102", + name: "第二项目", + rootPath: "C:\\Second", executionSpace: { - kind: 'local' as const, - rootPath: 'C:\\Second' - } - } + kind: "local" as const, + rootPath: "C:\\Second", + }, + }; vi.mocked(api.projects.list).mockResolvedValueOnce([ project, - secondProject - ]) + secondProject, + ]); vi.mocked(api.heartbeats.list).mockResolvedValue([ { - id: '00000000-0000-4000-8000-000000000701', - scope: { kind: 'projects', projectIds: [projectId] }, - name: '旧项目心跳', - timezone: 'Asia/Shanghai', - recurrence: { type: 'daily', localTime: '09:00' }, + id: "00000000-0000-4000-8000-000000000701", + scope: { kind: "projects", projectIds: [projectId] }, + name: "旧项目心跳", + timezone: "Asia/Shanghai", + recurrence: { type: "daily", localTime: "09:00" }, enabled: true, lookbackHours: 24, retentionDays: 30, - nextRunAt: '2026-08-05T01:00:00.000Z', - createdAt: '2026-08-01T00:00:00.000Z', - updatedAt: '2026-08-01T00:00:00.000Z' - } - ]) - render() + nextRunAt: "2026-08-05T01:00:00.000Z", + createdAt: "2026-08-01T00:00:00.000Z", + updatedAt: "2026-08-01T00:00:00.000Z", + }, + ]); + render(); - fireEvent.click( - screen.getByRole('button', { name: '智能心跳' }) - ) - fireEvent.click( - await screen.findByRole('tab', { name: '心跳计划' }) - ) - expect(await screen.findAllByText('旧项目心跳')).not.toHaveLength(0) - selectProjectOption(secondProject.name) - fireEvent.click( - screen.getByRole('button', { name: '智能心跳' }) - ) + fireEvent.click(screen.getByRole("button", { name: "智能心跳" })); + fireEvent.click(await screen.findByRole("tab", { name: "心跳计划" })); + expect(await screen.findAllByText("旧项目心跳")).not.toHaveLength(0); + selectProjectOption(secondProject.name); + fireEvent.click(screen.getByRole("button", { name: "智能心跳" })); - expect(await screen.findAllByText('旧项目心跳')).not.toHaveLength(0) - }) + expect(await screen.findAllByText("旧项目心跳")).not.toHaveLength(0); + }); - it('automatically snapshots the conversation project on new activity', async () => { - render() - await screen.findByText('项目：默认项目') + it("automatically snapshots the conversation project on new activity", async () => { + render(); + await screen.findByText("项目：默认项目"); - fireEvent.change(screen.getByLabelText('向 GoodBuddy 提问'), { - target: { value: '记录项目范围' } - }) - fireEvent.click(screen.getByLabelText('发送')) + fireEvent.change(screen.getByLabelText("向 GoodBuddy 提问"), { + target: { value: "记录项目范围" }, + }); + fireEvent.click(screen.getByLabelText("发送")); await waitFor(() => expect( vi @@ -11030,651 +10627,621 @@ describe('App', () => { .mock.calls.at(-1)?.[0] .find( (record) => - record.kind === 'request' && - record.title === '记录项目范围' - )?.scope + record.kind === "request" && record.title === "记录项目范围", + )?.scope, ).toEqual({ - kind: 'project', + kind: "project", projectId, - projectName: project.name - }) - ) - }) + projectName: project.name, + }), + ); + }); - it('migrates legacy activity and keeps its possible-loss warning visible', async () => { + it("migrates legacy activity and keeps its possible-loss warning visible", async () => { localStorage.setItem( ACTIVITY_STORAGE_KEY, JSON.stringify([ { - id: 'legacy-activity', - conversationId: 'legacy-conversation', - requestId: 'legacy-request', - scope: { kind: 'global' }, - kind: 'tool', - title: '旧版工具记录', - detail: 'x'.repeat(4_000), - status: 'completed', - createdAt: 1 - } - ]) - ) + id: "legacy-activity", + conversationId: "legacy-conversation", + requestId: "legacy-request", + scope: { kind: "global" }, + kind: "tool", + title: "旧版工具记录", + detail: "x".repeat(4_000), + status: "completed", + createdAt: 1, + }, + ]), + ); - render() - await screen.findByText('项目：默认项目') + render(); + await screen.findByText("项目：默认项目"); await waitFor(() => { expect(api.activityHistory.replace).toHaveBeenCalledWith( [ expect.objectContaining({ - id: 'legacy-activity', - detail: 'x'.repeat(4_000) - }) + id: "legacy-activity", + detail: "x".repeat(4_000), + }), ], - true - ) - expect(localStorage.getItem(ACTIVITY_STORAGE_KEY)).toBeNull() - }) + true, + ); + expect(localStorage.getItem(ACTIVITY_STORAGE_KEY)).toBeNull(); + }); - fireEvent.click(screen.getByRole('button', { name: '运行记录' })) + fireEvent.click(screen.getByRole("button", { name: "运行记录" })); expect( await screen.findByText( - '旧版本曾限制本机运行记录的数量和详情长度，因此更早的记录或部分详情可能已经丢失。新记录会完整保存。', + "旧版本曾限制本机运行记录的数量和详情长度，因此更早的记录或部分详情可能已经丢失。新记录会完整保存。", {}, - { timeout: 3_000 } - ) - ).toBeInTheDocument() - }) + { timeout: 3_000 }, + ), + ).toBeInTheDocument(); + }); - it('marks the current primary navigation page and hides decorative icons', async () => { - render() + it("marks the current primary navigation page and hides decorative icons", async () => { + render(); - const navigation = screen.getByRole('navigation', { - name: '主导航' - }) - const chat = within(navigation).getByRole('button', { name: '对话' }) - const knowledge = within(navigation).getByRole('button', { - name: '知识库' - }) + const navigation = screen.getByRole("navigation", { + name: "主导航", + }); + const chat = within(navigation).getByRole("button", { name: "对话" }); + const knowledge = within(navigation).getByRole("button", { + name: "知识库", + }); - expect(chat).toHaveAttribute('aria-current', 'page') - expect(knowledge).not.toHaveAttribute('aria-current') - for (const button of within(navigation).getAllByRole('button')) { - expect(button.querySelector('svg')).toHaveAttribute( - 'aria-hidden', - 'true' - ) + expect(chat).toHaveAttribute("aria-current", "page"); + expect(knowledge).not.toHaveAttribute("aria-current"); + for (const button of within(navigation).getAllByRole("button")) { + expect(button.querySelector("svg")).toHaveAttribute( + "aria-hidden", + "true", + ); } - fireEvent.click(knowledge) - expect(knowledge).toHaveAttribute('aria-current', 'page') - expect(chat).not.toHaveAttribute('aria-current') - }) + fireEvent.click(knowledge); + expect(knowledge).toHaveAttribute("aria-current", "page"); + expect(chat).not.toHaveAttribute("aria-current"); + }); - it('collapses the primary sidebar into an inert narrow-window overlay', async () => { - const originalWidth = window.innerWidth - const { container } = render() - const sidebar = container.querySelector('.sidebar') - const workspace = container.querySelector('.workspace') - expect(sidebar).not.toBeNull() - expect(workspace).not.toBeNull() + it("collapses the primary sidebar into an inert narrow-window overlay", async () => { + const originalWidth = window.innerWidth; + const { container } = render(); + const sidebar = container.querySelector(".sidebar"); + const workspace = container.querySelector(".workspace"); + expect(sidebar).not.toBeNull(); + expect(workspace).not.toBeNull(); if (!sidebar || !workspace) { - return + return; } try { - Object.defineProperty(window, 'innerWidth', { + Object.defineProperty(window, "innerWidth", { configurable: true, - value: 680 - }) - window.dispatchEvent(new Event('resize')) + value: 680, + }); + window.dispatchEvent(new Event("resize")); - await waitFor(() => - expect(sidebar).toHaveClass('sidebar--closed') - ) - expect(sidebar).toHaveAttribute('aria-hidden', 'true') - expect(sidebar).toHaveAttribute('inert') + await waitFor(() => expect(sidebar).toHaveClass("sidebar--closed")); + expect(sidebar).toHaveAttribute("aria-hidden", "true"); + expect(sidebar).toHaveAttribute("inert"); - const toggle = screen.getByRole('button', { name: '切换侧栏' }) - fireEvent.click(toggle) - expect(sidebar).not.toHaveClass('sidebar--closed') - expect(sidebar).toHaveAttribute('aria-hidden', 'false') - expect(sidebar).toHaveAttribute('aria-modal', 'true') - expect(sidebar).not.toHaveAttribute('inert') - expect(workspace).toHaveAttribute('aria-hidden', 'true') - expect(workspace).toHaveAttribute('inert') + const toggle = screen.getByRole("button", { name: "切换侧栏" }); + fireEvent.click(toggle); + expect(sidebar).not.toHaveClass("sidebar--closed"); + expect(sidebar).toHaveAttribute("aria-hidden", "false"); + expect(sidebar).toHaveAttribute("aria-modal", "true"); + expect(sidebar).not.toHaveAttribute("inert"); + expect(workspace).toHaveAttribute("aria-hidden", "true"); + expect(workspace).toHaveAttribute("inert"); await waitFor(() => expect( - within(sidebar).getByRole('button', { name: '对话' }) - ).toHaveFocus() - ) + within(sidebar).getByRole("button", { name: "对话" }), + ).toHaveFocus(), + ); - fireEvent.click( - within(sidebar).getByRole('button', { name: '知识库' }) - ) - await waitFor(() => expect(toggle).toHaveFocus()) - expect(sidebar).toHaveClass('sidebar--closed') - expect(workspace).not.toHaveAttribute('aria-hidden') - expect(workspace).not.toHaveAttribute('inert') + fireEvent.click(within(sidebar).getByRole("button", { name: "知识库" })); + await waitFor(() => expect(toggle).toHaveFocus()); + expect(sidebar).toHaveClass("sidebar--closed"); + expect(workspace).not.toHaveAttribute("aria-hidden"); + expect(workspace).not.toHaveAttribute("inert"); } finally { - Object.defineProperty(window, 'innerWidth', { + Object.defineProperty(window, "innerWidth", { configurable: true, - value: originalWidth - }) + value: originalWidth, + }); } - }) + }); - it('resizes and remembers the docked primary sidebar width', () => { - const originalWidth = window.innerWidth - Object.defineProperty(window, 'innerWidth', { + it("resizes and remembers the docked primary sidebar width", () => { + const originalWidth = window.innerWidth; + Object.defineProperty(window, "innerWidth", { configurable: true, - value: 1200 - }) - const { container, unmount } = render() + value: 1200, + }); + const { container, unmount } = render(); try { - const sidebar = - container.querySelector('.sidebar') - const separator = screen.getByRole('separator', { - name: '调整主侧栏宽度' - }) - const setPointerCapture = vi.fn() - const releasePointerCapture = vi.fn() + const sidebar = container.querySelector(".sidebar"); + const separator = screen.getByRole("separator", { + name: "调整主侧栏宽度", + }); + const setPointerCapture = vi.fn(); + const releasePointerCapture = vi.fn(); Object.defineProperties(separator, { setPointerCapture: { value: setPointerCapture }, hasPointerCapture: { value: () => true }, - releasePointerCapture: { value: releasePointerCapture } - }) + releasePointerCapture: { value: releasePointerCapture }, + }); - expect( - sidebar?.style.getPropertyValue('--primary-sidebar-width') - ).toBe('278px') - expect(separator).toHaveAttribute('aria-valuemin', '220') - expect(separator).toHaveAttribute('aria-valuemax', '420') + expect(sidebar?.style.getPropertyValue("--primary-sidebar-width")).toBe( + "278px", + ); + expect(separator).toHaveAttribute("aria-valuemin", "220"); + expect(separator).toHaveAttribute("aria-valuemax", "420"); fireEvent.pointerDown(separator, { button: 0, clientX: 320, - pointerId: 17 - }) + pointerId: 17, + }); fireEvent.pointerMove(separator, { clientX: 400, - pointerId: 17 - }) + pointerId: 17, + }); - expect(setPointerCapture).toHaveBeenCalledWith(17) - expect(sidebar).toHaveClass('sidebar--resizing') - expect( - sidebar?.style.getPropertyValue('--primary-sidebar-width') - ).toBe('400px') + expect(setPointerCapture).toHaveBeenCalledWith(17); + expect(sidebar).toHaveClass("sidebar--resizing"); + expect(sidebar?.style.getPropertyValue("--primary-sidebar-width")).toBe( + "400px", + ); - fireEvent.pointerUp(separator, { pointerId: 17 }) - expect(releasePointerCapture).toHaveBeenCalledWith(17) - expect(sidebar).not.toHaveClass('sidebar--resizing') - expect( - localStorage.getItem('goodbuddy.primary-sidebar-width.v1') - ).toBe('400') + fireEvent.pointerUp(separator, { pointerId: 17 }); + expect(releasePointerCapture).toHaveBeenCalledWith(17); + expect(sidebar).not.toHaveClass("sidebar--resizing"); + expect(localStorage.getItem("goodbuddy.primary-sidebar-width.v1")).toBe( + "400", + ); - fireEvent.keyDown(separator, { key: 'Home' }) - expect( - sidebar?.style.getPropertyValue('--primary-sidebar-width') - ).toBe('220px') + fireEvent.keyDown(separator, { key: "Home" }); + expect(sidebar?.style.getPropertyValue("--primary-sidebar-width")).toBe( + "220px", + ); - fireEvent.keyDown(separator, { key: 'ArrowRight' }) - expect( - sidebar?.style.getPropertyValue('--primary-sidebar-width') - ).toBe('236px') + fireEvent.keyDown(separator, { key: "ArrowRight" }); + expect(sidebar?.style.getPropertyValue("--primary-sidebar-width")).toBe( + "236px", + ); - fireEvent.keyDown(separator, { key: 'End' }) - expect( - sidebar?.style.getPropertyValue('--primary-sidebar-width') - ).toBe('420px') - expect(separator).toHaveAttribute('aria-valuenow', '420') + fireEvent.keyDown(separator, { key: "End" }); + expect(sidebar?.style.getPropertyValue("--primary-sidebar-width")).toBe( + "420px", + ); + expect(separator).toHaveAttribute("aria-valuenow", "420"); - unmount() - const persistedRender = render() + unmount(); + const persistedRender = render(); expect( persistedRender.container - .querySelector('.sidebar') - ?.style.getPropertyValue('--primary-sidebar-width') - ).toBe('420px') + .querySelector(".sidebar") + ?.style.getPropertyValue("--primary-sidebar-width"), + ).toBe("420px"); } finally { - Object.defineProperty(window, 'innerWidth', { + Object.defineProperty(window, "innerWidth", { configurable: true, - value: originalWidth - }) + value: originalWidth, + }); } - }) + }); - it('docks the assistant workbar without obscuring a narrow workspace', async () => { - const originalWidth = window.innerWidth - const { container } = render() + it("docks the assistant workbar without obscuring a narrow workspace", async () => { + const originalWidth = window.innerWidth; + const { container } = render(); try { - Object.defineProperty(window, 'innerWidth', { + Object.defineProperty(window, "innerWidth", { configurable: true, - value: 680 - }) - window.dispatchEvent(new Event('resize')) + value: 680, + }); + window.dispatchEvent(new Event("resize")); await waitFor(() => - expect(container.querySelector('.sidebar')) - .toHaveClass('sidebar--closed') - ) - fireEvent.click(screen.getByLabelText('切换助手工作栏')) + expect(container.querySelector(".sidebar")).toHaveClass( + "sidebar--closed", + ), + ); + fireEvent.click(screen.getByLabelText("切换助手工作栏")); - const workbar = screen.getByRole('complementary', { - name: '助手工作栏' - }) - expect(workbar).toHaveClass('assistant-sidebar--open') - expect(workbar.parentElement).toHaveClass('app-content') + const workbar = screen.getByRole("complementary", { + name: "助手工作栏", + }); + expect(workbar).toHaveClass("assistant-sidebar--open"); + expect(workbar.parentElement).toHaveClass("app-content"); expect( - workbar.parentElement?.querySelector('main.workspace') - ).toBeInTheDocument() - expect(workbar).not.toHaveAttribute('aria-modal') - expect(container.querySelector('main')).not.toHaveAttribute( - 'inert' - ) + workbar.parentElement?.querySelector("main.workspace"), + ).toBeInTheDocument(); + expect(workbar).not.toHaveAttribute("aria-modal"); + expect(container.querySelector("main")).not.toHaveAttribute("inert"); expect( - container.querySelector('.assistant-sidebar-backdrop') - ).not.toBeInTheDocument() + container.querySelector(".assistant-sidebar-backdrop"), + ).not.toBeInTheDocument(); } finally { - Object.defineProperty(window, 'innerWidth', { + Object.defineProperty(window, "innerWidth", { configurable: true, - value: originalWidth - }) + value: originalWidth, + }); } - }) + }); - it('opens Magic Notes as a global first-class workspace', async () => { + it("opens Magic Notes as a global first-class workspace", async () => { api.updates = { getSettings: vi.fn(async () => ({ checkUpdatesOnStartup: false, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: true, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), updateSettings: vi.fn(async () => ({ checkUpdatesOnStartup: false, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: true, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), check: vi.fn(), openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; try { - render() - await screen.findByText('项目：默认项目') - const magicNotesEntry = await screen.findByRole('button', { - name: '魔法笔记' - }) + render(); + await screen.findByText("项目：默认项目"); + const magicNotesEntry = await screen.findByRole("button", { + name: "魔法笔记", + }); - fireEvent.click(magicNotesEntry) + fireEvent.click(magicNotesEntry); expect( await screen.findByRole( - 'heading', - { name: '魔法笔记' }, - { timeout: 3000 } - ) - ).toBeInTheDocument() - expect(screen.getByText('全局')).toBeInTheDocument() + "heading", + { name: "魔法笔记" }, + { timeout: 3000 }, + ), + ).toBeInTheDocument(); + expect(screen.getByText("全局")).toBeInTheDocument(); expect( - screen.getByRole('button', { name: '新建笔记' }) - ).toBeInTheDocument() - await waitFor(() => - expect(api.magicNotes.list).toHaveBeenCalled() - ) - expect( - screen.queryByLabelText('切换助手工作栏') - ).toBeInTheDocument() + screen.getByRole("button", { name: "新建笔记" }), + ).toBeInTheDocument(); + await waitFor(() => expect(api.magicNotes.list).toHaveBeenCalled()); + expect(screen.queryByLabelText("切换助手工作栏")).toBeInTheDocument(); } finally { - delete api.updates + delete api.updates; } - }) + }); - it('shows and refreshes the incomplete Magic Notes todo count', async () => { + it("shows and refreshes the incomplete Magic Notes todo count", async () => { vi.mocked(api.magicNotes.getTodoStatus).mockResolvedValueOnce({ - incompleteCount: 4 - }) + incompleteCount: 4, + }); api.updates = { getSettings: vi.fn(async () => ({ checkUpdatesOnStartup: false, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: true, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), updateSettings: vi.fn(), check: vi.fn(), openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; try { - render() + render(); - expect( - await screen.findByLabelText('4 个未完成待办') - ).toHaveTextContent('4') + expect(await screen.findByLabelText("4 个未完成待办")).toHaveTextContent( + "4", + ); vi.mocked(api.magicNotes.getTodoStatus).mockResolvedValueOnce({ - incompleteCount: 0 - }) + incompleteCount: 0, + }); await act(async () => { - magicTodoStatusChangedListener?.() - }) + magicTodoStatusChangedListener?.(); + }); await waitFor(() => expect( - screen.queryByLabelText('4 个未完成待办') - ).not.toBeInTheDocument() - ) + screen.queryByLabelText("4 个未完成待办"), + ).not.toBeInTheDocument(), + ); vi.mocked(api.magicNotes.getTodoStatus).mockResolvedValueOnce({ - incompleteCount: 125 - }) + incompleteCount: 125, + }); await act(async () => { - magicTodoStatusChangedListener?.() - }) + magicTodoStatusChangedListener?.(); + }); expect( - await screen.findByLabelText('125 个未完成待办') - ).toHaveTextContent('99+') + await screen.findByLabelText("125 个未完成待办"), + ).toHaveTextContent("99+"); } finally { - delete api.updates + delete api.updates; } - }) + }); - it('hides the todo count when its Magic Notes setting is off', async () => { + it("hides the todo count when its Magic Notes setting is off", async () => { vi.mocked(api.magicNotes.getTodoStatus).mockResolvedValue({ - incompleteCount: 4 - }) + incompleteCount: 4, + }); api.updates = { getSettings: vi.fn(async () => ({ checkUpdatesOnStartup: false, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: true, magicNotesShowIncompleteTodoCount: false, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), updateSettings: vi.fn(), check: vi.fn(), openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; try { - render() + render(); expect( - await screen.findByRole('button', { name: '魔法笔记' }) - ).toBeInTheDocument() - expect( - screen.queryByLabelText('4 个未完成待办') - ).not.toBeInTheDocument() - expect(api.magicNotes.getTodoStatus).not.toHaveBeenCalled() + await screen.findByRole("button", { name: "魔法笔记" }), + ).toBeInTheDocument(); + expect(screen.queryByLabelText("4 个未完成待办")).not.toBeInTheDocument(); + expect(api.magicNotes.getTodoStatus).not.toHaveBeenCalled(); } finally { - delete api.updates + delete api.updates; } - }) + }); - it('hides Magic Notes when the platform feature is disabled', async () => { + it("hides Magic Notes when the platform feature is disabled", async () => { api.updates = { getSettings: vi.fn(async () => ({ checkUpdatesOnStartup: false, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: false, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), updateSettings: vi.fn(async () => ({ checkUpdatesOnStartup: false, - updateSource: 'github' as const, - modelDownloadSource: 'modelscope' as const, + updateSource: "github" as const, + modelDownloadSource: "modelscope" as const, localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: false, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate' as const, - magicNoteCommentFormat: 'combined' as const + magicNoteCommentMode: "immediate" as const, + magicNoteCommentFormat: "combined" as const, })), check: vi.fn(), openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; try { - render() + render(); await waitFor(() => expect( - screen.queryByRole('button', { name: '魔法笔记' }) - ).not.toBeInTheDocument() - ) + screen.queryByRole("button", { name: "魔法笔记" }), + ).not.toBeInTheDocument(), + ); } finally { - delete api.updates + delete api.updates; } - }) + }); - it('hides Magic Notes by default without an explicit setting', () => { - render() + it("hides Magic Notes by default without an explicit setting", () => { + render(); expect( - screen.queryByRole('button', { name: '魔法笔记' }) - ).not.toBeInTheDocument() - }) + screen.queryByRole("button", { name: "魔法笔记" }), + ).not.toBeInTheDocument(); + }); - it('renders Agent HTML by default and applies the platform switch immediately', async () => { - const conversationId = - '00000000-0000-4000-8000-000000000471' + it("renders Agent HTML by default and applies the platform switch immediately", async () => { + const conversationId = "00000000-0000-4000-8000-000000000471"; vi.mocked(api.conversations.list).mockResolvedValueOnce([ { id: conversationId, projectId, - title: 'HTML 预览会话', + title: "HTML 预览会话", updatedAt: 1_775_000_000_000, messages: [ { - id: '00000000-0000-4000-8000-000000000472', - role: 'assistant', + id: "00000000-0000-4000-8000-000000000472", + role: "assistant", content: `\`\`\`html

Rendered dashboard

\`\`\``, createdAt: 1_775_000_000_000, - state: 'complete' - } - ] - } - ]) + state: "complete", + }, + ], + }, + ]); let applicationSettings: ApplicationSettings = { checkUpdatesOnStartup: false, - updateSource: 'github', - modelDownloadSource: 'modelscope', + updateSource: "github", + modelDownloadSource: "modelscope", localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: false, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate', - magicNoteCommentFormat: 'combined' - } + magicNoteCommentMode: "immediate", + magicNoteCommentFormat: "combined", + }; const updateSettings = vi.fn< - NonNullable['updateSettings'] + NonNullable["updateSettings"] >(async (input) => { - applicationSettings = { ...applicationSettings, ...input } - return { ...applicationSettings } - }) + applicationSettings = { ...applicationSettings, ...input }; + return { ...applicationSettings }; + }); api.updates = { getSettings: vi.fn(async () => ({ ...applicationSettings })), updateSettings, check: vi.fn(), openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; - render() + render(); expect( - await screen.findByTitle('Agent 回复 HTML 静态预览') - ).toHaveAttribute('sandbox', '') + await screen.findByTitle("Agent 回复 HTML 静态预览"), + ).toHaveAttribute("sandbox", ""); - fireEvent.click( - await screen.findByRole('button', { name: /本地工作区/u }) - ) - fireEvent.click( - await screen.findByRole('tab', { name: '平台功能' }) - ) - const toggle = await screen.findByRole('switch', { - name: '在会话中渲染 HTML' - }) - expect(toggle).toBeChecked() - fireEvent.click(toggle) + fireEvent.click(await screen.findByRole("button", { name: /本地工作区/u })); + fireEvent.click(await screen.findByRole("tab", { name: "平台功能" })); + const toggle = await screen.findByRole("switch", { + name: "在会话中渲染 HTML", + }); + expect(toggle).toBeChecked(); + fireEvent.click(toggle); await waitFor(() => expect(updateSettings).toHaveBeenCalledWith({ - conversationHtmlRenderingEnabled: false - }) - ) + conversationHtmlRenderingEnabled: false, + }), + ); await waitFor(() => expect( - screen.queryByTitle('Agent 回复 HTML 静态预览') - ).not.toBeInTheDocument() - ) - expect(screen.getByText(/Rendered dashboard/u)).toBeInTheDocument() - }) + screen.queryByTitle("Agent 回复 HTML 静态预览"), + ).not.toBeInTheDocument(), + ); + expect(screen.getByText(/Rendered dashboard/u)).toBeInTheDocument(); + }); - it('keeps platform-feature switches in Settings without navigating', async () => { + it("keeps platform-feature switches in Settings without navigating", async () => { let applicationSettings: ApplicationSettings = { checkUpdatesOnStartup: false, - updateSource: 'github', - modelDownloadSource: 'modelscope', + updateSource: "github", + modelDownloadSource: "modelscope", localToolEnvironment: defaultLocalToolEnvironmentSettings, conversationHtmlRenderingEnabled: true, remoteProjectsEnabled: false, magicNotesEnabled: false, magicNotesShowIncompleteTodoCount: true, - magicNoteCommentMode: 'immediate', - magicNoteCommentFormat: 'combined' - } + magicNoteCommentMode: "immediate", + magicNoteCommentFormat: "combined", + }; api.updates = { getSettings: vi.fn(async () => ({ ...applicationSettings })), updateSettings: vi.fn(async (input) => { applicationSettings = { ...applicationSettings, - ...input - } - return { ...applicationSettings } + ...input, + }; + return { ...applicationSettings }; }), check: vi.fn(), openReleasePage: vi.fn(async () => {}), - onResult: vi.fn(() => () => {}) - } + onResult: vi.fn(() => () => {}), + }; try { - const { container } = render() + const { container } = render(); fireEvent.click( - await screen.findByRole('button', { - name: /本地工作区/u - }) - ) - await screen.findByRole('heading', { name: '设置中心' }) - fireEvent.click( - screen.getByRole('tab', { name: '平台功能' }) - ) - fireEvent.click( - await screen.findByRole('tab', { name: '魔法笔记' }) - ) - const toggle = await screen.findByRole('switch', { - name: '显示魔法笔记入口' - }) + await screen.findByRole("button", { + name: /本地工作区/u, + }), + ); + await screen.findByRole("heading", { name: "设置中心" }); + fireEvent.click(screen.getByRole("tab", { name: "平台功能" })); + fireEvent.click(await screen.findByRole("tab", { name: "魔法笔记" })); + const toggle = await screen.findByRole("switch", { + name: "显示魔法笔记入口", + }); - fireEvent.click(toggle) - await waitFor(() => expect(toggle).toBeChecked()) + fireEvent.click(toggle); + await waitFor(() => expect(toggle).toBeChecked()); expect( - screen.getByRole('heading', { name: '设置中心' }) - ).toBeInTheDocument() + screen.getByRole("heading", { name: "设置中心" }), + ).toBeInTheDocument(); expect( - screen.getByRole('button', { name: '魔法笔记' }) - ).toBeInTheDocument() + screen.getByRole("button", { name: "魔法笔记" }), + ).toBeInTheDocument(); expect( - container.querySelector('.magic-notes-page') - ).not.toBeInTheDocument() + container.querySelector(".magic-notes-page"), + ).not.toBeInTheDocument(); - fireEvent.click(toggle) - await waitFor(() => expect(toggle).not.toBeChecked()) + fireEvent.click(toggle); + await waitFor(() => expect(toggle).not.toBeChecked()); expect( - screen.getByRole('heading', { name: '设置中心' }) - ).toBeInTheDocument() + screen.getByRole("heading", { name: "设置中心" }), + ).toBeInTheDocument(); expect( - screen.queryByRole('button', { name: '魔法笔记' }) - ).not.toBeInTheDocument() + screen.queryByRole("button", { name: "魔法笔记" }), + ).not.toBeInTheDocument(); } finally { - delete api.updates + delete api.updates; } - }) + }); - it('gives the knowledge workspace the full content width', async () => { - render() + it("gives the knowledge workspace the full content width", async () => { + render(); - fireEvent.click(screen.getByRole('button', { name: '知识库' })) + fireEvent.click(screen.getByRole("button", { name: "知识库" })); + expect(await screen.findByLabelText("知识工作区")).toBeInTheDocument(); + expect(screen.queryByLabelText("切换助手工作栏")).toBeInTheDocument(); expect( - await screen.findByLabelText('知识工作区') - ).toBeInTheDocument() - expect( - screen.queryByLabelText('切换助手工作栏') - ).toBeInTheDocument() - expect( - screen.queryByRole('button', { name: /^专家角色：/u }) - ).not.toBeInTheDocument() - }) + screen.queryByRole("button", { name: /^专家角色：/u }), + ).not.toBeInTheDocument(); + }); - it('keeps Smart Heartbeat available when the runtime is not configured', async () => { + it("keeps Smart Heartbeat available when the runtime is not configured", async () => { vi.mocked(api.agent.getStatus).mockResolvedValue({ - id: 'setup', - label: '需要配置模型', + id: "setup", + label: "需要配置模型", available: false, supportsToolExecution: false, - detail: '请配置模型' - }) - render() + detail: "请配置模型", + }); + render(); expect( - await screen.findByRole('heading', { name: '设置中心' }) - ).toBeInTheDocument() - await waitFor(() => - expect(api.agent.getStatus).toHaveBeenCalledOnce() - ) - fireEvent.click( - screen.getByRole('button', { name: '智能心跳' }) - ) + await screen.findByRole("heading", { name: "设置中心" }), + ).toBeInTheDocument(); + await waitFor(() => expect(api.agent.getStatus).toHaveBeenCalledOnce()); + fireEvent.click(screen.getByRole("button", { name: "智能心跳" })); expect( - await screen.findByRole('heading', { name: '智能心跳' }) - ).toBeInTheDocument() - expect(api.agent.getStatus).toHaveBeenCalledOnce() - }) -}) + await screen.findByRole("heading", { name: "智能心跳" }), + ).toBeInTheDocument(); + expect(api.agent.getStatus).toHaveBeenCalledOnce(); + }); +}); diff --git a/src/renderer/src/App.tsx b/src/renderer/src/App.tsx index 54ae8a41..53163fb4 100644 --- a/src/renderer/src/App.tsx +++ b/src/renderer/src/App.tsx @@ -38,8 +38,8 @@ import { Sun, TerminalSquare, Trash2, - X -} from 'lucide-react' + X, +} from "lucide-react"; import { Component, Suspense, @@ -52,10 +52,10 @@ import { useRef, useState, type ReactNode, - type SetStateAction -} from 'react' -import { useTranslation } from 'react-i18next' -import type { TFunction } from 'i18next' + type SetStateAction, +} from "react"; +import { useTranslation } from "react-i18next"; +import type { TFunction } from "i18next"; import type { ApprovalDecision, AgentEvent, @@ -72,27 +72,27 @@ import type { RuntimeCustomizationSettings, RuntimeNativeSnapshot, RuntimeControl, - RuntimeSettings -} from '../../shared/contracts' + RuntimeSettings, +} from "../../shared/contracts"; import { defaultContextCompressionSettings, - maximumPastedImageBytes -} from '../../shared/contracts' + maximumPastedImageBytes, +} from "../../shared/contracts"; import { buildConversationSummaryHistory, estimatedContextRequestOverheadTokens, estimateMessagesTokens, - getEffectiveContextTriggerTokens -} from '../../shared/context-window' + getEffectiveContextTriggerTokens, +} from "../../shared/context-window"; import { agentRuntimeSelectionKey, agentRuntimeSelectionSchema, - type AgentRuntimeSelection -} from '../../shared/runtime-selection-contracts' + type AgentRuntimeSelection, +} from "../../shared/runtime-selection-contracts"; import { getDefaultRuntimeSelection, - getRuntimeSelectionForProvider -} from './runtime-selection' + getRuntimeSelectionForProvider, +} from "./runtime-selection"; import type { ActivityHistorySnapshot, AssistantProject, @@ -118,8 +118,8 @@ import type { ProjectCreateInput, InteractiveWorkMode, ProjectChannel, - WorkspaceChanges -} from '../../shared/assistant-contracts' + WorkspaceChanges, +} from "../../shared/assistant-contracts"; import { assistantIdSchema, conversationAttachmentSchema, @@ -130,153 +130,144 @@ import { conversationSubagentActivitySchema, interactiveWorkModes, normalizeInteractiveWorkMode, - projectChannelLabels -} from '../../shared/assistant-contracts' + projectChannelLabels, +} from "../../shared/assistant-contracts"; import { ChatTimeline, type ImageViewerItem, type Message, type SubagentActivity, - type ToolActivity -} from './ChatTimeline' + type ToolActivity, +} from "./ChatTimeline"; import { clearLegacyActivityHistory, loadLegacyActivityHistory, mergeActivityRecords, reconcileActivityRecords, upsertActivityRecord, - type ActivityRecord -} from './activity-store' + type ActivityRecord, +} from "./activity-store"; import { KnowledgeCitationDialog, - type KnowledgeCitationContextView -} from './KnowledgeCitationDialog' + type KnowledgeCitationContextView, +} from "./KnowledgeCitationDialog"; import { DestructiveConfirmActions, EmptyState, PageShell, SegmentedControl, - ScopeBadge -} from './WorkspacePrimitives' -import { - ProjectSwitcher -} from './ProjectSwitcher' + ScopeBadge, +} from "./WorkspacePrimitives"; +import { ProjectSwitcher } from "./ProjectSwitcher"; import { RightAssistantSidebar, type AssistantSidebarTab, type PendingSidebarApproval, - type SidebarArtifact -} from './RightAssistantSidebar' + type SidebarArtifact, +} from "./RightAssistantSidebar"; import { CustomTaskDialog, - type CustomTaskDestination -} from './CustomTaskDialog' -import { ConversationTaskStrip } from './ConversationTaskStrip' -import { ConversationInputQueue } from './ConversationInputQueue' -import { OverflowMarquee } from './OverflowMarquee' -import { findTaskSchedule } from './TaskScheduleActions' -import type { SettingsCategoryId } from './settings-categories' -import type { SettingsLeaveRequester } from './SettingsPanel' -import type { GlobalShortcutSettingsSnapshot } from '../../shared/shortcut' -import goodbuddyDarkIcon from './assets/goodbuddy-dark.png' -import goodbuddyLightIcon from './assets/goodbuddy-light.png' -import { - loadBrandingPreferences, - saveBrandingPreferences -} from './branding' -import { BrandLockup } from './BrandLockup' + type CustomTaskDestination, +} from "./CustomTaskDialog"; +import { ConversationTaskStrip } from "./ConversationTaskStrip"; +import { ConversationInputQueue } from "./ConversationInputQueue"; +import { OverflowMarquee } from "./OverflowMarquee"; +import { findTaskSchedule } from "./TaskScheduleActions"; +import type { SettingsCategoryId } from "./settings-categories"; +import type { SettingsLeaveRequester } from "./SettingsPanel"; +import type { GlobalShortcutSettingsSnapshot } from "../../shared/shortcut"; +import goodbuddyDarkIcon from "./assets/goodbuddy-dark.png"; +import goodbuddyLightIcon from "./assets/goodbuddy-light.png"; +import { loadBrandingPreferences, saveBrandingPreferences } from "./branding"; +import { BrandLockup } from "./BrandLockup"; import { applyAppearanceTheme, loadAppearanceTheme, resolveAppearanceTheme, saveAppearanceTheme, - type AppearanceTheme -} from './theme' + type AppearanceTheme, +} from "./theme"; import { describeSpeechRecognitionError, getSpeechRecognitionConstructor, prepareSpeechRecognition, startPcmRecording, - type PcmRecording -} from './speech-recognition' + type PcmRecording, +} from "./speech-recognition"; import type { AppNotificationInput, - AppNotificationTone -} from './notifications' -import type { ReleaseNotesSnapshot } from '../../shared/release-notes-contracts' -import type { - RemoteProjectRecoveryState -} from '../../shared/remote-project-recovery-contracts' -import { ReleaseNotesDialog } from './ReleaseNotesDialog' -import { scheduleIdleRoutePreload } from './idle-route-preload' -import { createPreloadableComponent } from './preloadable-component' + AppNotificationTone, +} from "./notifications"; +import type { ReleaseNotesSnapshot } from "../../shared/release-notes-contracts"; +import type { RemoteProjectRecoveryState } from "../../shared/remote-project-recovery-contracts"; +import { ReleaseNotesDialog } from "./ReleaseNotesDialog"; +import { scheduleIdleRoutePreload } from "./idle-route-preload"; +import { createPreloadableComponent } from "./preloadable-component"; import { formatConversationListTime, - type TimeFormatLocale -} from './time-format' -import { formatMediumDateTime } from './locale-formatters' -import { formatCompactTokens } from './token-format' + type TimeFormatLocale, +} from "./time-format"; +import { formatMediumDateTime } from "./locale-formatters"; +import { formatCompactTokens } from "./token-format"; import { filterKeepAliveEntries, pruneKeepAliveEntries, touchAndPruneKeepAliveEntries, - type KeepAliveCacheEntry -} from './keep-alive-cache' -import { activateModalFocus, trapTabFocus } from './dialog-focus' + type KeepAliveCacheEntry, +} from "./keep-alive-cache"; +import { activateModalFocus, trapTabFocus } from "./dialog-focus"; import { displayErrorMessage, - displayNetworkAwareErrorMessage -} from './error-message' -import { getProjectDisplayText } from './project-display' + displayNetworkAwareErrorMessage, +} from "./error-message"; +import { getProjectDisplayText } from "./project-display"; const knowledgeWorkspaceRoute = createPreloadableComponent( - () => import('./KnowledgeWorkspace'), - (module) => module.KnowledgeWorkspace -) + () => import("./KnowledgeWorkspace"), + (module) => module.KnowledgeWorkspace, +); const heartbeatCenterRoute = createPreloadableComponent( - () => import('./HeartbeatCenter'), - (module) => module.HeartbeatCenter -) + () => import("./HeartbeatCenter"), + (module) => module.HeartbeatCenter, +); const magicNotesWorkspaceRoute = createPreloadableComponent( - () => import('./MagicNotesWorkspace'), - (module) => module.MagicNotesWorkspace -) + () => import("./MagicNotesWorkspace"), + (module) => module.MagicNotesWorkspace, +); const settingsPanelRoute = createPreloadableComponent( - () => import('./SettingsPanel'), - (module) => module.SettingsPanel -) + () => import("./SettingsPanel"), + (module) => module.SettingsPanel, +); const activityPanelRoute = createPreloadableComponent( - () => import('./ActivityPanel'), - (module) => module.ActivityPanel -) -const idleRouteModuleLoaders = [ - heartbeatCenterRoute.preload -] as const + () => import("./ActivityPanel"), + (module) => module.ActivityPanel, +); +const idleRouteModuleLoaders = [heartbeatCenterRoute.preload] as const; -const KnowledgeWorkspace = knowledgeWorkspaceRoute.Component -const HeartbeatCenter = heartbeatCenterRoute.Component -const MagicNotesWorkspace = magicNotesWorkspaceRoute.Component -const SettingsPanel = settingsPanelRoute.Component -const ActivityPanel = activityPanelRoute.Component +const KnowledgeWorkspace = knowledgeWorkspaceRoute.Component; +const HeartbeatCenter = heartbeatCenterRoute.Component; +const MagicNotesWorkspace = magicNotesWorkspaceRoute.Component; +const SettingsPanel = settingsPanelRoute.Component; +const ActivityPanel = activityPanelRoute.Component; -const messageRenderBatchSize = 80 -const conversationPersistenceIntervalMs = 500 -const conversationSearchSnapshotDelayMs = 250 -const keepAliveExpirationMs = 60 * 60 * 1_000 -const keepAliveSweepIntervalMs = 5 * 60 * 1_000 -const maximumCachedConversations = 12 -const recentCachedConversations = 5 -const maximumCachedWorkspaceViews = 4 -const recentCachedWorkspaceViews = 3 +const messageRenderBatchSize = 80; +const conversationPersistenceIntervalMs = 500; +const conversationSearchSnapshotDelayMs = 250; +const keepAliveExpirationMs = 60 * 60 * 1_000; +const keepAliveSweepIntervalMs = 5 * 60 * 1_000; +const maximumCachedConversations = 12; +const recentCachedConversations = 5; +const maximumCachedWorkspaceViews = 4; +const recentCachedWorkspaceViews = 3; function sameConversationQueueItems( current: ConversationQueueItem[], - next: ConversationQueueItem[] + next: ConversationQueueItem[], ): boolean { return ( current.length === next.length && current.every((item, index) => { - const candidate = next[index] + const candidate = next[index]; return ( candidate !== undefined && item.id === candidate.id && @@ -287,40 +278,36 @@ function sameConversationQueueItems( item.scheduleRunId === candidate.scheduleRunId && item.scheduleId === candidate.scheduleId && item.taskId === candidate.taskId - ) + ); }) - ) + ); } type AppNotification = { - id: string - message: string - tone: AppNotificationTone - revision: number -} + id: string; + message: string; + tone: AppNotificationTone; + revision: number; +}; -type AppNotificationAction = AppNotificationInput | { dismiss: string } +type AppNotificationAction = AppNotificationInput | { dismiss: string }; function appNotificationReducer( current: AppNotification[], - action: AppNotificationAction + action: AppNotificationAction, ): AppNotification[] { - if ('dismiss' in action) { - return current.filter( - (notification) => notification.id !== action.dismiss - ) + if ("dismiss" in action) { + return current.filter((notification) => notification.id !== action.dismiss); } - const message = action.message.slice(0, 2_000) - const id = action.dedupeKey ?? `${action.tone}:${message}` - const existing = current.find( - (notification) => notification.id === id - ) + const message = action.message.slice(0, 2_000); + const id = action.dedupeKey ?? `${action.tone}:${message}`; + const existing = current.find((notification) => notification.id === id); if ( - existing?.tone === 'error' && - action.tone === 'error' && + existing?.tone === "error" && + action.tone === "error" && existing.message === message ) { - return current + return current; } const updated = [ ...current.filter((notification) => notification.id !== id), @@ -328,23 +315,19 @@ function appNotificationReducer( id, message, tone: action.tone, - revision: (existing?.revision ?? 0) + 1 - } - ] + revision: (existing?.revision ?? 0) + 1, + }, + ]; const errors = updated.filter( - (notification) => notification.tone === 'error' - ) + (notification) => notification.tone === "error", + ); const transient = updated - .filter((notification) => notification.tone !== 'error') - .slice(-4) - return [...errors, ...transient] + .filter((notification) => notification.tone !== "error") + .slice(-4); + return [...errors, ...transient]; } -function RouteLoadingStatus({ - label -}: { - label: string -}): React.JSX.Element { +function RouteLoadingStatus({ label }: { label: string }): React.JSX.Element { return ( - ) + ); } function KeepAliveRoute({ active, children, - route + route, }: { - active: boolean - children: ReactNode - route: string + active: boolean; + children: ReactNode; + route: string; }): React.JSX.Element { return ( - ) + ); } class RouteErrorBoundary extends Component< { children: ReactNode; fallback: ReactNode }, { failed: boolean } > { - state = { failed: false } + state = { failed: false }; static getDerivedStateFromError(): { failed: boolean } { - return { failed: true } + return { failed: true }; } render(): ReactNode { - return this.state.failed ? this.props.fallback : this.props.children + return this.state.failed ? this.props.fallback : this.props.children; } } function RouteLoadError({ message, - reloadLabel + reloadLabel, }: { - message: string - reloadLabel: string + message: string; + reloadLabel: string; }): React.JSX.Element { return (
@@ -411,50 +394,45 @@ function RouteLoadError({ {reloadLabel}
- ) + ); } function AppNotificationItem({ notification, - dispatch + dispatch, }: { - notification: AppNotification - dispatch: React.Dispatch + notification: AppNotification; + dispatch: React.Dispatch; }): React.JSX.Element { - const { t } = useTranslation('app') + const { t } = useTranslation("app"); useEffect(() => { - if (notification.tone === 'error') { - return + if (notification.tone === "error") { + return; } const timeout = window.setTimeout(() => { - dispatch({ dismiss: notification.id }) - }, 4_500) - return () => window.clearTimeout(timeout) - }, [ - dispatch, - notification.id, - notification.revision, - notification.tone - ]) + dispatch({ dismiss: notification.id }); + }, 4_500); + return () => window.clearTimeout(timeout); + }, [dispatch, notification.id, notification.revision, notification.tone]); const label = - notification.tone === 'success' - ? t('notifications.success') - : notification.tone === 'error' - ? t('notifications.error') - : t('notifications.info') + notification.tone === "success" + ? t("notifications.success") + : notification.tone === "error" + ? t("notifications.error") + : t("notifications.info"); const Icon = - notification.tone === 'success' + notification.tone === "success" ? CheckCircle2 - : notification.tone === 'error' + : notification.tone === "error" ? CircleAlert - : Info + : Info; return (
- ) + ); } function AppNotificationViewport({ notifications, - dispatch + dispatch, }: { - notifications: AppNotification[] - dispatch: React.Dispatch + notifications: AppNotification[]; + dispatch: React.Dispatch; }): React.JSX.Element | null { - const { t } = useTranslation('app') + const { t } = useTranslation("app"); if (notifications.length === 0) { - return null + return null; } return (
{notifications.map((notification) => ( @@ -497,49 +475,42 @@ function AppNotificationViewport({ /> ))}
- ) + ); } -function supportsSubagentSmartRouting( - workMode: string -): boolean { - return workMode === 'ask' +function supportsSubagentSmartRouting(workMode: string): boolean { + return workMode === "ask"; } -type Conversation = Omit & { - messages: Message[] -} +type Conversation = Omit & { + messages: Message[]; +}; type ActiveRun = { - conversationId: string - messageId: string - projectId?: string - runtimeSelectionKey: string -} + conversationId: string; + messageId: string; + projectId?: string; + runtimeSelectionKey: string; +}; type WorkspaceView = - | 'chat' - | 'magic-notes' - | 'knowledge' - | 'heartbeat' - | 'activity' - | 'settings' + "chat" | "magic-notes" | "knowledge" | "heartbeat" | "activity" | "settings"; const intentRoutePreloaders: Partial< Record Promise> > = { - 'magic-notes': magicNotesWorkspaceRoute.preload, + "magic-notes": magicNotesWorkspaceRoute.preload, knowledge: knowledgeWorkspaceRoute.preload, activity: activityPanelRoute.preload, - settings: settingsPanelRoute.preload -} + settings: settingsPanelRoute.preload, +}; function preloadWorkspaceRouteOnIntent(view: WorkspaceView): void { - const preload = intentRoutePreloaders[view] + const preload = intentRoutePreloaders[view]; if (preload) { void preload().catch(() => { // Click and programmatic navigation retain their local retry boundary. - }) + }); } } @@ -550,261 +521,246 @@ const emptyTokenUsage: TokenUsageSummary = { output: 0, cacheRead: 0, cacheWrite: 0, - totalTokens: 0 + totalTokens: 0, }, - records: [] -} + records: [], +}; -const storageKey = 'goodbuddy.conversations.v1' +const storageKey = "goodbuddy.conversations.v1"; -const activeProjectStorageKey = 'goodbuddy.active-project.v1' +const activeProjectStorageKey = "goodbuddy.active-project.v1"; -const primarySidebarWidthStorageKey = - 'goodbuddy.primary-sidebar-width.v1' -const defaultPrimarySidebarWidth = 278 -const compactPrimarySidebarWidth = 236 -const minimumPrimarySidebarWidth = 220 -const maximumPrimarySidebarWidth = 420 -const minimumPrimaryWorkspaceWidth = 480 -const primarySidebarKeyboardResizeStep = 16 +const primarySidebarWidthStorageKey = "goodbuddy.primary-sidebar-width.v1"; +const defaultPrimarySidebarWidth = 278; +const compactPrimarySidebarWidth = 236; +const minimumPrimarySidebarWidth = 220; +const maximumPrimarySidebarWidth = 420; +const minimumPrimaryWorkspaceWidth = 480; +const primarySidebarKeyboardResizeStep = 16; -const maxMessageContentLength = 1_000_000 +const maxMessageContentLength = 1_000_000; function getPrimarySidebarWidthLimits(viewportWidth: number): { - minimum: number - maximum: number + minimum: number; + maximum: number; } { - const availableWidth = Math.max(0, Math.floor(viewportWidth)) + const availableWidth = Math.max(0, Math.floor(viewportWidth)); return { minimum: minimumPrimarySidebarWidth, maximum: Math.max( minimumPrimarySidebarWidth, Math.min( maximumPrimarySidebarWidth, - availableWidth - minimumPrimaryWorkspaceWidth - ) - ) - } + availableWidth - minimumPrimaryWorkspaceWidth, + ), + ), + }; } function clampPrimarySidebarWidth( width: number, - viewportWidth: number + viewportWidth: number, ): number { - const limits = getPrimarySidebarWidthLimits(viewportWidth) - return Math.min( - limits.maximum, - Math.max(limits.minimum, Math.round(width)) - ) + const limits = getPrimarySidebarWidthLimits(viewportWidth); + return Math.min(limits.maximum, Math.max(limits.minimum, Math.round(width))); } function loadPrimarySidebarWidth(): number { const fallback = window.innerWidth <= 1020 ? compactPrimarySidebarWidth - : defaultPrimarySidebarWidth + : defaultPrimarySidebarWidth; try { const persistedWidth = Number( - localStorage.getItem(primarySidebarWidthStorageKey) - ) + localStorage.getItem(primarySidebarWidthStorageKey), + ); return clampPrimarySidebarWidth( Number.isFinite(persistedWidth) && persistedWidth > 0 ? persistedWidth : fallback, - window.innerWidth - ) + window.innerWidth, + ); } catch { - return clampPrimarySidebarWidth(fallback, window.innerWidth) + return clampPrimarySidebarWidth(fallback, window.innerWidth); } } function appendMessageContentBlock( blocks: ConversationMessageBlock[] | undefined, - type: 'text' | 'reasoning', - delta: string + type: "text" | "reasoning", + delta: string, ): ConversationMessageBlock[] | undefined { if (!blocks || !delta) { - return blocks + return blocks; } - const current = [...blocks] - const previous = current.at(-1) + const current = [...blocks]; + const previous = current.at(-1); if (previous?.type === type) { current[current.length - 1] = { ...previous, - content: `${previous.content}${delta}`.slice( - 0, - maxMessageContentLength - ) - } - return current + content: `${previous.content}${delta}`.slice(0, maxMessageContentLength), + }; + return current; } current.push({ id: crypto.randomUUID(), type, - content: delta.slice(0, maxMessageContentLength) - }) - return current + content: delta.slice(0, maxMessageContentLength), + }); + return current; } function upsertMessageToolBlock( blocks: ConversationMessageBlock[] | undefined, - tool: ToolActivity + tool: ToolActivity, ): ConversationMessageBlock[] | undefined { if (!blocks) { - return blocks + return blocks; } - const callId = tool.callId + const callId = tool.callId; const index = callId ? blocks.findIndex( - (block) => - block.type === 'tool' && block.tool.callId === callId + (block) => block.type === "tool" && block.tool.callId === callId, ) - : -1 + : -1; if (index >= 0) { return blocks.map((block, blockIndex) => - blockIndex === index && block.type === 'tool' + blockIndex === index && block.type === "tool" ? { ...block, tool } - : block - ) + : block, + ); } return [ ...blocks, { id: crypto.randomUUID(), - type: 'tool', - tool - } - ] + type: "tool", + tool, + }, + ]; } function upsertMessageSubagentBlock( blocks: ConversationMessageBlock[] | undefined, childTaskId: string, - runtimeCallId?: string + runtimeCallId?: string, ): ConversationMessageBlock[] | undefined { if (!blocks) { - return blocks + return blocks; } if ( blocks.some( - (block) => - block.type === 'subagent' && - block.childTaskId === childTaskId + (block) => block.type === "subagent" && block.childTaskId === childTaskId, ) ) { - return blocks + return blocks; } const provisionalIndex = runtimeCallId ? blocks.findIndex( - (block) => - block.type === 'tool' && - block.tool.callId === runtimeCallId + (block) => block.type === "tool" && block.tool.callId === runtimeCallId, ) - : -1 + : -1; if (provisionalIndex >= 0) { return blocks.map((block, index) => index === provisionalIndex ? { id: block.id, - type: 'subagent' as const, - childTaskId + type: "subagent" as const, + childTaskId, } - : block - ) + : block, + ); } return [ ...blocks, { id: crypto.randomUUID(), - type: 'subagent', - childTaskId - } - ] + type: "subagent", + childTaskId, + }, + ]; } function terminalizeMessageToolBlocks( blocks: ConversationMessageBlock[] | undefined, - state: 'failed' | 'cancelled' | 'interrupted' + state: "failed" | "cancelled" | "interrupted", ): ConversationMessageBlock[] | undefined { return blocks?.map((block) => - block.type === 'tool' && - (block.tool.state === 'pending' || block.tool.state === 'running') + block.type === "tool" && + (block.tool.state === "pending" || block.tool.state === "running") ? { ...block, tool: { ...block.tool, - state - } + state, + }, } - : block - ) + : block, + ); } function isErrorRepresentedByFailedTool( tools: ToolActivity[] | undefined, - errorMessage: string + errorMessage: string, ): boolean { return Boolean( tools?.some( (tool) => - tool.state === 'failed' && + tool.state === "failed" && ((tool.error && errorMessage.includes(tool.error)) || - (tool.callId && errorMessage.includes(tool.callId))) - ) - ) + (tool.callId && errorMessage.includes(tool.callId))), + ), + ); } -const chatBottomProximity = 96 +const chatBottomProximity = 96; function createConversation( projectId?: string, runtimeSelection?: AgentRuntimeSelection, - greeting = - '你好，我是 GoodBuddy。你可以直接向我提问、添加本地文件，或使用知识库整理和检索信息。需要我操作文件或调用工具时，请选择合适的 Agent Runtime 和工作模式。' + greeting = "你好，我是 GoodBuddy。你可以直接向我提问、添加本地文件，或使用知识库整理和检索信息。需要我操作文件或调用工具时，请选择合适的 Agent Runtime 和工作模式。", ): Conversation { - const now = Date.now() + const now = Date.now(); return { id: crypto.randomUUID(), projectId, runtimeSelection, knowledgeLibraryIds: [], - knowledgeRetrievalMode: 'auto', - title: '新对话', + knowledgeRetrievalMode: "auto", + title: "新对话", updatedAt: now, messages: [ { id: crypto.randomUUID(), - role: 'assistant', + role: "assistant", content: greeting, createdAt: now, - state: 'complete' - } - ] - } + state: "complete", + }, + ], + }; } function createConversationBranchTitle( sourceTitle: string, - suffix: string + suffix: string, ): string { - const trailing = ` · ${suffix}` + const trailing = ` · ${suffix}`; if (trailing.length >= 200) { - return suffix.slice(0, 200) + return suffix.slice(0, 200); } - return `${sourceTitle - .slice(0, 200 - trailing.length) - .trimEnd()}${trailing}` + return `${sourceTitle.slice(0, 200 - trailing.length).trimEnd()}${trailing}`; } function ConversationBranchBadge({ - sourceTitle + sourceTitle, }: { - sourceTitle: string + sourceTitle: string; }): React.JSX.Element { - const { t } = useTranslation('app') - const label = t('conversation.branch.badge', { - title: sourceTitle - }) + const { t } = useTranslation("app"); + const label = t("conversation.branch.badge", { + title: sourceTitle, + }); return ( - ) + ); } function isUnusedConversation(conversation: Conversation): boolean { return ( - conversation.title === '新对话' && + conversation.title === "新对话" && conversation.messages.length === 1 && - conversation.messages[0]?.role === 'assistant' - ) + conversation.messages[0]?.role === "assistant" + ); } function getConversationDisplayTitle( conversation: Conversation, - defaultTitle: string + defaultTitle: string, ): string { - return isUnusedConversation(conversation) - ? defaultTitle - : conversation.title + return isUnusedConversation(conversation) ? defaultTitle : conversation.title; } type ChatQuickAction = { - title: string - description: string - prompt: string -} + title: string; + description: string; + prompt: string; +}; type ChatScrollSnapshot = { - pinnedToBottom: boolean - scrollTop: number -} + pinnedToBottom: boolean; + scrollTop: number; +}; function ChatHistoryPane({ active, @@ -864,281 +818,264 @@ function ChatHistoryPane({ quickActions, scrollSnapshot, taskStrip, - visibleMessageCount + visibleMessageCount, }: { - active: boolean - artifactById: ReadonlyMap - conversationHtmlRenderingEnabled: boolean - conversation: Conversation - locale: TimeFormatLocale - onCopyMessage: (content: string) => Promise - onDownloadImage: (item: ImageViewerItem) => void - onOpenCitationContext: ( - reference: KnowledgeSearchReference - ) => Promise - onOpenCitationSource: ( - reference: KnowledgeSearchReference - ) => Promise - onOpenImage: (item: ImageViewerItem, trigger: HTMLElement) => void + active: boolean; + artifactById: ReadonlyMap; + conversationHtmlRenderingEnabled: boolean; + conversation: Conversation; + locale: TimeFormatLocale; + onCopyMessage: (content: string) => Promise; + onDownloadImage: (item: ImageViewerItem) => void; + onOpenCitationContext: (reference: KnowledgeSearchReference) => Promise; + onOpenCitationSource: (reference: KnowledgeSearchReference) => Promise; + onOpenImage: (item: ImageViewerItem, trigger: HTMLElement) => void; onRespondApproval: ( conversationId: string, messageId: string, approvalId: string, - decision: ApprovalDecision - ) => Promise + decision: ApprovalDecision, + ) => Promise; onRespondQuestion: ( conversationId: string, messageId: string, questionId: string, - answers?: AgentQuestionAnswer[] - ) => Promise - onRetry: (content: string) => void + answers?: AgentQuestionAnswer[], + ) => Promise; + onRetry: (content: string) => void; onScrollSnapshotChange: ( conversationId: string, - snapshot: ChatScrollSnapshot - ) => void - onSetInput: (value: string) => void - onVisibleMessageCountChange: ( - conversationId: string, - count: number - ) => void - quickActions: ChatQuickAction[] - scrollSnapshot?: ChatScrollSnapshot - taskStrip?: ReactNode - visibleMessageCount: number + snapshot: ChatScrollSnapshot, + ) => void; + onSetInput: (value: string) => void; + onVisibleMessageCountChange: (conversationId: string, count: number) => void; + quickActions: ChatQuickAction[]; + scrollSnapshot?: ChatScrollSnapshot; + taskStrip?: ReactNode; + visibleMessageCount: number; }): React.JSX.Element { - const { t } = useTranslation('app') - const headingId = `chat-heading-${conversation.id}` - const scrollRef = useRef(null) - const pinnedToBottomRef = useRef( - scrollSnapshot?.pinnedToBottom ?? true - ) - const latestScrollSnapshotRef = useRef(scrollSnapshot) - const restorePendingRef = useRef(true) - const prependScrollPositionRef = useRef<{ - scrollHeight: number - scrollTop: number - } | undefined>(undefined) - const finalRevealedMessageIdRef = useRef( - undefined - ) - const messageArticleRefs = useRef(new Map()) - const previousMessageCountRef = useRef(conversation.messages.length) + const { t } = useTranslation("app"); + const headingId = `chat-heading-${conversation.id}`; + const scrollRef = useRef(null); + const pinnedToBottomRef = useRef(scrollSnapshot?.pinnedToBottom ?? true); + const latestScrollSnapshotRef = useRef(scrollSnapshot); + const restorePendingRef = useRef(true); + const prependScrollPositionRef = useRef< + | { + scrollHeight: number; + scrollTop: number; + } + | undefined + >(undefined); + const finalRevealedMessageIdRef = useRef(undefined); + const messageArticleRefs = useRef(new Map()); + const previousMessageCountRef = useRef(conversation.messages.length); const [showScrollToBottom, setShowScrollToBottom] = useState( - scrollSnapshot ? !scrollSnapshot.pinnedToBottom : false - ) + scrollSnapshot ? !scrollSnapshot.pinnedToBottom : false, + ); const visibleMessageStartIndex = Math.max( 0, - conversation.messages.length - visibleMessageCount - ) + conversation.messages.length - visibleMessageCount, + ); const visibleMessages = useMemo( () => conversation.messages.slice(visibleMessageStartIndex), - [conversation.messages, visibleMessageStartIndex] - ) - const hiddenMessageCount = visibleMessageStartIndex + [conversation.messages, visibleMessageStartIndex], + ); + const hiddenMessageCount = visibleMessageStartIndex; const handleArticleRef = useCallback( (messageId: string, element: HTMLElement | null): void => { if (element) { - messageArticleRefs.current.set(messageId, element) + messageArticleRefs.current.set(messageId, element); } else { - messageArticleRefs.current.delete(messageId) + messageArticleRefs.current.delete(messageId); } }, - [] - ) + [], + ); const saveScrollPosition = useCallback( (scrollContainer: HTMLElement): boolean => { const distanceFromBottom = scrollContainer.scrollHeight - scrollContainer.scrollTop - - scrollContainer.clientHeight - const pinnedToBottom = distanceFromBottom <= chatBottomProximity + scrollContainer.clientHeight; + const pinnedToBottom = distanceFromBottom <= chatBottomProximity; latestScrollSnapshotRef.current = { pinnedToBottom, - scrollTop: scrollContainer.scrollTop - } - return pinnedToBottom + scrollTop: scrollContainer.scrollTop, + }; + return pinnedToBottom; }, - [] - ) + [], + ); const handleScrollRef = useCallback( (element: HTMLElement | null): void => { - const previous = scrollRef.current + const previous = scrollRef.current; if (previous && previous !== element) { - saveScrollPosition(previous) + saveScrollPosition(previous); if (!element && latestScrollSnapshotRef.current) { onScrollSnapshotChange( conversation.id, - latestScrollSnapshotRef.current - ) + latestScrollSnapshotRef.current, + ); } } - scrollRef.current = element + scrollRef.current = element; }, - [conversation.id, onScrollSnapshotChange, saveScrollPosition] - ) + [conversation.id, onScrollSnapshotChange, saveScrollPosition], + ); const updateScrollPosition = useCallback((): void => { - const scrollContainer = scrollRef.current + const scrollContainer = scrollRef.current; if (!scrollContainer) { - return + return; } - const atBottom = saveScrollPosition(scrollContainer) - pinnedToBottomRef.current = atBottom - setShowScrollToBottom(!atBottom) - }, [saveScrollPosition]) + const atBottom = saveScrollPosition(scrollContainer); + pinnedToBottomRef.current = atBottom; + setShowScrollToBottom(!atBottom); + }, [saveScrollPosition]); useLayoutEffect(() => { - const previousMessageCount = previousMessageCountRef.current + const previousMessageCount = previousMessageCountRef.current; if ( conversation.messages .slice(previousMessageCount) - .some((message) => message.role === 'user') + .some((message) => message.role === "user") ) { - pinnedToBottomRef.current = true + pinnedToBottomRef.current = true; } - previousMessageCountRef.current = conversation.messages.length - }, [conversation.messages]) + previousMessageCountRef.current = conversation.messages.length; + }, [conversation.messages]); useLayoutEffect(() => { if (!active) { - return + return; } - const scrollContainer = scrollRef.current + const scrollContainer = scrollRef.current; if (!scrollContainer) { - return + return; } if (restorePendingRef.current) { - restorePendingRef.current = false + restorePendingRef.current = false; if (scrollSnapshot && !scrollSnapshot.pinnedToBottom) { - pinnedToBottomRef.current = false - scrollContainer.scrollTop = scrollSnapshot.scrollTop - return + pinnedToBottomRef.current = false; + scrollContainer.scrollTop = scrollSnapshot.scrollTop; + return; } } if (pinnedToBottomRef.current) { scrollContainer.scrollTo({ top: scrollContainer.scrollHeight, - behavior: 'auto' - }) + behavior: "auto", + }); } - }, [ - active, - conversation.messages, - scrollSnapshot, - visibleMessageCount - ]) + }, [active, conversation.messages, scrollSnapshot, visibleMessageCount]); useLayoutEffect(() => { - if (!active || typeof ResizeObserver !== 'function') { - return + if (!active || typeof ResizeObserver !== "function") { + return; } - const scrollContainer = scrollRef.current + const scrollContainer = scrollRef.current; const messageList = - scrollContainer?.querySelector('.message-list') + scrollContainer?.querySelector(".message-list"); if (!scrollContainer || !messageList) { - return + return; } const observer = new ResizeObserver(() => { - if ( - restorePendingRef.current || - !pinnedToBottomRef.current - ) { - return + if (restorePendingRef.current || !pinnedToBottomRef.current) { + return; } scrollContainer.scrollTo({ top: scrollContainer.scrollHeight, - behavior: 'auto' - }) - }) - observer.observe(messageList) - return () => observer.disconnect() - }, [active]) + behavior: "auto", + }); + }); + observer.observe(messageList); + return () => observer.disconnect(); + }, [active]); useLayoutEffect(() => { - const previous = prependScrollPositionRef.current + const previous = prependScrollPositionRef.current; if (!previous) { - return + return; } - prependScrollPositionRef.current = undefined - const scrollContainer = scrollRef.current + prependScrollPositionRef.current = undefined; + const scrollContainer = scrollRef.current; if (!scrollContainer) { - return + return; } scrollContainer.scrollTop = previous.scrollTop + - (scrollContainer.scrollHeight - previous.scrollHeight) - const finalRevealedMessageId = finalRevealedMessageIdRef.current - finalRevealedMessageIdRef.current = undefined + (scrollContainer.scrollHeight - previous.scrollHeight); + const finalRevealedMessageId = finalRevealedMessageIdRef.current; + finalRevealedMessageIdRef.current = undefined; if (finalRevealedMessageId) { messageArticleRefs.current .get(finalRevealedMessageId) - ?.focus({ preventScroll: true }) + ?.focus({ preventScroll: true }); } - }, [visibleMessageCount]) + }, [visibleMessageCount]); const revealEarlierMessages = useCallback((): void => { - const scrollContainer = scrollRef.current + const scrollContainer = scrollRef.current; if (scrollContainer) { prependScrollPositionRef.current = { scrollHeight: scrollContainer.scrollHeight, - scrollTop: scrollContainer.scrollTop - } + scrollTop: scrollContainer.scrollTop, + }; } if ( visibleMessageCount + messageRenderBatchSize >= conversation.messages.length ) { - finalRevealedMessageIdRef.current = - conversation.messages[0]?.id + finalRevealedMessageIdRef.current = conversation.messages[0]?.id; } onVisibleMessageCountChange( conversation.id, - visibleMessageCount + messageRenderBatchSize - ) + visibleMessageCount + messageRenderBatchSize, + ); }, [ conversation.id, conversation.messages, onVisibleMessageCountChange, - visibleMessageCount - ]) + visibleMessageCount, + ]); const scrollToBottom = (): void => { - const scrollContainer = scrollRef.current + const scrollContainer = scrollRef.current; if (!scrollContainer) { - return + return; } - pinnedToBottomRef.current = true - setShowScrollToBottom(false) + pinnedToBottomRef.current = true; + setShowScrollToBottom(false); const reduceMotion = - typeof window.matchMedia === 'function' && - window.matchMedia('(prefers-reduced-motion: reduce)').matches + typeof window.matchMedia === "function" && + window.matchMedia("(prefers-reduced-motion: reduce)").matches; scrollContainer.scrollTo({ top: scrollContainer.scrollHeight, - behavior: reduceMotion ? 'auto' : 'smooth' - }) - } + behavior: reduceMotion ? "auto" : "smooth", + }); + }; return ( - ) + ); } function App(): React.JSX.Element { - const { i18n, t } = useTranslation('app') - const { t: tWorkspace } = useTranslation('workspace') - const tRef = useRef(t) + const { i18n, t } = useTranslation("app"); + const { t: tWorkspace } = useTranslation("workspace"); + const tRef = useRef(t); useEffect(() => { - tRef.current = t - }, [t]) - const locale = i18n.resolvedLanguage === 'en-US' ? 'en-US' : 'zh-CN' + tRef.current = t; + }, [t]); + const locale = i18n.resolvedLanguage === "en-US" ? "en-US" : "zh-CN"; const conversationMigrationStoragePresent = useRef( - hasConversationMigrationStorage() - ) + hasConversationMigrationStorage(), + ); const [conversations, setConversations] = useState(() => loadConversations( - t('conversation.greeting'), - t('conversation.interrupted') - ) - ) + t("conversation.greeting"), + t("conversation.interrupted"), + ), + ); const [activeId, setActiveIdState] = useState( - () => conversations[0]?.id ?? '' - ) - const activeConversationIdRef = useRef(activeId) - const conversationsRef = useRef(conversations) + () => conversations[0]?.id ?? "", + ); + const activeConversationIdRef = useRef(activeId); + const conversationsRef = useRef(conversations); const persistedLocalConversationsRef = useRef( - new Map() - ) - const conversationPersistenceQueueRef = - useRef>(Promise.resolve()) - const conversationPersistencePausedRef = useRef(false) - const deletingLocalConversationIdsRef = useRef(new Set()) - const flushConversationPersistenceAfterRenderRef = useRef(false) + new Map(), + ); + const conversationPersistenceQueueRef = useRef>( + Promise.resolve(), + ); + const conversationPersistencePausedRef = useRef(false); + const deletingLocalConversationIdsRef = useRef(new Set()); + const flushConversationPersistenceAfterRenderRef = useRef(false); const [unreadConversationIds, setUnreadConversationIds] = useState< Set - >(() => new Set()) - const [conversationStoreReady, setConversationStoreReady] = - useState(false) - const migrationConversations = useRef(conversations) - const [projects, setProjects] = useState([]) - const projectsRef = useRef(projects) - const [projectRecoveryByProjectId, setProjectRecoveryByProjectId] = - useState>({}) + >(() => new Set()); + const [conversationStoreReady, setConversationStoreReady] = useState(false); + const migrationConversations = useRef(conversations); + const [projects, setProjects] = useState([]); + const projectsRef = useRef(projects); + const [projectRecoveryByProjectId, setProjectRecoveryByProjectId] = useState< + Record + >({}); const [projectRecoverySnapshotReady, setProjectRecoverySnapshotReady] = - useState(false) - const projectRecoverySnapshotReadyRef = useRef(false) + useState(false); + const projectRecoverySnapshotReadyRef = useRef(false); const projectRecoveryByProjectIdRef = useRef< Record - >({}) - const retryingRecoveryProjectIdsRef = useRef(new Set()) - const [assistantTasks, setAssistantTasks] = useState([]) - const assistantTasksRef = useRef(assistantTasks) + >({}); + const retryingRecoveryProjectIdsRef = useRef(new Set()); + const [assistantTasks, setAssistantTasks] = useState([]); + const assistantTasksRef = useRef(assistantTasks); const [tokenUsage, setTokenUsage] = - useState(emptyTokenUsage) - const [workspaceChanges, setWorkspaceChanges] = - useState() + useState(emptyTokenUsage); + const [workspaceChanges, setWorkspaceChanges] = useState(); const [assistantArtifacts, setAssistantArtifacts] = useState< AssistantArtifact[] - >([]) + >([]); const assistantArtifactById = useMemo( () => - new Map( - assistantArtifacts.map((artifact) => [artifact.id, artifact]) - ), - [assistantArtifacts] - ) - const [assistantMemories, setAssistantMemories] = useState< - AssistantMemory[] - >([]) + new Map(assistantArtifacts.map((artifact) => [artifact.id, artifact])), + [assistantArtifacts], + ); + const [assistantMemories, setAssistantMemories] = useState( + [], + ); const [assistantSchedules, setAssistantSchedules] = useState< AssistantSchedule[] - >([]) + >([]); const [conversationQueueItems, setConversationQueueItems] = useState< ConversationQueueItem[] - >([]) - const dispatchedConversationQueueItems = useRef(new Set()) + >([]); + const dispatchedConversationQueueItems = useRef(new Set()); const conversationQueueDispatchRef = useRef< (dispatch: ConversationQueueDispatch) => void - >(() => undefined) + >(() => undefined); const [selectedAssistantTaskId, setSelectedAssistantTaskId] = - useState() + useState(); const [expandedTaskConversationIds, setExpandedTaskConversationIds] = - useState>(() => new Set()) + useState>(() => new Set()); const [customTaskDialog, setCustomTaskDialog] = useState<{ - defaultDestination: CustomTaskDestination - }>() + defaultDestination: CustomTaskDestination; + }>(); const [assistantHeartbeats, setAssistantHeartbeats] = useState< AssistantHeartbeatConfig[] - >([]) + >([]); const [heartbeatEntries, setHeartbeatEntries] = useState< AssistantHeartbeatEntry[] - >([]) - const [heartbeatRuns, setHeartbeatRuns] = useState< - AssistantHeartbeatRun[] - >([]) - const [heartbeatMemories, setHeartbeatMemories] = useState< - AssistantMemory[] - >([]) - const [heartbeatLoading, setHeartbeatLoading] = useState(true) - const [heartbeatLoadError, setHeartbeatLoadError] = useState() - const [assistantExperts, setAssistantExperts] = useState< - AssistantExpert[] - >([]) - const [selectedExpertId, setSelectedExpertId] = useState('') - const [activeProjectId, setActiveProjectId] = useState('') - const activeProjectIdRef = useRef(activeProjectId) - const workspaceChangesRequestRef = useRef(0) - const runtimeStatusRequestRef = useRef(0) - const runtimeSetupPromptedRef = useRef(false) - const runtimeStatusCacheRef = useRef<{ - key: string - settings: RuntimeSettings - } | undefined>(undefined) - const viewRef = useRef('chat') - const heartbeatLoadRequestRef = useRef(0) - const [workMode, setWorkMode] = - useState('ask') + >([]); + const [heartbeatRuns, setHeartbeatRuns] = useState( + [], + ); + const [heartbeatMemories, setHeartbeatMemories] = useState( + [], + ); + const [heartbeatLoading, setHeartbeatLoading] = useState(true); + const [heartbeatLoadError, setHeartbeatLoadError] = useState(); + const [assistantExperts, setAssistantExperts] = useState( + [], + ); + const [selectedExpertId, setSelectedExpertId] = useState(""); + const [activeProjectId, setActiveProjectId] = useState(""); + const activeProjectIdRef = useRef(activeProjectId); + const workspaceChangesRequestRef = useRef(0); + const runtimeStatusRequestRef = useRef(0); + const runtimeSwitchGenerationRef = useRef(0); + const runtimeSetupPromptedRef = useRef(false); + const runtimeStatusCacheRef = useRef< + | { + key: string; + settings: RuntimeSettings; + } + | undefined + >(undefined); + const viewRef = useRef("chat"); + const heartbeatLoadRequestRef = useRef(0); + const [workMode, setWorkMode] = useState("ask"); const [conversationDrafts, setConversationDrafts] = useState< Record - >({}) - const input = conversationDrafts[activeId] ?? '' + >({}); + const input = conversationDrafts[activeId] ?? ""; const setInput = useCallback( (update: SetStateAction): void => { setConversationDrafts((current) => { - const currentValue = current[activeId] ?? '' + const currentValue = current[activeId] ?? ""; const nextValue = - typeof update === 'function' - ? update(currentValue) - : update + typeof update === "function" ? update(currentValue) : update; if (nextValue === currentValue) { - return current + return current; } if (!nextValue) { - const next = { ...current } - delete next[activeId] - return next + const next = { ...current }; + delete next[activeId]; + return next; } - return { ...current, [activeId]: nextValue } - }) + return { ...current, [activeId]: nextValue }; + }); }, - [activeId] - ) - const [voiceListening, setVoiceListening] = useState(false) - const [voiceRecording, setVoiceRecording] = useState(false) - const voiceRecordingRef = useRef(undefined) - const voiceRequestIdRef = useRef(undefined) - const voiceStartingRef = useRef(false) - const voiceDisposedRef = useRef(false) - const startupUpdateCheckStartedRef = useRef(false) - const startupReleaseNotesStartedRef = useRef(false) - const [releaseNotes, setReleaseNotes] = - useState() - const [runtime, setRuntime] = useState() - const [runtimeStatusKey, setRuntimeStatusKey] = useState('') - const [runtimeSettings, setRuntimeSettings] = useState() - const [runtimeMenuOpen, setRuntimeMenuOpen] = useState(false) + [activeId], + ); + const [voiceListening, setVoiceListening] = useState(false); + const [voiceRecording, setVoiceRecording] = useState(false); + const voiceRecordingRef = useRef(undefined); + const voiceRequestIdRef = useRef(undefined); + const voiceStartingRef = useRef(false); + const voiceDisposedRef = useRef(false); + const startupUpdateCheckStartedRef = useRef(false); + const startupReleaseNotesStartedRef = useRef(false); + const [releaseNotes, setReleaseNotes] = useState(); + const [runtime, setRuntime] = useState(); + const [runtimeStatusKey, setRuntimeStatusKey] = useState(""); + const [runtimeSettings, setRuntimeSettings] = useState(); + const [runtimeMenuOpen, setRuntimeMenuOpen] = useState(false); const [composerMenuOpen, setComposerMenuOpen] = useState< - | 'expert' - | 'mode' - | 'runtime-agent' - | 'runtime-action' - | 'runtime-preset' + | "expert" + | "mode" + | "runtime-agent" + | "runtime-action" + | "runtime-preset" | undefined - >() + >(); const [runtimeCustomization, setRuntimeCustomization] = - useState() + useState(); const [runtimeNativeSnapshot, setRuntimeNativeSnapshot] = - useState() - const [selectedRuntimeAgent, setSelectedRuntimeAgent] = - useState('') - const [selectedRuntimeCommand, setSelectedRuntimeCommand] = - useState('') - const [selectedContinuePreset, setSelectedContinuePreset] = - useState('') + useState(); + const runtimeNativeCacheRef = useRef( + new Map< + string, + { + customization: RuntimeCustomizationSettings; + snapshot: RuntimeNativeSnapshot; + } + >(), + ); + const runtimeNativeRetryScopesRef = useRef(new Set()); + const [runtimeNativeRetry, setRuntimeNativeRetry] = useState(0); + const [selectedRuntimeAgent, setSelectedRuntimeAgent] = useState(""); + const [selectedRuntimeCommand, setSelectedRuntimeCommand] = useState(""); + const [selectedContinuePreset, setSelectedContinuePreset] = useState(""); const [runtimeContextCompacting, setRuntimeContextCompacting] = - useState(false) - const runtimeCustomizationRequestRef = useRef(0) - const runtimeMenuButtonRef = useRef(null) - const runtimeMenuRef = useRef(null) - const [runtimeSwitching, setRuntimeSwitching] = useState(false) + useState(false); + const runtimeCustomizationRequestRef = useRef(0); + const runtimeMenuButtonRef = useRef(null); + const runtimeMenuRef = useRef(null); + const [runtimeSwitching, setRuntimeSwitching] = useState(false); const [appearanceTheme, setAppearanceTheme] = - useState(loadAppearanceTheme) + useState(loadAppearanceTheme); const [brandingPreferences, setBrandingPreferences] = useState( - loadBrandingPreferences - ) + loadBrandingPreferences, + ); const [systemPrefersDark, setSystemPrefersDark] = useState( () => - typeof window.matchMedia === 'function' && - window.matchMedia('(prefers-color-scheme: dark)').matches - ) + typeof window.matchMedia === "function" && + window.matchMedia("(prefers-color-scheme: dark)").matches, + ); const resolvedAppearanceTheme = resolveAppearanceTheme( appearanceTheme, - systemPrefersDark - ) + systemPrefersDark, + ); const brandingSubtitle = - locale === 'en-US' + locale === "en-US" ? brandingPreferences.subtitleEnUS - : brandingPreferences.subtitleZhCN + : brandingPreferences.subtitleZhCN; const toggleAppearanceTheme = useCallback((): void => { - setAppearanceTheme( - resolvedAppearanceTheme === 'dark' ? 'light' : 'dark' - ) - }, [resolvedAppearanceTheme]) + setAppearanceTheme(resolvedAppearanceTheme === "dark" ? "light" : "dark"); + }, [resolvedAppearanceTheme]); const effectiveWorkMode = - workMode === 'execute' && - runtime?.supportsToolExecution === false - ? 'ask' - : workMode + workMode === "execute" && runtime?.supportsToolExecution === false + ? "ask" + : workMode; const setExpertMenuOpen = useCallback((open: boolean): void => { - setComposerMenuOpen(open ? 'expert' : undefined) + setComposerMenuOpen(open ? "expert" : undefined); if (open) { - setRuntimeMenuOpen(false) + setRuntimeMenuOpen(false); } - }, []) + }, []); const setModeMenuOpen = useCallback((open: boolean): void => { - setComposerMenuOpen(open ? 'mode' : undefined) + setComposerMenuOpen(open ? "mode" : undefined); if (open) { - setRuntimeMenuOpen(false) + setRuntimeMenuOpen(false); } - }, []) - const setRuntimeAgentMenuOpen = useCallback( - (open: boolean): void => { - setComposerMenuOpen(open ? 'runtime-agent' : undefined) - if (open) { - setRuntimeMenuOpen(false) - } - }, - [] - ) - const setRuntimeActionMenuOpen = useCallback( - (open: boolean): void => { - setComposerMenuOpen(open ? 'runtime-action' : undefined) - if (open) { - setRuntimeMenuOpen(false) - } - }, - [] - ) - const setRuntimePresetMenuOpen = useCallback( - (open: boolean): void => { - setComposerMenuOpen(open ? 'runtime-preset' : undefined) - if (open) { - setRuntimeMenuOpen(false) - } - }, - [] - ) - const assistantExpertOptions = useMemo< - ComposerMenuOption[] - >( + }, []); + const setRuntimeAgentMenuOpen = useCallback((open: boolean): void => { + setComposerMenuOpen(open ? "runtime-agent" : undefined); + if (open) { + setRuntimeMenuOpen(false); + } + }, []); + const setRuntimeActionMenuOpen = useCallback((open: boolean): void => { + setComposerMenuOpen(open ? "runtime-action" : undefined); + if (open) { + setRuntimeMenuOpen(false); + } + }, []); + const setRuntimePresetMenuOpen = useCallback((open: boolean): void => { + setComposerMenuOpen(open ? "runtime-preset" : undefined); + if (open) { + setRuntimeMenuOpen(false); + } + }, []); + const assistantExpertOptions = useMemo[]>( () => [ { - value: '', - label: t('composer.experts.general'), - description: t('composer.experts.generalDescription') + value: "", + label: t("composer.experts.general"), + description: t("composer.experts.generalDescription"), }, { - value: 'team', - label: t('composer.experts.team'), - description: t('composer.experts.teamDescription') + value: "team", + label: t("composer.experts.team"), + description: t("composer.experts.teamDescription"), }, ...assistantExperts.map((expert) => ({ value: expert.id, label: expert.name, description: - expert.description || t('composer.experts.customDescription') - })) + expert.description || t("composer.experts.customDescription"), + })), ], - [assistantExperts, t] - ) - const workModeOptions = useMemo< - ComposerMenuOption[] - >( + [assistantExperts, t], + ); + const workModeOptions = useMemo[]>( () => interactiveWorkModes.map((value) => ({ value, label: t(`composer.modes.${value}.label`), description: - value === 'execute' - ? t('composer.modes.execute.description') - : t('composer.modes.ask.description'), - disabled: - value === 'execute' && !runtime?.supportsToolExecution + value === "execute" + ? t("composer.modes.execute.description") + : t("composer.modes.ask.description"), + disabled: value === "execute" && !runtime?.supportsToolExecution, })), - [runtime?.supportsToolExecution, t] - ) + [runtime?.supportsToolExecution, t], + ); const quickActions = useMemo( () => [ { - title: t('chat.quickActions.summarize.title'), - description: t('chat.quickActions.summarize.description'), - prompt: t('chat.quickActions.summarize.prompt') + title: t("chat.quickActions.summarize.title"), + description: t("chat.quickActions.summarize.description"), + prompt: t("chat.quickActions.summarize.prompt"), }, { - title: t('chat.quickActions.analyzeError.title'), - description: t('chat.quickActions.analyzeError.description'), - prompt: t('chat.quickActions.analyzeError.prompt') + title: t("chat.quickActions.analyzeError.title"), + description: t("chat.quickActions.analyzeError.description"), + prompt: t("chat.quickActions.analyzeError.prompt"), }, { - title: t('chat.quickActions.write.title'), - description: t('chat.quickActions.write.description'), - prompt: t('chat.quickActions.write.prompt') - } + title: t("chat.quickActions.write.title"), + description: t("chat.quickActions.write.description"), + prompt: t("chat.quickActions.write.prompt"), + }, ], - [t] - ) - const [appInfo, setAppInfo] = useState() + [t], + ); + const [appInfo, setAppInfo] = useState(); const [narrowWindow, setNarrowWindow] = useState( - () => window.innerWidth < 900 - ) + () => window.innerWidth < 900, + ); const [sidebarOpen, setSidebarOpen] = useState( - () => window.innerWidth >= 900 - ) + () => window.innerWidth >= 900, + ); const [primarySidebarWidth, setPrimarySidebarWidth] = useState( - loadPrimarySidebarWidth - ) - const [primarySidebarResizing, setPrimarySidebarResizing] = - useState(false) + loadPrimarySidebarWidth, + ); + const [primarySidebarResizing, setPrimarySidebarResizing] = useState(false); const [assistantSidebarOpen, setAssistantSidebarOpen] = useState( - () => window.innerWidth >= 1280 - ) + () => window.innerWidth >= 1280, + ); const [assistantSidebarTab, setAssistantSidebarTab] = - useState('tasks') + useState("tasks"); const [browserStates, setBrowserStates] = useState< Record - >({}) - const [view, setViewState] = useState('chat') - const settingsEntryFocusRef = useRef( - undefined - ) - const settingsExitFocusRef = useRef( - undefined - ) - const settingsLeaveRequesterRef = useRef< - SettingsLeaveRequester | undefined - >(undefined) + >({}); + const [view, setViewState] = useState("chat"); + const settingsEntryFocusRef = useRef(undefined); + const settingsExitFocusRef = useRef(undefined); + const settingsLeaveRequesterRef = useRef( + undefined, + ); const [cachedWorkspaceViews, setCachedWorkspaceViews] = useState< KeepAliveCacheEntry[] - >(() => [{ key: 'chat', lastVisitedAt: Date.now() }]) + >(() => [{ key: "chat", lastVisitedAt: Date.now() }]); const [cachedConversationViews, setCachedConversationViews] = useState< KeepAliveCacheEntry[] - >(() => - activeId - ? [{ key: activeId, lastVisitedAt: Date.now() }] - : [] - ) - const commitView = useCallback( - (next: WorkspaceView): void => { - const previous = viewRef.current - const now = Date.now() - const runningConversationIds = new Set( - [...activeRuns.current.values()].map((run) => run.conversationId) + >(() => (activeId ? [{ key: activeId, lastVisitedAt: Date.now() }] : [])); + const commitView = useCallback((next: WorkspaceView): void => { + const previous = viewRef.current; + const now = Date.now(); + const runningConversationIds = new Set( + [...activeRuns.current.values()].map((run) => run.conversationId), + ); + preparingConversations.current.forEach((conversationId) => + runningConversationIds.add(conversationId), + ); + const protectedWorkspaceViews = new Set(); + if (runningConversationIds.size > 0) { + protectedWorkspaceViews.add("chat"); + protectedWorkspaceViews.add("activity"); + } + if (knowledgeOperationCountRef.current > 0) { + protectedWorkspaceViews.add("knowledge"); + protectedWorkspaceViews.add("activity"); + } + if ( + assistantTasksRef.current.some( + (task) => + task.status === "queued" || + task.status === "running" || + task.status === "waiting_approval", ) - preparingConversations.current.forEach((conversationId) => - runningConversationIds.add(conversationId) - ) - const protectedWorkspaceViews = new Set() - if (runningConversationIds.size > 0) { - protectedWorkspaceViews.add('chat') - protectedWorkspaceViews.add('activity') - } - if (knowledgeOperationCountRef.current > 0) { - protectedWorkspaceViews.add('knowledge') - protectedWorkspaceViews.add('activity') - } - if ( - assistantTasksRef.current.some( - (task) => - task.status === 'queued' || - task.status === 'running' || - task.status === 'waiting_approval' - ) - ) { - protectedWorkspaceViews.add('activity') - } - viewRef.current = next - setCachedWorkspaceViews((current) => - touchAndPruneKeepAliveEntries(current, next, now, { - expiresAfterMs: keepAliveExpirationMs, - maximumEntries: maximumCachedWorkspaceViews, - protectedKeys: protectedWorkspaceViews, - recentEntries: recentCachedWorkspaceViews - }) - ) - setViewState(next) - if (previous === 'settings' && next !== 'settings') { - const preferred = settingsExitFocusRef.current - const returnTarget = settingsEntryFocusRef.current - settingsExitFocusRef.current = undefined - settingsEntryFocusRef.current = undefined - requestAnimationFrame(() => { - const target = - preferred?.isConnected - ? preferred - : returnTarget?.isConnected - ? returnTarget - : next === 'chat' - ? inputRef.current - : document.querySelector( - `.primary-nav [aria-current="page"]` - ) - target?.focus() - }) - } - }, - [] - ) + ) { + protectedWorkspaceViews.add("activity"); + } + viewRef.current = next; + setCachedWorkspaceViews((current) => + touchAndPruneKeepAliveEntries(current, next, now, { + expiresAfterMs: keepAliveExpirationMs, + maximumEntries: maximumCachedWorkspaceViews, + protectedKeys: protectedWorkspaceViews, + recentEntries: recentCachedWorkspaceViews, + }), + ); + setViewState(next); + if (previous === "settings" && next !== "settings") { + const preferred = settingsExitFocusRef.current; + const returnTarget = settingsEntryFocusRef.current; + settingsExitFocusRef.current = undefined; + settingsEntryFocusRef.current = undefined; + requestAnimationFrame(() => { + const target = preferred?.isConnected + ? preferred + : returnTarget?.isConnected + ? returnTarget + : next === "chat" + ? inputRef.current + : document.querySelector( + `.primary-nav [aria-current="page"]`, + ); + target?.focus(); + }); + } + }, []); const setView = useCallback( (update: SetStateAction): void => { const next = - typeof update === 'function' - ? update(viewRef.current) - : update + typeof update === "function" ? update(viewRef.current) : update; if ( - viewRef.current !== 'settings' && - next === 'settings' && + viewRef.current !== "settings" && + next === "settings" && !settingsEntryFocusRef.current?.isConnected ) { - const activeElement = document.activeElement + const activeElement = document.activeElement; settingsEntryFocusRef.current = activeElement instanceof HTMLElement && activeElement !== document.body && activeElement.isConnected ? activeElement - : undefined + : undefined; } - if (viewRef.current === 'settings' && next !== 'settings') { - const requestLeave = settingsLeaveRequesterRef.current + if (viewRef.current === "settings" && next !== "settings") { + const requestLeave = settingsLeaveRequesterRef.current; if (requestLeave) { - requestLeave(() => commitView(next)) - return + requestLeave(() => commitView(next)); + return; } } - commitView(next) + commitView(next); }, - [commitView] - ) + [commitView], + ); const registerSettingsLeaveRequester = useCallback( (requester: SettingsLeaveRequester | undefined): void => { - settingsLeaveRequesterRef.current = requester + settingsLeaveRequesterRef.current = requester; }, - [] - ) + [], + ); const handleShortcutSettingsChanged = useCallback( (snapshot: GlobalShortcutSettingsSnapshot): void => { setAppInfo((current) => current ? { ...current, - shortcut: snapshot.registered - ? snapshot.displayAccelerator - : '', - shortcutStatus: snapshot.status + shortcut: snapshot.registered ? snapshot.displayAccelerator : "", + shortcutStatus: snapshot.status, } - : current - ) + : current, + ); }, - [] - ) - const setActiveId = useCallback( - (update: SetStateAction): void => { - const next = - typeof update === 'function' - ? update(activeConversationIdRef.current) - : update - activeConversationIdRef.current = next - if (next) { - const now = Date.now() - const runningConversationIds = new Set( - [...activeRuns.current.values()].map((run) => run.conversationId) - ) - preparingConversations.current.forEach((conversationId) => - runningConversationIds.add(conversationId) - ) - setCachedConversationViews((current) => - touchAndPruneKeepAliveEntries(current, next, now, { - expiresAfterMs: keepAliveExpirationMs, - maximumEntries: maximumCachedConversations, - protectedKeys: runningConversationIds, - recentEntries: recentCachedConversations - }) - ) - } - setActiveIdState(next) - }, - [] - ) + [], + ); + const setActiveId = useCallback((update: SetStateAction): void => { + const next = + typeof update === "function" + ? update(activeConversationIdRef.current) + : update; + activeConversationIdRef.current = next; + if (next) { + const now = Date.now(); + const runningConversationIds = new Set( + [...activeRuns.current.values()].map((run) => run.conversationId), + ); + preparingConversations.current.forEach((conversationId) => + runningConversationIds.add(conversationId), + ); + setCachedConversationViews((current) => + touchAndPruneKeepAliveEntries(current, next, now, { + expiresAfterMs: keepAliveExpirationMs, + maximumEntries: maximumCachedConversations, + protectedKeys: runningConversationIds, + recentEntries: recentCachedConversations, + }), + ); + } + setActiveIdState(next); + }, []); const [settingsInitialCategory, setSettingsInitialCategory] = - useState() + useState(); const [settingsInitialChannel, setSettingsInitialChannel] = - useState() - const [remoteProjectsEnabled, setRemoteProjectsEnabled] = - useState(false) + useState(); + const [remoteProjectsEnabled, setRemoteProjectsEnabled] = useState(false); const [ conversationHtmlRenderingEnabled, - setConversationHtmlRenderingEnabled - ] = useState(true) - const [magicNotesEnabled, setMagicNotesEnabled] = useState(false) + setConversationHtmlRenderingEnabled, + ] = useState(true); + const [magicNotesEnabled, setMagicNotesEnabled] = useState(false); const [ magicNotesShowIncompleteTodoCount, - setMagicNotesShowIncompleteTodoCount - ] = useState(true) - const [incompleteMagicTodoCount, setIncompleteMagicTodoCount] = - useState(0) - const [searchQuery, setSearchQuery] = useState('') - const deferredSearchQuery = useDeferredValue(searchQuery) + setMagicNotesShowIncompleteTodoCount, + ] = useState(true); + const [incompleteMagicTodoCount, setIncompleteMagicTodoCount] = useState(0); + const [searchQuery, setSearchQuery] = useState(""); + const deferredSearchQuery = useDeferredValue(searchQuery); const [searchConversationSnapshot, setSearchConversationSnapshot] = - useState(conversations) - const [conversationLoadError, setConversationLoadError] = - useState() - const [conversationLoadRetry, setConversationLoadRetry] = useState(0) - const [conversationActionsId, setConversationActionsId] = useState('') - const [confirmingConversationId, setConfirmingConversationId] = - useState('') - const [deletingConversationId, setDeletingConversationId] = - useState('') - const [branchingConversationId, setBranchingConversationId] = - useState('') - const [renamingConversationId, setRenamingConversationId] = useState('') - const [notifications, notify] = useReducer( - appNotificationReducer, - [] - ) + useState(conversations); + const [conversationLoadError, setConversationLoadError] = useState(); + const [conversationLoadRetry, setConversationLoadRetry] = useState(0); + const [conversationActionsId, setConversationActionsId] = useState(""); + const [confirmingConversationId, setConfirmingConversationId] = useState(""); + const [deletingConversationId, setDeletingConversationId] = useState(""); + const [branchingConversationId, setBranchingConversationId] = useState(""); + const [renamingConversationId, setRenamingConversationId] = useState(""); + const [notifications, notify] = useReducer(appNotificationReducer, []); const handleWindowControlError = useCallback( (message: string): void => { - notify({ tone: 'error', message }) + notify({ tone: "error", message }); }, - [notify] - ) - const [attachmentsByConversation, setAttachmentsByConversation] = - useState>({}) - const attachments = - attachmentsByConversation[activeId] ?? [] - const attachmentsRef = useRef( - new Map() - ) + [notify], + ); + const [attachmentsByConversation, setAttachmentsByConversation] = useState< + Record + >({}); + const attachments = attachmentsByConversation[activeId] ?? []; + const attachmentsRef = useRef(new Map()); const updateAttachments = useCallback( ( update: | ContextAttachment[] - | ((current: ContextAttachment[]) => ContextAttachment[]) + | ((current: ContextAttachment[]) => ContextAttachment[]), ): void => { - const current = attachmentsRef.current.get(activeId) ?? [] - const next = - typeof update === 'function' - ? update(current) - : update + const current = attachmentsRef.current.get(activeId) ?? []; + const next = typeof update === "function" ? update(current) : update; if (next.length > 0) { - attachmentsRef.current.set(activeId, next) + attachmentsRef.current.set(activeId, next); } else { - attachmentsRef.current.delete(activeId) + attachmentsRef.current.delete(activeId); } setAttachmentsByConversation((values) => { if (next.length > 0) { - return { ...values, [activeId]: next } + return { ...values, [activeId]: next }; } - const remaining = { ...values } - delete remaining[activeId] - return remaining - }) + const remaining = { ...values }; + delete remaining[activeId]; + return remaining; + }); }, - [activeId] - ) - const [contextError, setContextError] = useState() + [activeId], + ); + const [contextError, setContextError] = useState(); const [fileSelectionProgress, setFileSelectionProgress] = - useState() - const [selectingContextFiles, setSelectingContextFiles] = - useState(false) - const selectingContextFilesRef = useRef(false) - const [imageViewerItem, setImageViewerItem] = - useState() + useState(); + const [selectingContextFiles, setSelectingContextFiles] = useState(false); + const selectingContextFilesRef = useRef(false); + const [imageViewerItem, setImageViewerItem] = useState(); const [citationDialog, setCitationDialog] = useState<{ - reference: KnowledgeSearchReference - context?: KnowledgeCitationContextView - loading: boolean - error?: string - }>() - const imageViewerTriggerRef = useRef( - undefined - ) - const imageViewerDialogRef = useRef(null) - const imageViewerCloseRef = useRef(null) + reference: KnowledgeSearchReference; + context?: KnowledgeCitationContextView; + loading: boolean; + error?: string; + }>(); + const imageViewerTriggerRef = useRef(undefined); + const imageViewerDialogRef = useRef(null); + const imageViewerCloseRef = useRef(null); useEffect(() => { if (!imageViewerItem) { - return + return; } - return activateModalFocus(() => imageViewerCloseRef.current) - }, [imageViewerItem]) + return activateModalFocus(() => imageViewerCloseRef.current); + }, [imageViewerItem]); useEffect( () => window.goodbuddy.context.onFileSelectionProgress((progress) => { if (selectingContextFilesRef.current) { - setFileSelectionProgress(progress) + setFileSelectionProgress(progress); } }), - [] - ) - const [knowledgeSnapshot, setKnowledgeSnapshot] = useState({ - libraries: [], - sources: [], - documents: [], - graphNodes: [], - graphRelations: [], - evidence: [], - tasks: [] - }) - const [knowledgeLoading, setKnowledgeLoading] = useState(true) - const [knowledgeLoadError, setKnowledgeLoadError] = useState() - const [knowledgeOperationCount, setKnowledgeOperationCount] = useState(0) - const knowledgeOperationCountRef = useRef(knowledgeOperationCount) - const knowledgeLoadRequestRef = useRef(0) - const failedKnowledgeLibraryIdRef = useRef( - undefined - ) - const [knowledgeScopeOpen, setKnowledgeScopeOpen] = useState(false) - const knowledgeScopeTriggerRef = useRef(null) - const knowledgeScopePopoverRef = useRef(null) - const [legacyActivityHistory] = useState(loadLegacyActivityHistory) + [], + ); + const [knowledgeSnapshot, setKnowledgeSnapshot] = useState( + { + libraries: [], + sources: [], + documents: [], + graphNodes: [], + graphRelations: [], + evidence: [], + tasks: [], + }, + ); + const [knowledgeLoading, setKnowledgeLoading] = useState(true); + const [knowledgeLoadError, setKnowledgeLoadError] = useState(); + const [knowledgeOperationCount, setKnowledgeOperationCount] = useState(0); + const knowledgeOperationCountRef = useRef(knowledgeOperationCount); + const knowledgeLoadRequestRef = useRef(0); + const failedKnowledgeLibraryIdRef = useRef(undefined); + const [knowledgeScopeOpen, setKnowledgeScopeOpen] = useState(false); + const knowledgeScopeTriggerRef = useRef(null); + const knowledgeScopePopoverRef = useRef(null); + const [legacyActivityHistory] = useState(loadLegacyActivityHistory); const [activityRecords, setActivityRecords] = useState( - legacyActivityHistory.records - ) + legacyActivityHistory.records, + ); const [ legacyActivityHistoryMayBeIncomplete, - setLegacyActivityHistoryMayBeIncomplete - ] = useState(legacyActivityHistory.historyMayBeIncomplete) - const [activityHistoryReady, setActivityHistoryReady] = useState(false) - const activityRecordsRef = useRef(activityRecords) + setLegacyActivityHistoryMayBeIncomplete, + ] = useState(legacyActivityHistory.historyMayBeIncomplete); + const [activityHistoryReady, setActivityHistoryReady] = useState(false); + const activityRecordsRef = useRef(activityRecords); const legacyActivityHistoryMayBeIncompleteRef = useRef( - legacyActivityHistoryMayBeIncomplete - ) - const activeRuns = useRef(new Map()) - const preparingConversations = useRef(new Set()) + legacyActivityHistoryMayBeIncomplete, + ); + const activeRuns = useRef(new Map()); + const preparingConversations = useRef(new Set()); const [activeConversationIds, setActiveConversationIds] = useState< ReadonlySet - >(() => new Set()) + >(() => new Set()); const setConversationActivity = useCallback( (conversationId: string, active: boolean): void => { setActiveConversationIds((current) => { if (current.has(conversationId) === active) { - return current + return current; } - const next = new Set(current) + const next = new Set(current); if (active) { - next.add(conversationId) + next.add(conversationId); } else { - next.delete(conversationId) + next.delete(conversationId); } - return next - }) + return next; + }); }, - [] - ) - const hydratingArtifactIds = useRef(new Set()) - const inputRef = useRef(null) + [], + ); + const hydratingArtifactIds = useRef(new Set()); + const inputRef = useRef(null); const [chatScrollSnapshots, setChatScrollSnapshots] = useState< Record - >({}) - const retryMessage = useCallback((content: string): void => { - setInput(content) - inputRef.current?.focus() - }, [setInput]) + >({}); + const retryMessage = useCallback( + (content: string): void => { + setInput(content); + inputRef.current?.focus(); + }, + [setInput], + ); useEffect( () => scheduleIdleRoutePreload( idleRouteModuleLoaders, () => activeRuns.current.size === 0 && - preparingConversations.current.size === 0 + preparingConversations.current.size === 0, ), - [] - ) + [], + ); const [visibleMessageCounts, setVisibleMessageCounts] = useState< Record - >({}) - const sidebarRef = useRef(null) - const sidebarToggleRef = useRef(null) - const livePrimarySidebarWidthRef = useRef(primarySidebarWidth) + >({}); + const sidebarRef = useRef(null); + const sidebarToggleRef = useRef(null); + const livePrimarySidebarWidthRef = useRef(primarySidebarWidth); const primarySidebarResizePointerIdRef = useRef( - undefined - ) - const assistantSidebarToggleRef = useRef(null) + undefined, + ); + const assistantSidebarToggleRef = useRef(null); const conversationActionTriggerRefs = useRef( - new Map() - ) + new Map(), + ); const handleChatScrollSnapshotChange = useCallback( (conversationId: string, snapshot: ChatScrollSnapshot): void => { setChatScrollSnapshots((current) => ({ ...current, - [conversationId]: snapshot - })) + [conversationId]: snapshot, + })); }, - [] - ) + [], + ); const handleVisibleMessageCountChange = useCallback( (conversationId: string, count: number): void => { setVisibleMessageCounts((current) => ({ ...current, - [conversationId]: count - })) + [conversationId]: count, + })); }, - [] - ) + [], + ); const closeNarrowSidebar = useCallback((): void => { - setSidebarOpen(false) - requestAnimationFrame(() => sidebarToggleRef.current?.focus()) - }, []) + setSidebarOpen(false); + requestAnimationFrame(() => sidebarToggleRef.current?.focus()); + }, []); const resizePrimarySidebarFromClientX = useCallback( (clientX: number, commit: boolean): void => { - const width = clampPrimarySidebarWidth( - clientX, - window.innerWidth - ) - livePrimarySidebarWidthRef.current = width + const width = clampPrimarySidebarWidth(clientX, window.innerWidth); + livePrimarySidebarWidthRef.current = width; if (commit) { - setPrimarySidebarWidth(width) - return + setPrimarySidebarWidth(width); + return; } sidebarRef.current?.style.setProperty( - '--primary-sidebar-width', - `${width}px` - ) + "--primary-sidebar-width", + `${width}px`, + ); }, - [] - ) + [], + ); const finishPrimarySidebarResize = useCallback( (event: React.PointerEvent): void => { - if ( - primarySidebarResizePointerIdRef.current !== event.pointerId - ) { - return + if (primarySidebarResizePointerIdRef.current !== event.pointerId) { + return; } - primarySidebarResizePointerIdRef.current = undefined + primarySidebarResizePointerIdRef.current = undefined; if (event.currentTarget.hasPointerCapture(event.pointerId)) { - event.currentTarget.releasePointerCapture(event.pointerId) + event.currentTarget.releasePointerCapture(event.pointerId); } - setPrimarySidebarWidth(livePrimarySidebarWidthRef.current) - setPrimarySidebarResizing(false) + setPrimarySidebarWidth(livePrimarySidebarWidthRef.current); + setPrimarySidebarResizing(false); }, - [] - ) + [], + ); const resizePrimarySidebarWithKeyboard = useCallback( (event: React.KeyboardEvent): void => { if (narrowWindow || !sidebarOpen) { - return + return; } - const limits = getPrimarySidebarWidthLimits(window.innerWidth) + const limits = getPrimarySidebarWidthLimits(window.innerWidth); const nextWidth = - event.key === 'Home' + event.key === "Home" ? limits.minimum - : event.key === 'End' + : event.key === "End" ? limits.maximum - : event.key === 'ArrowLeft' - ? primarySidebarWidth - - primarySidebarKeyboardResizeStep - : event.key === 'ArrowRight' - ? primarySidebarWidth + - primarySidebarKeyboardResizeStep - : undefined + : event.key === "ArrowLeft" + ? primarySidebarWidth - primarySidebarKeyboardResizeStep + : event.key === "ArrowRight" + ? primarySidebarWidth + primarySidebarKeyboardResizeStep + : undefined; if (nextWidth === undefined) { - return + return; } - event.preventDefault() - const width = clampPrimarySidebarWidth( - nextWidth, - window.innerWidth - ) - livePrimarySidebarWidthRef.current = width - setPrimarySidebarWidth(width) + event.preventDefault(); + const width = clampPrimarySidebarWidth(nextWidth, window.innerWidth); + livePrimarySidebarWidthRef.current = width; + setPrimarySidebarWidth(width); }, - [narrowWindow, primarySidebarWidth, sidebarOpen] - ) + [narrowWindow, primarySidebarWidth, sidebarOpen], + ); const navigateFromSidebar = useCallback( (nextView: WorkspaceView, trigger: HTMLElement): void => { - if (viewRef.current !== 'settings' && nextView === 'settings') { - settingsEntryFocusRef.current = trigger - } else if ( - viewRef.current === 'settings' && - nextView !== 'settings' - ) { - settingsExitFocusRef.current = trigger + if (viewRef.current !== "settings" && nextView === "settings") { + settingsEntryFocusRef.current = trigger; + } else if (viewRef.current === "settings" && nextView !== "settings") { + settingsExitFocusRef.current = trigger; } - setView(nextView) + setView(nextView); if (narrowWindow) { - closeNarrowSidebar() + closeNarrowSidebar(); } }, - [closeNarrowSidebar, narrowWindow, setView] - ) + [closeNarrowSidebar, narrowWindow, setView], + ); useEffect(() => { if (!conversationStoreReady || !projectRecoverySnapshotReady) { - return + return; } const conversationIds = [ ...new Set( @@ -2884,62 +2721,60 @@ function App(): React.JSX.Element { ? projectRecoveryByProjectIdRef.current[ conversation.projectId ] - : undefined - ) + : undefined, + ), ) - .map((conversation) => conversation.id) - ) - ] + .map((conversation) => conversation.id), + ), + ]; void Promise.all( conversationIds.map((conversationId) => - window.goodbuddy.conversationQueue.ready(conversationId) - ) + window.goodbuddy.conversationQueue.ready(conversationId), + ), ).catch(() => { notify({ - tone: 'error', - message: tRef.current( - 'notices.conversationQueueResumeFailed' - ), - dedupeKey: 'conversation-queue-resume' - }) - }) - }, [conversationStoreReady, projectRecoverySnapshotReady]) + tone: "error", + message: tRef.current("notices.conversationQueueResumeFailed"), + dedupeKey: "conversation-queue-resume", + }); + }); + }, [conversationStoreReady, projectRecoverySnapshotReady]); useEffect(() => { const sweep = (): void => { - const now = Date.now() + const now = Date.now(); const conversationIds = new Set( - conversationsRef.current.map((conversation) => conversation.id) - ) + conversationsRef.current.map((conversation) => conversation.id), + ); const runningConversationIds = new Set( - [...activeRuns.current.values()].map((run) => run.conversationId) - ) + [...activeRuns.current.values()].map((run) => run.conversationId), + ); preparingConversations.current.forEach((conversationId) => - runningConversationIds.add(conversationId) - ) - const protectedWorkspaceViews = new Set() + runningConversationIds.add(conversationId), + ); + const protectedWorkspaceViews = new Set(); if (runningConversationIds.size > 0) { - protectedWorkspaceViews.add('chat') - protectedWorkspaceViews.add('activity') + protectedWorkspaceViews.add("chat"); + protectedWorkspaceViews.add("activity"); } if (knowledgeOperationCount > 0) { - protectedWorkspaceViews.add('knowledge') - protectedWorkspaceViews.add('activity') + protectedWorkspaceViews.add("knowledge"); + protectedWorkspaceViews.add("activity"); } if ( assistantTasks.some( (task) => - task.status === 'queued' || - task.status === 'running' || - task.status === 'waiting_approval' + task.status === "queued" || + task.status === "running" || + task.status === "waiting_approval", ) ) { - protectedWorkspaceViews.add('activity') + protectedWorkspaceViews.add("activity"); } setCachedConversationViews((current) => pruneKeepAliveEntries( filterKeepAliveEntries(current, (entry) => - conversationIds.has(entry.key) + conversationIds.has(entry.key), ), { currentKey: activeId, @@ -2947,10 +2782,10 @@ function App(): React.JSX.Element { maximumEntries: maximumCachedConversations, now, protectedKeys: runningConversationIds, - recentEntries: recentCachedConversations - } - ) - ) + recentEntries: recentCachedConversations, + }, + ), + ); setCachedWorkspaceViews((current) => pruneKeepAliveEntries(current, { currentKey: view, @@ -2958,377 +2793,359 @@ function App(): React.JSX.Element { maximumEntries: maximumCachedWorkspaceViews, now, protectedKeys: protectedWorkspaceViews, - recentEntries: recentCachedWorkspaceViews - }) - ) - } - const interval = window.setInterval(sweep, keepAliveSweepIntervalMs) - return () => window.clearInterval(interval) - }, [activeId, assistantTasks, knowledgeOperationCount, view]) + recentEntries: recentCachedWorkspaceViews, + }), + ); + }; + const interval = window.setInterval(sweep, keepAliveSweepIntervalMs); + return () => window.clearInterval(interval); + }, [activeId, assistantTasks, knowledgeOperationCount, view]); useEffect(() => { - livePrimarySidebarWidthRef.current = primarySidebarWidth + livePrimarySidebarWidthRef.current = primarySidebarWidth; try { localStorage.setItem( primarySidebarWidthStorageKey, - String(primarySidebarWidth) - ) + String(primarySidebarWidth), + ); } catch { // The current width remains usable when browser storage is unavailable. } - }, [primarySidebarWidth]) + }, [primarySidebarWidth]); useEffect(() => { const collapseSidebarAtNarrowWidth = (): void => { - const narrow = window.innerWidth < 900 - setNarrowWindow(narrow) + const narrow = window.innerWidth < 900; + setNarrowWindow(narrow); if (narrow) { - setSidebarOpen(false) - setPrimarySidebarResizing(false) - primarySidebarResizePointerIdRef.current = undefined + setSidebarOpen(false); + setPrimarySidebarResizing(false); + primarySidebarResizePointerIdRef.current = undefined; } else { setPrimarySidebarWidth((current) => - clampPrimarySidebarWidth(current, window.innerWidth) - ) + clampPrimarySidebarWidth(current, window.innerWidth), + ); } - } - window.addEventListener('resize', collapseSidebarAtNarrowWidth) + }; + window.addEventListener("resize", collapseSidebarAtNarrowWidth); return () => - window.removeEventListener('resize', collapseSidebarAtNarrowWidth) - }, []) + window.removeEventListener("resize", collapseSidebarAtNarrowWidth); + }, []); useEffect(() => { if (!narrowWindow || !sidebarOpen) { - return + return; } const focusFrame = requestAnimationFrame(() => { sidebarRef.current ?.querySelector( - '.primary-nav button[aria-current="page"], .primary-nav button' + '.primary-nav button[aria-current="page"], .primary-nav button', ) - ?.focus() - }) + ?.focus(); + }); const closeOnEscape = (event: KeyboardEvent): void => { - if (event.key === 'Escape') { - event.preventDefault() - closeNarrowSidebar() + if (event.key === "Escape") { + event.preventDefault(); + closeNarrowSidebar(); } - } - document.addEventListener('keydown', closeOnEscape) + }; + document.addEventListener("keydown", closeOnEscape); return () => { - cancelAnimationFrame(focusFrame) - document.removeEventListener('keydown', closeOnEscape) - } - }, [closeNarrowSidebar, narrowWindow, sidebarOpen]) + cancelAnimationFrame(focusFrame); + document.removeEventListener("keydown", closeOnEscape); + }; + }, [closeNarrowSidebar, narrowWindow, sidebarOpen]); useEffect(() => { if (!knowledgeScopeOpen) { - return + return; } const focusFrame = requestAnimationFrame(() => { knowledgeScopePopoverRef.current - ?.querySelector('input') - ?.focus() - }) + ?.querySelector("input") + ?.focus(); + }); const isScopeTarget = (target: EventTarget | null): boolean => target instanceof Node && (knowledgeScopePopoverRef.current?.contains(target) === true || - knowledgeScopeTriggerRef.current?.contains(target) === true) + knowledgeScopeTriggerRef.current?.contains(target) === true); const closeOnOutsidePointer = (event: PointerEvent): void => { if (!isScopeTarget(event.target)) { - setKnowledgeScopeOpen(false) + setKnowledgeScopeOpen(false); } - } + }; const closeOnEscape = (event: KeyboardEvent): void => { - if (event.key !== 'Escape') { - return + if (event.key !== "Escape") { + return; } - event.preventDefault() - setKnowledgeScopeOpen(false) - knowledgeScopeTriggerRef.current?.focus() - } - document.addEventListener('pointerdown', closeOnOutsidePointer) - document.addEventListener('keydown', closeOnEscape) + event.preventDefault(); + setKnowledgeScopeOpen(false); + knowledgeScopeTriggerRef.current?.focus(); + }; + document.addEventListener("pointerdown", closeOnOutsidePointer); + document.addEventListener("keydown", closeOnEscape); return () => { - cancelAnimationFrame(focusFrame) - document.removeEventListener('pointerdown', closeOnOutsidePointer) - document.removeEventListener('keydown', closeOnEscape) - } - }, [knowledgeScopeOpen]) + cancelAnimationFrame(focusFrame); + document.removeEventListener("pointerdown", closeOnOutsidePointer); + document.removeEventListener("keydown", closeOnEscape); + }; + }, [knowledgeScopeOpen]); useLayoutEffect(() => { - conversationsRef.current = conversations - }, [conversations]) + conversationsRef.current = conversations; + }, [conversations]); useEffect(() => { if (!searchQuery.trim()) { - return + return; } const timeout = window.setTimeout( () => setSearchConversationSnapshot(conversations), - conversationSearchSnapshotDelayMs - ) - return () => window.clearTimeout(timeout) - }, [conversations, searchQuery]) + conversationSearchSnapshotDelayMs, + ); + return () => window.clearTimeout(timeout); + }, [conversations, searchQuery]); useEffect(() => { - projectsRef.current = projects - }, [projects]) + projectsRef.current = projects; + }, [projects]); useLayoutEffect(() => { - projectRecoveryByProjectIdRef.current = - projectRecoveryByProjectId - }, [projectRecoveryByProjectId]) + projectRecoveryByProjectIdRef.current = projectRecoveryByProjectId; + }, [projectRecoveryByProjectId]); useEffect(() => { - assistantTasksRef.current = assistantTasks - }, [assistantTasks]) + assistantTasksRef.current = assistantTasks; + }, [assistantTasks]); useLayoutEffect(() => { - resizeComposerTextarea(inputRef.current) - }, [input]) + resizeComposerTextarea(inputRef.current); + }, [input]); useEffect(() => { - saveAppearanceTheme(appearanceTheme) - }, [appearanceTheme]) + saveAppearanceTheme(appearanceTheme); + }, [appearanceTheme]); useEffect(() => { - const updates = window.goodbuddy.updates + const updates = window.goodbuddy.updates; if (!updates || startupUpdateCheckStartedRef.current) { - return + return; } - startupUpdateCheckStartedRef.current = true - let updateCheckSource: 'github' | 'mirror' | undefined + startupUpdateCheckStartedRef.current = true; + let updateCheckSource: "github" | "mirror" | undefined; void updates .getSettings() .then(async (settings) => { setConversationHtmlRenderingEnabled( - settings.conversationHtmlRenderingEnabled !== false - ) - setRemoteProjectsEnabled(settings.remoteProjectsEnabled) - setMagicNotesEnabled(settings.magicNotesEnabled) + settings.conversationHtmlRenderingEnabled !== false, + ); + setRemoteProjectsEnabled(settings.remoteProjectsEnabled); + setMagicNotesEnabled(settings.magicNotesEnabled); setMagicNotesShowIncompleteTodoCount( - settings.magicNotesShowIncompleteTodoCount - ) + settings.magicNotesShowIncompleteTodoCount, + ); if (!settings.magicNotesEnabled) { - setView((current) => - current === 'magic-notes' ? 'chat' : current - ) + setView((current) => (current === "magic-notes" ? "chat" : current)); } if (!settings.checkUpdatesOnStartup) { - return + return; } - updateCheckSource = settings.updateSource - const result = await updates.check() + updateCheckSource = settings.updateSource; + const result = await updates.check(); if (result.updateAvailable) { notify({ - tone: 'info', - message: i18n.t('notices.updateAvailable', { - ns: 'app', - version: result.latestVersion + tone: "info", + message: i18n.t("notices.updateAvailable", { + ns: "app", + version: result.latestVersion, }), - dedupeKey: 'update-available' - }) + dedupeKey: "update-available", + }); } }) .catch((reason: unknown) => { if (!updateCheckSource) { - return + return; } notify({ - tone: 'error', - message: i18n.t('notices.startupUpdateCheckFailed', { - ns: 'app', - source: i18n.t( - `notices.updateSources.${updateCheckSource}`, - { ns: 'app' } - ), + tone: "error", + message: i18n.t("notices.startupUpdateCheckFailed", { + ns: "app", + source: i18n.t(`notices.updateSources.${updateCheckSource}`, { + ns: "app", + }), error: displayNetworkAwareErrorMessage( reason, - i18n.t('notices.updateCheckFailed', { ns: 'app' }), - i18n.t('notices.updateCheckNetwork', { ns: 'app' }) - ) + i18n.t("notices.updateCheckFailed", { ns: "app" }), + i18n.t("notices.updateCheckNetwork", { ns: "app" }), + ), }), - dedupeKey: 'startup-update-check' - }) - }) - }, [i18n, setView]) + dedupeKey: "startup-update-check", + }); + }); + }, [i18n, setView]); useEffect(() => { - const magicNotes = window.goodbuddy.magicNotes - if ( - !magicNotesEnabled || - !magicNotesShowIncompleteTodoCount - ) { - return + const magicNotes = window.goodbuddy.magicNotes; + if (!magicNotesEnabled || !magicNotesShowIncompleteTodoCount) { + return; } - let active = true - let requestId = 0 + let active = true; + let requestId = 0; const refresh = async (): Promise => { - const currentRequestId = ++requestId + const currentRequestId = ++requestId; try { - const status = await magicNotes.getTodoStatus() + const status = await magicNotes.getTodoStatus(); if (active && requestId === currentRequestId) { - setIncompleteMagicTodoCount(status.incompleteCount) + setIncompleteMagicTodoCount(status.incompleteCount); } } catch { if (active && requestId === currentRequestId) { - setIncompleteMagicTodoCount(0) + setIncompleteMagicTodoCount(0); } } - } + }; const removeListener = magicNotes.onTodoStatusChanged(() => { - void refresh() - }) - void refresh() + void refresh(); + }); + void refresh(); return () => { - active = false - removeListener() - } - }, [ - magicNotesEnabled, - magicNotesShowIncompleteTodoCount - ]) + active = false; + removeListener(); + }; + }, [magicNotesEnabled, magicNotesShowIncompleteTodoCount]); useEffect(() => { - const releaseNotesApi = window.goodbuddy.releaseNotes + const releaseNotesApi = window.goodbuddy.releaseNotes; if (!releaseNotesApi || startupReleaseNotesStartedRef.current) { - return + return; } - startupReleaseNotesStartedRef.current = true + startupReleaseNotesStartedRef.current = true; void releaseNotesApi .getPending() .then((snapshot) => { if (snapshot.releases.length > 0) { - setReleaseNotes(snapshot) + setReleaseNotes(snapshot); } }) - .catch(() => undefined) - }, []) + .catch(() => undefined); + }, []); useEffect(() => { - applyAppearanceTheme(resolvedAppearanceTheme) - }, [resolvedAppearanceTheme]) + applyAppearanceTheme(resolvedAppearanceTheme); + }, [resolvedAppearanceTheme]); useEffect(() => { - voiceDisposedRef.current = false + voiceDisposedRef.current = false; return () => { - voiceDisposedRef.current = true - voiceRecordingRef.current?.cancel() - const requestId = voiceRequestIdRef.current + voiceDisposedRef.current = true; + voiceRecordingRef.current?.cancel(); + const requestId = voiceRequestIdRef.current; if (requestId) { - void window.goodbuddy.speech?.cancel(requestId) + void window.goodbuddy.speech?.cancel(requestId); } - } - }, []) + }; + }, []); useEffect(() => { - if (appearanceTheme !== 'system') { - return + if (appearanceTheme !== "system") { + return; } - if (typeof window.matchMedia !== 'function') { - return + if (typeof window.matchMedia !== "function") { + return; } - const systemTheme = window.matchMedia('(prefers-color-scheme: dark)') + const systemTheme = window.matchMedia("(prefers-color-scheme: dark)"); const updateSystemTheme = (): void => { - setSystemPrefersDark(systemTheme.matches) - } - updateSystemTheme() - systemTheme.addEventListener('change', updateSystemTheme) + setSystemPrefersDark(systemTheme.matches); + }; + updateSystemTheme(); + systemTheme.addEventListener("change", updateSystemTheme); return () => { - systemTheme.removeEventListener('change', updateSystemTheme) - } - }, [appearanceTheme]) + systemTheme.removeEventListener("change", updateSystemTheme); + }; + }, [appearanceTheme]); useEffect(() => { - if (typeof window.matchMedia !== 'function') { - return + if (typeof window.matchMedia !== "function") { + return; } - const compactLayout = window.matchMedia('(max-width: 1279px)') + const compactLayout = window.matchMedia("(max-width: 1279px)"); const closeCompactAssistantSidebar = (): void => { if (compactLayout.matches) { - setAssistantSidebarOpen(false) + setAssistantSidebarOpen(false); } - } - closeCompactAssistantSidebar() - compactLayout.addEventListener('change', closeCompactAssistantSidebar) + }; + closeCompactAssistantSidebar(); + compactLayout.addEventListener("change", closeCompactAssistantSidebar); return () => { - compactLayout.removeEventListener( - 'change', - closeCompactAssistantSidebar - ) - } - }, []) + compactLayout.removeEventListener("change", closeCompactAssistantSidebar); + }; + }, []); const activeConversation = useMemo( () => conversations.find((conversation) => conversation.id === activeId), - [activeId, conversations] - ) + [activeId, conversations], + ); const enabledKnowledgeLibraryIds = - activeConversation?.knowledgeLibraryIds ?? [] + activeConversation?.knowledgeLibraryIds ?? []; const setEnabledKnowledgeLibraryIds = useCallback( (action: SetStateAction): void => { setConversations((current) => current.map((conversation) => { if (conversation.id !== activeId) { - return conversation + return conversation; } - const previous = conversation.knowledgeLibraryIds ?? [] - const next = - typeof action === 'function' ? action(previous) : action + const previous = conversation.knowledgeLibraryIds ?? []; + const next = typeof action === "function" ? action(previous) : action; if ( previous.length === next.length && previous.every((id, index) => id === next[index]) ) { - return conversation + return conversation; } return { ...conversation, knowledgeLibraryIds: next, - updatedAt: Date.now() - } - }) - ) + updatedAt: Date.now(), + }; + }), + ); }, - [activeId] - ) + [activeId], + ); const activeProject = useMemo( () => projects.find((project) => project.id === activeProjectId), - [activeProjectId, projects] - ) - const activeProjectUsesManagedSsh = - isManagedSshProject(activeProject) + [activeProjectId, projects], + ); + const activeProjectUsesManagedSsh = isManagedSshProject(activeProject); const activeProjectRecovery = activeProjectUsesManagedSsh && activeProject ? projectRecoveryByProjectId[activeProject.id] - : undefined + : undefined; const activeProjectRecoveryBlocked = activeProjectUsesManagedSsh && (!projectRecoverySnapshotReady || - isProjectRecoveryUnsettled(activeProjectRecovery)) + isProjectRecoveryUnsettled(activeProjectRecovery)); const cachedWorkspaceViewKeys = useMemo( () => new Set(cachedWorkspaceViews.map((entry) => entry.key)), - [cachedWorkspaceViews] - ) + [cachedWorkspaceViews], + ); const cachedConversations = useMemo(() => { const conversationById = new Map( - conversations.map((conversation) => [ - conversation.id, - conversation - ]) - ) + conversations.map((conversation) => [conversation.id, conversation]), + ); const cachedIds = [ activeId, - ...cachedConversationViews.map((entry) => entry.key) + ...cachedConversationViews.map((entry) => entry.key), ].filter( (conversationId, index, values) => - conversationId && values.indexOf(conversationId) === index - ) + conversationId && values.indexOf(conversationId) === index, + ); return cachedIds.flatMap((conversationId) => { - const conversation = conversationById.get(conversationId) - return conversation ? [conversation] : [] - }) - }, [activeId, cachedConversationViews, conversations]) + const conversation = conversationById.get(conversationId); + return conversation ? [conversation] : []; + }); + }, [activeId, cachedConversationViews, conversations]); const activeRuntimeSelection = useMemo( () => @@ -3336,541 +3153,544 @@ function App(): React.JSX.Element { (runtimeSettings ? getDefaultRuntimeSelection(runtimeSettings) : undefined), - [activeConversation?.runtimeSelection, runtimeSettings] - ) + [activeConversation?.runtimeSelection, runtimeSettings], + ); const activeRuntimeSelectionKey = activeRuntimeSelection ? agentRuntimeSelectionKey(activeRuntimeSelection) - : '' - const activeRuntimeSelectionRef = useRef(activeRuntimeSelection) + : ""; + const activeRuntimeSelectionRef = useRef(activeRuntimeSelection); useEffect(() => { - activeRuntimeSelectionRef.current = activeRuntimeSelection - }, [activeRuntimeSelection]) + activeRuntimeSelectionRef.current = activeRuntimeSelection; + }, [activeRuntimeSelection]); const runtimeLabels = useMemo( () => ({ - directModel: t('runtime.directModel'), - automatic: t('runtime.automatic'), - automaticSelection: t('runtime.automaticSelection'), - modelUnavailable: t('runtime.modelUnavailable') + directModel: t("runtime.directModel"), + automatic: t("runtime.automatic"), + automaticSelection: t("runtime.automaticSelection"), + modelUnavailable: t("runtime.modelUnavailable"), }), - [t] - ) + [t], + ); const configuredRuntimeLabels = useMemo( () => ({ - modelUnavailable: t('runtime.modelUnavailable'), - selectModel: t('runtime.selectModel'), - ownConfiguration: t('runtime.ownConfiguration'), + modelUnavailable: t("runtime.modelUnavailable"), + selectModel: t("runtime.selectModel"), + ownConfiguration: t("runtime.ownConfiguration"), useOwnConfiguration: (runtimeLabel: string) => - t('runtime.useOwnConfiguration', { runtime: runtimeLabel }) + t("runtime.useOwnConfiguration", { runtime: runtimeLabel }), }), - [t] - ) + [t], + ); const activeRuntimeLabel = getRuntimeSelectionLabel( activeRuntimeSelection, runtimeSettings, runtime, - runtimeLabels - ) + runtimeLabels, + ); const openCodeMenuSelection = runtimeSettings - ? getRuntimeSelectionForProvider('opencode', runtimeSettings) - : undefined + ? getRuntimeSelectionForProvider("opencode", runtimeSettings) + : undefined; const continueMenuSelection = runtimeSettings - ? getRuntimeSelectionForProvider('continue', runtimeSettings) - : undefined + ? getRuntimeSelectionForProvider("continue", runtimeSettings) + : undefined; const deepseekHarnessMenuSelection = runtimeSettings - ? getRuntimeSelectionForProvider( - 'deepseek-harness', - runtimeSettings - ) - : undefined + ? getRuntimeSelectionForProvider("deepseek-harness", runtimeSettings) + : undefined; const openCodeMenuSource = runtimeSettings ? getConfiguredAgentRuntimeSource( runtimeSettings, - 'opencode', - configuredRuntimeLabels + "opencode", + configuredRuntimeLabels, ) - : undefined + : undefined; const continueMenuSource = runtimeSettings ? getConfiguredAgentRuntimeSource( runtimeSettings, - 'continue', - configuredRuntimeLabels + "continue", + configuredRuntimeLabels, ) - : undefined + : undefined; const deepseekHarnessMenuSource = runtimeSettings ? getConfiguredAgentRuntimeSource( runtimeSettings, - 'deepseek-harness', - configuredRuntimeLabels + "deepseek-harness", + configuredRuntimeLabels, ) - : undefined + : undefined; // Conversation refreshes replace the conversation object, so the runtime // selection identity changes without the selection itself changing. Keying // this effect on the resolved scope keeps the runtime controls mounted // instead of clearing and refetching the snapshot on every refresh. const runtimeNativeScopeKey = [ - activeConversation?.remote ? 'remote' : 'local', - activeProjectUsesManagedSsh ? 'managed-ssh' : 'direct', - activeProjectId ?? '', - activeRuntimeSelectionKey - ].join('\u0000') - const runtimeNativeScopeKeyRef = useRef(undefined) + activeConversation?.remote ? "remote" : "local", + activeProjectUsesManagedSsh ? "managed-ssh" : "direct", + activeProjectId ?? "", + activeRuntimeSelectionKey, + ].join("\u0000"); + const runtimeNativeScopeKeyRef = useRef(undefined); + + useEffect(() => { + runtimeSwitchGenerationRef.current += 1; + const generation = runtimeSwitchGenerationRef.current; + queueMicrotask(() => { + if (runtimeSwitchGenerationRef.current === generation) { + setRuntimeSwitching(false); + } + }); + }, [activeId]); + + useEffect(() => { + let cancelled = false; + queueMicrotask(() => { + if (!cancelled) { + setSelectedRuntimeAgent(""); + setSelectedRuntimeCommand(""); + setSelectedContinuePreset(""); + } + }); + return () => { cancelled = true; }; + }, [activeId, runtimeNativeScopeKey]); useEffect(() => { if (runtimeNativeScopeKeyRef.current === runtimeNativeScopeKey) { - return + return; } - runtimeNativeScopeKeyRef.current = runtimeNativeScopeKey - const selection = activeRuntimeSelectionRef.current - const requestId = runtimeCustomizationRequestRef.current + 1 - runtimeCustomizationRequestRef.current = requestId + if ( + runtimeNativeScopeKeyRef.current !== undefined && + runtimeNativeScopeKeyRef.current !== runtimeNativeScopeKey + ) { + runtimeNativeRetryScopesRef.current.delete(runtimeNativeScopeKey); + } + runtimeNativeScopeKeyRef.current = runtimeNativeScopeKey; + const selection = activeRuntimeSelectionRef.current; + const cached = runtimeNativeCacheRef.current.get(runtimeNativeScopeKey); + const requestId = runtimeCustomizationRequestRef.current + 1; + runtimeCustomizationRequestRef.current = requestId; queueMicrotask(() => { if (runtimeCustomizationRequestRef.current !== requestId) { - return + return; } - setRuntimeNativeSnapshot(undefined) - setRuntimeCustomization(undefined) - setSelectedRuntimeAgent('') - setSelectedRuntimeCommand('') - setSelectedContinuePreset('') - }) + setRuntimeNativeSnapshot(cached?.snapshot); + setRuntimeCustomization(cached?.customization); + }); if ( !selection || activeConversation?.remote || activeProjectUsesManagedSsh || - (selection.provider !== 'opencode' && - selection.provider !== 'continue') + (selection.provider !== "opencode" && selection.provider !== "continue") ) { - return + return; } - const provider = selection.provider + const provider = selection.provider; void Promise.all([ window.goodbuddy.runtimeCustomization.getSettings(), window.goodbuddy.runtimeCustomization.getNativeSnapshot({ provider, - ...('profileId' in selection && selection.profileId + ...("profileId" in selection && selection.profileId ? { profileId: selection.profileId } : {}), - ...(activeProjectId ? { projectId: activeProjectId } : {}) - }) + ...(activeProjectId ? { projectId: activeProjectId } : {}), + }), ]) .then(([customization, snapshot]) => { if (runtimeCustomizationRequestRef.current !== requestId) { - return + return; } - setRuntimeCustomization(customization) - setRuntimeNativeSnapshot(snapshot) - setSelectedContinuePreset('') + if ( + snapshot.inventoryStatus === "unavailable" || + snapshot.inventoryStatus === "partial" + ) { + // Keep a healthy cached inventory; otherwise retain usable partial data. + runtimeNativeCacheRef.current.set( + runtimeNativeScopeKey, + cached?.snapshot.available ? cached : { customization, snapshot }, + ); + throw new Error(snapshot.detail); + } + setRuntimeCustomization(customization); + setRuntimeNativeSnapshot(snapshot); + runtimeNativeCacheRef.current.set(runtimeNativeScopeKey, { + customization, + snapshot, + }); + runtimeNativeRetryScopesRef.current.delete(runtimeNativeScopeKey); }) .catch(() => { if (runtimeCustomizationRequestRef.current === requestId) { - setRuntimeCustomization(undefined) - setRuntimeNativeSnapshot(undefined) + const retained = runtimeNativeCacheRef.current.get(runtimeNativeScopeKey); + setRuntimeCustomization(retained?.customization); + setRuntimeNativeSnapshot(retained?.snapshot); + runtimeNativeScopeKeyRef.current = undefined; + if (!runtimeNativeRetryScopesRef.current.has(runtimeNativeScopeKey)) { + runtimeNativeRetryScopesRef.current.add(runtimeNativeScopeKey); + setRuntimeNativeRetry((value) => value + 1); + } } - }) + }); }, [ activeConversation?.remote, activeProjectUsesManagedSsh, activeProjectId, - runtimeNativeScopeKey - ]) + runtimeNativeRetry, + runtimeNativeScopeKey, + ]); - const runtimeAgentOptions = useMemo< - ComposerMenuOption[] - >(() => { + const runtimeAgentOptions = useMemo[]>(() => { if ( - activeRuntimeSelection?.provider !== 'opencode' || + activeRuntimeSelection?.provider !== "opencode" || !runtimeNativeSnapshot ) { - return [] + return []; } - const configuredDefault = - runtimeCustomization?.opencode.defaultAgent + const configuredDefault = runtimeCustomization?.opencode.defaultAgent; return [ { - value: '', + value: "", label: configuredDefault - ? t('composer.runtimeControls.configuredAgent', { - name: configuredDefault + ? t("composer.runtimeControls.configuredAgent", { + name: configuredDefault, }) - : t('composer.runtimeControls.runtimeDefaultAgent'), + : t("composer.runtimeControls.runtimeDefaultAgent"), description: t( - 'composer.runtimeControls.runtimeDefaultAgentDescription' - ) + "composer.runtimeControls.runtimeDefaultAgentDescription", + ), }, ...runtimeNativeSnapshot.agents .filter( (agent) => - !agent.hidden && - (agent.mode === 'primary' || agent.mode === 'all') + !agent.hidden && (agent.mode === "primary" || agent.mode === "all"), ) .map((agent) => ({ value: agent.id, label: agent.name, description: - agent.description ?? - t('composer.runtimeControls.agentDescription') - })) - ] + agent.description ?? t("composer.runtimeControls.agentDescription"), + })), + ]; }, [ activeRuntimeSelection?.provider, runtimeCustomization?.opencode.defaultAgent, runtimeNativeSnapshot, - t - ]) + t, + ]); - const runtimePresetOptions = useMemo< - ComposerMenuOption[] - >(() => { + const runtimePresetOptions = useMemo[]>(() => { if ( - activeRuntimeSelection?.provider !== 'continue' || + activeRuntimeSelection?.provider !== "continue" || !runtimeCustomization ) { - return [] + return []; } return [ { - value: '', - label: t('composer.runtimeControls.noPreset'), - description: t( - 'composer.runtimeControls.noPresetDescription' - ) + value: "", + label: t("composer.runtimeControls.noPreset"), + description: t("composer.runtimeControls.noPresetDescription"), }, ...runtimeCustomization.continue.presets.map((preset) => ({ value: preset.id, label: preset.name, description: preset.description ?? - t('composer.runtimeControls.presetDescription', { + t("composer.runtimeControls.presetDescription", { rules: preset.rules.filter((rule) => rule.enabled).length, - prompts: preset.prompts.length - }) - })) - ] - }, [ - activeRuntimeSelection?.provider, - runtimeCustomization, - t - ]) + prompts: preset.prompts.length, + }), + })), + ]; + }, [activeRuntimeSelection?.provider, runtimeCustomization, t]); const runtimeActionOptions = useMemo(() => { if (!runtimeNativeSnapshot) { - return [] + return []; } - const nativePrompts = runtimeNativeSnapshot.prompts + const nativePrompts = runtimeNativeSnapshot.prompts; const selectedPreset = - activeRuntimeSelection?.provider === 'continue' + activeRuntimeSelection?.provider === "continue" ? runtimeCustomization?.continue.presets.find( (preset) => preset.id === (selectedContinuePreset || - runtimeCustomization.continue.defaultPresetId) + runtimeCustomization.continue.defaultPresetId), ) - : undefined + : undefined; return [ { - value: '', - label: t('composer.runtimeControls.noAction'), - description: t( - 'composer.runtimeControls.noActionDescription' - ) + value: "", + label: t("composer.runtimeControls.noAction"), + description: t("composer.runtimeControls.noActionDescription"), }, - ...(activeRuntimeSelection?.provider === 'opencode' + ...(activeRuntimeSelection?.provider === "opencode" ? runtimeNativeSnapshot.commands.map((command) => ({ - value: JSON.stringify(['command', command.id]), + value: JSON.stringify(["command", command.id]), label: `/${command.name}`, description: command.description ?? - t('composer.runtimeControls.commandDescription'), + t("composer.runtimeControls.commandDescription"), action: { - type: 'command' as const, - id: command.id - } + type: "command" as const, + id: command.id, + }, })) : []), ...nativePrompts.map((prompt) => ({ - value: JSON.stringify(['native-prompt', prompt.id]), + value: JSON.stringify(["native-prompt", prompt.id]), label: prompt.name, description: - prompt.description ?? - t('composer.runtimeControls.promptDescription'), + prompt.description ?? t("composer.runtimeControls.promptDescription"), action: { - type: 'prompt' as const, - prompt: prompt.prompt - } + type: "prompt" as const, + prompt: prompt.prompt, + }, })), ...(selectedPreset?.prompts.map((prompt) => ({ - value: JSON.stringify([ - 'preset-prompt', - selectedPreset.id, - prompt.id - ]), + value: JSON.stringify(["preset-prompt", selectedPreset.id, prompt.id]), label: prompt.name, description: - prompt.description ?? - t('composer.runtimeControls.promptDescription'), + prompt.description ?? t("composer.runtimeControls.promptDescription"), action: { - type: 'prompt' as const, - prompt: prompt.prompt - } - })) ?? []) - ] + type: "prompt" as const, + prompt: prompt.prompt, + }, + })) ?? []), + ]; }, [ activeRuntimeSelection?.provider, runtimeCustomization, runtimeNativeSnapshot, selectedContinuePreset, - t - ]) + t, + ]); const selectRuntimeAction = useCallback( (value: string): void => { if (!value) { - setSelectedRuntimeCommand('') - return + setSelectedRuntimeCommand(""); + return; } const choice = runtimeActionOptions.find( - (candidate) => candidate.value === value - ) - if (choice?.action?.type === 'command') { - setSelectedRuntimeCommand(choice.action.id) - return + (candidate) => candidate.value === value, + ); + if (choice?.action?.type === "command") { + setSelectedRuntimeCommand(choice.action.id); + return; } - if (choice?.action?.type === 'prompt') { - setInput(choice.action.prompt) - setSelectedRuntimeCommand('') + if (choice?.action?.type === "prompt") { + setInput(choice.action.prompt); + setSelectedRuntimeCommand(""); requestAnimationFrame(() => { - resizeComposerTextarea(inputRef.current) - inputRef.current?.focus() - }) + resizeComposerTextarea(inputRef.current); + inputRef.current?.focus(); + }); } }, - [runtimeActionOptions, setInput] - ) + [runtimeActionOptions, setInput], + ); useEffect(() => { if (!runtimeMenuOpen) { - return + return; } - const menu = runtimeMenuRef.current + const menu = runtimeMenuRef.current; if (!menu) { - return + return; } const menuItems = Array.from( menu.querySelectorAll( - '[role="menuitemradio"], [role="menuitem"]' - ) - ).filter((item) => !item.disabled) + '[role="menuitemradio"], [role="menuitem"]', + ), + ).filter((item) => !item.disabled); const initialItem = - menuItems.find( - (item) => item.getAttribute('aria-checked') === 'true' - ) ?? menuItems[0] + menuItems.find((item) => item.getAttribute("aria-checked") === "true") ?? + menuItems[0]; menuItems.forEach((item) => { - item.tabIndex = item === initialItem ? 0 : -1 - }) + item.tabIndex = item === initialItem ? 0 : -1; + }); const focusFrame = requestAnimationFrame(() => { - initialItem?.focus() - }) + initialItem?.focus(); + }); const isRuntimeMenuTarget = (target: EventTarget | null): boolean => target instanceof Node && (menu.contains(target) || - runtimeMenuButtonRef.current?.contains(target) === true) + runtimeMenuButtonRef.current?.contains(target) === true); const dismissOnOutsidePointer = (event: PointerEvent): void => { if (!isRuntimeMenuTarget(event.target)) { - setRuntimeMenuOpen(false) + setRuntimeMenuOpen(false); } - } + }; const dismissOnOutsideFocus = (event: FocusEvent): void => { if (!isRuntimeMenuTarget(event.target)) { - setRuntimeMenuOpen(false) + setRuntimeMenuOpen(false); } - } - document.addEventListener('pointerdown', dismissOnOutsidePointer) - document.addEventListener('focusin', dismissOnOutsideFocus) + }; + document.addEventListener("pointerdown", dismissOnOutsidePointer); + document.addEventListener("focusin", dismissOnOutsideFocus); return () => { - cancelAnimationFrame(focusFrame) - document.removeEventListener( - 'pointerdown', - dismissOnOutsidePointer - ) - document.removeEventListener('focusin', dismissOnOutsideFocus) - } - }, [activeRuntimeSelectionKey, runtimeMenuOpen]) + cancelAnimationFrame(focusFrame); + document.removeEventListener("pointerdown", dismissOnOutsidePointer); + document.removeEventListener("focusin", dismissOnOutsideFocus); + }; + }, [activeRuntimeSelectionKey, runtimeMenuOpen]); const conversationNavigationRef = useRef({ activeId, - conversations - }) + conversations, + }); useEffect(() => { conversationNavigationRef.current = { activeId, - conversations - } - }, [activeId, conversations]) + conversations, + }; + }, [activeId, conversations]); useEffect(() => { - const selection = activeRuntimeSelectionRef.current + const selection = activeRuntimeSelectionRef.current; if (!selection || !runtimeSettings) { - return + return; } if ( - runtimeStatusCacheRef.current?.key === - activeRuntimeSelectionKey && + runtimeStatusCacheRef.current?.key === activeRuntimeSelectionKey && runtimeStatusCacheRef.current.settings === runtimeSettings ) { - return + return; } runtimeStatusCacheRef.current = { key: activeRuntimeSelectionKey, - settings: runtimeSettings - } - const requestId = runtimeStatusRequestRef.current + 1 - runtimeStatusRequestRef.current = requestId - setRuntimeSwitching(false) - setRuntimeStatusKey('') + settings: runtimeSettings, + }; + const requestId = runtimeStatusRequestRef.current + 1; + runtimeStatusRequestRef.current = requestId; + setRuntimeSwitching(false); + setRuntimeStatusKey(""); void window.goodbuddy.agent .getStatus(selection) .then((status) => { if (runtimeStatusRequestRef.current !== requestId) { - return + return; } - setRuntime(status) - setRuntimeStatusKey(activeRuntimeSelectionKey) + setRuntime(status); + setRuntimeStatusKey(activeRuntimeSelectionKey); if (!status.available && !runtimeSetupPromptedRef.current) { - runtimeSetupPromptedRef.current = true - setView('settings') + runtimeSetupPromptedRef.current = true; + setView("settings"); } }) .catch((reason: unknown) => { if (runtimeStatusRequestRef.current !== requestId) { - return + return; } setRuntime({ - id: 'setup', - label: tRef.current('runtime.unavailable'), + id: "setup", + label: tRef.current("runtime.unavailable"), available: false, supportsToolExecution: false, detail: reason instanceof Error ? reason.message - : tRef.current('runtime.errors.readStatus') - }) - setRuntimeStatusKey(activeRuntimeSelectionKey) - }) - }, [ - activeRuntimeSelectionKey, - runtimeSettings, - setView - ]) + : tRef.current("runtime.errors.readStatus"), + }); + setRuntimeStatusKey(activeRuntimeSelectionKey); + }); + }, [activeRuntimeSelectionKey, runtimeSettings, setView]); const startNewConversation = useCallback( (projectId?: string): boolean => { - const project = projects.find( - (candidate) => candidate.id === projectId - ) - if (project?.kind === 'channel') { - setView('chat') + const project = projects.find((candidate) => candidate.id === projectId); + if (project?.kind === "channel") { + setView("chat"); notify({ - tone: 'info', - message: tRef.current('notices.channelConversationAutomatic'), - dedupeKey: 'channel-project-new-conversation' - }) - return false + tone: "info", + message: tRef.current("notices.channelConversationAutomatic"), + dedupeKey: "channel-project-new-conversation", + }); + return false; } - const navigation = conversationNavigationRef.current + const navigation = conversationNavigationRef.current; const currentConversation = navigation.conversations.find( - (conversation) => conversation.id === navigation.activeId - ) + (conversation) => conversation.id === navigation.activeId, + ); if ( currentConversation && currentConversation.projectId === projectId && isUnusedConversation(currentConversation) ) { - setView('chat') - requestAnimationFrame(() => inputRef.current?.focus()) - return true + setView("chat"); + requestAnimationFrame(() => inputRef.current?.focus()); + return true; } const conversation = createConversation( projectId, runtimeSettings ? getProjectDefaultRuntimeSelection(project, runtimeSettings) : undefined, - tRef.current('conversation.greeting') - ) - const nextConversations = [ - conversation, - ...navigation.conversations - ] + tRef.current("conversation.greeting"), + ); + const nextConversations = [conversation, ...navigation.conversations]; conversationNavigationRef.current = { activeId: conversation.id, - conversations: nextConversations - } - setConversations(nextConversations) - setActiveId(conversation.id) - setView('chat') - requestAnimationFrame(() => inputRef.current?.focus()) - return true + conversations: nextConversations, + }; + setConversations(nextConversations); + setActiveId(conversation.id); + setView("chat"); + requestAnimationFrame(() => inputRef.current?.focus()); + return true; }, - [notify, projects, runtimeSettings, setActiveId, setView] - ) + [notify, projects, runtimeSettings, setActiveId, setView], + ); const activeProjectDisplayName = activeProject ? getProjectDisplayText(activeProject, tWorkspace).name - : undefined + : undefined; const queuedConversationIds = useMemo( - () => - new Set( - conversationQueueItems.map((item) => item.conversationId) - ), - [conversationQueueItems] - ) + () => new Set(conversationQueueItems.map((item) => item.conversationId)), + [conversationQueueItems], + ); const filteredConversations = useMemo(() => { - const query = deferredSearchQuery.trim().toLocaleLowerCase() - const candidates = query - ? searchConversationSnapshot - : conversations + const query = deferredSearchQuery.trim().toLocaleLowerCase(); + const candidates = query ? searchConversationSnapshot : conversations; return candidates.filter( (conversation) => - (!activeProjectId || - conversation.projectId === activeProjectId) && - (activeProject?.kind !== 'channel' || + (!activeProjectId || conversation.projectId === activeProjectId) && + (activeProject?.kind !== "channel" || conversation.remote !== undefined) && (!query || - conversation.title.toLocaleLowerCase().includes(query) || - conversation.messages.some((message) => - message.content.toLocaleLowerCase().includes(query) - )) - ) + conversation.title.toLocaleLowerCase().includes(query) || + conversation.messages.some((message) => + message.content.toLocaleLowerCase().includes(query), + )), + ); }, [ activeProject, activeProjectId, conversations, deferredSearchQuery, - searchConversationSnapshot - ]) + searchConversationSnapshot, + ]); const productAssistantTasks = useMemo( () => assistantTasks.filter( - (task) => !task.parentTaskId && task.origin === 'schedule' + (task) => !task.parentTaskId && task.origin === "schedule", ), - [assistantTasks] - ) + [assistantTasks], + ); const tasksByConversation = useMemo(() => { - const grouped = new Map() + const grouped = new Map(); for (const task of productAssistantTasks) { if (!task.conversationId) { - continue + continue; } - const existing = grouped.get(task.conversationId) ?? [] - existing.push(task) - grouped.set(task.conversationId, existing) + const existing = grouped.get(task.conversationId) ?? []; + existing.push(task); + grouped.set(task.conversationId, existing); } for (const tasks of grouped.values()) { tasks.sort((left, right) => - right.createdAt.localeCompare(left.createdAt) - ) + right.createdAt.localeCompare(left.createdAt), + ); } - return grouped - }, [productAssistantTasks]) + return grouped; + }, [productAssistantTasks]); const conversationTitles = useMemo( () => new Map( @@ -3878,22 +3698,22 @@ function App(): React.JSX.Element { conversation.id, getConversationDisplayTitle( conversation, - t('conversation.defaultTitle') - ) - ]) + t("conversation.defaultTitle"), + ), + ]), ), - [conversations, t] - ) + [conversations, t], + ); const projectNames = useMemo( () => new Map( projects.map((project) => [ project.id, - getProjectDisplayText(project, tWorkspace).name - ]) + getProjectDisplayText(project, tWorkspace).name, + ]), ), - [projects, tWorkspace] - ) + [projects, tWorkspace], + ); const pendingSidebarApprovals = useMemo( () => conversations.flatMap((conversation) => @@ -3906,56 +3726,55 @@ function App(): React.JSX.Element { approvalId: message.approval.id, title: message.approval.title, description: message.approval.description, - toolName: message.approval.toolName - } + toolName: message.approval.toolName, + }, ] - : [] - ) + : [], + ), ), - [conversations] - ) + [conversations], + ); const sidebarArtifacts = useMemo( () => assistantArtifacts .filter( (artifact) => - !activeProjectId || artifact.projectId === activeProjectId + !activeProjectId || artifact.projectId === activeProjectId, ) .map((artifact) => ({ id: artifact.id, title: artifact.title, - content: artifact.content ?? '', + content: artifact.content ?? "", createdAt: new Date(artifact.createdAt).getTime(), - mimeType: artifact.mimeType + mimeType: artifact.mimeType, })), - [activeProjectId, assistantArtifacts] - ) + [activeProjectId, assistantArtifacts], + ); const pendingHeartbeatSuggestionCount = useMemo(() => { const memoryIds = new Set( - heartbeatEntries.flatMap((entry) => entry.proposedMemoryIds) - ) + heartbeatEntries.flatMap((entry) => entry.proposedMemoryIds), + ); const taskIds = new Set( - heartbeatEntries.flatMap((entry) => entry.followUpTaskIds) - ) + heartbeatEntries.flatMap((entry) => entry.followUpTaskIds), + ); return ( heartbeatMemories.filter( - (memory) => - memoryIds.has(memory.id) && memory.status === 'proposed' + (memory) => memoryIds.has(memory.id) && memory.status === "proposed", ).length + assistantTasks.filter( (task) => taskIds.has(task.id) && - task.status !== 'completed' && - task.status !== 'cancelled' + task.status !== "completed" && + task.status !== "cancelled", ).length - ) - }, [assistantTasks, heartbeatEntries, heartbeatMemories]) + ); + }, [assistantTasks, heartbeatEntries, heartbeatMemories]); const updateMessage = useCallback( ( conversationId: string, messageId: string, - update: (message: Message) => Message + update: (message: Message) => Message, ): void => { setConversations((current) => current.map((conversation) => @@ -3964,88 +3783,88 @@ function App(): React.JSX.Element { ...conversation, updatedAt: Date.now(), messages: conversation.messages.map((message) => - message.id === messageId ? update(message) : message - ) + message.id === messageId ? update(message) : message, + ), } - : conversation - ) - ) + : conversation, + ), + ); }, - [] - ) + [], + ); const recordActivity = useCallback( ( - record: Omit, - scopeOverride?: ActivityRecord['scope'] + record: Omit, + scopeOverride?: ActivityRecord["scope"], ): void => { const conversation = conversationsRef.current.find( - (candidate) => candidate.id === record.conversationId - ) + (candidate) => candidate.id === record.conversationId, + ); const project = conversation?.projectId ? projectsRef.current.find( - (candidate) => candidate.id === conversation.projectId + (candidate) => candidate.id === conversation.projectId, ) - : undefined - const scope: ActivityRecord['scope'] = + : undefined; + const scope: ActivityRecord["scope"] = scopeOverride ?? (!conversation - ? { kind: 'unavailable' } + ? { kind: "unavailable" } : !conversation.projectId - ? { kind: 'global' } + ? { kind: "global" } : project?.id && project.name ? { - kind: 'project', + kind: "project", projectId: project.id.slice(0, 256), - projectName: project.name.slice(0, 120) + projectName: project.name.slice(0, 120), } - : { kind: 'unavailable' }) + : { kind: "unavailable" }); setActivityRecords((current) => upsertActivityRecord(current, { ...record, title: record.title.slice(0, 240), scope, id: crypto.randomUUID(), - createdAt: Date.now() - }) - ) + createdAt: Date.now(), + }), + ); }, - [] - ) + [], + ); const updateRequestActivity = useCallback( ( requestId: string, - status: ActivityRecord['status'], - detail?: string + status: ActivityRecord["status"], + detail?: string, ): void => { setActivityRecords((current) => current.map((record) => - record.requestId === requestId && record.kind === 'request' + record.requestId === requestId && record.kind === "request" ? { ...record, status, - detail: detail ?? record.detail + detail: detail ?? record.detail, } - : record - ) - ) + : record, + ), + ); }, - [] - ) + [], + ); useEffect(() => { - const api = window.goodbuddy.channels + const api = window.goodbuddy.channels; if (!api) { - return + return; } return api.onRemoteActivity((activity) => { - if (activity.kind === 'result') { + if (activity.kind === "result") { updateRequestActivity( activity.requestId, activity.status, - activity.detail - ) + activity.detail, + ); } recordActivity( { @@ -4055,74 +3874,74 @@ function App(): React.JSX.Element { kind: activity.kind, title: activity.title, detail: activity.detail, - status: activity.status + status: activity.status, }, { - kind: 'project', + kind: "project", projectId: activity.projectId.slice(0, 256), - projectName: activity.projectName.slice(0, 120) - } - ) - }) - }, [recordActivity, updateRequestActivity]) + projectName: activity.projectName.slice(0, 120), + }, + ); + }); + }, [recordActivity, updateRequestActivity]); const refreshKnowledge = useCallback( async (libraryId?: string): Promise => { - const requestId = ++knowledgeLoadRequestRef.current + const requestId = ++knowledgeLoadRequestRef.current; try { const snapshot = - await window.goodbuddy.knowledge.getSnapshot(libraryId) + await window.goodbuddy.knowledge.getSnapshot(libraryId); if (requestId !== knowledgeLoadRequestRef.current) { - return snapshot + return snapshot; } - failedKnowledgeLibraryIdRef.current = undefined - setKnowledgeSnapshot(snapshot) - setKnowledgeLoadError(undefined) + failedKnowledgeLibraryIdRef.current = undefined; + setKnowledgeSnapshot(snapshot); + setKnowledgeLoadError(undefined); const availableIds = new Set( - snapshot.libraries.map((library) => library.id) - ) + snapshot.libraries.map((library) => library.id), + ); setConversations((current) => current.map((conversation) => { - const previous = conversation.knowledgeLibraryIds ?? [] - const next = previous.filter((id) => availableIds.has(id)) + const previous = conversation.knowledgeLibraryIds ?? []; + const next = previous.filter((id) => availableIds.has(id)); return previous.length === next.length ? conversation : { ...conversation, knowledgeLibraryIds: next, - updatedAt: Date.now() - } - }) - ) - return snapshot + updatedAt: Date.now(), + }; + }), + ); + return snapshot; } catch (reason) { if (requestId !== knowledgeLoadRequestRef.current) { - throw reason + throw reason; } - failedKnowledgeLibraryIdRef.current = libraryId + failedKnowledgeLibraryIdRef.current = libraryId; setKnowledgeLoadError( displayErrorMessage( reason, - tRef.current('notices.knowledgeReadFailed') - ) - ) - throw reason + tRef.current("notices.knowledgeReadFailed"), + ), + ); + throw reason; } }, - [] - ) + [], + ); const retryKnowledgeLoad = useCallback(async (): Promise => { - setKnowledgeLoading(true) - setKnowledgeLoadError(undefined) + setKnowledgeLoading(true); + setKnowledgeLoadError(undefined); try { - await refreshKnowledge(failedKnowledgeLibraryIdRef.current) + await refreshKnowledge(failedKnowledgeLibraryIdRef.current); } catch { // The recoverable page state is set by refreshKnowledge. } finally { - setKnowledgeLoading(false) + setKnowledgeLoading(false); } - }, [refreshKnowledge]) + }, [refreshKnowledge]); const switchRuntime = useCallback( async (selection: AgentRuntimeSelection): Promise => { @@ -4130,73 +3949,82 @@ function App(): React.JSX.Element { !runtimeSettings || !activeConversation || runtimeSwitching || - (activeProjectUsesManagedSsh && - selection.provider !== 'opencode') + (activeProjectUsesManagedSsh && selection.provider !== "opencode") ) { - return + return; } - runtimeMenuButtonRef.current?.focus() - setRuntimeSwitching(true) - setRuntimeMenuOpen(false) - const requestId = runtimeStatusRequestRef.current + 1 - runtimeStatusRequestRef.current = requestId + runtimeMenuButtonRef.current?.focus(); + setRuntimeSwitching(true); + setRuntimeMenuOpen(false); + const requestId = runtimeStatusRequestRef.current + 1; + runtimeStatusRequestRef.current = requestId; + const generation = runtimeSwitchGenerationRef.current; try { - const status = await window.goodbuddy.agent.getStatus(selection) - if (runtimeStatusRequestRef.current !== requestId) { - return + const status = await window.goodbuddy.agent.getStatus(selection); + if ( + runtimeStatusRequestRef.current !== requestId || + runtimeSwitchGenerationRef.current !== generation + ) { + return; } - const selectionKey = agentRuntimeSelectionKey(selection) + const selectionKey = agentRuntimeSelectionKey(selection); runtimeStatusCacheRef.current = { key: selectionKey, - settings: runtimeSettings - } + settings: runtimeSettings, + }; const label = getRuntimeSelectionLabel( selection, runtimeSettings, status, - runtimeLabels - ) + runtimeLabels, + ); setConversations((current) => current.map((conversation) => conversation.id === activeConversation.id ? { ...conversation, runtimeSelection: selection, - updatedAt: Date.now() + updatedAt: Date.now(), } - : conversation - ) - ) - setRuntime(status) - setRuntimeStatusKey(selectionKey) + : conversation, + ), + ); + setRuntime(status); + setRuntimeStatusKey(selectionKey); notify({ - tone: status.available ? 'success' : 'error', + tone: status.available ? "success" : "error", message: status.available - ? tRef.current('runtime.switched', { label }) - : tRef.current('runtime.selectionUnavailable', { + ? tRef.current("runtime.switched", { label }) + : tRef.current("runtime.selectionUnavailable", { label, - detail: status.detail + detail: status.detail, }), - dedupeKey: 'runtime-switch' - }) + dedupeKey: "runtime-switch", + }); } catch (reason) { - if (runtimeStatusRequestRef.current !== requestId) { - return + if ( + runtimeStatusRequestRef.current !== requestId || + runtimeSwitchGenerationRef.current !== generation + ) { + return; } notify({ - tone: 'error', + tone: "error", message: reason instanceof Error ? reason.message - : tRef.current('runtime.errors.switch'), - dedupeKey: 'runtime-switch' - }) + : tRef.current("runtime.errors.switch"), + dedupeKey: "runtime-switch", + }); } finally { - if (runtimeStatusRequestRef.current === requestId) { - setRuntimeSwitching(false) + if ( + runtimeStatusRequestRef.current === requestId && + runtimeSwitchGenerationRef.current === generation + ) { + setRuntimeSwitching(false); requestAnimationFrame(() => { - runtimeMenuButtonRef.current?.focus() - }) + runtimeMenuButtonRef.current?.focus(); + }); } } }, @@ -4205,105 +4033,97 @@ function App(): React.JSX.Element { activeProjectUsesManagedSsh, runtimeLabels, runtimeSettings, - runtimeSwitching - ] - ) + runtimeSwitching, + ], + ); const refreshTokenUsage = useCallback(async (): Promise => { - setTokenUsage(await window.goodbuddy.usage.getTokenSummary()) - }, []) + setTokenUsage(await window.goodbuddy.usage.getTokenSummary()); + }, []); const loadWorkspaceChanges = useCallback( async (projectId: string): Promise => { - const requestId = workspaceChangesRequestRef.current + 1 - workspaceChangesRequestRef.current = requestId - const changes = await window.goodbuddy.workspace.getChanges(projectId) + const requestId = workspaceChangesRequestRef.current + 1; + workspaceChangesRequestRef.current = requestId; + const changes = await window.goodbuddy.workspace.getChanges(projectId); if ( workspaceChangesRequestRef.current === requestId && activeProjectIdRef.current === projectId ) { - setWorkspaceChanges(changes) + setWorkspaceChanges(changes); } }, - [] - ) + [], + ); const releaseConversationQueueAfterRun = useCallback( - (run: Pick): void => { + (run: Pick): void => { requestAnimationFrame(() => { if ( run.projectId && ((!projectRecoverySnapshotReadyRef.current && isManagedSshProject( projectsRef.current.find( - (project) => project.id === run.projectId - ) + (project) => project.id === run.projectId, + ), )) || isProjectRecoveryUnsettled( - projectRecoveryByProjectIdRef.current[run.projectId] + projectRecoveryByProjectIdRef.current[run.projectId], )) ) { - return + return; } void window.goodbuddy.conversationQueue .ready(run.conversationId) .catch(() => { notify({ - tone: 'error', - message: tRef.current( - 'notices.conversationQueueResumeFailed' - ), - dedupeKey: 'conversation-queue-resume' - }) - }) - }) + tone: "error", + message: tRef.current("notices.conversationQueueResumeFailed"), + dedupeKey: "conversation-queue-resume", + }); + }); + }); }, - [notify] - ) + [notify], + ); const handleAgentEvent = useCallback( (event: AgentEvent): void => { - const run = activeRuns.current.get(event.requestId) + const run = activeRuns.current.get(event.requestId); if (!run) { - if (event.type !== 'approval') { - return + if (event.type !== "approval") { + return; } const attachScheduledApproval = ( - task: AssistantTask | undefined + task: AssistantTask | undefined, ): void => { - if ( - !task?.conversationId || - task.origin !== 'schedule' - ) { - return + if (!task?.conversationId || task.origin !== "schedule") { + return; } setAssistantTasks((current) => current.map((candidate) => candidate.id === task.id - ? { ...candidate, status: 'waiting_approval' } - : candidate - ) - ) - if ( - activeConversationIdRef.current !== task.conversationId - ) { + ? { ...candidate, status: "waiting_approval" } + : candidate, + ), + ); + if (activeConversationIdRef.current !== task.conversationId) { setUnreadConversationIds((current) => { - const next = new Set(current) - next.add(task.conversationId!) - return next - }) + const next = new Set(current); + next.add(task.conversationId!); + return next; + }); } setConversations((current) => current.map((conversation) => { if (conversation.id !== task.conversationId) { - return conversation + return conversation; } const existing = conversation.messages.find( - (message) => - message.approval?.id === event.approvalId - ) + (message) => message.approval?.id === event.approvalId, + ); if (existing) { - return conversation + return conversation; } return { ...conversation, @@ -4312,13 +4132,13 @@ function App(): React.JSX.Element { ...conversation.messages, { id: crypto.randomUUID(), - role: 'assistant', + role: "assistant", content: event.title, createdAt: Date.now(), - state: 'complete', + state: "complete", task: { id: task.id, - title: task.title + title: task.title, }, approval: { id: event.approvalId, @@ -4326,175 +4146,162 @@ function App(): React.JSX.Element { description: event.description, toolName: event.toolName, argumentSummary: event.argumentSummary, - allowPermanent: event.allowPermanent - } - } - ] - } - }) - ) - } + allowPermanent: event.allowPermanent, + }, + }, + ], + }; + }), + ); + }; const task = assistantTasksRef.current.find( - (candidate) => candidate.id === event.requestId - ) + (candidate) => candidate.id === event.requestId, + ); if (task) { - attachScheduledApproval(task) + attachScheduledApproval(task); } else { void window.goodbuddy.tasks .list() .then((tasks) => { - setAssistantTasks(tasks) - assistantTasksRef.current = tasks + setAssistantTasks(tasks); + assistantTasksRef.current = tasks; attachScheduledApproval( - tasks.find( - (candidate) => candidate.id === event.requestId - ) - ) + tasks.find((candidate) => candidate.id === event.requestId), + ); }) - .catch(() => undefined) + .catch(() => undefined); } - return + return; } setAssistantTasks((current) => { - let changed = false + let changed = false; const updated = current.map((task) => { if (task.id !== event.requestId) { - return task + return task; } - const status: AssistantTask['status'] = - event.type === 'approval' || event.type === 'question' - ? 'waiting_approval' - : event.type === 'done' - ? 'completed' - : event.type === 'error' + const status: AssistantTask["status"] = + event.type === "approval" || event.type === "question" + ? "waiting_approval" + : event.type === "done" + ? "completed" + : event.type === "error" ? event.status - : 'running' + : "running"; const completedAt = - event.type === 'done' || event.type === 'error' + event.type === "done" || event.type === "error" ? new Date().toISOString() - : task.completedAt - const error = - event.type === 'error' ? event.message : task.error + : task.completedAt; + const error = event.type === "error" ? event.message : task.error; if ( task.status === status && task.completedAt === completedAt && task.error === error ) { - return task + return task; } - changed = true + changed = true; return { ...task, status, completedAt, - error - } - }) - return changed ? updated : current - }) - if (event.type === 'done') { - if ( - run.projectId && - activeProjectIdRef.current === run.projectId - ) { + error, + }; + }); + return changed ? updated : current; + }); + if (event.type === "done") { + if (run.projectId && activeProjectIdRef.current === run.projectId) { void loadWorkspaceChanges(run.projectId).catch(() => notify({ - tone: 'error', - message: tRef.current( - 'notices.workspaceChangesReadFailed' - ) - }) - ) + tone: "error", + message: tRef.current("notices.workspaceChangesReadFailed"), + }), + ); } - if (viewRef.current === 'activity') { + if (viewRef.current === "activity") { void refreshTokenUsage().catch(() => notify({ - tone: 'error', - message: tRef.current('notices.tokenUsageReadFailed') - }) - ) + tone: "error", + message: tRef.current("notices.tokenUsageReadFailed"), + }), + ); } void window.goodbuddy.artifacts .list() .then((artifacts) => setAssistantArtifacts((current) => - mergeArtifacts(current, artifacts) - ) + mergeArtifacts(current, artifacts), + ), ) .catch(() => notify({ - tone: 'error', - message: tRef.current('notices.resultsRefreshFailed') - }) - ) - } else if (event.type === 'artifact') { - hydratingArtifactIds.current.add(event.artifactId) + tone: "error", + message: tRef.current("notices.resultsRefreshFailed"), + }), + ); + } else if (event.type === "artifact") { + hydratingArtifactIds.current.add(event.artifactId); void window.goodbuddy.artifacts .get(event.artifactId) .then((artifact) => setAssistantArtifacts((current) => - mergeArtifacts(current, [artifact]) - ) + mergeArtifacts(current, [artifact]), + ), ) .catch(() => notify({ - tone: 'error', - message: tRef.current( - 'notices.generatedImageReadFailed' - ) - }) + tone: "error", + message: tRef.current("notices.generatedImageReadFailed"), + }), ) .finally(() => { - hydratingArtifactIds.current.delete(event.artifactId) - }) + hydratingArtifactIds.current.delete(event.artifactId); + }); } - if (event.type === 'text') { + if (event.type === "text") { updateMessage(run.conversationId, run.messageId, (message) => { const remaining = Math.max( 0, - maxMessageContentLength - message.content.length - ) - const acceptedDelta = event.delta.slice(0, remaining) + maxMessageContentLength - message.content.length, + ); + const acceptedDelta = event.delta.slice(0, remaining); const blocks = appendMessageContentBlock( message.blocks, - 'text', - acceptedDelta - ) + "text", + acceptedDelta, + ); return { ...message, content: `${message.content}${acceptedDelta}`, blocks, status: event.delta.length > remaining - ? tRef.current('chat.status.responseTruncated') - : undefined - } - }) - } else if (event.type === 'reasoning') { + ? tRef.current("chat.status.responseTruncated") + : undefined, + }; + }); + } else if (event.type === "reasoning") { updateMessage(run.conversationId, run.messageId, (message) => { - const currentReasoning = message.reasoning ?? '' + const currentReasoning = message.reasoning ?? ""; const acceptedDelta = event.delta.slice( 0, - Math.max( - 0, - maxMessageContentLength - currentReasoning.length - ) - ) + Math.max(0, maxMessageContentLength - currentReasoning.length), + ); const blocks = appendMessageContentBlock( message.blocks, - 'reasoning', - acceptedDelta - ) + "reasoning", + acceptedDelta, + ); return { ...message, reasoning: `${currentReasoning}${acceptedDelta}`, status: undefined, - blocks - } - }) - } else if (event.type === 'context-metrics') { + blocks, + }; + }); + } else if (event.type === "context-metrics") { setConversations((current) => current.map((conversation) => conversation.id === run.conversationId @@ -4503,56 +4310,49 @@ function App(): React.JSX.Element { contextMetrics: { contextTokens: event.contextTokens, source: event.source, - basis: 'model-call', - runtimeSelectionKey: run.runtimeSelectionKey - } + basis: "model-call", + runtimeSelectionKey: run.runtimeSelectionKey, + }, } - : conversation - ) - ) - } else if (event.type === 'context-compression') { - const estimatedAfterTokens = event.estimatedAfterTokens - const conversationScoped = event.scope !== 'agent-run' - const scope = event.scope ?? 'conversation' + : conversation, + ), + ); + } else if (event.type === "context-compression") { + const estimatedAfterTokens = event.estimatedAfterTokens; + const conversationScoped = event.scope !== "agent-run"; + const scope = event.scope ?? "conversation"; const marker: ConversationContextCompressionMarker = { - state: - event.state === 'started' - ? 'compressing' - : event.state, - scope, - estimatedBeforeTokens: event.estimatedBeforeTokens, - estimatedAfterTokens: event.estimatedAfterTokens, - compressionCount: event.compressionCount - } + state: event.state === "started" ? "compressing" : event.state, + scope, + estimatedBeforeTokens: event.estimatedBeforeTokens, + estimatedAfterTokens: event.estimatedAfterTokens, + compressionCount: event.compressionCount, + }; updateMessage(run.conversationId, run.messageId, (message) => { const current = message.contextCompressions ?? - (message.contextCompression - ? [message.contextCompression] - : []) + (message.contextCompression ? [message.contextCompression] : []); const existingIndex = current.findIndex( - (compression) => - (compression.scope ?? 'conversation') === scope - ) + (compression) => (compression.scope ?? "conversation") === scope, + ); const contextCompressions = existingIndex >= 0 ? [ ...current.filter( - (_compression, index) => - index !== existingIndex + (_compression, index) => index !== existingIndex, ), - marker + marker, ] - : [...current, marker] + : [...current, marker]; return { ...message, contextCompression: undefined, - contextCompressions - } - }) + contextCompressions, + }; + }); if ( conversationScoped && - event.state === 'completed' && + event.state === "completed" && estimatedAfterTokens !== undefined ) { setConversations((current) => @@ -4563,60 +4363,55 @@ function App(): React.JSX.Element { contextMetrics: { runtimeSelectionKey: run.runtimeSelectionKey, contextTokens: estimatedAfterTokens, - source: 'estimated', - basis: 'conversation' + source: "estimated", + basis: "conversation", }, contextCompressionState: event.conversationState ?? - conversation.contextCompressionState + conversation.contextCompressionState, } - : conversation - ) - ) - } else if ( - conversationScoped && - event.conversationState - ) { + : conversation, + ), + ); + } else if (conversationScoped && event.conversationState) { setConversations((current) => current.map((conversation) => conversation.id === run.conversationId ? { ...conversation, - contextCompressionState: event.conversationState + contextCompressionState: event.conversationState, } - : conversation - ) - ) + : conversation, + ), + ); } - } else if (event.type === 'status') { + } else if (event.type === "status") { updateMessage(run.conversationId, run.messageId, (message) => ({ ...message, - status: event.message - })) - } else if (event.type === 'tool') { + status: event.message, + })); + } else if (event.type === "tool") { recordActivity({ conversationId: run.conversationId, requestId: event.requestId, callId: event.callId.slice(0, 256), - kind: 'tool', + kind: "tool", title: event.name, - detail: [event.summary, event.error] - .filter(Boolean) - .join('\n'), + detail: [event.summary, event.error].filter(Boolean).join("\n"), status: - event.state === 'pending' - ? 'pending' - : event.state === 'running' - ? 'running' - : event.state === 'failed' - ? 'failed' - : 'completed' - }) + event.state === "pending" + ? "pending" + : event.state === "running" + ? "running" + : event.state === "failed" + ? "failed" + : "completed", + }); updateMessage(run.conversationId, run.messageId, (message) => { - const tools = [...(message.tools ?? [])] + const tools = [...(message.tools ?? [])]; const index = tools.findIndex( - (tool) => tool.callId === event.callId.slice(0, 256) - ) + (tool) => tool.callId === event.callId.slice(0, 256), + ); const tool = { callId: event.callId.slice(0, 256), name: event.name, @@ -4624,45 +4419,45 @@ function App(): React.JSX.Element { summary: event.summary, input: event.input, output: event.output, - error: event.error - } + error: event.error, + }; if (index >= 0) { - tools[index] = tool + tools[index] = tool; } else { - tools.push(tool) + tools.push(tool); } - const blocks = upsertMessageToolBlock(message.blocks, tool) + const blocks = upsertMessageToolBlock(message.blocks, tool); return { ...message, tools, - blocks - } - }) - } else if (event.type === 'subagent') { + blocks, + }; + }); + } else if (event.type === "subagent") { const actor = - 'actor' in event + "actor" in event ? event.actor : { - kind: 'expert' as const, + kind: "expert" as const, expertId: event.expertId, - expertName: event.expertName - } + expertName: event.expertName, + }; const actorLabel = - actor.kind === 'direct-model' - ? tRef.current('chat.subagents.directModelLabel') - : actor.expertName - const childStatus = event.state + actor.kind === "direct-model" + ? tRef.current("chat.subagents.directModelLabel") + : actor.expertName; + const childStatus = event.state; const completedAt = - event.state === 'completed' || - event.state === 'failed' || - event.state === 'cancelled' + event.state === "completed" || + event.state === "failed" || + event.state === "cancelled" ? new Date().toISOString() - : undefined - if (actor.kind === 'expert') { + : undefined; + if (actor.kind === "expert") { setAssistantTasks((current) => { const existing = current.find( - (task) => task.id === event.childTaskId - ) + (task) => task.id === event.childTaskId, + ); const childTask: AssistantTask = { id: event.childTaskId, projectId: run.projectId, @@ -4670,32 +4465,29 @@ function App(): React.JSX.Element { parentTaskId: event.requestId, expertId: actor.expertId, routingMode: - event.routingMode === 'native' - ? undefined - : event.routingMode, + event.routingMode === "native" ? undefined : event.routingMode, title: actor.expertName, instructions: event.reason ?? - tRef.current('chat.subagents.fallbackTask', { - name: actor.expertName + tRef.current("chat.subagents.fallbackTask", { + name: actor.expertName, }), - origin: 'subagent', + origin: "subagent", status: childStatus, - createdAt: - existing?.createdAt ?? new Date().toISOString(), + createdAt: existing?.createdAt ?? new Date().toISOString(), startedAt: - event.state === 'running' - ? existing?.startedAt ?? new Date().toISOString() + event.state === "running" + ? (existing?.startedAt ?? new Date().toISOString()) : existing?.startedAt, completedAt: completedAt ?? existing?.completedAt, - error: event.error - } + error: event.error, + }; return existing ? current.map((task) => - task.id === event.childTaskId ? childTask : task + task.id === event.childTaskId ? childTask : task, ) - : [...current, childTask] - }) + : [...current, childTask]; + }); } if (event.runtimeCallId) { setActivityRecords((current) => @@ -4703,47 +4495,43 @@ function App(): React.JSX.Element { (record) => !( record.requestId === event.requestId && - record.kind === 'tool' && + record.kind === "tool" && record.callId === event.runtimeCallId - ) - ) - ) + ), + ), + ); } recordActivity({ conversationId: run.conversationId, requestId: event.requestId, callId: event.childTaskId, - kind: 'subagent', + kind: "subagent", title: actorLabel, detail: [ - actor.kind === 'direct-model' - ? tRef.current('chat.subagents.directModel') - : event.routingMode === 'smart' - ? tRef.current('chat.subagents.smart') - : event.routingMode === 'native' - ? tRef.current('chat.subagents.native') - : tRef.current('chat.subagents.manual'), - actor.kind === 'direct-model' && event.workMode - ? event.workMode === 'execute' - ? 'Execute' - : 'Ask' + actor.kind === "direct-model" + ? tRef.current("chat.subagents.directModel") + : event.routingMode === "smart" + ? tRef.current("chat.subagents.smart") + : event.routingMode === "native" + ? tRef.current("chat.subagents.native") + : tRef.current("chat.subagents.manual"), + actor.kind === "direct-model" && event.workMode + ? event.workMode === "execute" + ? "Execute" + : "Ask" : undefined, event.reason, - event.error + event.error, ] .filter(Boolean) - .join(' · '), - status: - event.state === 'queued' - ? 'pending' - : event.state - }) + .join(" · "), + status: event.state === "queued" ? "pending" : event.state, + }); updateMessage(run.conversationId, run.messageId, (message) => { - const subagents = [...(message.subagents ?? [])] + const subagents = [...(message.subagents ?? [])]; const index = subagents.findIndex( - (subagent) => - subagent.childTaskId === event.childTaskId - ) + (subagent) => subagent.childTaskId === event.childTaskId, + ); const commonSubagent = { childTaskId: event.childTaskId, routingMode: event.routingMode, @@ -4753,47 +4541,45 @@ function App(): React.JSX.Element { reason: event.reason, progress: event.progress, output: event.output, - error: event.error - } + error: event.error, + }; const subagent: SubagentActivity = - 'actor' in event + "actor" in event ? { ...commonSubagent, actor: event.actor } : { ...commonSubagent, expertId: event.expertId, - expertName: event.expertName - } + expertName: event.expertName, + }; if (index >= 0) { - subagents[index] = subagent + subagents[index] = subagent; } else { - subagents.push(subagent) + subagents.push(subagent); } - const runtimeCallId = event.runtimeCallId + const runtimeCallId = event.runtimeCallId; const blocks = upsertMessageSubagentBlock( message.blocks, event.childTaskId, - runtimeCallId - ) + runtimeCallId, + ); return { ...message, subagents, tools: runtimeCallId - ? message.tools?.filter( - (tool) => tool.callId !== runtimeCallId - ) + ? message.tools?.filter((tool) => tool.callId !== runtimeCallId) : message.tools, - blocks - } - }) - } else if (event.type === 'approval') { + blocks, + }; + }); + } else if (event.type === "approval") { recordActivity({ conversationId: run.conversationId, requestId: event.requestId, - kind: 'approval', + kind: "approval", title: event.title, detail: event.description, - status: 'pending' - }) + status: "pending", + }); updateMessage(run.conversationId, run.messageId, (message) => ({ ...message, status: undefined, @@ -4803,24 +4589,24 @@ function App(): React.JSX.Element { description: event.description, toolName: event.toolName, argumentSummary: event.argumentSummary, - allowPermanent: event.allowPermanent - } - })) - } else if (event.type === 'question') { + allowPermanent: event.allowPermanent, + }, + })); + } else if (event.type === "question") { updateMessage(run.conversationId, run.messageId, (message) => ({ ...message, status: undefined, - question: event - })) - } else if (event.type === 'artifact') { + question: event, + })); + } else if (event.type === "artifact") { updateMessage(run.conversationId, run.messageId, (message) => ({ ...message, artifactIds: [ - ...new Set([...(message.artifactIds ?? []), event.artifactId]) + ...new Set([...(message.artifactIds ?? []), event.artifactId]), ].slice(-8), - status: tRef.current('chat.status.savingImage') - })) - } else if (event.type === 'knowledge-retrieval') { + status: tRef.current("chat.status.savingImage"), + })); + } else if (event.type === "knowledge-retrieval") { updateMessage(run.conversationId, run.messageId, (message) => ({ ...message, knowledgeRetrieval: { @@ -4830,216 +4616,201 @@ function App(): React.JSX.Element { resultCount: event.resultCount, durationMs: event.durationMs, usedChannels: event.usedChannels, - warnings: event.warnings + warnings: event.warnings, }, status: - event.state === 'searching' - ? tRef.current('chat.knowledgeRetrieval.searching') - : undefined - })) - } else if (event.type === 'source-references') { + event.state === "searching" + ? tRef.current("chat.knowledgeRetrieval.searching") + : undefined, + })); + } else if (event.type === "source-references") { updateMessage(run.conversationId, run.messageId, (message) => { - const referenceKey = ( - reference: KnowledgeSearchReference - ): string => + const referenceKey = (reference: KnowledgeSearchReference): string => [ reference.libraryId, reference.documentId, - reference.chunkId ?? '', - reference.locator ?? '', - reference.snippet - ].join('\0') + reference.chunkId ?? "", + reference.locator ?? "", + reference.snippet, + ].join("\0"); const incoming = [ ...new Map( event.references.map((reference) => [ referenceKey(reference), - reference - ]) - ).values() - ] - const incomingKeys = new Set(incoming.map(referenceKey)) + reference, + ]), + ).values(), + ]; + const incomingKeys = new Set(incoming.map(referenceKey)); const references = [ ...incoming, ...(message.sourceReferences ?? []).filter( - (reference) => !incomingKeys.has(referenceKey(reference)) - ) - ].slice(0, 20) + (reference) => !incomingKeys.has(referenceKey(reference)), + ), + ].slice(0, 20); return { ...message, - sourceReferences: references - } - }) + sourceReferences: references, + }; + }); } else { const terminalStatus = - event.type === 'error' - ? event.status === 'cancelled' - ? 'cancelled' - : 'failed' - : 'completed' + event.type === "error" + ? event.status === "cancelled" + ? "cancelled" + : "failed" + : "completed"; const incompleteSubagentError = tRef.current( - 'chat.subagents.incomplete' - ) + "chat.subagents.incomplete", + ); const incompleteActivityDetail = tRef.current( - 'chat.status.activityIncomplete' - ) + "chat.status.activityIncomplete", + ); updateRequestActivity( event.requestId, terminalStatus, - event.type === 'error' + event.type === "error" ? event.message - : tRef.current('chat.status.taskCompleted') - ) + : tRef.current("chat.status.taskCompleted"), + ); setActivityRecords((current) => current.map((record) => record.requestId === event.requestId && - record.kind !== 'request' && - (record.status === 'pending' || - record.status === 'running') + record.kind !== "request" && + (record.status === "pending" || record.status === "running") ? { ...record, status: - event.type === 'done' - ? 'interrupted' - : terminalStatus, + event.type === "done" ? "interrupted" : terminalStatus, detail: `${record.detail}\n${ - event.type === 'done' + event.type === "done" ? incompleteActivityDetail : event.message - }` + }`, } - : record - ) - ) + : record, + ), + ); recordActivity({ conversationId: run.conversationId, requestId: event.requestId, - kind: 'result', + kind: "result", title: - event.type === 'error' - ? tRef.current('chat.status.taskFailed') - : tRef.current('chat.status.taskCompleted'), + event.type === "error" + ? tRef.current("chat.status.taskFailed") + : tRef.current("chat.status.taskCompleted"), detail: - event.type === 'error' + event.type === "error" ? event.message - : tRef.current('chat.status.runtimeCompleted'), - status: terminalStatus - }) + : tRef.current("chat.status.runtimeCompleted"), + status: terminalStatus, + }); setAssistantTasks((current) => current.map((task) => task.parentTaskId === event.requestId && - (task.status === 'queued' || task.status === 'running') + (task.status === "queued" || task.status === "running") ? { ...task, - status: - event.type === 'done' - ? 'failed' - : terminalStatus, + status: event.type === "done" ? "failed" : terminalStatus, completedAt: new Date().toISOString(), error: - event.type === 'error' + event.type === "error" ? event.message - : incompleteSubagentError + : incompleteSubagentError, } - : task - ) - ) + : task, + ), + ); updateMessage(run.conversationId, run.messageId, (message) => { const representedToolError = - event.type === 'error' && - isErrorRepresentedByFailedTool( - message.tools, - event.message - ) + event.type === "error" && + isErrorRepresentedByFailedTool(message.tools, event.message); const toolTerminalState = - event.type === 'error' - ? event.status === 'cancelled' - ? ('cancelled' as const) - : ('failed' as const) - : ('interrupted' as const) + event.type === "error" + ? event.status === "cancelled" + ? ("cancelled" as const) + : ("failed" as const) + : ("interrupted" as const); const fallbackError = - event.type === 'error' && - !representedToolError && - !message.content + event.type === "error" && !representedToolError && !message.content ? event.message.slice(0, maxMessageContentLength) - : '' + : ""; return { ...message, - state: event.type === 'error' ? 'error' : 'complete', + state: event.type === "error" ? "error" : "complete", status: - event.type === 'error' && !representedToolError + event.type === "error" && !representedToolError ? event.message - : event.type === 'done' - ? tRef.current('chat.status.taskCompleted') + : event.type === "done" + ? tRef.current("chat.status.taskCompleted") : undefined, contextCompression: - event.type === 'error' && - message.contextCompression?.state === 'compressing' + event.type === "error" && + message.contextCompression?.state === "compressing" ? { ...message.contextCompression, - state: 'failed' as const + state: "failed" as const, } : message.contextCompression, contextCompressions: - event.type === 'error' + event.type === "error" ? message.contextCompressions?.map((compression) => - compression.state === 'compressing' + compression.state === "compressing" ? { ...compression, - state: 'failed' as const + state: "failed" as const, } - : compression + : compression, ) : message.contextCompressions, approval: undefined, question: undefined, tools: toolTerminalState ? message.tools?.map((tool) => - tool.state === 'pending' || tool.state === 'running' + tool.state === "pending" || tool.state === "running" ? { ...tool, state: toolTerminalState } - : tool + : tool, ) : message.tools, subagents: message.subagents?.map((subagent) => - subagent.state === 'queued' || - subagent.state === 'running' + subagent.state === "queued" || subagent.state === "running" ? { ...subagent, state: - event.type === 'error' - ? event.status === 'cancelled' - ? ('cancelled' as const) - : ('failed' as const) - : ('failed' as const), - ...(event.type === 'error' && - event.status !== 'cancelled' + event.type === "error" + ? event.status === "cancelled" + ? ("cancelled" as const) + : ("failed" as const) + : ("failed" as const), + ...(event.type === "error" && event.status !== "cancelled" ? { error: event.message.slice(0, 1_000) } - : event.type === 'done' + : event.type === "done" ? { error: incompleteSubagentError } - : {}) + : {}), } - : subagent + : subagent, ), blocks: toolTerminalState ? terminalizeMessageToolBlocks( appendMessageContentBlock( message.blocks, - 'text', - fallbackError + "text", + fallbackError, ), - toolTerminalState + toolTerminalState, ) : appendMessageContentBlock( message.blocks, - 'text', - fallbackError + "text", + fallbackError, ), - content: fallbackError || message.content - } - }) - activeRuns.current.delete(event.requestId) - setConversationActivity(run.conversationId, false) - releaseConversationQueueAfterRun(run) - flushConversationPersistenceAfterRenderRef.current = true + content: fallbackError || message.content, + }; + }); + activeRuns.current.delete(event.requestId); + setConversationActivity(run.conversationId, false); + releaseConversationQueueAfterRun(run); + flushConversationPersistenceAfterRenderRef.current = true; } }, [ @@ -5049,464 +4820,432 @@ function App(): React.JSX.Element { releaseConversationQueueAfterRun, setConversationActivity, updateMessage, - updateRequestActivity - ] - ) + updateRequestActivity, + ], + ); useEffect(() => { - activeProjectIdRef.current = activeProjectId + activeProjectIdRef.current = activeProjectId; if (activeProjectId) { try { - localStorage.setItem(activeProjectStorageKey, activeProjectId) + localStorage.setItem(activeProjectStorageKey, activeProjectId); } catch { // The project selection still works when persistence is unavailable. } } - }, [activeProjectId]) + }, [activeProjectId]); useEffect(() => { - viewRef.current = view - }, [view]) + viewRef.current = view; + }, [view]); const persistLocalConversationChanges = useCallback((): void => { - const operation = conversationPersistenceQueueRef.current.then( - async () => { - if (conversationPersistencePausedRef.current) { - return - } - const { batch, acknowledgements } = - createLocalConversationSaveBatch( - conversationsRef.current, - persistedLocalConversationsRef.current, - deletingLocalConversationIdsRef.current - ) - if (batch.length === 0) { - return - } - await window.goodbuddy.conversations.saveLocal(batch) - for (const conversation of acknowledgements) { - persistedLocalConversationsRef.current.set( - conversation.id, - conversation - ) - } + const operation = conversationPersistenceQueueRef.current.then(async () => { + if (conversationPersistencePausedRef.current) { + return; } - ) - conversationPersistenceQueueRef.current = - operation.catch(() => undefined) + const { batch, acknowledgements } = createLocalConversationSaveBatch( + conversationsRef.current, + persistedLocalConversationsRef.current, + deletingLocalConversationIdsRef.current, + ); + if (batch.length === 0) { + return; + } + await window.goodbuddy.conversations.saveLocal(batch); + for (const conversation of acknowledgements) { + persistedLocalConversationsRef.current.set( + conversation.id, + conversation, + ); + } + }); + conversationPersistenceQueueRef.current = operation.catch(() => undefined); void operation.catch(() => { notify({ - tone: 'error', - message: tRef.current( - 'notices.conversationPersistenceFailed' - ), - dedupeKey: 'conversation-persistence' - }) - }) - }, []) + tone: "error", + message: tRef.current("notices.conversationPersistenceFailed"), + dedupeKey: "conversation-persistence", + }); + }); + }, []); useEffect(() => { if (!conversationStoreReady) { - return + return; } - persistLocalConversationChanges() + persistLocalConversationChanges(); const interval = window.setInterval( persistLocalConversationChanges, - conversationPersistenceIntervalMs - ) + conversationPersistenceIntervalMs, + ); return () => { - window.clearInterval(interval) - persistLocalConversationChanges() - } - }, [conversationStoreReady, persistLocalConversationChanges]) + window.clearInterval(interval); + persistLocalConversationChanges(); + }; + }, [conversationStoreReady, persistLocalConversationChanges]); useEffect(() => { if ( !conversationStoreReady || !flushConversationPersistenceAfterRenderRef.current ) { - return + return; } - flushConversationPersistenceAfterRenderRef.current = false - persistLocalConversationChanges() - }, [ - conversationStoreReady, - conversations, - persistLocalConversationChanges - ]) + flushConversationPersistenceAfterRenderRef.current = false; + persistLocalConversationChanges(); + }, [conversationStoreReady, conversations, persistLocalConversationChanges]); const persistActivityHistory = useCallback(async (): Promise => { if (!activityHistoryReady) { - return + return; } try { await window.goodbuddy.activityHistory.replace( activityRecordsRef.current, - legacyActivityHistoryMayBeIncompleteRef.current - ) + legacyActivityHistoryMayBeIncompleteRef.current, + ); } catch { notify({ - tone: 'error', - message: tRef.current('notices.activityHistoryPersistenceFailed'), - dedupeKey: 'activity-history-persistence' - }) + tone: "error", + message: tRef.current("notices.activityHistoryPersistenceFailed"), + dedupeKey: "activity-history-persistence", + }); } - }, [activityHistoryReady]) + }, [activityHistoryReady]); useEffect( () => window.goodbuddy.app.onBeforeQuit(async () => { - await persistActivityHistory() + await persistActivityHistory(); if (!conversationStoreReady) { - return + return; } - flushConversationPersistenceAfterRenderRef.current = false - persistLocalConversationChanges() - await conversationPersistenceQueueRef.current + flushConversationPersistenceAfterRenderRef.current = false; + persistLocalConversationChanges(); + await conversationPersistenceQueueRef.current; }), [ conversationStoreReady, persistActivityHistory, - persistLocalConversationChanges - ] - ) + persistLocalConversationChanges, + ], + ); useEffect(() => { if (!conversationStoreReady) { - return + return; } - let active = true - let refreshSequence = 0 - let refreshTimer: number | undefined - let refreshInFlight = false - let refreshQueued = false + let active = true; + let refreshSequence = 0; + let refreshTimer: number | undefined; + let refreshInFlight = false; + let refreshQueued = false; const queueRefresh = (): void => { - refreshSequence += 1 - refreshQueued = true + refreshSequence += 1; + refreshQueued = true; if (refreshInFlight || refreshTimer !== undefined) { - return + return; } refreshTimer = window.setTimeout(() => { - refreshTimer = undefined - refresh() - }, 50) - } + refreshTimer = undefined; + refresh(); + }, 50); + }; // A run whose assistant message has already reached a persisted // terminal state (for example a remote task that finished while the // desktop was disconnected) must stop counting as in-flight, otherwise // the renderer keeps showing "processing" forever. const settleActiveRunsFromPersistedMessages = ( - persisted: readonly ConversationSnapshot[] + persisted: readonly ConversationSnapshot[], ): void => { if (activeRuns.current.size === 0) { - return + return; } - const persistedMessageStates = new Map() + const persistedMessageStates = new Map(); for (const conversation of persisted) { for (const message of conversation.messages) { persistedMessageStates.set( `${conversation.id}\0${message.id}`, - message.state - ) + message.state, + ); } } for (const [requestId, run] of activeRuns.current) { const state = persistedMessageStates.get( - `${run.conversationId}\0${run.messageId}` - ) - if (state !== undefined && state !== 'streaming') { - activeRuns.current.delete(requestId) - setConversationActivity(run.conversationId, false) - releaseConversationQueueAfterRun(run) + `${run.conversationId}\0${run.messageId}`, + ); + if (state !== undefined && state !== "streaming") { + activeRuns.current.delete(requestId); + setConversationActivity(run.conversationId, false); + releaseConversationQueueAfterRun(run); } } - } + }; const refresh = (): void => { if (refreshInFlight) { - refreshQueued = true - return + refreshQueued = true; + return; } - refreshInFlight = true - refreshQueued = false - const sequence = refreshSequence + refreshInFlight = true; + refreshQueued = false; + const sequence = refreshSequence; const tasksRefresh = window.goodbuddy.tasks .list() .then((tasks) => { if (!active || sequence !== refreshSequence) { - return + return; } - setAssistantTasks(tasks) + setAssistantTasks(tasks); setActivityRecords((current) => reconcileActivityRecords( current, tasks, - new Set(activeRuns.current.keys()) - ) - ) + new Set(activeRuns.current.keys()), + ), + ); }) .catch(() => { if (active && sequence === refreshSequence) { notify({ - tone: 'error', - message: tRef.current('notices.taskHistoryReadFailed'), - dedupeKey: 'task-lifecycle-refresh' - }) + tone: "error", + message: tRef.current("notices.taskHistoryReadFailed"), + dedupeKey: "task-lifecycle-refresh", + }); } - }) + }); const schedulesRefresh = window.goodbuddy.schedules .list() .then((schedules) => { if (active && sequence === refreshSequence) { - setAssistantSchedules(schedules) + setAssistantSchedules(schedules); } }) .catch(() => { if (active && sequence === refreshSequence) { notify({ - tone: 'error', - message: tRef.current('notices.schedulesReadFailed'), - dedupeKey: 'schedule-lifecycle-refresh' - }) + tone: "error", + message: tRef.current("notices.schedulesReadFailed"), + dedupeKey: "schedule-lifecycle-refresh", + }); } - }) + }); const conversationsRefresh = window.goodbuddy.conversations .list() .then((persisted) => { if (!active || sequence !== refreshSequence) { - return + return; } const remote = persisted.filter( - (conversation) => conversation.remote - ) + (conversation) => conversation.remote, + ); const previousById = new Map( conversationsRef.current.map((conversation) => [ conversation.id, - conversation - ]) - ) + conversation, + ]), + ); const updated = remote.filter((conversation) => { - const previous = previousById.get(conversation.id) + const previous = previousById.get(conversation.id); return ( previous === undefined || conversation.updatedAt > previous.updatedAt - ) - }) + ); + }); const unread = updated.filter( (conversation) => - conversation.id !== activeConversationIdRef.current - ) + conversation.id !== activeConversationIdRef.current, + ); if (unread.length > 0) { setUnreadConversationIds((current) => { - const next = new Set(current) - unread.forEach((conversation) => - next.add(conversation.id) - ) - return next - }) + const next = new Set(current); + unread.forEach((conversation) => next.add(conversation.id)); + return next; + }); notify({ - tone: 'info', - message: tRef.current('notices.remoteMessage', { - channel: - projectChannelLabels[ - unread[0]!.remote!.channel - ] + tone: "info", + message: tRef.current("notices.remoteMessage", { + channel: projectChannelLabels[unread[0]!.remote!.channel], }), - dedupeKey: 'remote-channel-message' - }) + dedupeKey: "remote-channel-message", + }); } setConversations((current) => mergePersistedConversations( current, persisted, - persistedLocalConversationsRef.current - ) - ) - settleActiveRunsFromPersistedMessages(persisted) + persistedLocalConversationsRef.current, + ), + ); + settleActiveRunsFromPersistedMessages(persisted); }) .catch(() => { if (active && sequence === refreshSequence) { notify({ - tone: 'error', - message: tRef.current( - 'notices.remoteConversationRefreshFailed' - ), - dedupeKey: 'remote-conversation-refresh' - }) + tone: "error", + message: tRef.current("notices.remoteConversationRefreshFailed"), + dedupeKey: "remote-conversation-refresh", + }); } - }) + }); void Promise.allSettled([ tasksRefresh, schedulesRefresh, - conversationsRefresh + conversationsRefresh, ]).finally(() => { - refreshInFlight = false + refreshInFlight = false; if (active && refreshQueued) { - queueRefresh() + queueRefresh(); } - }) - } + }); + }; const refreshWhenVisible = (): void => { - if (document.visibilityState !== 'hidden') { - queueRefresh() + if (document.visibilityState !== "hidden") { + queueRefresh(); } - } - const remove = window.goodbuddy.conversations.onChanged(queueRefresh) - window.addEventListener('focus', refreshWhenVisible) - document.addEventListener('visibilitychange', refreshWhenVisible) + }; + const remove = window.goodbuddy.conversations.onChanged(queueRefresh); + window.addEventListener("focus", refreshWhenVisible); + document.addEventListener("visibilitychange", refreshWhenVisible); return () => { - active = false + active = false; if (refreshTimer !== undefined) { - window.clearTimeout(refreshTimer) + window.clearTimeout(refreshTimer); } - remove() - window.removeEventListener('focus', refreshWhenVisible) - document.removeEventListener('visibilitychange', refreshWhenVisible) - } + remove(); + window.removeEventListener("focus", refreshWhenVisible); + document.removeEventListener("visibilitychange", refreshWhenVisible); + }; }, [ conversationStoreReady, releaseConversationQueueAfterRun, - setConversationActivity - ]) + setConversationActivity, + ]); useEffect(() => { - let active = true - let refreshInFlight = false - let refreshQueued = false - let refreshTimer: number | undefined - let refreshAll = false - const pendingConversationIds = new Set() + let active = true; + let refreshInFlight = false; + let refreshQueued = false; + let refreshTimer: number | undefined; + let refreshAll = false; + const pendingConversationIds = new Set(); const scheduleRefresh = (): void => { if (refreshTimer !== undefined || refreshInFlight) { - refreshQueued = true - return + refreshQueued = true; + return; } refreshTimer = window.setTimeout(() => { - refreshTimer = undefined - refreshQueued = false + refreshTimer = undefined; + refreshQueued = false; const conversationIds = refreshAll ? undefined - : [...pendingConversationIds] - refreshAll = false - pendingConversationIds.clear() - refresh(conversationIds) - }, 0) - } + : [...pendingConversationIds]; + refreshAll = false; + pendingConversationIds.clear(); + refresh(conversationIds); + }, 0); + }; const refresh = (conversationIds?: string[]): void => { - refreshInFlight = true - const affectedConversationIds = new Set( - conversationIds ?? [] - ) + refreshInFlight = true; + const affectedConversationIds = new Set(conversationIds ?? []); const reads = conversationIds ? conversationIds.map(async (conversationId) => ({ conversationId, items: - await window.goodbuddy.conversationQueue.list( - conversationId - ) + await window.goodbuddy.conversationQueue.list(conversationId), })) : [ - window.goodbuddy.conversationQueue - .list() - .then((items) => ({ - conversationId: undefined, - items - })) - ] + window.goodbuddy.conversationQueue.list().then((items) => ({ + conversationId: undefined, + items, + })), + ]; void Promise.all(reads) .then((results) => { if (!active) { - return + return; } setConversationQueueItems((current) => { const allItems = results.find( - (result) => result.conversationId === undefined - )?.items + (result) => result.conversationId === undefined, + )?.items; const next = allItems ? allItems : [ ...current.filter( - (item) => - !affectedConversationIds.has( - item.conversationId - ) + (item) => !affectedConversationIds.has(item.conversationId), ), - ...results.flatMap((result) => result.items) + ...results.flatMap((result) => result.items), ].sort( (left, right) => left.createdAt.localeCompare(right.createdAt) || - left.id.localeCompare(right.id) - ) - return sameConversationQueueItems(current, next) - ? current - : next - }) + left.id.localeCompare(right.id), + ); + return sameConversationQueueItems(current, next) ? current : next; + }); }) .catch(() => { if (active) { notify({ - tone: 'error', - message: tRef.current( - 'notices.conversationQueueReadFailed' - ), - dedupeKey: 'conversation-queue-read' - }) + tone: "error", + message: tRef.current("notices.conversationQueueReadFailed"), + dedupeKey: "conversation-queue-read", + }); } }) .finally(() => { - refreshInFlight = false + refreshInFlight = false; if (active && refreshQueued) { - refreshQueued = false - scheduleRefresh() + refreshQueued = false; + scheduleRefresh(); } - }) - } + }); + }; const queueRefresh = (conversationId?: string): void => { if (conversationId) { - pendingConversationIds.add(conversationId) + pendingConversationIds.add(conversationId); } else { - refreshAll = true + refreshAll = true; } - scheduleRefresh() - } - refresh() - const remove = - window.goodbuddy.conversationQueue.onChanged(queueRefresh) - const removeDispatch = - window.goodbuddy.conversationQueue.onDispatch((dispatch) => - conversationQueueDispatchRef.current(dispatch) - ) + scheduleRefresh(); + }; + refresh(); + const remove = window.goodbuddy.conversationQueue.onChanged(queueRefresh); + const removeDispatch = window.goodbuddy.conversationQueue.onDispatch( + (dispatch) => conversationQueueDispatchRef.current(dispatch), + ); return () => { - active = false + active = false; if (refreshTimer !== undefined) { - window.clearTimeout(refreshTimer) + window.clearTimeout(refreshTimer); } - remove() - removeDispatch() - } - }, []) + remove(); + removeDispatch(); + }; + }, []); useEffect(() => { - let active = true + let active = true; void (async () => { - let snapshot: ActivityHistorySnapshot + let snapshot: ActivityHistorySnapshot; try { - snapshot = await window.goodbuddy.activityHistory.get() + snapshot = await window.goodbuddy.activityHistory.get(); } catch { if (active) { notify({ - tone: 'error', - message: tRef.current('notices.activityHistoryReadFailed'), - dedupeKey: 'activity-history-read' - }) + tone: "error", + message: tRef.current("notices.activityHistoryReadFailed"), + dedupeKey: "activity-history-read", + }); } - return + return; } if (!active) { - return + return; } const legacyHistoryMayBeIncomplete = legacyActivityHistory.historyMayBeIncomplete || - snapshot.legacyHistoryMayBeIncomplete + snapshot.legacyHistoryMayBeIncomplete; if ( legacyActivityHistory.records.length > 0 || legacyActivityHistory.historyMayBeIncomplete @@ -5515,424 +5254,387 @@ function App(): React.JSX.Element { await window.goodbuddy.activityHistory.replace( mergeActivityRecords( legacyActivityHistory.records, - snapshot.records + snapshot.records, ), - legacyHistoryMayBeIncomplete - ) - clearLegacyActivityHistory() + legacyHistoryMayBeIncomplete, + ); + clearLegacyActivityHistory(); } catch { notify({ - tone: 'error', - message: tRef.current( - 'notices.activityHistoryPersistenceFailed' - ), - dedupeKey: 'activity-history-persistence' - }) + tone: "error", + message: tRef.current("notices.activityHistoryPersistenceFailed"), + dedupeKey: "activity-history-persistence", + }); } if (!active) { - return + return; } } setActivityRecords((current) => - mergeActivityRecords(current, snapshot.records) - ) - setLegacyActivityHistoryMayBeIncomplete(legacyHistoryMayBeIncomplete) - setActivityHistoryReady(true) - })() + mergeActivityRecords(current, snapshot.records), + ); + setLegacyActivityHistoryMayBeIncomplete(legacyHistoryMayBeIncomplete); + setActivityHistoryReady(true); + })(); return () => { - active = false - } - }, [legacyActivityHistory]) + active = false; + }; + }, [legacyActivityHistory]); useEffect(() => { - activityRecordsRef.current = activityRecords + activityRecordsRef.current = activityRecords; legacyActivityHistoryMayBeIncompleteRef.current = - legacyActivityHistoryMayBeIncomplete + legacyActivityHistoryMayBeIncomplete; if (!activityHistoryReady) { - return + return; } const timeout = window.setTimeout(() => { - void persistActivityHistory() - }, 250) - return () => window.clearTimeout(timeout) + void persistActivityHistory(); + }, 250); + return () => window.clearTimeout(timeout); }, [ activityHistoryReady, activityRecords, legacyActivityHistoryMayBeIncomplete, - persistActivityHistory - ]) + persistActivityHistory, + ]); useEffect( () => () => { - void persistActivityHistory() + void persistActivityHistory(); }, - [persistActivityHistory] - ) + [persistActivityHistory], + ); const resumeProjectConversationQueues = useCallback( (projectId: string): void => { if ( (!projectRecoverySnapshotReadyRef.current && isManagedSshProject( - projectsRef.current.find( - (project) => project.id === projectId - ) + projectsRef.current.find((project) => project.id === projectId), )) || isProjectRecoveryUnsettled( - projectRecoveryByProjectIdRef.current[projectId] + projectRecoveryByProjectIdRef.current[projectId], ) ) { - return + return; } const conversationIds = conversationsRef.current .filter( (conversation) => - conversation.projectId === projectId && - !conversation.remote + conversation.projectId === projectId && !conversation.remote, ) - .map((conversation) => conversation.id) + .map((conversation) => conversation.id); if (conversationIds.length === 0) { - return + return; } void Promise.all( conversationIds.map((conversationId) => - window.goodbuddy.conversationQueue.ready(conversationId) - ) + window.goodbuddy.conversationQueue.ready(conversationId), + ), ).catch(() => { notify({ - tone: 'error', - message: tRef.current( - 'notices.conversationQueueResumeFailed' - ), - dedupeKey: `conversation-queue-resume:${projectId}` - }) - }) + tone: "error", + message: tRef.current("notices.conversationQueueResumeFailed"), + dedupeKey: `conversation-queue-resume:${projectId}`, + }); + }); }, - [notify] - ) + [notify], + ); const applyProjectRecoveryState = useCallback( (state: RemoteProjectRecoveryState): void => { - const previous = - projectRecoveryByProjectIdRef.current[state.projectId] + const previous = projectRecoveryByProjectIdRef.current[state.projectId]; if (previous) { if (previous.requestId !== state.requestId) { if ( - !['completed', 'failed'].includes(previous.stage) || - state.stage !== 'network' + !["completed", "failed"].includes(previous.stage) || + state.stage !== "network" ) { - return + return; } } else if ( remoteRecoveryStageOrder(state.stage) < remoteRecoveryStageOrder(previous.stage) || - (previous.stage === 'cursor' && - state.stage === 'cursor' && + (previous.stage === "cursor" && + state.stage === "cursor" && BigInt(state.current) < BigInt(previous.current)) || - ['completed', 'failed'].includes(previous.stage) + ["completed", "failed"].includes(previous.stage) ) { - return + return; } } const next = { ...projectRecoveryByProjectIdRef.current, - [state.projectId]: state - } - projectRecoveryByProjectIdRef.current = next - setProjectRecoveryByProjectId(next) - if ( - state.stage === 'completed' && - previous?.stage !== 'completed' - ) { - resumeProjectConversationQueues(state.projectId) + [state.projectId]: state, + }; + projectRecoveryByProjectIdRef.current = next; + setProjectRecoveryByProjectId(next); + if (state.stage === "completed" && previous?.stage !== "completed") { + resumeProjectConversationQueues(state.projectId); } }, - [resumeProjectConversationQueues] - ) + [resumeProjectConversationQueues], + ); useEffect(() => { - const recoveryApi = window.goodbuddy.projects.remote - let active = true + const recoveryApi = window.goodbuddy.projects.remote; + let active = true; const removeListener = recoveryApi.onRecoveryProgress((state) => { if (active) { - applyProjectRecoveryState(state) + applyProjectRecoveryState(state); } - }) + }); void recoveryApi.getRecoverySnapshot().then( (snapshot) => { if (active) { - snapshot.recoveries.forEach(applyProjectRecoveryState) - projectRecoverySnapshotReadyRef.current = true - setProjectRecoverySnapshotReady(true) + snapshot.recoveries.forEach(applyProjectRecoveryState); + projectRecoverySnapshotReadyRef.current = true; + setProjectRecoverySnapshotReady(true); } }, () => { // Main reports actionable recovery failures as project states. if (active) { - projectRecoverySnapshotReadyRef.current = true - setProjectRecoverySnapshotReady(true) + projectRecoverySnapshotReadyRef.current = true; + setProjectRecoverySnapshotReady(true); } - } - ) + }, + ); return () => { - active = false - removeListener() - } - }, [applyProjectRecoveryState]) + active = false; + removeListener(); + }; + }, [applyProjectRecoveryState]); const retryProjectRecovery = useCallback( async (projectId: string): Promise => { if (retryingRecoveryProjectIdsRef.current.has(projectId)) { - return + return; } - retryingRecoveryProjectIdsRef.current.add(projectId) + retryingRecoveryProjectIdsRef.current.add(projectId); try { const state = - await window.goodbuddy.projects.remote.retryRecovery( - projectId - ) - applyProjectRecoveryState(state) + await window.goodbuddy.projects.remote.retryRecovery(projectId); + applyProjectRecoveryState(state); } catch { // Preserve the existing local failure without duplicating a toast. } finally { - retryingRecoveryProjectIdsRef.current.delete(projectId) + retryingRecoveryProjectIdsRef.current.delete(projectId); } }, - [applyProjectRecoveryState] - ) + [applyProjectRecoveryState], + ); useEffect(() => { - let active = true + let active = true; const initialization = Promise.all([ window.goodbuddy.projects.list(false), - window.goodbuddy.conversations.list() - ]) - .then(async ([value, persistedConversations]) => { - if (!active || value.length === 0) { - return - } - setProjects(value) - const project = value.find(isOrdinaryLocalProject) - if (!project) { - throw new Error('没有可用的本地项目') - } - setActiveProjectId(project.id) - setWorkMode( - normalizeInteractiveWorkMode(project.defaultWorkMode) - ) - const persistedLocalConversations = - persistedConversations.filter( - (conversation) => !conversation.remote - ) - const persistedConversationIds = new Set( - persistedConversations.map((conversation) => conversation.id) - ) - const shouldMigrateLocalStorage = - conversationMigrationStoragePresent.current || - persistedConversations.length === 0 - const migratedLocalConversations = - shouldMigrateLocalStorage - ? migrationConversations.current - .filter( - (conversation) => - !conversation.remote && - !persistedConversationIds.has(conversation.id) - ) - .map((conversation) => - conversation.projectId || project.kind === 'channel' - ? conversation - : { ...conversation, projectId: project.id } - ) - : [] - let nextConversations: Conversation[] = [ - ...persistedConversations, - ...migratedLocalConversations - ] - let projectConversation = nextConversations.find( + window.goodbuddy.conversations.list(), + ]).then(async ([value, persistedConversations]) => { + if (!active || value.length === 0) { + return; + } + setProjects(value); + const project = value.find(isOrdinaryLocalProject); + if (!project) { + throw new Error("没有可用的本地项目"); + } + setActiveProjectId(project.id); + setWorkMode(normalizeInteractiveWorkMode(project.defaultWorkMode)); + const persistedLocalConversations = persistedConversations.filter( + (conversation) => !conversation.remote, + ); + const persistedConversationIds = new Set( + persistedConversations.map((conversation) => conversation.id), + ); + const shouldMigrateLocalStorage = + conversationMigrationStoragePresent.current || + persistedConversations.length === 0; + const migratedLocalConversations = shouldMigrateLocalStorage + ? migrationConversations.current + .filter( + (conversation) => + !conversation.remote && + !persistedConversationIds.has(conversation.id), + ) + .map((conversation) => + conversation.projectId || project.kind === "channel" + ? conversation + : { ...conversation, projectId: project.id }, + ) + : []; + let nextConversations: Conversation[] = [ + ...persistedConversations, + ...migratedLocalConversations, + ]; + let projectConversation = nextConversations.find( + (conversation) => + conversation.projectId === project.id && + (project.kind !== "channel" || conversation.remote !== undefined), + ); + if (!projectConversation && project.kind !== "channel") { + projectConversation = createConversation( + project.id, + undefined, + tRef.current("conversation.greeting"), + ); + nextConversations = [projectConversation, ...nextConversations]; + } + const acknowledgedLocalConversations = new Map( + persistedLocalConversations.map((conversation) => [ + conversation.id, + conversation, + ]), + ); + if ( + nextConversations.some( (conversation) => - conversation.projectId === project.id && - (project.kind !== 'channel' || - conversation.remote !== undefined) + !conversation.remote && + acknowledgedLocalConversations.get(conversation.id) !== + conversation, ) - if (!projectConversation && project.kind !== 'channel') { - projectConversation = createConversation( - project.id, - undefined, - tRef.current('conversation.greeting') - ) - nextConversations = [ - projectConversation, - ...nextConversations - ] - } - const acknowledgedLocalConversations = new Map( - persistedLocalConversations.map((conversation) => [ - conversation.id, - conversation - ]) - ) - if ( - nextConversations.some( - (conversation) => - !conversation.remote && - acknowledgedLocalConversations.get(conversation.id) !== - conversation - ) - ) { - if (persistedConversations.length === 0) { - const migratedSnapshots = - toConversationSnapshots(nextConversations) - await window.goodbuddy.conversations.replace( - migratedSnapshots - ) - const migratedSnapshotIds = new Set( - migratedSnapshots.map((conversation) => conversation.id) - ) - for (const conversation of nextConversations) { - if (migratedSnapshotIds.has(conversation.id)) { - acknowledgedLocalConversations.set( - conversation.id, - conversation - ) - } - } - } - while (true) { - const migration = createLocalConversationSaveBatch( - nextConversations, - acknowledgedLocalConversations, - new Set() - ) - if (migration.batch.length === 0) { - break - } - await window.goodbuddy.conversations.saveLocal( - migration.batch - ) - for (const conversation of migration.acknowledgements) { - acknowledgedLocalConversations.set( - conversation.id, - conversation - ) + ) { + if (persistedConversations.length === 0) { + const migratedSnapshots = toConversationSnapshots(nextConversations); + await window.goodbuddy.conversations.replace(migratedSnapshots); + const migratedSnapshotIds = new Set( + migratedSnapshots.map((conversation) => conversation.id), + ); + for (const conversation of nextConversations) { + if (migratedSnapshotIds.has(conversation.id)) { + acknowledgedLocalConversations.set(conversation.id, conversation); } } } - if (!active) { - return + while (true) { + const migration = createLocalConversationSaveBatch( + nextConversations, + acknowledgedLocalConversations, + new Set(), + ); + if (migration.batch.length === 0) { + break; + } + await window.goodbuddy.conversations.saveLocal(migration.batch); + for (const conversation of migration.acknowledgements) { + acknowledgedLocalConversations.set(conversation.id, conversation); + } } - persistedLocalConversationsRef.current = - acknowledgedLocalConversations - setConversations(nextConversations) - setActiveId(projectConversation?.id ?? '') - try { - localStorage.removeItem(storageKey) - conversationMigrationStoragePresent.current = false - } catch { - // The SQLite migration has already completed successfully. - } - setConversationStoreReady(true) - }) + } + if (!active) { + return; + } + persistedLocalConversationsRef.current = acknowledgedLocalConversations; + setConversations(nextConversations); + setActiveId(projectConversation?.id ?? ""); + try { + localStorage.removeItem(storageKey); + conversationMigrationStoragePresent.current = false; + } catch { + // The SQLite migration has already completed successfully. + } + setConversationStoreReady(true); + }); conversationPersistenceQueueRef.current = initialization.then( () => undefined, - () => undefined - ) + () => undefined, + ); void initialization.catch((reason: unknown) => { - if (active) { - setConversationLoadError( - displayErrorMessage( - reason, - tRef.current('notices.projectReadFailed') - ) - ) - } - }) + if (active) { + setConversationLoadError( + displayErrorMessage( + reason, + tRef.current("notices.projectReadFailed"), + ), + ); + } + }); return () => { - active = false - } - }, [conversationLoadRetry, setActiveId]) + active = false; + }; + }, [conversationLoadRetry, setActiveId]); useEffect(() => { if (!activeProjectId) { - return + return; } void window.goodbuddy.memory .list(activeProjectId) .then(setAssistantMemories) .catch(() => notify({ - tone: 'error', - message: tRef.current('notices.memoryReadFailed') - }) - ) - }, [activeProjectId]) + tone: "error", + message: tRef.current("notices.memoryReadFailed"), + }), + ); + }, [activeProjectId]); const refreshWorkspaceChanges = useCallback(async (): Promise => { if (!activeProjectId) { - workspaceChangesRequestRef.current += 1 - setWorkspaceChanges(undefined) - return + workspaceChangesRequestRef.current += 1; + setWorkspaceChanges(undefined); + return; } - await loadWorkspaceChanges(activeProjectId) - }, [activeProjectId, loadWorkspaceChanges]) + await loadWorkspaceChanges(activeProjectId); + }, [activeProjectId, loadWorkspaceChanges]); const listWorkspaceDirectory = useCallback( async (path: string) => { if (!activeProjectId) { - throw new Error(tRef.current('notices.selectProject')) + throw new Error(tRef.current("notices.selectProject")); } - return window.goodbuddy.workspace.listDirectory(activeProjectId, path) + return window.goodbuddy.workspace.listDirectory(activeProjectId, path); }, - [activeProjectId] - ) + [activeProjectId], + ); const loadWorkspaceFile = useCallback( async (path: string, offsetBytes = 0) => { if (!activeProjectId) { - throw new Error(tRef.current('notices.selectProject')) + throw new Error(tRef.current("notices.selectProject")); } return window.goodbuddy.workspace.readFile( activeProjectId, path, - offsetBytes - ) + offsetBytes, + ); }, - [activeProjectId] - ) - const loadWorkspaceDiff = useCallback(async (path: string) => { - if (!activeProjectId) throw new Error(tRef.current('notices.selectProject')) - return window.goodbuddy.workspace.getFileDiff(activeProjectId, path) - }, [activeProjectId]) + [activeProjectId], + ); + const loadWorkspaceDiff = useCallback( + async (path: string) => { + if (!activeProjectId) + throw new Error(tRef.current("notices.selectProject")); + return window.goodbuddy.workspace.getFileDiff(activeProjectId, path); + }, + [activeProjectId], + ); const openWorkspaceEntry = useCallback( - async ( - path: string, - type: 'file' | 'directory' - ): Promise => { + async (path: string, type: "file" | "directory"): Promise => { if (!activeProjectId) { - throw new Error(tRef.current('notices.selectProject')) + throw new Error(tRef.current("notices.selectProject")); } - await window.goodbuddy.workspace.openPath( - activeProjectId, - path, - type - ) + await window.goodbuddy.workspace.openPath(activeProjectId, path, type); }, - [activeProjectId] - ) + [activeProjectId], + ); useEffect(() => { - if (assistantSidebarTab !== 'workspace') { - return + if (assistantSidebarTab !== "workspace") { + return; } const timeout = setTimeout(() => { void refreshWorkspaceChanges().catch(() => { notify({ - tone: 'error', - message: tRef.current('notices.workspaceChangesReadFailed') - }) - }) - }, 0) - return () => clearTimeout(timeout) - }, [assistantSidebarTab, refreshWorkspaceChanges]) + tone: "error", + message: tRef.current("notices.workspaceChangesReadFailed"), + }); + }); + }, 0); + return () => clearTimeout(timeout); + }, [assistantSidebarTab, refreshWorkspaceChanges]); useEffect(() => { void window.goodbuddy.experts @@ -5940,296 +5642,287 @@ function App(): React.JSX.Element { .then(setAssistantExperts) .catch(() => notify({ - tone: 'error', - message: tRef.current('notices.expertsReadFailed') - }) - ) - }, []) + tone: "error", + message: tRef.current("notices.expertsReadFailed"), + }), + ); + }, []); useEffect(() => { if (!activeProjectId) { - return + return; } void window.goodbuddy.schedules .list() .then(setAssistantSchedules) .catch(() => notify({ - tone: 'error', - message: tRef.current('notices.schedulesReadFailed') - }) - ) - }, [activeProjectId]) + tone: "error", + message: tRef.current("notices.schedulesReadFailed"), + }), + ); + }, [activeProjectId]); const loadHeartbeats = useCallback(async () => { const [configs, memories] = await Promise.all([ window.goodbuddy.heartbeats.list(), - window.goodbuddy.memory.list() - ]) - const history = await window.goodbuddy.heartbeats.history() + window.goodbuddy.memory.list(), + ]); + const history = await window.goodbuddy.heartbeats.history(); return { configs, memories, runs: history.runs, - entries: history.entries - } - }, []) + entries: history.entries, + }; + }, []); const refreshHeartbeats = useCallback(async (): Promise => { - const requestId = ++heartbeatLoadRequestRef.current - const result = await loadHeartbeats() + const requestId = ++heartbeatLoadRequestRef.current; + const result = await loadHeartbeats(); if (requestId !== heartbeatLoadRequestRef.current) { - return + return; } - setAssistantHeartbeats(result.configs) - setHeartbeatMemories(result.memories) - setHeartbeatRuns(result.runs) - setHeartbeatEntries(result.entries) - }, [loadHeartbeats]) + setAssistantHeartbeats(result.configs); + setHeartbeatMemories(result.memories); + setHeartbeatRuns(result.runs); + setHeartbeatEntries(result.entries); + }, [loadHeartbeats]); useEffect(() => { if (projects.length === 0) { - return + return; } - const requestId = ++heartbeatLoadRequestRef.current + const requestId = ++heartbeatLoadRequestRef.current; const timeout = setTimeout(() => { if (requestId !== heartbeatLoadRequestRef.current) { - return + return; } - setHeartbeatLoading(true) - setHeartbeatLoadError(undefined) - setAssistantHeartbeats([]) - setHeartbeatRuns([]) - setHeartbeatEntries([]) + setHeartbeatLoading(true); + setHeartbeatLoadError(undefined); + setAssistantHeartbeats([]); + setHeartbeatRuns([]); + setHeartbeatEntries([]); void loadHeartbeats() .then((result) => { if (requestId !== heartbeatLoadRequestRef.current) { - return + return; } - setAssistantHeartbeats(result.configs) - setHeartbeatMemories(result.memories) - setHeartbeatRuns(result.runs) - setHeartbeatEntries(result.entries) - setHeartbeatLoadError(undefined) + setAssistantHeartbeats(result.configs); + setHeartbeatMemories(result.memories); + setHeartbeatRuns(result.runs); + setHeartbeatEntries(result.entries); + setHeartbeatLoadError(undefined); }) .catch((reason: unknown) => { if (requestId !== heartbeatLoadRequestRef.current) { - return + return; } setHeartbeatLoadError( displayErrorMessage( reason, - tRef.current('notices.heartbeatReadFailed') - ) - ) + tRef.current("notices.heartbeatReadFailed"), + ), + ); }) .finally(() => { if (requestId === heartbeatLoadRequestRef.current) { - setHeartbeatLoading(false) + setHeartbeatLoading(false); } - }) - }, 0) + }); + }, 0); return () => { - clearTimeout(timeout) + clearTimeout(timeout); if (requestId === heartbeatLoadRequestRef.current) { - heartbeatLoadRequestRef.current += 1 + heartbeatLoadRequestRef.current += 1; } - } - }, [loadHeartbeats, projects.length]) + }; + }, [loadHeartbeats, projects.length]); const refreshHeartbeatCenter = useCallback(async (): Promise => { const [artifacts] = await Promise.all([ window.goodbuddy.artifacts.list(), - refreshHeartbeats() - ]) - setAssistantArtifacts((current) => - mergeArtifacts(current, artifacts) - ) - }, [refreshHeartbeats]) + refreshHeartbeats(), + ]); + setAssistantArtifacts((current) => mergeArtifacts(current, artifacts)); + }, [refreshHeartbeats]); const retryHeartbeatLoad = useCallback(async (): Promise => { - setHeartbeatLoading(true) - setHeartbeatLoadError(undefined) + setHeartbeatLoading(true); + setHeartbeatLoadError(undefined); try { - await refreshHeartbeatCenter() - setHeartbeatLoadError(undefined) + await refreshHeartbeatCenter(); + setHeartbeatLoadError(undefined); } catch (reason) { setHeartbeatLoadError( displayErrorMessage( reason, - tRef.current('notices.heartbeatReadFailed') - ) - ) + tRef.current("notices.heartbeatReadFailed"), + ), + ); } finally { - setHeartbeatLoading(false) + setHeartbeatLoading(false); } - }, [refreshHeartbeatCenter]) + }, [refreshHeartbeatCenter]); const createHeartbeat = useCallback( async (input: HeartbeatCreateInput): Promise => { - await window.goodbuddy.heartbeats.create(input) - await refreshHeartbeats() + await window.goodbuddy.heartbeats.create(input); + await refreshHeartbeats(); }, - [refreshHeartbeats] - ) + [refreshHeartbeats], + ); const updateHeartbeat = useCallback( - async ( - heartbeatId: string, - input: HeartbeatUpdateInput - ): Promise => { - await window.goodbuddy.heartbeats.update(heartbeatId, input) - await refreshHeartbeats() + async (heartbeatId: string, input: HeartbeatUpdateInput): Promise => { + await window.goodbuddy.heartbeats.update(heartbeatId, input); + await refreshHeartbeats(); }, - [refreshHeartbeats] - ) + [refreshHeartbeats], + ); const removeHeartbeat = useCallback( async (heartbeatId: string): Promise => { - await window.goodbuddy.heartbeats.remove(heartbeatId) - await refreshHeartbeats() + await window.goodbuddy.heartbeats.remove(heartbeatId); + await refreshHeartbeats(); }, - [refreshHeartbeats] - ) + [refreshHeartbeats], + ); const runHeartbeat = useCallback( async (heartbeatId: string): Promise => { - await window.goodbuddy.heartbeats.runNow(heartbeatId) - await refreshHeartbeatCenter() + await window.goodbuddy.heartbeats.runNow(heartbeatId); + await refreshHeartbeatCenter(); }, - [refreshHeartbeatCenter] - ) + [refreshHeartbeatCenter], + ); const setHeartbeatPaused = useCallback( async (heartbeatId: string, paused: boolean): Promise => { - await window.goodbuddy.heartbeats.setPaused(heartbeatId, paused) - await refreshHeartbeats() + await window.goodbuddy.heartbeats.setPaused(heartbeatId, paused); + await refreshHeartbeats(); }, - [refreshHeartbeats] - ) + [refreshHeartbeats], + ); useEffect(() => { - if (view !== 'heartbeat') { - return + if (view !== "heartbeat") { + return; } - let refreshing = false + let refreshing = false; const refresh = (): void => { if (refreshing) { - return + return; } - refreshing = true + refreshing = true; void refreshHeartbeatCenter() .then(() => setHeartbeatLoadError(undefined)) .catch((reason: unknown) => setHeartbeatLoadError( displayErrorMessage( reason, - tRef.current('notices.heartbeatRefreshFailed') - ) - ) + tRef.current("notices.heartbeatRefreshFailed"), + ), + ), ) .finally(() => { - refreshing = false - }) - } - const timeout = setTimeout(refresh, 0) - const interval = setInterval(refresh, 30_000) + refreshing = false; + }); + }; + const timeout = setTimeout(refresh, 0); + const interval = setInterval(refresh, 30_000); return () => { - clearTimeout(timeout) - clearInterval(interval) - } - }, [refreshHeartbeatCenter, view]) + clearTimeout(timeout); + clearInterval(interval); + }; + }, [refreshHeartbeatCenter, view]); useEffect(() => { void window.goodbuddy.tasks .list() .then((tasks) => { - setAssistantTasks(tasks) + setAssistantTasks(tasks); setActivityRecords((current) => reconcileActivityRecords( current, tasks, - new Set(activeRuns.current.keys()) - ) - ) + new Set(activeRuns.current.keys()), + ), + ); }) .catch(() => notify({ - tone: 'error', - message: tRef.current('notices.taskHistoryReadFailed') - }) - ) - }, []) + tone: "error", + message: tRef.current("notices.taskHistoryReadFailed"), + }), + ); + }, []); useEffect(() => { - if (view !== 'activity') { - return + if (view !== "activity") { + return; } const timeout = setTimeout(() => { void refreshTokenUsage().catch(() => notify({ - tone: 'error', - message: tRef.current('notices.tokenUsageReadFailed') - }) - ) - }, 0) - return () => clearTimeout(timeout) - }, [refreshTokenUsage, view]) + tone: "error", + message: tRef.current("notices.tokenUsageReadFailed"), + }), + ); + }, 0); + return () => clearTimeout(timeout); + }, [refreshTokenUsage, view]); useEffect(() => { void window.goodbuddy.artifacts .list() .then((artifacts) => - setAssistantArtifacts((current) => - mergeArtifacts(current, artifacts) - ) + setAssistantArtifacts((current) => mergeArtifacts(current, artifacts)), ) .catch(() => notify({ - tone: 'error', - message: tRef.current('notices.resultHistoryReadFailed') - }) - ) - }, []) + tone: "error", + message: tRef.current("notices.resultHistoryReadFailed"), + }), + ); + }, []); useEffect(() => { const missingIds = [ ...new Set( (activeConversation?.messages ?? []).flatMap( - (message) => message.artifactIds ?? [] - ) - ) + (message) => message.artifactIds ?? [], + ), + ), ] .filter( (artifactId) => !assistantArtifactById.get(artifactId)?.content && - !hydratingArtifactIds.current.has(artifactId) + !hydratingArtifactIds.current.has(artifactId), ) - .slice(-32) + .slice(-32); if (missingIds.length === 0) { - return + return; } for (const artifactId of missingIds) { - hydratingArtifactIds.current.add(artifactId) + hydratingArtifactIds.current.add(artifactId); } void Promise.allSettled( missingIds.map((artifactId) => - window.goodbuddy.artifacts.get(artifactId) - ) + window.goodbuddy.artifacts.get(artifactId), + ), ).then((results) => { const artifacts = results.flatMap((result) => - result.status === 'fulfilled' ? [result.value] : [] - ) + result.status === "fulfilled" ? [result.value] : [], + ); if (artifacts.length > 0) { - setAssistantArtifacts((current) => - mergeArtifacts(current, artifacts) - ) + setAssistantArtifacts((current) => mergeArtifacts(current, artifacts)); } for (const artifactId of missingIds) { - hydratingArtifactIds.current.delete(artifactId) + hydratingArtifactIds.current.delete(artifactId); } - }) - }, [activeConversation, assistantArtifactById]) + }); + }, [activeConversation, assistantArtifactById]); useEffect(() => { const timeout = setTimeout(() => { @@ -6237,624 +5930,566 @@ function App(): React.JSX.Element { .catch(() => { // refreshKnowledge exposes a recoverable page-local error. }) - .finally(() => setKnowledgeLoading(false)) - }, 0) - return () => clearTimeout(timeout) - }, [refreshKnowledge]) + .finally(() => setKnowledgeLoading(false)); + }, 0); + return () => clearTimeout(timeout); + }, [refreshKnowledge]); useEffect(() => { if ( knowledgeLoadError || - (view !== 'knowledge' && knowledgeOperationCount === 0) + (view !== "knowledge" && knowledgeOperationCount === 0) ) { - return + return; } - const interval = setInterval(() => { - void refreshKnowledge( - knowledgeSnapshot.selectedLibraryId - ).catch(() => { - // The task center keeps the last successful snapshot while polling. - }) - }, knowledgeOperationCount > 0 ? 350 : 1_000) - return () => clearInterval(interval) + const interval = setInterval( + () => { + void refreshKnowledge(knowledgeSnapshot.selectedLibraryId).catch(() => { + // The task center keeps the last successful snapshot while polling. + }); + }, + knowledgeOperationCount > 0 ? 350 : 1_000, + ); + return () => clearInterval(interval); }, [ knowledgeLoadError, knowledgeOperationCount, knowledgeSnapshot.selectedLibraryId, refreshKnowledge, - view - ]) + view, + ]); useEffect(() => { void Promise.all([ window.goodbuddy.settings.getRuntime(), - window.goodbuddy.agent.getStatus() + window.goodbuddy.agent.getStatus(), ]) .then(([settings, status]) => { const selectionKey = agentRuntimeSelectionKey( - getDefaultRuntimeSelection(settings) - ) + getDefaultRuntimeSelection(settings), + ); runtimeStatusCacheRef.current = { key: selectionKey, - settings - } - setRuntimeSettings(settings) - setRuntime(status) - setRuntimeStatusKey(selectionKey) + settings, + }; + setRuntimeSettings(settings); + setRuntime(status); + setRuntimeStatusKey(selectionKey); if (!status.available && !runtimeSetupPromptedRef.current) { - runtimeSetupPromptedRef.current = true - setView('settings') + runtimeSetupPromptedRef.current = true; + setView("settings"); } }) .catch(() => notify({ - tone: 'error', - message: tRef.current('runtime.errors.readSettings') - }) - ) + tone: "error", + message: tRef.current("runtime.errors.readSettings"), + }), + ); void window.goodbuddy.app .getInfo() .then(setAppInfo) .catch(() => notify({ - tone: 'error', - message: tRef.current('notices.appInfoReadFailed') - }) - ) + tone: "error", + message: tRef.current("notices.appInfoReadFailed"), + }), + ); const removeAgentListener = - window.goodbuddy.agent.onEvent(handleAgentEvent) - const removeOpenSettingsListener = - window.goodbuddy.app.onOpenSettings(() => setView('settings')) + window.goodbuddy.agent.onEvent(handleAgentEvent); + const removeOpenSettingsListener = window.goodbuddy.app.onOpenSettings(() => + setView("settings"), + ); return () => { - removeAgentListener() - removeOpenSettingsListener() - } - }, [handleAgentEvent, setView]) + removeAgentListener(); + removeOpenSettingsListener(); + }; + }, [handleAgentEvent, setView]); - useEffect( - () => { - const browserApi = window.goodbuddy.browser - if (!browserApi) { - return + useEffect(() => { + const browserApi = window.goodbuddy.browser; + if (!browserApi) { + return; + } + return browserApi.onState((state) => { + setBrowserStates((current) => { + const previous = current[state.conversationId]; + return { + ...current, + [state.conversationId]: + state.status === "stopped" || + state.frameDataUrl || + !previous?.frameDataUrl + ? state + : { + ...state, + frameDataUrl: previous.frameDataUrl, + }, + }; + }); + if ( + state.status !== "stopped" && + state.conversationId === conversationNavigationRef.current.activeId + ) { + setAssistantSidebarOpen(true); + setAssistantSidebarTab("browser"); } - return browserApi.onState((state) => { - setBrowserStates((current) => { - const previous = current[state.conversationId] - return { - ...current, - [state.conversationId]: - state.status === 'stopped' || - state.frameDataUrl || - !previous?.frameDataUrl - ? state - : { - ...state, - frameDataUrl: previous.frameDataUrl - } - } - }) - if ( - state.status !== 'stopped' && - state.conversationId === - conversationNavigationRef.current.activeId - ) { - setAssistantSidebarOpen(true) - setAssistantSidebarTab('browser') - } - }) - }, - [] - ) + }); + }, []); useEffect( () => window.goodbuddy.app.onNewConversation(() => { - startNewConversation(activeProjectIdRef.current || undefined) + startNewConversation(activeProjectIdRef.current || undefined); }), - [startNewConversation] - ) + [startNewConversation], + ); useEffect(() => { - const handleNewConversationShortcut = ( - event: KeyboardEvent - ): void => { + const handleNewConversationShortcut = (event: KeyboardEvent): void => { if ( - event.key.toLocaleLowerCase() !== 'n' || + event.key.toLocaleLowerCase() !== "n" || (!event.ctrlKey && !event.metaKey) || event.altKey || event.shiftKey ) { - return + return; } - event.preventDefault() - startNewConversation(activeProjectIdRef.current || undefined) - } - document.addEventListener( - 'keydown', - handleNewConversationShortcut - ) + event.preventDefault(); + startNewConversation(activeProjectIdRef.current || undefined); + }; + document.addEventListener("keydown", handleNewConversationShortcut); return () => - document.removeEventListener( - 'keydown', - handleNewConversationShortcut - ) - }, [startNewConversation]) + document.removeEventListener("keydown", handleNewConversationShortcut); + }, [startNewConversation]); const commitProjectSelection = ( selected: AssistantProject, - candidateConversations = conversationsRef.current + candidateConversations = conversationsRef.current, ): void => { - setActiveProjectId(selected.id) - setWorkMode( - normalizeInteractiveWorkMode(selected.defaultWorkMode) - ) + setActiveProjectId(selected.id); + setWorkMode(normalizeInteractiveWorkMode(selected.defaultWorkMode)); const conversation = candidateConversations.find( (candidate) => candidate.projectId === selected.id && - ( - selected.kind !== 'channel' || - candidate.remote !== undefined - ) - ) + (selected.kind !== "channel" || candidate.remote !== undefined), + ); if (conversation) { if (isManagedSshProject(selected)) { const runtimeSelection = runtimeSettings - ? getRuntimeSelectionForProvider( - 'opencode', - runtimeSettings - ) - : ({ provider: 'opencode' } as const) + ? getRuntimeSelectionForProvider("opencode", runtimeSettings) + : ({ provider: "opencode" } as const); setConversations((current) => current.map((candidate) => candidate.projectId === selected.id && - candidate.runtimeSelection?.provider !== 'opencode' + candidate.runtimeSelection?.provider !== "opencode" ? { ...candidate, runtimeSelection, - updatedAt: Date.now() + updatedAt: Date.now(), } - : candidate - ) - ) + : candidate, + ), + ); } - setActiveId(conversation.id) - } else if (selected.kind === 'channel') { - setActiveId('') + setActiveId(conversation.id); + } else if (selected.kind === "channel") { + setActiveId(""); } else { const created = createConversation( selected.id, runtimeSettings - ? getProjectDefaultRuntimeSelection( - selected, - runtimeSettings - ) + ? getProjectDefaultRuntimeSelection(selected, runtimeSettings) : undefined, - t('conversation.greeting') - ) - setConversations((current) => [created, ...current]) - setActiveId(created.id) + t("conversation.greeting"), + ); + setConversations((current) => [created, ...current]); + setActiveId(created.id); } - setView('chat') - } + setView("chat"); + }; const selectProject = (projectId: string): void => { - const project = projects.find((candidate) => candidate.id === projectId) + const project = projects.find((candidate) => candidate.id === projectId); if (!project) { - return + return; } - if ( - project.executionSpace.kind === 'ssh' && - !remoteProjectsEnabled - ) { - return + if (project.executionSpace.kind === "ssh" && !remoteProjectsEnabled) { + return; } - commitProjectSelection(project) + commitProjectSelection(project); if ( !isProjectRecoveryUnsettled( - projectRecoveryByProjectIdRef.current[project.id] + projectRecoveryByProjectIdRef.current[project.id], ) ) { - resumeProjectConversationQueues(project.id) + resumeProjectConversationQueues(project.id); } - } + }; const createProject = async ( - input: ProjectCreateInput + input: ProjectCreateInput, ): Promise => { - const project = await window.goodbuddy.projects.create(input) - setProjects((current) => [project, ...current]) - setActiveProjectId(project.id) - setWorkMode(normalizeInteractiveWorkMode(project.defaultWorkMode)) + const project = await window.goodbuddy.projects.create(input); + setProjects((current) => [project, ...current]); + setActiveProjectId(project.id); + setWorkMode(normalizeInteractiveWorkMode(project.defaultWorkMode)); const conversation = createConversation( project.id, runtimeSettings ? getProjectDefaultRuntimeSelection(project, runtimeSettings) : undefined, - t('conversation.greeting') - ) - setConversations((current) => [conversation, ...current]) - setActiveId(conversation.id) - setView('chat') - return project - } + t("conversation.greeting"), + ); + setConversations((current) => [conversation, ...current]); + setActiveId(conversation.id); + setView("chat"); + return project; + }; const updateProject = async ( projectId: string, - input: ProjectCreateInput + input: ProjectCreateInput, ): Promise => { - const project = await window.goodbuddy.projects.update( - projectId, - input - ) + const project = await window.goodbuddy.projects.update(projectId, input); setProjects((current) => current.map((candidate) => - candidate.id === project.id ? project : candidate - ) - ) + candidate.id === project.id ? project : candidate, + ), + ); if (project.id === activeProjectId) { - setWorkMode( - normalizeInteractiveWorkMode(project.defaultWorkMode) - ) + setWorkMode(normalizeInteractiveWorkMode(project.defaultWorkMode)); } - return project - } + return project; + }; const loadCommittedRemoteProject = async ( - project: AssistantProject + project: AssistantProject, ): Promise => { - if ( - !remoteProjectsEnabled || - project.executionSpace.kind !== 'ssh' - ) { - return + if (!remoteProjectsEnabled || project.executionSpace.kind !== "ssh") { + return; } setProjects((current) => current.some((candidate) => candidate.id === project.id) ? current.map((candidate) => - candidate.id === project.id ? project : candidate + candidate.id === project.id ? project : candidate, ) - : [project, ...current] - ) - commitProjectSelection(project) + : [project, ...current], + ); + commitProjectSelection(project); if ( !isProjectRecoveryUnsettled( - projectRecoveryByProjectIdRef.current[project.id] + projectRecoveryByProjectIdRef.current[project.id], ) ) { - resumeProjectConversationQueues(project.id) + resumeProjectConversationQueues(project.id); } - } + }; - const handleRemoteProjectsEnabledChange = ( - enabled: boolean - ): void => { - setRemoteProjectsEnabled(enabled) + const handleRemoteProjectsEnabledChange = (enabled: boolean): void => { + setRemoteProjectsEnabled(enabled); if (enabled) { - return + return; } - if (activeProject?.executionSpace.kind !== 'ssh') { - return + if (activeProject?.executionSpace.kind !== "ssh") { + return; } - const localProject = projects.find(isOrdinaryLocalProject) + const localProject = projects.find(isOrdinaryLocalProject); if (localProject) { - commitProjectSelection(localProject) + commitProjectSelection(localProject); } - } + }; const archiveProject = async (projectId: string): Promise => { - await window.goodbuddy.projects.setArchived(projectId, true) - const remaining = projects.filter((project) => project.id !== projectId) - setProjects(remaining) + await window.goodbuddy.projects.setArchived(projectId, true); + const remaining = projects.filter((project) => project.id !== projectId); + setProjects(remaining); if (projectId === activeProjectIdRef.current) { const next = remaining.find(isOrdinaryLocalProject) ?? - (remoteProjectsEnabled ? remaining[0] : undefined) + (remoteProjectsEnabled ? remaining[0] : undefined); if (next) { - selectProject(next.id) + selectProject(next.id); } } - } + }; - const removeProjectsFromUi = ( - projectIds: readonly string[] - ): void => { + const removeProjectsFromUi = (projectIds: readonly string[]): void => { if (projectIds.length === 0) { - return + return; } - const deletedProjectIds = new Set(projectIds) - const referencesDeletedProject = ( - projectId: string | undefined - ): boolean => - projectId !== undefined && deletedProjectIds.has(projectId) + const deletedProjectIds = new Set(projectIds); + const referencesDeletedProject = (projectId: string | undefined): boolean => + projectId !== undefined && deletedProjectIds.has(projectId); const remainingProjects = projectsRef.current.filter( - (project) => !deletedProjectIds.has(project.id) - ) - const currentConversations = conversationsRef.current + (project) => !deletedProjectIds.has(project.id), + ); + const currentConversations = conversationsRef.current; const deletedConversationIds = new Set( currentConversations .filter((conversation) => - referencesDeletedProject(conversation.projectId) + referencesDeletedProject(conversation.projectId), ) - .map((conversation) => conversation.id) - ) + .map((conversation) => conversation.id), + ); const remainingConversations = currentConversations.filter( - (conversation) => - !referencesDeletedProject(conversation.projectId) - ) + (conversation) => !referencesDeletedProject(conversation.projectId), + ); setProjects((current) => - current.filter( - (project) => !deletedProjectIds.has(project.id) - ) - ) + current.filter((project) => !deletedProjectIds.has(project.id)), + ); setConversations((current) => current.filter( - (conversation) => - !referencesDeletedProject(conversation.projectId) - ) - ) + (conversation) => !referencesDeletedProject(conversation.projectId), + ), + ); setAssistantTasks((current) => - current.filter( - (task) => !referencesDeletedProject(task.projectId) - ) - ) + current.filter((task) => !referencesDeletedProject(task.projectId)), + ); setAssistantArtifacts((current) => current.filter( - (artifact) => - !referencesDeletedProject(artifact.projectId) - ) - ) + (artifact) => !referencesDeletedProject(artifact.projectId), + ), + ); setAssistantMemories((current) => current.filter( (memory) => !( - memory.scope === 'project' && + memory.scope === "project" && referencesDeletedProject(memory.scopeId) ) && !( - memory.scope === 'conversation' && + memory.scope === "conversation" && memory.scopeId !== undefined && deletedConversationIds.has(memory.scopeId) - ) - ) - ) + ), + ), + ); setAssistantSchedules((current) => current.filter( - (schedule) => - !referencesDeletedProject(schedule.projectId) - ) - ) + (schedule) => !referencesDeletedProject(schedule.projectId), + ), + ); setAssistantHeartbeats((current) => current.flatMap((heartbeat) => { - if (heartbeat.scope.kind === 'global') { - return heartbeat + if (heartbeat.scope.kind === "global") { + return heartbeat; } const projectIds = heartbeat.scope.projectIds.filter( - (id) => !deletedProjectIds.has(id) - ) + (id) => !deletedProjectIds.has(id), + ); return projectIds.length > 0 - ? [{ ...heartbeat, scope: { kind: 'projects', projectIds } }] - : [] - }) - ) + ? [{ ...heartbeat, scope: { kind: "projects", projectIds } }] + : []; + }), + ); if (!deletedProjectIds.has(activeProjectIdRef.current)) { - return + return; } - const next = remainingProjects.find(isOrdinaryLocalProject) + const next = remainingProjects.find(isOrdinaryLocalProject); if (next) { - commitProjectSelection(next, remainingConversations) - return + commitProjectSelection(next, remainingConversations); + return; } - setActiveProjectId('') - setActiveId('') - setView('chat') - } + setActiveProjectId(""); + setActiveId(""); + setView("chat"); + }; const deleteProject = async ( projectId: string, - confirmation: string + confirmation: string, ): Promise => { - await window.goodbuddy.projects.delete(projectId, confirmation) - removeProjectsFromUi([projectId]) - } + await window.goodbuddy.projects.delete(projectId, confirmation); + removeProjectsFromUi([projectId]); + }; const newConversation = (): boolean => { - return startNewConversation(activeProjectId || undefined) - } + return startNewConversation(activeProjectId || undefined); + }; const setMemoryStatus = async ( memoryId: string, - status: AssistantMemory['status'] + status: AssistantMemory["status"], ): Promise => { - await window.goodbuddy.memory.setStatus(memoryId, status) + await window.goodbuddy.memory.setStatus(memoryId, status); setAssistantMemories((current) => - status === 'rejected' + status === "rejected" ? current.filter((memory) => memory.id !== memoryId) : current.map((memory) => - memory.id === memoryId ? { ...memory, status } : memory - ) - ) + memory.id === memoryId ? { ...memory, status } : memory, + ), + ); setHeartbeatMemories((current) => - status === 'rejected' + status === "rejected" ? current.filter((memory) => memory.id !== memoryId) : current.map((memory) => - memory.id === memoryId ? { ...memory, status } : memory - ) - ) - } + memory.id === memoryId ? { ...memory, status } : memory, + ), + ); + }; const useHeartbeatTask = (task: AssistantTask): void => { if (task.projectId && task.projectId !== activeProjectId) { - setActiveProjectId(task.projectId) + setActiveProjectId(task.projectId); } if ( - !startNewConversation( - task.projectId ?? (activeProjectId || undefined) - ) + !startNewConversation(task.projectId ?? (activeProjectId || undefined)) ) { - return + return; } - setWorkMode('ask') + setWorkMode("ask"); setInput( - [ - t('notices.heartbeatTaskPrompt'), - task.title, - task.instructions - ].join('\n\n') - ) + [t("notices.heartbeatTaskPrompt"), task.title, task.instructions].join( + "\n\n", + ), + ); notify({ - tone: 'info', - message: t('notices.heartbeatTaskAdded', { title: task.title }) - }) - requestAnimationFrame(() => inputRef.current?.focus()) - } + tone: "info", + message: t("notices.heartbeatTaskAdded", { title: task.title }), + }); + requestAnimationFrame(() => inputRef.current?.focus()); + }; const setHeartbeatTaskStatus = async ( taskId: string, - status: 'completed' | 'cancelled' + status: "completed" | "cancelled", ): Promise => { - await window.goodbuddy.tasks.setStatus(taskId, status) + await window.goodbuddy.tasks.setStatus(taskId, status); setAssistantTasks((current) => current.map((task) => task.id === taskId ? { ...task, status, - completedAt: new Date().toISOString() + completedAt: new Date().toISOString(), } - : task - ) - ) - } + : task, + ), + ); + }; - const deleteConversation = async ( - conversationId: string - ): Promise => { + const deleteConversation = async (conversationId: string): Promise => { if (deletingConversationId) { - return + return; } - setDeletingConversationId(conversationId) + setDeletingConversationId(conversationId); const activeRequests = [...activeRuns.current.entries()] .filter(([, run]) => run.conversationId === conversationId) - .map(([requestId]) => requestId) + .map(([requestId]) => requestId); try { await Promise.all( activeRequests.map((requestId) => - window.goodbuddy.agent.cancel(requestId) - ) - ) + window.goodbuddy.agent.cancel(requestId), + ), + ); } catch { notify({ - tone: 'error', - message: t('notices.deleteConversationCancelFailed') - }) - setDeletingConversationId('') - return + tone: "error", + message: t("notices.deleteConversationCancelFailed"), + }); + setDeletingConversationId(""); + return; } const deletingConversation = conversations.find( - (conversation) => conversation.id === conversationId - ) + (conversation) => conversation.id === conversationId, + ); if (deletingConversation && !deletingConversation.remote) { - deletingLocalConversationIdsRef.current.add(conversationId) + deletingLocalConversationIdsRef.current.add(conversationId); try { - await conversationPersistenceQueueRef.current - await window.goodbuddy.conversations.deleteLocal(conversationId) - persistedLocalConversationsRef.current.delete(conversationId) + await conversationPersistenceQueueRef.current; + await window.goodbuddy.conversations.deleteLocal(conversationId); + persistedLocalConversationsRef.current.delete(conversationId); } catch { - deletingLocalConversationIdsRef.current.delete(conversationId) + deletingLocalConversationIdsRef.current.delete(conversationId); notify({ - tone: 'error', - message: t( - 'notices.deleteConversationPersistenceFailed' - ) - }) - setDeletingConversationId('') - return + tone: "error", + message: t("notices.deleteConversationPersistenceFailed"), + }); + setDeletingConversationId(""); + return; } } - setConfirmingConversationId('') - setDeletingConversationId('') + setConfirmingConversationId(""); + setDeletingConversationId(""); if (conversationActionsId === conversationId) { - setConversationActionsId('') + setConversationActionsId(""); } if (renamingConversationId === conversationId) { - setRenamingConversationId('') + setRenamingConversationId(""); } - const browserStop = window.goodbuddy.browser?.stop(conversationId) + const browserStop = window.goodbuddy.browser?.stop(conversationId); if (browserStop) { void browserStop.catch(() => { notify({ - tone: 'error', - message: t('notices.deletedConversationBrowserCloseFailed') - }) - }) + tone: "error", + message: t("notices.deletedConversationBrowserCloseFailed"), + }); + }); } setBrowserStates((current) => { - const next = { ...current } - delete next[conversationId] - return next - }) - const draftAttachments = - attachmentsRef.current.get(conversationId) ?? [] - attachmentsRef.current.delete(conversationId) + const next = { ...current }; + delete next[conversationId]; + return next; + }); + const draftAttachments = attachmentsRef.current.get(conversationId) ?? []; + attachmentsRef.current.delete(conversationId); for (const attachment of draftAttachments) { - void window.goodbuddy.context.remove(attachment.id) + void window.goodbuddy.context.remove(attachment.id); } setAttachmentsByConversation((current) => { - const next = { ...current } - delete next[conversationId] - return next - }) + const next = { ...current }; + delete next[conversationId]; + return next; + }); setConversationDrafts((current) => { - const next = { ...current } - delete next[conversationId] - return next - }) + const next = { ...current }; + delete next[conversationId]; + return next; + }); setChatScrollSnapshots((current) => { - const next = { ...current } - delete next[conversationId] - return next - }) + const next = { ...current }; + delete next[conversationId]; + return next; + }); setVisibleMessageCounts((current) => { - const next = { ...current } - delete next[conversationId] - return next - }) - setConversationActivity(conversationId, false) + const next = { ...current }; + delete next[conversationId]; + return next; + }); + setConversationActivity(conversationId, false); const remaining = conversations.filter( - (conversation) => conversation.id !== conversationId - ) - conversationsRef.current = remaining - deletingLocalConversationIdsRef.current.delete(conversationId) + (conversation) => conversation.id !== conversationId, + ); + conversationsRef.current = remaining; + deletingLocalConversationIdsRef.current.delete(conversationId); const projectRemaining = remaining.filter( - (conversation) => conversation.projectId === activeProjectId - ) - setConversations(remaining) + (conversation) => conversation.projectId === activeProjectId, + ); + setConversations(remaining); if (projectRemaining.length > 0) { if (conversationId === activeId) { - setActiveId(projectRemaining[0]?.id ?? '') + setActiveId(projectRemaining[0]?.id ?? ""); } - return + return; } - if (activeProject?.kind === 'channel') { - setActiveId('') - return + if (activeProject?.kind === "channel") { + setActiveId(""); + return; } const replacement = createConversation( activeProjectId || undefined, runtimeSettings - ? getProjectDefaultRuntimeSelection( - activeProject, - runtimeSettings - ) + ? getProjectDefaultRuntimeSelection(activeProject, runtimeSettings) : undefined, - t('conversation.greeting') - ) - setConversations((current) => [replacement, ...current]) - setActiveId(replacement.id) - } + t("conversation.greeting"), + ); + setConversations((current) => [replacement, ...current]); + setActiveId(replacement.id); + }; const branchConversation = async ( - sourceConversation: Conversation + sourceConversation: Conversation, ): Promise => { if ( branchingConversationId || @@ -6863,461 +6498,431 @@ function App(): React.JSX.Element { activeConversationIds.has(sourceConversation.id) || queuedConversationIds.has(sourceConversation.id) ) { - return + return; } - setBranchingConversationId(sourceConversation.id) + setBranchingConversationId(sourceConversation.id); try { - persistLocalConversationChanges() - await conversationPersistenceQueueRef.current + persistLocalConversationChanges(); + await conversationPersistenceQueueRef.current; if ( - persistedLocalConversationsRef.current.get( - sourceConversation.id - ) !== sourceConversation || + persistedLocalConversationsRef.current.get(sourceConversation.id) !== + sourceConversation || conversationsRef.current.find( - (conversation) => - conversation.id === sourceConversation.id + (conversation) => conversation.id === sourceConversation.id, ) !== sourceConversation ) { - throw new Error(t('notices.conversationPersistenceFailed')) + throw new Error(t("notices.conversationPersistenceFailed")); } const sourceTitle = getConversationDisplayTitle( sourceConversation, - t('conversation.defaultTitle') - ) + t("conversation.defaultTitle"), + ); const branch = await window.goodbuddy.conversations.branchLocal({ sourceConversationId: sourceConversation.id, title: createConversationBranchTitle( sourceTitle, - t('conversation.branch.suffix') - ) - }) - const nextBranch: Conversation = branch - persistedLocalConversationsRef.current.set( - nextBranch.id, - nextBranch - ) + t("conversation.branch.suffix"), + ), + }); + const nextBranch: Conversation = branch; + persistedLocalConversationsRef.current.set(nextBranch.id, nextBranch); setConversations((current) => [ nextBranch, - ...current.filter( - (conversation) => conversation.id !== nextBranch.id - ) - ]) - setSelectedAssistantTaskId(undefined) - setActiveId(nextBranch.id) - setView('chat') + ...current.filter((conversation) => conversation.id !== nextBranch.id), + ]); + setSelectedAssistantTaskId(undefined); + setActiveId(nextBranch.id); + setView("chat"); if (narrowWindow) { - closeNarrowSidebar() + closeNarrowSidebar(); } notify({ - tone: 'success', - message: t('notices.conversationBranched') - }) - requestAnimationFrame(() => inputRef.current?.focus()) + tone: "success", + message: t("notices.conversationBranched"), + }); + requestAnimationFrame(() => inputRef.current?.focus()); } catch (error) { notify({ - tone: 'error', + tone: "error", message: displayErrorMessage( error, - t('notices.conversationBranchFailed') - ) - }) - focusConversationActions(sourceConversation.id) + t("notices.conversationBranchFailed"), + ), + }); + focusConversationActions(sourceConversation.id); } finally { - setBranchingConversationId('') + setBranchingConversationId(""); } - } + }; const focusConversationActions = (conversationId: string): void => { requestAnimationFrame(() => - conversationActionTriggerRefs.current.get(conversationId)?.focus() - ) - } + conversationActionTriggerRefs.current.get(conversationId)?.focus(), + ); + }; - const saveTitle = ( - conversationId: string, - titleInput: string - ): void => { - const title = titleInput.trim().slice(0, 80) + const saveTitle = (conversationId: string, titleInput: string): void => { + const title = titleInput.trim().slice(0, 80); if (!title) { - return + return; } setConversations((current) => current.map((conversation) => conversation.id === conversationId ? { ...conversation, title, updatedAt: Date.now() } - : conversation - ) - ) - setRenamingConversationId('') - focusConversationActions(conversationId) - } + : conversation, + ), + ); + setRenamingConversationId(""); + focusConversationActions(conversationId); + }; const writeClipboardText = useCallback( - async ( - content: string, - successMessage: string - ): Promise => { + async (content: string, successMessage: string): Promise => { try { - await window.goodbuddy.clipboard.writeText(content) + await window.goodbuddy.clipboard.writeText(content); notify({ - tone: 'success', - message: successMessage - }) + tone: "success", + message: successMessage, + }); } catch { notify({ - tone: 'error', - message: t('notices.clipboardUnavailable') - }) + tone: "error", + message: t("notices.clipboardUnavailable"), + }); } }, - [t] - ) + [t], + ); const copyConversation = async ( - conversation: ConversationSnapshot + conversation: ConversationSnapshot, ): Promise => { const transcript = conversation.messages - .map( - (message) => - t('chat.exportSpeaker', { - speaker: - message.role === 'user' ? t('chat.user') : 'GoodBuddy', - content: `${message.content}${formatAttachmentList( - message.attachments, - t - )}` - }) + .map((message) => + t("chat.exportSpeaker", { + speaker: message.role === "user" ? t("chat.user") : "GoodBuddy", + content: `${message.content}${formatAttachmentList( + message.attachments, + t, + )}`, + }), ) - .join('\n\n') - await writeClipboardText( - transcript, - t('notices.conversationCopied') - ) - } + .join("\n\n"); + await writeClipboardText(transcript, t("notices.conversationCopied")); + }; const copyMessage = useCallback( (content: string): Promise => - writeClipboardText(content, t('notices.messageCopied')), - [t, writeClipboardText] - ) + writeClipboardText(content, t("notices.messageCopied")), + [t, writeClipboardText], + ); - const exportConversation = ( - conversation: ConversationSnapshot - ): void => { + const exportConversation = (conversation: ConversationSnapshot): void => { const markdown = [ `# ${conversation.title}`, - '', + "", ...conversation.messages.flatMap((message) => [ - `## ${message.role === 'user' ? t('chat.user') : 'GoodBuddy'}`, - '', - `${message.content}${formatAttachmentList( - message.attachments, - t - )}`, - '' - ]) - ].join('\n') + `## ${message.role === "user" ? t("chat.user") : "GoodBuddy"}`, + "", + `${message.content}${formatAttachmentList(message.attachments, t)}`, + "", + ]), + ].join("\n"); const blob = new Blob([markdown], { - type: 'text/markdown;charset=utf-8' - }) - const url = URL.createObjectURL(blob) - const anchor = document.createElement('a') - anchor.href = url + type: "text/markdown;charset=utf-8", + }); + const url = URL.createObjectURL(blob); + const anchor = document.createElement("a"); + anchor.href = url; anchor.download = `${ - conversation.title.replace(/[\\/:*?"<>|]/g, '_') || - t('conversation.exportFallbackName') - }.md` - anchor.click() - URL.revokeObjectURL(url) - notify({ tone: 'success', message: t('notices.conversationExported') }) - } + conversation.title.replace(/[\\/:*?"<>|]/g, "_") || + t("conversation.exportFallbackName") + }.md`; + anchor.click(); + URL.revokeObjectURL(url); + notify({ tone: "success", message: t("notices.conversationExported") }); + }; - const openImageViewer = useCallback(( - item: ImageViewerItem, - trigger: HTMLElement - ): void => { - if (!imageDataUrlPattern.test(item.src)) { - notify({ - tone: 'error', - message: tRef.current('notices.imageUnavailable') - }) - return - } - imageViewerTriggerRef.current = trigger - setImageViewerItem(item) - }, []) + const openImageViewer = useCallback( + (item: ImageViewerItem, trigger: HTMLElement): void => { + if (!imageDataUrlPattern.test(item.src)) { + notify({ + tone: "error", + message: tRef.current("notices.imageUnavailable"), + }); + return; + } + imageViewerTriggerRef.current = trigger; + setImageViewerItem(item); + }, + [], + ); const closeImageViewer = (): void => { - const trigger = imageViewerTriggerRef.current - setImageViewerItem(undefined) - imageViewerTriggerRef.current = undefined - requestAnimationFrame(() => trigger?.focus()) - } + const trigger = imageViewerTriggerRef.current; + setImageViewerItem(undefined); + imageViewerTriggerRef.current = undefined; + requestAnimationFrame(() => trigger?.focus()); + }; - const openCitationContext = useCallback(async ( - reference: KnowledgeSearchReference - ): Promise => { - setCitationDialog({ - reference, - loading: true - }) - if (!reference.chunkId) { + const openCitationContext = useCallback( + async (reference: KnowledgeSearchReference): Promise => { setCitationDialog({ reference, - loading: false, - error: tRef.current('chat.citations.contextUnavailable') - }) - return - } - try { - const context = - await window.goodbuddy.knowledge.getReferenceContext({ + loading: true, + }); + if (!reference.chunkId) { + setCitationDialog({ + reference, + loading: false, + error: tRef.current("chat.citations.contextUnavailable"), + }); + return; + } + try { + const context = await window.goodbuddy.knowledge.getReferenceContext({ knowledgeBaseId: reference.libraryId, documentId: reference.documentId, - chunkId: reference.chunkId - }) - setCitationDialog({ - reference, - loading: false, - context: { - libraryName: reference.libraryName, - documentName: context.documentTitle, - sourceName: context.sourceDisplayName, - locator: context.locator, - matchedContent: context.matchedContent, - contextContent: context.contextContent, - truncated: context.truncated - } - }) - } catch (reason) { - setCitationDialog({ - reference, - loading: false, - error: - reason instanceof Error - ? reason.message - : tRef.current('chat.citations.contextUnavailable') - }) - } - }, []) + chunkId: reference.chunkId, + }); + setCitationDialog({ + reference, + loading: false, + context: { + libraryName: reference.libraryName, + documentName: context.documentTitle, + sourceName: context.sourceDisplayName, + locator: context.locator, + matchedContent: context.matchedContent, + contextContent: context.contextContent, + truncated: context.truncated, + }, + }); + } catch (reason) { + setCitationDialog({ + reference, + loading: false, + error: + reason instanceof Error + ? reason.message + : tRef.current("chat.citations.contextUnavailable"), + }); + } + }, + [], + ); - const openCitationSource = useCallback(async ( - reference: KnowledgeSearchReference - ): Promise => { - if (!reference.chunkId) { - return - } - try { - await window.goodbuddy.knowledge.openReferenceSource({ - knowledgeBaseId: reference.libraryId, - documentId: reference.documentId, - chunkId: reference.chunkId - }) - } catch (reason) { - notify({ - tone: 'error', - message: - reason instanceof Error - ? reason.message - : tRef.current('chat.citations.openFailed') - }) - } - }, []) + const openCitationSource = useCallback( + async (reference: KnowledgeSearchReference): Promise => { + if (!reference.chunkId) { + return; + } + try { + await window.goodbuddy.knowledge.openReferenceSource({ + knowledgeBaseId: reference.libraryId, + documentId: reference.documentId, + chunkId: reference.chunkId, + }); + } catch (reason) { + notify({ + tone: "error", + message: + reason instanceof Error + ? reason.message + : tRef.current("chat.citations.openFailed"), + }); + } + }, + [], + ); const downloadImage = useCallback((item: ImageViewerItem): void => { if (!imageDataUrlPattern.test(item.src)) { notify({ - tone: 'error', - message: tRef.current('notices.imageUnavailable') - }) - return + tone: "error", + message: tRef.current("notices.imageUnavailable"), + }); + return; } - const anchor = document.createElement('a') - anchor.href = item.src + const anchor = document.createElement("a"); + anchor.href = item.src; anchor.download = getImageDownloadName( item.title, item.src, - tRef.current('chat.images.fallbackTitle') - ) - anchor.rel = 'noopener' - anchor.click() + tRef.current("chat.images.fallbackTitle"), + ); + anchor.rel = "noopener"; + anchor.click(); notify({ - tone: 'info', - message: tRef.current('notices.imageDownloadStarted') - }) - }, []) + tone: "info", + message: tRef.current("notices.imageDownloadStarted"), + }); + }, []); const submit = async ( - queuedDispatch?: ConversationQueueDispatch + queuedDispatch?: ConversationQueueDispatch, ): Promise => { - const queuedInput = queuedDispatch?.input + const queuedInput = queuedDispatch?.input; const releaseQueuedItem = async (): Promise => { if (!queuedDispatch) { - return + return; } try { await window.goodbuddy.conversationQueue.releaseUser( - queuedDispatch.item.id - ) + queuedDispatch.item.id, + ); } catch { notify({ - tone: 'error', - message: t('notices.conversationQueueReleaseFailed') - }) + tone: "error", + message: t("notices.conversationQueueReleaseFailed"), + }); } - } + }; const conversationSnapshot = queuedInput ? conversationsRef.current.find( - (conversation) => - conversation.id === queuedInput.conversationId + (conversation) => conversation.id === queuedInput.conversationId, ) - : activeConversation + : activeConversation; const recoveryProjectId = - queuedInput?.projectId ?? conversationSnapshot?.projectId + queuedInput?.projectId ?? conversationSnapshot?.projectId; if ( recoveryProjectId && - (((!projectRecoverySnapshotReadyRef.current && + ((!projectRecoverySnapshotReadyRef.current && isManagedSshProject( projectsRef.current.find( - (project) => project.id === recoveryProjectId - ) - ))) || + (project) => project.id === recoveryProjectId, + ), + )) || isProjectRecoveryUnsettled( - projectRecoveryByProjectIdRef.current[recoveryProjectId] + projectRecoveryByProjectIdRef.current[recoveryProjectId], )) ) { - await releaseQueuedItem() - return + await releaseQueuedItem(); + return; } const command = - !queuedInput && - activeRuntimeSelection?.provider === 'opencode' + !queuedInput && activeRuntimeSelection?.provider === "opencode" ? runtimeNativeSnapshot?.commands.find( - (candidate) => - candidate.id === selectedRuntimeCommand + (candidate) => candidate.id === selectedRuntimeCommand, ) - : undefined - const commandArguments = queuedInput ? '' : input.trim() - const prompt = queuedInput?.prompt ?? (command - ? `/${command.name}${ - commandArguments ? ` ${commandArguments}` : '' - }` - : commandArguments) + : undefined; + const commandArguments = queuedInput ? "" : input.trim(); + const prompt = + queuedInput?.prompt ?? + (command + ? `/${command.name}${commandArguments ? ` ${commandArguments}` : ""}` + : commandArguments); if (!prompt || !conversationSnapshot) { - await releaseQueuedItem() - return + await releaseQueuedItem(); + return; } if (!queuedInput && selectingContextFilesRef.current) { notify({ - tone: 'info', - message: t('composer.attachmentProgress.waitBeforeSending') - }) - return + tone: "info", + message: t("composer.attachmentProgress.waitBeforeSending"), + }); + return; } if (conversationSnapshot.remote) { notify({ - tone: 'info', - message: t('notices.remoteConversationReadOnly') - }) - await releaseQueuedItem() - return + tone: "info", + message: t("notices.remoteConversationReadOnly"), + }); + await releaseQueuedItem(); + return; } if (!queuedInput && !runtime) { notify({ - tone: 'info', - message: t('runtime.loadingRetry') - }) - return + tone: "info", + message: t("runtime.loadingRetry"), + }); + return; } if ( !queuedInput && - (runtimeSwitching || - runtimeStatusKey !== activeRuntimeSelectionKey) + (runtimeSwitching || runtimeStatusKey !== activeRuntimeSelectionKey) ) { notify({ - tone: 'info', - message: t('runtime.updatingRetry') - }) - return + tone: "info", + message: t("runtime.updatingRetry"), + }); + return; } if (!queuedInput && !runtime?.available) { - return + return; } - const requestId = crypto.randomUUID() - const conversationId = conversationSnapshot.id - const attachmentSnapshot = ( - queuedInput?.attachments ?? attachments - ).slice(0, 8) - const historySnapshot = conversationSnapshot.messages + const requestId = crypto.randomUUID(); + const conversationId = conversationSnapshot.id; + const attachmentSnapshot = (queuedInput?.attachments ?? attachments).slice( + 0, + 8, + ); + const historySnapshot = conversationSnapshot.messages; const retainedHistorySnapshot = historySnapshot .filter( - (message) => - message.state === 'complete' && message.content.trim() + (message) => message.state === "complete" && message.content.trim(), ) - .slice(-500) + .slice(-500); const projectIdSnapshot = queuedInput ? queuedInput.projectId - : activeProjectId || undefined + : activeProjectId || undefined; const knowledgeRetrievalModeSnapshot = queuedInput?.knowledgeRetrievalMode ?? conversationSnapshot.knowledgeRetrievalMode ?? - 'auto' + "auto"; const runtimeSelectionSnapshot = - queuedInput?.runtimeSelection ?? activeRuntimeSelection + queuedInput?.runtimeSelection ?? activeRuntimeSelection; if (!runtimeSelectionSnapshot) { - notify({ tone: 'info', message: t('runtime.notSelected') }) - await releaseQueuedItem() - return + notify({ tone: "info", message: t("runtime.notSelected") }); + await releaseQueuedItem(); + return; } - const runtimeControlSnapshot: RuntimeControl | undefined = - queuedInput - ? queuedInput.runtimeControl - : runtimeSelectionSnapshot.provider === 'opencode' && - (selectedRuntimeAgent || command) + const runtimeControlSnapshot: RuntimeControl | undefined = queuedInput + ? queuedInput.runtimeControl + : runtimeSelectionSnapshot.provider === "opencode" && + (selectedRuntimeAgent || command) + ? { + provider: "opencode", + ...(selectedRuntimeAgent ? { agent: selectedRuntimeAgent } : {}), + ...(command + ? { + command: { + name: command.name, + arguments: commandArguments, + }, + } + : {}), + } + : runtimeSelectionSnapshot.provider === "continue" && + selectedContinuePreset ? { - provider: 'opencode', - ...(selectedRuntimeAgent - ? { agent: selectedRuntimeAgent } - : {}), - ...(command - ? { - command: { - name: command.name, - arguments: commandArguments - } - } - : {}) + provider: "continue", + presetId: selectedContinuePreset, } - : runtimeSelectionSnapshot.provider === 'continue' && - selectedContinuePreset - ? { - provider: 'continue', - presetId: selectedContinuePreset - } - : undefined + : undefined; const selectedExpertSnapshot = queuedInput ? queuedInput.teamMode - ? 'team' - : queuedInput.expertId ?? '' - : runtime?.capability === 'image-generation' - ? '' - : selectedExpertId - const workModeSnapshot = - normalizeInteractiveWorkMode( - queuedInput?.workMode ?? effectiveWorkMode - ) + ? "team" + : (queuedInput.expertId ?? "") + : runtime?.capability === "image-generation" + ? "" + : selectedExpertId; + const workModeSnapshot = normalizeInteractiveWorkMode( + queuedInput?.workMode ?? effectiveWorkMode, + ); const knowledgeLibraryIdsSnapshot = - queuedInput?.knowledgeLibraryIds ?? enabledKnowledgeLibraryIds + queuedInput?.knowledgeLibraryIds ?? enabledKnowledgeLibraryIds; const smartRoutingSnapshot = queuedInput?.smartRouting ?? (!queuedInput && - runtime?.capability !== 'image-generation' && + runtime?.capability !== "image-generation" && runtimeSettings?.subagentSmartRoutingEnabled === true && !selectedExpertSnapshot && supportsSubagentSmartRouting(workModeSnapshot) ? true - : undefined) + : undefined); if (!queuedInput) { const queueInput: ConversationQueueUserInput = { @@ -7327,112 +6932,91 @@ function App(): React.JSX.Element { ...(runtimeControlSnapshot ? { runtimeControl: runtimeControlSnapshot } : {}), - ...(selectedExpertSnapshot && - selectedExpertSnapshot !== 'team' + ...(selectedExpertSnapshot && selectedExpertSnapshot !== "team" ? { expertId: selectedExpertSnapshot } : {}), - ...(selectedExpertSnapshot === 'team' - ? { teamMode: true } - : {}), - ...(smartRoutingSnapshot - ? { smartRouting: true } - : {}), + ...(selectedExpertSnapshot === "team" ? { teamMode: true } : {}), + ...(smartRoutingSnapshot ? { smartRouting: true } : {}), workMode: workModeSnapshot, includeMemoryContext: !command, prompt, attachments: attachmentSnapshot, knowledgeLibraryIds: knowledgeLibraryIdsSnapshot, - knowledgeRetrievalMode: knowledgeRetrievalModeSnapshot - } + knowledgeRetrievalMode: knowledgeRetrievalModeSnapshot, + }; try { - persistLocalConversationChanges() - await conversationPersistenceQueueRef.current + persistLocalConversationChanges(); + await conversationPersistenceQueueRef.current; if ( - !persistedLocalConversationsRef.current.has( - conversationSnapshot.id - ) + !persistedLocalConversationsRef.current.has(conversationSnapshot.id) ) { - throw new Error( - t('notices.conversationPersistenceFailed') - ) + throw new Error(t("notices.conversationPersistenceFailed")); } - await window.goodbuddy.conversationQueue.enqueueUser(queueInput) - setComposerMenuOpen(undefined) - setRuntimeMenuOpen(false) - setInput('') - updateAttachments([]) + await window.goodbuddy.conversationQueue.enqueueUser(queueInput); + setComposerMenuOpen(undefined); + setRuntimeMenuOpen(false); + setInput(""); + updateAttachments([]); if (command) { - setSelectedRuntimeCommand('') + setSelectedRuntimeCommand(""); } } catch (reason) { notify({ - tone: 'error', - message: displayErrorMessage( - reason, - t('notices.sendFailed') - ) - }) + tone: "error", + message: displayErrorMessage(reason, t("notices.sendFailed")), + }); } - return + return; } - if ( - dispatchedConversationQueueItems.current.has( - queuedDispatch.item.id - ) - ) { - return + if (dispatchedConversationQueueItems.current.has(queuedDispatch.item.id)) { + return; } - dispatchedConversationQueueItems.current.add( - queuedDispatch.item.id - ) - setComposerMenuOpen(undefined) - setRuntimeMenuOpen(false) - preparingConversations.current.add(conversationId) - setConversationActivity(conversationId, true) + dispatchedConversationQueueItems.current.add(queuedDispatch.item.id); + setComposerMenuOpen(undefined); + setRuntimeMenuOpen(false); + preparingConversations.current.add(conversationId); + setConversationActivity(conversationId, true); const userMessage: Message = { id: crypto.randomUUID(), queueItemId: queuedDispatch.item.id, - role: 'user', + role: "user", content: prompt, createdAt: Date.now(), - state: 'complete', + state: "complete", attachments: - attachmentSnapshot.length > 0 ? attachmentSnapshot : undefined - } + attachmentSnapshot.length > 0 ? attachmentSnapshot : undefined, + }; setConversations((current) => current.map((conversation) => conversation.id === conversationId ? { ...conversation, title: - conversation.title === '新对话' + conversation.title === "新对话" ? prompt.slice(0, 24) : conversation.title, updatedAt: Date.now(), - messages: [ - ...conversation.messages.slice(-499), - userMessage - ] + messages: [...conversation.messages.slice(-499), userMessage], } - : conversation - ) - ) + : conversation, + ), + ); const memoryContext = queuedInput.includeMemoryContext ? buildMemoryContext(assistantMemories) - : '' + : ""; const executionPrompt = memoryContext ? `${prompt}\n\n${memoryContext}` - : prompt + : prompt; const assistantMessage: Message = { id: crypto.randomUUID(), - role: 'assistant', - content: '', + role: "assistant", + content: "", blocks: [], createdAt: Date.now(), - state: 'streaming', - status: t('runtime.connecting') - } + state: "streaming", + status: t("runtime.connecting"), + }; activeRuns.current.set(requestId, { conversationId, @@ -7442,13 +7026,13 @@ function App(): React.JSX.Element { runtimeSettings ? resolveContextMetricsRuntimeSelection( runtimeSelectionSnapshot, - runtimeSettings + runtimeSettings, ) - : runtimeSelectionSnapshot - ) - }) - preparingConversations.current.delete(conversationId) - const startedAt = new Date().toISOString() + : runtimeSelectionSnapshot, + ), + }); + preparingConversations.current.delete(conversationId); + const startedAt = new Date().toISOString(); setAssistantTasks((current) => [ { @@ -7457,22 +7041,22 @@ function App(): React.JSX.Element { conversationId, title: prompt.slice(0, 120), instructions: prompt, - origin: 'user' as const, - status: 'running' as const, + origin: "user" as const, + status: "running" as const, createdAt: startedAt, - startedAt + startedAt, }, - ...current - ].slice(0, 100) - ) + ...current, + ].slice(0, 100), + ); recordActivity({ conversationId, requestId, - kind: 'request', + kind: "request", title: prompt.slice(0, 120), - detail: t('notices.userStartedTask'), - status: 'running' - }) + detail: t("notices.userStartedTask"), + status: "running", + }); setConversations((current) => current.map((conversation) => conversation.id === conversationId @@ -7481,12 +7065,12 @@ function App(): React.JSX.Element { updatedAt: Date.now(), messages: [ ...conversation.messages.slice(-499), - assistantMessage - ] + assistantMessage, + ], } - : conversation - ) - ) + : conversation, + ), + ); try { await window.goodbuddy.agent.run({ requestId, @@ -7496,39 +7080,34 @@ function App(): React.JSX.Element { runtimeSelection: runtimeSelectionSnapshot, runtimeControl: runtimeControlSnapshot, expertId: - selectedExpertSnapshot && selectedExpertSnapshot !== 'team' + selectedExpertSnapshot && selectedExpertSnapshot !== "team" ? selectedExpertSnapshot : undefined, - teamMode: selectedExpertSnapshot === 'team', + teamMode: selectedExpertSnapshot === "team", smartRouting: smartRoutingSnapshot, workMode: workModeSnapshot, prompt: executionPrompt, knowledgeLibraryIds: knowledgeLibraryIdsSnapshot, knowledgeRetrievalMode: knowledgeRetrievalModeSnapshot, - contextIds: attachmentSnapshot.map( - (attachment) => attachment.id - ), - contextCompressionState: - conversationSnapshot.contextCompressionState, + contextIds: attachmentSnapshot.map((attachment) => attachment.id), + contextCompressionState: conversationSnapshot.contextCompressionState, history: retainedHistorySnapshot.map((message) => ({ role: message.role, - content: message.content + content: message.content, })), - historyMessageIds: retainedHistorySnapshot.map( - (message) => message.id - ), + historyMessageIds: retainedHistorySnapshot.map((message) => message.id), currentUserMessageId: userMessage.id, - currentAssistantMessageId: assistantMessage.id - }) + currentAssistantMessageId: assistantMessage.id, + }); for (const attachment of attachmentSnapshot) { - void window.goodbuddy.context.remove(attachment.id) + void window.goodbuddy.context.remove(attachment.id); } } catch (error) { - preparingConversations.current.delete(conversationId) - activeRuns.current.delete(requestId) - setConversationActivity(conversationId, false) + preparingConversations.current.delete(conversationId); + activeRuns.current.delete(requestId); + setConversationActivity(conversationId, false); for (const attachment of attachmentSnapshot) { - void window.goodbuddy.context.remove(attachment.id) + void window.goodbuddy.context.remove(attachment.id); } setConversations((current) => current.map((conversation) => @@ -7536,7 +7115,7 @@ function App(): React.JSX.Element { ? { ...conversation, title: - conversationSnapshot.title === '新对话' && + conversationSnapshot.title === "新对话" && conversation.title === prompt.slice(0, 24) ? conversationSnapshot.title : conversation.title, @@ -7544,95 +7123,87 @@ function App(): React.JSX.Element { messages: conversation.messages.filter( (message) => message.id !== userMessage.id && - message.id !== assistantMessage.id - ) + message.id !== assistantMessage.id, + ), } - : conversation - ) - ) + : conversation, + ), + ); setAssistantTasks((current) => - current.filter((task) => task.id !== requestId) - ) + current.filter((task) => task.id !== requestId), + ); setActivityRecords((current) => - current.filter((record) => record.requestId !== requestId) - ) + current.filter((record) => record.requestId !== requestId), + ); notify({ - tone: 'error', - message: displayErrorMessage( - error, - t('notices.sendFailed') - ) - }) - await releaseQueuedItem() + tone: "error", + message: displayErrorMessage(error, t("notices.sendFailed")), + }); + await releaseQueuedItem(); } finally { - dispatchedConversationQueueItems.current.delete( - queuedDispatch.item.id - ) + dispatchedConversationQueueItems.current.delete(queuedDispatch.item.id); } - } + }; useLayoutEffect(() => { conversationQueueDispatchRef.current = (dispatch) => { - void submit(dispatch) - } - }) + void submit(dispatch); + }; + }); const compactRuntimeContext = async (): Promise => { if ( !activeConversation || !activeRuntimeSelection || - (activeRuntimeSelection.provider !== 'opencode' && - activeRuntimeSelection.provider !== 'continue') || + (activeRuntimeSelection.provider !== "opencode" && + activeRuntimeSelection.provider !== "continue") || runtimeContextCompacting || isRunning ) { - return + return; } const history = activeConversation.messages .filter( - (message) => - message.state === 'complete' && message.content.trim() + (message) => message.state === "complete" && message.content.trim(), ) - .slice(-500) + .slice(-500); if (history.length < 2) { notify({ - tone: 'info', - message: t('composer.context.nothingToCompact'), - dedupeKey: 'runtime-context-compact' - }) - return + tone: "info", + message: t("composer.context.nothingToCompact"), + dedupeKey: "runtime-context-compact", + }); + return; } - const requestId = crypto.randomUUID() - setRuntimeContextCompacting(true) + const requestId = crypto.randomUUID(); + setRuntimeContextCompacting(true); try { - const result = - await window.goodbuddy.agent.compactConversation({ - requestId, - conversationId: activeConversation.id, - projectId: activeConversation.projectId, - runtimeSelection: activeRuntimeSelection, - history: history.map((message) => ({ - role: message.role, - content: message.content - })), - historyMessageIds: history.map((message) => message.id), - contextCompressionState: - activeConversation.contextCompressionState - }) + const result = await window.goodbuddy.agent.compactConversation({ + requestId, + conversationId: activeConversation.id, + projectId: activeConversation.projectId, + runtimeSelection: activeRuntimeSelection, + history: history.map((message) => ({ + role: message.role, + content: message.content, + })), + historyMessageIds: history.map((message) => message.id), + contextCompressionState: activeConversation.contextCompressionState, + }); if (result.contextCompressionState) { - const state = result.contextCompressionState + const state = result.contextCompressionState; const remainingHistory = history.slice( - Math.min(state.coveredMessageCount, history.length) - ) + Math.min(state.coveredMessageCount, history.length), + ); const estimatedAfterTokens = estimatedContextRequestOverheadTokens + estimateMessagesTokens([ ...buildConversationSummaryHistory(state.summary), ...remainingHistory.map((message) => ({ role: message.role, - content: message.content - })) - ]) + content: message.content, + })), + ]); setConversations((current) => current.map((conversation) => conversation.id === activeConversation.id @@ -7640,769 +7211,744 @@ function App(): React.JSX.Element { ...conversation, contextCompressionState: state, contextMetrics: { - runtimeSelectionKey: - activeRuntimeSelectionKey, + runtimeSelectionKey: activeRuntimeSelectionKey, contextTokens: estimatedAfterTokens, - source: 'estimated', - basis: 'conversation' + source: "estimated", + basis: "conversation", }, - updatedAt: Date.now() + updatedAt: Date.now(), } - : conversation - ) - ) + : conversation, + ), + ); } notify({ - tone: result.compacted ? 'success' : 'info', + tone: result.compacted ? "success" : "info", message: result.detail, - dedupeKey: 'runtime-context-compact' - }) + dedupeKey: "runtime-context-compact", + }); } catch (reason) { notify({ - tone: 'error', + tone: "error", message: reason instanceof Error ? reason.message - : t('composer.context.compactFailed'), - dedupeKey: 'runtime-context-compact' - }) + : t("composer.context.compactFailed"), + dedupeKey: "runtime-context-compact", + }); } finally { - setRuntimeContextCompacting(false) + setRuntimeContextCompacting(false); } - } + }; const stop = async (): Promise => { const requestId = [...activeRuns.current.entries()].find( - ([, run]) => run.conversationId === activeId - )?.[0] + ([, run]) => run.conversationId === activeId, + )?.[0]; if (requestId) { try { - await window.goodbuddy.agent.cancel(requestId) + await window.goodbuddy.agent.cancel(requestId); } catch { - notify({ tone: 'error', message: t('notices.stopFailed') }) + notify({ tone: "error", message: t("notices.stopFailed") }); } } - } + }; - const respondToApproval = useCallback(async ( - conversationId: string, - messageId: string, - approvalId: string, - decision: ApprovalDecision - ): Promise => { - try { - await window.goodbuddy.agent.respondApproval(approvalId, decision) - const approved = decision !== 'deny' - const decisionLabel = { - deny: tRef.current('chat.approval.decisionDeny'), - once: tRef.current('chat.approval.decisionOnce'), - session: tRef.current('chat.approval.decisionSession'), - permanent: tRef.current('chat.approval.decisionPermanent') - }[decision] - setActivityRecords((current) => { - let updated = false - return current.map((record) => { - if ( - !updated && - record.conversationId === conversationId && - record.kind === 'approval' && - record.status === 'pending' - ) { - updated = true - return { - ...record, - status: approved ? ('completed' as const) : ('denied' as const), - detail: `${record.detail}\n${tRef.current( - 'notices.userDecision', - { decision: decisionLabel } - )}` + const respondToApproval = useCallback( + async ( + conversationId: string, + messageId: string, + approvalId: string, + decision: ApprovalDecision, + ): Promise => { + try { + await window.goodbuddy.agent.respondApproval(approvalId, decision); + const approved = decision !== "deny"; + const decisionLabel = { + deny: tRef.current("chat.approval.decisionDeny"), + once: tRef.current("chat.approval.decisionOnce"), + session: tRef.current("chat.approval.decisionSession"), + permanent: tRef.current("chat.approval.decisionPermanent"), + }[decision]; + setActivityRecords((current) => { + let updated = false; + return current.map((record) => { + if ( + !updated && + record.conversationId === conversationId && + record.kind === "approval" && + record.status === "pending" + ) { + updated = true; + return { + ...record, + status: approved ? ("completed" as const) : ("denied" as const), + detail: `${record.detail}\n${tRef.current( + "notices.userDecision", + { decision: decisionLabel }, + )}`, + }; } - } - return record - }) - }) - updateMessage(conversationId, messageId, (message) => ({ - ...message, - approval: undefined, - status: - approved && message.task - ? undefined - : approved - ? tRef.current('chat.approval.executing', { - decision: decisionLabel - }) - : tRef.current('chat.approval.denied') - })) - } catch { - updateMessage(conversationId, messageId, (message) => ({ - ...message, - status: tRef.current('chat.approval.responseFailed') - })) - } - }, [updateMessage]) + return record; + }); + }); + updateMessage(conversationId, messageId, (message) => ({ + ...message, + approval: undefined, + status: + approved && message.task + ? undefined + : approved + ? tRef.current("chat.approval.executing", { + decision: decisionLabel, + }) + : tRef.current("chat.approval.denied"), + })); + } catch { + updateMessage(conversationId, messageId, (message) => ({ + ...message, + status: tRef.current("chat.approval.responseFailed"), + })); + } + }, + [updateMessage], + ); - const respondToQuestion = useCallback(async ( - conversationId: string, - messageId: string, - questionId: string, - answers?: AgentQuestionAnswer[] - ): Promise => { - await window.goodbuddy.agent.respondQuestion(questionId, answers) - updateMessage(conversationId, messageId, (message) => ({ - ...message, - question: undefined, - status: answers - ? tRef.current('chat.status.answerSubmitted') - : tRef.current('chat.status.questionSkipped') - })) - }, [updateMessage]) + const respondToQuestion = useCallback( + async ( + conversationId: string, + messageId: string, + questionId: string, + answers?: AgentQuestionAnswer[], + ): Promise => { + await window.goodbuddy.agent.respondQuestion(questionId, answers); + updateMessage(conversationId, messageId, (message) => ({ + ...message, + question: undefined, + status: answers + ? tRef.current("chat.status.answerSubmitted") + : tRef.current("chat.status.questionSkipped"), + })); + }, + [updateMessage], + ); const addContext = async ( - action: () => Promise + action: () => Promise, ): Promise => { - const conversationId = activeId - setContextError(undefined) + const conversationId = activeId; + setContextError(undefined); try { - const result = await action() - const selected = Array.isArray(result) ? result : [result] - const current = - attachmentsRef.current.get(conversationId) ?? [] + const result = await action(); + const selected = Array.isArray(result) ? result : [result]; + const current = attachmentsRef.current.get(conversationId) ?? []; const unique = selected.filter( - (item) => - !current.some((existing) => existing.id === item.id) - ) - const accepted = unique.slice( - 0, - Math.max(0, 8 - current.length) - ) + (item) => !current.some((existing) => existing.id === item.id), + ); + const accepted = unique.slice(0, Math.max(0, 8 - current.length)); for (const attachment of unique.slice(accepted.length)) { - void window.goodbuddy.context.remove(attachment.id) + void window.goodbuddy.context.remove(attachment.id); } - updateAttachments([...current, ...accepted]) + updateAttachments([...current, ...accepted]); if (accepted.length < unique.length) { - setContextError(t('composer.errors.attachmentLimit')) + setContextError(t("composer.errors.attachmentLimit")); } } catch (reason) { setContextError( reason instanceof Error ? reason.message - : t('composer.errors.addContext') - ) + : t("composer.errors.addContext"), + ); } - } + }; const selectContextFiles = async (): Promise => { if (selectingContextFilesRef.current) { - return + return; } - selectingContextFilesRef.current = true - setSelectingContextFiles(true) - setFileSelectionProgress(undefined) + selectingContextFilesRef.current = true; + setSelectingContextFiles(true); + setFileSelectionProgress(undefined); try { - await addContext(() => window.goodbuddy.context.selectFiles()) + await addContext(() => window.goodbuddy.context.selectFiles()); } finally { - selectingContextFilesRef.current = false - setSelectingContextFiles(false) - setFileSelectionProgress(undefined) + selectingContextFilesRef.current = false; + setSelectingContextFiles(false); + setFileSelectionProgress(undefined); } - } + }; const startWebSpeechInput = async (): Promise => { - const SpeechRecognition = - getSpeechRecognitionConstructor(window) + const SpeechRecognition = getSpeechRecognitionConstructor(window); if (!SpeechRecognition) { notify({ - tone: 'info', - message: t('composer.voice.unsupported') - }) - return + tone: "info", + message: t("composer.voice.unsupported"), + }); + return; } - setVoiceListening(true) - setVoiceRecording(false) - let started = false + setVoiceListening(true); + setVoiceRecording(false); + let started = false; try { const prepared = await prepareSpeechRecognition( SpeechRecognition, - 'zh-CN', + "zh-CN", () => { notify({ - tone: 'info', - message: t('composer.voice.downloadingPack'), - dedupeKey: 'speech-status' - }) - } - ) - const { recognition } = prepared + tone: "info", + message: t("composer.voice.downloadingPack"), + dedupeKey: "speech-status", + }); + }, + ); + const { recognition } = prepared; recognition.onresult = (event) => { - const transcript = event.results[0]?.[0]?.transcript?.trim() + const transcript = event.results[0]?.[0]?.transcript?.trim(); if (transcript) { setInput((current) => - current ? `${current} ${transcript}` : transcript - ) + current ? `${current} ${transcript}` : transcript, + ); notify({ - tone: 'success', - message: t('composer.voice.transcribed'), - dedupeKey: 'speech-status' - }) + tone: "success", + message: t("composer.voice.transcribed"), + dedupeKey: "speech-status", + }); } - } + }; recognition.onerror = (event) => { notify({ - tone: 'error', + tone: "error", message: describeSpeechRecognitionError(event), - dedupeKey: 'speech-status' - }) - setVoiceListening(false) - setVoiceRecording(false) - } + dedupeKey: "speech-status", + }); + setVoiceListening(false); + setVoiceRecording(false); + }; recognition.onend = () => { - setVoiceListening(false) - setVoiceRecording(false) - } - recognition.start() - started = true - setVoiceRecording(true) + setVoiceListening(false); + setVoiceRecording(false); + }; + recognition.start(); + started = true; + setVoiceRecording(true); notify({ - tone: 'info', + tone: "info", message: prepared.local - ? t('composer.voice.localListening') - : t('composer.voice.systemListening'), - dedupeKey: 'speech-status' - }) + ? t("composer.voice.localListening") + : t("composer.voice.systemListening"), + dedupeKey: "speech-status", + }); } catch (reason) { notify({ - tone: 'error', + tone: "error", message: reason instanceof Error ? reason.message - : t('composer.voice.startFailed'), - dedupeKey: 'speech-status' - }) + : t("composer.voice.startFailed"), + dedupeKey: "speech-status", + }); } finally { if (!started) { - setVoiceListening(false) - setVoiceRecording(false) + setVoiceListening(false); + setVoiceRecording(false); } } - } + }; const startVoiceInput = async (): Promise => { - const speech = window.goodbuddy.speech + const speech = window.goodbuddy.speech; if (!speech) { - await startWebSpeechInput() - return + await startWebSpeechInput(); + return; } const audioWindow = window as typeof window & { - webkitAudioContext?: typeof AudioContext - } + webkitAudioContext?: typeof AudioContext; + }; const AudioContextType = - audioWindow.AudioContext ?? audioWindow.webkitAudioContext + audioWindow.AudioContext ?? audioWindow.webkitAudioContext; if (!navigator.mediaDevices?.getUserMedia || !AudioContextType) { notify({ - tone: 'error', - message: t('composer.voice.microphoneUnavailable'), - dedupeKey: 'speech-status' - }) - return + tone: "error", + message: t("composer.voice.microphoneUnavailable"), + dedupeKey: "speech-status", + }); + return; } - setVoiceListening(true) - setVoiceRecording(false) - voiceStartingRef.current = true + setVoiceListening(true); + setVoiceRecording(false); + voiceStartingRef.current = true; try { const recording = await startPcmRecording( navigator.mediaDevices, - AudioContextType - ) - voiceStartingRef.current = false + AudioContextType, + ); + voiceStartingRef.current = false; if (voiceDisposedRef.current) { - void recording.result.catch(() => undefined) - recording.cancel() - return + void recording.result.catch(() => undefined); + recording.cancel(); + return; } - voiceRecordingRef.current = recording - setVoiceRecording(true) + voiceRecordingRef.current = recording; + setVoiceRecording(true); notify({ - tone: 'info', - message: t('composer.voice.recording'), - dedupeKey: 'speech-status' - }) + tone: "info", + message: t("composer.voice.recording"), + dedupeKey: "speech-status", + }); void recording.result .then(async ({ audio, sampleRate }) => { - voiceRecordingRef.current = undefined - setVoiceRecording(false) - const requestId = crypto.randomUUID() - voiceRequestIdRef.current = requestId + voiceRecordingRef.current = undefined; + setVoiceRecording(false); + const requestId = crypto.randomUUID(); + voiceRequestIdRef.current = requestId; notify({ - tone: 'info', - message: t('composer.voice.localRecognizing'), - dedupeKey: 'speech-status' - }) + tone: "info", + message: t("composer.voice.localRecognizing"), + dedupeKey: "speech-status", + }); const result = await speech.transcribe({ requestId, sampleRate, - audio - }) + audio, + }); if (voiceRequestIdRef.current !== requestId) { - return + return; } - const transcript = result.text.trim() + const transcript = result.text.trim(); if (!transcript) { notify({ - tone: 'info', - message: t('composer.voice.noSpeech'), - dedupeKey: 'speech-status' - }) - return + tone: "info", + message: t("composer.voice.noSpeech"), + dedupeKey: "speech-status", + }); + return; } setInput((current) => - current ? `${current} ${transcript}` : transcript - ) + current ? `${current} ${transcript}` : transcript, + ); notify({ - tone: 'success', - message: t('composer.voice.transcribed'), - dedupeKey: 'speech-status' - }) + tone: "success", + message: t("composer.voice.transcribed"), + dedupeKey: "speech-status", + }); }) .catch((reason: unknown) => { notify({ tone: - reason instanceof Error && - reason.name === 'AbortError' - ? 'info' - : 'error', + reason instanceof Error && reason.name === "AbortError" + ? "info" + : "error", message: - reason instanceof Error && - reason.name === 'AbortError' - ? t('composer.voice.cancelled') + reason instanceof Error && reason.name === "AbortError" + ? t("composer.voice.cancelled") : reason instanceof Error ? reason.message - : t('composer.voice.localFailed'), - dedupeKey: 'speech-status' - }) + : t("composer.voice.localFailed"), + dedupeKey: "speech-status", + }); }) .finally(() => { - voiceRequestIdRef.current = undefined - setVoiceListening(false) - setVoiceRecording(false) - }) + voiceRequestIdRef.current = undefined; + setVoiceListening(false); + setVoiceRecording(false); + }); } catch (reason) { - voiceStartingRef.current = false - setVoiceListening(false) - setVoiceRecording(false) + voiceStartingRef.current = false; + setVoiceListening(false); + setVoiceRecording(false); notify({ - tone: 'error', + tone: "error", message: - reason instanceof Error && - reason.name === 'NotAllowedError' - ? t('composer.voice.permissionDenied') + reason instanceof Error && reason.name === "NotAllowedError" + ? t("composer.voice.permissionDenied") : reason instanceof Error ? reason.message - : t('composer.voice.recordingStartFailed'), - dedupeKey: 'speech-status' - }) + : t("composer.voice.recordingStartFailed"), + dedupeKey: "speech-status", + }); } - } + }; const toggleVoiceInput = (): void => { if (voiceStartingRef.current) { - return + return; } - const recording = voiceRecordingRef.current + const recording = voiceRecordingRef.current; if (recording) { - setVoiceRecording(false) - recording.stop() + setVoiceRecording(false); + recording.stop(); notify({ - tone: 'info', - message: t('composer.voice.preparing'), - dedupeKey: 'speech-status' - }) - return + tone: "info", + message: t("composer.voice.preparing"), + dedupeKey: "speech-status", + }); + return; } - const requestId = voiceRequestIdRef.current + const requestId = voiceRequestIdRef.current; if (requestId) { - voiceRequestIdRef.current = undefined - void window.goodbuddy.speech?.cancel(requestId) + voiceRequestIdRef.current = undefined; + void window.goodbuddy.speech?.cancel(requestId); notify({ - tone: 'info', - message: t('composer.voice.cancelled'), - dedupeKey: 'speech-status' - }) - setVoiceListening(false) - setVoiceRecording(false) - return + tone: "info", + message: t("composer.voice.cancelled"), + dedupeKey: "speech-status", + }); + setVoiceListening(false); + setVoiceRecording(false); + return; } - void startVoiceInput() - } + void startVoiceInput(); + }; const refreshSelectedKnowledge = async (): Promise => { - await refreshKnowledge(knowledgeSnapshot.selectedLibraryId) - } + await refreshKnowledge(knowledgeSnapshot.selectedLibraryId); + }; const createKnowledgeLibrary = async ( - input: Parameters< - typeof window.goodbuddy.knowledge.createLibrary - >[0] + input: Parameters[0], ): Promise => { - const library = await window.goodbuddy.knowledge.createLibrary(input) - setEnabledKnowledgeLibraryIds((current) => [...current, library.id]) - await refreshKnowledge(library.id) - } + const library = await window.goodbuddy.knowledge.createLibrary(input); + setEnabledKnowledgeLibraryIds((current) => [...current, library.id]); + await refreshKnowledge(library.id); + }; const deleteKnowledgeLibrary = async (libraryId: string): Promise => { - await window.goodbuddy.knowledge.deleteLibrary(libraryId) - await refreshKnowledge() - } + await window.goodbuddy.knowledge.deleteLibrary(libraryId); + await refreshKnowledge(); + }; const runKnowledgeSourceAction = async ( - action: () => Promise + action: () => Promise, ): Promise => { setKnowledgeOperationCount((count) => { - const next = count + 1 - knowledgeOperationCountRef.current = next - return next - }) + const next = count + 1; + knowledgeOperationCountRef.current = next; + return next; + }); try { - const result = await action() - await refreshSelectedKnowledge() - return result + const result = await action(); + await refreshSelectedKnowledge(); + return result; } catch (error) { - await refreshSelectedKnowledge().catch(() => undefined) - throw error + await refreshSelectedKnowledge().catch(() => undefined); + throw error; } finally { setKnowledgeOperationCount((count) => { - const next = Math.max(0, count - 1) - knowledgeOperationCountRef.current = next - return next - }) + const next = Math.max(0, count - 1); + knowledgeOperationCountRef.current = next; + return next; + }); } - } + }; const openActivityConversation = (conversationId: string): void => { const conversation = conversations.find( - (candidate) => candidate.id === conversationId - ) + (candidate) => candidate.id === conversationId, + ); if (!conversation) { notify({ - tone: 'info', - message: t('notices.conversationDeleted') - }) - return + tone: "info", + message: t("notices.conversationDeleted"), + }); + return; } if (conversation.projectId) { const project = projects.find( - (candidate) => candidate.id === conversation.projectId - ) - setActiveProjectId(conversation.projectId) + (candidate) => candidate.id === conversation.projectId, + ); + setActiveProjectId(conversation.projectId); if (project) { - setWorkMode( - normalizeInteractiveWorkMode(project.defaultWorkMode) - ) + setWorkMode(normalizeInteractiveWorkMode(project.defaultWorkMode)); } } - setActiveId(conversationId) + setActiveId(conversationId); setUnreadConversationIds((current) => { if (!current.has(conversationId)) { - return current + return current; } - const next = new Set(current) - next.delete(conversationId) - return next - }) - setView('chat') - } + const next = new Set(current); + next.delete(conversationId); + return next; + }); + setView("chat"); + }; const openAssistantTask = (task: AssistantTask): void => { if (!task.conversationId) { notify({ - tone: 'info', - message: t('notices.conversationDeleted') - }) - return + tone: "info", + message: t("notices.conversationDeleted"), + }); + return; } const conversation = conversations.find( - (candidate) => candidate.id === task.conversationId - ) + (candidate) => candidate.id === task.conversationId, + ); if (!conversation) { notify({ - tone: 'info', - message: t('notices.conversationDeleted') - }) - return + tone: "info", + message: t("notices.conversationDeleted"), + }); + return; } if (task.projectId) { const project = projects.find( - (candidate) => candidate.id === task.projectId - ) - setActiveProjectId(task.projectId) + (candidate) => candidate.id === task.projectId, + ); + setActiveProjectId(task.projectId); if (project) { - setWorkMode( - normalizeInteractiveWorkMode(project.defaultWorkMode) - ) + setWorkMode(normalizeInteractiveWorkMode(project.defaultWorkMode)); } } - setSelectedAssistantTaskId(task.id) + setSelectedAssistantTaskId(task.id); setExpandedTaskConversationIds((current) => { - const next = new Set(current) - next.add(conversation.id) - return next - }) - setActiveId(conversation.id) - setView('chat') - } + const next = new Set(current); + next.add(conversation.id); + return next; + }); + setActiveId(conversation.id); + setView("chat"); + }; const openCustomTaskDialog = ( - defaultDestination: CustomTaskDestination + defaultDestination: CustomTaskDestination, ): void => { - if (!activeProject || activeProject.kind !== 'user') { + if (!activeProject || activeProject.kind !== "user") { notify({ - tone: 'info', - message: t('customTask.errors.projectUnavailable') - }) - return + tone: "info", + message: t("customTask.errors.projectUnavailable"), + }); + return; } - setCustomTaskDialog({ defaultDestination }) - } + setCustomTaskDialog({ defaultDestination }); + }; const createCustomTask = async ( - input: Parameters[0] + input: Parameters[0], ): Promise => { if (input.conversationId) { - persistLocalConversationChanges() - await conversationPersistenceQueueRef.current + persistLocalConversationChanges(); + await conversationPersistenceQueueRef.current; } - const schedule = await window.goodbuddy.schedules.create(input) + const schedule = await window.goodbuddy.schedules.create(input); setAssistantSchedules((current) => [ schedule, - ...current.filter((item) => item.id !== schedule.id) - ]) - setSelectedAssistantTaskId(schedule.taskId) + ...current.filter((item) => item.id !== schedule.id), + ]); + setSelectedAssistantTaskId(schedule.taskId); setExpandedTaskConversationIds((current) => { - const next = new Set(current) - next.add(schedule.conversationId) - return next - }) + const next = new Set(current); + next.add(schedule.conversationId); + return next; + }); const [conversationResult, taskResult, scheduleResult] = await Promise.allSettled([ window.goodbuddy.conversations.list(), window.goodbuddy.tasks.list(), - window.goodbuddy.schedules.list() - ]) - if (conversationResult.status === 'fulfilled') { + window.goodbuddy.schedules.list(), + ]); + if (conversationResult.status === "fulfilled") { setConversations((current) => mergePersistedConversations( current, conversationResult.value, - persistedLocalConversationsRef.current - ) - ) + persistedLocalConversationsRef.current, + ), + ); } else { notify({ - tone: 'error', - message: t('notices.remoteConversationRefreshFailed'), - dedupeKey: 'custom-task-conversation-refresh' - }) + tone: "error", + message: t("notices.remoteConversationRefreshFailed"), + dedupeKey: "custom-task-conversation-refresh", + }); } - if (taskResult.status === 'fulfilled') { - setAssistantTasks(taskResult.value) + if (taskResult.status === "fulfilled") { + setAssistantTasks(taskResult.value); } - if (scheduleResult.status === 'fulfilled') { - setAssistantSchedules(scheduleResult.value) + if (scheduleResult.status === "fulfilled") { + setAssistantSchedules(scheduleResult.value); } if ( - taskResult.status === 'rejected' || - scheduleResult.status === 'rejected' + taskResult.status === "rejected" || + scheduleResult.status === "rejected" ) { notify({ - tone: 'error', - message: t('notices.taskHistoryReadFailed'), - dedupeKey: 'custom-task-discovery-refresh' - }) + tone: "error", + message: t("notices.taskHistoryReadFailed"), + dedupeKey: "custom-task-discovery-refresh", + }); } if (schedule.projectId) { const project = projects.find( - (candidate) => candidate.id === schedule.projectId - ) - setActiveProjectId(schedule.projectId) + (candidate) => candidate.id === schedule.projectId, + ); + setActiveProjectId(schedule.projectId); if (project) { - setWorkMode( - normalizeInteractiveWorkMode(project.defaultWorkMode) - ) + setWorkMode(normalizeInteractiveWorkMode(project.defaultWorkMode)); } } - setActiveId(schedule.conversationId) - setView('chat') - return schedule - } + setActiveId(schedule.conversationId); + setView("chat"); + return schedule; + }; - const runAssistantSchedule = async ( - scheduleId: string - ): Promise => { - await window.goodbuddy.schedules.runNow(scheduleId) + const runAssistantSchedule = async (scheduleId: string): Promise => { + await window.goodbuddy.schedules.runNow(scheduleId); notify({ - tone: 'success', - message: t('notices.scheduleStarted') - }) - } + tone: "success", + message: t("notices.scheduleStarted"), + }); + }; const setAssistantScheduleEnabled = async ( scheduleId: string, - enabled: boolean + enabled: boolean, ): Promise => { - await window.goodbuddy.schedules.setEnabled(scheduleId, enabled) + await window.goodbuddy.schedules.setEnabled(scheduleId, enabled); setAssistantSchedules((current) => current.map((schedule) => - schedule.id === scheduleId - ? { ...schedule, enabled } - : schedule - ) - ) - } + schedule.id === scheduleId ? { ...schedule, enabled } : schedule, + ), + ); + }; - const removeAssistantSchedule = async ( - scheduleId: string - ): Promise => { - await window.goodbuddy.schedules.remove(scheduleId) + const removeAssistantSchedule = async (scheduleId: string): Promise => { + await window.goodbuddy.schedules.remove(scheduleId); setAssistantSchedules((current) => - current.filter((schedule) => schedule.id !== scheduleId) - ) - } + current.filter((schedule) => schedule.id !== scheduleId), + ); + }; const clearLocalData = async (): Promise => { - conversationPersistencePausedRef.current = true + conversationPersistencePausedRef.current = true; try { - await conversationPersistenceQueueRef.current + await conversationPersistenceQueueRef.current; for (const requestId of activeRuns.current.keys()) { - await window.goodbuddy.agent.cancel(requestId) + await window.goodbuddy.agent.cancel(requestId); } - activeRuns.current.clear() - setActiveConversationIds(new Set()) + activeRuns.current.clear(); + setActiveConversationIds(new Set()); for (const attachments of attachmentsRef.current.values()) { for (const attachment of attachments) { - await window.goodbuddy.context.remove(attachment.id) + await window.goodbuddy.context.remove(attachment.id); } } - attachmentsRef.current.clear() - setAttachmentsByConversation({}) + attachmentsRef.current.clear(); + setAttachmentsByConversation({}); for (const library of knowledgeSnapshot.libraries) { - await window.goodbuddy.knowledge.deleteLibrary(library.id) + await window.goodbuddy.knowledge.deleteLibrary(library.id); } - await window.goodbuddy.app.clearLocalData() + await window.goodbuddy.app.clearLocalData(); const conversation = createConversation( activeProjectId || undefined, runtimeSettings - ? getProjectDefaultRuntimeSelection( - activeProject, - runtimeSettings - ) + ? getProjectDefaultRuntimeSelection(activeProject, runtimeSettings) : undefined, - t('conversation.greeting') - ) - conversationsRef.current = [conversation] - persistedLocalConversationsRef.current.clear() - setConversations([conversation]) - setActiveId(conversation.id) - legacyActivityHistoryMayBeIncompleteRef.current = false - setLegacyActivityHistoryMayBeIncomplete(false) - setActivityRecords([]) - setAssistantTasks([]) - setTokenUsage(emptyTokenUsage) - setAssistantArtifacts([]) - setAssistantMemories([]) - setAssistantSchedules([]) - setAssistantHeartbeats([]) - setHeartbeatEntries([]) - setHeartbeatRuns([]) - setHeartbeatMemories([]) + t("conversation.greeting"), + ); + conversationsRef.current = [conversation]; + persistedLocalConversationsRef.current.clear(); + setConversations([conversation]); + setActiveId(conversation.id); + legacyActivityHistoryMayBeIncompleteRef.current = false; + setLegacyActivityHistoryMayBeIncomplete(false); + setActivityRecords([]); + setAssistantTasks([]); + setTokenUsage(emptyTokenUsage); + setAssistantArtifacts([]); + setAssistantMemories([]); + setAssistantSchedules([]); + setAssistantHeartbeats([]); + setHeartbeatEntries([]); + setHeartbeatRuns([]); + setHeartbeatMemories([]); setKnowledgeSnapshot({ libraries: [], sources: [], documents: [], graphNodes: [], graphRelations: [], - evidence: [] - }) - updateAttachments([]) - setInput('') - setView('chat') + evidence: [], + }); + updateAttachments([]); + setInput(""); + setView("chat"); notify({ - tone: 'success', - message: t('notices.localDataCleared') - }) + tone: "success", + message: t("notices.localDataCleared"), + }); } finally { - conversationPersistencePausedRef.current = false - persistLocalConversationChanges() + conversationPersistencePausedRef.current = false; + persistLocalConversationChanges(); } - } + }; const isRunning = activeConversation?.messages.some( - (message) => message.state === 'streaming' - ) ?? false + (message) => message.state === "streaming", + ) ?? false; const activeConversationQueueItems = useMemo( () => - conversationQueueItems.filter( - (item) => item.conversationId === activeId - ), - [activeId, conversationQueueItems] - ) + conversationQueueItems.filter((item) => item.conversationId === activeId), + [activeId, conversationQueueItems], + ); const conversationExecutionRunning = isRunning || assistantTasks.some( (task) => task.conversationId === activeId && - (task.status === 'running' || - task.status === 'waiting_approval') - ) + (task.status === "running" || task.status === "waiting_approval"), + ); const handleConversationQueueError = useCallback( (message: string): void => { notify({ - tone: 'error', - message - }) + tone: "error", + message, + }); }, - [notify] - ) + [notify], + ); const interruptConversationQueueItem = useCallback( (itemId: string) => window.goodbuddy.conversationQueue.interruptAndRun(itemId), - [] - ) + [], + ); const removeConversationQueueItem = useCallback( - (itemId: string) => - window.goodbuddy.conversationQueue.remove(itemId), - [] - ) + (itemId: string) => window.goodbuddy.conversationQueue.remove(itemId), + [], + ); const runtimeAgentControlAvailable = - activeRuntimeSelection?.provider === 'opencode' && - runtimeAgentOptions.length > 1 + activeRuntimeSelection?.provider === "opencode" && + runtimeAgentOptions.length > 1; const runtimePresetControlAvailable = - activeRuntimeSelection?.provider === 'continue' && - runtimePresetOptions.length > 1 + activeRuntimeSelection?.provider === "continue" && + runtimePresetOptions.length > 1; const runtimeActionControlAvailable = - (activeRuntimeSelection?.provider === 'opencode' || - activeRuntimeSelection?.provider === 'continue') && - runtimeActionOptions.length > 1 + (activeRuntimeSelection?.provider === "opencode" || + activeRuntimeSelection?.provider === "continue") && + runtimeActionOptions.length > 1; const runtimeControlsAvailable = runtimeAgentControlAvailable || runtimePresetControlAvailable || - runtimeActionControlAvailable + runtimeActionControlAvailable; const runtimeControlsProvider = runtimeControlsAvailable - ? activeRuntimeSelection?.provider === 'opencode' - ? 'OpenCode' - : activeRuntimeSelection?.provider === 'continue' - ? 'Continue' + ? activeRuntimeSelection?.provider === "opencode" + ? "OpenCode" + : activeRuntimeSelection?.provider === "continue" + ? "Continue" : undefined - : undefined + : undefined; const runtimeControlsLabel = runtimeControlsProvider - ? t('composer.runtimeControls.groupLabel', { - runtime: runtimeControlsProvider + ? t("composer.runtimeControls.groupLabel", { + runtime: runtimeControlsProvider, }) - : '' + : ""; const runtimeContextCompactAvailable = - (activeRuntimeSelection?.provider === 'opencode' || - activeRuntimeSelection?.provider === 'continue') && - runtimeNativeSnapshot?.context.manualCompact === true + (activeRuntimeSelection?.provider === "opencode" || + activeRuntimeSelection?.provider === "continue") && + runtimeNativeSnapshot?.context.manualCompact === true; const composerContextMetrics = useMemo(() => { if ( @@ -8411,146 +7957,136 @@ function App(): React.JSX.Element { !activeRuntimeSelection || activeConversation.remote ) { - return undefined + return undefined; } - const resolvedRuntimeSelection = - resolveContextMetricsRuntimeSelection( - activeRuntimeSelection, - runtimeSettings - ) + const resolvedRuntimeSelection = resolveContextMetricsRuntimeSelection( + activeRuntimeSelection, + runtimeSettings, + ); const activeModelProfile = - 'profileId' in resolvedRuntimeSelection && + "profileId" in resolvedRuntimeSelection && resolvedRuntimeSelection.profileId ? runtimeSettings.modelProfiles.find( - (candidate) => - candidate.id === resolvedRuntimeSelection.profileId + (candidate) => candidate.id === resolvedRuntimeSelection.profileId, ) - : undefined - if (activeModelProfile?.protocol === 'openai-images-generations') { - return undefined + : undefined; + if (activeModelProfile?.protocol === "openai-images-generations") { + return undefined; } - const latest = activeConversation.contextMetrics + const latest = activeConversation.contextMetrics; const applicableLatest = latest?.runtimeSelectionKey === agentRuntimeSelectionKey(resolvedRuntimeSelection) ? latest - : undefined + : undefined; if (!applicableLatest) { - return undefined + return undefined; } const compressionSettings = - runtimeSettings.contextCompression ?? - defaultContextCompressionSettings - const contextTokens = applicableLatest.contextTokens - const contextWindowTokens = - activeModelProfile?.contextWindowTokens + runtimeSettings.contextCompression ?? defaultContextCompressionSettings; + const contextTokens = applicableLatest.contextTokens; + const contextWindowTokens = activeModelProfile?.contextWindowTokens; const effectiveTriggerTokens = getEffectiveContextTriggerTokens({ triggerTokens: compressionSettings.triggerTokens, - contextWindowTokens - }) - const denominatorTokens = contextWindowTokens + contextWindowTokens, + }); + const denominatorTokens = contextWindowTokens; const percentage = denominatorTokens === undefined ? undefined - : Math.round( - (contextTokens / denominatorTokens) * 100 - ) + : Math.round((contextTokens / denominatorTokens) * 100); return { contextTokens, effectiveTriggerTokens, contextWindowTokens, compressionEnabled: - resolvedRuntimeSelection.provider === 'model' && + resolvedRuntimeSelection.provider === "model" && compressionSettings.enabled, source: applicableLatest.source, basis: applicableLatest.basis ?? - (applicableLatest.source === 'estimated' && + (applicableLatest.source === "estimated" && activeConversation.contextCompressionState - ? 'conversation' - : 'model-call'), + ? "conversation" + : "model-call"), denominatorTokens, - percentage - } - }, [ - activeConversation, - activeRuntimeSelection, - runtimeSettings - ]) + percentage, + }; + }, [activeConversation, activeRuntimeSelection, runtimeSettings]); - const mainSidebarOpen = narrowWindow && sidebarOpen - const canResizePrimarySidebar = sidebarOpen && !narrowWindow - const primarySidebarWidthLimits = - getPrimarySidebarWidthLimits(window.innerWidth) - const backgroundIsolated = mainSidebarOpen + const mainSidebarOpen = narrowWindow && sidebarOpen; + const canResizePrimarySidebar = sidebarOpen && !narrowWindow; + const primarySidebarWidthLimits = getPrimarySidebarWidthLimits( + window.innerWidth, + ); + const backgroundIsolated = mainSidebarOpen; const runtimeState = runtimeSwitching || !runtime || runtimeStatusKey !== activeRuntimeSelectionKey - ? 'connecting' + ? "connecting" : runtime.available - ? 'ready' - : 'unavailable' - const runtimeDetailId = 'topbar-runtime-detail' + ? "ready" + : "unavailable"; + const runtimeDetailId = "topbar-runtime-detail"; const runtimeDetail = - runtimeState === 'connecting' - ? t('runtime.connecting') + runtimeState === "connecting" + ? t("runtime.connecting") : runtime?.detail || - (runtimeState === 'unavailable' - ? t('runtime.unavailable') - : t('runtime.state.ready')) - const composerContextErrorId = 'composer-context-error' + (runtimeState === "unavailable" + ? t("runtime.unavailable") + : t("runtime.state.ready")); + const composerContextErrorId = "composer-context-error"; const runBrowserCommand = async ( command: ( browserApi: NonNullable, - conversationId: string - ) => Promise + conversationId: string, + ) => Promise, ): Promise => { if (!activeId) { - return + return; } - const browserApi = window.goodbuddy.browser + const browserApi = window.goodbuddy.browser; if (!browserApi) { notify({ - tone: 'error', - message: t('notices.browserControlUnavailable') - }) - return + tone: "error", + message: t("notices.browserControlUnavailable"), + }); + return; } - await command(browserApi, activeId) - } + await command(browserApi, activeId); + }; return (