shennian 0.3.86 → 0.4.0

package/dist/src/agents/workbuddy.js

@@ -1,444 +1,3 @@
-// @arch docs/architecture/cli/agent-adapters.md
-// @test src/__tests__/workbuddy-adapter.test.ts
-import { spawnSync } from 'node:child_process';
-import { createInterface } from 'node:readline';
-import { randomUUID } from 'node:crypto';
-import path from 'node:path';
-import { AgentAdapter, registerAgent, } from './adapter.js';
-import { resolveBuiltinCommand, spawnAgentCommand } from './command-spec.js';
-import { buildAgentProcessEnv, buildManagedAgentProcessEnv } from '../agent-env.js';
-import { buildPlatformInstructions, ensurePlatformInstructionsFile, } from './platform-instructions.js';
-import { resolveWindowsSystemRoot } from '../windows-shell.js';
-import { resolveManagedAgentAdapterEnforcement, } from '../security/managed-agent-policy.js';
-import { registerManagedAgentRuntimeSecurity } from '../security/managed-agent-runtime.js';
-import { resolveManagedRoomIpcPaths } from '../daemon-ipc/paths.js';
-import { compileWorkBuddyManagedPermissions, createWorkBuddyManagedConfigFiles, disposeWorkBuddyManagedConfigFiles, WORKBUDDY_MANAGED_RUNTIME_SECURITY, } from './workbuddy-managed-permissions.js';
-registerManagedAgentRuntimeSecurity('workbuddy', WORKBUDDY_MANAGED_RUNTIME_SECURITY);
-const WORKBUDDY_EFFORTS = new Set(['minimal', 'low', 'medium', 'high', 'xhigh', 'max']);
-export function normalizeWorkBuddyModelId(modelId) {
-    const trimmed = modelId?.trim();
-    return trimmed && trimmed !== 'auto' ? trimmed : undefined;
-}
-export function normalizeWorkBuddyReasoningEffort(value) {
-    const trimmed = value?.trim();
-    if (!trimmed)
-        return undefined;
-    if (WORKBUDDY_EFFORTS.has(trimmed))
-        return trimmed;
-    throw new Error(`Unsupported WorkBuddy reasoning effort "${trimmed}". Supported values: minimal, low, medium, high, xhigh, max.`);
-}
-export function workBuddySystemPromptArgs(systemPrompt, platform, promptFile) {
-    if (!systemPrompt)
-        return [];
-    if (platform === 'win32') {
-        if (!promptFile)
-            throw new Error('WorkBuddy system prompt file is required on Windows.');
-        return ['--system-prompt-file', promptFile];
-    }
-    return ['--append-system-prompt', systemPrompt];
-}
-export function workBuddyRoomActivationText(text) {
-    const blocks = text.split(/\n\n(?=<[^>\n]+>\n)/);
-    const current = blocks[blocks.length - 1] ?? text;
-    return current
-        .replace(/^\[群消息\][ \t]+<[^>\n]+>\n/, '')
-        .replace(/^\[群消息\]\n?/, '')
-        .replace(/^<[^>\n]+>\n/, '')
-        .trim();
-}
-export function terminateWorkBuddyProcess(proc, platform = process.platform, terminateWindowsTree = terminateWindowsProcessTree) {
-    if (platform === 'win32' && proc.pid && terminateWindowsTree(proc.pid))
-        return;
-    proc.kill('SIGTERM');
-}
-function terminateWindowsProcessTree(pid) {
-    const taskkill = path.win32.join(resolveWindowsSystemRoot(), 'System32', 'taskkill.exe');
-    const result = spawnSync(taskkill, ['/pid', String(pid), '/T', '/F'], {
-        stdio: 'ignore',
-        windowsHide: true,
-        timeout: 5_000,
-    });
-    return result.status === 0;
-}
-export class WorkBuddyAdapter extends AgentAdapter {
-    type = 'workbuddy';
-    process = null;
-    agentSessionId = null;
-    workDir = null;
-    seq = 0;
-    runId = '';
-    terminalState = 'open';
-    hasEmittedText = false;
-    runPhase = null;
-    persistNativeSession = true;
-    externalChannel = null;
-    shennianSessionId = null;
-    extraEnv = {};
-    managedAgentPolicy = null;
-    managedConfigDir = null;
-    managedProcessEnv = null;
-    configure(options) {
-        this.shennianSessionId = options.sessionId ?? null;
-        this.externalChannel = options.externalChannel ?? null;
-        this.extraEnv = options.env ?? {};
-        if ('managedAgentPolicy' in options) {
-            this.managedAgentPolicy = options.managedAgentPolicy ?? null;
-        }
-        if (this.managedAgentPolicy)
-            this.workDir = this.managedAgentPolicy.adapterPolicy.workDir;
-    }
-    async start(_sessionId, workDir, agentSessionId, options) {
-        this.workDir = this.managedAgentPolicy?.adapterPolicy.workDir ?? workDir;
-        this.seq = 0;
-        this.persistNativeSession = options?.persistNativeSession !== false;
-        if (agentSessionId)
-            this.agentSessionId = agentSessionId;
-    }
-    async send(text, modelId, reasoningEffort, _attachments, agentSystemPrompt, context) {
-        const managedInvocation = context?.source === 'room_activation' ||
-            this.shennianSessionId?.startsWith('room-managed-') === true;
-        if (managedInvocation && !this.managedAgentPolicy) {
-            throw new Error('managed_agent_policy_missing');
-        }
-        await this.killProcess();
-        this.runId = randomUUID();
-        this.resetRunState();
-        const managedRoomActivation = context?.source === 'room_activation';
-        // CodeBuddy's resumed native Session treats the Room's `<speaker>` block
-        // envelope as already-consumed input and exits without calling the tool.
-        // Give it the current addressed message only. Additional history remains
-        // available through the scoped Shennian CLI read command.
-        const promptText = managedRoomActivation ? workBuddyRoomActivationText(text) : text;
-        const args = ['-p', promptText, '--output-format', 'stream-json', '--verbose'];
-        if (!this.persistNativeSession)
-            args.push('--no-session-persistence');
-        const resolvedModelId = normalizeWorkBuddyModelId(modelId);
-        if (resolvedModelId)
-            args.push('--model', resolvedModelId);
-        if (managedRoomActivation)
-            args.push('--max-turns', '8');
-        // Windows launches the npm WorkBuddy shim through cmd.exe, whose command
-        // line is capped at 8191 characters. The managed bootstrap only needs a
-        // native Session id, so avoid attaching the large platform prompt until
-        // that Session exists. Room activations still receive the full prompt.
-        const isManagedBootstrap = this.shennianSessionId?.startsWith('room-managed-') &&
-            !this.agentSessionId &&
-            !managedRoomActivation;
-        const systemPrompt = isManagedBootstrap
-            ? null
-            : buildPlatformInstructions(this.workDir ?? process.cwd(), this.externalChannel, this.shennianSessionId ?? undefined, agentSystemPrompt);
-        if (systemPrompt) {
-            // The npm shim is launched through cmd.exe on Windows. Room context can
-            // exceed its 8191-character command-line limit, so pass the same private
-            // generated instructions by file instead of truncating the activation.
-            const promptFile = process.platform === 'win32'
-                ? ensurePlatformInstructionsFile(this.workDir ?? process.cwd(), this.externalChannel, this.shennianSessionId ?? undefined, agentSystemPrompt)
-                : undefined;
-            args.push(...workBuddySystemPromptArgs(systemPrompt, process.platform, promptFile));
-        }
-        const effort = normalizeWorkBuddyReasoningEffort(reasoningEffort);
-        if (effort)
-            args.push('--effort', effort);
-        if (this.agentSessionId)
-            args.push('--resume', this.agentSessionId);
-        args.push(...(await this.prepareManagedLaunch()));
-        this.spawnAndParse(args);
-    }
-    async resume(agentSessionId) {
-        await this.killProcess();
-        this.agentSessionId = agentSessionId;
-        this.resetRunState();
-    }
-    async stop() {
-        await this.killProcess();
-    }
-    async getStatus() {
-        return {
-            active: this.process != null && this.terminalState === 'open',
-            runId: this.runId || null,
-            runPhase: this.runPhase,
-            canStop: this.process != null && this.terminalState === 'open',
-        };
-    }
-    spawnAndParse(args) {
-        const spec = resolveBuiltinCommand('workbuddy');
-        if (!spec) {
-            void this.disposeManagedConfig();
-            this.emit('error', new Error('WorkBuddy was not found. Install WorkBuddy Desktop or the CodeBuddy CLI, then retry.'));
-            return;
-        }
-        let proc;
-        try {
-            proc = spawnAgentCommand(spec, args, {
-                cwd: this.workDir ?? undefined,
-                stdio: ['ignore', 'pipe', 'pipe'],
-                env: this.buildProcessEnv(),
-            });
-        }
-        catch (error) {
-            void this.disposeManagedConfig();
-            throw error;
-        }
-        this.process = proc;
-        const lines = createInterface({ input: proc.stdout });
-        lines.on('line', (line) => {
-            if (!line.trim())
-                return;
-            try {
-                this.handleStreamEvent(JSON.parse(line));
-            }
-            catch {
-                // WorkBuddy may write non-protocol notices to stdout. Never surface
-                // unstructured output because it may contain local connector details.
-            }
-        });
-        let stderr = '';
-        proc.stderr?.on('data', (chunk) => {
-            stderr = `${stderr}${chunk.toString()}`.slice(-8_000);
-        });
-        proc.on('close', (code) => {
-            if (this.process !== proc)
-                return;
-            this.process = null;
-            this.runPhase = null;
-            void this.disposeManagedConfig();
-            this.handleProcessClose(code, stderr);
-        });
-        proc.on('error', (error) => {
-            if (this.process !== proc)
-                return;
-            this.process = null;
-            this.runPhase = null;
-            void this.disposeManagedConfig();
-            const missing = error.code === 'ENOENT';
-            this.emitErrorIfOpen({
-                state: 'error',
-                message: missing
-                    ? 'WorkBuddy was not found. Install WorkBuddy Desktop or the CodeBuddy CLI, then retry.'
-                    : sanitizeWorkBuddyError(error.message),
-            });
-        });
-    }
-    handleStreamEvent(event) {
-        if (event.type === 'system' && event.subtype === 'init' && event.session_id) {
-            this.agentSessionId = event.session_id;
-            this.runPhase = 'thinking';
-            this.emitEvent({ state: 'start', agentSessionId: event.session_id });
-            return;
-        }
-        if (event.type === 'control_request' && event.request?.subtype === 'can_use_tool') {
-            this.runPhase = 'waiting_approval';
-            this.emitEvent({
-                state: 'approval-pending',
-                name: event.request.tool_name,
-                args: event.request.input,
-                approval: {
-                    title: `WorkBuddy 请求使用 ${event.request.tool_name || '工具'}`,
-                    source: 'workbuddy',
-                    actionHint: '请在 WorkBuddy 或神念中确认后继续。',
-                },
-            });
-            return;
-        }
-        if (event.type === 'assistant') {
-            const blocks = event.message?.content;
-            if (blocks?.length) {
-                for (const block of blocks)
-                    this.handleContentBlock(block);
-            }
-            else if (event.text) {
-                this.emitText(event.text, Boolean(event.thinking));
-            }
-            return;
-        }
-        if (event.type !== 'result')
-            return;
-        if (event.subtype === 'tool_result') {
-            this.runPhase = 'tool_running';
-            this.emitEvent({
-                state: 'tool-result',
-                name: event.name,
-                result: stringifyToolResult(event.content),
-            });
-            return;
-        }
-        if (event.subtype === 'error' || event.error) {
-            this.emitErrorIfOpen({
-                state: 'error',
-                message: sanitizeWorkBuddyError(event.error || event.result || 'WorkBuddy failed.'),
-            });
-            return;
-        }
-        if (!this.hasEmittedText && event.result)
-            this.emitText(event.result, false);
-        if (event.session_id)
-            this.agentSessionId = event.session_id;
-        const usage = event.usage ?? event.message?.usage;
-        this.emitFinalIfOpen({
-            state: 'final',
-            agentSessionId: this.agentSessionId ?? undefined,
-            usage: usage
-                ? { inputTokens: usage.input_tokens ?? 0, outputTokens: usage.output_tokens ?? 0 }
-                : undefined,
-        });
-    }
-    handleContentBlock(block) {
-        if (block.type === 'text' && block.text) {
-            this.emitText(block.text, false);
-        }
-        else if (block.type === 'thinking' && block.thinking) {
-            this.emitText(block.thinking, true);
-        }
-        else if (block.type === 'tool_use') {
-            this.runPhase = 'tool_running';
-            this.emitEvent({ state: 'tool-call', name: block.name, args: block.input });
-        }
-        else if (block.type === 'tool_result') {
-            this.emitEvent({
-                state: 'tool-result',
-                name: block.name,
-                result: stringifyToolResult(block.content),
-            });
-        }
-    }
-    emitText(text, thinking) {
-        this.runPhase = thinking ? 'thinking' : 'streaming_text';
-        const prefix = !thinking && this.hasEmittedText ? '\n\n' : '';
-        this.emitEvent({ state: 'delta', text: prefix + text, thinking: thinking || undefined });
-        if (!thinking)
-            this.hasEmittedText = true;
-    }
-    handleProcessClose(code, stderr) {
-        if (code !== 0 && code !== null) {
-            this.emitErrorIfOpen({
-                state: 'error',
-                message: sanitizeWorkBuddyError(stderr) || `WorkBuddy exited with code ${code}.`,
-            });
-            return;
-        }
-        this.emitFinalIfOpen({ state: 'final', agentSessionId: this.agentSessionId ?? undefined });
-    }
-    emitFinalIfOpen(partial) {
-        if (this.terminalState !== 'open')
-            return;
-        this.terminalState = 'final';
-        this.runPhase = null;
-        this.emitEvent(partial);
-    }
-    emitErrorIfOpen(partial) {
-        if (this.terminalState !== 'open')
-            return;
-        this.terminalState = 'error';
-        this.runPhase = null;
-        this.emitEvent(partial);
-    }
-    emitEvent(partial) {
-        this.emit('agentEvent', { ...partial, runId: this.runId, seq: this.seq++ });
-    }
-    resetRunState() {
-        this.seq = 0;
-        this.terminalState = 'open';
-        this.hasEmittedText = false;
-        this.runPhase = null;
-    }
-    async killProcess() {
-        const proc = this.process;
-        if (!proc) {
-            await this.disposeManagedConfig();
-            return;
-        }
-        this.process = null;
-        this.runPhase = null;
-        terminateWorkBuddyProcess(proc);
-        await new Promise((resolve) => {
-            let settled = false;
-            const finish = () => {
-                if (settled)
-                    return;
-                settled = true;
-                resolve();
-            };
-            proc.once('close', finish);
-            setTimeout(() => {
-                terminateWorkBuddyProcess(proc);
-                finish();
-            }, 3_000).unref?.();
-        });
-        await this.disposeManagedConfig();
-    }
-    async prepareManagedLaunch() {
-        const compiled = this.managedAgentPolicy;
-        if (!compiled) {
-            this.managedProcessEnv = null;
-            return [
-                '--permission-mode',
-                'bypassPermissions',
-                '--setting-sources',
-                'user,project,local',
-                '--tools',
-                'default',
-            ];
-        }
-        const daemonIpc = resolveManagedRoomIpcPaths();
-        const nativeConfig = compileWorkBuddyManagedPermissions(resolveManagedAgentAdapterEnforcement(compiled), compiled.adapterPolicy.workDir, {
-            internalUnixSockets: daemonIpc.transport === 'unix_socket' ? [daemonIpc.endpoint] : [],
-        });
-        const files = await createWorkBuddyManagedConfigFiles(nativeConfig);
-        this.managedConfigDir = files.directory;
-        const env = buildManagedAgentProcessEnv({
-            agentType: 'workbuddy',
-            providerEnv: this.extraEnv,
-            sessionEnv: this.extraEnv,
-            runtimeEnv: { NO_COLOR: '1' },
-        });
-        // These values are generated from the verified policy rather than inherited
-        // from the daemon. They contain no credentials and cannot enable bypass mode.
-        Object.assign(env, nativeConfig.runtimeEnv);
-        delete env.CODEBUDDY_IS_SANDBOX;
-        this.managedProcessEnv = env;
-        return [
-            '--permission-mode',
-            'dontAsk',
-            '--subagent-permission-mode',
-            'dontAsk',
-            '--settings',
-            files.settingsPath,
-            '--setting-sources',
-            'none',
-            '--strict-mcp-config',
-            '--mcp-config',
-            '{}',
-            '--tools',
-            nativeConfig.toolNames.join(','),
-        ];
-    }
-    buildProcessEnv() {
-        return this.managedProcessEnv ?? buildAgentProcessEnv(this.extraEnv);
-    }
-    async disposeManagedConfig() {
-        const directory = this.managedConfigDir;
-        this.managedConfigDir = null;
-        this.managedProcessEnv = null;
-        if (!directory)
-            return;
-        await disposeWorkBuddyManagedConfigFiles(directory).catch(() => { });
-    }
-}
-function stringifyToolResult(value) {
-    if (typeof value === 'string')
-        return value;
-    try {
-        return JSON.stringify(value);
-    }
-    catch {
-        return '[unserializable WorkBuddy tool result]';
-    }
-}
-export function sanitizeWorkBuddyError(value) {
-    return value
-        .replace(/(Authorization\s*:\s*)Bearer\s+[A-Za-z0-9._~+/-]+/gi, '$1Bearer [redacted]')
-        .replace(/(api[-_ ]?key|access[-_ ]?token|refresh[-_ ]?token|connector[-_ ]?token)(\s*[:=]\s*)[^\s,;]+/gi, '$1$2[redacted]')
-        .trim()
-        .slice(0, 2_000);
-}
-registerAgent('workbuddy', () => new WorkBuddyAdapter());
+import{spawnSync as y}from"node:child_process";import{createInterface as w}from"node:readline";import{randomUUID as k}from"node:crypto";import E from"node:path";import{AgentAdapter as S,registerAgent as P}from"./adapter.js";import{resolveBuiltinCommand as I,spawnAgentCommand as x}from"./command-spec.js";import{buildAgentProcessEnv as _,buildManagedAgentProcessEnv as B}from"../agent-env.js";import{buildPlatformInstructions as C,ensurePlatformInstructionsFile as A}from"./platform-instructions.js";import{resolveWindowsSystemRoot as v}from"../windows-shell.js";import{resolveManagedAgentAdapterEnforcement as W}from"../security/managed-agent-policy.js";import{registerManagedAgentRuntimeSecurity as D}from"../security/managed-agent-runtime.js";import{resolveManagedRoomIpcPaths as T}from"../daemon-ipc/paths.js";import{compileWorkBuddyManagedPermissions as M,createWorkBuddyManagedConfigFiles as O,disposeWorkBuddyManagedConfigFiles as R,WORKBUDDY_MANAGED_RUNTIME_SECURITY as N}from"./workbuddy-managed-permissions.js";D("workbuddy",N);const b=new Set(["minimal","low","medium","high","xhigh","max"]);function q(n){const e=n?.trim();return e&&e!=="auto"?e:void 0}function F(n){const e=n?.trim();if(e){if(b.has(e))return e;throw new Error(`Unsupported WorkBuddy reasoning effort "${e}". Supported values: minimal, low, medium, high, xhigh, max.`)}}function U(n,e,t){if(!n)return[];if(e==="win32"){if(!t)throw new Error("WorkBuddy system prompt file is required on Windows.");return["--system-prompt-file",t]}return["--append-system-prompt",n]}function $(n){const e=n.split(/\n\n(?=<[^>\n]+>\n)/);return(e[e.length-1]??n).replace(/^\[群消息\][ \t]+<[^>\n]+>\n/,"").replace(/^\[群消息\]\n?/,"").replace(/^<[^>\n]+>\n/,"").trim()}function c(n,e=process.platform,t=z){e==="win32"&&n.pid&&t(n.pid)||n.kill("SIGTERM")}function z(n){const e=E.win32.join(v(),"System32","taskkill.exe");return y(e,["/pid",String(n),"/T","/F"],{stdio:"ignore",windowsHide:!0,timeout:5e3}).status===0}class j extends S{type="workbuddy";process=null;agentSessionId=null;workDir=null;seq=0;runId="";terminalState="open";hasEmittedText=!1;runPhase=null;persistNativeSession=!0;externalChannel=null;shennianSessionId=null;extraEnv={};managedAgentPolicy=null;managedConfigDir=null;managedProcessEnv=null;configure(e){this.shennianSessionId=e.sessionId??null,this.externalChannel=e.externalChannel??null,this.extraEnv=e.env??{},"managedAgentPolicy"in e&&(this.managedAgentPolicy=e.managedAgentPolicy??null),this.managedAgentPolicy&&(this.workDir=this.managedAgentPolicy.adapterPolicy.workDir)}async start(e,t,s,r){this.workDir=this.managedAgentPolicy?.adapterPolicy.workDir??t,this.seq=0,this.persistNativeSession=r?.persistNativeSession!==!1,s&&(this.agentSessionId=s)}async send(e,t,s,r,o,i){if((i?.source==="room_activation"||this.shennianSessionId?.startsWith("room-managed-")===!0)&&!this.managedAgentPolicy)throw new Error("managed_agent_policy_missing");await this.killProcess(),this.runId=k(),this.resetRunState();const d=i?.source==="room_activation",a=["-p",d?$(e):e,"--output-format","stream-json","--verbose"];this.persistNativeSession||a.push("--no-session-persistence");const h=q(t);h&&a.push("--model",h),d&&a.push("--max-turns","8");const m=this.shennianSessionId?.startsWith("room-managed-")&&!this.agentSessionId&&!d?null:C(this.workDir??process.cwd(),this.externalChannel,this.shennianSessionId??void 0,o);if(m){const f=process.platform==="win32"?A(this.workDir??process.cwd(),this.externalChannel,this.shennianSessionId??void 0,o):void 0;a.push(...U(m,process.platform,f))}const p=F(s);p&&a.push("--effort",p),this.agentSessionId&&a.push("--resume",this.agentSessionId),a.push(...await this.prepareManagedLaunch()),this.spawnAndParse(a)}async resume(e){await this.killProcess(),this.agentSessionId=e,this.resetRunState()}async stop(){await this.killProcess()}async getStatus(){return{active:this.process!=null&&this.terminalState==="open",runId:this.runId||null,runPhase:this.runPhase,canStop:this.process!=null&&this.terminalState==="open"}}spawnAndParse(e){const t=I("workbuddy");if(!t){this.disposeManagedConfig(),this.emit("error",new Error("WorkBuddy was not found. Install WorkBuddy Desktop or the CodeBuddy CLI, then retry."));return}let s;try{s=x(t,e,{cwd:this.workDir??void 0,stdio:["ignore","pipe","pipe"],env:this.buildProcessEnv()})}catch(i){throw this.disposeManagedConfig(),i}this.process=s,w({input:s.stdout}).on("line",i=>{if(i.trim())try{this.handleStreamEvent(JSON.parse(i))}catch{}});let o="";s.stderr?.on("data",i=>{o=`${o}${i.toString()}`.slice(-8e3)}),s.on("close",i=>{this.process===s&&(this.process=null,this.runPhase=null,this.disposeManagedConfig(),this.handleProcessClose(i,o))}),s.on("error",i=>{if(this.process!==s)return;this.process=null,this.runPhase=null,this.disposeManagedConfig();const l=i.code==="ENOENT";this.emitErrorIfOpen({state:"error",message:l?"WorkBuddy was not found. Install WorkBuddy Desktop or the CodeBuddy CLI, then retry.":u(i.message)})})}handleStreamEvent(e){if(e.type==="system"&&e.subtype==="init"&&e.session_id){this.agentSessionId=e.session_id,this.runPhase="thinking",this.emitEvent({state:"start",agentSessionId:e.session_id});return}if(e.type==="control_request"&&e.request?.subtype==="can_use_tool"){this.runPhase="waiting_approval",this.emitEvent({state:"approval-pending",name:e.request.tool_name,args:e.request.input,approval:{title:`WorkBuddy \u8BF7\u6C42\u4F7F\u7528 ${e.request.tool_name||"\u5DE5\u5177"}`,source:"workbuddy",actionHint:"\u8BF7\u5728 WorkBuddy \u6216\u795E\u5FF5\u4E2D\u786E\u8BA4\u540E\u7EE7\u7EED\u3002"}});return}if(e.type==="assistant"){const s=e.message?.content;if(s?.length)for(const r of s)this.handleContentBlock(r);else e.text&&this.emitText(e.text,!!e.thinking);return}if(e.type!=="result")return;if(e.subtype==="tool_result"){this.runPhase="tool_running",this.emitEvent({state:"tool-result",name:e.name,result:g(e.content)});return}if(e.subtype==="error"||e.error){this.emitErrorIfOpen({state:"error",message:u(e.error||e.result||"WorkBuddy failed.")});return}!this.hasEmittedText&&e.result&&this.emitText(e.result,!1),e.session_id&&(this.agentSessionId=e.session_id);const t=e.usage??e.message?.usage;this.emitFinalIfOpen({state:"final",agentSessionId:this.agentSessionId??void 0,usage:t?{inputTokens:t.input_tokens??0,outputTokens:t.output_tokens??0}:void 0})}handleContentBlock(e){e.type==="text"&&e.text?this.emitText(e.text,!1):e.type==="thinking"&&e.thinking?this.emitText(e.thinking,!0):e.type==="tool_use"?(this.runPhase="tool_running",this.emitEvent({state:"tool-call",name:e.name,args:e.input})):e.type==="tool_result"&&this.emitEvent({state:"tool-result",name:e.name,result:g(e.content)})}emitText(e,t){this.runPhase=t?"thinking":"streaming_text";const s=!t&&this.hasEmittedText?`
+
+`:"";this.emitEvent({state:"delta",text:s+e,thinking:t||void 0}),t||(this.hasEmittedText=!0)}handleProcessClose(e,t){if(e!==0&&e!==null){this.emitErrorIfOpen({state:"error",message:u(t)||`WorkBuddy exited with code ${e}.`});return}this.emitFinalIfOpen({state:"final",agentSessionId:this.agentSessionId??void 0})}emitFinalIfOpen(e){this.terminalState==="open"&&(this.terminalState="final",this.runPhase=null,this.emitEvent(e))}emitErrorIfOpen(e){this.terminalState==="open"&&(this.terminalState="error",this.runPhase=null,this.emitEvent(e))}emitEvent(e){this.emit("agentEvent",{...e,runId:this.runId,seq:this.seq++})}resetRunState(){this.seq=0,this.terminalState="open",this.hasEmittedText=!1,this.runPhase=null}async killProcess(){const e=this.process;if(!e){await this.disposeManagedConfig();return}this.process=null,this.runPhase=null,c(e),await new Promise(t=>{let s=!1;const r=()=>{s||(s=!0,t())};e.once("close",r),setTimeout(()=>{c(e),r()},3e3).unref?.()}),await this.disposeManagedConfig()}async prepareManagedLaunch(){const e=this.managedAgentPolicy;if(!e)return this.managedProcessEnv=null,["--permission-mode","bypassPermissions","--setting-sources","user,project,local","--tools","default"];const t=T(),s=M(W(e),e.adapterPolicy.workDir,{internalUnixSockets:t.transport==="unix_socket"?[t.endpoint]:[]}),r=await O(s);this.managedConfigDir=r.directory;const o=B({agentType:"workbuddy",providerEnv:this.extraEnv,sessionEnv:this.extraEnv,runtimeEnv:{NO_COLOR:"1"}});return Object.assign(o,s.runtimeEnv),delete o.CODEBUDDY_IS_SANDBOX,this.managedProcessEnv=o,["--permission-mode","dontAsk","--subagent-permission-mode","dontAsk","--settings",r.settingsPath,"--setting-sources","none","--strict-mcp-config","--mcp-config","{}","--tools",s.toolNames.join(",")]}buildProcessEnv(){return this.managedProcessEnv??_(this.extraEnv)}async disposeManagedConfig(){const e=this.managedConfigDir;this.managedConfigDir=null,this.managedProcessEnv=null,e&&await R(e).catch(()=>{})}}function g(n){if(typeof n=="string")return n;try{return JSON.stringify(n)}catch{return"[unserializable WorkBuddy tool result]"}}function u(n){return n.replace(/(Authorization\s*:\s*)Bearer\s+[A-Za-z0-9._~+/-]+/gi,"$1Bearer [redacted]").replace(/(api[-_ ]?key|access[-_ ]?token|refresh[-_ ]?token|connector[-_ ]?token)(\s*[:=]\s*)[^\s,;]+/gi,"$1$2[redacted]").trim().slice(0,2e3)}P("workbuddy",()=>new j);export{j as WorkBuddyAdapter,q as normalizeWorkBuddyModelId,F as normalizeWorkBuddyReasoningEffort,u as sanitizeWorkBuddyError,c as terminateWorkBuddyProcess,$ as workBuddyRoomActivationText,U as workBuddySystemPromptArgs};

