@automagik/genie 4.260504.1 → 4.260504.2

package/dist/dispatch-client.js

@@ -1,4 +1,4 @@
 // @bun
 var __require=import.meta.require;import{appendFileSync,mkdirSync,statSync}from"fs";import{connect}from"net";import{homedir}from"os";import{dirname,join}from"path";function defaultSocketPath(){if(process.env.GENIE_HOOK_SOCK)return process.env.GENIE_HOOK_SOCK;let home=process.env.GENIE_HOME??join(homedir(),".genie");return join(home,"hook.sock")}function fallbackLogPath(){let home=process.env.GENIE_HOME??join(homedir(),".genie");return join(home,"hook-fallback.log")}var MAX_FRAME_BYTES=1048576,FALLBACK_LOG_MAX_BYTES=104857600,DEFAULT_TIMEOUT_MS=5000;function readStdinSync(){let chunks=[],total=0,fd=0,buf=Buffer.alloc(65536);while(!0){let n;try{n=__require("fs").readSync(fd,buf,0,buf.length,null)}catch{break}if(n===0)break;if(chunks.push(Buffer.from(buf.subarray(0,n))),total+=n,total>MAX_FRAME_BYTES)break}return Buffer.concat(chunks,Math.min(total,MAX_FRAME_BYTES))}function summarizePayload(payload){try{let obj=JSON.parse(payload.toString("utf-8")),event=typeof obj.hook_event_name==="string"?obj.hook_event_name:null,tool=typeof obj.tool_name==="string"?obj.tool_name:null,command=null,ti=obj.tool_input;if(ti&&typeof ti.command==="string")command=ti.command.split(`
 `)[0].slice(0,256);return{event,tool,command}}catch{return{event:null,tool:null,command:null}}}function appendFallback(record){let path=fallbackLogPath();try{mkdirSync(dirname(path),{recursive:!0});let writeFresh=!1;try{if(statSync(path).size>=FALLBACK_LOG_MAX_BYTES)writeFresh=!0}catch{}let line=`${JSON.stringify(record)}
-`;if(writeFresh)__require("fs").writeFileSync(path,line);else appendFileSync(path,line)}catch{}}function buildFrame(payload){let header=Buffer.alloc(4);return header.writeUInt32BE(payload.length,0),Buffer.concat([header,payload])}function parseStep(acc,length){if(length===-1){if(acc.length<4)return{kind:"incomplete",length:-1};let declared=acc.readUInt32BE(0);if(declared>MAX_FRAME_BYTES)return{kind:"error",reason:`oversized frame ${declared}`};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}}async function roundtrip(socketPath,payload,timeoutMs){return new Promise((resolve)=>{let settled=!1,finish=(result)=>{if(settled)return;settled=!0,resolve(result)},timer=setTimeout(()=>{try{sock.destroy()}catch{}finish({reply:null,reason:`timeout after ${timeoutMs}ms`})},timeoutMs);timer.unref();let sock=connect(socketPath);sock.once("connect",()=>{sock.write(buildFrame(payload))});let acc=Buffer.alloc(0),length=-1;sock.on("data",(chunk)=>{acc=acc.length===0?Buffer.from(chunk):Buffer.concat([acc,Buffer.from(chunk)],acc.length+chunk.length);let step=parseStep(acc,length);if(step.kind==="incomplete"){length=step.length;return}clearTimeout(timer);try{sock.destroy()}catch{}if(step.kind==="done")finish({reply:step.body.toString("utf-8")});else finish({reply:null,reason:step.reason})}),sock.once("error",(err)=>{clearTimeout(timer);let code=err.code??"unknown";finish({reply:null,reason:`connect error: ${code}`})}),sock.once("end",()=>{if(!settled)clearTimeout(timer),finish({reply:null,reason:"socket closed before reply"})})})}async function runDispatchClient(){let payload=readStdinSync();if(payload.length===0)return 0;let socketPath=process.env.GENIE_HOOK_SOCK??defaultSocketPath(),timeoutMs=Number.parseInt(process.env.GENIE_HOOK_TIMEOUT_MS??"",10),effectiveTimeout=Number.isFinite(timeoutMs)&&timeoutMs>0?timeoutMs:DEFAULT_TIMEOUT_MS,result=await roundtrip(socketPath,payload,effectiveTimeout);if(result.reply!==null){if(result.reply.length>0)process.stdout.write(result.reply);return 0}let summary=summarizePayload(payload);return appendFallback({ts:new Date().toISOString(),event:summary.event,tool:summary.tool,command:summary.command,agent_id:process.env.GENIE_AGENT_NAME??null,reason:result.reason??"unknown"}),0}if(import.meta.main)runDispatchClient().then((code)=>{process.exit(code)}).catch(()=>{process.exit(0)});export{runDispatchClient};
+`;if(writeFresh)__require("fs").writeFileSync(path,line);else appendFileSync(path,line)}catch{}}function buildFrame(payload){let header=Buffer.alloc(4);return header.writeUInt32BE(payload.length,0),Buffer.concat([header,payload])}function parseStep(acc,length){if(length===-1){if(acc.length<4)return{kind:"incomplete",length:-1};let declared=acc.readUInt32BE(0);if(declared>MAX_FRAME_BYTES)return{kind:"error",reason:`oversized frame ${declared}`};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}}async function roundtrip(socketPath,payload,timeoutMs){return new Promise((resolve)=>{let settled=!1,finish=(result)=>{if(settled)return;settled=!0,resolve(result)},timer=setTimeout(()=>{try{sock.destroy()}catch{}finish({reply:null,reason:`timeout after ${timeoutMs}ms`})},timeoutMs);timer.unref();let sock=connect(socketPath);sock.once("connect",()=>{sock.write(buildFrame(payload))});let acc=Buffer.alloc(0),length=-1;sock.on("data",(chunk)=>{acc=acc.length===0?Buffer.from(chunk):Buffer.concat([acc,Buffer.from(chunk)],acc.length+chunk.length);let step=parseStep(acc,length);if(step.kind==="incomplete"){length=step.length;return}clearTimeout(timer);try{sock.destroy()}catch{}if(step.kind==="done")finish({reply:step.body.toString("utf-8")});else finish({reply:null,reason:step.reason})}),sock.once("error",(err)=>{clearTimeout(timer);let code=err.code??"unknown";finish({reply:null,reason:`connect error: ${code}`})}),sock.once("end",()=>{if(!settled)clearTimeout(timer),finish({reply:null,reason:"socket closed before reply"})})})}async function runDispatchClient(){let payload=readStdinSync();if(payload.length===0)return 0;let socketPath=process.env.GENIE_HOOK_SOCK??defaultSocketPath(),timeoutMs=Number.parseInt(process.env.GENIE_HOOK_TIMEOUT_MS??"",10),effectiveTimeout=Number.isFinite(timeoutMs)&&timeoutMs>0?timeoutMs:DEFAULT_TIMEOUT_MS,result=await roundtrip(socketPath,payload,effectiveTimeout);if(result.reply!==null){if(result.reply.length>0)process.stdout.write(result.reply);return 0}let summary=summarizePayload(payload),agentId=process.env.GENIE_AGENT_ID??process.env.GENIE_AGENT_NAME??null;return appendFallback({ts:new Date().toISOString(),event:summary.event,tool:summary.tool,command:summary.command,agent_id:agentId,reason:result.reason??"unknown"}),0}if(import.meta.main)runDispatchClient().then((code)=>{process.exit(code)}).catch(()=>{process.exit(0)});export{runDispatchClient};