@automagik/genie 4.260501.4 → 4.260501.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/genie.js
CHANGED
|
@@ -695,14 +695,14 @@ ${bin} set-option -w pane-active-border-style "fg=$COLOR"
|
|
|
695
695
|
)
|
|
696
696
|
`)}async function recordMigrationComplete(target,sourceDb,rowsCopied){await ensureMigrationStateTable(target),await target.unsafe(`INSERT INTO _genie_migration_state (source_db, rows_copied)
|
|
697
697
|
VALUES ($1, $2::jsonb)
|
|
698
|
-
ON CONFLICT (source_db) DO UPDATE SET completed_at = now(), rows_copied = EXCLUDED.rows_copied`,[sourceDb,JSON.stringify(rowsCopied)])}var V1_DB_NAME="genie",PG_USER="postgres",PG_PASS="postgres",PG_HOST="127.0.0.1",promptedThisProcess=!1;var init_v1_migration_prompt=__esm(()=>{init_src()});var exports_db={};__export(exports_db,{shutdown:()=>shutdown,runRetention:()=>runRetention,resolveTcpPgPassword:()=>resolveTcpPgPassword,resolvePgserveSocketDir:()=>resolvePgserveSocketDir,resolvePgserveLibpqSocketPath:()=>resolvePgserveLibpqSocketPath,resolvePgserveDaemonPidPath:()=>resolvePgserveDaemonPidPath,resolvePgserveControlSocketPath:()=>resolvePgserveControlSocketPath,resolvePgserveAuthPassword:()=>resolvePgserveAuthPassword,resolveDatabaseName:()=>resolveDatabaseName,resetConnection:()=>resetConnection,readPostmasterDiscovery:()=>readPostmasterDiscovery,probePgserveDaemon:()=>probePgserveDaemon,pinCwdToGeniePackageDir:()=>pinCwdToGeniePackageDir,isSocketMode:()=>isSocketMode,isConnected:()=>isConnected,isAvailable:()=>isAvailable,getOrStartDaemon:()=>getOrStartDaemon,getLockfilePath:()=>getLockfilePath,getDataDir:()=>getDataDir,getConnection:()=>getConnection,getAuxiliaryPortBase:()=>getAuxiliaryPortBase,getActivePort:()=>getActivePort,ensurePgserve:()=>ensurePgserve,checkRootGuard:()=>checkRootGuard,autoStartDaemon:()=>autoStartDaemon,__setSpawnDaemonForTest:()=>__setSpawnDaemonForTest,DB_NAME:()=>DB_NAME});import{execFileSync,execSync as execSync3,spawn}from"child_process";import{existsSync as existsSync13,mkdirSync as mkdirSync6,readFileSync as readFileSync9,renameSync,unlinkSync as unlinkSync3,writeFileSync as writeFileSync6}from"fs";import{createConnection}from"net";import{homedir as homedir11}from"os";import{basename as basename2,dirname as dirname5,join as join17}from"path";import{pathToFileURL}from"url";function resolvePgserveSocketDir(){let xdg=process.env.XDG_RUNTIME_DIR,base=xdg&&xdg.length>0?xdg:"/tmp";return join17(base,"pgserve")}function resolveDatabaseName(){let testDbName=process.env.GENIE_TEST_DB_NAME;if(testDbName&&testDbName.length>0)return testDbName;return DB_NAME}function resolvePgserveAuthPassword(){let password=process.env.PGPASSWORD;return password&&password.length>0?password:DB_NAME}function resolvePgserveTimeoutMs(){let parsed=Number(process.env.GENIE_PGSERVE_TIMEOUT);return Number.isFinite(parsed)&&parsed>0?parsed:16000}function resolvePgConnectTimeoutSeconds(useSocket){let parsed=Number(process.env.GENIE_PG_CONNECT_TIMEOUT);if(Number.isFinite(parsed)&&parsed>0)return Math.ceil(parsed);if(!useSocket)return 5;return Math.max(16,Math.ceil(resolvePgserveTimeoutMs()/1000))}function resolveTcpPgPassword(){return resolvePgserveAuthPassword()}function resolvePgserveLibpqSocketPath(){return join17(resolvePgserveSocketDir(),".s.PGSQL.5432")}function resolvePgserveControlSocketPath(){return join17(resolvePgserveSocketDir(),"control.sock")}function resolvePgserveDaemonPidPath(){return join17(resolvePgserveSocketDir(),"pgserve.pid")}function readPostmasterDiscovery(){let file=join17(resolvePgserveSocketDir(),"admin.json");if(!existsSync13(file))return null;let raw;try{raw=readFileSync9(file,"utf-8")}catch{return null}let parsed;try{parsed=JSON.parse(raw)}catch{return null}if(!parsed||typeof parsed!=="object")return null;let obj=parsed,socketDir=typeof obj.socketDir==="string"&&obj.socketDir.length>0?obj.socketDir:null,port=typeof obj.port==="number"&&Number.isFinite(obj.port)&&obj.port>0?obj.port:null,pid=typeof obj.pid==="number"&&Number.isFinite(obj.pid)&&obj.pid>0?obj.pid:null;if(!socketDir||!port)return null;if(!existsSync13(join17(socketDir,`.s.PGSQL.${port}`)))return null;return{socketDir,port,pid:pid??-1}}function probePgserveDaemon(){let socketPresent=existsSync13(resolvePgserveLibpqSocketPath()),pid=liveDaemonPid(readDaemonPid(resolvePgserveDaemonPidPath()));if(socketPresent&&pid!==null)return{running:!0,pid,socketPresent:!0};if(!socketPresent&&pid===null)return{running:!1,pid:null,socketPresent:!1,reason:"no daemon"};return{running:!1,pid,socketPresent,reason:socketPresent?"socket present but pid stale":"pid alive but no socket"}}function readDaemonPid(pidPath){if(!existsSync13(pidPath))return null;try{let raw=readFileSync9(pidPath,"utf-8").trim(),parsed=Number.parseInt(raw,10);return Number.isInteger(parsed)&&parsed>0?parsed:null}catch{return null}}function liveDaemonPid(pid){if(pid===null)return null;try{return process.kill(pid,0),pid}catch(err){return err.code==="EPERM"?pid:null}}function resolveGeniePackageDir(){if(geniePackageDirCache!==void 0)return geniePackageDirCache;let PACKAGE_NAME="@automagik/genie",MAX_WALK_DEPTH=10,current=dirname5(import.meta.dir??__dirname);for(let depth=0;depth<MAX_WALK_DEPTH;depth++){let candidate=join17(current,"package.json");try{if(existsSync13(candidate)){if(JSON.parse(readFileSync9(candidate,"utf-8"))?.name===PACKAGE_NAME)return geniePackageDirCache=current,current}}catch{}let parent=dirname5(current);if(parent===current)break;current=parent}return geniePackageDirCache=null,null}function pinCwdToGeniePackageDir(){let previous=process.cwd();if(daemonCwdPinned)return{previous,pinned:previous};let dir=resolveGeniePackageDir();if(!dir)return{previous,pinned:null};try{if(process.cwd()!==dir)process.chdir(dir);return daemonCwdPinned=!0,{previous,pinned:dir}}catch{return{previous,pinned:null}}}function findLocalPgserveRoot(){let candidates=[join17(import.meta.dir,"..","node_modules","pgserve"),join17(import.meta.dir,"..","..","node_modules","pgserve")];for(let root of candidates)if(existsSync13(join17(root,"package.json")))return root;return null}function resolveLocalPgserveEntry(){let root=findLocalPgserveRoot();if(root===null)return null;try{let pkg=JSON.parse(readFileSync9(join17(root,"package.json"),"utf-8"));return join17(root,pkg.main??"src/index.js")}catch{return join17(root,"src/index.js")}}async function importPgserveSdk(){let localEntry=resolveLocalPgserveEntry();if(localEntry!==null&&existsSync13(localEntry))try{return await import(pathToFileURL(localEntry).href)}catch{}try{return await import("pgserve")}catch{return null}}function findPgserveDaemonCommand(){let localRoot=findLocalPgserveRoot();if(localRoot!==null){let localCommand=resolvePgservePackageCommand(localRoot);if(localCommand!==null)return localCommand}try{let resolved=__require.resolve("pgserve/bin/pgserve-wrapper.cjs"),packageCommand=resolvePgservePackageCommand(join17(dirname5(resolved),".."));if(packageCommand!==null)return packageCommand}catch{}let globalBin=join17(homedir11(),".bun","bin","pgserve");if(existsSync13(globalBin))return{command:globalBin,argsPrefix:[],display:globalBin};try{let fromPath=execSync3("which pgserve",{encoding:"utf-8",timeout:3000}).trim();if(fromPath.length>0)return{command:fromPath,argsPrefix:[],display:fromPath}}catch{}return null}function resolvePgservePackageCommand(root){let script=join17(root,"bin","postgres-server.js"),bun=findBunRuntime();if(existsSync13(script)&&bun!==null)return{command:bun,argsPrefix:[script],display:`${bun} ${script}`};let wrapper=join17(root,"bin","pgserve-wrapper.cjs");if(existsSync13(wrapper))return{command:wrapper,argsPrefix:[],display:wrapper};return null}function findBunRuntime(){if(process.execPath&&existsSync13(process.execPath)&&basename2(process.execPath).startsWith("bun"))return process.execPath;let homeBun=join17(homedir11(),".bun","bin",process.platform==="win32"?"bun.exe":"bun");if(existsSync13(homeBun))return homeBun;try{let fromPath=execSync3("which bun",{encoding:"utf-8",timeout:3000}).trim();if(fromPath.length>0)return fromPath}catch{}return null}function sleep(ms){return new Promise((r)=>setTimeout(r,ms))}async function getOrStartDaemon(){if(process.env.GENIE_PG_DISABLE_AUTOSTART==="1"||isPgAutostartDisabled()){let state=probePgserveDaemon();if(state.running&&await isPgserveSocketResponsive())return state;throw Error("pgserve daemon unavailable and PG autostart is disabled")}let initial=probePgserveDaemon();if(initial.running){if(await isPgserveSocketResponsive())return initial;await recoverUnresponsivePgserveDaemon(initial)}if(process.env.CI==="true"&&process.env.GENIE_PG_ALLOW_CI_AUTOSTART!=="1")throw Error("pgserve v2 daemon socket not present and CI=true. Either start `pgserve daemon` in the workflow or set GENIE_PG_ALLOW_CI_AUTOSTART=1 to opt in.");let rootErr=checkRootGuard();if(rootErr!==null)throw Error(rootErr);if(initial.reason==="socket present but pid stale"&&await isPgserveSocketResponsive())return{running:!0,pid:null,socketPresent:!0,reason:"socket completes pgserve greeting but pid file is stale"};if(daemonStartPromise)return await daemonStartPromise,probePgserveDaemon();daemonStartPromise=startPgserveDaemonOnce(initial);try{await daemonStartPromise}finally{daemonStartPromise=null}return probePgserveDaemon()}function cleanPartialDaemonState(initial){if(initial.reason==="pid alive but no socket"&&initial.pid!==null)unlinkIfPresent(resolvePgserveDaemonPidPath());if(initial.reason==="socket present but pid stale")removeStalePgserveSocketArtifacts()}async function recoverUnresponsivePgserveDaemon(state){if(state.pid!==null&&isLikelyPgserveDaemonProcess(state.pid)){if(await signalPgserveDaemonPid(state.pid,"SIGTERM"),liveDaemonPid(state.pid)!==null)await signalPgserveDaemonPid(state.pid,"SIGKILL")}removeStalePgserveSocketArtifacts()}function isLikelyPgserveDaemonProcess(pid){try{let command=execFileSync("ps",["-p",String(pid),"-o","command="],{encoding:"utf-8",timeout:1000}).trim();return command.includes("pgserve")||command.includes("postgres-server.js")||command.includes("postgres")&&command.includes(DATA_DIR)}catch{return!1}}async function signalPgserveDaemonPid(pid,signal){let signaled=!1;try{process.kill(-pid,signal),signaled=!0}catch{}try{process.kill(pid,signal),signaled=!0}catch{}if(!signaled)return;let deadline=Date.now()+(signal==="SIGTERM"?1000:250);while(Date.now()<deadline){if(liveDaemonPid(pid)===null)return;await sleep(50)}}async function isPgserveSocketResponsive(){let candidates=[resolvePgserveLibpqSocketPath(),resolvePgserveControlSocketPath()].filter((path2)=>existsSync13(path2));for(let path2 of candidates)if(await canCompletePgserveGreet(path2))return!0;return!1}function canCompletePgserveGreet(path2){return new Promise((resolve3)=>{let settled=!1,timer2=null,socket=null,finish=(ok)=>{if(settled)return;if(settled=!0,timer2)clearTimeout(timer2);socket?.removeAllListeners(),socket?.destroy(),resolve3(ok)},request=Buffer.alloc(8);request.writeUInt32BE(8,0),request.writeUInt32BE(PG_SSL_REQUEST_CODE,4),socket=createConnection(path2),timer2=setTimeout(()=>finish(!1),PGSERVE_GREET_TIMEOUT_MS),timer2.unref(),socket.once("connect",()=>socket?.write(request)),socket.once("data",(chunk)=>finish(chunk[0]===78||chunk[0]===83)),socket.once("error",()=>finish(!1))})}function removeStalePgserveSocketArtifacts(){for(let path2 of[resolvePgserveDaemonPidPath(),resolvePgserveLibpqSocketPath(),resolvePgserveControlSocketPath()])unlinkIfPresent(path2)}function unlinkIfPresent(path2){try{unlinkSync3(path2)}catch{}}async function startPgserveDaemonOnce(initial){cleanPartialDaemonState(initial);let orphan=detectOrphanDataDirLock();if(orphan!==null)await evictOrphanDataDirHolder(orphan);let daemonCommand=findPgserveDaemonCommand();if(daemonCommand===null){if(await tryEnsureDaemonWithSdk())return;throw Error("pgserve binary not found. Install with `bun add pgserve@^2.0.2` (or `npm i pgserve@^2.0.2`), or start `pgserve daemon` manually before running genie.")}mkdirSync6(resolvePgserveSocketDir(),{recursive:!0,mode:448});let maxConnections=process.env.GENIE_PG_MAX_CONNECTIONS??"10000",child=spawn(daemonCommand.command,[...daemonCommand.argsPrefix,"daemon","--data",DATA_DIR,"--log","warn","--max-connections",maxConnections],{detached:!0,stdio:"ignore",env:{...process.env}});try{await waitForDaemonSocket(daemonCommand,child),child.unref()}catch(err){throw await terminatePgserveTree(child),err}}function detectOrphanDataDirLock(){let pidFile=join17(DATA_DIR,"postmaster.pid");if(!existsSync13(pidFile))return null;let raw;try{raw=readFileSync9(pidFile,"utf-8")}catch{return null}let firstLine=raw.split(`
|
|
699
|
-
`,1)[0]?.trim()??"",pid=Number.parseInt(firstLine,10);if(!Number.isInteger(pid)||pid<=0)return null;if(liveDaemonPid(pid)===null)return null;let cmd;try{cmd=execFileSync("ps",["-p",String(pid),"-o","command="],{encoding:"utf-8",timeout:1000}).trim()}catch{return null}return cmd.includes("pgserve")||cmd.includes("postgres-server.js")||cmd.includes("postgres")&&cmd.includes(DATA_DIR)?{pid,cmd}:null}async function evictOrphanDataDirHolder(holder){let target=holder.pid;try{let ppidStr=execFileSync("ps",["-p",String(holder.pid),"-o","ppid="],{encoding:"utf-8",timeout:1000}).trim(),ppid=Number.parseInt(ppidStr,10);if(Number.isInteger(ppid)&&ppid>1){let pcmd=execFileSync("ps",["-p",String(ppid),"-o","command="],{encoding:"utf-8",timeout:1000}).trim();if(pcmd.includes("pgserve")||pcmd.includes("postgres-server.js"))target=ppid}}catch{}if(await signalPgserveDaemonPid(target,"SIGTERM"),liveDaemonPid(target)!==null)await signalPgserveDaemonPid(target,"SIGKILL");unlinkIfPresent(join17(DATA_DIR,"postmaster.pid")),removeStalePgserveSocketArtifacts()}async function waitForDaemonSocket(daemonCommand,child){let socketPath=resolvePgserveLibpqSocketPath(),timeout=resolvePgserveTimeoutMs(),deadline=Date.now()+timeout,childExit=null;child?.once("exit",(code,signal)=>{childExit=signal?`signal ${signal}`:`exit code ${code??"unknown"}`});while(Date.now()<deadline){if(existsSync13(socketPath)&&await isPgserveSocketResponsive())return;if(childExit!==null){let holder=detectOrphanDataDirLock(),holderHint=holder!==null?` Data directory ${DATA_DIR} is held by PID ${holder.pid} (${holder.cmd}); kill it (or run \`genie serve restart\`) and retry.`:"";throw Error(`pgserve v2 daemon exited before binding ${socketPath} (${childExit}).${holderHint} Try starting it manually: ${formatPgserveDaemonCommand(daemonCommand)}`)}await sleep(250)}throw Error(`pgserve v2 daemon did not bind ${socketPath} within ${Math.round(timeout/1000)}s. Try starting it manually: ${formatPgserveDaemonCommand(daemonCommand)}`)}function formatPgserveDaemonCommand(daemonCommand){return`${daemonCommand.display} daemon --data ${DATA_DIR} --log warn`}
|
|
698
|
+
ON CONFLICT (source_db) DO UPDATE SET completed_at = now(), rows_copied = EXCLUDED.rows_copied`,[sourceDb,JSON.stringify(rowsCopied)])}var V1_DB_NAME="genie",PG_USER="postgres",PG_PASS="postgres",PG_HOST="127.0.0.1",promptedThisProcess=!1;var init_v1_migration_prompt=__esm(()=>{init_src()});var exports_db={};__export(exports_db,{shutdown:()=>shutdown,runRetention:()=>runRetention,resolveTcpPgPassword:()=>resolveTcpPgPassword,resolvePgserveSocketDir:()=>resolvePgserveSocketDir,resolvePgserveLibpqSocketPath:()=>resolvePgserveLibpqSocketPath,resolvePgserveDaemonPidPath:()=>resolvePgserveDaemonPidPath,resolvePgserveControlSocketPath:()=>resolvePgserveControlSocketPath,resolvePgserveAuthPassword:()=>resolvePgserveAuthPassword,resolveDatabaseName:()=>resolveDatabaseName,resetConnection:()=>resetConnection,readPostmasterDiscovery:()=>readPostmasterDiscovery,probePgserveDaemon:()=>probePgserveDaemon,pinCwdToGeniePackageDir:()=>pinCwdToGeniePackageDir,isSocketMode:()=>isSocketMode,isConnected:()=>isConnected,isAvailable:()=>isAvailable,getOrStartDaemon:()=>getOrStartDaemon,getLockfilePath:()=>getLockfilePath,getDataDir:()=>getDataDir,getConnection:()=>getConnection,getAuxiliaryPortBase:()=>getAuxiliaryPortBase,getActivePort:()=>getActivePort,ensurePgserve:()=>ensurePgserve,checkRootGuard:()=>checkRootGuard,autoStartDaemon:()=>autoStartDaemon,__setSpawnDaemonForTest:()=>__setSpawnDaemonForTest,DB_NAME:()=>DB_NAME});import{execFileSync,execSync as execSync3,spawn}from"child_process";import{existsSync as existsSync13,mkdirSync as mkdirSync6,readFileSync as readFileSync9,renameSync,unlinkSync as unlinkSync3,writeFileSync as writeFileSync6}from"fs";import{createConnection}from"net";import{homedir as homedir11}from"os";import{basename as basename2,dirname as dirname5,join as join17}from"path";function resolvePgserveSocketDir(){let xdg=process.env.XDG_RUNTIME_DIR,base=xdg&&xdg.length>0?xdg:"/tmp";return join17(base,"pgserve")}function resolveDatabaseName(){let testDbName=process.env.GENIE_TEST_DB_NAME;if(testDbName&&testDbName.length>0)return testDbName;return DB_NAME}function resolvePgserveAuthPassword(){let password=process.env.PGPASSWORD;return password&&password.length>0?password:DB_NAME}function resolvePgserveTimeoutMs(){let parsed=Number(process.env.GENIE_PGSERVE_TIMEOUT);return Number.isFinite(parsed)&&parsed>0?parsed:16000}function resolvePgConnectTimeoutSeconds(useSocket){let parsed=Number(process.env.GENIE_PG_CONNECT_TIMEOUT);if(Number.isFinite(parsed)&&parsed>0)return Math.ceil(parsed);if(!useSocket)return 5;return Math.max(16,Math.ceil(resolvePgserveTimeoutMs()/1000))}function resolveTcpPgPassword(){return resolvePgserveAuthPassword()}function resolvePgserveLibpqSocketPath(){return join17(resolvePgserveSocketDir(),".s.PGSQL.5432")}function resolvePgserveControlSocketPath(){return join17(resolvePgserveSocketDir(),"control.sock")}function resolvePgserveDaemonPidPath(){return join17(resolvePgserveSocketDir(),"pgserve.pid")}function readPostmasterDiscovery(){let file=join17(resolvePgserveSocketDir(),"admin.json");if(!existsSync13(file))return null;let raw;try{raw=readFileSync9(file,"utf-8")}catch{return null}let parsed;try{parsed=JSON.parse(raw)}catch{return null}if(!parsed||typeof parsed!=="object")return null;let obj=parsed,socketDir=typeof obj.socketDir==="string"&&obj.socketDir.length>0?obj.socketDir:null,port=typeof obj.port==="number"&&Number.isFinite(obj.port)&&obj.port>0?obj.port:null,pid=typeof obj.pid==="number"&&Number.isFinite(obj.pid)&&obj.pid>0?obj.pid:null;if(!socketDir||!port)return null;if(!existsSync13(join17(socketDir,`.s.PGSQL.${port}`)))return null;return{socketDir,port,pid:pid??-1}}function probePgserveDaemon(){let socketPresent=existsSync13(resolvePgserveLibpqSocketPath()),pid=liveDaemonPid(readDaemonPid(resolvePgserveDaemonPidPath()));if(socketPresent&&pid!==null)return{running:!0,pid,socketPresent:!0};if(!socketPresent&&pid===null)return{running:!1,pid:null,socketPresent:!1,reason:"no daemon"};return{running:!1,pid,socketPresent,reason:socketPresent?"socket present but pid stale":"pid alive but no socket"}}function readDaemonPid(pidPath){if(!existsSync13(pidPath))return null;try{let raw=readFileSync9(pidPath,"utf-8").trim(),parsed=Number.parseInt(raw,10);return Number.isInteger(parsed)&&parsed>0?parsed:null}catch{return null}}function liveDaemonPid(pid){if(pid===null)return null;try{return process.kill(pid,0),pid}catch(err){return err.code==="EPERM"?pid:null}}function resolveGeniePackageDir(){if(geniePackageDirCache!==void 0)return geniePackageDirCache;let PACKAGE_NAME="@automagik/genie",MAX_WALK_DEPTH=10,current=dirname5(import.meta.dir??__dirname);for(let depth=0;depth<MAX_WALK_DEPTH;depth++){let candidate=join17(current,"package.json");try{if(existsSync13(candidate)){if(JSON.parse(readFileSync9(candidate,"utf-8"))?.name===PACKAGE_NAME)return geniePackageDirCache=current,current}}catch{}let parent=dirname5(current);if(parent===current)break;current=parent}return geniePackageDirCache=null,null}function pinCwdToGeniePackageDir(){let previous=process.cwd();if(daemonCwdPinned)return{previous,pinned:previous};let dir=resolveGeniePackageDir();if(!dir)return{previous,pinned:null};try{if(process.cwd()!==dir)process.chdir(dir);return daemonCwdPinned=!0,{previous,pinned:dir}}catch{return{previous,pinned:null}}}function findLocalPgserveRoot(){let candidates=[join17(import.meta.dir,"..","node_modules","pgserve"),join17(import.meta.dir,"..","..","node_modules","pgserve")];for(let root of candidates)if(existsSync13(join17(root,"package.json")))return root;return null}function findPgserveDaemonCommand(){let localRoot=findLocalPgserveRoot();if(localRoot!==null){let localCommand=resolvePgservePackageCommand(localRoot);if(localCommand!==null)return localCommand}let globalBin=join17(homedir11(),".bun","bin","pgserve");if(existsSync13(globalBin))return{command:globalBin,argsPrefix:[],display:globalBin};try{let fromPath=execSync3("which pgserve",{encoding:"utf-8",timeout:3000}).trim();if(fromPath.length>0)return{command:fromPath,argsPrefix:[],display:fromPath}}catch{}return null}function resolvePgservePackageCommand(root){let script=join17(root,"bin","postgres-server.js"),bun=findBunRuntime();if(existsSync13(script)&&bun!==null)return{command:bun,argsPrefix:[script],display:`${bun} ${script}`};let wrapper=join17(root,"bin","pgserve-wrapper.cjs");if(existsSync13(wrapper))return{command:wrapper,argsPrefix:[],display:wrapper};return null}function findBunRuntime(){if(process.execPath&&existsSync13(process.execPath)&&basename2(process.execPath).startsWith("bun"))return process.execPath;let homeBun=join17(homedir11(),".bun","bin",process.platform==="win32"?"bun.exe":"bun");if(existsSync13(homeBun))return homeBun;try{let fromPath=execSync3("which bun",{encoding:"utf-8",timeout:3000}).trim();if(fromPath.length>0)return fromPath}catch{}return null}function sleep(ms){return new Promise((r)=>setTimeout(r,ms))}async function getOrStartDaemon(){if(process.env.GENIE_PG_DISABLE_AUTOSTART==="1"||isPgAutostartDisabled()){let state=probePgserveDaemon();if(state.running&&await isPgserveSocketResponsive())return state;throw Error("pgserve daemon unavailable and PG autostart is disabled")}let initial=probePgserveDaemon();if(initial.running){if(await isPgserveSocketResponsive())return initial;await recoverUnresponsivePgserveDaemon(initial)}if(process.env.CI==="true"&&process.env.GENIE_PG_ALLOW_CI_AUTOSTART!=="1")throw Error("pgserve v2 daemon socket not present and CI=true. Either start `pgserve daemon` in the workflow or set GENIE_PG_ALLOW_CI_AUTOSTART=1 to opt in.");let rootErr=checkRootGuard();if(rootErr!==null)throw Error(rootErr);if(initial.reason==="socket present but pid stale"&&await isPgserveSocketResponsive())return{running:!0,pid:null,socketPresent:!0,reason:"socket completes pgserve greeting but pid file is stale"};if(daemonStartPromise)return await daemonStartPromise,probePgserveDaemon();daemonStartPromise=startPgserveDaemonOnce(initial);try{await daemonStartPromise}finally{daemonStartPromise=null}return probePgserveDaemon()}function cleanPartialDaemonState(initial){if(initial.reason==="pid alive but no socket"&&initial.pid!==null)unlinkIfPresent(resolvePgserveDaemonPidPath());if(initial.reason==="socket present but pid stale")removeStalePgserveSocketArtifacts()}async function recoverUnresponsivePgserveDaemon(state){if(state.pid!==null&&isLikelyPgserveDaemonProcess(state.pid)){if(await signalPgserveDaemonPid(state.pid,"SIGTERM"),liveDaemonPid(state.pid)!==null)await signalPgserveDaemonPid(state.pid,"SIGKILL")}removeStalePgserveSocketArtifacts()}function isLikelyPgserveDaemonProcess(pid){try{let command=execFileSync("ps",["-p",String(pid),"-o","command="],{encoding:"utf-8",timeout:1000}).trim();return command.includes("pgserve")||command.includes("postgres-server.js")||command.includes("postgres")&&command.includes(DATA_DIR)}catch{return!1}}async function signalPgserveDaemonPid(pid,signal){let signaled=!1;try{process.kill(-pid,signal),signaled=!0}catch{}try{process.kill(pid,signal),signaled=!0}catch{}if(!signaled)return;let deadline=Date.now()+(signal==="SIGTERM"?1000:250);while(Date.now()<deadline){if(liveDaemonPid(pid)===null)return;await sleep(50)}}async function isPgserveSocketResponsive(){let candidates=[resolvePgserveLibpqSocketPath(),resolvePgserveControlSocketPath()].filter((path2)=>existsSync13(path2));for(let path2 of candidates)if(await canCompletePgserveGreet(path2))return!0;return!1}function canCompletePgserveGreet(path2){return new Promise((resolve3)=>{let settled=!1,timer2=null,socket=null,finish=(ok)=>{if(settled)return;if(settled=!0,timer2)clearTimeout(timer2);socket?.removeAllListeners(),socket?.destroy(),resolve3(ok)},request=Buffer.alloc(8);request.writeUInt32BE(8,0),request.writeUInt32BE(PG_SSL_REQUEST_CODE,4),socket=createConnection(path2),timer2=setTimeout(()=>finish(!1),PGSERVE_GREET_TIMEOUT_MS),timer2.unref(),socket.once("connect",()=>socket?.write(request)),socket.once("data",(chunk)=>finish(chunk[0]===78||chunk[0]===83)),socket.once("error",()=>finish(!1))})}function removeStalePgserveSocketArtifacts(){for(let path2 of[resolvePgserveDaemonPidPath(),resolvePgserveLibpqSocketPath(),resolvePgserveControlSocketPath()])unlinkIfPresent(path2)}function unlinkIfPresent(path2){try{unlinkSync3(path2)}catch{}}async function startPgserveDaemonOnce(initial){cleanPartialDaemonState(initial);let orphan=detectOrphanDataDirLock();if(orphan!==null)await evictOrphanDataDirHolder(orphan);let daemonCommand=findPgserveDaemonCommand();if(daemonCommand===null)throw Error("pgserve binary not found. Install with `bun add -g pgserve@^2.1.0` (or `npm i -g pgserve@^2.1.0`), or run `pgserve install` to register the canonical pm2-supervised daemon.");mkdirSync6(resolvePgserveSocketDir(),{recursive:!0,mode:448});let maxConnections=process.env.GENIE_PG_MAX_CONNECTIONS??"10000",child=spawn(daemonCommand.command,[...daemonCommand.argsPrefix,"daemon","--data",DATA_DIR,"--log","warn","--max-connections",maxConnections],{detached:!0,stdio:"ignore",env:{...process.env}});try{await waitForDaemonSocket(daemonCommand,child),child.unref()}catch(err){throw await terminatePgserveTree(child),err}}function detectOrphanDataDirLock(){let pidFile=join17(DATA_DIR,"postmaster.pid");if(!existsSync13(pidFile))return null;let raw;try{raw=readFileSync9(pidFile,"utf-8")}catch{return null}let firstLine=raw.split(`
|
|
699
|
+
`,1)[0]?.trim()??"",pid=Number.parseInt(firstLine,10);if(!Number.isInteger(pid)||pid<=0)return null;if(liveDaemonPid(pid)===null)return null;let cmd;try{cmd=execFileSync("ps",["-p",String(pid),"-o","command="],{encoding:"utf-8",timeout:1000}).trim()}catch{return null}return cmd.includes("pgserve")||cmd.includes("postgres-server.js")||cmd.includes("postgres")&&cmd.includes(DATA_DIR)?{pid,cmd}:null}async function evictOrphanDataDirHolder(holder){let target=holder.pid;try{let ppidStr=execFileSync("ps",["-p",String(holder.pid),"-o","ppid="],{encoding:"utf-8",timeout:1000}).trim(),ppid=Number.parseInt(ppidStr,10);if(Number.isInteger(ppid)&&ppid>1){let pcmd=execFileSync("ps",["-p",String(ppid),"-o","command="],{encoding:"utf-8",timeout:1000}).trim();if(pcmd.includes("pgserve")||pcmd.includes("postgres-server.js"))target=ppid}}catch{}if(await signalPgserveDaemonPid(target,"SIGTERM"),liveDaemonPid(target)!==null)await signalPgserveDaemonPid(target,"SIGKILL");unlinkIfPresent(join17(DATA_DIR,"postmaster.pid")),removeStalePgserveSocketArtifacts()}async function waitForDaemonSocket(daemonCommand,child){let socketPath=resolvePgserveLibpqSocketPath(),timeout=resolvePgserveTimeoutMs(),deadline=Date.now()+timeout,childExit=null;child?.once("exit",(code,signal)=>{childExit=signal?`signal ${signal}`:`exit code ${code??"unknown"}`});while(Date.now()<deadline){if(existsSync13(socketPath)&&await isPgserveSocketResponsive())return;if(childExit!==null){let holder=detectOrphanDataDirLock(),holderHint=holder!==null?` Data directory ${DATA_DIR} is held by PID ${holder.pid} (${holder.cmd}); kill it (or run \`genie serve restart\`) and retry.`:"";throw Error(`pgserve v2 daemon exited before binding ${socketPath} (${childExit}).${holderHint} Try starting it manually: ${formatPgserveDaemonCommand(daemonCommand)}`)}await sleep(250)}throw Error(`pgserve v2 daemon did not bind ${socketPath} within ${Math.round(timeout/1000)}s. Try starting it manually: ${formatPgserveDaemonCommand(daemonCommand)}`)}function formatPgserveDaemonCommand(daemonCommand){return`${daemonCommand.display} daemon --data ${DATA_DIR} --log warn`}function maskCredentials(url){return url.replace(/\/\/.*@/,"//***@")}function checkRootGuard(){if(process.getuid?.()!==0)return null;if(process.env.GENIE_ALLOW_ROOT==="1")return null;return"pgserve cannot start under uid 0 (root) \u2014 PostgreSQL refuses to run as root for security reasons. "+"Run genie as a non-root user, or set GENIE_ALLOW_ROOT=1 to attempt anyway. See: https://github.com/automagik-dev/genie/issues/1226"}function isPgAutostartDisabled(){let value=process.env.GENIE_PG_NO_AUTOSTART;return value!==void 0&&TRUTHY_ENV.has(value.trim().toLowerCase())}function selfHealPostgres(dataDir){try{execSync3(`pkill -9 -f "postgres.*${dataDir.replace(/\//g,"\\/")}" 2>/dev/null || true`,{stdio:"ignore",timeout:5000}),execSync3(`pkill -9 -f "pgserve.*${dataDir.replace(/\//g,"\\/")}" 2>/dev/null || true`,{stdio:"ignore",timeout:5000})}catch{}let pidFile=join17(dataDir,"postmaster.pid");if(existsSync13(pidFile))try{unlinkSync3(pidFile)}catch{}try{execSync3("ipcs -m 2>/dev/null | awk '$6 == 0 {print $2}' | xargs -I{} ipcrm -m {} 2>/dev/null || true",{stdio:"ignore",timeout:5000})}catch{}}function signalPgserveTree(child,signal){let pid=child.pid;if(pid===void 0){try{child.kill(signal)}catch{}return}try{if(process.platform==="win32")child.kill(signal);else process.kill(-pid,signal)}catch{try{child.kill(signal)}catch{}}}async function terminatePgserveTree(child){if(signalPgserveTree(child,"SIGTERM"),!await new Promise((resolve3)=>{if(child.exitCode!==null||child.signalCode!==null){resolve3(!0);return}let timer2=setTimeout(()=>resolve3(!1),1000);timer2.unref(),child.once("exit",()=>{clearTimeout(timer2),resolve3(!0)})}))signalPgserveTree(child,"SIGKILL")}function getPort(){let envPort=process.env.GENIE_PG_PORT;if(envPort){let parsed=Number.parseInt(envPort,10);if(!Number.isNaN(parsed)&&parsed>0&&parsed<65536)return parsed}return DEFAULT_PORT}async function isPostgresHealthy(port){try{return await Promise.race([(async()=>{let pg=(await Promise.resolve().then(() => (init_src(),exports_src))).default,probe=pg({host:DEFAULT_HOST,port,database:DB_NAME,username:DB_NAME,password:resolveTcpPgPassword(),max:1,connect_timeout:3,idle_timeout:1});try{return await probe`SELECT 1`,await probe.end({timeout:2}),!0}catch{try{await probe.end({timeout:1})}catch{}return!1}})(),new Promise((resolve3)=>{setTimeout(()=>resolve3(!1),4000).unref()})])}catch{return!1}}function readLockfile(){try{let content=readFileSync9(LOCKFILE_PATH,"utf-8").trim(),port=Number.parseInt(content,10);if(!Number.isNaN(port)&&port>0&&port<65536)return port}catch{}return null}function writeLockfile(port){try{mkdirSync6(GENIE_HOME,{recursive:!0});let tmpPath=`${LOCKFILE_PATH}.tmp.${process.pid}`;writeFileSync6(tmpPath,String(port),"utf-8"),renameSync(tmpPath,LOCKFILE_PATH)}catch{}}function removeLockfile(){try{unlinkSync3(LOCKFILE_PATH)}catch{}}async function runRetention(sql){if(retentionRan)return;try{await sql.unsafe(`
|
|
700
700
|
DELETE FROM heartbeats WHERE created_at < now() - interval '7 days';
|
|
701
701
|
DELETE FROM machine_snapshots WHERE created_at < now() - interval '30 days';
|
|
702
702
|
DELETE FROM audit_events WHERE entity_type LIKE 'otel_%' AND created_at < now() - interval '30 days';
|
|
703
703
|
DELETE FROM genie_runtime_events WHERE created_at < now() - interval '14 days';
|
|
704
704
|
`),retentionRan=!0}catch(retErr){retentionRan=!0;let msg=retErr instanceof Error?retErr.message:String(retErr);process.stderr.write(`[genie] retention cleanup warning: ${msg}
|
|
705
|
-
`)}}async function ensurePgserve(){if(activePort===null){let testPort=await resolveTestPort();if(testPort!==null)return activePort=testPort,process.env.GENIE_PG_AVAILABLE="true",testPort}if(ensurePromise)return ensurePromise;ensurePromise=_ensurePgserve();try{return await ensurePromise}finally{ensurePromise=null}}function __setSpawnDaemonForTest(fn){spawnDaemon=fn??(()=>{let bunPath=process.execPath??"bun",genieBin=process.argv[1]??"genie";spawn(bunPath,[genieBin,"serve","start","--headless","--foreground"],{detached:!0,stdio:"ignore",env:{...process.env,GENIE_IS_DAEMON:"1"}}).unref()})}function readPidFile(pidPath){try{return readFileSync9(pidPath,"utf-8").trim()||null}catch{return null}}function parsePidFile(raw){let sepIdx=raw.indexOf(":"),pid,recordedStartTime;if(sepIdx<0)pid=Number.parseInt(raw,10),recordedStartTime=null;else{pid=Number.parseInt(raw.slice(0,sepIdx),10);let tail=raw.slice(sepIdx+1).trim();recordedStartTime=tail===""||tail==="unknown"?null:tail}if(Number.isNaN(pid)||pid<=0)return null;return{pid,recordedStartTime}}function isServeAlive(pid,recordedStartTime){try{process.kill(pid,0)}catch{return!1}if(recordedStartTime===null)return!1;let currentStartTime=getProcessStartTime(pid);return currentStartTime!==null&¤tStartTime===recordedStartTime}function unlinkQuiet(path2){try{unlinkSync3(path2)}catch{}}async function autoStartDaemon(){let home=process.env.GENIE_HOME??GENIE_HOME,pidPath=join17(home,"serve.pid"),raw=readPidFile(pidPath);if(!raw){lastAutoStartOutcome="missing",lastAutoStartPid=null,spawnDaemon();return}let parsed=parsePidFile(raw);if(!parsed){unlinkQuiet(pidPath),lastAutoStartOutcome="stale",lastAutoStartPid=null,spawnDaemon();return}if(isServeAlive(parsed.pid,parsed.recordedStartTime)){lastAutoStartOutcome="alive",lastAutoStartPid=parsed.pid;return}unlinkQuiet(pidPath),lastAutoStartOutcome="stale",lastAutoStartPid=parsed.pid,spawnDaemon()}async function resolveTestPort(){let raw=process.env.GENIE_TEST_PG_PORT;if(!raw)return null;let parsed=Number.parseInt(raw,10);if(Number.isNaN(parsed)||parsed<=0||parsed>=65536||!await isPostgresHealthy(parsed))throw Error(`GENIE_TEST_PG_PORT=${raw} set but not reachable`);return parsed}async function tryExistingPort(port){let portFromFile=readLockfile();if(portFromFile!==null&&await isPostgresHealthy(portFromFile))return activePort=portFromFile,process.env.GENIE_PG_AVAILABLE="true",portFromFile;if(await isPostgresHealthy(port))return activePort=port,process.env.GENIE_PG_AVAILABLE="true",writeLockfile(port),port;return null}async function spawnPgserveDirect(port){mkdirSync6(DATA_DIR,{recursive:!0}),selfHealPostgres(DATA_DIR);try{let startedPort=await startPgserveOnPort(port);return registerExitHandler(),startedPort}catch(err){process.env.GENIE_PG_AVAILABLE="false";let message=err instanceof Error?err.message:String(err);throw Error(`pgserve failed to start: ${maskCredentials(message)}`)}}async function waitForDaemonPort(){let deadline=Date.now()+16000;while(Date.now()<deadline){let p=readLockfile();if(p!==null&&await isPostgresHealthy(p))return activePort=p,process.env.GENIE_PG_AVAILABLE="true",p;await new Promise((r)=>setTimeout(r,500))}return null}function throwDaemonTimeout(outcomeAtStart,pidAtStart){process.env.GENIE_PG_AVAILABLE="false";let home=process.env.GENIE_HOME??GENIE_HOME,pidPath=join17(home,"serve.pid"),hasPidFile=existsSync13(pidPath),currentPort=readLockfile()??getPort();if(outcomeAtStart==="stale")throw Error(`Stale ~/.genie/serve.pid (PID ${pidAtStart??"unknown"} was not our serve). Removed and retried \u2014 if this persists, run: genie serve start`);if(!hasPidFile)throw Error("genie serve not running. Run: genie serve start");throw Error(`genie serve is running (PID ${pidAtStart??outcomeAtStart??"unknown"}) but pgserve did not respond on port ${currentPort} within 16s. Try: genie serve restart, or check ~/.genie/logs/scheduler.log`)}async function _ensurePgserve(){if(activePort!==null)return activePort;let port=getPort(),existing=await tryExistingPort(port);if(existing!==null)return existing;if(process.env.CI==="true")throw process.env.GENIE_PG_AVAILABLE="false",Error("pgserve not available in CI");if(isPgAutostartDisabled())throw process.env.GENIE_PG_AVAILABLE="false",Error("pgserve unavailable and GENIE_PG_NO_AUTOSTART=1");let rootErr=checkRootGuard();if(rootErr!==null)throw process.env.GENIE_PG_AVAILABLE="false",Error(rootErr);if(process.env.GENIE_IS_DAEMON==="1")return spawnPgserveDirect(port);await autoStartDaemon();let outcomeAtStart=lastAutoStartOutcome,pidAtStart=lastAutoStartPid,waited=await waitForDaemonPort();if(waited!==null)return waited;throwDaemonTimeout(outcomeAtStart,pidAtStart)}function findPgserveBin(){
|
|
705
|
+
`)}}async function ensurePgserve(){if(activePort===null){let testPort=await resolveTestPort();if(testPort!==null)return activePort=testPort,process.env.GENIE_PG_AVAILABLE="true",testPort}if(ensurePromise)return ensurePromise;ensurePromise=_ensurePgserve();try{return await ensurePromise}finally{ensurePromise=null}}function __setSpawnDaemonForTest(fn){spawnDaemon=fn??(()=>{let bunPath=process.execPath??"bun",genieBin=process.argv[1]??"genie";spawn(bunPath,[genieBin,"serve","start","--headless","--foreground"],{detached:!0,stdio:"ignore",env:{...process.env,GENIE_IS_DAEMON:"1"}}).unref()})}function readPidFile(pidPath){try{return readFileSync9(pidPath,"utf-8").trim()||null}catch{return null}}function parsePidFile(raw){let sepIdx=raw.indexOf(":"),pid,recordedStartTime;if(sepIdx<0)pid=Number.parseInt(raw,10),recordedStartTime=null;else{pid=Number.parseInt(raw.slice(0,sepIdx),10);let tail=raw.slice(sepIdx+1).trim();recordedStartTime=tail===""||tail==="unknown"?null:tail}if(Number.isNaN(pid)||pid<=0)return null;return{pid,recordedStartTime}}function isServeAlive(pid,recordedStartTime){try{process.kill(pid,0)}catch{return!1}if(recordedStartTime===null)return!1;let currentStartTime=getProcessStartTime(pid);return currentStartTime!==null&¤tStartTime===recordedStartTime}function unlinkQuiet(path2){try{unlinkSync3(path2)}catch{}}async function autoStartDaemon(){let home=process.env.GENIE_HOME??GENIE_HOME,pidPath=join17(home,"serve.pid"),raw=readPidFile(pidPath);if(!raw){lastAutoStartOutcome="missing",lastAutoStartPid=null,spawnDaemon();return}let parsed=parsePidFile(raw);if(!parsed){unlinkQuiet(pidPath),lastAutoStartOutcome="stale",lastAutoStartPid=null,spawnDaemon();return}if(isServeAlive(parsed.pid,parsed.recordedStartTime)){lastAutoStartOutcome="alive",lastAutoStartPid=parsed.pid;return}unlinkQuiet(pidPath),lastAutoStartOutcome="stale",lastAutoStartPid=parsed.pid,spawnDaemon()}async function resolveTestPort(){let raw=process.env.GENIE_TEST_PG_PORT;if(!raw)return null;let parsed=Number.parseInt(raw,10);if(Number.isNaN(parsed)||parsed<=0||parsed>=65536||!await isPostgresHealthy(parsed))throw Error(`GENIE_TEST_PG_PORT=${raw} set but not reachable`);return parsed}async function tryExistingPort(port){let portFromFile=readLockfile();if(portFromFile!==null&&await isPostgresHealthy(portFromFile))return activePort=portFromFile,process.env.GENIE_PG_AVAILABLE="true",portFromFile;if(await isPostgresHealthy(port))return activePort=port,process.env.GENIE_PG_AVAILABLE="true",writeLockfile(port),port;return null}async function spawnPgserveDirect(port){mkdirSync6(DATA_DIR,{recursive:!0}),selfHealPostgres(DATA_DIR);try{let startedPort=await startPgserveOnPort(port);return registerExitHandler(),startedPort}catch(err){process.env.GENIE_PG_AVAILABLE="false";let message=err instanceof Error?err.message:String(err);throw Error(`pgserve failed to start: ${maskCredentials(message)}`)}}async function waitForDaemonPort(){let deadline=Date.now()+16000;while(Date.now()<deadline){let p=readLockfile();if(p!==null&&await isPostgresHealthy(p))return activePort=p,process.env.GENIE_PG_AVAILABLE="true",p;await new Promise((r)=>setTimeout(r,500))}return null}function throwDaemonTimeout(outcomeAtStart,pidAtStart){process.env.GENIE_PG_AVAILABLE="false";let home=process.env.GENIE_HOME??GENIE_HOME,pidPath=join17(home,"serve.pid"),hasPidFile=existsSync13(pidPath),currentPort=readLockfile()??getPort();if(outcomeAtStart==="stale")throw Error(`Stale ~/.genie/serve.pid (PID ${pidAtStart??"unknown"} was not our serve). Removed and retried \u2014 if this persists, run: genie serve start`);if(!hasPidFile)throw Error("genie serve not running. Run: genie serve start");throw Error(`genie serve is running (PID ${pidAtStart??outcomeAtStart??"unknown"}) but pgserve did not respond on port ${currentPort} within 16s. Try: genie serve restart, or check ~/.genie/logs/scheduler.log`)}async function _ensurePgserve(){if(activePort!==null)return activePort;let port=getPort(),existing=await tryExistingPort(port);if(existing!==null)return existing;if(process.env.CI==="true")throw process.env.GENIE_PG_AVAILABLE="false",Error("pgserve not available in CI");if(isPgAutostartDisabled())throw process.env.GENIE_PG_AVAILABLE="false",Error("pgserve unavailable and GENIE_PG_NO_AUTOSTART=1");let rootErr=checkRootGuard();if(rootErr!==null)throw process.env.GENIE_PG_AVAILABLE="false",Error(rootErr);if(process.env.GENIE_IS_DAEMON==="1")return spawnPgserveDirect(port);await autoStartDaemon();let outcomeAtStart=lastAutoStartOutcome,pidAtStart=lastAutoStartPid,waited=await waitForDaemonPort();if(waited!==null)return waited;throwDaemonTimeout(outcomeAtStart,pidAtStart)}function findPgserveBin(){let globalBin=join17(homedir11(),".bun","bin","pgserve");if(existsSync13(globalBin))return globalBin;try{return execSync3("which pgserve",{encoding:"utf-8",timeout:3000}).trim()}catch{return"pgserve"}}async function startPgserveOnPort(port){mkdirSync6(DATA_DIR,{recursive:!0});let child=spawn(findPgserveBin(),["--port",String(port),"--host",DEFAULT_HOST,"--data",DATA_DIR,"--log","warn","--no-stats","--no-cluster","--pgvector","--max-connections",process.env.GENIE_PG_MAX_CONNECTIONS??"10000"],{detached:!0,stdio:"ignore"});child.unref(),pgserveChild=child;let timeout=Number(process.env.GENIE_PGSERVE_TIMEOUT)||30000,deadline=Date.now()+timeout;while(Date.now()<deadline){if(await isPostgresHealthy(port))return activePort=port,ownsLockfile=!0,process.env.GENIE_PG_AVAILABLE="true",writeLockfile(port),port;await new Promise((r)=>setTimeout(r,500))}if(await terminatePgserveTree(child),pgserveChild===child)pgserveChild=null;throw selfHealPostgres(DATA_DIR),Error(`pgserve failed to start on port ${port} (timeout after ${timeout/1000}s)`)}function registerExitHandler(){if(exitHandlerRegistered)return;exitHandlerRegistered=!0;let cleanup=()=>{if(sqlClient){let dying=sqlClient;sqlClient=null,dying.end({timeout:1}).catch(()=>{})}if(pgserveChild)signalPgserveTree(pgserveChild,"SIGTERM"),pgserveChild=null;if(ownsLockfile)removeLockfile(),ownsLockfile=!1};process.on("exit",cleanup),process.on("beforeExit",()=>{shutdown().catch(()=>{})}),process.on("SIGINT",()=>{cleanup(),process.exit(130)}),process.on("SIGTERM",()=>{cleanup(),process.exit(143)})}async function healthCheckCachedClient(){if(!sqlClient)return null;try{return await sqlClient`SELECT 1`,sqlClient}catch{let dying=sqlClient;if(!dying)return null;return sqlClient=null,activePort=null,dying.end({timeout:5}).catch(()=>{}),null}}async function runPostConnectSetup(client,isTestMode,timings){let _t2=Date.now(),skipBoot=isTestMode||process.env.GENIE_SKIP_DB_BOOT==="1";if(!skipBoot)await runMigrations(client);let _t3=Date.now();if(!skipBoot&&(needsSeed()||await needsSeededTeams(client)))await runSeed(client);if(!skipBoot)await maybePromptV1Migration(client);let _t4=Date.now(),_t5=_t4;if(process.env.GENIE_PROFILE_DB)console.error(`[db-profile] pgserve=${timings.t1-timings.t0}ms migrate=${_t3-_t2}ms seed=${_t4-_t3}ms retention=skipped total=${_t5-timings.t0}ms`)}async function getConnection(){let cached=await healthCheckCachedClient();if(cached)return cached;if(buildPromise)return buildPromise;buildPromise=_buildConnection();try{return await buildPromise}finally{buildPromise=null}}function shouldUseUnixSocket(){if(process.env.GENIE_PG_FORCE_TCP==="1")return!1;if(process.env.GENIE_TEST_PG_PORT)return!1;return!0}async function maybePrintBanner(client,isTestMode){if(bannerPrinted||isTestMode)return;if(process.env.GENIE_QUIET==="1"||process.env.GENIE_NO_BANNER==="1"){bannerPrinted=!0;return}try{let db=(await client.unsafe("SELECT current_database() AS db"))[0]?.db;if(typeof db==="string"&&db.length>0)process.stderr.write(`[pgserve] connected to ${db}
|
|
706
706
|
`)}catch{}bannerPrinted=!0}async function _buildConnection(){let _t0=Date.now(),useSocket=shouldUseUnixSocket();if(useSocket)await getOrStartDaemon();let discovery=useSocket?readPostmasterDiscovery():null,port=useSocket?discovery?.port??5432:await ensurePgserve(),_t1=Date.now(),pgModule=(await Promise.resolve().then(() => (init_src(),exports_src))).default,database=resolveDatabaseName(),isTestMode=Boolean(process.env.GENIE_TEST_DB_NAME),pgWireCredential=useSocket?resolvePgserveAuthPassword():resolveTcpPgPassword(),host=useSocket?discovery?.socketDir??resolvePgserveSocketDir():DEFAULT_HOST,cliShortLived=!daemonCwdPinned&&!isTestMode&&process.env.GENIE_SKIP_DB_BOOT==="1",originalCwd=process.cwd(),pkgDir=resolveGeniePackageDir(),shouldRestoreCwd=!daemonCwdPinned&&!isTestMode,pinned=!1;if(pkgDir&&process.cwd()!==pkgDir)try{process.chdir(pkgDir),pinned=!0}catch(err){let msg=err instanceof Error?err.message:String(err);process.stderr.write(`[pgserve] WARN: failed to pin cwd to ${pkgDir}: ${msg}
|
|
707
707
|
`)}else if(!pkgDir)process.stderr.write(`[pgserve] WARN: could not resolve genie package dir; pgserve fingerprint may be unstable
|
|
708
708
|
`);let client=pgModule({host,port,database,username:DB_NAME,[PG_AUTH_FIELD]:pgWireCredential,max:cliShortLived?1:50,idle_timeout:cliShortLived?0:1,connect_timeout:resolvePgConnectTimeoutSeconds(useSocket),onnotice:()=>{},connection:{client_min_messages:"warning"}});sqlClient=client;try{if(await client`SELECT 1`,await runPostConnectSetup(client,isTestMode,{t0:_t0,t1:_t1}),await maybePrintBanner(client,isTestMode),useSocket)activePort=SOCKET_PORT_SENTINEL,process.env.GENIE_PG_AVAILABLE="true"}catch(err){if(sqlClient===client)sqlClient=null;throw activePort=null,client.end({timeout:2}).catch(()=>{}),err}finally{if(pinned&&shouldRestoreCwd&&process.cwd()!==originalCwd)try{process.chdir(originalCwd)}catch(err){let msg=err instanceof Error?err.message:String(err);process.stderr.write(`[pgserve] WARN: failed to restore cwd to ${originalCwd}: ${msg}
|
|
@@ -2132,7 +2132,7 @@ ${bodyHash}`}function signOmniRequest(method,path3,body){let paths=keyPaths(),ho
|
|
|
2132
2132
|
[omni-signature] Run \`genie omni handshake\` to register this host and enable signed requests.
|
|
2133
2133
|
`)}return null}let timestamp2=new Date().toISOString(),canonical=canonicalSigningInput(timestamp2,method,path3,body),signature=sign(null,Buffer.from(canonical,"utf-8"),key).toString("base64url");return{"X-Genie-Host-Id":host.hostId,"X-Genie-Timestamp":timestamp2,"X-Genie-Signature":signature}}var warnedMissingKey=!1,cachedKey=null,cachedKeyPath=null;var init_omni_signature=()=>{};var exports_frontmatter_writer={};__export(exports_frontmatter_writer,{writeFrontmatter:()=>writeFrontmatter,serializeSdkConfig:()=>serializeSdkConfig});import{readFileSync as readFileSync27,writeFileSync as writeFileSync16}from"fs";function writeFrontmatter(filePath,updates){let content=readFileSync27(filePath,"utf-8"),{yamlObj,body}=splitFrontmatter(content),merged={...yamlObj,...stripUndefined(updates)},output=`---
|
|
2134
2134
|
${dump(merged,{lineWidth:-1,noRefs:!0,sortKeys:!1,quotingType:'"'})}---
|
|
2135
|
-
${body}`;writeFileSync16(filePath,output,"utf-8")}function serializeSdkConfig(sdk){let result2={};for(let[key,value]of Object.entries(sdk)){if(value===void 0||value===null)continue;if(Array.isArray(value)&&value.length===0)continue;if(typeof value==="object"&&!Array.isArray(value)&&Object.keys(value).length===0)continue;result2[key]=value}return result2}function splitFrontmatter(content){let match=content.match(/^---\n([\s\S]*?)\n---\n?([\s\S]*)$/);if(!match)return{yamlObj:{},body:content};let yamlStr=match[1],body=match[2],yamlObj={};try{let parsed=load(yamlStr);if(typeof parsed==="object"&&parsed!==null)yamlObj=parsed}catch{}return{yamlObj,body}}function stripUndefined(obj){let result2={};for(let[key,value]of Object.entries(obj))if(value!==void 0)result2[key]=value;return result2}var init_frontmatter_writer=__esm(()=>{init_js_yaml()});import{existsSync as existsSync43,realpathSync as realpathSync5}from"fs";import{homedir as homedir34}from"os";import{join as join52,resolve as resolve7}from"path";function getCwd(deps){return deps.cwd??process.cwd()}function getHomeDir(deps){return deps.homeDir??homedir34()}function getWorkspaceRoot(deps){if("workspaceRoot"in deps)return deps.workspaceRoot??null;return findWorkspace()?.root??null}function getConfig(deps){return deps.config??loadGenieConfigSync()}function expandHome(path3,homeDir){if(path3==="~")return homeDir;if(path3.startsWith("~/"))return join52(homeDir,path3.slice(2));return path3}function normalizeBrainVaultPath(path3,deps){return resolve7(getCwd(deps),expandHome(path3,getHomeDir(deps)))}function canonicalBrainVaultPath(path3,deps){let realpath=deps.realpath??realpathSync5;try{return realpath(path3)}catch{return resolve7(path3)}}function dedupeBrainVaultPaths(paths,deps={}){let seen=new Set,deduped=[];for(let path3 of paths){let canonical=canonicalBrainVaultPath(path3,deps);if(seen.has(canonical))continue;seen.add(canonical),deduped.push(canonical)}return deduped}function hasBrainJson(path3,deps){return(deps.exists??existsSync43)(join52(path3,"brain.json"))}function filterVaultsWithBrainJson(paths,source,deps){let warn=deps.warn??console.warn,label=source==="config"?"configured":source==="registry"?"registered":"legacy",valid=[];for(let path3 of paths){if(hasBrainJson(path3,deps)){valid.push(path3);continue}warn(` Brain server: skipped ${label} vault ${path3} (missing brain.json)`)}return valid}function normalizeAndDedupe(paths,deps){return dedupeBrainVaultPaths(paths.map((path3)=>normalizeBrainVaultPath(path3,deps)),deps)}function pushPathFromEntry(entry2,paths){let found=!1;for(let field of REGISTRY_PATH_FIELDS){let value=entry2[field];if(typeof value==="string"&&value.trim().length>0)paths.push(value),found=!0}return found}function isRegistryRecord(value){return Boolean(value)&&typeof value==="object"&&!Array.isArray(value)}function collectRegistryArray(values2,paths){for(let item of values2)collectRegistryPaths(item,paths)}function collectRegistryObject(entry2,paths){let foundPath=pushPathFromEntry(entry2,paths);for(let field of REGISTRY_COLLECTION_FIELDS)if(field in entry2)collectRegistryPaths(entry2[field],paths);if(!foundPath){for(let item of Object.values(entry2))if(item&&typeof item==="object")collectRegistryPaths(item,paths)}}function collectRegistryPaths(value,paths){if(typeof value==="string"){if(value.trim().length>0)paths.push(value);return}if(Array.isArray(value)){collectRegistryArray(value,paths);return}if(isRegistryRecord(value))collectRegistryObject(value,paths)}function countRegistryItems(value,fallback){if(Array.isArray(value))return value.length;if(!isRegistryRecord(value))return fallback;for(let field of REGISTRY_COLLECTION_FIELDS){let collection=value[field];if(Array.isArray(collection))return collection.length;if(isRegistryRecord(collection))return Object.keys(collection).length}return fallback}async function discoverRegisteredBrainVaultPaths(deps){let brain=deps.brain;if(!brain)return{paths:[],registryCount:0};let registryCalls=["listBrains","listBrainVaults","listVaults","listRegisteredBrains","listRegisteredBrainVaults","getBrainRegistry","readBrainRegistry"];for(let name of registryCalls){let read=brain[name];if(typeof read!=="function")continue;try{let result2=await read.call(brain),paths=[];if(collectRegistryPaths(result2,paths),paths.length>0)return{paths,registryCount:countRegistryItems(result2,paths.length)}}catch(err){let msg=err instanceof Error?err.message:String(err);(deps.warn??console.warn)(` Brain registry: ${name} failed: ${msg}`)}}return{paths:[],registryCount:0}}async function findLegacyBrainVault(deps={}){let cwd=getCwd(deps),homeDir=getHomeDir(deps),workspaceRoot=getWorkspaceRoot(deps),candidates=[workspaceRoot?join52(workspaceRoot,"brain"):void 0,cwd,join52(cwd,"brain"),join52(homeDir,"brain")].filter((path3)=>typeof path3==="string");for(let path3 of normalizeAndDedupe(candidates,deps))if(hasBrainJson(path3,deps))return path3;return null}async function findBrainVault(deps={}){return(await resolveBrainVaults(deps)).paths[0]??null}async function resolveBrainVaults(deps={}){let configuredPaths=getConfig(deps).brain?.paths;if(Array.isArray(configuredPaths)&&configuredPaths.length>0)return{source:"config",paths:filterVaultsWithBrainJson(normalizeAndDedupe(configuredPaths,deps),"config",deps)};let registered=await discoverRegisteredBrainVaultPaths(deps);if(registered.paths.length>0)return{source:"registry",paths:filterVaultsWithBrainJson(normalizeAndDedupe(registered.paths,deps),"registry",deps),registryCount:registered.registryCount};let legacyPath=await findLegacyBrainVault(deps);return{source:"legacy",paths:legacyPath?[legacyPath]:[]}}function normalizeStartupConcurrency(value){if(typeof value!=="number"||!Number.isFinite(value))return DEFAULT_BRAIN_START_CONCURRENCY;return Math.max(1,Math.floor(value))}async function allSettledBounded(items,concurrency,worker){let results=Array(items.length),nextIndex=0;async function runWorker(){while(!0){let index=nextIndex++;if(index>=items.length)return;try{results[index]={status:"fulfilled",value:await worker(items[index],index)}}catch(reason){results[index]={status:"rejected",reason}}}}return await Promise.allSettled(Array.from({length:Math.min(concurrency,items.length)},()=>runWorker())),results}function warnRegistryDrift(resolution,startedCount,resolvedCount,deps){if(resolution.source!=="registry")return;let expectedCount=resolution.registryCount??resolvedCount;if(startedCount===expectedCount)return;(deps.warn??console.warn)(` Brain server: registry drift: started ${startedCount}/${expectedCount} registered vault(s) (${resolvedCount} resolved valid path(s))`)}async function startResolvedBrainVaults(resolution,brain,geniePgPort,deps={}){let startEmbeddedBrainServer=brain.startEmbeddedBrainServer;if(!startEmbeddedBrainServer)return[];let warn=deps.warn??console.warn,log2=deps.log??console.log,handles=[],paths=dedupeBrainVaultPaths(resolution.paths,deps),concurrency=normalizeStartupConcurrency(deps.startupConcurrency),results=await allSettledBounded(paths,concurrency,async(brainPath)=>{let handle=await startEmbeddedBrainServer.call(brain,{brainPath,geniePgPort});return log2(` Brain server ready on port ${handle.port} (${brainPath})`),{brainPath,port:handle.port,stop:handle.stop}});for(let index=0;index<results.length;index++){let result2=results[index];if(result2.status==="fulfilled")handles.push(result2.value);else{let msg=result2.reason instanceof Error?result2.reason.message:String(result2.reason);warn(` Brain server: failed for ${paths[index]}: ${msg}`)}}return warnRegistryDrift(resolution,handles.length,paths.length,deps),handles}var REGISTRY_PATH_FIELDS,REGISTRY_COLLECTION_FIELDS,DEFAULT_BRAIN_START_CONCURRENCY=4;var init_brain_vaults=__esm(()=>{init_genie_config2();init_workspace();REGISTRY_PATH_FIELDS=["homePath","brainPath","vaultPath","path","root","dir"],REGISTRY_COLLECTION_FIELDS=["paths","brains","vaults","entries","items","registered"]});var exports_tui_disable={};__export(exports_tui_disable,{noticeTuiSkipped:()=>noticeTuiSkipped,isTuiDisabled:()=>isTuiDisabled});function isTuiDisabled(){let envVal=process.env.GENIE_TUI_DISABLE;if(envVal&&TRUTHY2.has(envVal.trim().toLowerCase()))return!0;if(process.argv.includes("--no-tui"))return!0;return!1}function noticeTuiSkipped(context){let reason=process.env.GENIE_TUI_DISABLE?"GENIE_TUI_DISABLE is set":"--no-tui flag present";console.error(`genie: TUI ${context} skipped (${reason}). See https://github.com/automagik-dev/genie for status of the upstream OpenTUI kqueue spin on macOS.`)}var TRUTHY2;var init_tui_disable=__esm(()=>{TRUTHY2=new Set(["1","true","yes","on"])});var exports_service_registry={};__export(exports_service_registry,{unregisterService:()=>unregisterService,registerService:()=>registerService,reapDeadServices:()=>reapDeadServices,killAllServices:()=>killAllServices,getRegisteredServices:()=>getRegisteredServices,clearRegistry:()=>clearRegistry});function registerService(name,pid){registry.set(name,{pid,name,startedAt:new Date})}function unregisterService(name){registry.delete(name)}function getRegisteredServices(){return Array.from(registry.values())}function reapDeadServices(){let reaped=[];for(let[name,entry2]of registry)try{process.kill(entry2.pid,0)}catch{registry.delete(name),reaped.push(name)}return reaped}function killAllServices(){for(let[_name,entry2]of registry)try{process.kill(entry2.pid,"SIGTERM")}catch{registry.delete(_name)}let deadline=Date.now()+3000,checkInterval=200,stillAlive=()=>{for(let[,entry2]of registry)try{return process.kill(entry2.pid,0),!0}catch{}return!1};while(Date.now()<deadline&&stillAlive()){let waitUntil=Date.now()+checkInterval;while(Date.now()<waitUntil);}for(let[name,entry2]of registry){try{process.kill(entry2.pid,0),process.kill(entry2.pid,"SIGKILL")}catch{}registry.delete(name)}}function clearRegistry(){registry.clear()}var registry;var init_service_registry=__esm(()=>{registry=new Map});function parseDuration(input){let match=input.trim().match(DURATION_RE);if(!match)throw Error(`Invalid duration: "${input}". Expected format: 10m, 2h, 24h, 1d`);let value=Number.parseFloat(match[1]),unit=match[2].toLowerCase(),ms=value*{s:1000,sec:1000,m:60000,min:60000,h:3600000,hr:3600000,d:86400000,day:86400000}[unit];if(ms<=0)throw Error(`Duration must be positive: "${input}"`);return ms}function expandRange(range,step,min,max){if(step===0)throw Error("Cron step value cannot be 0");if(range==="*"){let out=[];for(let i2=min;i2<=max;i2+=step)out.push(i2);return out}if(range.includes("-")){let[start,end]=range.split("-").map(Number),out=[];for(let i2=start;i2<=end;i2+=step)out.push(i2);return out}return[Number.parseInt(range,10)]}function parseCronField(field,min,max){let values2=new Set;for(let part of field.split(",")){let stepMatch=part.match(/^(.+)\/(\d+)$/),step=stepMatch?Number.parseInt(stepMatch[2],10):1,range=stepMatch?stepMatch[1]:part;for(let v of expandRange(range,step,min,max))values2.add(v)}return[...values2].sort((a,b2)=>a-b2)}function getTimeParts(date,tz){if(!tz)return{month:date.getMonth()+1,dom:date.getDate(),dow:date.getDay(),hour:date.getHours(),minute:date.getMinutes()};let parts=new Intl.DateTimeFormat("en-US",{timeZone:tz,year:"numeric",month:"numeric",day:"numeric",hour:"numeric",minute:"numeric",weekday:"short",hour12:!1}).formatToParts(date),get3=(type2)=>Number(parts.find((p)=>p.type===type2)?.value??0),dayMap={Sun:0,Mon:1,Tue:2,Wed:3,Thu:4,Fri:5,Sat:6},weekday=parts.find((p)=>p.type==="weekday")?.value??"Sun";return{month:get3("month"),dom:get3("day"),dow:dayMap[weekday]??0,hour:get3("hour")===24?0:get3("hour"),minute:get3("minute")}}function parseOpts(afterOrOpts){if(afterOrOpts instanceof Date)return{after:afterOrOpts};if(afterOrOpts)return{after:afterOrOpts.after,timezone:afterOrOpts.timezone};return{}}function advanceToNextDay(candidate,tz){candidate.setTime(candidate.getTime()+86400000);let tp=getTimeParts(candidate,tz);candidate.setTime(candidate.getTime()-tp.hour*3600000-tp.minute*60000)}function parseCronExpr(cronExpr){let parts=cronExpr.trim().split(/\s+/);if(parts.length<5)throw Error(`Invalid cron expression: "${cronExpr}"`);let[minField,hourField,domField,monthField,dowField]=parts;return{minutes:parseCronField(minField,0,59),hours:parseCronField(hourField,0,23),doms:parseCronField(domField,1,31),months:parseCronField(monthField,1,12),dows:parseCronField(dowField,0,6),domRestricted:domField!=="*",dowRestricted:dowField!=="*"}}function computeNextCronDue(cronExpr,afterOrOpts){let{after,timezone}=parseOpts(afterOrOpts),cron=parseCronExpr(cronExpr),candidate=new Date((after??new Date).getTime());candidate.setSeconds(0,0),candidate.setTime(candidate.getTime()+60000);let limit=new Date(candidate.getTime()+31622400000);while(candidate<=limit){let tp=getTimeParts(candidate,timezone);if(!cron.months.includes(tp.month)){advanceToNextDay(candidate,timezone);continue}if(!(cron.domRestricted&&cron.dowRestricted?cron.doms.includes(tp.dom)||cron.dows.includes(tp.dow):cron.doms.includes(tp.dom)&&cron.dows.includes(tp.dow))){advanceToNextDay(candidate,timezone);continue}if(!cron.hours.includes(tp.hour)){candidate.setTime(candidate.getTime()+3600000-tp.minute*60000);continue}if(cron.minutes.includes(tp.minute))return candidate;candidate.setTime(candidate.getTime()+60000)}throw Error(`No next cron occurrence found for "${cronExpr}" within 366 days`)}var DURATION_RE;var init_cron=__esm(()=>{DURATION_RE=/^(\d+(?:\.\d+)?)\s*(s|sec|m|min|h|hr|d|day)s?$/i});import{randomUUID as randomUUID7}from"crypto";function parseNotifyPayload(channel,raw){switch(channel){case"genie_task_stage":{let parts=raw.split(":");if(parts.length<3)return null;return{channel,eventType:"task.stage_change",payload:{taskId:parts[0],fromStage:parts[1],toStage:parts[2]},taskId:parts[0],summary:`Task ${parts[0]} moved from ${parts[1]} to ${parts[2]}`}}case"genie_executor_state":{let parts=raw.split(":");if(parts.length<4)return null;let eventType=parts[3]==="error"?"executor.error":"executor.state_change";return{channel,eventType,payload:{executorId:parts[0],agentId:parts[1],oldState:parts[2],newState:parts[3]},agentId:parts[1],summary:`${parts[1]} state: ${parts[2]} \u2192 ${parts[3]}`}}case"genie_message":{let parts=raw.split(":");if(parts.length<2)return null;return{channel,eventType:"task.comment",payload:{messageId:parts[0],conversationId:parts[1]},summary:`New message in conversation ${parts[1]}`}}case"genie_audit_event":{let parts=raw.split(":");if(parts.length<3)return null;return{channel,eventType:`${parts[0]}.${parts[2]}`,payload:{entityType:parts[0],entityId:parts[1],auditEventType:parts[2]},summary:`${parts[0]} ${parts[1]}: ${parts[2]}`}}default:return null}}function isActionableEvent(eventType){return ACTIONABLE_EVENTS.has(eventType)||eventType.startsWith("request.")}async function resolveTargetTeams(event){if(!isActionableEvent(event.eventType))return[];let active=(await listTeams()).filter((t)=>t.status==="in_progress");if(event.agentId){let aid=event.agentId;return active.filter((t)=>t.members.includes(aid)||t.leader===aid).map((t)=>t.name)}return active.map((t)=>t.name)}async function writeMailbox(repoPath,leader,message,traceId){try{await(await getConnection())`
|
|
2135
|
+
${body}`;writeFileSync16(filePath,output,"utf-8")}function serializeSdkConfig(sdk){let result2={};for(let[key,value]of Object.entries(sdk)){if(value===void 0||value===null)continue;if(Array.isArray(value)&&value.length===0)continue;if(typeof value==="object"&&!Array.isArray(value)&&Object.keys(value).length===0)continue;result2[key]=value}return result2}function splitFrontmatter(content){let match=content.match(/^---\n([\s\S]*?)\n---\n?([\s\S]*)$/);if(!match)return{yamlObj:{},body:content};let yamlStr=match[1],body=match[2],yamlObj={};try{let parsed=load(yamlStr);if(typeof parsed==="object"&&parsed!==null)yamlObj=parsed}catch{}return{yamlObj,body}}function stripUndefined(obj){let result2={};for(let[key,value]of Object.entries(obj))if(value!==void 0)result2[key]=value;return result2}var init_frontmatter_writer=__esm(()=>{init_js_yaml()});import{existsSync as existsSync43,realpathSync as realpathSync5}from"fs";import{homedir as homedir34}from"os";import{join as join52,resolve as resolve7}from"path";function getCwd(deps){return deps.cwd??process.cwd()}function getHomeDir(deps){return deps.homeDir??homedir34()}function getWorkspaceRoot(deps){if("workspaceRoot"in deps)return deps.workspaceRoot??null;return findWorkspace()?.root??null}function getConfig(deps){return deps.config??loadGenieConfigSync()}function expandHome(path3,homeDir){if(path3==="~")return homeDir;if(path3.startsWith("~/"))return join52(homeDir,path3.slice(2));return path3}function normalizeBrainVaultPath(path3,deps){return resolve7(getCwd(deps),expandHome(path3,getHomeDir(deps)))}function canonicalBrainVaultPath(path3,deps){let realpath=deps.realpath??realpathSync5;try{return realpath(path3)}catch{return resolve7(path3)}}function dedupeBrainVaultPaths(paths,deps={}){let seen=new Set,deduped=[];for(let path3 of paths){let canonical=canonicalBrainVaultPath(path3,deps);if(seen.has(canonical))continue;seen.add(canonical),deduped.push(canonical)}return deduped}function hasBrainJson(path3,deps){return(deps.exists??existsSync43)(join52(path3,"brain.json"))}function filterVaultsWithBrainJson(paths,source,deps){let warn=deps.warn??console.warn,label=source==="config"?"configured":source==="registry"?"registered":"legacy",valid=[];for(let path3 of paths){if(hasBrainJson(path3,deps)){valid.push(path3);continue}warn(` Brain server: skipped ${label} vault ${path3} (missing brain.json)`)}return valid}function normalizeAndDedupe(paths,deps){return dedupeBrainVaultPaths(paths.map((path3)=>normalizeBrainVaultPath(path3,deps)),deps)}function pushPathFromEntry(entry2,paths){let found=!1;for(let field of REGISTRY_PATH_FIELDS){let value=entry2[field];if(typeof value==="string"&&value.trim().length>0)paths.push(value),found=!0}return found}function isRegistryRecord(value){return Boolean(value)&&typeof value==="object"&&!Array.isArray(value)}function collectRegistryArray(values2,paths){for(let item of values2)collectRegistryPaths(item,paths)}function collectRegistryObject(entry2,paths){let foundPath=pushPathFromEntry(entry2,paths);for(let field of REGISTRY_COLLECTION_FIELDS)if(field in entry2)collectRegistryPaths(entry2[field],paths);if(!foundPath){for(let item of Object.values(entry2))if(item&&typeof item==="object")collectRegistryPaths(item,paths)}}function collectRegistryPaths(value,paths){if(typeof value==="string"){if(value.trim().length>0)paths.push(value);return}if(Array.isArray(value)){collectRegistryArray(value,paths);return}if(isRegistryRecord(value))collectRegistryObject(value,paths)}function countRegistryItems(value,fallback){if(Array.isArray(value))return value.length;if(!isRegistryRecord(value))return fallback;for(let field of REGISTRY_COLLECTION_FIELDS){let collection=value[field];if(Array.isArray(collection))return collection.length;if(isRegistryRecord(collection))return Object.keys(collection).length}return fallback}async function discoverRegisteredBrainVaultPaths(deps){let brain=deps.brain;if(!brain)return{paths:[],registryCount:0};let registryCalls=["listBrains","listBrainVaults","listVaults","listRegisteredBrains","listRegisteredBrainVaults","getBrainRegistry","readBrainRegistry"];for(let name of registryCalls){let read=brain[name];if(typeof read!=="function")continue;try{let result2=await read.call(brain),paths=[];if(collectRegistryPaths(result2,paths),paths.length>0)return{paths,registryCount:countRegistryItems(result2,paths.length)}}catch(err){let msg=err instanceof Error?err.message:String(err);(deps.warn??console.warn)(` Brain registry: ${name} failed: ${msg}`)}}return{paths:[],registryCount:0}}async function findLegacyBrainVault(deps={}){let cwd=getCwd(deps),homeDir=getHomeDir(deps),workspaceRoot=getWorkspaceRoot(deps),candidates=[workspaceRoot?join52(workspaceRoot,"brain"):void 0,cwd,join52(cwd,"brain"),join52(homeDir,"brain")].filter((path3)=>typeof path3==="string");for(let path3 of normalizeAndDedupe(candidates,deps))if(hasBrainJson(path3,deps))return path3;return null}async function findBrainVault(deps={}){return(await resolveBrainVaults(deps)).paths[0]??null}async function resolveBrainVaults(deps={}){let configuredPaths=getConfig(deps).brain?.paths;if(Array.isArray(configuredPaths)&&configuredPaths.length>0)return{source:"config",paths:filterVaultsWithBrainJson(normalizeAndDedupe(configuredPaths,deps),"config",deps)};let registered=await discoverRegisteredBrainVaultPaths(deps);if(registered.paths.length>0)return{source:"registry",paths:filterVaultsWithBrainJson(normalizeAndDedupe(registered.paths,deps),"registry",deps),registryCount:registered.registryCount};let legacyPath=await findLegacyBrainVault(deps);return{source:"legacy",paths:legacyPath?[legacyPath]:[]}}function normalizeStartupConcurrency(value){if(typeof value!=="number"||!Number.isFinite(value))return DEFAULT_BRAIN_START_CONCURRENCY;return Math.max(1,Math.floor(value))}async function allSettledBounded(items,concurrency,worker){let results=Array(items.length),nextIndex=0;async function runWorker(){while(!0){let index=nextIndex++;if(index>=items.length)return;try{results[index]={status:"fulfilled",value:await worker(items[index],index)}}catch(reason){results[index]={status:"rejected",reason}}}}return await Promise.allSettled(Array.from({length:Math.min(concurrency,items.length)},()=>runWorker())),results}function warnRegistryDrift(resolution,startedCount,resolvedCount,deps){if(resolution.source!=="registry")return;let expectedCount=resolution.registryCount??resolvedCount;if(startedCount===expectedCount)return;(deps.warn??console.warn)(` Brain server: registry drift: started ${startedCount}/${expectedCount} registered vault(s) (${resolvedCount} resolved valid path(s))`)}async function startResolvedBrainVaults(resolution,brain,geniePgPort,deps={}){let startEmbeddedBrainServer=brain.startEmbeddedBrainServer;if(!startEmbeddedBrainServer)return[];let warn=deps.warn??console.warn,log2=deps.log??console.log,handles=[],paths=dedupeBrainVaultPaths(resolution.paths,deps),concurrency=normalizeStartupConcurrency(deps.startupConcurrency),results=await allSettledBounded(paths,concurrency,async(brainPath)=>{let handle=await startEmbeddedBrainServer.call(brain,{brainPath,geniePgPort});return log2(` Brain server ready on port ${handle.port} (${brainPath})`),{brainPath,port:handle.port,stop:handle.stop}});for(let index=0;index<results.length;index++){let result2=results[index];if(result2.status==="fulfilled")handles.push(result2.value);else{let msg=result2.reason instanceof Error?result2.reason.message:String(result2.reason);warn(` Brain server: failed for ${paths[index]}: ${msg}`)}}return warnRegistryDrift(resolution,handles.length,paths.length,deps),handles}var REGISTRY_PATH_FIELDS,REGISTRY_COLLECTION_FIELDS,DEFAULT_BRAIN_START_CONCURRENCY=4;var init_brain_vaults=__esm(()=>{init_genie_config2();init_workspace();REGISTRY_PATH_FIELDS=["homePath","brainPath","vaultPath","path","root","dir"],REGISTRY_COLLECTION_FIELDS=["paths","brains","vaults","entries","items","registered"]});var exports_tui_disable={};__export(exports_tui_disable,{noticeTuiSkipped:()=>noticeTuiSkipped,isTuiDisabled:()=>isTuiDisabled});function isTuiDisabled(){let envVal=process.env.GENIE_TUI_DISABLE;if(envVal&&TRUTHY2.has(envVal.trim().toLowerCase()))return!0;if(process.argv.includes("--no-tui"))return!0;if(process.stdout.isTTY===!1)return!0;return!1}function noticeTuiSkipped(context){let reason=process.env.GENIE_TUI_DISABLE?"GENIE_TUI_DISABLE is set":"--no-tui flag present";console.error(`genie: TUI ${context} skipped (${reason}). See https://github.com/automagik-dev/genie for status of the upstream OpenTUI kqueue spin on macOS.`)}var TRUTHY2;var init_tui_disable=__esm(()=>{TRUTHY2=new Set(["1","true","yes","on"])});var exports_service_registry={};__export(exports_service_registry,{unregisterService:()=>unregisterService,registerService:()=>registerService,reapDeadServices:()=>reapDeadServices,killAllServices:()=>killAllServices,getRegisteredServices:()=>getRegisteredServices,clearRegistry:()=>clearRegistry});function registerService(name,pid){registry.set(name,{pid,name,startedAt:new Date})}function unregisterService(name){registry.delete(name)}function getRegisteredServices(){return Array.from(registry.values())}function reapDeadServices(){let reaped=[];for(let[name,entry2]of registry)try{process.kill(entry2.pid,0)}catch{registry.delete(name),reaped.push(name)}return reaped}function killAllServices(){for(let[_name,entry2]of registry)try{process.kill(entry2.pid,"SIGTERM")}catch{registry.delete(_name)}let deadline=Date.now()+3000,checkInterval=200,stillAlive=()=>{for(let[,entry2]of registry)try{return process.kill(entry2.pid,0),!0}catch{}return!1};while(Date.now()<deadline&&stillAlive()){let waitUntil=Date.now()+checkInterval;while(Date.now()<waitUntil);}for(let[name,entry2]of registry){try{process.kill(entry2.pid,0),process.kill(entry2.pid,"SIGKILL")}catch{}registry.delete(name)}}function clearRegistry(){registry.clear()}var registry;var init_service_registry=__esm(()=>{registry=new Map});function parseDuration(input){let match=input.trim().match(DURATION_RE);if(!match)throw Error(`Invalid duration: "${input}". Expected format: 10m, 2h, 24h, 1d`);let value=Number.parseFloat(match[1]),unit=match[2].toLowerCase(),ms=value*{s:1000,sec:1000,m:60000,min:60000,h:3600000,hr:3600000,d:86400000,day:86400000}[unit];if(ms<=0)throw Error(`Duration must be positive: "${input}"`);return ms}function expandRange(range,step,min,max){if(step===0)throw Error("Cron step value cannot be 0");if(range==="*"){let out=[];for(let i2=min;i2<=max;i2+=step)out.push(i2);return out}if(range.includes("-")){let[start,end]=range.split("-").map(Number),out=[];for(let i2=start;i2<=end;i2+=step)out.push(i2);return out}return[Number.parseInt(range,10)]}function parseCronField(field,min,max){let values2=new Set;for(let part of field.split(",")){let stepMatch=part.match(/^(.+)\/(\d+)$/),step=stepMatch?Number.parseInt(stepMatch[2],10):1,range=stepMatch?stepMatch[1]:part;for(let v of expandRange(range,step,min,max))values2.add(v)}return[...values2].sort((a,b2)=>a-b2)}function getTimeParts(date,tz){if(!tz)return{month:date.getMonth()+1,dom:date.getDate(),dow:date.getDay(),hour:date.getHours(),minute:date.getMinutes()};let parts=new Intl.DateTimeFormat("en-US",{timeZone:tz,year:"numeric",month:"numeric",day:"numeric",hour:"numeric",minute:"numeric",weekday:"short",hour12:!1}).formatToParts(date),get3=(type2)=>Number(parts.find((p)=>p.type===type2)?.value??0),dayMap={Sun:0,Mon:1,Tue:2,Wed:3,Thu:4,Fri:5,Sat:6},weekday=parts.find((p)=>p.type==="weekday")?.value??"Sun";return{month:get3("month"),dom:get3("day"),dow:dayMap[weekday]??0,hour:get3("hour")===24?0:get3("hour"),minute:get3("minute")}}function parseOpts(afterOrOpts){if(afterOrOpts instanceof Date)return{after:afterOrOpts};if(afterOrOpts)return{after:afterOrOpts.after,timezone:afterOrOpts.timezone};return{}}function advanceToNextDay(candidate,tz){candidate.setTime(candidate.getTime()+86400000);let tp=getTimeParts(candidate,tz);candidate.setTime(candidate.getTime()-tp.hour*3600000-tp.minute*60000)}function parseCronExpr(cronExpr){let parts=cronExpr.trim().split(/\s+/);if(parts.length<5)throw Error(`Invalid cron expression: "${cronExpr}"`);let[minField,hourField,domField,monthField,dowField]=parts;return{minutes:parseCronField(minField,0,59),hours:parseCronField(hourField,0,23),doms:parseCronField(domField,1,31),months:parseCronField(monthField,1,12),dows:parseCronField(dowField,0,6),domRestricted:domField!=="*",dowRestricted:dowField!=="*"}}function computeNextCronDue(cronExpr,afterOrOpts){let{after,timezone}=parseOpts(afterOrOpts),cron=parseCronExpr(cronExpr),candidate=new Date((after??new Date).getTime());candidate.setSeconds(0,0),candidate.setTime(candidate.getTime()+60000);let limit=new Date(candidate.getTime()+31622400000);while(candidate<=limit){let tp=getTimeParts(candidate,timezone);if(!cron.months.includes(tp.month)){advanceToNextDay(candidate,timezone);continue}if(!(cron.domRestricted&&cron.dowRestricted?cron.doms.includes(tp.dom)||cron.dows.includes(tp.dow):cron.doms.includes(tp.dom)&&cron.dows.includes(tp.dow))){advanceToNextDay(candidate,timezone);continue}if(!cron.hours.includes(tp.hour)){candidate.setTime(candidate.getTime()+3600000-tp.minute*60000);continue}if(cron.minutes.includes(tp.minute))return candidate;candidate.setTime(candidate.getTime()+60000)}throw Error(`No next cron occurrence found for "${cronExpr}" within 366 days`)}var DURATION_RE;var init_cron=__esm(()=>{DURATION_RE=/^(\d+(?:\.\d+)?)\s*(s|sec|m|min|h|hr|d|day)s?$/i});import{randomUUID as randomUUID7}from"crypto";function parseNotifyPayload(channel,raw){switch(channel){case"genie_task_stage":{let parts=raw.split(":");if(parts.length<3)return null;return{channel,eventType:"task.stage_change",payload:{taskId:parts[0],fromStage:parts[1],toStage:parts[2]},taskId:parts[0],summary:`Task ${parts[0]} moved from ${parts[1]} to ${parts[2]}`}}case"genie_executor_state":{let parts=raw.split(":");if(parts.length<4)return null;let eventType=parts[3]==="error"?"executor.error":"executor.state_change";return{channel,eventType,payload:{executorId:parts[0],agentId:parts[1],oldState:parts[2],newState:parts[3]},agentId:parts[1],summary:`${parts[1]} state: ${parts[2]} \u2192 ${parts[3]}`}}case"genie_message":{let parts=raw.split(":");if(parts.length<2)return null;return{channel,eventType:"task.comment",payload:{messageId:parts[0],conversationId:parts[1]},summary:`New message in conversation ${parts[1]}`}}case"genie_audit_event":{let parts=raw.split(":");if(parts.length<3)return null;return{channel,eventType:`${parts[0]}.${parts[2]}`,payload:{entityType:parts[0],entityId:parts[1],auditEventType:parts[2]},summary:`${parts[0]} ${parts[1]}: ${parts[2]}`}}default:return null}}function isActionableEvent(eventType){return ACTIONABLE_EVENTS.has(eventType)||eventType.startsWith("request.")}async function resolveTargetTeams(event){if(!isActionableEvent(event.eventType))return[];let active=(await listTeams()).filter((t)=>t.status==="in_progress");if(event.agentId){let aid=event.agentId;return active.filter((t)=>t.members.includes(aid)||t.leader===aid).map((t)=>t.name)}return active.map((t)=>t.name)}async function writeMailbox(repoPath,leader,message,traceId){try{await(await getConnection())`
|
|
2136
2136
|
INSERT INTO mailbox (id, repo_path, "from", "to", body, trace_id, created_at)
|
|
2137
2137
|
VALUES (${`mail-${traceId}`}, ${repoPath}, 'system', ${leader}, ${message}, ${traceId}, now())
|
|
2138
2138
|
`}catch{try{await send(repoPath,"system",leader,message)}catch{}}}async function deliverToHierarchy(leader,teamName,message,traceId){let sql=await getConnection(),current=leader,visited=new Set([current]);for(;;){let reportsTo=(await sql`
|
|
@@ -2527,7 +2527,7 @@ ${queryContent}`;let{messages:queryMessages}=state.provider.runQuery({agentId:se
|
|
|
2527
2527
|
`}}class TurnTracker{turns=new Map;open(sessionKey2,turnId,messageId){this.turns.set(sessionKey2,{turnId,sessionKey:sessionKey2,messageId,startedAt:Date.now(),closed:!1})}close(sessionKey2,action){let turn=this.turns.get(sessionKey2);if(turn&&!turn.closed)turn.closed=!0,turn.closedAction=action}isOpen(sessionKey2){let turn=this.turns.get(sessionKey2);return turn!==void 0&&!turn.closed}getTurnId(sessionKey2){return this.turns.get(sessionKey2)?.turnId}getByTurnId(turnId){for(let turn of this.turns.values())if(turn.turnId===turnId)return turn;return}delete(sessionKey2){this.turns.delete(sessionKey2)}}var exports_omni_bridge={};__export(exports_omni_bridge,{OmniBridge:()=>OmniBridge});import{closeSync as closeSync4,existsSync as existsSync46,mkdirSync as mkdirSync20,openSync as openSync4,readFileSync as readFileSync28,unlinkSync as unlinkSync9,writeSync}from"fs";import{dirname as dirname15}from"path";function withTimeout2(p,ms,label){return new Promise((resolve11,reject)=>{let timer2=setTimeout(()=>reject(Error(`${label} timed out after ${ms}ms`)),ms);timer2.unref?.(),p.then((v)=>{clearTimeout(timer2),resolve11(v)},(err)=>{clearTimeout(timer2),reject(err)})})}function isPgConnectionError(err){if(!err||typeof err!=="object")return!1;let e=err,code=e.code??"";if(["ECONNREFUSED","ECONNRESET","ETIMEDOUT","ENOTFOUND","EPIPE","EHOSTUNREACH"].includes(code))return!0;let msg=e.message??String(err);return/ECONNREFUSED|ECONNRESET|ETIMEDOUT|ENOTFOUND|EPIPE|connection terminated|connection closed|server closed the connection|the database system is shutting down/i.test(msg)}class OmniBridge{nc=null;sub=null;executor;turnTracker=new TurnTracker;heartbeatPublisher=null;sessions=new Map;messageQueue=[];recentMessageIds=new Map;idleCheckTimer=null;sc=import_nats4.StringCodec();sql=null;pgAvailable=!1;pgProvider;natsConnectFn;queueConfig;queue=null;sessionStore=null;natsUrl;idleTimeoutMs;maxConcurrent;executorType;pidfilePath=null;startedAtMs=0;pingSub=null;signalCleanup=null;constructor(config={}){if(this.natsUrl=config.natsUrl??process.env.GENIE_NATS_URL??DEFAULT_NATS_URL2,this.idleTimeoutMs=config.idleTimeoutMs??(process.env.GENIE_IDLE_TIMEOUT_MS?Number(process.env.GENIE_IDLE_TIMEOUT_MS):DEFAULT_IDLE_TIMEOUT_MS2),this.maxConcurrent=config.maxConcurrent??(process.env.GENIE_MAX_CONCURRENT?Number(process.env.GENIE_MAX_CONCURRENT):DEFAULT_MAX_CONCURRENT),this.pgProvider=config.pgProvider??(async()=>{let{getConnection:getConnection2}=await Promise.resolve().then(() => (init_db(),exports_db));return await getConnection2()}),this.natsConnectFn=config.natsConnectFn??import_nats4.connect,this.queueConfig=config.queue??{},this.executorType=resolveExecutorType(config.executorType),this.executorType==="sdk")this.executor=new ClaudeSdkOmniExecutor;else this.executor=new ClaudeCodeOmniExecutor}async start(){if(this.nc){console.log("[omni-bridge] Already running");return}console.log(`[omni-bridge] Connecting to NATS at ${this.natsUrl}...`),this.nc=await this.natsConnectFn({servers:this.natsUrl,name:"genie-omni-bridge",reconnect:!0,maxReconnectAttempts:-1,reconnectTimeWait:2000}),console.log("[omni-bridge] Connected to NATS"),await this.setupPg(),this.wireExecutorHooks(),this.subscribeOmniChannels(),this.startedAtMs=Date.now(),this.subscribePingChannel(),await this.claimPidfile(),this.armSignalCleanup(),this.idleCheckTimer=setInterval(()=>this.checkIdleSessions(),IDLE_CHECK_INTERVAL_MS),console.log(`[omni-bridge] Listening on omni.message.> (max_concurrent=${this.maxConcurrent}, idle_timeout=${this.idleTimeoutMs}ms)`)}async setupPg(){if(await this.probePg(),this.pgAvailable&&this.sql)this.sessionStore=new BridgeSessionStore(this.sql),await this.recoverSessions();if(this.executorType==="sdk"&&this.pgAvailable&&this.sql)this.queue=new OmniQueue(this.sql,(_req,msg)=>this.routeMessage(msg),this.queueConfig),await this.queue.recoverStale(),this.queue.start()}wireExecutorHooks(){this.executor.setSafePgCall(this.safePgCall.bind(this));let sc=this.sc,nc=this.nc;if(!nc)return;let publish=(topic,payload)=>{nc.publish(topic,sc.encode(payload))};this.executor.setNatsPublish(publish),this.heartbeatPublisher=new HeartbeatPublisher({publish})}subscribeOmniChannels(){if(!this.nc)return;this.sub=this.nc.subscribe("omni.message.>",{queue:"genie-bridge"}),this.processSubscription();let turnSubs=["omni.turn.open.>","omni.turn.done.>","omni.turn.nudge.>","omni.turn.timeout.>"];for(let topic of turnSubs){let sub=this.nc.subscribe(topic,{queue:"genie-bridge"});this.processTurnEvents(sub)}let sessionResetSub=this.nc.subscribe("omni.session.reset.>",{queue:"genie-bridge"});this.processSessionResetEvents(sessionResetSub)}subscribePingChannel(){if(!this.nc)return;this.pingSub=this.nc.subscribe(BRIDGE_PING_SUBJECT);let pingSub=this.pingSub,pingSc=this.sc,pingStart=this.startedAtMs;(async()=>{for await(let m of pingSub){let pong={ok:!0,pid:process.pid,uptimeMs:Date.now()-pingStart,subjects:["omni.message.>","omni.turn.open.>","omni.session.reset.>",BRIDGE_PING_SUBJECT]};try{m.respond(pingSc.encode(JSON.stringify(pong)))}catch{}}})().catch(()=>{})}async claimPidfile(){this.pidfilePath=getBridgePidfilePath();try{mkdirSync20(dirname15(this.pidfilePath),{recursive:!0}),this.evictStalePidfile(this.pidfilePath);let fd=openSync4(this.pidfilePath,"wx"),payload=JSON.stringify({pid:process.pid,startedAt:this.startedAtMs,subjects:["omni.message.>","omni.turn.open.>","omni.session.reset.>",BRIDGE_PING_SUBJECT],natsUrl:this.natsUrl});writeSync(fd,payload),closeSync4(fd)}catch(err){await this.rollbackStartOnPidfileError(err)}}evictStalePidfile(path3){if(!existsSync46(path3))return;let stalePid=null;try{let raw=readFileSync28(path3,"utf8"),parsed=JSON.parse(raw);if(typeof parsed.pid==="number"&&Number.isFinite(parsed.pid))stalePid=parsed.pid}catch{}if(stalePid!==null&&this.isPidAlive(stalePid))throw Error(`pidfile locked by PID ${stalePid}`);try{unlinkSync9(path3)}catch{}}isPidAlive(pid){try{return process.kill(pid,0),!0}catch(probeErr){return probeErr.code!=="ESRCH"}}async rollbackStartOnPidfileError(err){this.pidfilePath=null;let detail=err instanceof Error?err.message:String(err);try{if(this.pingSub)this.pingSub.unsubscribe()}catch{}this.pingSub=null;try{await this.nc?.drain()}catch{}throw this.nc=null,Error(`[omni-bridge] pidfile locked at ${getBridgePidfilePath()}: ${detail}`)}armSignalCleanup(){let onSignal=()=>{if(this.pidfilePath){try{unlinkSync9(this.pidfilePath)}catch{}this.pidfilePath=null}};process.once("SIGTERM",onSignal),process.once("SIGINT",onSignal),this.signalCleanup=()=>{process.removeListener("SIGTERM",onSignal),process.removeListener("SIGINT",onSignal)}}async stop(){if(!this.nc){console.log("[omni-bridge] Not running");return}if(console.log("[omni-bridge] Shutting down..."),this.queue)this.queue.stop(),this.queue=null;if(this.idleCheckTimer)clearInterval(this.idleCheckTimer),this.idleCheckTimer=null;if(this.heartbeatPublisher?.stopAll(),this.heartbeatPublisher=null,await this.shutdownActiveSessions(),this.unsubscribeChannels(),this.clearPidfileOnStop(),this.signalCleanup)this.signalCleanup(),this.signalCleanup=null;try{await this.nc.drain()}catch{}this.nc=null,this.sql=null,this.pgAvailable=!1,this.sessionStore=null,console.log("[omni-bridge] Stopped")}async shutdownActiveSessions(){for(let[key,entry2]of this.sessions){if(entry2.idleTimer)clearTimeout(entry2.idleTimer);if(entry2.spawning||!entry2.session)continue;if(entry2.session.executorType==="tmux"){console.log(`[omni-bridge] Detaching from tmux session ${key} (pane stays alive)`);continue}try{await this.executor.shutdown(entry2.session)}catch(err){console.warn(`[omni-bridge] Error shutting down session ${key}:`,err)}let closeId=entry2.pgBridgeSessionId;if(closeId&&this.sessionStore)await this.safePgCall("session_close_sdk",(sql)=>new BridgeSessionStore(sql).close(closeId),void 0)}this.sessions.clear()}unsubscribeChannels(){if(this.sub)this.sub.unsubscribe(),this.sub=null;if(this.pingSub){try{this.pingSub.unsubscribe()}catch{}this.pingSub=null}}clearPidfileOnStop(){if(!this.pidfilePath)return;try{unlinkSync9(this.pidfilePath)}catch{}this.pidfilePath=null}async status(){let now=Date.now(),activeFromPg=null,executorIds=[];if(this.pgAvailable&&this.sql){let rows=await this.safePgCall("status_active_count",async(sql)=>sql`
|
|
2528
2528
|
SELECT id FROM executors
|
|
2529
2529
|
WHERE ended_at IS NULL AND metadata->>'source' = 'omni'
|
|
2530
|
-
`,null);if(rows)activeFromPg=rows.length,executorIds=rows.map((r)=>r.id)}let pgQueue=null;if(this.queue)pgQueue=await this.safePgCall("status_queue_stats",(_sql)=>this.queue?.stats()??Promise.resolve(null),null);return{connected:this.nc!==null,natsUrl:this.natsUrl,pgAvailable:this.pgAvailable,activeSessions:activeFromPg??this.sessions.size,maxConcurrent:this.maxConcurrent,idleTimeoutMs:this.idleTimeoutMs,queueDepth:pgQueue?pgQueue.pending+pgQueue.processing:this.messageQueue.length,executorType:this.executorType,executorIds,pgQueue,sessions:Array.from(this.sessions.entries()).map(([key,entry2])=>({id:key,agentName:entry2.session.agentName,chatId:entry2.session.chatId,instanceId:entry2.instanceId,executorType:entry2.session.executorType,spawning:entry2.spawning,idleMs:now-entry2.session.lastActivityAt,bufferSize:entry2.buffer.length}))}}async probePg(){try{let sql=await withTimeout2(this.pgProvider(),PG_STARTUP_PROBE_TIMEOUT_MS,"PG provider startup");await withTimeout2(Promise.resolve(sql`SELECT 1`),PG_STARTUP_PROBE_TIMEOUT_MS,"PG SELECT 1 probe"),this.sql=sql,this.pgAvailable=!0,console.log("[omni-bridge] PG reachable \u2014 session recovery enabled")}catch(err){this.sql=null,this.pgAvailable=!1;let msg=err instanceof Error?err.message:String(err);if(isPgConnectionError(err)){console.warn(`[omni-bridge] PG unavailable \u2014 session recovery disabled (${msg})`);return}throw Error(`[omni-bridge] PG schema mismatch or setup error: ${msg}. ${"Run `bun run migrate` (or the equivalent migration command) and retry."}`)}}async recoverSessions(){if(!this.sessionStore)return;let orphanedCount=await this.safePgCall("recover_orphan_all",(sql)=>new BridgeSessionStore(sql).markAllOrphaned(),0);if(orphanedCount>0)console.log(`[omni-bridge] Startup cleanup: orphaned ${orphanedCount} stale session(s) from previous run`)}async safePgCall(op,fn,fallback,ctx){if(!this.pgAvailable||!this.sql)return fallback;let sql=this.sql;try{return await withTimeout2(fn(sql),PG_RUNTIME_QUERY_TIMEOUT_MS,`safePgCall(${op})`)}catch(err){let msg=err instanceof Error?err.message:String(err),execPart=ctx?.executorId?` executor_id=${ctx.executorId}`:"",chatPart=ctx?.chatId?` chat_id=${ctx.chatId}`:"";if(console.warn(`[omni-bridge] safePgCall(${op}) failed${execPart}${chatPart}: ${msg}`),isPgConnectionError(err))this.pgAvailable=!1,this.sql=null,console.warn("[omni-bridge] PG connection lost \u2014 switching to degraded mode");return fallback}}fillSubjectMetadata(parsed,subject){let parts=subject.split(".");if(parts.length<4)return;parsed.instanceId=parsed.instanceId||parts[2],parsed.chatId=parsed.chatId||parts[3]}isDuplicateMessage(messageId){if(!messageId)return!1;if(this.recentMessageIds.has(messageId))return!0;if(this.recentMessageIds.set(messageId,Date.now()),this.recentMessageIds.size>1000){let cutoff=Date.now()-60000;for(let[id,ts3]of this.recentMessageIds)if(ts3<cutoff)this.recentMessageIds.delete(id)}return!1}async dispatchMessage(parsed){if(this.queue){let env=parsed.env??{};await this.queue.enqueue(parsed,env);return}let key=`${parsed.agent}:${parsed.chatId}`,hasSession=this.sessions.has(key);console.log(`[omni-bridge] Routing message for ${key} (hasSession=${hasSession}, queue=${!!this.queue})`),await this.routeMessage(parsed),console.log(`[omni-bridge] routeMessage done for ${key}`)}async processSubscription(){if(!this.sub)return;for await(let msg of this.sub)try{let data=this.sc.decode(msg.data),parsed=JSON.parse(data);this.fillSubjectMetadata(parsed,msg.subject),console.log(`[omni-bridge] NATS message received: ${msg.subject} agent=${parsed.agent} chat=${parsed.chatId}`);let messageId=parsed.messageId;if(this.isDuplicateMessage(messageId)){console.log(`[omni-bridge] Dedup: skipping duplicate messageId=${messageId}`);continue}if(!parsed.chatId||!parsed.agent){console.warn("[omni-bridge] Dropping message: missing chatId or agent",msg.subject);continue}await this.dispatchMessage(parsed)}catch(err){console.error("[omni-bridge] Error processing message:",err)}}async processTurnEvents(sub){for await(let msg of sub)try{let payload=JSON.parse(this.sc.decode(msg.data)),parts=msg.subject.split("."),eventType=parts[2],instanceId=parts[3],chatId=parts.slice(4).join(".");console.log(`[omni-bridge] Turn event: ${eventType} instance=${instanceId} chat=${chatId}`);let sessionKey2=this.findSessionKey(instanceId,chatId);if(!sessionKey2&&payload.turnId){if(sessionKey2=this.findSessionKeyByTurnId(payload.turnId),sessionKey2)console.log(`[omni-bridge] Matched session via turnId fallback: ${sessionKey2}`)}if(sessionKey2)await this.routeTurnEvent(eventType,sessionKey2,payload);else console.log(`[omni-bridge] No session found for turn.${eventType} (instance=${instanceId}, chat=${chatId})`)}catch(err){console.warn("[omni-bridge] Error processing turn event:",err)}}async routeTurnEvent(eventType,sessionKey2,payload){switch(eventType){case"open":this.turnTracker.open(sessionKey2,payload.turnId,payload.messageId),this.startHeartbeat(sessionKey2,payload);break;case"done":this.turnTracker.close(sessionKey2,payload.action),this.heartbeatPublisher?.stop(sessionKey2),await this.handleTurnDone(sessionKey2);break;case"nudge":await this.handleTurnNudge(sessionKey2,payload.message);break;case"timeout":this.heartbeatPublisher?.stop(sessionKey2),await this.handleTurnTimeout(sessionKey2);break}}startHeartbeat(sessionKey2,payload){let publisher=this.heartbeatPublisher;if(!publisher)return;let entry2=this.sessions.get(sessionKey2);if(!entry2)return;let turnId=payload.turnId;if(!turnId)return;publisher.start(sessionKey2,{instanceId:entry2.instanceId,chatId:entry2.session?.chatId??"",turnId,isBusy:async()=>{let live=this.sessions.get(sessionKey2);if(!live?.session)return!1;try{return await this.executor.isBusy(live.session)}catch{return!1}}})}findSessionKeyByTurnId(turnId){for(let[key]of this.sessions)if(this.turnTracker.getTurnId(key)===turnId)return key;return}chatIdMap=new Map;findSessionKey(instanceId,chatId){let resolvedChatId=this.chatIdMap.get(chatId);for(let[key,entry2]of this.sessions){if(entry2.instanceId!==instanceId)continue;if(entry2.session?.chatId===chatId)return key;if(resolvedChatId&&entry2.session?.chatId===resolvedChatId)return key;if(entry2.spawning&&key.endsWith(`:${chatId}`))return key;if(resolvedChatId&&entry2.spawning&&key.endsWith(`:${resolvedChatId}`))return key}return}async handleTurnNudge(sessionKey2,nudgeText){let entry2=this.sessions.get(sessionKey2);if(!entry2?.session)return;try{await this.executor.injectNudge(entry2.session,nudgeText)}catch(err){console.warn(`[omni-bridge] Failed to inject nudge for ${sessionKey2}:`,err)}}async processSessionResetEvents(sub){for await(let msg of sub)try{let parts=msg.subject.split(".");if(parts.length<5){console.warn(`[omni-bridge] Malformed session-reset subject: ${msg.subject}`);continue}let instanceId=parts[3],chatId=parts.slice(4).join("."),action;try{action=JSON.parse(this.sc.decode(msg.data)).action}catch{}await this.handleSessionReset(instanceId,chatId,action)}catch(err){console.warn("[omni-bridge] Error processing session reset event:",err)}}async handleSessionReset(instanceId,chatId,action){let sessionKey2=this.findSessionKey(instanceId,chatId);if(!sessionKey2){console.log(`[omni-bridge] Session reset for cold chat ${instanceId}/${chatId} \u2014 no-op`);return}let entry2=this.sessions.get(sessionKey2);if(!entry2)return;let actionTag=action?` (action=${action})`:"";if(entry2.spawning){console.log(`[omni-bridge] Session reset for spawning ${sessionKey2}${actionTag}, marking cancelled`),entry2.cancelled=!0,entry2.buffer=[],this.turnTracker.close(sessionKey2,"reset"),this.heartbeatPublisher?.stop(sessionKey2),await this.removeSession(sessionKey2),await this.drainQueue();return}if(!entry2.session)return;console.log(`[omni-bridge] Session reset for ${sessionKey2}${actionTag}, evicting`),this.turnTracker.close(sessionKey2,"reset"),this.heartbeatPublisher?.stop(sessionKey2);try{await this.executor.shutdown(entry2.session)}catch(err){console.warn(`[omni-bridge] Error shutting down reset session ${sessionKey2}:`,err)}await this.removeSession(sessionKey2),await this.drainQueue()}async handleTurnDone(sessionKey2){if(!this.sessions.get(sessionKey2)?.session)return;console.log(`[omni-bridge] Turn done for ${sessionKey2}, session stays alive for next message`),this.resetIdleTimer(sessionKey2)}async handleTurnTimeout(sessionKey2){let entry2=this.sessions.get(sessionKey2);if(!entry2?.session)return;console.warn(`[omni-bridge] Turn timed out for ${sessionKey2}, evicting session`),this.turnTracker.close(sessionKey2,"timeout"),this.heartbeatPublisher?.stop(sessionKey2);try{await this.executor.shutdown(entry2.session)}catch(err){console.warn(`[omni-bridge] Error shutting down timed-out session ${sessionKey2}:`,err)}this.sessions.delete(sessionKey2)}async routeMessage(message){let key=`${message.agent}:${message.chatId}`,entry2=this.sessions.get(key);if(entry2){if(entry2.spawning){if(entry2.buffer.length<MAX_BUFFER_PER_CHAT)entry2.buffer.push(message);else console.warn(`[omni-bridge] Buffer full (${MAX_BUFFER_PER_CHAT}) for ${key}, dropping message from ${message.sender}`),await this.publishBufferFullReply(message);return}if(await this.executor.isAlive(entry2.session)){await this.executor.deliver(entry2.session,message),this.resetIdleTimer(key);let bsId=entry2.pgBridgeSessionId;if(bsId&&this.sessionStore)this.safePgCall("session_activity",(sql)=>new BridgeSessionStore(sql).recordActivity(bsId),void 0,{chatId:message.chatId});return}await this.removeSession(key)}await this.spawnSession(message)}async spawnSession(message){let key=`${message.agent}:${message.chatId}`,existing=this.sessions.get(key);if(existing){if(existing.buffer.length<MAX_BUFFER_PER_CHAT)existing.buffer.push(message),console.log(`[omni-bridge] Buffered message for existing session ${key} (buffer=${existing.buffer.length})`);return}if(this.sessions.size>=this.maxConcurrent){this.messageQueue.push(message),await this.publishAutoReply(message),console.log(`[omni-bridge] Max concurrent (${this.maxConcurrent}) reached, queued message for ${key}`);return}let placeholder={session:null,instanceId:message.instanceId,spawning:!0,buffer:[message],idleTimer:null};this.sessions.set(key,placeholder);try{let raw=message,payloadEnv=raw.env,spawnEnv={OMNI_API_KEY:payloadEnv?.OMNI_API_KEY??process.env.OMNI_API_KEY??"",OMNI_INSTANCE:payloadEnv?.OMNI_INSTANCE??message.instanceId,OMNI_CHAT:payloadEnv?.OMNI_CHAT??message.chatId,OMNI_MESSAGE:payloadEnv?.OMNI_MESSAGE??raw.messageId??"",OMNI_TURN_ID:payloadEnv?.OMNI_TURN_ID||"",OMNI_SENDER_NAME:payloadEnv?.OMNI_SENDER_NAME??message.sender??""};console.log(`[omni-bridge] Spawning session for ${key}...`);let session=await this.executor.spawn(message.agent,message.chatId,spawnEnv,message.content);if(placeholder.cancelled){console.log(`[omni-bridge] Spawn for ${key} completed but was cancelled by reset, shutting down`);try{await this.executor.shutdown(session)}catch(err){console.warn(`[omni-bridge] Error shutting down cancelled spawn for ${key}:`,err)}return}if(placeholder.session=session,placeholder.spawning=!1,this.sessionStore){let claudeSessionId=session.sdk?.claudeSessionId??session.tmux?.claudeSessionId,pgId=await this.safePgCall("session_create",(sql)=>new BridgeSessionStore(sql).create({instanceId:message.instanceId,chatId:message.chatId,agentName:message.agent,executorId:session.sdk?.executorId,tmuxPaneId:session.tmux?.paneId,claudeSessionId}),void 0,{chatId:message.chatId});if(pgId)placeholder.pgBridgeSessionId=pgId}for(let buffered of placeholder.buffer)await this.executor.deliver(session,buffered);placeholder.buffer=[],this.resetIdleTimer(key);let sessionTag=session.executorType==="tmux"?`(tmux pane=${session.tmux?.paneId})`:"(executor=sdk)";console.log(`[omni-bridge] Session active: ${key} ${sessionTag}`)}catch(err){console.error(`[omni-bridge] Failed to spawn session for ${key}:`,err);let lostMessages=placeholder.buffer;if(lostMessages.length>0)console.warn(`[omni-bridge] Re-queuing ${lostMessages.length} buffered message(s) from failed spawn for ${key}`),this.messageQueue.push(...lostMessages);this.sessions.delete(key)}}resetIdleTimer(key){let entry2=this.sessions.get(key);if(!entry2)return;if(entry2.idleTimer)clearTimeout(entry2.idleTimer);entry2.idleTimer=setTimeout(async()=>{console.log(`[omni-bridge] Idle timeout for ${key}, shutting down...`);try{await this.executor.shutdown(entry2.session)}catch{}await this.removeSession(key),await this.drainQueue()},this.idleTimeoutMs)}async checkIdleSessions(){let now=Date.now();for(let[key,entry2]of this.sessions){if(entry2.spawning)continue;if(!await this.executor.isAlive(entry2.session)){console.log(`[omni-bridge] Dead session detected: ${key}`),await this.removeSession(key);continue}let idleMs=now-entry2.session.lastActivityAt;if(idleMs>this.idleTimeoutMs){console.log(`[omni-bridge] Forcing idle shutdown: ${key} (idle ${Math.round(idleMs/1000)}s)`);try{await this.executor.shutdown(entry2.session)}catch{}await this.removeSession(key)}}}async removeSession(key){let entry2=this.sessions.get(key);if(entry2?.idleTimer)clearTimeout(entry2.idleTimer);this.heartbeatPublisher?.stop(key);let closeId=entry2?.pgBridgeSessionId;if(closeId&&this.sessionStore)await this.safePgCall("session_close",(sql)=>new BridgeSessionStore(sql).close(closeId),void 0);this.sessions.delete(key)}async drainQueue(){while(this.messageQueue.length>0){if(this.sessions.size>=this.maxConcurrent)break;let message=this.messageQueue.shift();if(message)await this.spawnSession(message)}}async publishBufferFullReply(message){if(!this.nc)return;let topic=`omni.reply.${message.instanceId}.${message.chatId}`,reply={content:"Fila de mensagens cheia, por favor aguarde e tente novamente.",agent:message.agent,chat_id:message.chatId,instance_id:message.instanceId,timestamp:new Date().toISOString(),auto_reply:!0};this.nc.publish(topic,this.sc.encode(JSON.stringify(reply)))}async publishAutoReply(message){if(!this.nc)return;let topic=`omni.reply.${message.instanceId}.${message.chatId}`,reply={content:"Aguarde um momento, estou atendendo outros clientes.",agent:message.agent,chat_id:message.chatId,instance_id:message.instanceId,timestamp:new Date().toISOString(),auto_reply:!0};this.nc.publish(topic,this.sc.encode(JSON.stringify(reply)))}}var import_nats4,DEFAULT_NATS_URL2="localhost:4222",DEFAULT_IDLE_TIMEOUT_MS2=900000,DEFAULT_MAX_CONCURRENT=20,MAX_BUFFER_PER_CHAT=50,IDLE_CHECK_INTERVAL_MS=30000,PG_STARTUP_PROBE_TIMEOUT_MS=5000,PG_RUNTIME_QUERY_TIMEOUT_MS=2000;var init_omni_bridge=__esm(()=>{init_bridge_status();init_executor_config();init_agent_heartbeat();init_claude_code2();init_claude_sdk2();import_nats4=__toESM(require_mod4(),1)});import{existsSync as existsSync47,mkdirSync as mkdirSync21,readFileSync as readFileSync29,readdirSync as readdirSync10,renameSync as renameSync7,statSync as statSync12,writeFileSync as writeFileSync18}from"fs";import{homedir as homedir38}from"os";import{dirname as dirname16,join as join59,resolve as resolvePath4}from"path";import{pathToFileURL as pathToFileURL2}from"url";function scanDir(dirAbs){if(!existsSync47(dirAbs))return[];return readdirSync10(dirAbs).filter((name)=>name.endsWith(".ts")&&!name.startsWith("_")&&!name.endsWith(".test.ts")).map((name)=>join59(dirAbs,name)).filter((p)=>{try{return statSync12(p).isFile()}catch{return!1}})}function discoverHooks(opts){let home=homedir38(),discovered=[],teamsRoot=join59(home,".claude","teams");if(existsSync47(teamsRoot))for(let teamName of readdirSync10(teamsRoot)){if(teamName.startsWith("_"))continue;let teamHooksDir=join59(teamsRoot,teamName,"hooks");for(let path3 of scanDir(teamHooksDir))discovered.push({path:path3,scope:"team",teamName})}if(opts.repoRoot){let repoHooksDir=join59(opts.repoRoot,".genie","hooks"),repoRemoteUrl=opts.resolveRepoRemoteUrl?.(opts.repoRoot)??void 0;for(let path3 of scanDir(repoHooksDir))discovered.push({path:path3,scope:"repo",repoRemoteUrl})}let globalHooksDir=join59(home,".genie","hooks");for(let path3 of scanDir(globalHooksDir))discovered.push({path:path3,scope:"global"});return discovered}function quarantine(filePath,errorMessage){let dir=dirname16(filePath),quarantineDir=join59(dir,"_quarantine");if(!existsSync47(quarantineDir))mkdirSync21(quarantineDir,{recursive:!0});let basename11=filePath.split("/").pop()??"unknown.ts",target=join59(quarantineDir,basename11);try{renameSync7(filePath,target)}catch{}try{writeFileSync18(`${target}.error`,errorMessage,"utf-8")}catch{}return target}function validateHandler(imported){if(typeof imported!=="object"||imported===null)return"expected default export to be an object";let handler=imported;if(handler.version!=="1")return`unknown handler version: ${String(handler.version)} (expected '1')`;if(typeof handler.name!=="string"||handler.name.length===0)return"handler.name must be a non-empty string";if(typeof handler.event!=="string")return"handler.event must be a HookEventName string";if(typeof handler.priority!=="number"||Number.isNaN(handler.priority))return"handler.priority must be a finite number";if(typeof handler.fn!=="function")return"handler.fn must be a function";return null}async function loadExternalHooks(opts={}){let trustPath=opts.trustPath??defaultTrustPath(),trustFile;try{trustFile=readTrustFile(trustPath)}catch(err){let msg=err instanceof Error?err.message:String(err);return console.warn(`[hook-loader] trust file unreadable at ${trustPath}: ${msg} \u2014 refusing to load any external hooks`),[]}let importer=opts.importer??((url)=>import(url.href)),discovered=discoverHooks(opts),loaded=[],seenNames=new Map,outcomes=[];for(let source of discovered){let verify=verifyTrust(source.path,trustFile,{currentRepoRemoteUrl:source.repoRemoteUrl});if(!verify.trusted){outcomes.push({kind:"untrusted",reason:verify.reason,source});continue}let imported;try{imported=await importer(pathToFileURL2(resolvePath4(source.path)))}catch(err){let msg=err instanceof Error?`${err.name}: ${err.message}`:String(err),quarantined=quarantine(source.path,msg);outcomes.push({kind:"broken",error:msg,source,quarantined});continue}let handlerCandidate=imported.default,validationError=validateHandler(handlerCandidate);if(validationError!==null){let quarantined=quarantine(source.path,validationError);outcomes.push({kind:"broken",error:validationError,source,quarantined});continue}let stamped={...handlerCandidate,source:source.scope,manifest_path:source.path},existing=seenNames.get(stamped.name);if(existing){if(console.warn(`[hook-loader] handler name collision: "${stamped.name}" registered from ${existing.path} (${existing.scope}) shadows ${source.path} (${source.scope})`),opts.strict)throw Error(`--strict-hooks: handler name collision on "${stamped.name}" between ${existing.path} and ${source.path}`);outcomes.push({kind:"shadowed",handler:stamped,source,shadowedBy:existing});continue}parseCapabilities(readFileSync29(source.path,"utf-8")),seenNames.set(stamped.name,source),loaded.push(stamped),outcomes.push({kind:"loaded",handler:stamped,source})}let builtins=getRegistry().filter((h)=>h.source==="builtin");return setRegistry([...builtins,...loaded]),outcomes}var init_loader=__esm(()=>{init_hooks();init_trust()});var exports_hook_socket={};__export(exports_hook_socket,{startHookSocket:()=>startHookSocket,defaultHookSocketPath:()=>defaultHookSocketPath});import{existsSync as existsSync48,unlinkSync as unlinkSync10}from"fs";import{mkdir as mkdir7}from"fs/promises";import{createServer,connect as netConnect}from"net";import{homedir as homedir39}from"os";import{dirname as dirname17,join as join60}from"path";function defaultHookSocketPath(){if(process.env.GENIE_HOOK_SOCK)return process.env.GENIE_HOOK_SOCK;let home=process.env.GENIE_HOME??join60(homedir39(),".genie");return join60(home,"hook.sock")}async function detectStaleAndCleanup(socketPath){if(!existsSync48(socketPath))return"clean";if(await new Promise((resolve11)=>{let probe=netConnect(socketPath),finish=(alive)=>{try{probe.destroy()}catch{}resolve11(alive)};probe.once("connect",()=>finish(!0)),probe.once("error",()=>finish(!1)),setTimeout(()=>finish(!1),200).unref()}))return"live";try{return unlinkSync10(socketPath),"stale-removed"}catch(err){throw Error(`Failed to remove stale hook socket at ${socketPath}: ${err.message}`)}}function parseFrame(acc,length){if(length===-1){if(acc.length<4)return{kind:"incomplete",length:-1};let declared=acc.readUInt32BE(0);if(declared>MAX_FRAME_BYTES2)return{kind:"error",reason:`frame length ${declared} exceeds MAX_FRAME_BYTES`};if(declared===0)return{kind:"done",body:Buffer.alloc(0)};if(acc.length>=4+declared)return{kind:"done",body:acc.subarray(4,4+declared)};return{kind:"incomplete",length:declared}}if(acc.length>=4+length)return{kind:"done",body:acc.subarray(4,4+length)};return{kind:"incomplete",length}}function readOneFrame(socket){return new Promise((resolve11,reject)=>{let acc=Buffer.alloc(0),length=-1,cleanup=()=>{socket.off("data",onData),socket.off("end",onEnd),socket.off("error",onError)},onData=(chunk)=>{acc=acc.length===0?Buffer.from(chunk):Buffer.concat([acc,Buffer.from(chunk)],acc.length+chunk.length);let result2=parseFrame(acc,length);if(result2.kind==="incomplete"){length=result2.length;return}if(cleanup(),result2.kind==="done")resolve11(result2.body);else reject(Error(result2.reason))},onEnd=()=>{cleanup(),reject(Error(`socket closed after ${acc.length} bytes (expected ${length===-1?"4+":4+length})`))},onError=(err)=>{cleanup(),reject(err)};socket.on("data",onData),socket.on("end",onEnd),socket.on("error",onError)})}function writeFrame(socket,payload){let body=Buffer.from(payload,"utf-8");if(body.length>MAX_FRAME_BYTES2){let empty=Buffer.alloc(4);socket.end(empty);return}let header=Buffer.alloc(4);header.writeUInt32BE(body.length,0),socket.end(Buffer.concat([header,body]))}async function handleConnection(socket){socket.setNoDelay(!0);try{let body=await readOneFrame(socket),stdin=body.length===0?"":body.toString("utf-8"),reply=await dispatch(stdin);writeFrame(socket,reply??"")}catch(err){try{writeFrame(socket,"")}catch{}if(process.env.GENIE_HOOK_SOCK_DEBUG==="1")console.warn(`[hook-socket] connection error: ${err.message}`)}}async function startHookSocket(options={}){if(process.env.GENIE_WIDE_EMIT===void 0)process.env.GENIE_WIDE_EMIT="1";let loaderOutcomes=[];if(options.loadExternal!==!1)loaderOutcomes=await loadExternalHooks({repoRoot:options.repoRoot,strict:options.strict}),summarizeLoaderOutcomes(loaderOutcomes);let socketPath=defaultHookSocketPath();await mkdir7(dirname17(socketPath),{recursive:!0});let state=await detectStaleAndCleanup(socketPath);if(state==="live")throw Error(`hook socket at ${socketPath} is already live \u2014 another genie serve daemon is running. Refusing to start.`);if(state==="stale-removed")console.log(`hook-socket: removed stale socket at ${socketPath}`);let liveSockets=new Set,server3=createServer((socket)=>{liveSockets.add(socket),socket.once("close",()=>liveSockets.delete(socket)),handleConnection(socket)});await new Promise((resolve11,reject)=>{let onError=(err)=>{server3.off("listening",onListening),reject(err)},onListening=()=>{server3.off("error",onError),resolve11()};server3.once("error",onError),server3.once("listening",onListening),server3.listen(socketPath)}),console.log(`hook-socket: listening at ${socketPath}`);let stopped=!1;return{path:socketPath,stop:async()=>{if(stopped)return;stopped=!0;for(let sock of liveSockets)try{sock.destroy()}catch{}if(liveSockets.clear(),await new Promise((resolve11)=>{server3.close(()=>resolve11())}),existsSync48(socketPath))try{unlinkSync10(socketPath)}catch{}},loaderOutcomes}}function summarizeLoaderOutcomes(outcomes){if(outcomes.length===0)return;let loaded=0,untrusted=0,broken=0,shadowed=0;for(let outcome of outcomes)if(outcome.kind==="loaded")loaded+=1;else if(outcome.kind==="untrusted")untrusted+=1;else if(outcome.kind==="broken")broken+=1;else shadowed+=1;let parts=[`hook-loader: ${loaded} loaded`];if(untrusted>0)parts.push(`${untrusted} untrusted`);if(broken>0)parts.push(`${broken} broken (quarantined)`);if(shadowed>0)parts.push(`${shadowed} shadowed`);console.log(parts.join(", "))}var MAX_FRAME_BYTES2=1048576;var init_hook_socket=__esm(()=>{init_hooks();init_loader()});var exports_serve={};__export(exports_serve,{writeStoppingLockSync:()=>writeStoppingLockSync,startBrainServerIfEnabled:()=>startBrainServerIfEnabled,registerServeCommands:()=>registerServeCommands,isTuiSessionReady:()=>isTuiSessionReady,isStoppingLockActive:()=>isStoppingLockActive,isServeRunning:()=>isServeRunning,getTuiQuitBindingArgs:()=>getTuiQuitBindingArgs,getTuiKeybindings:()=>getTuiKeybindings,ensureTuiSession:()=>ensureTuiSession,clearStoppingLock:()=>clearStoppingLock,autoStartServe:()=>autoStartServe});import{execSync as execSync11,spawn as spawn4,spawnSync as spawnSync6}from"child_process";import{closeSync as closeSync5,existsSync as existsSync49,mkdirSync as mkdirSync22,openSync as openSync5,readFileSync as readFileSync30,unlinkSync as unlinkSync11,writeFileSync as writeFileSync19,writeSync as writeSync2}from"fs";import{homedir as homedir40}from"os";import{dirname as dirname18,join as join61}from"path";function genieHome3(){return process.env.GENIE_HOME??join61(homedir40(),".genie")}function servePidPath(){return join61(genieHome3(),"serve.pid")}function serveStartupStatusPath(){return join61(genieHome3(),"state",`serve-startup-${process.pid}-${Date.now()}.json`)}function readServePid(){let path3=servePidPath();if(!existsSync49(path3))return null;let raw=readFileSync30(path3,"utf-8").trim();if(raw==="")return null;let sepIdx=raw.indexOf(":");if(sepIdx<0){let pid2=Number.parseInt(raw,10);if(Number.isNaN(pid2)||pid2<=0)return null;return{pid:pid2,startTime:null}}let pidPart=raw.slice(0,sepIdx),startTimePart=raw.slice(sepIdx+1).trim(),pid=Number.parseInt(pidPart,10);if(Number.isNaN(pid)||pid<=0)return null;return{pid,startTime:startTimePart===""||startTimePart==="unknown"?null:startTimePart}}function removeServePid(){let path3=servePidPath();if(!existsSync49(path3))return;try{let current=readServePid();if(current&¤t.pid!==process.pid)return;unlinkSync11(path3)}catch{}}function isProcessAlive(pid){try{return process.kill(pid,0),!0}catch{return!1}}function stoppingLockPath(){return join61(genieHome3(),"serve.stopping.lock")}function writeStoppingLockSync(ttlMs=STOPPING_LOCK_TTL_MS){mkdirSync22(genieHome3(),{recursive:!0}),writeFileSync19(stoppingLockPath(),String(Date.now()+ttlMs),"utf-8")}function clearStoppingLock(){try{unlinkSync11(stoppingLockPath())}catch{}}function isStoppingLockActive(){let path3=stoppingLockPath();if(!existsSync49(path3))return!1;let raw;try{raw=readFileSync30(path3,"utf-8").trim()}catch{return!1}let expiresAt=Number.parseInt(raw,10);if(raw===""||Number.isNaN(expiresAt)||expiresAt<=Date.now())return clearStoppingLock(),!1;return!0}function tuiTmuxConf(){return[join61(genieHome3(),"tui-tmux.conf")].find((p)=>existsSync49(p))??"/dev/null"}function tuiTmux(subcmd){return`${tmuxBin()} -L genie-tui -f ${tuiTmuxConf()} ${subcmd}`}function isGenieTmuxRunning(){try{return execSync11(genieTmuxCmd("list-sessions"),{stdio:"ignore"}),!0}catch{return!1}}function getTuiKeybindings(sessionName=TUI_SESSION){return[`bind-key -T root Tab if-shell "[ '#{pane_index}' = '0' ]" "select-pane -t ${sessionName}:0.1" "select-pane -t ${sessionName}:0.0"`,`bind-key -T root C-1 select-pane -t ${sessionName}:0.0`,`bind-key -T root C-2 select-pane -t ${sessionName}:0.1`,`bind-key -T root C-b if-shell "[ $(tmux display-message -p '#\\{pane_width\\}' -t ${sessionName}:0.0) -gt 5 ]" "resize-pane -t ${sessionName}:0.0 -x 0" "resize-pane -t ${sessionName}:0.0 -x ${NAV_WIDTH}"`,`bind-key -T root C-t select-pane -t ${sessionName}:0.0 \\; send-keys -t ${sessionName}:0.0 C-t`,"bind-key -T root C-d detach-client",`bind-key -T root C-q select-pane -t ${sessionName}:0.0 \\; send-keys -t ${sessionName}:0.0 C-q`]}function getTuiQuitBindingArgs(sessionName=TUI_SESSION){return["bind-key","-T","root","C-q","select-pane","-t",`${sessionName}:0.0`,"\\;","send-keys","-t",`${sessionName}:0.0`,"C-q"]}function applyTuiStyle(){let cmds=[`set-option -t ${TUI_SESSION} pane-border-style 'fg=${TUI_STYLE.inactiveBorder}'`,`set-option -t ${TUI_SESSION} pane-active-border-style 'fg=${TUI_STYLE.activeBorder}'`,...process.env.GENIE_TMUX_MOUSE!=="off"?[`set-option -t ${TUI_SESSION} mouse on`]:[],`set-option -t ${TUI_SESSION} status off`,`set-option -t ${TUI_SESSION} pane-border-status off`];for(let cmd of cmds)try{execSync11(tuiTmux(cmd),{stdio:"ignore"})}catch{}}function setupTuiKeybindings(){for(let cmd of getTuiKeybindings())try{if(cmd.startsWith("bind-key -T root C-q "))spawnSync6(tmuxBin(),["-L","genie-tui","-f",tuiTmuxConf(),...getTuiQuitBindingArgs()],{stdio:"ignore"});else execSync11(tuiTmux(cmd),{stdio:"ignore"})}catch{}}function startTuiTmuxServer(){try{execSync11(tuiTmux(`has-session -t ${TUI_SESSION}`),{stdio:"ignore"});let panes2=execSync11(tuiTmux(`list-panes -t ${TUI_SESSION}:0 -F '#{pane_id}'`),{encoding:"utf-8"}).trim().split(`
|
|
2530
|
+
`,null);if(rows)activeFromPg=rows.length,executorIds=rows.map((r)=>r.id)}let pgQueue=null;if(this.queue)pgQueue=await this.safePgCall("status_queue_stats",(_sql)=>this.queue?.stats()??Promise.resolve(null),null);return{connected:this.nc!==null,natsUrl:this.natsUrl,pgAvailable:this.pgAvailable,activeSessions:activeFromPg??this.sessions.size,maxConcurrent:this.maxConcurrent,idleTimeoutMs:this.idleTimeoutMs,queueDepth:pgQueue?pgQueue.pending+pgQueue.processing:this.messageQueue.length,executorType:this.executorType,executorIds,pgQueue,sessions:Array.from(this.sessions.entries()).map(([key,entry2])=>({id:key,agentName:entry2.session.agentName,chatId:entry2.session.chatId,instanceId:entry2.instanceId,executorType:entry2.session.executorType,spawning:entry2.spawning,idleMs:now-entry2.session.lastActivityAt,bufferSize:entry2.buffer.length}))}}async probePg(){try{let sql=await withTimeout2(this.pgProvider(),PG_STARTUP_PROBE_TIMEOUT_MS,"PG provider startup");await withTimeout2(Promise.resolve(sql`SELECT 1`),PG_STARTUP_PROBE_TIMEOUT_MS,"PG SELECT 1 probe"),this.sql=sql,this.pgAvailable=!0,console.log("[omni-bridge] PG reachable \u2014 session recovery enabled")}catch(err){this.sql=null,this.pgAvailable=!1;let msg=err instanceof Error?err.message:String(err);if(isPgConnectionError(err)){console.warn(`[omni-bridge] PG unavailable \u2014 session recovery disabled (${msg})`);return}throw Error(`[omni-bridge] PG schema mismatch or setup error: ${msg}. ${"Run `bun run migrate` (or the equivalent migration command) and retry."}`)}}async recoverSessions(){if(!this.sessionStore)return;let orphanedCount=await this.safePgCall("recover_orphan_all",(sql)=>new BridgeSessionStore(sql).markAllOrphaned(),0);if(orphanedCount>0)console.log(`[omni-bridge] Startup cleanup: orphaned ${orphanedCount} stale session(s) from previous run`)}async safePgCall(op,fn,fallback,ctx){if(!this.pgAvailable||!this.sql)return fallback;let sql=this.sql;try{return await withTimeout2(fn(sql),PG_RUNTIME_QUERY_TIMEOUT_MS,`safePgCall(${op})`)}catch(err){let msg=err instanceof Error?err.message:String(err),execPart=ctx?.executorId?` executor_id=${ctx.executorId}`:"",chatPart=ctx?.chatId?` chat_id=${ctx.chatId}`:"";if(console.warn(`[omni-bridge] safePgCall(${op}) failed${execPart}${chatPart}: ${msg}`),isPgConnectionError(err))this.pgAvailable=!1,this.sql=null,console.warn("[omni-bridge] PG connection lost \u2014 switching to degraded mode");return fallback}}fillSubjectMetadata(parsed,subject){let parts=subject.split(".");if(parts.length<4)return;parsed.instanceId=parsed.instanceId||parts[2],parsed.chatId=parsed.chatId||parts[3]}isDuplicateMessage(messageId){if(!messageId)return!1;if(this.recentMessageIds.has(messageId))return!0;if(this.recentMessageIds.set(messageId,Date.now()),this.recentMessageIds.size>1000){let cutoff=Date.now()-60000;for(let[id,ts3]of this.recentMessageIds)if(ts3<cutoff)this.recentMessageIds.delete(id)}return!1}async dispatchMessage(parsed){if(this.queue){let env=parsed.env??{};await this.queue.enqueue(parsed,env);return}let key=`${parsed.agent}:${parsed.chatId}`,hasSession=this.sessions.has(key);console.log(`[omni-bridge] Routing message for ${key} (hasSession=${hasSession}, queue=${!!this.queue})`),await this.routeMessage(parsed),console.log(`[omni-bridge] routeMessage done for ${key}`)}async processSubscription(){if(!this.sub)return;for await(let msg of this.sub)try{let data=this.sc.decode(msg.data),parsed=JSON.parse(data);this.fillSubjectMetadata(parsed,msg.subject),console.log(`[omni-bridge] NATS message received: ${msg.subject} agent=${parsed.agent} chat=${parsed.chatId}`);let messageId=parsed.messageId;if(this.isDuplicateMessage(messageId)){console.log(`[omni-bridge] Dedup: skipping duplicate messageId=${messageId}`);continue}if(!parsed.chatId||!parsed.agent){console.warn("[omni-bridge] Dropping message: missing chatId or agent",msg.subject);continue}await this.dispatchMessage(parsed)}catch(err){console.error("[omni-bridge] Error processing message:",err)}}async processTurnEvents(sub){for await(let msg of sub)try{let payload=JSON.parse(this.sc.decode(msg.data)),parts=msg.subject.split("."),eventType=parts[2],instanceId=parts[3],chatId=parts.slice(4).join(".");console.log(`[omni-bridge] Turn event: ${eventType} instance=${instanceId} chat=${chatId}`);let sessionKey2=this.findSessionKey(instanceId,chatId);if(!sessionKey2&&payload.turnId){if(sessionKey2=this.findSessionKeyByTurnId(payload.turnId),sessionKey2)console.log(`[omni-bridge] Matched session via turnId fallback: ${sessionKey2}`)}if(sessionKey2)await this.routeTurnEvent(eventType,sessionKey2,payload);else console.log(`[omni-bridge] No session found for turn.${eventType} (instance=${instanceId}, chat=${chatId})`)}catch(err){console.warn("[omni-bridge] Error processing turn event:",err)}}async routeTurnEvent(eventType,sessionKey2,payload){switch(eventType){case"open":this.turnTracker.open(sessionKey2,payload.turnId,payload.messageId),this.startHeartbeat(sessionKey2,payload);break;case"done":this.turnTracker.close(sessionKey2,payload.action),this.heartbeatPublisher?.stop(sessionKey2),await this.handleTurnDone(sessionKey2);break;case"nudge":await this.handleTurnNudge(sessionKey2,payload.message);break;case"timeout":this.heartbeatPublisher?.stop(sessionKey2),await this.handleTurnTimeout(sessionKey2);break}}startHeartbeat(sessionKey2,payload){let publisher=this.heartbeatPublisher;if(!publisher)return;let entry2=this.sessions.get(sessionKey2);if(!entry2)return;let turnId=payload.turnId;if(!turnId)return;publisher.start(sessionKey2,{instanceId:entry2.instanceId,chatId:entry2.session?.chatId??"",turnId,isBusy:async()=>{let live=this.sessions.get(sessionKey2);if(!live?.session)return!1;try{return await this.executor.isBusy(live.session)}catch{return!1}}})}findSessionKeyByTurnId(turnId){for(let[key]of this.sessions)if(this.turnTracker.getTurnId(key)===turnId)return key;return}chatIdMap=new Map;findSessionKey(instanceId,chatId){let resolvedChatId=this.chatIdMap.get(chatId);for(let[key,entry2]of this.sessions){if(entry2.instanceId!==instanceId)continue;if(entry2.session?.chatId===chatId)return key;if(resolvedChatId&&entry2.session?.chatId===resolvedChatId)return key;if(entry2.spawning&&key.endsWith(`:${chatId}`))return key;if(resolvedChatId&&entry2.spawning&&key.endsWith(`:${resolvedChatId}`))return key}return}async handleTurnNudge(sessionKey2,nudgeText){let entry2=this.sessions.get(sessionKey2);if(!entry2?.session)return;try{await this.executor.injectNudge(entry2.session,nudgeText)}catch(err){console.warn(`[omni-bridge] Failed to inject nudge for ${sessionKey2}:`,err)}}async processSessionResetEvents(sub){for await(let msg of sub)try{let parts=msg.subject.split(".");if(parts.length<5){console.warn(`[omni-bridge] Malformed session-reset subject: ${msg.subject}`);continue}let instanceId=parts[3],chatId=parts.slice(4).join("."),action;try{action=JSON.parse(this.sc.decode(msg.data)).action}catch{}await this.handleSessionReset(instanceId,chatId,action)}catch(err){console.warn("[omni-bridge] Error processing session reset event:",err)}}async handleSessionReset(instanceId,chatId,action){let sessionKey2=this.findSessionKey(instanceId,chatId);if(!sessionKey2){console.log(`[omni-bridge] Session reset for cold chat ${instanceId}/${chatId} \u2014 no-op`);return}let entry2=this.sessions.get(sessionKey2);if(!entry2)return;let actionTag=action?` (action=${action})`:"";if(entry2.spawning){console.log(`[omni-bridge] Session reset for spawning ${sessionKey2}${actionTag}, marking cancelled`),entry2.cancelled=!0,entry2.buffer=[],this.turnTracker.close(sessionKey2,"reset"),this.heartbeatPublisher?.stop(sessionKey2),await this.removeSession(sessionKey2),await this.drainQueue();return}if(!entry2.session)return;console.log(`[omni-bridge] Session reset for ${sessionKey2}${actionTag}, evicting`),this.turnTracker.close(sessionKey2,"reset"),this.heartbeatPublisher?.stop(sessionKey2);try{await this.executor.shutdown(entry2.session)}catch(err){console.warn(`[omni-bridge] Error shutting down reset session ${sessionKey2}:`,err)}await this.removeSession(sessionKey2),await this.drainQueue()}async handleTurnDone(sessionKey2){if(!this.sessions.get(sessionKey2)?.session)return;console.log(`[omni-bridge] Turn done for ${sessionKey2}, session stays alive for next message`),this.resetIdleTimer(sessionKey2)}async handleTurnTimeout(sessionKey2){let entry2=this.sessions.get(sessionKey2);if(!entry2?.session)return;console.warn(`[omni-bridge] Turn timed out for ${sessionKey2}, evicting session`),this.turnTracker.close(sessionKey2,"timeout"),this.heartbeatPublisher?.stop(sessionKey2);try{await this.executor.shutdown(entry2.session)}catch(err){console.warn(`[omni-bridge] Error shutting down timed-out session ${sessionKey2}:`,err)}this.sessions.delete(sessionKey2)}async routeMessage(message){let key=`${message.agent}:${message.chatId}`,entry2=this.sessions.get(key);if(entry2){if(entry2.spawning){if(entry2.buffer.length<MAX_BUFFER_PER_CHAT)entry2.buffer.push(message);else console.warn(`[omni-bridge] Buffer full (${MAX_BUFFER_PER_CHAT}) for ${key}, dropping message from ${message.sender}`),await this.publishBufferFullReply(message);return}if(await this.executor.isAlive(entry2.session)){await this.executor.deliver(entry2.session,message),this.resetIdleTimer(key);let bsId=entry2.pgBridgeSessionId;if(bsId&&this.sessionStore)this.safePgCall("session_activity",(sql)=>new BridgeSessionStore(sql).recordActivity(bsId),void 0,{chatId:message.chatId});return}await this.removeSession(key)}await this.spawnSession(message)}async spawnSession(message){let key=`${message.agent}:${message.chatId}`,existing=this.sessions.get(key);if(existing){if(existing.buffer.length<MAX_BUFFER_PER_CHAT)existing.buffer.push(message),console.log(`[omni-bridge] Buffered message for existing session ${key} (buffer=${existing.buffer.length})`);return}if(this.sessions.size>=this.maxConcurrent){this.messageQueue.push(message),await this.publishAutoReply(message),console.log(`[omni-bridge] Max concurrent (${this.maxConcurrent}) reached, queued message for ${key}`);return}let placeholder={session:null,instanceId:message.instanceId,spawning:!0,buffer:[message],idleTimer:null};this.sessions.set(key,placeholder);try{let raw=message,payloadEnv=raw.env,spawnEnv={OMNI_API_KEY:payloadEnv?.OMNI_API_KEY??process.env.OMNI_API_KEY??"",OMNI_INSTANCE:payloadEnv?.OMNI_INSTANCE??message.instanceId,OMNI_CHAT:payloadEnv?.OMNI_CHAT??message.chatId,OMNI_MESSAGE:payloadEnv?.OMNI_MESSAGE??raw.messageId??"",OMNI_TURN_ID:payloadEnv?.OMNI_TURN_ID||"",OMNI_SENDER_NAME:payloadEnv?.OMNI_SENDER_NAME??message.sender??""};console.log(`[omni-bridge] Spawning session for ${key}...`);let session=await this.executor.spawn(message.agent,message.chatId,spawnEnv,message.content);if(placeholder.cancelled){console.log(`[omni-bridge] Spawn for ${key} completed but was cancelled by reset, shutting down`);try{await this.executor.shutdown(session)}catch(err){console.warn(`[omni-bridge] Error shutting down cancelled spawn for ${key}:`,err)}return}if(placeholder.session=session,placeholder.spawning=!1,this.sessionStore){let claudeSessionId=session.sdk?.claudeSessionId??session.tmux?.claudeSessionId,pgId=await this.safePgCall("session_create",(sql)=>new BridgeSessionStore(sql).create({instanceId:message.instanceId,chatId:message.chatId,agentName:message.agent,executorId:session.sdk?.executorId,tmuxPaneId:session.tmux?.paneId,claudeSessionId}),void 0,{chatId:message.chatId});if(pgId)placeholder.pgBridgeSessionId=pgId}for(let buffered of placeholder.buffer)await this.executor.deliver(session,buffered);placeholder.buffer=[],this.resetIdleTimer(key);let sessionTag=session.executorType==="tmux"?`(tmux pane=${session.tmux?.paneId})`:"(executor=sdk)";console.log(`[omni-bridge] Session active: ${key} ${sessionTag}`)}catch(err){console.error(`[omni-bridge] Failed to spawn session for ${key}:`,err);let lostMessages=placeholder.buffer;if(lostMessages.length>0)console.warn(`[omni-bridge] Re-queuing ${lostMessages.length} buffered message(s) from failed spawn for ${key}`),this.messageQueue.push(...lostMessages);this.sessions.delete(key)}}resetIdleTimer(key){let entry2=this.sessions.get(key);if(!entry2)return;if(entry2.idleTimer)clearTimeout(entry2.idleTimer);entry2.idleTimer=setTimeout(async()=>{console.log(`[omni-bridge] Idle timeout for ${key}, shutting down...`);try{await this.executor.shutdown(entry2.session)}catch{}await this.removeSession(key),await this.drainQueue()},this.idleTimeoutMs)}async checkIdleSessions(){let now=Date.now();for(let[key,entry2]of this.sessions){if(entry2.spawning)continue;if(!await this.executor.isAlive(entry2.session)){console.log(`[omni-bridge] Dead session detected: ${key}`),await this.removeSession(key);continue}let idleMs=now-entry2.session.lastActivityAt;if(idleMs>this.idleTimeoutMs){console.log(`[omni-bridge] Forcing idle shutdown: ${key} (idle ${Math.round(idleMs/1000)}s)`);try{await this.executor.shutdown(entry2.session)}catch{}await this.removeSession(key)}}}async removeSession(key){let entry2=this.sessions.get(key);if(entry2?.idleTimer)clearTimeout(entry2.idleTimer);this.heartbeatPublisher?.stop(key);let closeId=entry2?.pgBridgeSessionId;if(closeId&&this.sessionStore)await this.safePgCall("session_close",(sql)=>new BridgeSessionStore(sql).close(closeId),void 0);this.sessions.delete(key)}async drainQueue(){while(this.messageQueue.length>0){if(this.sessions.size>=this.maxConcurrent)break;let message=this.messageQueue.shift();if(message)await this.spawnSession(message)}}async publishBufferFullReply(message){if(!this.nc)return;let topic=`omni.reply.${message.instanceId}.${message.chatId}`,reply={content:"Fila de mensagens cheia, por favor aguarde e tente novamente.",agent:message.agent,chat_id:message.chatId,instance_id:message.instanceId,timestamp:new Date().toISOString(),auto_reply:!0};this.nc.publish(topic,this.sc.encode(JSON.stringify(reply)))}async publishAutoReply(message){if(!this.nc)return;let topic=`omni.reply.${message.instanceId}.${message.chatId}`,reply={content:"Aguarde um momento, estou atendendo outros clientes.",agent:message.agent,chat_id:message.chatId,instance_id:message.instanceId,timestamp:new Date().toISOString(),auto_reply:!0};this.nc.publish(topic,this.sc.encode(JSON.stringify(reply)))}}var import_nats4,DEFAULT_NATS_URL2="localhost:4222",DEFAULT_IDLE_TIMEOUT_MS2=900000,DEFAULT_MAX_CONCURRENT=20,MAX_BUFFER_PER_CHAT=50,IDLE_CHECK_INTERVAL_MS=30000,PG_STARTUP_PROBE_TIMEOUT_MS=5000,PG_RUNTIME_QUERY_TIMEOUT_MS=2000;var init_omni_bridge=__esm(()=>{init_bridge_status();init_executor_config();init_agent_heartbeat();init_claude_code2();init_claude_sdk2();import_nats4=__toESM(require_mod4(),1)});import{existsSync as existsSync47,mkdirSync as mkdirSync21,readFileSync as readFileSync29,readdirSync as readdirSync10,renameSync as renameSync7,statSync as statSync12,writeFileSync as writeFileSync18}from"fs";import{homedir as homedir38}from"os";import{dirname as dirname16,join as join59,resolve as resolvePath4}from"path";import{pathToFileURL}from"url";function scanDir(dirAbs){if(!existsSync47(dirAbs))return[];return readdirSync10(dirAbs).filter((name)=>name.endsWith(".ts")&&!name.startsWith("_")&&!name.endsWith(".test.ts")).map((name)=>join59(dirAbs,name)).filter((p)=>{try{return statSync12(p).isFile()}catch{return!1}})}function discoverHooks(opts){let home=homedir38(),discovered=[],teamsRoot=join59(home,".claude","teams");if(existsSync47(teamsRoot))for(let teamName of readdirSync10(teamsRoot)){if(teamName.startsWith("_"))continue;let teamHooksDir=join59(teamsRoot,teamName,"hooks");for(let path3 of scanDir(teamHooksDir))discovered.push({path:path3,scope:"team",teamName})}if(opts.repoRoot){let repoHooksDir=join59(opts.repoRoot,".genie","hooks"),repoRemoteUrl=opts.resolveRepoRemoteUrl?.(opts.repoRoot)??void 0;for(let path3 of scanDir(repoHooksDir))discovered.push({path:path3,scope:"repo",repoRemoteUrl})}let globalHooksDir=join59(home,".genie","hooks");for(let path3 of scanDir(globalHooksDir))discovered.push({path:path3,scope:"global"});return discovered}function quarantine(filePath,errorMessage){let dir=dirname16(filePath),quarantineDir=join59(dir,"_quarantine");if(!existsSync47(quarantineDir))mkdirSync21(quarantineDir,{recursive:!0});let basename11=filePath.split("/").pop()??"unknown.ts",target=join59(quarantineDir,basename11);try{renameSync7(filePath,target)}catch{}try{writeFileSync18(`${target}.error`,errorMessage,"utf-8")}catch{}return target}function validateHandler(imported){if(typeof imported!=="object"||imported===null)return"expected default export to be an object";let handler=imported;if(handler.version!=="1")return`unknown handler version: ${String(handler.version)} (expected '1')`;if(typeof handler.name!=="string"||handler.name.length===0)return"handler.name must be a non-empty string";if(typeof handler.event!=="string")return"handler.event must be a HookEventName string";if(typeof handler.priority!=="number"||Number.isNaN(handler.priority))return"handler.priority must be a finite number";if(typeof handler.fn!=="function")return"handler.fn must be a function";return null}async function loadExternalHooks(opts={}){let trustPath=opts.trustPath??defaultTrustPath(),trustFile;try{trustFile=readTrustFile(trustPath)}catch(err){let msg=err instanceof Error?err.message:String(err);return console.warn(`[hook-loader] trust file unreadable at ${trustPath}: ${msg} \u2014 refusing to load any external hooks`),[]}let importer=opts.importer??((url)=>import(url.href)),discovered=discoverHooks(opts),loaded=[],seenNames=new Map,outcomes=[];for(let source of discovered){let verify=verifyTrust(source.path,trustFile,{currentRepoRemoteUrl:source.repoRemoteUrl});if(!verify.trusted){outcomes.push({kind:"untrusted",reason:verify.reason,source});continue}let imported;try{imported=await importer(pathToFileURL(resolvePath4(source.path)))}catch(err){let msg=err instanceof Error?`${err.name}: ${err.message}`:String(err),quarantined=quarantine(source.path,msg);outcomes.push({kind:"broken",error:msg,source,quarantined});continue}let handlerCandidate=imported.default,validationError=validateHandler(handlerCandidate);if(validationError!==null){let quarantined=quarantine(source.path,validationError);outcomes.push({kind:"broken",error:validationError,source,quarantined});continue}let stamped={...handlerCandidate,source:source.scope,manifest_path:source.path},existing=seenNames.get(stamped.name);if(existing){if(console.warn(`[hook-loader] handler name collision: "${stamped.name}" registered from ${existing.path} (${existing.scope}) shadows ${source.path} (${source.scope})`),opts.strict)throw Error(`--strict-hooks: handler name collision on "${stamped.name}" between ${existing.path} and ${source.path}`);outcomes.push({kind:"shadowed",handler:stamped,source,shadowedBy:existing});continue}parseCapabilities(readFileSync29(source.path,"utf-8")),seenNames.set(stamped.name,source),loaded.push(stamped),outcomes.push({kind:"loaded",handler:stamped,source})}let builtins=getRegistry().filter((h)=>h.source==="builtin");return setRegistry([...builtins,...loaded]),outcomes}var init_loader=__esm(()=>{init_hooks();init_trust()});var exports_hook_socket={};__export(exports_hook_socket,{startHookSocket:()=>startHookSocket,defaultHookSocketPath:()=>defaultHookSocketPath});import{existsSync as existsSync48,unlinkSync as unlinkSync10}from"fs";import{mkdir as mkdir7}from"fs/promises";import{createServer,connect as netConnect}from"net";import{homedir as homedir39}from"os";import{dirname as dirname17,join as join60}from"path";function defaultHookSocketPath(){if(process.env.GENIE_HOOK_SOCK)return process.env.GENIE_HOOK_SOCK;let home=process.env.GENIE_HOME??join60(homedir39(),".genie");return join60(home,"hook.sock")}async function detectStaleAndCleanup(socketPath){if(!existsSync48(socketPath))return"clean";if(await new Promise((resolve11)=>{let probe=netConnect(socketPath),finish=(alive)=>{try{probe.destroy()}catch{}resolve11(alive)};probe.once("connect",()=>finish(!0)),probe.once("error",()=>finish(!1)),setTimeout(()=>finish(!1),200).unref()}))return"live";try{return unlinkSync10(socketPath),"stale-removed"}catch(err){throw Error(`Failed to remove stale hook socket at ${socketPath}: ${err.message}`)}}function parseFrame(acc,length){if(length===-1){if(acc.length<4)return{kind:"incomplete",length:-1};let declared=acc.readUInt32BE(0);if(declared>MAX_FRAME_BYTES2)return{kind:"error",reason:`frame length ${declared} exceeds MAX_FRAME_BYTES`};if(declared===0)return{kind:"done",body:Buffer.alloc(0)};if(acc.length>=4+declared)return{kind:"done",body:acc.subarray(4,4+declared)};return{kind:"incomplete",length:declared}}if(acc.length>=4+length)return{kind:"done",body:acc.subarray(4,4+length)};return{kind:"incomplete",length}}function readOneFrame(socket){return new Promise((resolve11,reject)=>{let acc=Buffer.alloc(0),length=-1,cleanup=()=>{socket.off("data",onData),socket.off("end",onEnd),socket.off("error",onError)},onData=(chunk)=>{acc=acc.length===0?Buffer.from(chunk):Buffer.concat([acc,Buffer.from(chunk)],acc.length+chunk.length);let result2=parseFrame(acc,length);if(result2.kind==="incomplete"){length=result2.length;return}if(cleanup(),result2.kind==="done")resolve11(result2.body);else reject(Error(result2.reason))},onEnd=()=>{cleanup(),reject(Error(`socket closed after ${acc.length} bytes (expected ${length===-1?"4+":4+length})`))},onError=(err)=>{cleanup(),reject(err)};socket.on("data",onData),socket.on("end",onEnd),socket.on("error",onError)})}function writeFrame(socket,payload){let body=Buffer.from(payload,"utf-8");if(body.length>MAX_FRAME_BYTES2){let empty=Buffer.alloc(4);socket.end(empty);return}let header=Buffer.alloc(4);header.writeUInt32BE(body.length,0),socket.end(Buffer.concat([header,body]))}async function handleConnection(socket){socket.setNoDelay(!0);try{let body=await readOneFrame(socket),stdin=body.length===0?"":body.toString("utf-8"),reply=await dispatch(stdin);writeFrame(socket,reply??"")}catch(err){try{writeFrame(socket,"")}catch{}if(process.env.GENIE_HOOK_SOCK_DEBUG==="1")console.warn(`[hook-socket] connection error: ${err.message}`)}}async function startHookSocket(options={}){if(process.env.GENIE_WIDE_EMIT===void 0)process.env.GENIE_WIDE_EMIT="1";let loaderOutcomes=[];if(options.loadExternal!==!1)loaderOutcomes=await loadExternalHooks({repoRoot:options.repoRoot,strict:options.strict}),summarizeLoaderOutcomes(loaderOutcomes);let socketPath=defaultHookSocketPath();await mkdir7(dirname17(socketPath),{recursive:!0});let state=await detectStaleAndCleanup(socketPath);if(state==="live")throw Error(`hook socket at ${socketPath} is already live \u2014 another genie serve daemon is running. Refusing to start.`);if(state==="stale-removed")console.log(`hook-socket: removed stale socket at ${socketPath}`);let liveSockets=new Set,server3=createServer((socket)=>{liveSockets.add(socket),socket.once("close",()=>liveSockets.delete(socket)),handleConnection(socket)});await new Promise((resolve11,reject)=>{let onError=(err)=>{server3.off("listening",onListening),reject(err)},onListening=()=>{server3.off("error",onError),resolve11()};server3.once("error",onError),server3.once("listening",onListening),server3.listen(socketPath)}),console.log(`hook-socket: listening at ${socketPath}`);let stopped=!1;return{path:socketPath,stop:async()=>{if(stopped)return;stopped=!0;for(let sock of liveSockets)try{sock.destroy()}catch{}if(liveSockets.clear(),await new Promise((resolve11)=>{server3.close(()=>resolve11())}),existsSync48(socketPath))try{unlinkSync10(socketPath)}catch{}},loaderOutcomes}}function summarizeLoaderOutcomes(outcomes){if(outcomes.length===0)return;let loaded=0,untrusted=0,broken=0,shadowed=0;for(let outcome of outcomes)if(outcome.kind==="loaded")loaded+=1;else if(outcome.kind==="untrusted")untrusted+=1;else if(outcome.kind==="broken")broken+=1;else shadowed+=1;let parts=[`hook-loader: ${loaded} loaded`];if(untrusted>0)parts.push(`${untrusted} untrusted`);if(broken>0)parts.push(`${broken} broken (quarantined)`);if(shadowed>0)parts.push(`${shadowed} shadowed`);console.log(parts.join(", "))}var MAX_FRAME_BYTES2=1048576;var init_hook_socket=__esm(()=>{init_hooks();init_loader()});var exports_serve={};__export(exports_serve,{writeStoppingLockSync:()=>writeStoppingLockSync,startBrainServerIfEnabled:()=>startBrainServerIfEnabled,registerServeCommands:()=>registerServeCommands,isTuiSessionReady:()=>isTuiSessionReady,isStoppingLockActive:()=>isStoppingLockActive,isServeRunning:()=>isServeRunning,getTuiQuitBindingArgs:()=>getTuiQuitBindingArgs,getTuiKeybindings:()=>getTuiKeybindings,ensureTuiSession:()=>ensureTuiSession,clearStoppingLock:()=>clearStoppingLock,autoStartServe:()=>autoStartServe});import{execSync as execSync11,spawn as spawn4,spawnSync as spawnSync6}from"child_process";import{closeSync as closeSync5,existsSync as existsSync49,mkdirSync as mkdirSync22,openSync as openSync5,readFileSync as readFileSync30,unlinkSync as unlinkSync11,writeFileSync as writeFileSync19,writeSync as writeSync2}from"fs";import{homedir as homedir40}from"os";import{dirname as dirname18,join as join61}from"path";function genieHome3(){return process.env.GENIE_HOME??join61(homedir40(),".genie")}function servePidPath(){return join61(genieHome3(),"serve.pid")}function serveStartupStatusPath(){return join61(genieHome3(),"state",`serve-startup-${process.pid}-${Date.now()}.json`)}function readServePid(){let path3=servePidPath();if(!existsSync49(path3))return null;let raw=readFileSync30(path3,"utf-8").trim();if(raw==="")return null;let sepIdx=raw.indexOf(":");if(sepIdx<0){let pid2=Number.parseInt(raw,10);if(Number.isNaN(pid2)||pid2<=0)return null;return{pid:pid2,startTime:null}}let pidPart=raw.slice(0,sepIdx),startTimePart=raw.slice(sepIdx+1).trim(),pid=Number.parseInt(pidPart,10);if(Number.isNaN(pid)||pid<=0)return null;return{pid,startTime:startTimePart===""||startTimePart==="unknown"?null:startTimePart}}function removeServePid(){let path3=servePidPath();if(!existsSync49(path3))return;try{let current=readServePid();if(current&¤t.pid!==process.pid)return;unlinkSync11(path3)}catch{}}function isProcessAlive(pid){try{return process.kill(pid,0),!0}catch{return!1}}function stoppingLockPath(){return join61(genieHome3(),"serve.stopping.lock")}function writeStoppingLockSync(ttlMs=STOPPING_LOCK_TTL_MS){mkdirSync22(genieHome3(),{recursive:!0}),writeFileSync19(stoppingLockPath(),String(Date.now()+ttlMs),"utf-8")}function clearStoppingLock(){try{unlinkSync11(stoppingLockPath())}catch{}}function isStoppingLockActive(){let path3=stoppingLockPath();if(!existsSync49(path3))return!1;let raw;try{raw=readFileSync30(path3,"utf-8").trim()}catch{return!1}let expiresAt=Number.parseInt(raw,10);if(raw===""||Number.isNaN(expiresAt)||expiresAt<=Date.now())return clearStoppingLock(),!1;return!0}function tuiTmuxConf(){return[join61(genieHome3(),"tui-tmux.conf")].find((p)=>existsSync49(p))??"/dev/null"}function tuiTmux(subcmd){return`${tmuxBin()} -L genie-tui -f ${tuiTmuxConf()} ${subcmd}`}function isGenieTmuxRunning(){try{return execSync11(genieTmuxCmd("list-sessions"),{stdio:"ignore"}),!0}catch{return!1}}function getTuiKeybindings(sessionName=TUI_SESSION){return[`bind-key -T root Tab if-shell "[ '#{pane_index}' = '0' ]" "select-pane -t ${sessionName}:0.1" "select-pane -t ${sessionName}:0.0"`,`bind-key -T root C-1 select-pane -t ${sessionName}:0.0`,`bind-key -T root C-2 select-pane -t ${sessionName}:0.1`,`bind-key -T root C-b if-shell "[ $(tmux display-message -p '#\\{pane_width\\}' -t ${sessionName}:0.0) -gt 5 ]" "resize-pane -t ${sessionName}:0.0 -x 0" "resize-pane -t ${sessionName}:0.0 -x ${NAV_WIDTH}"`,`bind-key -T root C-t select-pane -t ${sessionName}:0.0 \\; send-keys -t ${sessionName}:0.0 C-t`,"bind-key -T root C-d detach-client",`bind-key -T root C-q select-pane -t ${sessionName}:0.0 \\; send-keys -t ${sessionName}:0.0 C-q`]}function getTuiQuitBindingArgs(sessionName=TUI_SESSION){return["bind-key","-T","root","C-q","select-pane","-t",`${sessionName}:0.0`,"\\;","send-keys","-t",`${sessionName}:0.0`,"C-q"]}function applyTuiStyle(){let cmds=[`set-option -t ${TUI_SESSION} pane-border-style 'fg=${TUI_STYLE.inactiveBorder}'`,`set-option -t ${TUI_SESSION} pane-active-border-style 'fg=${TUI_STYLE.activeBorder}'`,...process.env.GENIE_TMUX_MOUSE!=="off"?[`set-option -t ${TUI_SESSION} mouse on`]:[],`set-option -t ${TUI_SESSION} status off`,`set-option -t ${TUI_SESSION} pane-border-status off`];for(let cmd of cmds)try{execSync11(tuiTmux(cmd),{stdio:"ignore"})}catch{}}function setupTuiKeybindings(){for(let cmd of getTuiKeybindings())try{if(cmd.startsWith("bind-key -T root C-q "))spawnSync6(tmuxBin(),["-L","genie-tui","-f",tuiTmuxConf(),...getTuiQuitBindingArgs()],{stdio:"ignore"});else execSync11(tuiTmux(cmd),{stdio:"ignore"})}catch{}}function startTuiTmuxServer(){try{execSync11(tuiTmux(`has-session -t ${TUI_SESSION}`),{stdio:"ignore"});let panes2=execSync11(tuiTmux(`list-panes -t ${TUI_SESSION}:0 -F '#{pane_id}'`),{encoding:"utf-8"}).trim().split(`
|
|
2531
2531
|
`);if(panes2.length>=2){try{execSync11(tuiTmux(`respawn-pane -k -t ${panes2[1]} 'cat'`),{stdio:"ignore"})}catch{}return{leftPane:panes2[0],rightPane:panes2[1]}}let cols2=Number.parseInt(execSync11(tuiTmux(`display-message -t ${TUI_SESSION}:0 -p '#{window_width}'`),{encoding:"utf-8"}).trim(),10)||120;execSync11(tuiTmux(`split-window -h -t ${TUI_SESSION}:0 -l ${cols2-NAV_WIDTH-1}`),{stdio:"ignore"});let refreshed=execSync11(tuiTmux(`list-panes -t ${TUI_SESSION}:0 -F '#{pane_id}'`),{encoding:"utf-8"}).trim().split(`
|
|
2532
2532
|
`);applyTuiStyle(),setupTuiKeybindings();try{execSync11(tuiTmux(`select-pane -t ${refreshed[0]}`),{stdio:"ignore"})}catch{}return{leftPane:refreshed[0],rightPane:refreshed[1]||refreshed[0]}}catch{}let cols=120;execSync11(tuiTmux(`new-session -d -s ${TUI_SESSION} -x ${cols} -y ${40}`),{stdio:"ignore"}),execSync11(tuiTmux(`split-window -h -t ${TUI_SESSION}:0 -l ${cols-NAV_WIDTH-1}`),{stdio:"ignore"});let panes=execSync11(tuiTmux(`list-panes -t ${TUI_SESSION}:0 -F '#{pane_id}'`),{encoding:"utf-8"}).trim().split(`
|
|
2533
2533
|
`);applyTuiStyle(),setupTuiKeybindings();try{execSync11(tuiTmux(`select-pane -t ${panes[0]}`),{stdio:"ignore"})}catch{}return{leftPane:panes[0],rightPane:panes[1]||panes[0]}}function sendTuiLaunchScript(leftPane,rightPane,workspaceRoot){let home=genieHome3(),bunPath=process.execPath||"bun",genieBin=process.argv[1]||"genie",scriptPath=join61(home,"tui-launch.sh"),logsDir=join61(home,"logs"),crashLog=join61(logsDir,"tui-crash.log"),envVars=["GENIE_TUI_PANE=left",`GENIE_TUI_RIGHT=${rightPane}`];if(workspaceRoot)envVars.push(`GENIE_TUI_WORKSPACE=${workspaceRoot}`);let content=["#!/bin/sh",`mkdir -p '${logsDir}'`,`exec 2>> '${crashLog}'`,`printf -- '--- tui-launch %s pid=%s ---\\n' "$(date -u +%Y-%m-%dT%H:%M:%SZ)" "$$" >&2`,`export ${envVars.join(`
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@automagik/genie",
|
|
3
|
-
"version": "4.260501.
|
|
3
|
+
"version": "4.260501.5",
|
|
4
4
|
"description": "Collaborative terminal toolkit for human + AI workflows. NOTE: the npm distribution is being soft-deprecated — the canonical install is `curl -fsSL https://get.automagik.dev/genie | bash` (cosign + SLSA verified). See https://automagik.dev/genie/security/distribution-sovereignty",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"bin": {
|
|
@@ -60,7 +60,6 @@
|
|
|
60
60
|
"ignore": "7.0.5",
|
|
61
61
|
"js-yaml": "4.1.1",
|
|
62
62
|
"nats": "2.29.3",
|
|
63
|
-
"pgserve": "^2.0.8",
|
|
64
63
|
"postgres": "3.4.9",
|
|
65
64
|
"react": "19.2.5",
|
|
66
65
|
"react-dom": "19.2.5",
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "genie",
|
|
3
|
-
"version": "4.260501.
|
|
3
|
+
"version": "4.260501.5",
|
|
4
4
|
"description": "Human-AI partnership for Claude Code. Share a terminal, orchestrate workers, evolve together. Brainstorm ideas, turn them into wishes, execute with /work, validate with /review, and ship as one team.",
|
|
5
5
|
"author": {
|
|
6
6
|
"name": "Namastex Labs"
|