package/dist/src/commands/logout.d.ts

@@ -0,0 +1,3 @@
+import type { Command } from 'commander';
+export declare function logout(): Promise<void>;
+export declare function registerLogoutCommand(program: Command): void;

package/dist/src/commands/logout.js

@@ -0,0 +1 @@
+import{execFile as c}from"node:child_process";import s from"node:fs";import{fileURLToPath as l}from"node:url";import{promisify as u}from"node:util";import{getConfigPath as f,saveConfig as d}from"../config/index.js";const g=u(c);async function m(){const r=l(new URL("../../bin/shennian.js",import.meta.url)),n=async t=>{const{stdout:a}=await g(process.execPath,[r,t,"--json"],{timeout:3e4,windowsHide:!0});return JSON.parse(a)},e=await n("status");if(e.error||e.running!==!1||e.remoteAccessDisabled!==!0){const t=await n("stop");if(t.error||t.running!==!1||t.remoteAccessDisabled!==!0)throw new Error("Background service did not stop completely")}const i=f();if(!s.existsSync(i))return;const o=JSON.parse(s.readFileSync(i,"utf8").replace(/^\uFEFF/,""));if(!o||typeof o!="object"||Array.isArray(o))throw new Error("Invalid configuration");delete o.machineId,delete o.machineToken,delete o.accountNamespaceId,delete o.accessToken,delete o.refreshToken,d(o)}function k(r){r.command("logout").description("Stop the background service and clear local Shennian login credentials").option("--json","Print machine-readable result JSON").action(async n=>{try{await m();const e="Logged out of Shennian. Run shennian pair to sign in again.";console.log(n.json?JSON.stringify({ok:!0,message:e}):e)}catch{const e="Logout incomplete. Check shennian status and retry shennian logout.";n.json?console.log(JSON.stringify({ok:!1,error:"logout_failed",message:e,nextAction:"retry_logout"})):console.error(e),process.exitCode=1}})}export{m as logout,k as registerLogoutCommand};

