shennian 0.3.86 → 0.4.0

package/dist/src/agents/cursor.js

@@ -1,314 +1 @@
-// @arch docs/architecture/cli/agent-adapters.md
-// @test src/__tests__/cursor-adapter.test.ts
-import { createInterface } from 'node:readline';
-import { randomUUID } from 'node:crypto';
-import { AgentAdapter, registerAgent } from './adapter.js';
-import { resolveBuiltinCommand, spawnResolvedCommand } from './command-spec.js';
-import { buildAgentProcessEnv, buildManagedAgentProcessEnv } from '../agent-env.js';
-import { buildInlineChannelPreamble } from './platform-instructions.js';
-import { resolveManagedAgentAdapterEnforcement, } from '../security/managed-agent-policy.js';
-import { registerManagedAgentRuntimeSecurity } from '../security/managed-agent-runtime.js';
-import { compileCursorManagedPermissions, createCursorManagedConfigDirectory, CURSOR_MANAGED_RUNTIME_SECURITY, disposeCursorManagedConfigDirectory, } from './cursor-managed-permissions.js';
-registerManagedAgentRuntimeSecurity('cursor', CURSOR_MANAGED_RUNTIME_SECURITY);
-export class CursorAdapter extends AgentAdapter {
-    type = 'cursor';
-    process = null;
-    chatId = null;
-    workDir = null;
-    seq = 0;
-    runId = '';
-    emittedText = '';
-    externalChannel = null;
-    shennianSessionId = null;
-    extraEnv = {};
-    managedAgentPolicy = null;
-    managedConfigDir = 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) {
-        this.workDir = this.managedAgentPolicy?.adapterPolicy.workDir ?? workDir;
-        this.seq = 0;
-        if (agentSessionId)
-            this.chatId = agentSessionId;
-    }
-    async send(text, modelId, _reasoningEffort, _attachments, agentSystemPrompt, context) {
-        if (context?.source === 'room_activation' && !this.managedAgentPolicy) {
-            throw new Error('managed_agent_policy_missing');
-        }
-        await this.killProcess();
-        this.runId = randomUUID();
-        this.resetRunState();
-        const outgoing = `${buildInlineChannelPreamble(this.workDir ?? process.cwd(), this.externalChannel, this.shennianSessionId ?? undefined, agentSystemPrompt)}${text}`;
-        const args = ['-p', outgoing, '--output-format', 'stream-json', '--stream-partial-output'];
-        if (modelId) {
-            args.push('--model', modelId);
-        }
-        if (this.chatId) {
-            args.push('--continue');
-        }
-        await this.spawnAndParse(args);
-    }
-    async resume(agentSessionId) {
-        await this.killProcess();
-        this.chatId = agentSessionId;
-        this.runId = randomUUID();
-        this.resetRunState();
-        await this.spawnAndParse([
-            '--resume',
-            agentSessionId,
-            '--output-format',
-            'stream-json',
-            '--stream-partial-output',
-        ]);
-    }
-    async stop() {
-        await this.killProcess();
-    }
-    async spawnAndParse(args) {
-        const spec = resolveBuiltinCommand('cursor');
-        if (!spec) {
-            this.emit('error', new Error('Cursor Agent CLI not found. Expected "agent" or "cursor agent".'));
-            return;
-        }
-        const managedLaunch = await this.prepareManagedLaunch();
-        let proc;
-        try {
-            proc = spawnResolvedCommand(spec, [...args, ...managedLaunch.args], {
-                cwd: this.workDir ?? undefined,
-                stdio: ['ignore', 'pipe', 'pipe'],
-                env: managedLaunch.env,
-            });
-        }
-        catch (error) {
-            await this.disposeManagedConfig(managedLaunch.configDir);
-            throw error;
-        }
-        this.process = proc;
-        this.managedConfigDir = managedLaunch.configDir;
-        const rl = createInterface({ input: proc.stdout });
-        rl.on('line', (line) => {
-            if (!line.trim())
-                return;
-            try {
-                const obj = JSON.parse(line);
-                this.handleStreamEvent(obj);
-            }
-            catch {
-                // 非 JSON 行当作纯文本 delta
-                this.emitEvent({ state: 'delta', text: line });
-            }
-        });
-        let stderrBuf = '';
-        proc.stderr?.on('data', (chunk) => {
-            stderrBuf += chunk.toString();
-        });
-        proc.on('close', (code) => {
-            if (this.process !== proc)
-                return;
-            this.process = null;
-            void this.disposeManagedConfig(managedLaunch.configDir);
-            if (code !== 0 && code !== null) {
-                const msg = stderrBuf.trim() || `cursor agent exited with code ${code}`;
-                this.emitEvent({ state: 'error', message: msg });
-            }
-        });
-        proc.on('error', (err) => {
-            if (this.process !== proc)
-                return;
-            this.process = null;
-            void this.disposeManagedConfig(managedLaunch.configDir);
-            if (err.code === 'ENOENT') {
-                this.emit('error', new Error('Command "cursor" not found. Is Cursor CLI installed?'));
-            }
-            else {
-                this.emit('error', err);
-            }
-        });
-    }
-    async prepareManagedLaunch() {
-        const compiled = this.managedAgentPolicy;
-        if (!compiled) {
-            return {
-                args: ['--trust'],
-                env: buildAgentProcessEnv(this.extraEnv),
-                configDir: null,
-            };
-        }
-        const nativeConfig = compileCursorManagedPermissions(resolveManagedAgentAdapterEnforcement(compiled), compiled.adapterPolicy.workDir);
-        const configDir = await createCursorManagedConfigDirectory(nativeConfig);
-        const env = buildManagedAgentProcessEnv({
-            agentType: 'cursor',
-            providerEnv: this.extraEnv,
-            sessionEnv: this.extraEnv,
-            runtimeEnv: { NO_COLOR: '1' },
-        });
-        // CURSOR_CONFIG_DIR contains only Shennian-generated, non-secret policy.
-        // It is assigned after allowlist filtering so Cursor cannot fall back to
-        // the user's mutable global CLI configuration.
-        env.CURSOR_CONFIG_DIR = configDir;
-        return { args: nativeConfig.args, env, configDir };
-    }
-    async disposeManagedConfig(configDir) {
-        if (!configDir)
-            return;
-        if (this.managedConfigDir === configDir)
-            this.managedConfigDir = null;
-        await disposeCursorManagedConfigDirectory(configDir).catch(() => { });
-    }
-    handleStreamEvent(obj) {
-        if (obj.session_id && !this.chatId) {
-            this.chatId = obj.session_id;
-        }
-        if (obj.chat_id && !this.chatId) {
-            this.chatId = obj.chat_id;
-        }
-        const eventType = obj.type ?? obj.event;
-        switch (eventType) {
-            case 'system': {
-                // init event — session_id already captured above
-                break;
-            }
-            case 'assistant': {
-                // Cursor's --stream-partial-output currently emits duplicate buffered
-                // flushes. Only timestamped assistant events without model_call_id are
-                // safe to treat as live deltas; use result.result as the terminal
-                // canonical text.
-                if (obj.message?.content) {
-                    if (!this.isLiveAssistantDelta(obj))
-                        break;
-                    for (const block of obj.message.content) {
-                        if (block.type === 'text' && block.text) {
-                            this.emitTextDelta(block.text);
-                        }
-                        else if (block.type === 'thinking' && block.thinking) {
-                            this.emitEvent({ state: 'delta', text: block.thinking, thinking: true });
-                        }
-                        else if (block.type === 'tool_use') {
-                            this.emitEvent({ state: 'tool-call', name: block.name, args: block.input });
-                        }
-                    }
-                    break;
-                }
-                // Legacy flat
-                if (obj.delta ?? obj.text) {
-                    this.emitEvent({ state: 'delta', text: (obj.delta ?? obj.text) });
-                }
-                break;
-            }
-            case 'user':
-                break; // echo of user message
-            case 'result': {
-                if (obj.subtype === 'success' || !obj.is_error) {
-                    this.emitFinalResultDelta(obj.result);
-                    const u = obj.usage;
-                    this.emitEvent({
-                        state: 'final',
-                        agentSessionId: this.chatId ?? undefined,
-                        usage: u
-                            ? {
-                                inputTokens: u.inputTokens ?? u.input_tokens ?? 0,
-                                outputTokens: u.outputTokens ?? u.output_tokens ?? 0,
-                            }
-                            : undefined,
-                    });
-                }
-                else {
-                    this.emitEvent({
-                        state: 'error',
-                        message: obj.result ?? obj.error ?? 'unknown error',
-                    });
-                }
-                break;
-            }
-            case 'error': {
-                this.emitEvent({
-                    state: 'error',
-                    message: obj.error ?? 'unknown error',
-                });
-                break;
-            }
-            // Legacy event types
-            case 'text':
-            case 'delta':
-            case 'content': {
-                const text = obj.delta ?? obj.text;
-                if (text)
-                    this.emitEvent({ state: 'delta', text });
-                break;
-            }
-            case 'done': {
-                this.emitEvent({ state: 'final', agentSessionId: this.chatId ?? undefined });
-                break;
-            }
-            default: {
-                if (obj.done) {
-                    this.emitEvent({ state: 'final', agentSessionId: this.chatId ?? undefined });
-                }
-                else if (obj.delta ?? obj.text) {
-                    this.emitEvent({ state: 'delta', text: (obj.delta ?? obj.text) });
-                }
-            }
-        }
-    }
-    emitEvent(partial) {
-        const event = {
-            ...partial,
-            runId: this.runId,
-            seq: this.seq++,
-        };
-        this.emit('agentEvent', event);
-    }
-    async killProcess() {
-        const proc = this.process;
-        if (!proc) {
-            await this.disposeManagedConfig(this.managedConfigDir);
-            return;
-        }
-        this.process = null;
-        proc.kill('SIGTERM');
-        await new Promise((resolve) => {
-            proc.on('close', resolve);
-            setTimeout(() => {
-                proc.kill('SIGKILL');
-                resolve();
-            }, 3000);
-        });
-        await this.disposeManagedConfig(this.managedConfigDir);
-    }
-    resetRunState() {
-        this.seq = 0;
-        this.emittedText = '';
-    }
-    isLiveAssistantDelta(obj) {
-        return obj.timestamp_ms != null && !obj.model_call_id;
-    }
-    emitTextDelta(text) {
-        if (!text)
-            return;
-        this.emittedText += text;
-        this.emitEvent({ state: 'delta', text });
-    }
-    emitFinalResultDelta(result) {
-        if (!result)
-            return;
-        if (!this.emittedText) {
-            this.emitTextDelta(result);
-            return;
-        }
-        if (!result.startsWith(this.emittedText))
-            return;
-        const tail = result.slice(this.emittedText.length);
-        if (tail) {
-            this.emitTextDelta(tail);
-        }
-    }
-}
-registerAgent('cursor', () => new CursorAdapter());
+import{createInterface as l}from"node:readline";import{randomUUID as d}from"node:crypto";import{AgentAdapter as c,registerAgent as h}from"./adapter.js";import{resolveBuiltinCommand as u,spawnResolvedCommand as m}from"./command-spec.js";import{buildAgentProcessEnv as g,buildManagedAgentProcessEnv as f}from"../agent-env.js";import{buildInlineChannelPreamble as p}from"./platform-instructions.js";import{resolveManagedAgentAdapterEnforcement as v}from"../security/managed-agent-policy.js";import{registerManagedAgentRuntimeSecurity as E}from"../security/managed-agent-runtime.js";import{compileCursorManagedPermissions as w,createCursorManagedConfigDirectory as k,CURSOR_MANAGED_RUNTIME_SECURITY as y,disposeCursorManagedConfigDirectory as C}from"./cursor-managed-permissions.js";E("cursor",y);class x extends c{type="cursor";process=null;chatId=null;workDir=null;seq=0;runId="";emittedText="";externalChannel=null;shennianSessionId=null;extraEnv={};managedAgentPolicy=null;managedConfigDir=null;configure(t){this.shennianSessionId=t.sessionId??null,this.externalChannel=t.externalChannel??null,this.extraEnv=t.env??{},"managedAgentPolicy"in t&&(this.managedAgentPolicy=t.managedAgentPolicy??null),this.managedAgentPolicy&&(this.workDir=this.managedAgentPolicy.adapterPolicy.workDir)}async start(t,s,e){this.workDir=this.managedAgentPolicy?.adapterPolicy.workDir??s,this.seq=0,e&&(this.chatId=e)}async send(t,s,e,n,o,r){if(r?.source==="room_activation"&&!this.managedAgentPolicy)throw new Error("managed_agent_policy_missing");await this.killProcess(),this.runId=d(),this.resetRunState();const a=["-p",`${p(this.workDir??process.cwd(),this.externalChannel,this.shennianSessionId??void 0,o)}${t}`,"--output-format","stream-json","--stream-partial-output"];s&&a.push("--model",s),this.chatId&&a.push("--continue"),await this.spawnAndParse(a)}async resume(t){await this.killProcess(),this.chatId=t,this.runId=d(),this.resetRunState(),await this.spawnAndParse(["--resume",t,"--output-format","stream-json","--stream-partial-output"])}async stop(){await this.killProcess()}async spawnAndParse(t){const s=u("cursor");if(!s){this.emit("error",new Error('Cursor Agent CLI not found. Expected "agent" or "cursor agent".'));return}const e=await this.prepareManagedLaunch();let n;try{n=m(s,[...t,...e.args],{cwd:this.workDir??void 0,stdio:["ignore","pipe","pipe"],env:e.env})}catch(i){throw await this.disposeManagedConfig(e.configDir),i}this.process=n,this.managedConfigDir=e.configDir,l({input:n.stdout}).on("line",i=>{if(i.trim())try{const a=JSON.parse(i);this.handleStreamEvent(a)}catch{this.emitEvent({state:"delta",text:i})}});let r="";n.stderr?.on("data",i=>{r+=i.toString()}),n.on("close",i=>{if(this.process===n&&(this.process=null,this.disposeManagedConfig(e.configDir),i!==0&&i!==null)){const a=r.trim()||`cursor agent exited with code ${i}`;this.emitEvent({state:"error",message:a})}}),n.on("error",i=>{this.process===n&&(this.process=null,this.disposeManagedConfig(e.configDir),i.code==="ENOENT"?this.emit("error",new Error('Command "cursor" not found. Is Cursor CLI installed?')):this.emit("error",i))})}async prepareManagedLaunch(){const t=this.managedAgentPolicy;if(!t)return{args:["--trust"],env:g(this.extraEnv),configDir:null};const s=w(v(t),t.adapterPolicy.workDir),e=await k(s),n=f({agentType:"cursor",providerEnv:this.extraEnv,sessionEnv:this.extraEnv,runtimeEnv:{NO_COLOR:"1"}});return n.CURSOR_CONFIG_DIR=e,{args:s.args,env:n,configDir:e}}async disposeManagedConfig(t){t&&(this.managedConfigDir===t&&(this.managedConfigDir=null),await C(t).catch(()=>{}))}handleStreamEvent(t){switch(t.session_id&&!this.chatId&&(this.chatId=t.session_id),t.chat_id&&!this.chatId&&(this.chatId=t.chat_id),t.type??t.event){case"system":break;case"assistant":{if(t.message?.content){if(!this.isLiveAssistantDelta(t))break;for(const e of t.message.content)e.type==="text"&&e.text?this.emitTextDelta(e.text):e.type==="thinking"&&e.thinking?this.emitEvent({state:"delta",text:e.thinking,thinking:!0}):e.type==="tool_use"&&this.emitEvent({state:"tool-call",name:e.name,args:e.input});break}(t.delta??t.text)&&this.emitEvent({state:"delta",text:t.delta??t.text});break}case"user":break;case"result":{if(t.subtype==="success"||!t.is_error){this.emitFinalResultDelta(t.result);const e=t.usage;this.emitEvent({state:"final",agentSessionId:this.chatId??void 0,usage:e?{inputTokens:e.inputTokens??e.input_tokens??0,outputTokens:e.outputTokens??e.output_tokens??0}:void 0})}else this.emitEvent({state:"error",message:t.result??t.error??"unknown error"});break}case"error":{this.emitEvent({state:"error",message:t.error??"unknown error"});break}case"text":case"delta":case"content":{const e=t.delta??t.text;e&&this.emitEvent({state:"delta",text:e});break}case"done":{this.emitEvent({state:"final",agentSessionId:this.chatId??void 0});break}default:t.done?this.emitEvent({state:"final",agentSessionId:this.chatId??void 0}):(t.delta??t.text)&&this.emitEvent({state:"delta",text:t.delta??t.text})}}emitEvent(t){const s={...t,runId:this.runId,seq:this.seq++};this.emit("agentEvent",s)}async killProcess(){const t=this.process;if(!t){await this.disposeManagedConfig(this.managedConfigDir);return}this.process=null,t.kill("SIGTERM"),await new Promise(s=>{t.on("close",s),setTimeout(()=>{t.kill("SIGKILL"),s()},3e3)}),await this.disposeManagedConfig(this.managedConfigDir)}resetRunState(){this.seq=0,this.emittedText=""}isLiveAssistantDelta(t){return t.timestamp_ms!=null&&!t.model_call_id}emitTextDelta(t){t&&(this.emittedText+=t,this.emitEvent({state:"delta",text:t}))}emitFinalResultDelta(t){if(!t)return;if(!this.emittedText){this.emitTextDelta(t);return}if(!t.startsWith(this.emittedText))return;const s=t.slice(this.emittedText.length);s&&this.emitTextDelta(s)}}h("cursor",()=>new x);export{x as CursorAdapter};