shennian 0.3.86 → 0.4.0

package/dist/src/session/queue.js

@@ -1,12 +1,12 @@
-import p from"node:fs";import{randomUUID as D}from"node:crypto";import{CODEX_DESKTOP_SEND_UNAVAILABLE as O,buildUserMessagePayload as z}from"@shennian/wire";import{isCodexThreadTailInterrupted as j}from"../native-fusion/codex-files.js";import{mergeProjectedSessions as F}from"./projection.js";import{parsePersonalAttachmentReferences as _}from"./attachment-reference.js";import{assertFinishedAttachmentReferences as w,markAttachmentReferences as T}from"./attachment-transfer-store.js";import{resolvePersonalSyncV1Path as C}from"../personal-sync/paths.js";import{getPersonalSyncIdentity as M}from"../personal-sync/epoch.js";import{getDaemonInstallationId as S}from"../personal-sync/identity.js";import{appendMessage as v,getMessageDeliveryStatus as k,recordSession as E,tombstoneMessage as K,upsertMessage as b}from"./store.js";import{requestSessionIndexPublish as x}from"./index-publisher.js";class J extends Error{constructor(e){super(e),this.name="TerminalQueuedChatDispatchError"}}function U(){return{sessions:{}}}function $(){return new Date().toISOString()}function V(s){return`enqueue-send-${s}-${D()}`}function g(){try{const s=JSON.parse(p.readFileSync(P(),"utf-8"));return{sessions:s.sessions&&typeof s.sessions=="object"?s.sessions:{}}}catch(s){if(s.code==="ENOENT"||s instanceof Error&&s.message.startsWith("ENOENT:"))return U();throw s}}function y(s){const e=C();p.mkdirSync(e,{recursive:!0,mode:448});const n=P(),t=`${n}.tmp-${process.pid}-${D()}`;try{p.writeFileSync(t,`${JSON.stringify(s,null,2)}
-`,{encoding:"utf8",mode:384,flag:"wx"});const i=p.openSync(t,"r");if(typeof i=="number")try{p.fsyncSync(i)}finally{p.closeSync(i)}p.renameSync(t,n)}finally{try{p.unlinkSync(t)}catch{}}}function P(){return C("outbound-queue.json")}function A(s){const e=$();return{...M(S()),id:s.queueMessageId||s.clientMessageId||`queue-${D()}`,sessionId:s.sessionId,text:s.text,agentType:s.agentType,sessionMode:s.sessionMode,managerConfig:s.managerConfig,workDir:s.workDir,agentSessionId:s.agentSessionId??null,modelId:s.modelId??null,systemPrompt:s.systemPrompt??null,managerDefaultWorkerAgentType:s.managerDefaultWorkerAgentType??null,managerDefaultWorkerModelId:s.managerDefaultWorkerModelId??null,reasoningEffort:s.reasoningEffort??null,clientMessageId:s.clientMessageId??null,attachments:_(s.attachments),externalChannel:s.externalChannel??null,replyTarget:s.replyTarget??null,origin:s.origin,createdAt:e,updatedAt:e}}function q(s){return{id:s.clientMessageId??s.id,sessionId:s.sessionId,role:"user",ts:Date.parse(s.createdAt),payload:z(s.text,s.attachments)}}class fe{opts;draining=new Set;prepareManagedRoomTurn=null;constructor(e){this.opts=e,setTimeout(()=>this.drainIdleQueues(),0).unref?.()}getSnapshot(e){return{...M(S()),sessionId:e,busy:!!this.opts.getRuntime().sessions.get(e)?.currentRunId,pending:(g().sessions[e]??[]).map(W)}}setManagedRoomTurnPreparer(e){this.prepareManagedRoomTurn=e}async enqueueManagedRoomBatch(e){if(e.length===0)return;const n=this.opts.getRuntime(),t=g(),i=new Set,r=new Map;let a=!1;for(const o of e){const d=o.params;if(!d.sessionId||!d.text||!d.agentType||!d.workDir)throw new Error("sessionId, text, agentType and workDir are required");const l=A({...M(S()),...d});l.origin="room",l.managedRoom=o.context;const f=t.sessions[d.sessionId]??[],u=Q(f,l),I=l.clientMessageId??l.id,h=k(d.sessionId,I);if(!u&&h.state==="not_found")f.push(l),t.sessions[d.sessionId]=f,r.set(`${l.sessionId}\0${I}`,l),i.add(d.sessionId),a=!0;else if(u&&h.state==="not_found")r.set(`${u.sessionId}\0${I}`,u),i.add(d.sessionId);else if(u&&h.state==="queued")i.add(d.sessionId);else if(u){const m=f.filter(c=>c.id!==u.id);m.length?t.sessions[d.sessionId]=m:delete t.sessions[d.sessionId],a=!0}}a&&y(t);for(const o of r.values())v(o.sessionId,q(o),"queued"),E({sessionId:o.sessionId,agentType:o.agentType,sessionMode:o.sessionMode,managerConfig:o.managerConfig,workDir:o.workDir,agentSessionId:o.agentSessionId,modelId:o.modelId});if(i.size!==0){x(n.client);for(const o of i)this.logLifecycle("enqueue.persisted",o,{origin:"room",pendingCount:g().sessions[o]?.length??0}),this.broadcast(o),this.drainNext(o)}}async handleEnqueue(e){const n=this.opts.getRuntime(),t=e.params;if(F(t.sessionListProjection),!t.sessionId||!t.text||!t.agentType||!t.workDir){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId, text, agentType and workDir are required"});return}const i=n.resolvePath(t.workDir),r=_(t.attachments);r?.length&&w(i,r);const a={text:t.text,attachments:r,localized:!1},o=n.sessions.get(t.sessionId);if(t.agentType==="codex"&&t.agentSessionId&&o?.currentRunId?.startsWith("dispatch:")&&j(t.agentSessionId)){n.client.sendRes({type:"res",id:e.id,ok:!1,error:O});return}const d=await this.isSessionBusy(t.sessionId),l=g(),f=l.sessions[t.sessionId]??[];if(!d&&!f.length){await this.opts.dispatchReq({...e,id:V(e.id),method:"chat.send",params:{...t,text:a.text,responseId:e.id,clientMessageId:t.clientMessageId??t.queueMessageId,waitForDispatch:!0,attachments:a.attachments}});return}const u=A({...t,text:a.text,attachments:a.attachments}),I=u.clientMessageId??u.id,h=k(t.sessionId,I);if(h.state!=="not_found"){n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:h.state==="queued",queueMessageId:u.id,...h,queue:this.getSnapshot(t.sessionId)}});return}const m=Q(f,u);m||(l.sessions[t.sessionId]=[...f,u],y(l));const c=m??u;E({sessionId:c.sessionId,agentType:c.agentType,sessionMode:c.sessionMode,managerConfig:c.managerConfig,workDir:c.workDir,agentSessionId:c.agentSessionId,modelId:c.modelId}),v(c.sessionId,q(c),"queued"),T(n.resolvePath(c.workDir),c.attachments??[]),x(n.client),this.logLifecycle("enqueue.persisted",t.sessionId,{origin:t.origin??"app",queueMessageId:c.id,pendingCount:(g().sessions[t.sessionId]??[]).length}),this.broadcast(t.sessionId),n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:!0,queueMessageId:m?.id??u.id,...k(t.sessionId,m?.clientMessageId??m?.id??u.clientMessageId??u.id),queue:this.getSnapshot(t.sessionId),...a.localized?{localizedAttachments:!0}:{}}}),d||this.drainNext(t.sessionId)}async handleGet(e){const n=this.opts.getRuntime(),t=e.params;if(!t.sessionId){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId is required"});return}n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}}),this.getSnapshot(t.sessionId).busy||this.drainNext(t.sessionId)}async handleEdit(e){const n=this.opts.getRuntime(),t=e.params;if(!t.sessionId||!t.queueMessageId||!t.text){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId, queueMessageId and text are required"});return}const i=g(),r=i.sessions[t.sessionId]??[],a=r.findIndex(l=>l.id===t.queueMessageId);if(a<0){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}if(r[a].managedRoom){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const o=_(t.attachments),d=n.resolvePath(r[a].workDir);o?.length&&w(d,o),r[a]={...r[a],text:t.text,attachments:o,updatedAt:$()},i.sessions[t.sessionId]=r,y(i),b(t.sessionId,q(r[a]),"queued"),r[a].attachments?.length&&T(d,r[a].attachments??[]),x(n.client),this.broadcast(t.sessionId),n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}})}async handleDelete(e){const n=this.opts.getRuntime(),t=e.params;if(!t.sessionId||!t.queueMessageId){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId and queueMessageId are required"});return}const i=g(),r=i.sessions[t.sessionId]??[],a=r.find(d=>d.id===t.queueMessageId);if(a?.managedRoom){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const o=r.filter(d=>d.id!==t.queueMessageId);if(o.length===r.length){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}o.length?i.sessions[t.sessionId]=o:delete i.sessions[t.sessionId],y(i),K(t.sessionId,a?.clientMessageId??a?.id??t.queueMessageId),x(n.client),this.broadcast(t.sessionId),n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}})}noteTerminal(e){this.drainNext(e)}purgeSession(e){G(e),this.draining.delete(e)}drainIdleQueues(){const e=g();for(const n of Object.keys(e.sessions))this.drainNext(n)}async drainNext(e){if(this.draining.has(e))return;this.draining.add(e);let n=!1,t=[];try{if(await this.isSessionBusy(e)){this.logLifecycle("drain.deferred",e,{pendingCount:g().sessions[e]?.length??0});return}const i=g(),r=i.sessions[e]??[],a=r.filter(l=>{const f=l.clientMessageId??l.id,u=k(e,f).state;return u==="queued"||u==="not_found"});a.length!==r.length&&(a.length?i.sessions[e]=a:delete i.sessions[e],y(i),this.logLifecycle("drain.pruned-terminal",e,{prunedCount:r.length-a.length,pendingCount:a.length}));const o=H(a);if(o.length===0){this.broadcast(e);return}t=o,this.logLifecycle("dispatch.start",e,{batchCount:o.length,queueMessageId:o[0].id,origin:o[0].origin??"app"});for(const l of o)b(e,q(l),"queued");await this.dispatchQueuedMessages(o);const d=this.removeBatch(e,o);this.logLifecycle("dispatch.accepted",e,{batchCount:o.length,queueMessageId:o[0].id,pendingCount:d}),n=d>0,this.broadcast(e)}catch(i){if(i instanceof J){const r=this.removeBatch(e,t);this.logLifecycle("dispatch.terminal-failure",e,{batchCount:t.length,queueMessageId:t[0]?.id,pendingCount:r,errorCode:L(i.message,"terminal_dispatch_failed")},"error"),n=r>0}else this.logLifecycle("dispatch.transient-failure",e,{errorCode:L(i instanceof Error?i.message:String(i),"transient_dispatch_failed")},"error");this.broadcast(e)}finally{this.draining.delete(e),n&&queueMicrotask(()=>{this.drainNext(e)})}}removeBatch(e,n){if(n.length===0)return g().sessions[e]?.length??0;const t=g(),i=new Set(n.map(a=>a.id)),r=(t.sessions[e]??[]).filter(a=>!i.has(a.id));return r.length?t.sessions[e]=r:delete t.sessions[e],y(t),r.length}logLifecycle(e,n,t,i="info"){const r=`[chat.queue] ${JSON.stringify({event:e,sessionId:n,...t})}`;i==="error"?console.error(r):console.info(r)}async dispatchQueuedMessages(e){const n=e[0],t=e.flatMap(o=>o.managedRoom?[o.managedRoom]:[]),i=t.length>0?await this.prepareManagedRoomTurn?.(t):void 0;if(t.length>0&&!i)throw new Error("managed_room_turn_preparer_unavailable");const r=e.length===1?n.text:X(e),a=Z(e.flatMap(o=>o.attachments??[]));await this.opts.dispatchQueuedReq({type:"req",id:`queue-send-${n.id}-${Date.now()}`,method:"chat.send",params:{...M(S()),sessionId:n.sessionId,text:r,agentType:n.agentType,sessionMode:t.length>0?"standard":n.sessionMode,managerConfig:t.length>0?null:n.managerConfig,workDir:n.workDir,agentSessionId:n.agentSessionId??null,modelId:n.modelId??void 0,systemPrompt:i?.systemPrompt??n.systemPrompt??null,managerDefaultWorkerAgentType:t.length>0?null:n.managerDefaultWorkerAgentType,managerDefaultWorkerModelId:t.length>0?null:n.managerDefaultWorkerModelId,reasoningEffort:n.reasoningEffort??void 0,clientMessageId:n.clientMessageId??n.id,attachments:a,externalChannel:n.externalChannel,replyTarget:n.replyTarget,...i??{},roomActivation:t.length>0||void 0,suppressExternalOwnerEcho:t.length>0||void 0,queuedBatchItems:e.map(W),waitForDispatch:!0}})}async isSessionBusy(e){const n=this.opts.getRuntime().sessions.get(e);if(!n)return!1;if(n.currentRunId)return!0;if(!n.adapter.getStatus)return!1;const t=await n.adapter.getStatus().catch(()=>null);return t?.active?(t.runId&&(n.currentRunId=t.runId,n.currentRunPhase=t.runPhase??n.currentRunPhase),!0):!1}broadcast(e){this.opts.getRuntime().client.sendEvent({type:"event",event:"session.queue.update",payload:{queue:this.getSnapshot(e)}})}}function G(s){const e=g();e.sessions[s]&&(delete e.sessions[s],y(e))}function Q(s,e){const n=R(e);if(n)return s.find(i=>R(i)===n)??null;if(e.origin!=="external")return null;const t=B(e);return s.find(i=>B(i)===t)??null}function W(s){const{origin:e,managedRoom:n,...t}=s;return t}function H(s){const e=s[0];if(!e)return[];const n=N(e),t=[];for(const i of s){if(N(i)!==n)break;t.push(i)}return t}function N(s){return JSON.stringify({agentType:s.agentType,sessionMode:s.sessionMode??null,workDir:s.workDir,agentSessionId:s.agentSessionId??null,modelId:s.modelId??null,systemPrompt:s.systemPrompt??null,reasoningEffort:s.reasoningEffort??null,managerConfig:s.managerConfig??null,managerDefaultWorkerAgentType:s.managerDefaultWorkerAgentType??null,managerDefaultWorkerModelId:s.managerDefaultWorkerModelId??null,externalChannel:s.externalChannel??null,replyTarget:s.replyTarget??null,roomRegistryKey:s.managedRoom?.registryKey??null})}function X(s){return s.every(e=>e.managedRoom)?`[\u7FA4\u6D88\u606F]
-${s.map(e=>Y(e.text)).filter(Boolean).join(`
+import h from"node:fs";import{randomUUID as q}from"node:crypto";import{CODEX_DESKTOP_SEND_UNAVAILABLE as z,buildUserMessagePayload as j}from"@shennian/wire";import{isCodexThreadTailInterrupted as K}from"../native-fusion/codex-files.js";import{mergeProjectedSessions as J}from"./projection.js";import{parsePersonalAttachmentReferences as D}from"./attachment-reference.js";import{assertFinishedAttachmentReferences as w,markAttachmentReferences as T}from"./attachment-transfer-store.js";import{resolvePersonalSyncV1Path as C}from"../personal-sync/paths.js";import{getPersonalSyncIdentity as S}from"../personal-sync/epoch.js";import{getDaemonInstallationId as R}from"../personal-sync/identity.js";import{appendMessage as v,getMessageDeliveryStatus as k,listSessionRecords as U,recordSession as b,tombstoneMessage as E,upsertMessage as P}from"./store.js";import{requestSessionIndexPublish as M}from"./index-publisher.js";class V extends Error{constructor(e){super(e),this.name="TerminalQueuedChatDispatchError"}}function G(){return{sessions:{}}}function $(){return new Date().toISOString()}function H(s){return`enqueue-send-${s}-${q()}`}function g(){try{const s=JSON.parse(h.readFileSync(A(),"utf-8"));return{sessions:s.sessions&&typeof s.sessions=="object"?s.sessions:{}}}catch(s){if(s.code==="ENOENT"||s instanceof Error&&s.message.startsWith("ENOENT:"))return G();throw s}}function p(s){const e=C();h.mkdirSync(e,{recursive:!0,mode:448});const n=A(),t=`${n}.tmp-${process.pid}-${q()}`;try{h.writeFileSync(t,`${JSON.stringify(s,null,2)}
+`,{encoding:"utf8",mode:384,flag:"wx"});const i=h.openSync(t,"r");if(typeof i=="number")try{h.fsyncSync(i)}finally{h.closeSync(i)}h.renameSync(t,n)}finally{try{h.unlinkSync(t)}catch{}}}function A(){return C("outbound-queue.json")}function Q(s){const e=$();return{...S(R()),id:s.queueMessageId||s.clientMessageId||`queue-${q()}`,sessionId:s.sessionId,text:s.text,agentType:s.agentType,sessionMode:s.sessionMode,managerConfig:s.managerConfig,workDir:s.workDir,agentSessionId:s.agentSessionId??null,modelId:s.modelId??null,systemPrompt:s.systemPrompt??null,managerDefaultWorkerAgentType:s.managerDefaultWorkerAgentType??null,managerDefaultWorkerModelId:s.managerDefaultWorkerModelId??null,reasoningEffort:s.reasoningEffort??null,clientMessageId:s.clientMessageId??null,attachments:D(s.attachments),externalChannel:s.externalChannel??null,replyTarget:s.replyTarget??null,origin:s.origin,createdAt:e,updatedAt:e}}function x(s){return{id:s.clientMessageId??s.id,sessionId:s.sessionId,role:"user",ts:Date.parse(s.createdAt),payload:j(s.text,s.attachments)}}class he{opts;draining=new Set;inFlightMessageIds=new Map;prepareManagedRoomTurn=null;constructor(e){this.opts=e,setTimeout(()=>this.drainIdleQueues(),0).unref?.()}getSnapshot(e){return{...S(R()),sessionId:e,busy:!!this.opts.getRuntime().sessions.get(e)?.currentRunId,pending:(g().sessions[e]??[]).map(B)}}setManagedRoomTurnPreparer(e){this.prepareManagedRoomTurn=e}async enqueueManagedRoomBatch(e){if(e.length===0)return;const n=this.opts.getRuntime(),t=g(),i=new Set,r=new Map;let a=!1;for(const o of e){const d=o.params;if(!d.sessionId||!d.text||!d.agentType||!d.workDir)throw new Error("sessionId, text, agentType and workDir are required");const u=Q({...S(R()),...d});u.origin="room",u.managedRoom=o.context;const f=t.sessions[d.sessionId]??[],l=N(f,u),I=u.clientMessageId??u.id,y=k(d.sessionId,I);if(!l&&y.state==="not_found")f.push(u),t.sessions[d.sessionId]=f,r.set(`${u.sessionId}\0${I}`,u),i.add(d.sessionId),a=!0;else if(l&&y.state==="not_found")r.set(`${l.sessionId}\0${I}`,l),i.add(d.sessionId);else if(l&&y.state==="queued")i.add(d.sessionId);else if(l){const m=f.filter(c=>c.id!==l.id);m.length?t.sessions[d.sessionId]=m:delete t.sessions[d.sessionId],a=!0}}a&&p(t);for(const o of r.values())v(o.sessionId,x(o),"queued"),b({sessionId:o.sessionId,agentType:o.agentType,sessionMode:o.sessionMode,managerConfig:o.managerConfig,workDir:o.workDir,agentSessionId:W(n,o),modelId:o.modelId});if(i.size!==0){M(n.client);for(const o of i)this.logLifecycle("enqueue.persisted",o,{origin:"room",pendingCount:g().sessions[o]?.length??0}),this.broadcast(o),this.drainNext(o)}}async handleEnqueue(e){const n=this.opts.getRuntime(),t=e.params;if(J(t.sessionListProjection),!t.sessionId||!t.text||!t.agentType||!t.workDir){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId, text, agentType and workDir are required"});return}const i=n.resolvePath(t.workDir),r=D(t.attachments);r?.length&&w(i,r);const a={text:t.text,attachments:r,localized:!1},o=n.sessions.get(t.sessionId);if(t.agentType==="codex"&&t.agentSessionId&&o?.currentRunId?.startsWith("dispatch:")&&K(t.agentSessionId)){n.client.sendRes({type:"res",id:e.id,ok:!1,error:z});return}const d=await this.isSessionBusy(t.sessionId),u=g(),f=u.sessions[t.sessionId]??[];if(!d&&!f.length){await this.opts.dispatchReq({...e,id:H(e.id),method:"chat.send",params:{...t,text:a.text,responseId:e.id,clientMessageId:t.clientMessageId??t.queueMessageId,waitForDispatch:!0,attachments:a.attachments}});return}const l=Q({...t,text:a.text,attachments:a.attachments}),I=l.clientMessageId??l.id,y=k(t.sessionId,I);if(y.state!=="not_found"){n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:y.state==="queued",queueMessageId:l.id,...y,queue:this.getSnapshot(t.sessionId)}});return}const m=N(f,l);m||(u.sessions[t.sessionId]=[...f,l],p(u));const c=m??l;b({sessionId:c.sessionId,agentType:c.agentType,sessionMode:c.sessionMode,managerConfig:c.managerConfig,workDir:c.workDir,agentSessionId:c.agentSessionId,modelId:c.modelId}),v(c.sessionId,x(c),"queued"),T(n.resolvePath(c.workDir),c.attachments??[]),M(n.client),this.logLifecycle("enqueue.persisted",t.sessionId,{origin:t.origin??"app",queueMessageId:c.id,pendingCount:(g().sessions[t.sessionId]??[]).length}),this.broadcast(t.sessionId),n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:!0,queueMessageId:m?.id??l.id,...k(t.sessionId,m?.clientMessageId??m?.id??l.clientMessageId??l.id),queue:this.getSnapshot(t.sessionId),...a.localized?{localizedAttachments:!0}:{}}}),d||this.drainNext(t.sessionId)}async handleGet(e){const n=this.opts.getRuntime(),t=e.params;if(!t.sessionId){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId is required"});return}n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}}),this.getSnapshot(t.sessionId).busy||this.drainNext(t.sessionId)}async handleEdit(e){const n=this.opts.getRuntime(),t=e.params;if(!t.sessionId||!t.queueMessageId||!t.text){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId, queueMessageId and text are required"});return}const i=g(),r=i.sessions[t.sessionId]??[],a=r.findIndex(u=>u.id===t.queueMessageId);if(a<0){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}if(r[a].managedRoom){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const o=D(t.attachments),d=n.resolvePath(r[a].workDir);o?.length&&w(d,o),r[a]={...r[a],text:t.text,attachments:o,updatedAt:$()},i.sessions[t.sessionId]=r,p(i),P(t.sessionId,x(r[a]),"queued"),r[a].attachments?.length&&T(d,r[a].attachments??[]),M(n.client),this.broadcast(t.sessionId),n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}})}async handleDelete(e){const n=this.opts.getRuntime(),t=e.params;if(!t.sessionId||!t.queueMessageId){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId and queueMessageId are required"});return}const i=g(),r=i.sessions[t.sessionId]??[],a=r.find(d=>d.id===t.queueMessageId);if(a?.managedRoom){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const o=r.filter(d=>d.id!==t.queueMessageId);if(o.length===r.length){n.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}o.length?i.sessions[t.sessionId]=o:delete i.sessions[t.sessionId],p(i),E(t.sessionId,a?.clientMessageId??a?.id??t.queueMessageId),M(n.client),this.broadcast(t.sessionId),n.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}})}noteTerminal(e){this.drainNext(e)}purgeSession(e){X(e),this.draining.delete(e)}discardPendingManagedRoomMessages(e){const n=g(),t=n.sessions[e]??[],i=this.inFlightMessageIds.get(e)??new Set,r=t.filter(o=>o.managedRoom&&!i.has(o.id));if(r.length===0)return 0;const a=t.filter(o=>!r.includes(o));a.length?n.sessions[e]=a:delete n.sessions[e],p(n);for(const o of r)E(e,o.clientMessageId??o.id);return M(this.opts.getRuntime().client),this.logLifecycle("managed-room.discarded-paused",e,{discardedCount:r.length,pendingCount:a.length}),this.broadcast(e),r.length}drainIdleQueues(){const e=g();for(const n of Object.keys(e.sessions))this.drainNext(n)}async drainNext(e){if(this.draining.has(e))return;this.draining.add(e);let n=!1,t=[];try{if(await this.isSessionBusy(e)){this.logLifecycle("drain.deferred",e,{pendingCount:g().sessions[e]?.length??0});return}const i=g(),r=i.sessions[e]??[],a=r.filter(u=>{const f=u.clientMessageId??u.id,l=k(e,f).state;return l==="queued"||l==="not_found"});a.length!==r.length&&(a.length?i.sessions[e]=a:delete i.sessions[e],p(i),this.logLifecycle("drain.pruned-terminal",e,{prunedCount:r.length-a.length,pendingCount:a.length}));const o=Y(a);if(o.length===0){this.broadcast(e);return}t=o,this.inFlightMessageIds.set(e,new Set(o.map(u=>u.id))),this.logLifecycle("dispatch.start",e,{batchCount:o.length,queueMessageId:o[0].id,origin:o[0].origin??"app"});for(const u of o)P(e,x(u),"queued");await this.dispatchQueuedMessages(o);const d=this.removeBatch(e,o);this.logLifecycle("dispatch.accepted",e,{batchCount:o.length,queueMessageId:o[0].id,pendingCount:d}),n=d>0,this.broadcast(e)}catch(i){if(i instanceof V){const r=this.removeBatch(e,t);this.logLifecycle("dispatch.terminal-failure",e,{batchCount:t.length,queueMessageId:t[0]?.id,pendingCount:r,errorCode:O(i.message,"terminal_dispatch_failed")},"error"),n=r>0}else this.logLifecycle("dispatch.transient-failure",e,{errorCode:O(i instanceof Error?i.message:String(i),"transient_dispatch_failed")},"error");this.broadcast(e)}finally{this.inFlightMessageIds.delete(e),this.draining.delete(e),n&&queueMicrotask(()=>{this.drainNext(e)})}}removeBatch(e,n){if(n.length===0)return g().sessions[e]?.length??0;const t=g(),i=new Set(n.map(a=>a.id)),r=(t.sessions[e]??[]).filter(a=>!i.has(a.id));return r.length?t.sessions[e]=r:delete t.sessions[e],p(t),r.length}logLifecycle(e,n,t,i="info"){const r=`[chat.queue] ${JSON.stringify({event:e,sessionId:n,...t})}`;i==="error"?console.error(r):console.info(r)}async dispatchQueuedMessages(e){const n=e[0],t=e.flatMap(d=>d.managedRoom?[d.managedRoom]:[]),i=t.length>0?await this.prepareManagedRoomTurn?.(t):void 0;if(t.length>0&&!i)throw new Error("managed_room_turn_preparer_unavailable");const r=e.length===1?n.text:Z(e),a=te(e.flatMap(d=>d.attachments??[])),o=t.length>0?W(this.opts.getRuntime(),n):n.agentSessionId??null;await this.opts.dispatchQueuedReq({type:"req",id:`queue-send-${n.id}-${Date.now()}`,method:"chat.send",params:{...S(R()),sessionId:n.sessionId,text:r,agentType:n.agentType,sessionMode:t.length>0?"standard":n.sessionMode,managerConfig:t.length>0?null:n.managerConfig,workDir:n.workDir,agentSessionId:o,modelId:n.modelId??void 0,systemPrompt:i?.systemPrompt??n.systemPrompt??null,managerDefaultWorkerAgentType:t.length>0?null:n.managerDefaultWorkerAgentType,managerDefaultWorkerModelId:t.length>0?null:n.managerDefaultWorkerModelId,reasoningEffort:n.reasoningEffort??void 0,clientMessageId:n.clientMessageId??n.id,attachments:a,externalChannel:n.externalChannel,replyTarget:n.replyTarget,...i??{},roomActivation:t.length>0||void 0,suppressExternalOwnerEcho:t.length>0||void 0,queuedBatchItems:e.map(B),waitForDispatch:!0}})}async isSessionBusy(e){const n=this.opts.getRuntime().sessions.get(e);if(!n)return!1;if(n.currentRunId)return!0;if(!n.adapter.getStatus)return!1;const t=await n.adapter.getStatus().catch(()=>null);return t?.active?(t.runId&&(n.currentRunId=t.runId,n.currentRunPhase=t.runPhase??n.currentRunPhase),!0):!1}broadcast(e){this.opts.getRuntime().client.sendEvent({type:"event",event:"session.queue.update",payload:{queue:this.getSnapshot(e)}})}}function W(s,e){if(s.sessions.has(e.sessionId))return null;const t=U().find(i=>i.sessionId===e.sessionId)?.agentSessionId?.trim();return!t||t.startsWith("managed:")?null:t}function X(s){const e=g();e.sessions[s]&&(delete e.sessions[s],p(e))}function N(s,e){const n=_(e);if(n)return s.find(i=>_(i)===n)??null;if(e.origin!=="external")return null;const t=F(e);return s.find(i=>F(i)===t)??null}function B(s){const{origin:e,managedRoom:n,...t}=s;return t}function Y(s){const e=s[0];if(!e)return[];const n=L(e),t=[];for(const i of s){if(L(i)!==n)break;t.push(i)}return t}function L(s){return JSON.stringify({agentType:s.agentType,sessionMode:s.sessionMode??null,workDir:s.workDir,agentSessionId:s.agentSessionId??null,modelId:s.modelId??null,systemPrompt:s.systemPrompt??null,reasoningEffort:s.reasoningEffort??null,managerConfig:s.managerConfig??null,managerDefaultWorkerAgentType:s.managerDefaultWorkerAgentType??null,managerDefaultWorkerModelId:s.managerDefaultWorkerModelId??null,externalChannel:s.externalChannel??null,replyTarget:s.replyTarget??null,roomRegistryKey:s.managedRoom?.registryKey??null})}function Z(s){return s.every(e=>e.managedRoom)?`[\u7FA4\u6D88\u606F]
+${s.map(e=>ee(e.text)).filter(Boolean).join(`
 
 `)}`:`[\u8FDE\u7EED\u7528\u6237\u6D88\u606F]
 ${s.map((e,n)=>`<\u6D88\u606F ${n+1}>
 ${e.text.trim()}`).join(`
 
-`)}`}function Y(s){const e=s.trim(),n=e.match(/^\[群消息\][ \t]+<([^>\n]+)>\n([\s\S]*)$/);return n?`<${n[1]}>
+`)}`}function ee(s){const e=s.trim(),n=e.match(/^\[群消息\][ \t]+<([^>\n]+)>\n([\s\S]*)$/);return n?`<${n[1]}>
 ${n[2]}`:e.startsWith(`[\u7FA4\u6D88\u606F]
