shennian 0.4.58 → 0.4.60

package/dist/publish-build-manifest.json

@@ -2,7 +2,7 @@
   "schemaVersion": 1,
   "jsMinified": true,
   "minifier": "esbuild",
-  "fileCount": 200,
+  "fileCount": 201,
   "files": [
     {
       "file": "bin/shennian.js",
@@ -366,8 +366,8 @@
     },
     {
       "file": "src/commands/upgrade.js",
-      "beforeBytes": 12351,
-      "afterBytes": 6348
+      "beforeBytes": 13052,
+      "afterBytes": 6572
     },
     {
       "file": "src/config/index.js",
@@ -471,8 +471,8 @@
     },
     {
       "file": "src/manager/runtime.js",
-      "beforeBytes": 28472,
-      "afterBytes": 14502
+      "beforeBytes": 28570,
+      "afterBytes": 14555
     },
     {
       "file": "src/manager/worker-ipc.js",
@@ -906,8 +906,8 @@
     },
     {
       "file": "src/session/queue.js",
-      "beforeBytes": 42174,
-      "afterBytes": 19010
+      "beforeBytes": 42777,
+      "afterBytes": 19219
     },
     {
       "file": "src/session/queued-chat-dispatch.js",
@@ -964,10 +964,15 @@
       "beforeBytes": 2185,
       "afterBytes": 1207
     },
+    {
+      "file": "src/upgrade/distribution.js",
+      "beforeBytes": 2160,
+      "afterBytes": 1364
+    },
     {
       "file": "src/upgrade/engine.js",
-      "beforeBytes": 16155,
-      "afterBytes": 6796
+      "beforeBytes": 16718,
+      "afterBytes": 7089
     },
     {
       "file": "src/upgrade/scheduler.js",
@@ -996,8 +1001,8 @@
     },
     {
       "file": "src/upgrade/updater-worker.js",
-      "beforeBytes": 30817,
-      "afterBytes": 15224
+      "beforeBytes": 33374,
+      "afterBytes": 16485
     },
     {
       "file": "src/windows-shell.js",

package/dist/src/commands/upgrade.js

@@ -1 +1 @@
-import c from"chalk";import{execSync as x,spawn as C}from"node:child_process";import{randomUUID as T}from"node:crypto";import I from"node:os";import{createInterface as b}from"node:readline/promises";import{getCurrentVersion as h,fetchLatestVersion as w,compareVersions as k,checkForUpdate as E,acquireUpgradeLock as V,recordUpgradeFailure as $}from"../upgrade/engine.js";import{callDaemonIpc as P}from"../daemon-ipc/client.js";import{getDaemonStatus as M}from"./daemon.js";import{ensureUpdaterInstalled as v}from"../upgrade/updater-bootstrap.js";import{launchUpdater as U,createUpdaterJob as F,waitForUpdaterReady as A,confirmUpdaterHandoff as _}from"../upgrade/updater-launcher.js";import{prepareRecoveryServiceEntry as D}from"../upgrade/updater-service-entry.js";import{resolveShennianPath as H}from"../config/index.js";import{assertUpdaterAvailable as L,prepareUpdaterCandidate as q}from"../upgrade/updater-worker.js";import{getUpgradeChannel as R,setUpgradeChannel as N}from"../upgrade/channel.js";const S=["\u280B","\u2819","\u2839","\u2838","\u283C","\u2834","\u2826","\u2827","\u2807","\u280F"];function G(n,r){return n!=="major"||r}function oe(n={}){const r=n.platform??I.platform(),i=n.kill??process.kill,e=n.pid??process.pid,d=n.env??process.env,t=n.spawn??C,o=n.execSync??x;if(r==="win32"){const a=d.ComSpec||"cmd.exe";try{t(a,["/d","/s","/c","timeout /t 2 /nobreak >nul & schtasks /Run /TN ShennianAgent >nul"],{detached:!0,stdio:"ignore",windowsHide:!0}).unref(),i(e,"SIGTERM");return}catch{try{o('schtasks /run /tn "ShennianAgent"',{stdio:"pipe",windowsHide:!0})}catch{}}}else try{t("sh",["-c","sleep 2; shennian start >/dev/null 2>&1"],{detached:!0,stdio:"ignore",env:d}).unref()}catch{}i(e,"SIGTERM")}function ae(n){n.command("upgrade").description("Upgrade shennian CLI to the latest version").option("--channel <channel>","Persist stable (npm latest) or preview (npm next); does not install").option("--check","Only check for updates, do not install").option("--to <version>","Upgrade to a specific version").option("--yes","Confirm a major version upgrade (required in non-interactive environments)").action(async r=>{if(r.channel!==void 0){if(r.to||r.yes)throw new Error("--channel cannot be combined with --to or --yes");if(N(r.channel),console.log(`Upgrade channel: ${R()} (saved; no installation)`),!r.check)return}console.log(`Upgrade channel: ${R()}`);const i=h();console.log(c.gray(`Current version: ${i}`)),console.log();let e=0;const d=setInterval(()=>{process.stdout.write(`\r  ${S[e%S.length]} Checking for updates...`),e++},80);let t;try{t=r.to??await w(),clearInterval(d),process.stdout.write("\r"+" ".repeat(40)+"\r")}catch(a){clearInterval(d),process.stdout.write("\r"+" ".repeat(40)+"\r"),console.error(c.red(`\u2717 Failed to check for updates: ${a instanceof Error?a.message:String(a)}`)),process.exit(1)}const o=k(i,t);if(o==="none"&&!r.to){console.log(c.green(`\u2713 Already up to date (${i})`));return}if(console.log(`  Available: ${c.green(t)} ${o==="major"?c.red("(major)"):o==="minor"?c.yellow("(minor)"):c.green("(patch)")}`),console.log(),!r.check){if(o==="major"&&(console.log(c.yellow("  \u26A0 This is a major version update and may contain breaking changes.")),console.log(c.gray("    Please review the changelog before upgrading.")),console.log(),!r.yes)){(!process.stdin.isTTY||!process.stdout.isTTY)&&(console.error(c.red("\u2717 Major upgrades require explicit confirmation; rerun with --yes.")),process.exit(1));const a=b({input:process.stdin,output:process.stdout}),p=await a.question(`  Type ${t} to confirm this major upgrade: `);if(a.close(),p.trim()!==t){console.log(c.gray("  Upgrade cancelled."));return}}try{await P({method:"upgrade.start",params:{targetVersion:t,confirmedMajor:o==="major"||r.yes===!0},timeoutMs:1e4})}catch(a){if(M({cleanupStale:!0}).running)throw a;const f=v();U({fromVersion:i,targetVersion:t,oldDaemonPid:null},{readiness:f})}console.log(),console.log(c.green(`\u2713 Upgrade scheduled ${i} \u2192 ${t}`)),console.log(c.gray("  The independent updater will verify, roll back if needed, and restart Shennian."))}})}async function se(n,r,i,e={}){const d=s=>{n.sendEvent({type:"event",event:"upgrade",payload:s})},t=e.currentVersion??h();let o;try{d({state:"checking",machineId:"self"}),o=i??await w()}catch(s){const l=`Version check failed: ${s instanceof Error?s.message:String(s)}`;return e.suppressResponse||n.sendRes({type:"res",id:r,ok:!1,error:l}),{ok:!1,error:l}}const a=k(t,o);if(a==="none")return e.suppressResponse||n.sendRes({type:"res",id:r,ok:!0,payload:{alreadyLatest:!0,version:t}}),{ok:!0,alreadyLatest:!0,version:t};if(!G(a,e.confirmedMajor===!0))return e.suppressResponse||n.sendRes({type:"res",id:r,ok:!1,error:"Major upgrades require explicit second confirmation"}),{ok:!1,error:"Major upgrades require explicit second confirmation"};const p=T();let f;try{f=V();try{L(H("runtime","updater"))}catch(s){throw f.release(),s}}catch(s){const l=s instanceof Error?s.message:String(s);return e.suppressResponse||n.sendRes({type:"res",id:r,ok:!1,error:l}),{ok:!1,error:l}}const g={from:t,to:o,jobId:p},y=async()=>{let s=!1;try{const l=v(),u={fromVersion:t,targetVersion:o,oldDaemonPid:process.pid,jobId:p,requireHandoffAck:!0},m=F(u);if(d({state:"downloading",machineId:"self",version:o,jobId:p}),await q(m),D(m,l.workerPath),U(u,{readiness:l}),await A(m),e.canHandoff&&!e.canHandoff())throw new Error("upgrade_waiting_for_idle");const j=e.prepareForUpgrade?.(p)??{activeRunCount:0};return s=!0,_(m),d({state:"restarting",machineId:"self",...g,activeRunCount:j.activeRunCount}),{ok:!0,...g,resultPath:m.resultPath}}catch(l){s&&e.cancelUpgradePreparation?.();const u=l instanceof Error?l.message:String(l);return u!=="upgrade_waiting_for_idle"&&$(o,u),d({state:"failed",machineId:"self",error:u,jobId:p}),{ok:!1,error:u}}finally{f.release()}};return e.suppressResponse||n.sendRes({type:"res",id:r,ok:!0,payload:g}),e.waitForCompletion?y():(y(),{ok:!0,...g})}async function ie(n,r,i={}){const e=await E(i.currentVersion);n.sendRes({type:"res",id:r,ok:!0,payload:e.hasUpdate?{hasUpdate:!0,current:e.current,latest:e.latest,changeType:e.changeType}:{hasUpdate:!1,current:i.currentVersion??h()}})}export{se as handleUpgradeStart,ie as handleUpgradeStatus,G as hasMajorUpgradeConfirmation,ae as registerUpgradeCommand,oe as restartCurrentDaemonAfterUpgrade};
+import c from"chalk";import{execSync as P,spawn as V}from"node:child_process";import{randomUUID as w}from"node:crypto";import $ from"node:os";import{createInterface as A}from"node:readline/promises";import{getCurrentVersion as y,fetchLatestVersion as k,compareVersions as v,checkForUpdate as M,acquireUpgradeLock as F,recordUpgradeFailure as O}from"../upgrade/engine.js";import{callDaemonIpc as _}from"../daemon-ipc/client.js";import{getDaemonStatus as D}from"./daemon.js";import{ensureUpdaterInstalled as U}from"../upgrade/updater-bootstrap.js";import{launchUpdater as S,createUpdaterJob as R,waitForUpdaterReady as H,confirmUpdaterHandoff as L}from"../upgrade/updater-launcher.js";import{prepareRecoveryServiceEntry as j}from"../upgrade/updater-service-entry.js";import{resolveShennianPath as q}from"../config/index.js";import{assertUpdaterAvailable as N,prepareUpdaterCandidate as C}from"../upgrade/updater-worker.js";import{getUpgradeChannel as x,setUpgradeChannel as G}from"../upgrade/channel.js";import{fetchOwnedCliArtifact as T,getCliUpdateSourceOrder as b}from"../upgrade/distribution.js";const I=["\u280B","\u2819","\u2839","\u2838","\u283C","\u2834","\u2826","\u2827","\u2807","\u280F"];function Y(t,r){return t!=="major"||r}function ie(t={}){const r=t.platform??$.platform(),i=t.kill??process.kill,e=t.pid??process.pid,l=t.env??process.env,n=t.spawn??V,o=t.execSync??P;if(r==="win32"){const a=l.ComSpec||"cmd.exe";try{n(a,["/d","/s","/c","timeout /t 2 /nobreak >nul & schtasks /Run /TN ShennianAgent >nul"],{detached:!0,stdio:"ignore",windowsHide:!0}).unref(),i(e,"SIGTERM");return}catch{try{o('schtasks /run /tn "ShennianAgent"',{stdio:"pipe",windowsHide:!0})}catch{}}}else try{n("sh",["-c","sleep 2; shennian start >/dev/null 2>&1"],{detached:!0,stdio:"ignore",env:l}).unref()}catch{}i(e,"SIGTERM")}function ce(t){t.command("upgrade").description("Upgrade shennian CLI to the latest version").option("--channel <channel>","Persist stable (npm latest) or preview (npm next); does not install").option("--check","Only check for updates, do not install").option("--to <version>","Upgrade to a specific version").option("--yes","Confirm a major version upgrade (required in non-interactive environments)").action(async r=>{if(r.channel!==void 0){if(r.to||r.yes)throw new Error("--channel cannot be combined with --to or --yes");if(G(r.channel),console.log(`Upgrade channel: ${x()} (saved; no installation)`),!r.check)return}console.log(`Upgrade channel: ${x()}`);const i=y();console.log(c.gray(`Current version: ${i}`)),console.log();let e=0;const l=setInterval(()=>{process.stdout.write(`\r  ${I[e%I.length]} Checking for updates...`),e++},80);let n;try{n=r.to??await k(),clearInterval(l),process.stdout.write("\r"+" ".repeat(40)+"\r")}catch(a){clearInterval(l),process.stdout.write("\r"+" ".repeat(40)+"\r"),console.error(c.red(`\u2717 Failed to check for updates: ${a instanceof Error?a.message:String(a)}`)),process.exit(1)}const o=v(i,n);if(o==="none"&&!r.to){console.log(c.green(`\u2713 Already up to date (${i})`));return}if(console.log(`  Available: ${c.green(n)} ${o==="major"?c.red("(major)"):o==="minor"?c.yellow("(minor)"):c.green("(patch)")}`),console.log(),!r.check){if(o==="major"&&(console.log(c.yellow("  \u26A0 This is a major version update and may contain breaking changes.")),console.log(c.gray("    Please review the changelog before upgrading.")),console.log(),!r.yes)){(!process.stdin.isTTY||!process.stdout.isTTY)&&(console.error(c.red("\u2717 Major upgrades require explicit confirmation; rerun with --yes.")),process.exit(1));const a=A({input:process.stdin,output:process.stdout}),p=await a.question(`  Type ${n} to confirm this major upgrade: `);if(a.close(),p.trim()!==n){console.log(c.gray("  Upgrade cancelled."));return}}try{await _({method:"upgrade.start",params:{targetVersion:n,confirmedMajor:o==="major"||r.yes===!0},timeoutMs:1e4})}catch(a){if(D({cleanupStale:!0}).running)throw a;const m=U(),u={fromVersion:i,targetVersion:n,oldDaemonPid:null,jobId:w()},g=R(u);await C(g,{fetchOwnedArtifact:T,sourceOrder:b()}),j(g,m.workerPath),S(u,{readiness:m})}console.log(),console.log(c.green(`\u2713 Upgrade scheduled ${i} \u2192 ${n}`)),console.log(c.gray("  The independent updater will verify, roll back if needed, and restart Shennian."))}})}async function de(t,r,i,e={}){const l=s=>{t.sendEvent({type:"event",event:"upgrade",payload:s})},n=e.currentVersion??y();let o;try{l({state:"checking",machineId:"self"}),o=i??await k()}catch(s){const d=`Version check failed: ${s instanceof Error?s.message:String(s)}`;return e.suppressResponse||t.sendRes({type:"res",id:r,ok:!1,error:d}),{ok:!1,error:d}}const a=v(n,o);if(a==="none")return e.suppressResponse||t.sendRes({type:"res",id:r,ok:!0,payload:{alreadyLatest:!0,version:n}}),{ok:!0,alreadyLatest:!0,version:n};if(!Y(a,e.confirmedMajor===!0))return e.suppressResponse||t.sendRes({type:"res",id:r,ok:!1,error:"Major upgrades require explicit second confirmation"}),{ok:!1,error:"Major upgrades require explicit second confirmation"};const p=w();let m;try{m=F();try{N(q("runtime","updater"))}catch(s){throw m.release(),s}}catch(s){const d=s instanceof Error?s.message:String(s);return e.suppressResponse||t.sendRes({type:"res",id:r,ok:!1,error:d}),{ok:!1,error:d}}const u={from:n,to:o,jobId:p},g=async()=>{let s=!1;try{const d=U(),f={fromVersion:n,targetVersion:o,oldDaemonPid:process.pid,jobId:p,requireHandoffAck:!0},h=R(f);if(l({state:"downloading",machineId:"self",version:o,jobId:p}),await C(h,{fetchOwnedArtifact:T,sourceOrder:b()}),j(h,d.workerPath),S(f,{readiness:d}),await H(h),e.canHandoff&&!e.canHandoff())throw new Error("upgrade_waiting_for_idle");const E=e.prepareForUpgrade?.(p)??{activeRunCount:0};return s=!0,L(h),l({state:"restarting",machineId:"self",...u,activeRunCount:E.activeRunCount}),{ok:!0,...u,resultPath:h.resultPath}}catch(d){s&&e.cancelUpgradePreparation?.();const f=d instanceof Error?d.message:String(d);return f!=="upgrade_waiting_for_idle"&&O(o,f),l({state:"failed",machineId:"self",error:f,jobId:p}),{ok:!1,error:f}}finally{m.release()}};return e.suppressResponse||t.sendRes({type:"res",id:r,ok:!0,payload:u}),e.waitForCompletion?g():(g(),{ok:!0,...u})}async function le(t,r,i={}){const e=await M(i.currentVersion);t.sendRes({type:"res",id:r,ok:!0,payload:e.hasUpdate?{hasUpdate:!0,current:e.current,latest:e.latest,changeType:e.changeType}:{hasUpdate:!1,current:i.currentVersion??y()}})}export{de as handleUpgradeStart,le as handleUpgradeStatus,Y as hasMajorUpgradeConfirmation,ce as registerUpgradeCommand,ie as restartCurrentDaemonAfterUpgrade};

package/dist/src/manager/runtime.d.ts

@@ -54,6 +54,7 @@ export declare class ManagerRuntimeService {
     setManagedAgentPolicy(sessionId: string, policy: CompiledManagedAgentPolicy | null): void;
     getManagedAgentPolicyForWorker(managerSessionId: string, workDir: string): CompiledManagedAgentPolicy | null;
     isNativeSkillManager(managerSessionId: string, workDir: string): boolean;
+    isManagedWorkerSession(sessionId: string): boolean;
     setManagerWorkerDefaults(sessionId: string, agentType?: AgentType | null, modelId?: string | null): void;
     getManagerWorkerDefaults(sessionId: string): {
         agentType: AgentType | null;

package/dist/src/manager/runtime.js

@@ -1,10 +1,10 @@
-import _ from"node:http";import{createHash as N,randomBytes as x,randomUUID as f}from"node:crypto";import h from"node:fs";import v from"node:os";import y from"node:path";import{ManagerRegistry as T}from"./registry.js";import{appendMessage as D}from"../session/store.js";import{requestSessionIndexPublish as q}from"../session/index-publisher.js";import{RetiredChannelRuntime as W}from"./retired-external-channel-compat.js";import{resolveShennianPath as P}from"../config/index.js";import{buildExternalChannelInstructions as O}from"../agents/external-channel-instructions.js";import{getPersonalSyncIdentity as I}from"../personal-sync/epoch.js";import{handleWorkerIpcRoute as E}from"./worker-ipc.js";import{deriveWorkerTitle as C,formatWorkerMarkdown as H}from"./worker-markdown.js";const B=Number(process.env.SHENNIAN_MANAGER_IPC_BODY_MAX_BYTES||2*1024*1024),m=12*6e4;let R=null;function ae(o){R=o}function oe(){return R}function c(o){return y.resolve(o||v.homedir())}function g(o,e,t){o.writeHead(e,{"content-type":"application/json; charset=utf-8"}),o.end(JSON.stringify(t))}function M(o){return o.replace(/\r\n/g,`
-`).trim()}function $(o){return o==="worker.final"?"completed":o==="worker.error"?"failed":o==="worker.aborted"?"stopped":"progress"}function U(o,e){return`manager-inbox:${N("sha256").update(o).update("\0").update(e.map(r=>r.id).join("\0")).digest("hex").slice(0,32)}`}async function j(o){const e=[];let t=0;for await(const r of o){const n=Buffer.isBuffer(r)?r:Buffer.from(r);if(t+=n.length,t>B)throw new Error("Request body too large");e.push(n)}return e.length===0?{}:JSON.parse(Buffer.concat(e).toString("utf8"))}function G(o,e,t,r){o.client.sendRes({type:"res",id:e,ok:t,...t?{payload:r}:{error:String(r.error||"unknown error")}})}function A(){return F()?y.join(v.tmpdir(),"shennian-vitest-runtime",String(process.pid),"manager-ipc.json"):P("runtime","manager-ipc.json")}function F(){return(process.env.VITEST==="true"||!!process.env.VITEST_WORKER_ID)&&!process.env.SHENNIAN_HOME?.trim()}class ue{opts;managedPolicies=new Map;registry=new T;channelRuntime;server=null;ipcUrl=null;ipcToken=x(24).toString("hex");healthTimer=null;startPromise=null;workerTextAcc=new Map;constructor(e){this.opts=e,this.channelRuntime=e.channelRuntime??new W}async start(){return this.startPromise?this.startPromise:(this.startPromise=this.doStart(),this.startPromise)}async doStart(){if(this.server)return;this.server=_.createServer((t,r)=>{this.handleIpc(t,r)}),this.server.requestTimeout=m,this.server.timeout=m,this.server.keepAliveTimeout=m,this.server.headersTimeout=m+5e3,await new Promise((t,r)=>{this.server.once("error",r),this.server.listen(0,"127.0.0.1",()=>t())});const e=this.server.address();typeof e=="object"&&e&&(this.ipcUrl=`http://127.0.0.1:${e.port}`,this.writeIpcRuntimeFile()),this.server.unref(),await this.channelRuntime.start(),this.healthTimer=setInterval(()=>this.scanWorkerHealth(),6e4),this.healthTimer.unref(),this.recoverPersistedInboxDeliveries()}async ready(){await this.start()}async stop(){this.healthTimer&&clearInterval(this.healthTimer),this.healthTimer=null,await this.channelRuntime.stop(),await new Promise(e=>{if(!this.server)return e();this.server.close(()=>e())}),this.server=null,this.ipcUrl=null,this.removeIpcRuntimeFile()}writeIpcRuntimeFile(){if(this.ipcUrl)try{const e=A();h.mkdirSync(y.dirname(e),{recursive:!0}),h.writeFileSync(e,JSON.stringify({url:this.ipcUrl,token:this.ipcToken,pid:process.pid,updatedAt:new Date().toISOString()},null,2),{mode:384}),h.chmodSync(e,384)}catch{}}removeIpcRuntimeFile(){try{const e=A(),t=JSON.parse(h.readFileSync(e,"utf-8"));(t.url===this.ipcUrl||t.token===this.ipcToken)&&h.unlinkSync(e)}catch{}}getInjectedEnv(e,t,r,n){return this.ipcUrl?{SHENNIAN_MANAGER_SESSION_ID:e,SHENNIAN_MANAGER_AGENT_SESSION_ID:t??"",SHENNIAN_MANAGER_WORKDIR:c(r),SHENNIAN_MANAGER_MODEL:n,SHENNIAN_MANAGER_IPC_URL:this.ipcUrl,SHENNIAN_MANAGER_IPC_TOKEN:this.ipcToken}:{}}registerManager(e){this.registry.upsertManager({...e,workDir:c(e.workDir),machineId:process.env.SHENNIAN_MACHINE_ID??null,managedRoomBound:e.managedRoomBound??(this.managedPolicies.has(e.sessionId)||void 0)})}setManagedAgentPolicy(e,t){if(t){this.managedPolicies.set(e,t);const r=this.registry.getManager(e);r&&!r.managedRoomBound&&this.registry.upsertManager({...r,managedRoomBound:!0})}else this.managedPolicies.delete(e)}getManagedAgentPolicyForWorker(e,t){const r=this.registry.getManager(e),n=this.managedPolicies.get(e);return!r||!n||c(t)!==r.workDir?null:n}isNativeSkillManager(e,t){const r=this.registry.getManager(e);return r?.controllerKind==="native_skill"&&c(t)===r.workDir}setManagerWorkerDefaults(e,t,r){const n=this.registry.getManager(e);n&&this.registry.upsertManager({...n,defaultWorkerAgentType:t??null,defaultWorkerModelId:r??null})}getManagerWorkerDefaults(e){const t=this.registry.getManager(e);return{agentType:t?.defaultWorkerAgentType??null,modelId:t?.defaultWorkerModelId??null}}noteManagerAgentSession(e,t,r,n){const s=this.registry.getManager(e);this.registry.upsertManager({sessionId:e,agentSessionId:t,workDir:c(r),machineId:process.env.SHENNIAN_MACHINE_ID??null,modelId:n,agentType:s?.agentType,providerModelId:s?.providerModelId})}noteAgentEvent(e,t){const r=this.findWorker(e);if(!r)return;const n={lastActivityAt:new Date().toISOString(),runId:t.runId};t.agentSessionId&&(n.agentSessionId=t.agentSessionId);const s=`${e}:${t.runId}`;let i=t.message||r.summary||"(\u65E0\u53EF\u89C1\u5185\u5BB9)";if(t.state==="delta"&&t.text&&!t.thinking){const u=(this.workerTextAcc.get(s)??"")+t.text;this.workerTextAcc.set(s,u);const a=M(u);a&&(n.summary=a.length>160?`${a.slice(0,160)}...`:a)}if(t.state==="final"||t.state==="error"||t.state==="aborted"){n.status=t.state;const u=M(this.workerTextAcc.get(s)??"");u&&(n.summary=u.length>240?`${u.slice(0,240)}...`:u,i=t.message||u),this.workerTextAcc.delete(s)}else t.state==="start"&&(n.status="running");const d=this.registry.updateWorker(e,n);d&&(t.state==="final"||t.state==="error"||t.state==="aborted")&&this.enqueueWorkerEvent(d.managedBy,d,t.state,t.runId,i)}findWorker(e){return this.registry.load().workers[e]}async handleAppReq(e){G(this.opts.getRuntime(),e.id,!1,{error:"feature_retired"})}async handleIpc(e,t){if(e.headers.authorization!==`Bearer ${this.ipcToken}`){g(t,401,{ok:!1,error:"Unauthorized"});return}try{const r=new URL(e.url??"/","http://127.0.0.1"),n=await j(e);if(r.pathname.startsWith("/channel/")||r.pathname.startsWith("/external/")||r.pathname.startsWith("/wechat-rpa/")){g(t,410,{ok:!1,error:"feature_retired"});return}const s=await E({registry:this.registry,getRuntime:this.opts.getRuntime,dispatchReq:this.opts.dispatchReq,dispatchChatEnqueue:this.dispatchChatEnqueue.bind(this),appendManagerVisibleMarkdown:this.appendManagerVisibleMarkdown.bind(this)},r.pathname,n,String(e.headers["x-shennian-manager-session-id"]||""));if(s){g(t,s.status,s.body);return}g(t,404,{ok:!1,error:`Unknown manager IPC path: ${r.pathname}`})}catch(r){g(t,400,{ok:!1,error:r instanceof Error?r.message:String(r)})}}async executeScopedWorkerAction(e,t,r){const n={"manager.sessions.list":"/sessions/list","manager.sessions.start":"/sessions/start","manager.sessions.send":"/sessions/send","manager.sessions.queue":"/sessions/queue","manager.sessions.queue.edit":"/sessions/queue/edit","manager.sessions.queue.delete":"/sessions/queue/delete","manager.sessions.stop":"/sessions/stop","manager.sessions.read":"/sessions/read","manager.memory.path":"/memory/path"},s=await E({registry:this.registry,getRuntime:this.opts.getRuntime,dispatchReq:this.opts.dispatchReq,dispatchChatEnqueue:this.dispatchChatEnqueue.bind(this),appendManagerVisibleMarkdown:this.appendManagerVisibleMarkdown.bind(this)},n[t],r,e);if(!s||s.status<200||s.status>=300)throw new Error(String(s?.body.error||"Managed Manager action failed"));return JSON.parse(JSON.stringify(s.body))}async dispatchChatEnqueue(e,t,r,n,s,i,d){if((await this.opts.dispatchReq({type:"req",id:`manager-enqueue-${f()}`,method:"chat.enqueue",params:{...I(),sessionId:e,text:n,agentType:t,workDir:r,agentSessionId:s,modelId:i,managedWorkerParentSessionId:d}}))?.durableQueueHandoff===!1)throw new Error("manager_worker_queue_handoff_unconfirmed")}appendManagerVisibleMarkdown(e,t,r){this.registry.getManager(e)?.controllerKind!=="native_skill"&&(D(e,{id:r,sessionId:e,role:"agent",ts:Date.now(),payload:t},"committed"),q(this.opts.getRuntime().client))}enqueueWorkerEvent(e,t,r,n,s){const i=this.registry.enqueueInboxEvent({managerSessionId:e,workerSessionId:t.sessionId,kind:r==="final"?"worker.final":r==="error"?"worker.error":"worker.aborted",priority:"normal",runId:n??null,summary:s||t.summary||"(\u65E0\u53EF\u89C1\u6458\u8981)"}),d=this.registry.getManager(e);i&&d?.controllerKind!=="native_skill"&&this.drainManagerInbox(e)}scanWorkerHealth(){const e=Date.now(),t=this.registry.load(),r=this.opts.getRuntime();for(const n of Object.values(t.workers)){if(n.status!=="running")continue;const s=e-Date.parse(n.createdAt);if(s<10*6e4)continue;const i=e-Date.parse(n.lastActivityAt),d=r.sessions.get(n.sessionId)?.currentRunId,u=!!r.chatQueue?.getSnapshot(n.sessionId).pending.length;if(!d){if(!u&&i>=10*6e4){const w="\u672A\u627E\u5230\u6D3B\u52A8\u4E2D\u7684 Provider Run \u6216\u5F85\u5904\u7406\u6D88\u606F\uFF1B\u5DF2\u5C06\u9057\u7559\u7684\u8FD0\u884C\u72B6\u6001\u6536\u655B\u4E3A\u5931\u8D25\u3002\u539F Worker \u4F1A\u8BDD\u4ECD\u4FDD\u7559\uFF0C\u53EF\u7EE7\u7EED\u53D1\u9001\u65B0\u6D88\u606F\u3002",p=this.registry.updateWorker(n.sessionId,{status:"error",lastActivityAt:new Date(e).toISOString(),summary:w});p&&this.enqueueWorkerEvent(p.managedBy,p,"error",n.runId??void 0,w)}continue}const a=n.healthNotifiedAt?Date.parse(n.healthNotifiedAt):0;if(a&&e-a<10*6e4)continue;const l=t.managers[n.managedBy];if(!l)continue;const S=Math.max(0,Math.floor(i/6e4)),b=M(n.runId?this.workerTextAcc.get(`${n.sessionId}:${n.runId}`)??"":""),k=`\u5DF2\u8FD0\u884C ${Math.floor(s/6e4)} \u5206\u949F\uFF0C\u5C1A\u672A\u7ED3\u675F\u3002
+import _ from"node:http";import{createHash as N,randomBytes as x,randomUUID as f}from"node:crypto";import h from"node:fs";import v from"node:os";import y from"node:path";import{ManagerRegistry as T}from"./registry.js";import{appendMessage as D}from"../session/store.js";import{requestSessionIndexPublish as W}from"../session/index-publisher.js";import{RetiredChannelRuntime as q}from"./retired-external-channel-compat.js";import{resolveShennianPath as P}from"../config/index.js";import{buildExternalChannelInstructions as O}from"../agents/external-channel-instructions.js";import{getPersonalSyncIdentity as I}from"../personal-sync/epoch.js";import{handleWorkerIpcRoute as E}from"./worker-ipc.js";import{deriveWorkerTitle as C,formatWorkerMarkdown as B}from"./worker-markdown.js";const H=Number(process.env.SHENNIAN_MANAGER_IPC_BODY_MAX_BYTES||2*1024*1024),m=12*6e4;let R=null;function ae(o){R=o}function oe(){return R}function c(o){return y.resolve(o||v.homedir())}function g(o,e,t){o.writeHead(e,{"content-type":"application/json; charset=utf-8"}),o.end(JSON.stringify(t))}function M(o){return o.replace(/\r\n/g,`
+`).trim()}function $(o){return o==="worker.final"?"completed":o==="worker.error"?"failed":o==="worker.aborted"?"stopped":"progress"}function U(o,e){return`manager-inbox:${N("sha256").update(o).update("\0").update(e.map(r=>r.id).join("\0")).digest("hex").slice(0,32)}`}async function j(o){const e=[];let t=0;for await(const r of o){const n=Buffer.isBuffer(r)?r:Buffer.from(r);if(t+=n.length,t>H)throw new Error("Request body too large");e.push(n)}return e.length===0?{}:JSON.parse(Buffer.concat(e).toString("utf8"))}function G(o,e,t,r){o.client.sendRes({type:"res",id:e,ok:t,...t?{payload:r}:{error:String(r.error||"unknown error")}})}function S(){return F()?y.join(v.tmpdir(),"shennian-vitest-runtime",String(process.pid),"manager-ipc.json"):P("runtime","manager-ipc.json")}function F(){return(process.env.VITEST==="true"||!!process.env.VITEST_WORKER_ID)&&!process.env.SHENNIAN_HOME?.trim()}class ue{opts;managedPolicies=new Map;registry=new T;channelRuntime;server=null;ipcUrl=null;ipcToken=x(24).toString("hex");healthTimer=null;startPromise=null;workerTextAcc=new Map;constructor(e){this.opts=e,this.channelRuntime=e.channelRuntime??new q}async start(){return this.startPromise?this.startPromise:(this.startPromise=this.doStart(),this.startPromise)}async doStart(){if(this.server)return;this.server=_.createServer((t,r)=>{this.handleIpc(t,r)}),this.server.requestTimeout=m,this.server.timeout=m,this.server.keepAliveTimeout=m,this.server.headersTimeout=m+5e3,await new Promise((t,r)=>{this.server.once("error",r),this.server.listen(0,"127.0.0.1",()=>t())});const e=this.server.address();typeof e=="object"&&e&&(this.ipcUrl=`http://127.0.0.1:${e.port}`,this.writeIpcRuntimeFile()),this.server.unref(),await this.channelRuntime.start(),this.healthTimer=setInterval(()=>this.scanWorkerHealth(),6e4),this.healthTimer.unref(),this.recoverPersistedInboxDeliveries()}async ready(){await this.start()}async stop(){this.healthTimer&&clearInterval(this.healthTimer),this.healthTimer=null,await this.channelRuntime.stop(),await new Promise(e=>{if(!this.server)return e();this.server.close(()=>e())}),this.server=null,this.ipcUrl=null,this.removeIpcRuntimeFile()}writeIpcRuntimeFile(){if(this.ipcUrl)try{const e=S();h.mkdirSync(y.dirname(e),{recursive:!0}),h.writeFileSync(e,JSON.stringify({url:this.ipcUrl,token:this.ipcToken,pid:process.pid,updatedAt:new Date().toISOString()},null,2),{mode:384}),h.chmodSync(e,384)}catch{}}removeIpcRuntimeFile(){try{const e=S(),t=JSON.parse(h.readFileSync(e,"utf-8"));(t.url===this.ipcUrl||t.token===this.ipcToken)&&h.unlinkSync(e)}catch{}}getInjectedEnv(e,t,r,n){return this.ipcUrl?{SHENNIAN_MANAGER_SESSION_ID:e,SHENNIAN_MANAGER_AGENT_SESSION_ID:t??"",SHENNIAN_MANAGER_WORKDIR:c(r),SHENNIAN_MANAGER_MODEL:n,SHENNIAN_MANAGER_IPC_URL:this.ipcUrl,SHENNIAN_MANAGER_IPC_TOKEN:this.ipcToken}:{}}registerManager(e){this.registry.upsertManager({...e,workDir:c(e.workDir),machineId:process.env.SHENNIAN_MACHINE_ID??null,managedRoomBound:e.managedRoomBound??(this.managedPolicies.has(e.sessionId)||void 0)})}setManagedAgentPolicy(e,t){if(t){this.managedPolicies.set(e,t);const r=this.registry.getManager(e);r&&!r.managedRoomBound&&this.registry.upsertManager({...r,managedRoomBound:!0})}else this.managedPolicies.delete(e)}getManagedAgentPolicyForWorker(e,t){const r=this.registry.getManager(e),n=this.managedPolicies.get(e);return!r||!n||c(t)!==r.workDir?null:n}isNativeSkillManager(e,t){const r=this.registry.getManager(e);return r?.controllerKind==="native_skill"&&c(t)===r.workDir}isManagedWorkerSession(e){return!!this.findWorker(e)}setManagerWorkerDefaults(e,t,r){const n=this.registry.getManager(e);n&&this.registry.upsertManager({...n,defaultWorkerAgentType:t??null,defaultWorkerModelId:r??null})}getManagerWorkerDefaults(e){const t=this.registry.getManager(e);return{agentType:t?.defaultWorkerAgentType??null,modelId:t?.defaultWorkerModelId??null}}noteManagerAgentSession(e,t,r,n){const s=this.registry.getManager(e);this.registry.upsertManager({sessionId:e,agentSessionId:t,workDir:c(r),machineId:process.env.SHENNIAN_MACHINE_ID??null,modelId:n,agentType:s?.agentType,providerModelId:s?.providerModelId})}noteAgentEvent(e,t){const r=this.findWorker(e);if(!r)return;const n={lastActivityAt:new Date().toISOString(),runId:t.runId};t.agentSessionId&&(n.agentSessionId=t.agentSessionId);const s=`${e}:${t.runId}`;let i=t.message||r.summary||"(\u65E0\u53EF\u89C1\u5185\u5BB9)";if(t.state==="delta"&&t.text&&!t.thinking){const u=(this.workerTextAcc.get(s)??"")+t.text;this.workerTextAcc.set(s,u);const a=M(u);a&&(n.summary=a.length>160?`${a.slice(0,160)}...`:a)}if(t.state==="final"||t.state==="error"||t.state==="aborted"){n.status=t.state;const u=M(this.workerTextAcc.get(s)??"");u&&(n.summary=u.length>240?`${u.slice(0,240)}...`:u,i=t.message||u),this.workerTextAcc.delete(s)}else t.state==="start"&&(n.status="running");const d=this.registry.updateWorker(e,n);d&&(t.state==="final"||t.state==="error"||t.state==="aborted")&&this.enqueueWorkerEvent(d.managedBy,d,t.state,t.runId,i)}findWorker(e){return this.registry.load().workers[e]}async handleAppReq(e){G(this.opts.getRuntime(),e.id,!1,{error:"feature_retired"})}async handleIpc(e,t){if(e.headers.authorization!==`Bearer ${this.ipcToken}`){g(t,401,{ok:!1,error:"Unauthorized"});return}try{const r=new URL(e.url??"/","http://127.0.0.1"),n=await j(e);if(r.pathname.startsWith("/channel/")||r.pathname.startsWith("/external/")||r.pathname.startsWith("/wechat-rpa/")){g(t,410,{ok:!1,error:"feature_retired"});return}const s=await E({registry:this.registry,getRuntime:this.opts.getRuntime,dispatchReq:this.opts.dispatchReq,dispatchChatEnqueue:this.dispatchChatEnqueue.bind(this),appendManagerVisibleMarkdown:this.appendManagerVisibleMarkdown.bind(this)},r.pathname,n,String(e.headers["x-shennian-manager-session-id"]||""));if(s){g(t,s.status,s.body);return}g(t,404,{ok:!1,error:`Unknown manager IPC path: ${r.pathname}`})}catch(r){g(t,400,{ok:!1,error:r instanceof Error?r.message:String(r)})}}async executeScopedWorkerAction(e,t,r){const n={"manager.sessions.list":"/sessions/list","manager.sessions.start":"/sessions/start","manager.sessions.send":"/sessions/send","manager.sessions.queue":"/sessions/queue","manager.sessions.queue.edit":"/sessions/queue/edit","manager.sessions.queue.delete":"/sessions/queue/delete","manager.sessions.stop":"/sessions/stop","manager.sessions.read":"/sessions/read","manager.memory.path":"/memory/path"},s=await E({registry:this.registry,getRuntime:this.opts.getRuntime,dispatchReq:this.opts.dispatchReq,dispatchChatEnqueue:this.dispatchChatEnqueue.bind(this),appendManagerVisibleMarkdown:this.appendManagerVisibleMarkdown.bind(this)},n[t],r,e);if(!s||s.status<200||s.status>=300)throw new Error(String(s?.body.error||"Managed Manager action failed"));return JSON.parse(JSON.stringify(s.body))}async dispatchChatEnqueue(e,t,r,n,s,i,d){if((await this.opts.dispatchReq({type:"req",id:`manager-enqueue-${f()}`,method:"chat.enqueue",params:{...I(),sessionId:e,text:n,agentType:t,workDir:r,agentSessionId:s,modelId:i,managedWorkerParentSessionId:d}}))?.durableQueueHandoff===!1)throw new Error("manager_worker_queue_handoff_unconfirmed")}appendManagerVisibleMarkdown(e,t,r){this.registry.getManager(e)?.controllerKind!=="native_skill"&&(D(e,{id:r,sessionId:e,role:"agent",ts:Date.now(),payload:t},"committed"),W(this.opts.getRuntime().client))}enqueueWorkerEvent(e,t,r,n,s){const i=this.registry.enqueueInboxEvent({managerSessionId:e,workerSessionId:t.sessionId,kind:r==="final"?"worker.final":r==="error"?"worker.error":"worker.aborted",priority:"normal",runId:n??null,summary:s||t.summary||"(\u65E0\u53EF\u89C1\u6458\u8981)"}),d=this.registry.getManager(e);i&&d?.controllerKind!=="native_skill"&&this.drainManagerInbox(e)}scanWorkerHealth(){const e=Date.now(),t=this.registry.load(),r=this.opts.getRuntime();for(const n of Object.values(t.workers)){if(n.status!=="running")continue;const s=e-Date.parse(n.createdAt);if(s<10*6e4)continue;const i=e-Date.parse(n.lastActivityAt),d=r.sessions.get(n.sessionId)?.currentRunId,u=!!r.chatQueue?.getSnapshot(n.sessionId).pending.length;if(!d){if(!u&&i>=10*6e4){const w="\u672A\u627E\u5230\u6D3B\u52A8\u4E2D\u7684 Provider Run \u6216\u5F85\u5904\u7406\u6D88\u606F\uFF1B\u5DF2\u5C06\u9057\u7559\u7684\u8FD0\u884C\u72B6\u6001\u6536\u655B\u4E3A\u5931\u8D25\u3002\u539F Worker \u4F1A\u8BDD\u4ECD\u4FDD\u7559\uFF0C\u53EF\u7EE7\u7EED\u53D1\u9001\u65B0\u6D88\u606F\u3002",p=this.registry.updateWorker(n.sessionId,{status:"error",lastActivityAt:new Date(e).toISOString(),summary:w});p&&this.enqueueWorkerEvent(p.managedBy,p,"error",n.runId??void 0,w)}continue}const a=n.healthNotifiedAt?Date.parse(n.healthNotifiedAt):0;if(a&&e-a<10*6e4)continue;const l=t.managers[n.managedBy];if(!l)continue;const A=Math.max(0,Math.floor(i/6e4)),b=M(n.runId?this.workerTextAcc.get(`${n.sessionId}:${n.runId}`)??"":""),k=`\u5DF2\u8FD0\u884C ${Math.floor(s/6e4)} \u5206\u949F\uFF0C\u5C1A\u672A\u7ED3\u675F\u3002
 
 \u5F53\u524D\u53EF\u89C1\u8FDB\u5C55\uFF1A
 - \u6700\u8FD1\u6587\u672C\uFF1A${b||n.summary||"(\u65E0\u53EF\u89C1\u5185\u5BB9)"}
 - \u6700\u8FD1\u6D3B\u52A8\u65F6\u95F4\uFF1A${n.lastActivityAt}
-- \u6700\u8FD1 ${S} \u5206\u949F\u6CA1\u6709\u65B0\u6D3B\u52A8\u3002`;this.registry.updateWorker(n.sessionId,{healthNotifiedAt:new Date(e).toISOString(),lastHealthSummary:k}),this.registry.enqueueInboxEvent({managerSessionId:l.sessionId,workerSessionId:n.sessionId,kind:"worker.health",priority:"normal",runId:n.runId??null,summary:k})&&l.controllerKind!=="native_skill"&&this.drainManagerInbox(l.sessionId)}}async interruptAndResumeManager(e,t,r){this.opts.getRuntime().sessions.get(e.sessionId)?.currentRunId&&await this.opts.dispatchReq({type:"req",id:`manager-interrupt-${f()}`,method:"chat.abort",params:{...I(),sessionId:e.sessionId}}),await this.dispatchManagerContinuation(e,t,!1,r)}bindManagerAdapterEvents(e,t){t.on("agentEvent",r=>{if(r.state==="start"&&r.agentSessionId){const n=this.registry.getManager(e);n&&this.noteManagerAgentSession(e,r.agentSessionId,n.workDir,n.modelId)}r.state==="start"&&(this.acknowledgeManagerInboxDelivery(e),this.updateManagerStatus(e,"running")),(r.state==="final"||r.state==="error"||r.state==="aborted")&&(this.updateManagerStatus(e,"idle"),queueMicrotask(()=>{this.drainManagerInbox(e)}))})}updateManagerStatus(e,t){const r=this.registry.getManager(e);r&&this.registry.upsertManager({...r,status:t})}async drainManagerInbox(e){const t=this.registry.getManager(e);if(!t||t.deliveringInboxEventIds?.length)return;const r=this.registry.listInboxEvents(e);if(!r.length)return;const n=r.find(a=>a.priority==="high"),s=this.opts.getRuntime().sessions.get(e);if(t.status!=="idle"&&!n||s?.currentRunId&&!n)return;const i=n?[n]:r.slice(0,20),d=i.map(a=>{const l=this.registry.getWorkerForManager(e,a.workerSessionId);return H({kind:$(a.kind),sessionId:a.workerSessionId,title:l?.title||C(l?.userGoal||l?.summary||a.summary),body:a.summary})}).join(`
+- \u6700\u8FD1 ${A} \u5206\u949F\u6CA1\u6709\u65B0\u6D3B\u52A8\u3002`;this.registry.updateWorker(n.sessionId,{healthNotifiedAt:new Date(e).toISOString(),lastHealthSummary:k}),this.registry.enqueueInboxEvent({managerSessionId:l.sessionId,workerSessionId:n.sessionId,kind:"worker.health",priority:"normal",runId:n.runId??null,summary:k})&&l.controllerKind!=="native_skill"&&this.drainManagerInbox(l.sessionId)}}async interruptAndResumeManager(e,t,r){this.opts.getRuntime().sessions.get(e.sessionId)?.currentRunId&&await this.opts.dispatchReq({type:"req",id:`manager-interrupt-${f()}`,method:"chat.abort",params:{...I(),sessionId:e.sessionId}}),await this.dispatchManagerContinuation(e,t,!1,r)}bindManagerAdapterEvents(e,t){t.on("agentEvent",r=>{if(r.state==="start"&&r.agentSessionId){const n=this.registry.getManager(e);n&&this.noteManagerAgentSession(e,r.agentSessionId,n.workDir,n.modelId)}r.state==="start"&&(this.acknowledgeManagerInboxDelivery(e),this.updateManagerStatus(e,"running")),(r.state==="final"||r.state==="error"||r.state==="aborted")&&(this.updateManagerStatus(e,"idle"),queueMicrotask(()=>{this.drainManagerInbox(e)}))})}updateManagerStatus(e,t){const r=this.registry.getManager(e);r&&this.registry.upsertManager({...r,status:t})}async drainManagerInbox(e){const t=this.registry.getManager(e);if(!t||t.deliveringInboxEventIds?.length)return;const r=this.registry.listInboxEvents(e);if(!r.length)return;const n=r.find(a=>a.priority==="high"),s=this.opts.getRuntime().sessions.get(e);if(t.status!=="idle"&&!n||s?.currentRunId&&!n)return;const i=n?[n]:r.slice(0,20),d=i.map(a=>{const l=this.registry.getWorkerForManager(e,a.workerSessionId);return B({kind:$(a.kind),sessionId:a.workerSessionId,title:l?.title||C(l?.userGoal||l?.summary||a.summary),body:a.summary})}).join(`
 
 ---
 

package/dist/src/session/queue.js

@@ -1 +1 @@
-import{NativeSourceDeletionStore as U}from"../native-fusion/native-source-deletions.js";import{QueueRetryScheduler as J}from"./queue-retry.js";import{findDuplicateQueuedMessage as E,stripLocalQueueFields as v,takeCompatibleBatch as G,formatQueuedBatch as Q,dedupeQueuedAttachments as B,classifyQueueError as F,isControllableQueueMessage as T,takeControllableMessage as H,normalizeQueuedModelId as q,resolveManagedRoomNativeSessionId as L}from"./queue-helpers.js";import{randomUUID as V}from"node:crypto";import{CODEX_DESKTOP_SEND_UNAVAILABLE as X,buildUserMessagePayload as Y}from"@shennian/wire";import{isCodexThreadTailInterrupted as Z}from"../native-fusion/codex-files.js";import{mergeProjectedSessions as ee}from"./projection.js";import{parsePersonalAttachmentReferences as x}from"./attachment-reference.js";import{assertFinishedAttachmentReferences as W,markAttachmentReferences as b}from"./attachment-transfer-store.js";import{getPersonalSyncIdentity as R}from"../personal-sync/epoch.js";import{getDaemonInstallationId as D}from"../personal-sync/identity.js";import{appendMessage as j,getMessageDeliveryStatus as _,listSessionRecords as N,recordSession as z,tombstoneMessage as $,upsertMessage as C}from"./store.js";import{requestSessionIndexPublish as S}from"./index-publisher.js";import{purgeQueuedSessionData as te,readQueue as p,writeQueue as M}from"./queue-storage.js";import{getNativeSourceAgentType as se}from"./handlers/chat-events.js";import{purgeQueuedSessionData as Se}from"./queue-storage.js";class A extends Error{constructor(e){super(e),this.name="TerminalQueuedChatDispatchError"}}function P(){return new Date().toISOString()}function O(u){const e=P(),s=u.managedWorkerParentSessionId?{kind:"manager_worker",parentSessionId:u.managedWorkerParentSessionId}:u.managerEventId?{kind:"manager_event",eventId:u.managerEventId}:u.managedAgentDirect?{kind:"managed_direct"}:{kind:"personal"};return{...R(D()),id:u.queueMessageId||u.clientMessageId||`queue-${V()}`,sessionId:u.sessionId,text:u.text,agentType:u.agentType,sessionMode:u.sessionMode,managerConfig:u.managerConfig,workDir:u.workDir,agentSessionId:u.agentSessionId??null,modelId:q(u.modelId)??null,systemPrompt:u.systemPrompt??null,managerDefaultWorkerAgentType:u.managerDefaultWorkerAgentType??null,managerDefaultWorkerModelId:u.managerDefaultWorkerModelId??null,reasoningEffort:u.reasoningEffort??null,clientMessageId:u.clientMessageId??null,attachments:x(u.attachments),externalChannel:u.externalChannel??null,replyTarget:u.replyTarget??null,origin:u.origin,createdAt:e,updatedAt:e,trigger:s}}function w(u){return{id:u.clientMessageId??u.id,sessionId:u.sessionId,role:u.trigger?.kind==="manager_event"?"agent":"user",ts:Date.parse(u.createdAt),payload:Y(u.text,u.attachments)}}class ye{opts;draining=new Set;promoting=new Set;disposed=!1;retries=new J(e=>{this.wakeSession(e)});deletions=new U;inFlightMessageIds=new Map;deferredLogState=new Map;prepareManagedRoomTurn=null;prepareManagedAgentDirectTurn=null;constructor(e){this.opts=e,setTimeout(()=>this.drainIdleQueues(),0).unref?.()}stop(){this.disposed=!0,this.retries.stop()}getSnapshot(e){const s=p().sessions[e]??[],t=this.inFlightMessageIds.get(e);return{...R(D()),sessionId:e,busy:!!this.opts.getRuntime().sessions.get(e)?.currentRunId,pending:s.map(v),controllablePending:s.filter(o=>T(o)&&!t?.has(o.id)).map(v)}}setManagedRoomTurnPreparer(e){this.prepareManagedRoomTurn=e}setManagedAgentDirectTurnPreparer(e){this.prepareManagedAgentDirectTurn=e}async enqueueManagedRoomBatch(e){if(e.length===0)return;const s=this.opts.getRuntime(),t=p(),o=new Set,i=new Map;let n=!1;for(const r of e){const a=r.params;if(this.deletions.hasSession(a.sessionId)){delete t.sessions[a.sessionId],n=!0;continue}if(!a.sessionId||!a.text||!a.agentType||!a.workDir)throw new Error("sessionId, text, agentType and workDir are required");const g=O({...R(D()),...a});g.origin="room",g.managedRoom=r.context,g.trigger={kind:"room_activation",...r.context};const c=t.sessions[a.sessionId]??[],h=E(c,g),I=g.clientMessageId??g.id,m=_(a.sessionId,I);if(!h&&m.state==="not_found")c.push(g),t.sessions[a.sessionId]=c,i.set(`${g.sessionId}\0${I}`,g),o.add(a.sessionId),n=!0;else if(h&&m.state==="not_found")i.set(`${h.sessionId}\0${I}`,h),o.add(a.sessionId);else if(h&&m.state==="queued")o.add(a.sessionId);else if(h){const l=c.filter(k=>k.id!==h.id);l.length?t.sessions[a.sessionId]=l:delete t.sessions[a.sessionId],n=!0}}n&&M(t);let d;for(const r of o)try{for(const a of i.values())a.sessionId===r&&(j(r,w(a),"queued"),z({sessionId:r,agentType:a.agentType,sessionMode:a.sessionMode,managerConfig:a.managerConfig,workDir:a.workDir,agentSessionId:L(s,a),modelId:a.modelId}));this.logLifecycle("enqueue.persisted",r,{origin:"room",pendingCount:p().sessions[r]?.length??0}),this.broadcast(r),this.wakeSession(r)}catch(a){d=a,this.logLifecycle("enqueue.commit-failed",r,{failureClass:F(a instanceof Error?a.message:"","canonical_commit_failed")},"error"),this.retries.schedule(r,!0)}if(o.size&&S(s.client),d)throw d}async handleEnqueue(e){const s=this.opts.getRuntime(),t=e.params;if(ee(t.sessionListProjection),!t.sessionId||!t.text||!t.agentType||!t.workDir)return s.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId, text, agentType and workDir are required"}),!1;const o=t.managedAgentDirect?await this.prepareManagedAgentDirectTurn?.(t.sessionId):null;if(t.managedAgentDirect&&!o)throw new Error("managed_agent_context_unavailable");const i=o?N().find(K=>K.sessionId===t.sessionId):null,n=o?{...t,...o,agentType:o.agentType,sessionMode:o.sessionMode,managerConfig:o.managerConfig,workDir:o.workDir,agentSessionId:o.agentSessionId,modelId:i?.modelId??void 0,systemPrompt:o.systemPrompt,managerDefaultWorkerAgentType:o.managerDefaultWorkerAgentType,managerDefaultWorkerModelId:o.managerDefaultWorkerModelId}:t,d=s.resolvePath(n.workDir),r=x(n.attachments);r?.length&&W(d,r);const a={text:n.text,attachments:r,localized:!1},g=s.sessions.get(n.sessionId);if(n.agentType==="codex"&&n.agentSessionId&&g?.currentRunId?.startsWith("dispatch:")&&Z(n.agentSessionId))return s.client.sendRes({type:"res",id:e.id,ok:!1,error:X}),!1;o?.managedRoomContextToken&&s.revokeManagedRoomContextToken?.(o.managedRoomContextToken);const c=await this.isSessionBusy(n.sessionId),h=p(),I=h.sessions[n.sessionId]??[],m=O({...n,text:a.text,attachments:a.attachments,systemPrompt:o?null:n.systemPrompt});o&&(m.managedAgentDirect=!0,m.trigger?.kind!=="manager_event"&&(m.trigger={kind:"managed_direct"}));const l=m.clientMessageId??m.id,k=_(n.sessionId,l);if(k.state!=="not_found")return s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:k.state==="queued",queueMessageId:m.id,...k,queue:this.getSnapshot(n.sessionId)}}),!0;const y=E(I,m);y||(h.sessions[n.sessionId]=[...I,m],M(h));const f=y??m;return z({sessionId:f.sessionId,agentType:f.agentType,sessionMode:f.sessionMode,managerConfig:f.managerConfig,workDir:f.workDir,agentSessionId:f.agentSessionId,modelId:f.modelId}),j(f.sessionId,w(f),"queued"),b(s.resolvePath(f.workDir),f.attachments??[]),S(s.client),this.logLifecycle("enqueue.persisted",n.sessionId,{origin:n.origin??"app",queueMessageId:f.id,pendingCount:(p().sessions[n.sessionId]??[]).length}),this.broadcast(n.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:!0,queueMessageId:y?.id??m.id,..._(n.sessionId,y?.clientMessageId??y?.id??m.clientMessageId??m.id),queue:this.getSnapshot(n.sessionId),...a.localized?{localizedAttachments:!0}:{}}}),c?this.retries.schedule(n.sessionId):this.wakeSession(n.sessionId),!0}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)}})}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 o=p(),i=o.sessions[t.sessionId]??[],n=i.findIndex(a=>a.id===t.queueMessageId);if(n<0){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}if(!T(i[n])||this.inFlightMessageIds.get(t.sessionId)?.has(t.queueMessageId)){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const d=x(t.attachments),r=s.resolvePath(i[n].workDir);d?.length&&W(r,d),i[n]={...i[n],text:t.text,attachments:d,updatedAt:P()},o.sessions[t.sessionId]=i,M(o),C(t.sessionId,w(i[n]),"queued"),i[n].attachments?.length&&b(r,i[n].attachments??[]),S(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 o=p(),i=o.sessions[t.sessionId]??[],n=i.find(r=>r.id===t.queueMessageId);if(n&&(!T(n)||this.inFlightMessageIds.get(t.sessionId)?.has(t.queueMessageId))){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const d=i.filter(r=>r.id!==t.queueMessageId);if(d.length===i.length){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}d.length?o.sessions[t.sessionId]=d:delete o.sessions[t.sessionId],M(o),$(t.sessionId,n?.clientMessageId??n?.id??t.queueMessageId),S(s.client),this.broadcast(t.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}})}async handlePromote(e){const s=this.opts.getRuntime(),t=e.params,o=n=>s.client.sendRes({type:"res",id:e.id,ok:!1,error:n});if(!t.sessionId||!t.queueMessageId){o("sessionId and queueMessageId are required");return}if(this.promoting.has(t.sessionId)){o("queue_promotion_in_progress");return}this.promoting.add(t.sessionId);let i=[];try{for(;this.draining.has(t.sessionId);)await new Promise(r=>setTimeout(r,5));if(i=H(p().sessions[t.sessionId]??[],t.queueMessageId),!i.length){o("Queued message not found");return}this.inFlightMessageIds.set(t.sessionId,new Set(i.map(r=>r.id)));const n=s.sessions.get(t.sessionId);let d="idle";if(n?.currentRunId)if(this.canInjectActiveBatch(n,i)){const r=i.length===1?i[0].text:Q(i),a=i[0].clientMessageId??i[0].id;s.nativeFusion?.registerManagedSend({sessionId:t.sessionId,agentType:i[0].agentType,sourceAgentType:se(i[0].agentType,i[0].modelId),canonicalMessageId:a,sourceSessionKey:n.agentSessionId,text:r,managedEchoPolicy:"suppress_following"});const g=Date.now();try{await n.adapter.injectActiveInput({text:r,attachments:B(i.flatMap(c=>c.attachments??[])),clientMessageId:a})}catch(c){throw s.nativeFusion?.cancelManagedSend(a),c}for(const c of i)C(t.sessionId,{...w(c),ts:g},"executing");this.removeBatch(t.sessionId,i),d="injected"}else{if(!this.opts.interruptQueuedSession)throw new Error("queue_interrupt_unavailable");await this.opts.interruptQueuedSession(t.sessionId),await this.dispatchQueuedMessages(i),this.removeBatch(t.sessionId,i),d="restarted"}else await this.dispatchQueuedMessages(i),this.removeBatch(t.sessionId,i);S(s.client),this.broadcast(t.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId),promotedMessageIds:i.map(r=>r.id),mode:d}})}catch(n){o(n instanceof Error?n.message:"queue_promotion_failed"),this.broadcast(t.sessionId)}finally{this.inFlightMessageIds.delete(t.sessionId),this.promoting.delete(t.sessionId),p().sessions[t.sessionId]?.length&&this.wakeSession(t.sessionId)}}noteTerminal(e){this.wakeSession(e)}purgeSession(e){te(e),this.retries.cancel(e)}discardPendingManagedRoomMessages(e){if(this.deletions.hasSession(e)){const d=p().sessions[e]?.length??0;return this.purgeSession(e),d}const s=p(),t=s.sessions[e]??[],o=this.inFlightMessageIds.get(e)??new Set,i=t.filter(d=>d.managedRoom&&!o.has(d.id));if(i.length===0)return 0;const n=t.filter(d=>!i.includes(d));n.length?s.sessions[e]=n:delete s.sessions[e],M(s);for(const d of i)$(e,d.clientMessageId??d.id);return S(this.opts.getRuntime().client),this.logLifecycle("managed-room.discarded-paused",e,{discardedCount:i.length,pendingCount:n.length}),this.broadcast(e),i.length}wakeSession(e){this.drainNext(e).catch(()=>{this.logLifecycle("drain.retry-required",e,{},"error"),this.retries.schedule(e,!0)})}drainIdleQueues(){if(this.disposed)return;const e=p();for(const s of Object.keys(e.sessions))this.wakeSession(s)}async drainNext(e){if(this.disposed||this.draining.has(e)||this.promoting.has(e))return;this.draining.add(e);let s=!1,t=!1,o=[];try{if(this.deletions.hasSession(e)){this.purgeSession(e);return}if(await this.isSessionBusy(e)){const g=p().sessions[e]?.length??0,c=this.deferredLogState.get(e),h=Date.now();(!c||c.pendingCount!==g||h-c.at>=3e4)&&(this.logLifecycle("drain.deferred",e,{pendingCount:g}),this.deferredLogState.set(e,{at:h,pendingCount:g}));return}this.deferredLogState.delete(e);const i=p(),n=i.sessions[e]??[],d=n.filter(g=>{const c=g.clientMessageId??g.id,h=_(e,c).state;return h==="queued"||h==="not_found"});d.length!==n.length&&(d.length?i.sessions[e]=d:delete i.sessions[e],M(i),this.logLifecycle("drain.pruned-terminal",e,{prunedCount:n.length-d.length,pendingCount:d.length}));const r=G(d);if(r.length===0){this.broadcast(e);return}o=r,this.inFlightMessageIds.set(e,new Set(r.map(g=>g.id))),this.logLifecycle("dispatch.start",e,{batchCount:r.length,queueMessageId:r[0].id,origin:r[0].origin??"app"});for(const g of r)C(e,w(g),"queued");await this.dispatchQueuedMessages(r);const a=this.removeBatch(e,r);this.logLifecycle("dispatch.accepted",e,{batchCount:r.length,queueMessageId:r[0].id,pendingCount:a}),this.deferredLogState.delete(e),s=a>0,this.broadcast(e)}catch(i){const n=F(i instanceof Error?i.message:String(i),i instanceof A?"terminal_dispatch_failed":"transient_dispatch_failed");if(i instanceof A||n==="managed_claim_source_binding_conflict"){const r=this.removeBatch(e,o);this.logLifecycle("dispatch.terminal-failure",e,{batchCount:o.length,queueMessageId:o[0]?.id,pendingCount:r,failureClass:n},"error"),s=r>0}else t=!0,this.logLifecycle("dispatch.transient-failure",e,{failureClass:n},"error");this.broadcast(e)}finally{this.inFlightMessageIds.delete(e),this.draining.delete(e),p().sessions[e]?.length?s?queueMicrotask(()=>this.wakeSession(e)):this.retries.schedule(e,t):this.retries.cancel(e)}}canInjectActiveBatch(e,s){if(!e.adapter.supportsActiveInput||!e.adapter.injectActiveInput||!e.journalRunId)return!1;const t=this.opts.getRuntime().runJournal?.get(s[0].sessionId,e.journalRunId),o=s[0].trigger?.kind==="managed_direct"||s[0].managedAgentDirect?null:"personal";return!!(o&&s[0].systemPrompt==null&&t&&t.source===o&&t.deliveryMode==="direct"&&t.agentType===s[0].agentType&&t.workDir===e.workDir&&t.agentSessionId===e.agentSessionId&&t.modelId===(s[0].modelId??null)&&t.reasoningEffort===(s[0].reasoningEffort??null))}removeBatch(e,s){if(s.length===0)return p().sessions[e]?.length??0;const t=p(),o=new Set(s.map(n=>n.id)),i=(t.sessions[e]??[]).filter(n=>!o.has(n.id));return i.length?t.sessions[e]=i:delete t.sessions[e],M(t),i.length}logLifecycle(e,s,t,o="info"){const i=`[chat.queue] ${JSON.stringify({at:P(),event:e,sessionId:s,...t})}`;o==="error"?console.error(i):console.info(i)}async dispatchQueuedMessages(e){const s=e[0],t=e.flatMap(l=>{if(l.trigger?.kind==="room_activation"){const{kind:k,...y}=l.trigger;return[y]}return l.managedRoom?[l.managedRoom]:[]}),o=e.some(l=>l.trigger?.kind==="managed_direct"||!!l.managedAgentDirect),i=e.find(l=>l.trigger?.kind==="manager_event")?.trigger,n=e.find(l=>l.trigger?.kind==="manager_worker")?.trigger,d=t.length>0?await this.prepareManagedRoomTurn?.(t):o?await this.prepareManagedAgentDirectTurn?.(s.sessionId):void 0;if(t.length>0&&!d)throw new Error("managed_room_turn_preparer_unavailable");if(o&&!d)throw new Error("managed_agent_context_unavailable");const r=new Set((p().sessions[s.sessionId]??[]).map(l=>l.id));if(this.disposed||this.deletions.hasSession(s.sessionId)||e.some(l=>!r.has(l.id)))throw d?.managedRoomContextToken&&this.opts.getRuntime().revokeManagedRoomContextToken?.(d.managedRoomContextToken),this.disposed?new Error("queue_stopped"):new A("queued_dispatch_cancelled");const a=d??null,g=a?.sessionMode==="manager"?a.managerConfig:null,c=e.length===1?s.text:Q(e),h=B(e.flatMap(l=>l.attachments??[])),I=o?d.agentSessionId:t.length>0?L(this.opts.getRuntime(),s):s.agentSessionId??null,m=d?N().find(l=>l.sessionId===s.sessionId):null;await this.opts.dispatchQueuedReq({type:"req",id:`queue-send-${s.id}-${Date.now()}`,method:"chat.send",params:{...R(D()),sessionId:s.sessionId,text:c,agentType:o?d.agentType:s.agentType,sessionMode:a?.sessionMode??s.sessionMode,managerConfig:a?g:s.managerConfig,workDir:o?d.workDir:s.workDir,agentSessionId:I,modelId:q(m?.modelId)??q(s.modelId),systemPrompt:d?.systemPrompt??s.systemPrompt??null,managerDefaultWorkerAgentType:a?a.managerDefaultWorkerAgentType:s.managerDefaultWorkerAgentType,managerDefaultWorkerModelId:a?a.managerDefaultWorkerModelId:s.managerDefaultWorkerModelId,reasoningEffort:s.reasoningEffort??void 0,clientMessageId:s.clientMessageId??s.id,attachments:h,externalChannel:s.externalChannel,replyTarget:s.replyTarget,roomContextRegistryKey:d?.roomContextRegistryKey,activationCauseMessageId:t.at(-1)?.causeMessageId,rootCauseMessageId:t.at(-1)?.rootCauseMessageId,activationSequence:t.at(-1)?.sequence,activationHopCount:t.at(-1)?.hopCount,managedRecoverySource:i?.kind==="manager_event"?"manager_event":o?"managed_direct":t.length>0?"managed_room":void 0,managerEventId:i?.kind==="manager_event"?i.eventId:void 0,managedWorkerParentSessionId:n?.kind==="manager_worker"?n.parentSessionId:void 0,managerWorkerNativeFull:n?.kind==="manager_worker"&&this.opts.getRuntime().managerRuntime?.isNativeSkillManager(n.parentSessionId,s.workDir)||void 0,managedRoomContextToken:d?.managedRoomContextToken,suppressExternalOwnerEcho:t.length>0||void 0,queuedBatchItems:e.map(v),waitForDispatch:!0}},d??void 0)}async isSessionBusy(e){const s=this.opts.getRuntime().recoveryBarrier;if(s?.isDispatchFrozen()||s?.isRecovering(e))return!0;const t=this.opts.getRuntime().sessions.get(e);if(!t)return!1;if(t.currentRunId){if(!(t.currentRunId.startsWith("dispatch:")&&!!this.opts.getRuntime().runJournal&&!t.journalRunId)||(t.adapter.getStatus?await t.adapter.getStatus().catch(()=>null):null)?.active)return!0;this.logLifecycle("drain.recovered-stale-dispatch",e,{staleRunId:t.currentRunId}),t.currentRunId=null,t.currentRunPhase=null}if(!t.adapter.getStatus)return!1;const o=await t.adapter.getStatus().catch(()=>null);return o?.active?(o.runId&&(t.currentRunId=o.runId,t.currentRunPhase=o.runPhase??t.currentRunPhase),!0):!1}broadcast(e){this.opts.getRuntime().client.sendEvent({type:"event",event:"session.queue.update",payload:{queue:this.getSnapshot(e)}})}}export{ye as ChatQueueManager,A as TerminalQueuedChatDispatchError,Se as purgeQueuedSessionData};
+import{NativeSourceDeletionStore as U}from"../native-fusion/native-source-deletions.js";import{QueueRetryScheduler as J}from"./queue-retry.js";import{findDuplicateQueuedMessage as E,stripLocalQueueFields as v,takeCompatibleBatch as G,formatQueuedBatch as Q,dedupeQueuedAttachments as B,classifyQueueError as F,isControllableQueueMessage as T,takeControllableMessage as H,normalizeQueuedModelId as q,resolveManagedRoomNativeSessionId as W}from"./queue-helpers.js";import{randomUUID as V}from"node:crypto";import{CODEX_DESKTOP_SEND_UNAVAILABLE as X,buildUserMessagePayload as Y}from"@shennian/wire";import{isCodexThreadTailInterrupted as Z}from"../native-fusion/codex-files.js";import{mergeProjectedSessions as ee}from"./projection.js";import{parsePersonalAttachmentReferences as C}from"./attachment-reference.js";import{assertFinishedAttachmentReferences as L,markAttachmentReferences as b}from"./attachment-transfer-store.js";import{getPersonalSyncIdentity as R}from"../personal-sync/epoch.js";import{getDaemonInstallationId as D}from"../personal-sync/identity.js";import{appendMessage as j,getMessageDeliveryStatus as _,listSessionRecords as N,recordSession as z,tombstoneMessage as $,upsertMessage as x}from"./store.js";import{requestSessionIndexPublish as S}from"./index-publisher.js";import{purgeQueuedSessionData as te,readQueue as p,writeQueue as M}from"./queue-storage.js";import{getNativeSourceAgentType as se}from"./handlers/chat-events.js";import{purgeQueuedSessionData as Se}from"./queue-storage.js";class A extends Error{constructor(e){super(e),this.name="TerminalQueuedChatDispatchError"}}function P(){return new Date().toISOString()}function O(u){const e=P(),s=u.managedWorkerParentSessionId?{kind:"manager_worker",parentSessionId:u.managedWorkerParentSessionId}:u.managerEventId?{kind:"manager_event",eventId:u.managerEventId}:u.managedAgentDirect?{kind:"managed_direct"}:{kind:"personal"};return{...R(D()),id:u.queueMessageId||u.clientMessageId||`queue-${V()}`,sessionId:u.sessionId,text:u.text,agentType:u.agentType,sessionMode:u.sessionMode,managerConfig:u.managerConfig,workDir:u.workDir,agentSessionId:u.agentSessionId??null,modelId:q(u.modelId)??null,systemPrompt:u.systemPrompt??null,managerDefaultWorkerAgentType:u.managerDefaultWorkerAgentType??null,managerDefaultWorkerModelId:u.managerDefaultWorkerModelId??null,reasoningEffort:u.reasoningEffort??null,clientMessageId:u.clientMessageId??null,attachments:C(u.attachments),externalChannel:u.externalChannel??null,replyTarget:u.replyTarget??null,origin:u.origin,createdAt:e,updatedAt:e,trigger:s}}function w(u){return{id:u.clientMessageId??u.id,sessionId:u.sessionId,role:u.trigger?.kind==="manager_event"?"agent":"user",ts:Date.parse(u.createdAt),payload:Y(u.text,u.attachments)}}class ye{opts;draining=new Set;promoting=new Set;disposed=!1;retries=new J(e=>{this.wakeSession(e)});deletions=new U;inFlightMessageIds=new Map;deferredLogState=new Map;prepareManagedRoomTurn=null;prepareManagedAgentDirectTurn=null;constructor(e){this.opts=e,setTimeout(()=>this.drainIdleQueues(),0).unref?.()}stop(){this.disposed=!0,this.retries.stop()}getSnapshot(e){const s=p().sessions[e]??[],t=this.inFlightMessageIds.get(e);return{...R(D()),sessionId:e,busy:!!this.opts.getRuntime().sessions.get(e)?.currentRunId,pending:s.map(v),controllablePending:s.filter(o=>T(o)&&!t?.has(o.id)).map(v)}}setManagedRoomTurnPreparer(e){this.prepareManagedRoomTurn=e}setManagedAgentDirectTurnPreparer(e){this.prepareManagedAgentDirectTurn=e}async enqueueManagedRoomBatch(e){if(e.length===0)return;const s=this.opts.getRuntime(),t=p(),o=new Set,i=new Map;let n=!1;for(const d of e){const a=d.params;if(this.deletions.hasSession(a.sessionId)){delete t.sessions[a.sessionId],n=!0;continue}if(!a.sessionId||!a.text||!a.agentType||!a.workDir)throw new Error("sessionId, text, agentType and workDir are required");const g=O({...R(D()),...a});g.origin="room",g.managedRoom=d.context,g.trigger={kind:"room_activation",...d.context};const c=t.sessions[a.sessionId]??[],h=E(c,g),I=g.clientMessageId??g.id,m=_(a.sessionId,I);if(!h&&m.state==="not_found")c.push(g),t.sessions[a.sessionId]=c,i.set(`${g.sessionId}\0${I}`,g),o.add(a.sessionId),n=!0;else if(h&&m.state==="not_found")i.set(`${h.sessionId}\0${I}`,h),o.add(a.sessionId);else if(h&&m.state==="queued")o.add(a.sessionId);else if(h){const l=c.filter(k=>k.id!==h.id);l.length?t.sessions[a.sessionId]=l:delete t.sessions[a.sessionId],n=!0}}n&&M(t);let r;for(const d of o)try{for(const a of i.values())a.sessionId===d&&(j(d,w(a),"queued"),z({sessionId:d,agentType:a.agentType,sessionMode:a.sessionMode,managerConfig:a.managerConfig,workDir:a.workDir,agentSessionId:W(s,a),modelId:a.modelId}));this.logLifecycle("enqueue.persisted",d,{origin:"room",pendingCount:p().sessions[d]?.length??0}),this.broadcast(d),this.wakeSession(d)}catch(a){r=a,this.logLifecycle("enqueue.commit-failed",d,{failureClass:F(a instanceof Error?a.message:"","canonical_commit_failed")},"error"),this.retries.schedule(d,!0)}if(o.size&&S(s.client),r)throw r}async handleEnqueue(e){const s=this.opts.getRuntime(),t=e.params;if(ee(t.sessionListProjection),!t.sessionId||!t.text||!t.agentType||!t.workDir)return s.client.sendRes({type:"res",id:e.id,ok:!1,error:"sessionId, text, agentType and workDir are required"}),!1;const o=t.managedAgentDirect?await this.prepareManagedAgentDirectTurn?.(t.sessionId):null;if(t.managedAgentDirect&&!o)throw new Error("managed_agent_context_unavailable");const i=o?N().find(K=>K.sessionId===t.sessionId):null,n=o?{...t,...o,agentType:o.agentType,sessionMode:o.sessionMode,managerConfig:o.managerConfig,workDir:o.workDir,agentSessionId:o.agentSessionId,modelId:i?.modelId??void 0,systemPrompt:o.systemPrompt,managerDefaultWorkerAgentType:o.managerDefaultWorkerAgentType,managerDefaultWorkerModelId:o.managerDefaultWorkerModelId}:t,r=s.resolvePath(n.workDir),d=C(n.attachments);d?.length&&L(r,d);const a={text:n.text,attachments:d,localized:!1},g=s.sessions.get(n.sessionId);if(n.agentType==="codex"&&n.agentSessionId&&g?.currentRunId?.startsWith("dispatch:")&&Z(n.agentSessionId))return s.client.sendRes({type:"res",id:e.id,ok:!1,error:X}),!1;o?.managedRoomContextToken&&s.revokeManagedRoomContextToken?.(o.managedRoomContextToken);const c=await this.isSessionBusy(n.sessionId),h=p(),I=h.sessions[n.sessionId]??[],m=O({...n,text:a.text,attachments:a.attachments,systemPrompt:o?null:n.systemPrompt});o&&(m.managedAgentDirect=!0,m.trigger?.kind!=="manager_event"&&(m.trigger={kind:"managed_direct"}));const l=m.clientMessageId??m.id,k=_(n.sessionId,l);if(k.state!=="not_found")return s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:k.state==="queued",queueMessageId:m.id,...k,queue:this.getSnapshot(n.sessionId)}}),!0;const y=E(I,m);y||(h.sessions[n.sessionId]=[...I,m],M(h));const f=y??m;return z({sessionId:f.sessionId,agentType:f.agentType,sessionMode:f.sessionMode,managerConfig:f.managerConfig,workDir:f.workDir,agentSessionId:f.agentSessionId,modelId:f.modelId}),j(f.sessionId,w(f),"queued"),b(s.resolvePath(f.workDir),f.attachments??[]),S(s.client),this.logLifecycle("enqueue.persisted",n.sessionId,{origin:n.origin??"app",queueMessageId:f.id,pendingCount:(p().sessions[n.sessionId]??[]).length}),this.broadcast(n.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queued:!0,queueMessageId:y?.id??m.id,..._(n.sessionId,y?.clientMessageId??y?.id??m.clientMessageId??m.id),queue:this.getSnapshot(n.sessionId),...a.localized?{localizedAttachments:!0}:{}}}),c?this.retries.schedule(n.sessionId):this.wakeSession(n.sessionId),!0}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)}})}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 o=p(),i=o.sessions[t.sessionId]??[],n=i.findIndex(a=>a.id===t.queueMessageId);if(n<0){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}if(!T(i[n])||this.inFlightMessageIds.get(t.sessionId)?.has(t.queueMessageId)){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const r=C(t.attachments),d=s.resolvePath(i[n].workDir);r?.length&&L(d,r),i[n]={...i[n],text:t.text,attachments:r,updatedAt:P()},o.sessions[t.sessionId]=i,M(o),x(t.sessionId,w(i[n]),"queued"),i[n].attachments?.length&&b(d,i[n].attachments??[]),S(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 o=p(),i=o.sessions[t.sessionId]??[],n=i.find(d=>d.id===t.queueMessageId);if(n&&(!T(n)||this.inFlightMessageIds.get(t.sessionId)?.has(t.queueMessageId))){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"managed_agent_queue_mutation_forbidden"});return}const r=i.filter(d=>d.id!==t.queueMessageId);if(r.length===i.length){s.client.sendRes({type:"res",id:e.id,ok:!1,error:"Queued message not found"});return}r.length?o.sessions[t.sessionId]=r:delete o.sessions[t.sessionId],M(o),$(t.sessionId,n?.clientMessageId??n?.id??t.queueMessageId),S(s.client),this.broadcast(t.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId)}})}async handlePromote(e){const s=this.opts.getRuntime(),t=e.params,o=n=>s.client.sendRes({type:"res",id:e.id,ok:!1,error:n});if(!t.sessionId||!t.queueMessageId){o("sessionId and queueMessageId are required");return}if(this.promoting.has(t.sessionId)){o("queue_promotion_in_progress");return}this.promoting.add(t.sessionId);let i=[];try{for(;this.draining.has(t.sessionId);)await new Promise(d=>setTimeout(d,5));if(i=H(p().sessions[t.sessionId]??[],t.queueMessageId),!i.length){o("Queued message not found");return}this.inFlightMessageIds.set(t.sessionId,new Set(i.map(d=>d.id)));const n=s.sessions.get(t.sessionId);let r="idle";if(n?.currentRunId)if(this.canInjectActiveBatch(n,i)){const d=i.length===1?i[0].text:Q(i),a=i[0].clientMessageId??i[0].id;s.nativeFusion?.registerManagedSend({sessionId:t.sessionId,agentType:i[0].agentType,sourceAgentType:se(i[0].agentType,i[0].modelId),canonicalMessageId:a,sourceSessionKey:n.agentSessionId,text:d,managedEchoPolicy:"suppress_following"});const g=Date.now();try{await n.adapter.injectActiveInput({text:d,attachments:B(i.flatMap(c=>c.attachments??[])),clientMessageId:a})}catch(c){throw s.nativeFusion?.cancelManagedSend(a),c}for(const c of i)x(t.sessionId,{...w(c),ts:g},"executing");this.removeBatch(t.sessionId,i),r="injected"}else{if(!this.opts.interruptQueuedSession)throw new Error("queue_interrupt_unavailable");await this.opts.interruptQueuedSession(t.sessionId),await this.dispatchQueuedMessages(i),this.removeBatch(t.sessionId,i),r="restarted"}else await this.dispatchQueuedMessages(i),this.removeBatch(t.sessionId,i);S(s.client),this.broadcast(t.sessionId),s.client.sendRes({type:"res",id:e.id,ok:!0,payload:{queue:this.getSnapshot(t.sessionId),promotedMessageIds:i.map(d=>d.id),mode:r}})}catch(n){o(n instanceof Error?n.message:"queue_promotion_failed"),this.broadcast(t.sessionId)}finally{this.inFlightMessageIds.delete(t.sessionId),this.promoting.delete(t.sessionId),p().sessions[t.sessionId]?.length&&this.wakeSession(t.sessionId)}}noteTerminal(e){this.wakeSession(e)}purgeSession(e){te(e),this.retries.cancel(e)}discardPendingManagedRoomMessages(e){if(this.deletions.hasSession(e)){const r=p().sessions[e]?.length??0;return this.purgeSession(e),r}const s=p(),t=s.sessions[e]??[],o=this.inFlightMessageIds.get(e)??new Set,i=t.filter(r=>r.managedRoom&&!o.has(r.id));if(i.length===0)return 0;const n=t.filter(r=>!i.includes(r));n.length?s.sessions[e]=n:delete s.sessions[e],M(s);for(const r of i)$(e,r.clientMessageId??r.id);return S(this.opts.getRuntime().client),this.logLifecycle("managed-room.discarded-paused",e,{discardedCount:i.length,pendingCount:n.length}),this.broadcast(e),i.length}wakeSession(e){this.drainNext(e).catch(()=>{this.logLifecycle("drain.retry-required",e,{},"error"),this.retries.schedule(e,!0)})}drainIdleQueues(){if(this.disposed)return;const e=p();for(const s of Object.keys(e.sessions))this.wakeSession(s)}async drainNext(e){if(this.disposed||this.draining.has(e)||this.promoting.has(e))return;this.draining.add(e);let s=!1,t=!1,o=[];try{if(this.deletions.hasSession(e)){this.purgeSession(e);return}if(await this.isSessionBusy(e)){const g=p().sessions[e]?.length??0,c=this.deferredLogState.get(e),h=Date.now();(!c||c.pendingCount!==g||h-c.at>=3e4)&&(this.logLifecycle("drain.deferred",e,{pendingCount:g}),this.deferredLogState.set(e,{at:h,pendingCount:g}));return}this.deferredLogState.delete(e);const i=p(),n=i.sessions[e]??[],r=n.filter(g=>{const c=g.clientMessageId??g.id,h=_(e,c).state;return h==="queued"||h==="not_found"});r.length!==n.length&&(r.length?i.sessions[e]=r:delete i.sessions[e],M(i),this.logLifecycle("drain.pruned-terminal",e,{prunedCount:n.length-r.length,pendingCount:r.length}));const d=G(r);if(d.length===0){this.broadcast(e);return}o=d,this.inFlightMessageIds.set(e,new Set(d.map(g=>g.id))),this.logLifecycle("dispatch.start",e,{batchCount:d.length,queueMessageId:d[0].id,origin:d[0].origin??"app"});for(const g of d)x(e,w(g),"queued");await this.dispatchQueuedMessages(d);const a=this.removeBatch(e,d);this.logLifecycle("dispatch.accepted",e,{batchCount:d.length,queueMessageId:d[0].id,pendingCount:a}),this.deferredLogState.delete(e),s=a>0,this.broadcast(e)}catch(i){const n=F(i instanceof Error?i.message:String(i),i instanceof A?"terminal_dispatch_failed":"transient_dispatch_failed"),r=o[0],d=!!(r&&!r.trigger&&!r.clientMessageId&&Date.now()-Date.parse(r.createdAt)>=10*6e4&&this.opts.getRuntime().managerRuntime?.isManagedWorkerSession(e));if(i instanceof A||n==="managed_claim_source_binding_conflict"||d){const g=this.removeBatch(e,o);this.logLifecycle("dispatch.terminal-failure",e,{batchCount:o.length,queueMessageId:o[0]?.id,pendingCount:g,failureClass:d?"stale_legacy_managed_worker_dispatch_failure":n},"error"),s=g>0}else t=!0,this.logLifecycle("dispatch.transient-failure",e,{failureClass:n},"error");this.broadcast(e)}finally{this.inFlightMessageIds.delete(e),this.draining.delete(e),p().sessions[e]?.length?s?queueMicrotask(()=>this.wakeSession(e)):this.retries.schedule(e,t):this.retries.cancel(e)}}canInjectActiveBatch(e,s){if(!e.adapter.supportsActiveInput||!e.adapter.injectActiveInput||!e.journalRunId)return!1;const t=this.opts.getRuntime().runJournal?.get(s[0].sessionId,e.journalRunId),o=s[0].trigger?.kind==="managed_direct"||s[0].managedAgentDirect?null:"personal";return!!(o&&s[0].systemPrompt==null&&t&&t.source===o&&t.deliveryMode==="direct"&&t.agentType===s[0].agentType&&t.workDir===e.workDir&&t.agentSessionId===e.agentSessionId&&t.modelId===(s[0].modelId??null)&&t.reasoningEffort===(s[0].reasoningEffort??null))}removeBatch(e,s){if(s.length===0)return p().sessions[e]?.length??0;const t=p(),o=new Set(s.map(n=>n.id)),i=(t.sessions[e]??[]).filter(n=>!o.has(n.id));return i.length?t.sessions[e]=i:delete t.sessions[e],M(t),i.length}logLifecycle(e,s,t,o="info"){const i=`[chat.queue] ${JSON.stringify({at:P(),event:e,sessionId:s,...t})}`;o==="error"?console.error(i):console.info(i)}async dispatchQueuedMessages(e){const s=e[0],t=e.flatMap(l=>{if(l.trigger?.kind==="room_activation"){const{kind:k,...y}=l.trigger;return[y]}return l.managedRoom?[l.managedRoom]:[]}),o=e.some(l=>l.trigger?.kind==="managed_direct"||!!l.managedAgentDirect),i=e.find(l=>l.trigger?.kind==="manager_event")?.trigger,n=e.find(l=>l.trigger?.kind==="manager_worker")?.trigger,r=t.length>0?await this.prepareManagedRoomTurn?.(t):o?await this.prepareManagedAgentDirectTurn?.(s.sessionId):void 0;if(t.length>0&&!r)throw new Error("managed_room_turn_preparer_unavailable");if(o&&!r)throw new Error("managed_agent_context_unavailable");const d=new Set((p().sessions[s.sessionId]??[]).map(l=>l.id));if(this.disposed||this.deletions.hasSession(s.sessionId)||e.some(l=>!d.has(l.id)))throw r?.managedRoomContextToken&&this.opts.getRuntime().revokeManagedRoomContextToken?.(r.managedRoomContextToken),this.disposed?new Error("queue_stopped"):new A("queued_dispatch_cancelled");const a=r??null,g=a?.sessionMode==="manager"?a.managerConfig:null,c=e.length===1?s.text:Q(e),h=B(e.flatMap(l=>l.attachments??[])),I=o?r.agentSessionId:t.length>0?W(this.opts.getRuntime(),s):s.agentSessionId??null,m=r?N().find(l=>l.sessionId===s.sessionId):null;await this.opts.dispatchQueuedReq({type:"req",id:`queue-send-${s.id}-${Date.now()}`,method:"chat.send",params:{...R(D()),sessionId:s.sessionId,text:c,agentType:o?r.agentType:s.agentType,sessionMode:a?.sessionMode??s.sessionMode,managerConfig:a?g:s.managerConfig,workDir:o?r.workDir:s.workDir,agentSessionId:I,modelId:q(m?.modelId)??q(s.modelId),systemPrompt:r?.systemPrompt??s.systemPrompt??null,managerDefaultWorkerAgentType:a?a.managerDefaultWorkerAgentType:s.managerDefaultWorkerAgentType,managerDefaultWorkerModelId:a?a.managerDefaultWorkerModelId:s.managerDefaultWorkerModelId,reasoningEffort:s.reasoningEffort??void 0,clientMessageId:s.clientMessageId??s.id,attachments:h,externalChannel:s.externalChannel,replyTarget:s.replyTarget,roomContextRegistryKey:r?.roomContextRegistryKey,activationCauseMessageId:t.at(-1)?.causeMessageId,rootCauseMessageId:t.at(-1)?.rootCauseMessageId,activationSequence:t.at(-1)?.sequence,activationHopCount:t.at(-1)?.hopCount,managedRecoverySource:i?.kind==="manager_event"?"manager_event":o?"managed_direct":t.length>0?"managed_room":void 0,managerEventId:i?.kind==="manager_event"?i.eventId:void 0,managedWorkerParentSessionId:n?.kind==="manager_worker"?n.parentSessionId:void 0,managerWorkerNativeFull:n?.kind==="manager_worker"&&this.opts.getRuntime().managerRuntime?.isNativeSkillManager(n.parentSessionId,s.workDir)||void 0,managedRoomContextToken:r?.managedRoomContextToken,suppressExternalOwnerEcho:t.length>0||void 0,queuedBatchItems:e.map(v),waitForDispatch:!0}},r??void 0)}async isSessionBusy(e){const s=this.opts.getRuntime().recoveryBarrier;if(s?.isDispatchFrozen()||s?.isRecovering(e))return!0;const t=this.opts.getRuntime().sessions.get(e);if(!t)return!1;if(t.currentRunId){if(!(t.currentRunId.startsWith("dispatch:")&&!!this.opts.getRuntime().runJournal&&!t.journalRunId)||(t.adapter.getStatus?await t.adapter.getStatus().catch(()=>null):null)?.active)return!0;this.logLifecycle("drain.recovered-stale-dispatch",e,{staleRunId:t.currentRunId}),t.currentRunId=null,t.currentRunPhase=null}if(!t.adapter.getStatus)return!1;const o=await t.adapter.getStatus().catch(()=>null);return o?.active?(o.runId&&(t.currentRunId=o.runId,t.currentRunPhase=o.runPhase??t.currentRunPhase),!0):!1}broadcast(e){this.opts.getRuntime().client.sendEvent({type:"event",event:"session.queue.update",payload:{queue:this.getSnapshot(e)}})}}export{ye as ChatQueueManager,A as TerminalQueuedChatDispatchError,Se as purgeQueuedSessionData};