package/dist/src/daemon-ipc/paths.js

@@ -1 +1 @@
-import s from"node:crypto";import a from"node:os";import t from"node:path";import{getShennianDir as i}from"../config/index.js";const f="SHENNIAN_MANAGED_ROOM_IPC_ROOT";function c(e=i(),n=process.platform){const o=t.join(e,"runtime"),r=t.join(o,"daemon-ipc.secret");if(n==="win32"){const p=s.createHash("sha256").update(t.resolve(e).toLowerCase()).digest("hex").slice(0,20);return{runtimeDirectory:o,secretFile:r,endpoint:`\\\\.\\pipe\\shennian-daemon-${p}`,transport:"windows_named_pipe"}}return{runtimeDirectory:o,secretFile:r,endpoint:t.join(o,"daemon.sock"),transport:"unix_socket"}}function m(e=i(),n=a.tmpdir()){const o=s.createHash("sha256").update(t.resolve(e)).digest("hex").slice(0,20);return t.join(n,`sn-mr-${o}`)}function l(e=m(),n=process.platform){return c(e,n)}export{f as MANAGED_ROOM_IPC_ROOT_ENV,m as getManagedRoomIpcRoot,c as resolveDaemonIpcPaths,l as resolveManagedRoomIpcPaths};
+import s from"node:crypto";import p from"node:fs";import c from"node:os";import n from"node:path";import{getShennianDir as a}from"../config/index.js";const R="SHENNIAN_MANAGED_ROOM_IPC_ROOT";function m(e=a(),o=process.platform){const t=n.join(e,"runtime"),r=n.join(t,"daemon-ipc.secret");if(o==="win32"){const i=s.createHash("sha256").update(n.resolve(e).toLowerCase()).digest("hex").slice(0,20);return{runtimeDirectory:t,secretFile:r,endpoint:`\\\\.\\pipe\\shennian-daemon-${i}`,transport:"windows_named_pipe"}}return{runtimeDirectory:t,secretFile:r,endpoint:n.join(t,"daemon.sock"),transport:"unix_socket"}}function d(e=a(),o=c.tmpdir()){const t=s.createHash("sha256").update(n.resolve(e)).digest("hex").slice(0,20);let r=o;try{r=f(o)}catch{}return n.join(r,`sn-mr-${t}`)}function x(e=d(),o=process.platform){return m(e,o)}function f(e){return p.realpathSync.native(e)}export{R as MANAGED_ROOM_IPC_ROOT_ENV,d as getManagedRoomIpcRoot,m as resolveDaemonIpcPaths,x as resolveManagedRoomIpcPaths};