-`)?e.slice(6):e}function B(s){const e=R(s);return e?`id:${e}`:["body",ee(s.text),...(s.attachments??[]).map(n=>n.path||`${n.name}:${n.mimeType}`)].join(`
-`)}function R(s){const e=typeof s.clientMessageId=="string"?s.clientMessageId.trim():"";if(e)return e;const n=typeof s.id=="string"?s.id.trim():"";return n&&!n.startsWith("queue-")?n:""}function Z(s){const e=new Set,n=[];for(const t of s){const i=t.path||`${t.name}:${t.mimeType}`;e.has(i)||(e.add(i),n.push(t))}return n.length?n:void 0}function ee(s){return String(s||"").replace(/\s+/g," ").trim()}function L(s,e){const n=s.toLowerCase();return n.includes("no rollout found")?"missing_rollout":n.includes("managed_room_permission_snapshot_mismatch")?"managed_room_permission_snapshot_mismatch":n.includes("managed_room_prompt_snapshot_mismatch")?"managed_room_prompt_snapshot_mismatch":n.includes("managed_room_turn_preparer_unavailable")?"managed_room_turn_preparer_unavailable":n.includes("timeout")?"timeout":n.includes("connection")?"connection_error":e}export{fe as ChatQueueManager,J as TerminalQueuedChatDispatchError,G as purgeQueuedSessionData};
+`)?e.slice(6):e}function F(s){const e=_(s);return e?`id:${e}`:["body",ne(s.text),...(s.attachments??[]).map(n=>n.path||`${n.name}:${n.mimeType}`)].join(`
+`)}function _(s){const e=typeof s.clientMessageId=="string"?s.clientMessageId.trim():"";if(e)return e;const n=typeof s.id=="string"?s.id.trim():"";return n&&!n.startsWith("queue-")?n:""}function te(s){const e=new Set,n=[];for(const t of s){const i=t.path||`${t.name}:${t.mimeType}`;e.has(i)||(e.add(i),n.push(t))}return n.length?n:void 0}function ne(s){return String(s||"").replace(/\s+/g," ").trim()}function O(s,e){const n=s.toLowerCase();return n.includes("no rollout found")?"missing_rollout":n.includes("managed_room_permission_snapshot_mismatch")?"managed_room_permission_snapshot_mismatch":n.includes("managed_room_prompt_snapshot_mismatch")?"managed_room_prompt_snapshot_mismatch":n.includes("managed_room_turn_preparer_unavailable")?"managed_room_turn_preparer_unavailable":n.includes("timeout")?"timeout":n.includes("connection")?"connection_error":e}export{he as ChatQueueManager,V as TerminalQueuedChatDispatchError,X as purgeQueuedSessionData};