package/dist/src/upgrade/distribution.d.ts

@@ -0,0 +1,21 @@
+import { type UpgradeChannel } from './channel.js';
+export type CliDistribution = {
+    schemaVersion: 1;
+    channel: UpgradeChannel;
+    version: string;
+    artifact: {
+        url: string;
+        size: number;
+        integrity: string;
+    };
+};
+export type CliUpdateSourceOrder = 'npm-first' | 'oss-first';
+export declare function getCliUpdateSourceOrder(value?: string | undefined): CliUpdateSourceOrder;
+export declare function parseCliDistribution(value: unknown, channel: UpgradeChannel): CliDistribution;
+export declare function fetchCliDistribution(options?: {
+    baseUrl?: string;
+    channel?: UpgradeChannel;
+    machineToken?: string;
+    timeoutMs?: number;
+}): Promise<CliDistribution>;
+export declare function fetchOwnedCliArtifact(version: string): Promise<CliDistribution['artifact'] | null>;

package/dist/src/upgrade/distribution.js

@@ -0,0 +1 @@
+import{loadConfig as a}from"../config/index.js";import{regionToUrl as o}from"../region.js";import{getUpgradeChannel as c}from"./channel.js";function m(e=process.env.SHENNIAN_CLI_UPDATE_SOURCE_ORDER){return e==="oss-first"?"oss-first":"npm-first"}function l(e,n){const r=e,t=r?.artifact;if(r?.schemaVersion!==1||r.channel!==n||typeof r.version!="string"||!/^\d+\.\d+\.\d+$/.test(r.version)||typeof t?.url!="string"||!/^https?:\/\//.test(t.url)||!Number.isSafeInteger(t.size)||Number(t.size)<=0||typeof t.integrity!="string"||!/^sha512-[A-Za-z0-9+/]+={0,2}$/.test(t.integrity))throw new Error("Shennian release API returned an invalid CLI distribution");return r}async function u(e={}){const n=a(),r=e.channel??c(),t=e.machineToken??n.machineToken;if(!t)throw new Error("Shennian release API requires a paired Client");const s=new URL("/api/releases/cli",e.baseUrl??n.serverUrl??o("cn"));s.searchParams.set("channel",r);const i=await fetch(s,{headers:{Accept:"application/json",Authorization:`Bearer ${t}`},signal:AbortSignal.timeout(e.timeoutMs??5e3)});if(!i.ok)throw new Error(`Shennian release API returned ${i.status}`);return l(await i.json(),r)}async function d(e){try{const n=await u();return n.version===e?n.artifact:null}catch{return null}}export{u as fetchCliDistribution,d as fetchOwnedCliArtifact,m as getCliUpdateSourceOrder,l as parseCliDistribution};

package/dist/src/upgrade/engine.js

@@ -1,3 +1,3 @@
-import s from"node:fs";import c from"node:path";import{execSync as F,execFile as D}from"node:child_process";import{promisify as T}from"node:util";import{fileURLToPath as J}from"node:url";import{getShennianDir as U,resolveShennianPath as m}from"../config/index.js";import{getUpgradeDistTag as O}from"./channel.js";const f=T(D),y=m("upgrade-attempt.json"),L=3,P="https://registry.npmjs.org",I=10*6e4,_=[5*6e4,30*6e4,120*6e4,360*6e4];function g(){try{const r=h();if(r)return JSON.parse(s.readFileSync(r,"utf-8")).version}catch{}return"0.0.0"}function h(r=J(import.meta.url)){let e=c.dirname(r);for(let n=0;n<6;n++){const t=c.join(e,"package.json");if(s.existsSync(t)&&JSON.parse(s.readFileSync(t,"utf-8")).name==="shennian")return t;e=c.dirname(e)}return null}async function G(){const r=await fetch(`https://registry.npmjs.org/shennian/${O()}`,{headers:{Accept:"application/json"},signal:AbortSignal.timeout(1e4)});if(!r.ok)throw new Error(`npm registry returned ${r.status}`);const e=await r.json();if(typeof e?.version!="string"||!/^\d+\.\d+\.\d+$/.test(e.version))throw new Error("npm registry returned an unsupported CLI version (expected numeric major.minor.patch)");return e.version}function H(r,e){const n=p=>p.replace(/^v/,"").split(".").map(Number),[t,a,o]=n(r),[i,l,u]=n(e);return i>t?"major":i===t&&l>a?"minor":i===t&&l===a&&u>o?"patch":"none"}function B(){return F("npm root -g",{encoding:"utf-8",stdio:"pipe",windowsHide:!0}).trim()}function Y(r,e=process.platform){const n=e==="win32"?c.win32:c.posix,t=n.dirname(r),a=n.dirname(t);if(n.basename(a)!=="node_modules")return null;const o=n.dirname(a);return n.basename(o)==="lib"?n.dirname(o):o}function N(){const r=h();return r?Y(r):null}function z(){const r=h();return r?c.join(c.dirname(r),"dist","bin","shennian.js"):c.join(B(),"shennian","dist","bin","shennian.js")}function K(){try{return JSON.parse(s.readFileSync(y,"utf-8"))}catch{return null}}function d(r){s.mkdirSync(U(),{recursive:!0}),s.writeFileSync(y,JSON.stringify(r,null,2))}function X(){try{s.unlinkSync(y)}catch{}}function $(){return m("upgrade-failures.json")}function Z(){return m("upgrade.lock")}function q(r=Z(),e={}){const n=e.now??Date.now(),t=e.pid??process.pid,a=e.isProcessAlive??(u=>{try{return process.kill(u,0),!0}catch{return!1}});s.mkdirSync(c.dirname(r),{recursive:!0});const o=()=>s.openSync(r,"wx",384);let i;try{i=o()}catch(u){if(u.code!=="EEXIST")throw u;let p=!1;try{const A=JSON.parse(s.readFileSync(r,"utf-8")),b=Number(A.pid),E=Number(A.createdAt);p=!Number.isInteger(b)||!Number.isFinite(E)||n-E>I||!a(b)}catch{p=!0}if(!p)throw new Error("Another Shennian upgrade is already running");s.rmSync(r,{force:!0}),i=o()}s.writeFileSync(i,JSON.stringify({pid:t,createdAt:n})),s.closeSync(i);let l=!1;return{release(){if(!l){l=!0;try{s.unlinkSync(r)}catch{}}}}}function w(r=$()){try{const e=JSON.parse(s.readFileSync(r,"utf-8"));return!e||typeof e!="object"||!e.versions?{versions:{}}:{versions:e.versions}}catch{return{versions:{}}}}function j(r){s.mkdirSync(U(),{recursive:!0}),s.writeFileSync($(),JSON.stringify(r,null,2))}function Q(r){return _[Math.min(Math.max(r-1,0),_.length-1)]}function cr(r,e=Date.now(),n=w()){const t=n.versions[r];return!!(t&&t.nextRetryAt>e)}function ur(r,e,n=Date.now()){const t=w(),a=t.versions[r],o=(a?.failureCount??0)+1,i={version:r,failureCount:o,firstFailedAt:a?.firstFailedAt??n,lastFailedAt:n,nextRetryAt:n+Q(o),error:e};return t.versions[r]=i,j(t),i}function k(r){const e=w();e.versions[r]&&(delete e.versions[r],j(e))}async function lr(){const r=K();return r?Date.now()-r.attemptAt>600*1e3?(X(),null):(r.attemptCount++,r.attemptCount>=L?(d(r),r):(d(r),null)):null}async function pr(r,e,n={}){let t;try{t=q()}catch(a){return{ok:!1,error:a instanceof Error?a.message:String(a)}}try{return await W(r,e,n)}finally{t.release()}}async function W(r,e,n={}){const t=n.currentVersion??g(),a=g();if(t===r)return k(r),{ok:!1,error:`Already on version ${r}`};e({step:"checking"});try{F("npm --version",{stdio:"pipe",windowsHide:!0})}catch{return{ok:!1,error:"npm is not available in PATH"}}if(a===r){e({step:"verifying",version:r});try{await v(r)}catch(o){return{ok:!1,error:`Smoke test failed: ${o instanceof Error?o.message:String(o)}`}}return d({from:t,to:r,attemptCount:0,attemptAt:Date.now()}),k(r),e({step:"restarting",from:t,to:r}),{ok:!0,from:t,to:r}}e({step:"backing-up"}),e({step:"installing",version:r});try{await M(r)}catch(o){const i=await R(t);return{ok:!1,error:`npm install failed: ${o instanceof Error?o.message:String(o)}${i.ok?"":`; rollback failed: ${i.error}`}`,rolledBack:i.ok}}e({step:"verifying",version:r});try{await v(r)}catch(o){const i=await R(t);return{ok:!1,error:`Smoke test failed: ${o instanceof Error?o.message:String(o)}${i.ok?"":`; rollback failed: ${i.error}`}`,rolledBack:i.ok}}return d({from:t,to:r,attemptCount:0,attemptAt:Date.now()}),k(r),e({step:"restarting",from:t,to:r}),{ok:!0,from:t,to:r}}function C(r){if(!/^[0-9]+\.[0-9]+\.[0-9]+(?:-[0-9A-Za-z.-]+)?(?:\+[0-9A-Za-z.-]+)?$/.test(r))throw new Error(`Invalid npm version: ${r}`)}function S(){return process.platform==="win32"?"npm.cmd":"npm"}function V(r,e){return{timeout:e,windowsHide:!0,shell:r==="win32"}}function x(r){return V(process.platform,r)}async function M(r){C(r);const e=N(),n=["install","-g",`shennian@${r}`,...e?["--prefix",e]:[]];try{await f(S(),n,x(12e4))}catch(t){const a=t instanceof Error?t.message:String(t);try{await f(S(),[...n,"--registry",P],{...x(12e4)})}catch(o){const i=o instanceof Error?o.message:String(o);throw new Error(`${a}
+import s from"node:fs";import c from"node:path";import{execSync as F,execFile as R}from"node:child_process";import{promisify as T}from"node:util";import{fileURLToPath as J}from"node:url";import{getShennianDir as U,resolveShennianPath as m}from"../config/index.js";import{getUpgradeDistTag as O}from"./channel.js";import{fetchCliDistribution as L}from"./distribution.js";const f=T(R),y=m("upgrade-attempt.json"),P=3,I="https://registry.npmjs.org",G=10*6e4,_=[5*6e4,30*6e4,120*6e4,360*6e4];function g(){try{const r=h();if(r)return JSON.parse(s.readFileSync(r,"utf-8")).version}catch{}return"0.0.0"}function h(r=J(import.meta.url)){let e=c.dirname(r);for(let n=0;n<6;n++){const t=c.join(e,"package.json");if(s.existsSync(t)&&JSON.parse(s.readFileSync(t,"utf-8")).name==="shennian")return t;e=c.dirname(e)}return null}async function H(){let r;try{const e=await fetch(`https://registry.npmjs.org/shennian/${O()}`,{headers:{Accept:"application/json"},signal:AbortSignal.timeout(1e4)});if(!e.ok)throw new Error(`npm registry returned ${e.status}`);const n=await e.json();if(typeof n?.version!="string"||!/^\d+\.\d+\.\d+$/.test(n.version))throw new Error("npm registry returned an unsupported CLI version (expected numeric major.minor.patch)");return n.version}catch(e){r=e}try{return(await L()).version}catch(e){const n=r instanceof Error?r.message:String(r),t=e instanceof Error?e.message:String(e);throw new Error(`npm registry unavailable: ${n}; Shennian fallback failed: ${t}`)}}function B(r,e){const n=p=>p.replace(/^v/,"").split(".").map(Number),[t,a,o]=n(r),[i,u,l]=n(e);return i>t?"major":i===t&&u>a?"minor":i===t&&u===a&&l>o?"patch":"none"}function Y(){return F("npm root -g",{encoding:"utf-8",stdio:"pipe",windowsHide:!0}).trim()}function z(r,e=process.platform){const n=e==="win32"?c.win32:c.posix,t=n.dirname(r),a=n.dirname(t);if(n.basename(a)!=="node_modules")return null;const o=n.dirname(a);return n.basename(o)==="lib"?n.dirname(o):o}function $(){const r=h();return r?z(r):null}function K(){const r=h();return r?c.join(c.dirname(r),"dist","bin","shennian.js"):c.join(Y(),"shennian","dist","bin","shennian.js")}function X(){try{return JSON.parse(s.readFileSync(y,"utf-8"))}catch{return null}}function d(r){s.mkdirSync(U(),{recursive:!0}),s.writeFileSync(y,JSON.stringify(r,null,2))}function Z(){try{s.unlinkSync(y)}catch{}}function N(){return m("upgrade-failures.json")}function q(){return m("upgrade.lock")}function Q(r=q(),e={}){const n=e.now??Date.now(),t=e.pid??process.pid,a=e.isProcessAlive??(l=>{try{return process.kill(l,0),!0}catch{return!1}});s.mkdirSync(c.dirname(r),{recursive:!0});const o=()=>s.openSync(r,"wx",384);let i;try{i=o()}catch(l){if(l.code!=="EEXIST")throw l;let p=!1;try{const A=JSON.parse(s.readFileSync(r,"utf-8")),E=Number(A.pid),b=Number(A.createdAt);p=!Number.isInteger(E)||!Number.isFinite(b)||n-b>G||!a(E)}catch{p=!0}if(!p)throw new Error("Another Shennian upgrade is already running");s.rmSync(r,{force:!0}),i=o()}s.writeFileSync(i,JSON.stringify({pid:t,createdAt:n})),s.closeSync(i);let u=!1;return{release(){if(!u){u=!0;try{s.unlinkSync(r)}catch{}}}}}function w(r=N()){try{const e=JSON.parse(s.readFileSync(r,"utf-8"));return!e||typeof e!="object"||!e.versions?{versions:{}}:{versions:e.versions}}catch{return{versions:{}}}}function j(r){s.mkdirSync(U(),{recursive:!0}),s.writeFileSync(N(),JSON.stringify(r,null,2))}function W(r){return _[Math.min(Math.max(r-1,0),_.length-1)]}function ur(r,e=Date.now(),n=w()){const t=n.versions[r];return!!(t&&t.nextRetryAt>e)}function pr(r,e,n=Date.now()){const t=w(),a=t.versions[r],o=(a?.failureCount??0)+1,i={version:r,failureCount:o,firstFailedAt:a?.firstFailedAt??n,lastFailedAt:n,nextRetryAt:n+W(o),error:e};return t.versions[r]=i,j(t),i}function k(r){const e=w();e.versions[r]&&(delete e.versions[r],j(e))}async function fr(){const r=X();return r?Date.now()-r.attemptAt>600*1e3?(Z(),null):(r.attemptCount++,r.attemptCount>=P?(d(r),r):(d(r),null)):null}async function dr(r,e,n={}){let t;try{t=Q()}catch(a){return{ok:!1,error:a instanceof Error?a.message:String(a)}}try{return await V(r,e,n)}finally{t.release()}}async function V(r,e,n={}){const t=n.currentVersion??g(),a=g();if(t===r)return k(r),{ok:!1,error:`Already on version ${r}`};e({step:"checking"});try{F("npm --version",{stdio:"pipe",windowsHide:!0})}catch{return{ok:!1,error:"npm is not available in PATH"}}if(a===r){e({step:"verifying",version:r});try{await v(r)}catch(o){return{ok:!1,error:`Smoke test failed: ${o instanceof Error?o.message:String(o)}`}}return d({from:t,to:r,attemptCount:0,attemptAt:Date.now()}),k(r),e({step:"restarting",from:t,to:r}),{ok:!0,from:t,to:r}}e({step:"backing-up"}),e({step:"installing",version:r});try{await M(r)}catch(o){const i=await D(t);return{ok:!1,error:`npm install failed: ${o instanceof Error?o.message:String(o)}${i.ok?"":`; rollback failed: ${i.error}`}`,rolledBack:i.ok}}e({step:"verifying",version:r});try{await v(r)}catch(o){const i=await D(t);return{ok:!1,error:`Smoke test failed: ${o instanceof Error?o.message:String(o)}${i.ok?"":`; rollback failed: ${i.error}`}`,rolledBack:i.ok}}return d({from:t,to:r,attemptCount:0,attemptAt:Date.now()}),k(r),e({step:"restarting",from:t,to:r}),{ok:!0,from:t,to:r}}function C(r){if(!/^[0-9]+\.[0-9]+\.[0-9]+(?:-[0-9A-Za-z.-]+)?(?:\+[0-9A-Za-z.-]+)?$/.test(r))throw new Error(`Invalid npm version: ${r}`)}function S(){return process.platform==="win32"?"npm.cmd":"npm"}function rr(r,e){return{timeout:e,windowsHide:!0,shell:r==="win32"}}function x(r){return rr(process.platform,r)}async function M(r){C(r);const e=$(),n=["install","-g",`shennian@${r}`,...e?["--prefix",e]:[]];try{await f(S(),n,x(12e4))}catch(t){const a=t instanceof Error?t.message:String(t);try{await f(S(),[...n,"--registry",I],{...x(12e4)})}catch(o){const i=o instanceof Error?o.message:String(o);throw new Error(`${a}
 --- npmjs fallback ---
-${i}`)}}}async function v(r){C(r);const e=z(),{stdout:n}=await f(process.execPath,[e,"--version"],{timeout:1e4,windowsHide:!0});if(n.trim()!==r)throw new Error(`Version verification failed: expected ${r}, got ${n.trim()||"empty output"}`);const t=N();await f(S(),["ls","-g",`shennian@${r}`,"--omit=dev","--depth=0",...t?["--prefix",t]:[]],{...x(3e4)})}async function rr(r,e={}){await(e.install??M)(r),await(e.verify??v)(r)}async function R(r){try{return await rr(r),{ok:!0}}catch(e){return{ok:!1,error:e instanceof Error?e.message:String(e)}}}async function fr(r){const e=r??g(),n=await G(),t=H(e,n);return t==="none"?{hasUpdate:!1}:{hasUpdate:!0,current:e,latest:n,changeType:t}}export{q as acquireUpgradeLock,fr as checkForUpdate,X as clearUpgradeAttempt,k as clearUpgradeFailure,H as compareVersions,G as fetchLatestVersion,h as findPackageJson,g as getCurrentVersion,Q as getUpgradeRetryDelayMs,lr as handleStartupCrashCheck,M as installShennianVersion,cr as isUpgradeVersionInCooldown,V as npmExecOptionsForPlatform,pr as performUpgrade,K as readUpgradeAttempt,w as readUpgradeFailures,ur as recordUpgradeFailure,Y as resolveNpmGlobalPrefixFromPackageJson,rr as rollbackToExactVersion,v as verifyInstalledVersion,d as writeUpgradeAttempt};
+${i}`)}}}async function v(r){C(r);const e=K(),{stdout:n}=await f(process.execPath,[e,"--version"],{timeout:1e4,windowsHide:!0});if(n.trim()!==r)throw new Error(`Version verification failed: expected ${r}, got ${n.trim()||"empty output"}`);const t=$();await f(S(),["ls","-g",`shennian@${r}`,"--omit=dev","--depth=0",...t?["--prefix",t]:[]],{...x(3e4)})}async function er(r,e={}){await(e.install??M)(r),await(e.verify??v)(r)}async function D(r){try{return await er(r),{ok:!0}}catch(e){return{ok:!1,error:e instanceof Error?e.message:String(e)}}}async function mr(r){const e=r??g(),n=await H(),t=B(e,n);return t==="none"?{hasUpdate:!1}:{hasUpdate:!0,current:e,latest:n,changeType:t}}export{Q as acquireUpgradeLock,mr as checkForUpdate,Z as clearUpgradeAttempt,k as clearUpgradeFailure,B as compareVersions,H as fetchLatestVersion,h as findPackageJson,g as getCurrentVersion,W as getUpgradeRetryDelayMs,fr as handleStartupCrashCheck,M as installShennianVersion,ur as isUpgradeVersionInCooldown,rr as npmExecOptionsForPlatform,dr as performUpgrade,X as readUpgradeAttempt,w as readUpgradeFailures,pr as recordUpgradeFailure,z as resolveNpmGlobalPrefixFromPackageJson,er as rollbackToExactVersion,v as verifyInstalledVersion,d as writeUpgradeAttempt};

package/dist/src/upgrade/updater-worker.d.ts

@@ -29,6 +29,9 @@ type WorkerDeps = {
     execFile?: typeof execFile;
     healthTimeoutMs?: number;
     serviceRecovery?: boolean;
+    fetch?: typeof globalThis.fetch;
+    fetchOwnedArtifact?: (version: string) => Promise<OwnedArtifact | null>;
+    sourceOrder?: 'npm-first' | 'oss-first';
 };
 export declare function updaterPaths(job: Job): {
     active: string;
@@ -44,6 +47,11 @@ export declare function assertUpdaterAvailable(root: string): void;
 export declare function prepareUpdaterCandidate(job: Job, deps?: WorkerDeps): Promise<void>;
 /** Validate an already materialized candidate too (offline repair/artifact transport). */
 export declare function finalizeUpdaterCandidate(job: Job, deps?: WorkerDeps): Promise<void>;
+type OwnedArtifact = {
+    url: string;
+    size: number;
+    integrity: string;
+};
 export declare function runUpdaterJob(jobPath: string, deps?: WorkerDeps): Promise<void>;
 export declare function recoverServiceEntry(root: string): Promise<void>;
 export {};

package/dist/src/upgrade/updater-worker.js

@@ -1,2 +1,2 @@
-import o from"node:fs";import D from"node:os";import i from"node:path";import{execFile as U}from"node:child_process";import{promisify as Z}from"node:util";import{fileURLToPath as q,pathToFileURL as K}from"node:url";const j=Z(U),L=1,B="https://registry.npmjs.org",W=30*6e4;function A(t){return(t instanceof Error?t.message:String(t)).replace(/[A-Za-z]:\\[^\s]+|\/(?:[^\s/]+\/)+[^\s]+/g,"<path>").replace(/(token|password|secret|authorization)\s*[=:]\s*[^\s]+/gi,"$1=<redacted>").replace(/\s+/g," ").slice(0,500)}function k(t,r){o.mkdirSync(i.dirname(t),{recursive:!0,mode:448});const e=`${t}.tmp-${process.pid}-${Date.now()}`;try{o.writeFileSync(e,`${JSON.stringify(r,null,2)}
-`,{mode:384,flag:"wx"});const n=o.openSync(e,process.platform==="win32"?"r+":"r");try{o.fsyncSync(n)}finally{o.closeSync(n)}if(o.renameSync(e,t),process.platform!=="win32"){const s=o.openSync(i.dirname(t),"r");try{o.fsyncSync(s)}finally{o.closeSync(s)}}}finally{try{o.unlinkSync(e)}catch{}}}function O(t){const r=o.lstatSync(t);if(r.isSymbolicLink()||!r.isFile())throw new Error("invalid_job");const e=JSON.parse(o.readFileSync(t,"utf8")),n=w=>{if(typeof e[w]!="string"||!String(e[w]).trim())throw new Error("invalid_job");return String(e[w])},s=/^[0-9]+\.[0-9]+\.[0-9]+(?:-[0-9A-Za-z.-]+)?(?:\+[0-9A-Za-z.-]+)?$/,l=n("fromVersion"),c=n("targetVersion");if(e.protocolVersion!==L||e.packageName!=="shennian")throw new Error("invalid_job");if(!s.test(l)||!s.test(c))throw new Error("invalid_job");if(!Number.isFinite(Date.parse(n("createdAt"))))throw new Error("invalid_job");if(!/^[A-Za-z0-9_-]{1,128}$/.test(n("jobId")))throw new Error("invalid_job");if(!["darwin","linux","win32"].includes(n("platform")))throw new Error("invalid_job");const f=e.oldDaemonPid==null?null:Number(e.oldDaemonPid),a=Number(e.daemonExitTimeoutMs),u=Number(e.commandTimeoutMs);if(f!=null&&(!Number.isSafeInteger(f)||f<1))throw new Error("invalid_job");if(!Number.isSafeInteger(a)||a<1e3)throw new Error("invalid_job");if(!Number.isSafeInteger(u)||u<1e4)throw new Error("invalid_job");return{protocolVersion:1,jobId:n("jobId"),fromVersion:l,targetVersion:c,npmPrefix:n("npmPrefix"),npmExecutable:n("npmExecutable"),nodeExecutable:n("nodeExecutable"),packageName:"shennian",operation:e.operation==="rollback"?"rollback":"upgrade",oldDaemonPid:f,platform:n("platform"),createdAt:n("createdAt"),daemonExitTimeoutMs:a,commandTimeoutMs:u,resultPath:n("resultPath"),attemptPath:n("attemptPath"),isolatedDaemonStart:e.isolatedDaemonStart===!0,requireHandoffAck:e.requireHandoffAck===!0,...typeof e.rollbackJobId=="string"&&/^[A-Za-z0-9_-]{1,128}$/.test(e.rollbackJobId)?{rollbackJobId:e.rollbackJobId}:{}}}function T(t,r){return r==="win32"?i.win32.join(t,"node_modules","shennian"):i.posix.join(t,"lib","node_modules","shennian")}function x(t){const r=T(t.npmPrefix,t.platform),e=i.join(i.dirname(r),`.shennian-upgrade-${t.rollbackJobId??t.jobId}`),n=i.join(e,"candidate");return{active:r,transaction:e,prefix:n,candidate:T(n,t.platform),previous:i.join(e,"previous"),rejected:i.join(e,"rejected"),marker:i.join(e,"transaction.json")}}function m(t){try{return JSON.parse(o.readFileSync(t,"utf8"))}catch{return null}}function N(t){const r=o.lstatSync(t);if(!r.isDirectory()||r.isSymbolicLink())throw new Error("unsafe_installation")}function y(t,r){N(t);const e=o.realpathSync(t),n=a=>{const u=i.relative(e,o.realpathSync(a));if(u.startsWith(".."+i.sep)||u===".."||i.isAbsolute(u))throw new Error("external_package_dependency")},s=a=>{for(const u of o.readdirSync(a,{withFileTypes:!0})){const w=i.join(a,u.name);u.isSymbolicLink()?n(w):u.isDirectory()&&s(w)}};s(t);const l=m(i.join(t,"package.json"));if(l?.name!=="shennian"||l.version!==r||l.bin?.shennian!=="dist/bin/shennian.js")throw new Error("verification_failed");n(i.join(t,"dist","bin","shennian.js"));const c=new Set,f=a=>{const u=o.realpathSync(a);if(c.has(u))return;c.add(u);const w=m(i.join(a,"package.json"));if(!w)throw new Error("incomplete_package");for(const v of Object.keys(w.dependencies??{})){if(Object.hasOwn(w.optionalDependencies??{},v))continue;if(!/^(?:@[a-z0-9._-]+\/)?[a-z0-9._-]+$/i.test(v))throw new Error("invalid_dependency");let g=a,S=!1;for(;g===t||g.startsWith(t+i.sep);){const _=i.join(g,"node_modules",v);if(o.existsSync(i.join(_,"package.json"))){n(_),f(_),S=!0;break}g=i.dirname(g)}if(!S)throw new Error("incomplete_package")}};f(t)}function me(t){const r=m(i.join(t,"updater.lock"));if($(r))throw new Error("upgrade_locked")}async function X(t,r={}){const e=x(t);N(i.dirname(e.active)),y(e.active,t.fromVersion),o.mkdirSync(e.transaction,{mode:448});const n=r.execFile??j;try{await te({...t,npmPrefix:e.prefix},t.targetVersion,n),await Y(t,r)}catch(s){throw o.rmSync(e.transaction,{recursive:!0,force:!0}),Object.assign(new Error(A(s)),{code:"candidate_preparation_failed"})}}async function Y(t,r={}){const e=x(t);N(e.transaction);const n=m(e.marker);if(n&&n.phase!=="prepared")throw new Error("transaction_already_started");y(e.active,t.fromVersion),y(e.candidate,t.targetVersion);const s=r.execFile??j;await E({...t,npmPrefix:e.prefix},t.targetVersion,s),await s(t.npmExecutable,["ls","-g",`shennian@${t.targetVersion}`,"--all","--omit=dev","--offline","--depth=100","--prefix",e.prefix],h(t,3e4)),k(e.marker,{jobId:t.jobId,fromVersion:t.fromVersion,targetVersion:t.targetVersion,phase:"prepared"})}function F(t,r){if(o.renameSync(t,r),process.platform!=="win32")for(const e of new Set([i.dirname(t),i.dirname(r)])){const n=o.openSync(e,"r");try{o.fsyncSync(n)}finally{o.closeSync(n)}}}function Q(t){const r=x(t),e=i.dirname(t.resultPath);for(const n of o.readdirSync(i.dirname(r.active),{withFileTypes:!0})){if(!n.isDirectory()||!/^\.shennian-upgrade-[A-Za-z0-9_-]+$/.test(n.name))continue;const s=i.join(i.dirname(r.active),n.name);if(s===r.transaction)continue;const l=m(i.join(s,"transaction.json"));if(!l?.jobId||n.name!==`.shennian-upgrade-${l.jobId}`)continue;const c=m(i.join(e,`${l.jobId}.json`));c?.jobId===l.jobId&&["healthy","rolled_back"].includes(c.state??"")&&o.rmSync(s,{recursive:!0,force:!0})}}function ee(t){const r=i.join(i.dirname(t.attemptPath),"upgrade-failures.json");let e={versions:{}};try{e=JSON.parse(o.readFileSync(r,"utf8"))}catch{}const n=e.versions?.[t.targetVersion],s=Math.max(0,Number(n?.failureCount)||0)+1,l=Date.now(),c=[5*6e4,30*6e4,120*6e4,360*6e4];k(r,{versions:{...e.versions,[t.targetVersion]:{version:t.targetVersion,failureCount:s,firstFailedAt:n?.firstFailedAt??l,lastFailedAt:l,nextRetryAt:l+c[Math.min(s-1,c.length-1)],error:"upgrade_rolled_back"}}})}function b(t,r){const e=x(t),n=m(e.marker);if(!n||n.jobId!==(t.rollbackJobId??t.jobId))throw new Error("invalid_transaction");k(e.marker,{...n,phase:r})}function $(t){const r=Date.now()-D.uptime()*1e3;return!!(t?.pid&&(t.createdAt===void 0||t.createdAt>=r-2e3)&&I(t.pid))}function I(t){try{return process.kill(t,0),!0}catch{return!1}}async function J(t,r){if(r.serviceRecovery||!t.oldDaemonPid||!(r.isAlive??I)(t.oldDaemonPid))return;const e=r.kill??((l,c)=>process.kill(l,c)),n=r.sleep??(l=>new Promise(c=>setTimeout(c,l)));e(t.oldDaemonPid,"SIGTERM");const s=(r.now??Date.now)()+t.daemonExitTimeoutMs;for(;(r.now??Date.now)()<s;){if(!(r.isAlive??I)(t.oldDaemonPid))return;await n(100)}if(r.kill)e(t.oldDaemonPid,"SIGKILL");else if(t.platform==="win32")try{await(r.execFile??j)("taskkill",["/PID",String(t.oldDaemonPid),"/T","/F"],{timeout:15e3,windowsHide:!0,shell:!0})}catch{}else try{process.kill(-t.oldDaemonPid,"SIGKILL")}catch{try{process.kill(t.oldDaemonPid,"SIGKILL")}catch{}}if(await n(250),(r.isAlive??I)(t.oldDaemonPid))throw new Error("daemon_exit_timeout")}function h(t,r=t.commandTimeoutMs){return{timeout:r,windowsHide:!0,shell:t.platform==="win32"}}async function te(t,r,e){const n=["install","-g",`shennian@${r}`,"--prefix",t.npmPrefix];try{await e(t.npmExecutable,n,h(t))}catch{await e(t.npmExecutable,[...n,"--registry",B],h(t))}}async function E(t,r,e){const n=T(t.npmPrefix,t.platform),s=i.join(n,"dist","bin","shennian.js"),{stdout:l}=await e(t.nodeExecutable,[s,"--version"],h(t,1e4));if(l.trim()!==r)throw new Error("verification_failed");return s}async function P(t,r,e){if(t.isolatedDaemonStart){await e(t.nodeExecutable,[r,"start","--json"],h(t,3e4));return}try{if(t.platform==="darwin"){const n=typeof process.getuid=="function"?process.getuid():D.userInfo().uid;try{await e("launchctl",["bootstrap",`gui/${n}`,i.join(D.homedir(),"Library/LaunchAgents/com.shennian.agent.plist")],h(t,15e3))}catch{}await e("launchctl",["kickstart","-k",`gui/${n}/com.shennian.agent`],h(t,15e3));return}if(t.platform==="linux"){await e("systemctl",["--user","restart","shennian"],h(t,15e3));return}if(t.platform==="win32"){await e("schtasks",["/Run","/TN","ShennianAgent"],h(t,15e3));return}}catch{}await e(t.nodeExecutable,[r,"start","--json"],h(t,3e4))}async function C(t,r){if(!t.isolatedDaemonStart)try{t.platform==="darwin"?await r("launchctl",["bootout",`gui/${process.getuid?.()??D.userInfo().uid}/com.shennian.agent`],h(t,15e3)):t.platform==="linux"?await r("systemctl",["--user","stop","shennian"],h(t,15e3)):t.platform==="win32"&&await r("schtasks",["/End","/TN","ShennianAgent"],h(t,15e3))}catch{}}function re(t){return(r,e={})=>k(t.resultPath,{protocolVersion:L,jobId:t.jobId,state:r,fromVersion:t.fromVersion,targetVersion:t.targetVersion,operation:t.operation,...e,updatedAt:new Date().toISOString()})}function ne(t,r){const e=i.dirname(i.dirname(t)),n=i.join(e,"updater.lock");o.mkdirSync(e,{recursive:!0,mode:448});try{const s=o.openSync(n,"wx",384);o.writeFileSync(s,JSON.stringify({jobId:r.jobId,phase:"worker",pid:process.pid,createdAt:Date.now()})),o.closeSync(s)}catch(s){if(s.code!=="EEXIST")throw s;try{const c=JSON.parse(o.readFileSync(n,"utf8"));if((c.jobId!==r.jobId||c.phase!=="handoff"&&c.pid!==r.oldDaemonPid)&&$(c))throw new Error("upgrade_locked")}catch(c){if(c instanceof Error&&c.message==="upgrade_locked")throw c}o.rmSync(n,{force:!0});const l=o.openSync(n,"wx",384);o.writeFileSync(l,JSON.stringify({jobId:r.jobId,phase:"worker",pid:process.pid,createdAt:Date.now()})),o.closeSync(l)}return()=>{try{o.unlinkSync(n)}catch{}}}async function M(t,r={}){let e;try{e=O(t)}catch{throw new Error("invalid_job")}if(!r.serviceRecovery&&(r.now??Date.now)()-Date.parse(e.createdAt)>W)throw new Error("job_expired");const n=i.dirname(i.dirname(i.resolve(t)));if(i.basename(i.dirname(i.resolve(t)))!=="jobs"||i.basename(i.resolve(t))!==`${e.jobId}.json`)throw new Error("invalid_job");if(i.dirname(i.resolve(e.resultPath))!==i.join(n,"results"))throw new Error("invalid_job");if(i.basename(i.resolve(e.resultPath))!==`${e.jobId}.json`)throw new Error("invalid_job");const s=i.dirname(i.dirname(n));if(i.resolve(e.attemptPath)!==i.join(s,"upgrade-attempt.json"))throw new Error("invalid_job");const l=ne(t,e),c=re(e),f=r.execFile??j,a=x(e),u=i.join(n,"active-job.json"),w=r.sleep??(d=>new Promise(p=>setTimeout(p,d)));let v=!1,g={};const S=d=>{d&&ee(e),k(e.attemptPath,{from:d?e.targetVersion:e.fromVersion,to:d?e.fromVersion:e.targetVersion,attemptCount:0,attemptAt:Date.now(),jobId:e.jobId,operation:d?"rollback":"upgrade"}),c("starting",d?{operation:"rollback",...g,effectiveVersion:e.fromVersion,targetVersion:e.fromVersion,fromVersion:e.targetVersion}:{})},_=async d=>{if(r.serviceRecovery)return;const p=(r.now??Date.now)()+(r.healthTimeoutMs??9e4);for(;(r.now??Date.now)()<p;){if(m(e.resultPath)?.state===(d?"rolled_back":"healthy"))return;await w(250)}throw new Error("daemon_health_timeout")},H=async()=>{const d=m(i.join(s,"daemon.pid"));d?.version===e.targetVersion&&Number.isSafeInteger(d.pid)&&typeof d.pid=="number"&&d.pid!==process.pid&&await J({...e,oldDaemonPid:d.pid},r)},R=async()=>{if(b(e,"rolling_back"),r.serviceRecovery||await C(e,f),await H(),!o.existsSync(a.previous))y(a.active,e.fromVersion);else{if(y(a.previous,e.fromVersion),o.existsSync(a.active)){if(o.existsSync(a.rejected))throw new Error("rollback_destination_exists");F(a.active,a.rejected)}F(a.previous,a.active)}b(e,"restored"),y(a.active,e.fromVersion);const d=await E(e,e.fromVersion,f);S(!0),r.serviceRecovery||await P(e,d,f),await _(!0)};try{const d=m(e.resultPath);if(d?.jobId===e.jobId&&["healthy","rolled_back","failed"].includes(d.state??""))return;const p=m(a.marker);if(p&&(p.jobId!==(e.rollbackJobId??e.jobId)||p.fromVersion!==(e.operation==="rollback"?e.targetVersion:e.fromVersion)))throw new Error("invalid_transaction");if(e.operation==="rollback"){e={...e,fromVersion:e.targetVersion,targetVersion:e.fromVersion},v=!0,k(u,{jobPath:t}),await J(e,r),await R();return}if(p&&p.phase!=="prepared"){v=!0,p.phase==="restored"||p.phase==="rolling_back"&&!o.existsSync(a.previous)?(y(a.active,e.fromVersion),b(e,"restored"),S(!0),r.serviceRecovery||await P(e,await E(e,e.fromVersion,f),f),await _(!0)):p.phase==="switching"&&!o.existsSync(a.previous)?(y(a.active,e.fromVersion),b(e,"restored"),S(!0),r.serviceRecovery||await P(e,await E(e,e.fromVersion,f),f),await _(!0)):await R();return}if(p||(c("installing"),await X(e,r)),y(a.candidate,e.targetVersion),y(a.active,e.fromVersion),c("waiting_for_daemon"),e.requireHandoffAck){const G=(r.now??Date.now)()+1e4;for(;m(`${e.resultPath}.handoff`)?.jobId!==e.jobId;){if((r.now??Date.now)()>=G)throw new Error("handoff_not_confirmed");await w(50)}}await w(250),k(u,{jobPath:t}),r.serviceRecovery||await C(e,f),await J(e,r),b(e,"switching"),v=!0,F(a.active,a.previous),F(a.candidate,a.active),c("verifying"),y(a.active,e.targetVersion);const V=await E(e,e.targetVersion,f);b(e,"starting"),S(!1),r.serviceRecovery||await P(e,V,f),await _(!1)}catch(d){const p=d.code??(d instanceof Error?d.message:"internal_error");if(!v){c("failed",{errorCode:p,errorSummary:A(d)}),(!e.oldDaemonPid||!(r.isAlive??I)(e.oldDaemonPid))&&(y(a.active,e.fromVersion),r.serviceRecovery||await P(e,await E(e,e.fromVersion,f),f));return}g={errorCode:p,errorSummary:A(d)};try{await R()}catch(V){throw c("failed",{errorCode:"rollback_failed",errorSummary:A(V)}),V}}finally{const d=m(e.resultPath);if(["healthy","rolled_back"].includes(d?.state??""))try{Q(e)}catch{}l()}}async function ie(t){const r=m(i.join(t,"active-job.json"));if(!r?.jobPath)return;const e=i.resolve(r.jobPath);if(i.dirname(e)!==i.join(i.resolve(t),"jobs"))throw new Error("invalid_job");const n=O(e);for(;;){const s=m(n.resultPath);if(["healthy","rolled_back","failed"].includes(s?.state??""))return;const l=m(i.join(t,"updater.lock")),c=l?.pid!==process.pid&&$(l),f=m(x(n).marker);if(c){if(s?.state==="starting"&&["starting","restored"].includes(f?.phase??""))return;await new Promise(a=>setTimeout(a,250));continue}await M(e,{serviceRecovery:!0});return}}function z(t){try{return o.realpathSync.native(t)}catch{return i.resolve(t)}}const oe=process.argv[1]?z(process.argv[1]):"",ae=z(q(import.meta.url));if(oe===ae)if(process.argv[2]==="--service"){const[,,,t,r,...e]=process.argv;ie(t).then(async()=>{process.argv=[process.execPath,r,...e],await import(K(r).href)}).catch(()=>{process.exitCode=1})}else{const t=process.argv[2];t?M(t).catch(()=>{process.exitCode=1}):process.exitCode=2}export{me as assertUpdaterAvailable,Y as finalizeUpdaterCandidate,X as prepareUpdaterCandidate,ie as recoverServiceEntry,M as runUpdaterJob,x as updaterPaths,y as verifyLocalPackage};
+import o from"node:fs";import D from"node:os";import i from"node:path";import{createHash as U}from"node:crypto";import{execFile as B}from"node:child_process";import{promisify as q}from"node:util";import{fileURLToPath as K,pathToFileURL as W}from"node:url";const A=q(B),J=1,X="https://registry.npmjs.org",Y=30*6e4;function j(e){return(e instanceof Error?e.message:String(e)).replace(/[A-Za-z]:\\[^\s]+|\/(?:[^\s/]+\/)+[^\s]+/g,"<path>").replace(/(token|password|secret|authorization)\s*[=:]\s*[^\s]+/gi,"$1=<redacted>").replace(/\s+/g," ").slice(0,500)}function k(e,t){o.mkdirSync(i.dirname(e),{recursive:!0,mode:448});const r=`${e}.tmp-${process.pid}-${Date.now()}`;try{o.writeFileSync(r,`${JSON.stringify(t,null,2)}
+`,{mode:384,flag:"wx"});const n=o.openSync(r,process.platform==="win32"?"r+":"r");try{o.fsyncSync(n)}finally{o.closeSync(n)}if(o.renameSync(r,e),process.platform!=="win32"){const c=o.openSync(i.dirname(e),"r");try{o.fsyncSync(c)}finally{o.closeSync(c)}}}finally{try{o.unlinkSync(r)}catch{}}}function L(e){const t=o.lstatSync(e);if(t.isSymbolicLink()||!t.isFile())throw new Error("invalid_job");const r=JSON.parse(o.readFileSync(e,"utf8")),n=w=>{if(typeof r[w]!="string"||!String(r[w]).trim())throw new Error("invalid_job");return String(r[w])},c=/^[0-9]+\.[0-9]+\.[0-9]+(?:-[0-9A-Za-z.-]+)?(?:\+[0-9A-Za-z.-]+)?$/,l=n("fromVersion"),a=n("targetVersion");if(r.protocolVersion!==J||r.packageName!=="shennian")throw new Error("invalid_job");if(!c.test(l)||!c.test(a))throw new Error("invalid_job");if(!Number.isFinite(Date.parse(n("createdAt"))))throw new Error("invalid_job");if(!/^[A-Za-z0-9_-]{1,128}$/.test(n("jobId")))throw new Error("invalid_job");if(!["darwin","linux","win32"].includes(n("platform")))throw new Error("invalid_job");const f=r.oldDaemonPid==null?null:Number(r.oldDaemonPid),s=Number(r.daemonExitTimeoutMs),u=Number(r.commandTimeoutMs);if(f!=null&&(!Number.isSafeInteger(f)||f<1))throw new Error("invalid_job");if(!Number.isSafeInteger(s)||s<1e3)throw new Error("invalid_job");if(!Number.isSafeInteger(u)||u<1e4)throw new Error("invalid_job");return{protocolVersion:1,jobId:n("jobId"),fromVersion:l,targetVersion:a,npmPrefix:n("npmPrefix"),npmExecutable:n("npmExecutable"),nodeExecutable:n("nodeExecutable"),packageName:"shennian",operation:r.operation==="rollback"?"rollback":"upgrade",oldDaemonPid:f,platform:n("platform"),createdAt:n("createdAt"),daemonExitTimeoutMs:s,commandTimeoutMs:u,resultPath:n("resultPath"),attemptPath:n("attemptPath"),isolatedDaemonStart:r.isolatedDaemonStart===!0,requireHandoffAck:r.requireHandoffAck===!0,...typeof r.rollbackJobId=="string"&&/^[A-Za-z0-9_-]{1,128}$/.test(r.rollbackJobId)?{rollbackJobId:r.rollbackJobId}:{}}}function O(e,t){return t==="win32"?i.win32.join(e,"node_modules","shennian"):i.posix.join(e,"lib","node_modules","shennian")}function x(e){const t=O(e.npmPrefix,e.platform),r=i.join(i.dirname(t),`.shennian-upgrade-${e.rollbackJobId??e.jobId}`),n=i.join(r,"candidate");return{active:t,transaction:r,prefix:n,candidate:O(n,e.platform),previous:i.join(r,"previous"),rejected:i.join(r,"rejected"),marker:i.join(r,"transaction.json")}}function m(e){try{return JSON.parse(o.readFileSync(e,"utf8"))}catch{return null}}function T(e){const t=o.lstatSync(e);if(!t.isDirectory()||t.isSymbolicLink())throw new Error("unsafe_installation")}function y(e,t){T(e);const r=o.realpathSync(e),n=s=>{const u=i.relative(r,o.realpathSync(s));if(u.startsWith(".."+i.sep)||u===".."||i.isAbsolute(u))throw new Error("external_package_dependency")},c=s=>{for(const u of o.readdirSync(s,{withFileTypes:!0})){const w=i.join(s,u.name);u.isSymbolicLink()?n(w):u.isDirectory()&&c(w)}};c(e);const l=m(i.join(e,"package.json"));if(l?.name!=="shennian"||l.version!==t||l.bin?.shennian!=="dist/bin/shennian.js")throw new Error("verification_failed");n(i.join(e,"dist","bin","shennian.js"));const a=new Set,f=s=>{const u=o.realpathSync(s);if(a.has(u))return;a.add(u);const w=m(i.join(s,"package.json"));if(!w)throw new Error("incomplete_package");for(const g of Object.keys(w.dependencies??{})){if(Object.hasOwn(w.optionalDependencies??{},g))continue;if(!/^(?:@[a-z0-9._-]+\/)?[a-z0-9._-]+$/i.test(g))throw new Error("invalid_dependency");let v=s,_=!1;for(;v===e||v.startsWith(e+i.sep);){const S=i.join(v,"node_modules",g);if(o.existsSync(i.join(S,"package.json"))){n(S),f(S),_=!0;break}v=i.dirname(v)}if(!_)throw new Error("incomplete_package")}};f(e)}function ye(e){const t=m(i.join(e,"updater.lock"));if($(t))throw new Error("upgrade_locked")}async function Q(e,t={}){const r=x(e);T(i.dirname(r.active)),y(r.active,e.fromVersion),o.mkdirSync(r.transaction,{mode:448});try{await ie(e,r,t),await ee(e,t)}catch(n){throw o.rmSync(r.transaction,{recursive:!0,force:!0}),Object.assign(new Error(j(n)),{code:"candidate_preparation_failed"})}}async function ee(e,t={}){const r=x(e);T(r.transaction);const n=m(r.marker);if(n&&n.phase!=="prepared")throw new Error("transaction_already_started");y(r.active,e.fromVersion),y(r.candidate,e.targetVersion);const c=t.execFile??A;await b({...e,npmPrefix:r.prefix},e.targetVersion,c),await c(e.npmExecutable,["ls","-g",`shennian@${e.targetVersion}`,"--all","--omit=dev","--offline","--depth=100","--prefix",r.prefix],h(e,3e4)),k(r.marker,{jobId:e.jobId,fromVersion:e.fromVersion,targetVersion:e.targetVersion,phase:"prepared"})}function F(e,t){if(o.renameSync(e,t),process.platform!=="win32")for(const r of new Set([i.dirname(e),i.dirname(t)])){const n=o.openSync(r,"r");try{o.fsyncSync(n)}finally{o.closeSync(n)}}}function re(e){const t=x(e),r=i.dirname(e.resultPath);for(const n of o.readdirSync(i.dirname(t.active),{withFileTypes:!0})){if(!n.isDirectory()||!/^\.shennian-upgrade-[A-Za-z0-9_-]+$/.test(n.name))continue;const c=i.join(i.dirname(t.active),n.name);if(c===t.transaction)continue;const l=m(i.join(c,"transaction.json"));if(!l?.jobId||n.name!==`.shennian-upgrade-${l.jobId}`)continue;const a=m(i.join(r,`${l.jobId}.json`));a?.jobId===l.jobId&&["healthy","rolled_back"].includes(a.state??"")&&o.rmSync(c,{recursive:!0,force:!0})}}function te(e){const t=i.join(i.dirname(e.attemptPath),"upgrade-failures.json");let r={versions:{}};try{r=JSON.parse(o.readFileSync(t,"utf8"))}catch{}const n=r.versions?.[e.targetVersion],c=Math.max(0,Number(n?.failureCount)||0)+1,l=Date.now(),a=[5*6e4,30*6e4,120*6e4,360*6e4];k(t,{versions:{...r.versions,[e.targetVersion]:{version:e.targetVersion,failureCount:c,firstFailedAt:n?.firstFailedAt??l,lastFailedAt:l,nextRetryAt:l+a[Math.min(c-1,a.length-1)],error:"upgrade_rolled_back"}}})}function E(e,t){const r=x(e),n=m(r.marker);if(!n||n.jobId!==(e.rollbackJobId??e.jobId))throw new Error("invalid_transaction");k(r.marker,{...n,phase:t})}function $(e){const t=Date.now()-D.uptime()*1e3;return!!(e?.pid&&(e.createdAt===void 0||e.createdAt>=t-2e3)&&I(e.pid))}function I(e){try{return process.kill(e,0),!0}catch{return!1}}async function N(e,t){if(t.serviceRecovery||!e.oldDaemonPid||!(t.isAlive??I)(e.oldDaemonPid))return;const r=t.kill??((l,a)=>process.kill(l,a)),n=t.sleep??(l=>new Promise(a=>setTimeout(a,l)));r(e.oldDaemonPid,"SIGTERM");const c=(t.now??Date.now)()+e.daemonExitTimeoutMs;for(;(t.now??Date.now)()<c;){if(!(t.isAlive??I)(e.oldDaemonPid))return;await n(100)}if(t.kill)r(e.oldDaemonPid,"SIGKILL");else if(e.platform==="win32")try{await(t.execFile??A)("taskkill",["/PID",String(e.oldDaemonPid),"/T","/F"],{timeout:15e3,windowsHide:!0,shell:!0})}catch{}else try{process.kill(-e.oldDaemonPid,"SIGKILL")}catch{try{process.kill(e.oldDaemonPid,"SIGKILL")}catch{}}if(await n(250),(t.isAlive??I)(e.oldDaemonPid))throw new Error("daemon_exit_timeout")}function h(e,t=e.commandTimeoutMs){return{timeout:t,windowsHide:!0,shell:e.platform==="win32"}}async function z(e,t,r){const n=["install","-g",`shennian@${t}`,"--prefix",e.npmPrefix];try{await r(e.npmExecutable,n,h(e))}catch{await r(e.npmExecutable,[...n,"--registry",X],h(e))}}async function ne(e,t,r,n,c){try{if(typeof e.url!="string"||!/^https?:\/\//.test(e.url)||!Number.isSafeInteger(e.size)||e.size<=0||e.size>200*1024*1024||typeof e.integrity!="string"||!/^sha512-[A-Za-z0-9+/]+={0,2}$/.test(e.integrity))return null;const l=await c(e.url,{signal:AbortSignal.timeout(r)});if(!l.ok)return null;const a=Buffer.from(await l.arrayBuffer());if(a.byteLength!==e.size||`sha512-${U("sha512").update(a).digest("base64")}`!==e.integrity)return null;const s=i.join(n,`shennian-${t}.tgz`);return o.writeFileSync(s,a,{mode:384}),s}catch{return null}}async function ie(e,t,r){const n=r.execFile??A,c=async()=>{const a=await r.fetchOwnedArtifact?.(e.targetVersion);if(!a)return!1;const f=await ne(a,e.targetVersion,e.commandTimeoutMs,t.transaction,r.fetch??globalThis.fetch);if(!f)return!1;try{return await n(e.npmExecutable,["install","-g",f,"--prefix",t.prefix,"--offline","--ignore-scripts","--no-audit","--no-fund"],h(e)),!0}catch{return o.rmSync(t.prefix,{recursive:!0,force:!0}),!1}};if((r.sourceOrder??"npm-first")==="oss-first"){if(await c())return;await z({...e,npmPrefix:t.prefix},e.targetVersion,n);return}try{await z({...e,npmPrefix:t.prefix},e.targetVersion,n);return}catch(a){if(o.rmSync(t.prefix,{recursive:!0,force:!0}),await c())return;throw a}}async function b(e,t,r){const n=O(e.npmPrefix,e.platform),c=i.join(n,"dist","bin","shennian.js"),{stdout:l}=await r(e.nodeExecutable,[c,"--version"],h(e,1e4));if(l.trim()!==t)throw new Error("verification_failed");return c}async function P(e,t,r){if(e.isolatedDaemonStart){await r(e.nodeExecutable,[t,"start","--json"],h(e,3e4));return}try{if(e.platform==="darwin"){const n=typeof process.getuid=="function"?process.getuid():D.userInfo().uid;try{await r("launchctl",["bootstrap",`gui/${n}`,i.join(D.homedir(),"Library/LaunchAgents/com.shennian.agent.plist")],h(e,15e3))}catch{}await r("launchctl",["kickstart","-k",`gui/${n}/com.shennian.agent`],h(e,15e3));return}if(e.platform==="linux"){await r("systemctl",["--user","restart","shennian"],h(e,15e3));return}if(e.platform==="win32"){await r("schtasks",["/Run","/TN","ShennianAgent"],h(e,15e3));return}}catch{}await r(e.nodeExecutable,[t,"start","--json"],h(e,3e4))}async function C(e,t){if(!e.isolatedDaemonStart)try{e.platform==="darwin"?await t("launchctl",["bootout",`gui/${process.getuid?.()??D.userInfo().uid}/com.shennian.agent`],h(e,15e3)):e.platform==="linux"?await t("systemctl",["--user","stop","shennian"],h(e,15e3)):e.platform==="win32"&&await t("schtasks",["/End","/TN","ShennianAgent"],h(e,15e3))}catch{}}function oe(e){return(t,r={})=>k(e.resultPath,{protocolVersion:J,jobId:e.jobId,state:t,fromVersion:e.fromVersion,targetVersion:e.targetVersion,operation:e.operation,...r,updatedAt:new Date().toISOString()})}function ae(e,t){const r=i.dirname(i.dirname(e)),n=i.join(r,"updater.lock");o.mkdirSync(r,{recursive:!0,mode:448});try{const c=o.openSync(n,"wx",384);o.writeFileSync(c,JSON.stringify({jobId:t.jobId,phase:"worker",pid:process.pid,createdAt:Date.now()})),o.closeSync(c)}catch(c){if(c.code!=="EEXIST")throw c;try{const a=JSON.parse(o.readFileSync(n,"utf8"));if((a.jobId!==t.jobId||a.phase!=="handoff"&&a.pid!==t.oldDaemonPid)&&$(a))throw new Error("upgrade_locked")}catch(a){if(a instanceof Error&&a.message==="upgrade_locked")throw a}o.rmSync(n,{force:!0});const l=o.openSync(n,"wx",384);o.writeFileSync(l,JSON.stringify({jobId:t.jobId,phase:"worker",pid:process.pid,createdAt:Date.now()})),o.closeSync(l)}return()=>{try{o.unlinkSync(n)}catch{}}}async function M(e,t={}){let r;try{r=L(e)}catch{throw new Error("invalid_job")}if(!t.serviceRecovery&&(t.now??Date.now)()-Date.parse(r.createdAt)>Y)throw new Error("job_expired");const n=i.dirname(i.dirname(i.resolve(e)));if(i.basename(i.dirname(i.resolve(e)))!=="jobs"||i.basename(i.resolve(e))!==`${r.jobId}.json`)throw new Error("invalid_job");if(i.dirname(i.resolve(r.resultPath))!==i.join(n,"results"))throw new Error("invalid_job");if(i.basename(i.resolve(r.resultPath))!==`${r.jobId}.json`)throw new Error("invalid_job");const c=i.dirname(i.dirname(n));if(i.resolve(r.attemptPath)!==i.join(c,"upgrade-attempt.json"))throw new Error("invalid_job");const l=ae(e,r),a=oe(r),f=t.execFile??A,s=x(r),u=i.join(n,"active-job.json"),w=t.sleep??(d=>new Promise(p=>setTimeout(p,d)));let g=!1,v={};const _=d=>{d&&te(r),k(r.attemptPath,{from:d?r.targetVersion:r.fromVersion,to:d?r.fromVersion:r.targetVersion,attemptCount:0,attemptAt:Date.now(),jobId:r.jobId,operation:d?"rollback":"upgrade"}),a("starting",d?{operation:"rollback",...v,effectiveVersion:r.fromVersion,targetVersion:r.fromVersion,fromVersion:r.targetVersion}:{})},S=async d=>{if(t.serviceRecovery)return;const p=(t.now??Date.now)()+(t.healthTimeoutMs??9e4);for(;(t.now??Date.now)()<p;){if(m(r.resultPath)?.state===(d?"rolled_back":"healthy"))return;await w(250)}throw new Error("daemon_health_timeout")},Z=async()=>{const d=m(i.join(c,"daemon.pid"));d?.version===r.targetVersion&&Number.isSafeInteger(d.pid)&&typeof d.pid=="number"&&d.pid!==process.pid&&await N({...r,oldDaemonPid:d.pid},t)},R=async()=>{if(E(r,"rolling_back"),t.serviceRecovery||await C(r,f),await Z(),!o.existsSync(s.previous))y(s.active,r.fromVersion);else{if(y(s.previous,r.fromVersion),o.existsSync(s.active)){if(o.existsSync(s.rejected))throw new Error("rollback_destination_exists");F(s.active,s.rejected)}F(s.previous,s.active)}E(r,"restored"),y(s.active,r.fromVersion);const d=await b(r,r.fromVersion,f);_(!0),t.serviceRecovery||await P(r,d,f),await S(!0)};try{const d=m(r.resultPath);if(d?.jobId===r.jobId&&["healthy","rolled_back","failed"].includes(d.state??""))return;const p=m(s.marker);if(p&&(p.jobId!==(r.rollbackJobId??r.jobId)||p.fromVersion!==(r.operation==="rollback"?r.targetVersion:r.fromVersion)))throw new Error("invalid_transaction");if(r.operation==="rollback"){r={...r,fromVersion:r.targetVersion,targetVersion:r.fromVersion},g=!0,k(u,{jobPath:e}),await N(r,t),await R();return}if(p&&p.phase!=="prepared"){g=!0,p.phase==="restored"||p.phase==="rolling_back"&&!o.existsSync(s.previous)?(y(s.active,r.fromVersion),E(r,"restored"),_(!0),t.serviceRecovery||await P(r,await b(r,r.fromVersion,f),f),await S(!0)):p.phase==="switching"&&!o.existsSync(s.previous)?(y(s.active,r.fromVersion),E(r,"restored"),_(!0),t.serviceRecovery||await P(r,await b(r,r.fromVersion,f),f),await S(!0)):await R();return}if(p||(a("installing"),await Q(r,t)),y(s.candidate,r.targetVersion),y(s.active,r.fromVersion),a("waiting_for_daemon"),r.requireHandoffAck){const G=(t.now??Date.now)()+1e4;for(;m(`${r.resultPath}.handoff`)?.jobId!==r.jobId;){if((t.now??Date.now)()>=G)throw new Error("handoff_not_confirmed");await w(50)}}await w(250),k(u,{jobPath:e}),t.serviceRecovery||await C(r,f),await N(r,t),E(r,"switching"),g=!0,F(s.active,s.previous),F(s.candidate,s.active),a("verifying"),y(s.active,r.targetVersion);const V=await b(r,r.targetVersion,f);E(r,"starting"),_(!1),t.serviceRecovery||await P(r,V,f),await S(!1)}catch(d){const p=d.code??(d instanceof Error?d.message:"internal_error");if(!g){a("failed",{errorCode:p,errorSummary:j(d)}),(!r.oldDaemonPid||!(t.isAlive??I)(r.oldDaemonPid))&&(y(s.active,r.fromVersion),t.serviceRecovery||await P(r,await b(r,r.fromVersion,f),f));return}v={errorCode:p,errorSummary:j(d)};try{await R()}catch(V){throw a("failed",{errorCode:"rollback_failed",errorSummary:j(V)}),V}}finally{const d=m(r.resultPath);if(["healthy","rolled_back"].includes(d?.state??""))try{re(r)}catch{}l()}}async function se(e){const t=m(i.join(e,"active-job.json"));if(!t?.jobPath)return;const r=i.resolve(t.jobPath);if(i.dirname(r)!==i.join(i.resolve(e),"jobs"))throw new Error("invalid_job");const n=L(r);for(;;){const c=m(n.resultPath);if(["healthy","rolled_back","failed"].includes(c?.state??""))return;const l=m(i.join(e,"updater.lock")),a=l?.pid!==process.pid&&$(l),f=m(x(n).marker);if(a){if(c?.state==="starting"&&["starting","restored"].includes(f?.phase??""))return;await new Promise(s=>setTimeout(s,250));continue}await M(r,{serviceRecovery:!0});return}}function H(e){try{return o.realpathSync.native(e)}catch{return i.resolve(e)}}const ce=process.argv[1]?H(process.argv[1]):"",le=H(K(import.meta.url));if(ce===le)if(process.argv[2]==="--service"){const[,,,e,t,...r]=process.argv;se(e).then(async()=>{process.argv=[process.execPath,t,...r],await import(W(t).href)}).catch(()=>{process.exitCode=1})}else{const e=process.argv[2];e?M(e).catch(()=>{process.exitCode=1}):process.exitCode=2}export{ye as assertUpdaterAvailable,ee as finalizeUpdaterCandidate,Q as prepareUpdaterCandidate,se as recoverServiceEntry,M as runUpdaterJob,x as updaterPaths,y as verifyLocalPackage};