shennian 0.3.85 → 0.3.86

package/dist/src/session/queue.js

@@ -1,12 +1,12 @@
-import p from"node:fs";import{randomUUID as R}from"node:crypto";import{CODEX_DESKTOP_SEND_UNAVAILABLE as O,buildUserMessagePayload as j}from"@shennian/wire";import{isCodexThreadTailInterrupted as z}from"../native-fusion/codex-files.js";import{mergeProjectedSessions as F}from"./projection.js";import{parsePersonalAttachmentReferences as D}from"./attachment-reference.js";import{assertFinishedAttachmentReferences as T,markAttachmentReferences as v}from"./attachment-transfer-store.js";import{resolvePersonalSyncV1Path as $}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 E,getMessageDeliveryStatus as q,recordSession as b,tombstoneMessage as K,upsertMessage as P}from"./store.js";import{requestSessionIndexPublish as k}from"./index-publisher.js";function L(){return{sessions:{}}}function C(){return new Date().toISOString()}function U(n){return`enqueue-send-${n}-${R()}`}function g(){try{const n=JSON.parse(p.readFileSync(A(),"utf-8"));return{sessions:n.sessions&&typeof n.sessions=="object"?n.sessions:{}}}catch(n){if(n.code==="ENOENT"||n instanceof Error&&n.message.startsWith("ENOENT:"))return L();throw n}}function I(n){const e=$();p.mkdirSync(e,{recursive:!0,mode:448});const s=A(),t=`${s}.tmp-${process.pid}-${R()}`;try{p.writeFileSync(t,`${JSON.stringify(n,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,s)}finally{try{p.unlinkSync(t)}catch{}}}function A(){return $("outbound-queue.json")}function _(n){const e=C();return{...M(S()),id:n.queueMessageId||n.clientMessageId||`queue-${R()}`,sessionId:n.sessionId,text:n.text,agentType:n.agentType,sessionMode:n.sessionMode,managerConfig:n.managerConfig,workDir:n.workDir,agentSessionId:n.agentSessionId??null,modelId:n.modelId??null,systemPrompt:n.systemPrompt??null,managerDefaultWorkerAgentType:n.managerDefaultWorkerAgentType??null,managerDefaultWorkerModelId:n.managerDefaultWorkerModelId??null,reasoningEffort:n.reasoningEffort??null,clientMessageId:n.clientMessageId??null,attachments:D(n.attachments),externalChannel:n.externalChannel??null,replyTarget:n.replyTarget??null,origin:n.origin,createdAt:e,updatedAt:e}}function x(n){return{id:n.clientMessageId??n.id,sessionId:n.sessionId,role:"user",ts:Date.parse(n.createdAt),payload:j(n.text,n.attachments)}}class ce{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 s=this.opts.getRuntime(),t=g(),i=new Set,a=new Map;let d=!1;for(const o of e){const r=o.params;if(!r.sessionId||!r.text||!r.agentType||!r.workDir)throw new Error("sessionId, text, agentType and workDir are required");const l=_({...M(S()),...r});l.origin="room",l.managedRoom=o.context;const f=t.sessions[r.sessionId]??[],u=Q(f,l),y=l.clientMessageId??l.id,h=q(r.sessionId,y);if(!u&&h.state==="not_found")f.push(l),t.sessions[r.sessionId]=f,a.set(`${l.sessionId}\0${y}`,l),i.add(r.sessionId),d=!0;else if(u&&h.state==="not_found")a.set(`${u.sessionId}\0${y}`,u),i.add(r.sessionId);else if(u&&h.state==="queued")i.add(r.sessionId);else if(u){const m=f.filter(c=>c.id!==u.id);m.length?t.sessions[r.sessionId]=m:delete t.sessions[r.sessionId],d=!0}}d&&I(t);for(const o of a.values())E(o.sessionId,x(o),"queued"),b({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){k(s.client);for(const o of i)this.broadcast(o),this.drainNext(o)}}async handleEnqueue(e){const s=this.opts.getRuntime(),t=e.params;if(F(t.sessionListProjection),!t.sessionId||!t.text||!t.agentType||!t.workDir){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId, text, agentType and workDir are required"});return}const i=s.resolvePath(t.workDir),a=D(t.attachments);a?.length&&T(i,a);const d={text:t.text,attachments:a,localized:!1},o=s.sessions.get(t.sessionId);if(t.agentType==="codex"&&t.agentSessionId&&o?.currentRunId?.startsWith("dispatch:")&&z(t.agentSessionId)){s.client.sendRes({type:"res",id:e.id,ok:!1,error:O});return}const r=await this.isSessionBusy(t.sessionId),l=g(),f=l.sessions[t.sessionId]??[];if(!r&&!f.length){await this.opts.dispatchReq({...e,id:U(e.id),method:"chat.send",params:{...t,text:d.text,responseId:e.id,clientMessageId:t.clientMessageId??t.queueMessageId,waitForDispatch:!0,attachments:d.attachments}});return}const u=_({...t,text:d.text,attachments:d.attachments}),y=u.clientMessageId??u.id,h=q(t.sessionId,y);if(h.state!=="not_found"){s.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],I(l));const c=m??u;b({sessionId:c.sessionId,agentType:c.agentType,sessionMode:c.sessionMode,managerConfig:c.managerConfig,workDir:c.workDir,agentSessionId:c.agentSessionId,modelId:c.modelId}),E(c.sessionId,x(c),"queued"),v(s.resolvePath(c.workDir),c.attachments??[]),k(s.client),this.broadcast(t.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:!0,queueMessageId:m?.id??u.id,...q(t.sessionId,m?.clientMessageId??m?.id??u.clientMessageId??u.id),queue:this.getSnapshot(t.sessionId),...d.localized?{localizedAttachments:!0}:{}}}),r||this.drainNext(t.sessionId)}async handleGet(e){const s=this.opts.getRuntime(),t=e.params;if(!t.sessionId){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId is required"});return}s.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 s=this.opts.getRuntime(),t=e.params;if(!t.sessionId||!t.queueMessageId||!t.text){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId, queueMessageId and text are required"});return}const i=g(),a=i.sessions[t.sessionId]??[],d=a.findIndex(l=>l.id===t.queueMessageId);if(d<0){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}if(a[d].managedRoom){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const o=D(t.attachments),r=s.resolvePath(a[d].workDir);o?.length&&T(r,o),a[d]={...a[d],text:t.text,attachments:o,updatedAt:C()},i.sessions[t.sessionId]=a,I(i),P(t.sessionId,x(a[d]),"queued"),a[d].attachments?.length&&v(r,a[d].attachments??[]),k(s.client),this.broadcast(t.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}})}async handleDelete(e){const s=this.opts.getRuntime(),t=e.params;if(!t.sessionId||!t.queueMessageId){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId and queueMessageId are required"});return}const i=g(),a=i.sessions[t.sessionId]??[],d=a.find(r=>r.id===t.queueMessageId);if(d?.managedRoom){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const o=a.filter(r=>r.id!==t.queueMessageId);if(o.length===a.length){s.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],I(i),K(t.sessionId,d?.clientMessageId??d?.id??t.queueMessageId),k(s.client),this.broadcast(t.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}})}noteTerminal(e){this.drainNext(e)}purgeSession(e){J(e),this.draining.delete(e)}drainIdleQueues(){const e=g();for(const s of Object.keys(e.sessions))this.drainNext(s)}async drainNext(e){if(!this.draining.has(e)){this.draining.add(e);try{if(await this.isSessionBusy(e))return;const t=g().sessions[e]??[],i=V(t);if(i.length===0){this.broadcast(e);return}for(const r of i)P(e,x(r),"queued");await this.dispatchQueuedMessages(i);const a=g(),d=new Set(i.map(r=>r.id)),o=(a.sessions[e]??[]).filter(r=>!d.has(r.id));o.length?a.sessions[e]=o:delete a.sessions[e],I(a),this.broadcast(e)}catch(s){console.error(`[chat.queue] dispatch failed sessionId=${e}: ${s instanceof Error?s.message:String(s)}`),this.broadcast(e)}finally{this.draining.delete(e)}}}async dispatchQueuedMessages(e){const s=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 a=e.length===1?s.text:G(e),d=X(e.flatMap(o=>o.attachments??[]));await this.opts.dispatchQueuedReq({type:"req",id:`queue-send-${s.id}-${Date.now()}`,method:"chat.send",params:{...M(S()),sessionId:s.sessionId,text:a,agentType:s.agentType,sessionMode:t.length>0?"standard":s.sessionMode,managerConfig:t.length>0?null:s.managerConfig,workDir:s.workDir,agentSessionId:s.agentSessionId??null,modelId:s.modelId??void 0,systemPrompt:i?.systemPrompt??s.systemPrompt??null,managerDefaultWorkerAgentType:t.length>0?null:s.managerDefaultWorkerAgentType,managerDefaultWorkerModelId:t.length>0?null:s.managerDefaultWorkerModelId,reasoningEffort:s.reasoningEffort??void 0,clientMessageId:s.clientMessageId??s.id,attachments:d,externalChannel:s.externalChannel,replyTarget:s.replyTarget,...i??{},roomActivation:t.length>0||void 0,suppressExternalOwnerEcho:t.length>0||void 0,queuedBatchItems:e.map(W),waitForDispatch:!0}})}async isSessionBusy(e){const s=this.opts.getRuntime().sessions.get(e);if(!s)return!1;if(s.currentRunId)return!0;if(!s.adapter.getStatus)return!1;const t=await s.adapter.getStatus().catch(()=>null);return t?.active?(t.runId&&(s.currentRunId=t.runId,s.currentRunPhase=t.runPhase??s.currentRunPhase),!0):!1}broadcast(e){this.opts.getRuntime().client.sendEvent({type:"event",event:"session.queue.update",payload:{queue:this.getSnapshot(e)}})}}function J(n){const e=g();e.sessions[n]&&(delete e.sessions[n],I(e))}function Q(n,e){const s=w(e);if(s)return n.find(i=>w(i)===s)??null;if(e.origin!=="external")return null;const t=B(e);return n.find(i=>B(i)===t)??null}function W(n){const{origin:e,managedRoom:s,...t}=n;return t}function V(n){const e=n[0];if(!e)return[];const s=N(e),t=[];for(const i of n){if(N(i)!==s)break;t.push(i)}return t}function N(n){return JSON.stringify({agentType:n.agentType,sessionMode:n.sessionMode??null,workDir:n.workDir,agentSessionId:n.agentSessionId??null,modelId:n.modelId??null,systemPrompt:n.systemPrompt??null,reasoningEffort:n.reasoningEffort??null,managerConfig:n.managerConfig??null,managerDefaultWorkerAgentType:n.managerDefaultWorkerAgentType??null,managerDefaultWorkerModelId:n.managerDefaultWorkerModelId??null,externalChannel:n.externalChannel??null,replyTarget:n.replyTarget??null,roomRegistryKey:n.managedRoom?.registryKey??null})}function G(n){return n.every(e=>e.managedRoom)?`[\u7FA4\u6D88\u606F]
-${n.map(e=>H(e.text)).filter(Boolean).join(`
+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(`
 
 `)}`:`[\u8FDE\u7EED\u7528\u6237\u6D88\u606F]
-${n.map((e,s)=>`<\u6D88\u606F ${s+1}>
+${s.map((e,n)=>`<\u6D88\u606F ${n+1}>
 ${e.text.trim()}`).join(`
 
-`)}`}function H(n){const e=n.trim(),s=e.match(/^\[群消息\][ \t]+<([^>\n]+)>\n([\s\S]*)$/);return s?`<${s[1]}>
-${s[2]}`:e.startsWith(`[\u7FA4\u6D88\u606F]
-`)?e.slice(6):e}function B(n){const e=w(n);return e?`id:${e}`:["body",Y(n.text),...(n.attachments??[]).map(s=>s.path||`${s.name}:${s.mimeType}`)].join(`
-`)}function w(n){const e=typeof n.clientMessageId=="string"?n.clientMessageId.trim():"";if(e)return e;const s=typeof n.id=="string"?n.id.trim():"";return s&&!s.startsWith("queue-")?s:""}function X(n){const e=new Set,s=[];for(const t of n){const i=t.path||`${t.name}:${t.mimeType}`;e.has(i)||(e.add(i),s.push(t))}return s.length?s:void 0}function Y(n){return String(n||"").replace(/\s+/g," ").trim()}export{ce as ChatQueueManager,J as purgeQueuedSessionData};
+`)}`}function Y(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};

package/dist/src/upgrade/scheduler.js

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

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

@@ -6,7 +6,7 @@ export const DEFAULT_MANAGED_AGENT_PERMISSION_POLICY = {
     fileAccess: 'workspace-write',
     externalCapabilities: {
         commandNetwork: false,
-        webSearch: false,
+        webSearch: true,
         browser: false,
         computerUse: false,
         mcp: false,

package/package.json

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