package/dist/src/upgrade/scheduler.js

@@ -1,126 +1 @@
-// @arch docs/architecture/cli/upgrade.md#自动升级
-// @test packages/cli/src/__tests__/upgrade-scheduler.test.ts
-import { checkForUpdate, isUpgradeVersionInCooldown, } from './engine.js';
-const DEFAULT_RELEASE_POLICY = {
-    automaticPatchEnabled: false,
-    rolloutPercentage: 0,
-    pausedVersions: [],
-    blockedVersions: [],
-    killSwitch: true,
-};
-export function deterministicRolloutBucket(machineId) {
-    let hash = 2166136261;
-    for (let i = 0; i < machineId.length; i++) {
-        hash ^= machineId.charCodeAt(i);
-        hash = Math.imul(hash, 16777619);
-    }
-    return (hash >>> 0) % 100;
-}
-export function isAutomaticPatchAllowed(policy, machineId, currentVersion, targetVersion) {
-    if (policy.killSwitch || !policy.automaticPatchEnabled)
-        return false;
-    if (policy.blockedVersions.includes(targetVersion))
-        return false;
-    if (policy.pausedVersions.includes(targetVersion) || policy.pausedVersions.includes(currentVersion))
-        return false;
-    const rollout = Math.max(0, Math.min(100, Math.floor(policy.rolloutPercentage)));
-    return deterministicRolloutBucket(machineId) < rollout;
-}
-export async function fetchCliReleasePolicy(serverUrl) {
-    const url = new URL('/api/releases/cli-policy', serverUrl);
-    const response = await fetch(url, {
-        headers: { Accept: 'application/json' },
-        signal: AbortSignal.timeout(10_000),
-    });
-    if (!response.ok)
-        throw new Error(`release policy returned ${response.status}`);
-    const raw = await response.json();
-    return {
-        automaticPatchEnabled: raw.automaticPatchEnabled === true,
-        rolloutPercentage: typeof raw.rolloutPercentage === 'number' ? raw.rolloutPercentage : 0,
-        pausedVersions: Array.isArray(raw.pausedVersions) ? raw.pausedVersions.filter((v) => typeof v === 'string') : [],
-        blockedVersions: Array.isArray(raw.blockedVersions) ? raw.blockedVersions.filter((v) => typeof v === 'string') : [],
-        killSwitch: raw.killSwitch !== false,
-    };
-}
-export class AutoUpgradeScheduler {
-    deps;
-    policy = 'none';
-    timer = null;
-    started = false;
-    upgradeInFlight = false;
-    constructor(deps) {
-        this.deps = deps;
-    }
-    start() {
-        if (this.started)
-            return;
-        this.started = true;
-        if (this.policy === 'patch')
-            this.arm(this.deps.initialDelayMs ?? 30_000);
-    }
-    stop() {
-        this.started = false;
-        if (this.timer)
-            (this.deps.clearTimeout ?? clearTimeout)(this.timer);
-        this.timer = null;
-    }
-    setPolicy(policy) {
-        this.policy = policy;
-        if (!this.started)
-            return;
-        if (policy === 'none') {
-            if (this.timer)
-                (this.deps.clearTimeout ?? clearTimeout)(this.timer);
-            this.timer = null;
-            return;
-        }
-        if (!this.timer)
-            this.arm(0);
-    }
-    getPolicy() {
-        return this.policy;
-    }
-    async runOnce() {
-        if (this.policy !== 'patch' || this.upgradeInFlight)
-            return;
-        const idle = this.deps.getIdleState();
-        if (!idle.idle)
-            return;
-        const update = await (this.deps.checkForUpdate ?? checkForUpdate)(this.deps.currentVersion);
-        if (!update.hasUpdate || update.changeType !== 'patch')
-            return;
-        if ((this.deps.inCooldown ?? isUpgradeVersionInCooldown)(update.latest))
-            return;
-        let releasePolicy = DEFAULT_RELEASE_POLICY;
-        try {
-            releasePolicy = await (this.deps.fetchReleasePolicy ?? (() => fetchCliReleasePolicy(this.deps.serverUrl)))();
-        }
-        catch {
-            return; // fail closed: a missing control-plane policy never permits an automatic install
-        }
-        if (!isAutomaticPatchAllowed(releasePolicy, this.deps.machineId, update.current, update.latest))
-            return;
-        this.upgradeInFlight = true;
-        try {
-            await this.deps.startUpgrade(update.latest);
-        }
-        finally {
-            this.upgradeInFlight = false;
-        }
-    }
-    arm(delayMs) {
-        if (!this.started || this.timer)
-            return;
-        const setTimer = this.deps.setTimeout ?? setTimeout;
-        this.timer = setTimer(() => {
-            this.timer = null;
-            void this.runOnce().catch(() => { }).finally(() => {
-                if (this.started && this.policy === 'patch') {
-                    this.arm(this.deps.intervalMs ?? 5 * 60_000);
-                }
-            });
-        }, delayMs);
-        this.timer.unref?.();
-    }
-}
+import{checkForUpdate as a,isUpgradeVersionInCooldown as l}from"./engine.js";const n={automaticPatchEnabled:!1,rolloutPercentage:0,pausedVersions:[],blockedVersions:[],killSwitch:!0};function o(s){let e=2166136261;for(let t=0;t<s.length;t++)e^=s.charCodeAt(t),e=Math.imul(e,16777619);return(e>>>0)%100}function c(s,e,t,i){if(s.killSwitch||!s.automaticPatchEnabled||s.blockedVersions.includes(i)||s.pausedVersions.includes(i)||s.pausedVersions.includes(t))return!1;const r=Math.max(0,Math.min(100,Math.floor(s.rolloutPercentage)));return o(e)<r}async function h(s){const e=new URL("/api/releases/cli-policy",s),t=await fetch(e,{headers:{Accept:"application/json"},signal:AbortSignal.timeout(1e4)});if(!t.ok)throw new Error(`release policy returned ${t.status}`);const i=await t.json();return{automaticPatchEnabled:i.automaticPatchEnabled===!0,rolloutPercentage:typeof i.rolloutPercentage=="number"?i.rolloutPercentage:0,pausedVersions:Array.isArray(i.pausedVersions)?i.pausedVersions.filter(r=>typeof r=="string"):[],blockedVersions:Array.isArray(i.blockedVersions)?i.blockedVersions.filter(r=>typeof r=="string"):[],killSwitch:i.killSwitch!==!1}}class d{deps;policy="none";timer=null;started=!1;upgradeInFlight=!1;constructor(e){this.deps=e}start(){this.started||(this.started=!0,this.policy==="patch"&&this.arm(this.deps.initialDelayMs??3e4))}stop(){this.started=!1,this.timer&&(this.deps.clearTimeout??clearTimeout)(this.timer),this.timer=null}setPolicy(e){if(this.policy=e,!!this.started){if(e==="none"){this.timer&&(this.deps.clearTimeout??clearTimeout)(this.timer),this.timer=null;return}this.timer||this.arm(0)}}getPolicy(){return this.policy}async runOnce(){if(this.policy!=="patch"||this.upgradeInFlight||!this.deps.getIdleState().idle)return;const t=await(this.deps.checkForUpdate??a)(this.deps.currentVersion);if(!t.hasUpdate||t.changeType!=="patch"||(this.deps.inCooldown??l)(t.latest))return;let i=n;try{i=await(this.deps.fetchReleasePolicy??(()=>h(this.deps.serverUrl)))()}catch{return}if(c(i,this.deps.machineId,t.current,t.latest)){this.upgradeInFlight=!0;try{await this.deps.startUpgrade(t.latest)}finally{this.upgradeInFlight=!1}}}arm(e){if(!this.started||this.timer)return;const t=this.deps.setTimeout??setTimeout;this.timer=t(()=>{this.timer=null,this.runOnce().catch(()=>{}).finally(()=>{this.started&&this.policy==="patch"&&this.arm(this.deps.intervalMs??5*6e4)})},e),this.timer.unref?.()}}export{d as AutoUpgradeScheduler,o as deterministicRolloutBucket,h as fetchCliReleasePolicy,c as isAutomaticPatchAllowed};

package/node_modules/@shennian/wire/dist/agent-workspace.js

@@ -3,10 +3,10 @@
 export const MANAGED_AGENT_PERMISSION_POLICY_VERSION = 1;
 export const DEFAULT_MANAGED_AGENT_PERMISSION_POLICY = {
     version: MANAGED_AGENT_PERMISSION_POLICY_VERSION,
-    fileAccess: 'workspace-write',
+    fileAccess: 'workspace-read',
     externalCapabilities: {
         commandNetwork: false,
-        webSearch: true,
+        webSearch: false,
         browser: false,
         computerUse: false,
         mcp: false,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shennian",
-  "version": "0.3.86",
+  "version": "0.4.0",
   "description": "Shennian — AI Agent Control Plane CLI",
   "type": "module",
   "bin": {