shennian 0.3.85 → 0.3.86

package/dist/src/agents/cursor.js

@@ -1 +1,314 @@
-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};
+// @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());