@z_ptah/agent 0.0.9 → 0.0.10
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/index.js +3 -3
- package/package.json +1 -1
package/dist/index.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
|
-
import"dotenv/config";import{execSync as F}from"child_process";function T(e){let o=process.env[e];if(!o)throw new Error(`Missing required environment variable: ${e}`);return o}function W(){if(process.env.CLAUDE_PATH)return process.env.CLAUDE_PATH;try{return F("which claude",{encoding:"utf-8"}).trim()}catch{return"/usr/bin/claude"}}var s={serverUrl:T("PTAH_SERVER_URL"),token:T("PTAH_AGENT_TOKEN"),healthPort:Number(process.env.AGENT_HEALTH_PORT||9090),heartbeatIntervalMs:Number(process.env.HEARTBEAT_INTERVAL_MS||3e4),reconnectIntervalMs:Number(process.env.RECONNECT_INTERVAL_MS||5e3),projectsDir:process.env.PROJECTS_DIR||"/projects",claudePath:W()};import b from"ws";import{spawn as q}from"child_process";import{mkdirSync as X}from"fs";function D(e){let{commandId:o,sessionId:t,projectDir:r,message:n,claudeSessionId:a}=e,
|
|
3
|
-
`).filter(Boolean);for(let V of J){let E=JSON.parse(V);typeof E.session_id=="string"&&(
|
|
2
|
+
import"dotenv/config";import{execSync as F}from"child_process";function T(e){let o=process.env[e];if(!o)throw new Error(`Missing required environment variable: ${e}`);return o}function W(){if(process.env.CLAUDE_PATH)return process.env.CLAUDE_PATH;try{return F("which claude",{encoding:"utf-8"}).trim()}catch{return"/usr/bin/claude"}}var s={serverUrl:T("PTAH_SERVER_URL"),token:T("PTAH_AGENT_TOKEN"),healthPort:Number(process.env.AGENT_HEALTH_PORT||9090),heartbeatIntervalMs:Number(process.env.HEARTBEAT_INTERVAL_MS||3e4),reconnectIntervalMs:Number(process.env.RECONNECT_INTERVAL_MS||5e3),projectsDir:process.env.PROJECTS_DIR||"/projects",claudePath:W()};import b from"ws";import{spawn as q}from"child_process";import{mkdirSync as X}from"fs";function D(e){let{commandId:o,sessionId:t,projectDir:r,message:n,claudeSessionId:a}=e,d=["-p",n,"--output-format","stream-json","--allowedTools","Read,Edit,Write,Glob,Grep"];a&&d.push("--resume",a),X(r,{recursive:!0}),console.log(`Starting Claude for session ${t} in ${r}`);let u=q(s.claudePath,d,{cwd:r,env:{...process.env},stdio:["pipe","pipe","pipe"]}),f="";u.stdout.on("data",c=>{let p=c.toString();f+=p,i({type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:p,isComplete:!1}})}),u.stderr.on("data",c=>{let p=c.toString(),h={type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:`[stderr] ${p}`,isComplete:!1}};i(h)}),u.on("close",c=>{console.log(`Claude exited with code ${c} for session ${t}`),w(t);let p;try{let J=f.split(`
|
|
3
|
+
`).filter(Boolean);for(let V of J){let E=JSON.parse(V);typeof E.session_id=="string"&&(p=E.session_id)}}catch{}let h={type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:c===0?"":`Process exited with code ${c}`,isComplete:!0,claudeSessionId:p??a}};i(h)}),u.on("error",c=>{console.error(`Claude spawn error for session ${t}:`,c),w(t);let p={type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:`Error spawning Claude: ${c.message}`,isComplete:!0}};i(p)})}var x=new Map;function R(e){if(x.has(e.sessionId)){console.warn(`Session ${e.sessionId} already has an active Claude process`);return}x.set(e.sessionId,{commandId:e.commandId}),D(e)}function w(e){x.delete(e)}import{spawn as z}from"child_process";async function k(e){return new Promise((o,t)=>{console.log(`Building Docker in ${e}...`);let r=z("docker",["compose","up","-d","--build"],{cwd:e,stdio:"inherit"});r.on("close",n=>{n===0?(console.log("Docker build successful"),o()):t(new Error(`Docker build failed with code ${n}`))}),r.on("error",n=>{t(new Error(`Docker build error: ${n.message}`))})})}import{writeFile as K}from"fs/promises";import{spawn as Q}from"child_process";function y(e,o){return new Promise((t,r)=>{let n=Q(e,o,{stdio:"inherit"});n.on("close",a=>{a===0?t():r(new Error(`${e} failed with code ${a}`))}),n.on("error",r)})}async function N(e,o,t,r){let n=/^[a-z0-9][a-z0-9-]*$/;if(!n.test(e)||!n.test(o))throw new Error("Invalid project or org code for Nginx config");if(t<1024||t>65535||r<1024||r>65535)throw new Error("Invalid port number");let a=`${e}.${o}.mybptah.com`,d=`
|
|
4
4
|
server {
|
|
5
5
|
listen 80;
|
|
6
6
|
server_name ${a};
|
|
@@ -25,6 +25,6 @@ server {
|
|
|
25
25
|
proxy_cache_bypass $http_upgrade;
|
|
26
26
|
}
|
|
27
27
|
}
|
|
28
|
-
`,u=`/etc/nginx/sites-available/${a}`,f=`/etc/nginx/sites-enabled/${a}`;await K(u,
|
|
28
|
+
`,u=`/etc/nginx/sites-available/${a}`,f=`/etc/nginx/sites-enabled/${a}`;await K(u,d);try{await y("ln",["-sf",u,f])}catch{}return await y("nginx",["-t"]),await y("nginx",["-s","reload"]),console.log(`Nginx configured for ${a}`),a}async function O(e){try{await y("certbot",["--nginx","-d",e,"--non-interactive","--agree-tos","--email","ssl@mybptah.com"]),console.log(`SSL configured for ${e}`)}catch(o){console.error(`SSL setup failed for ${e}:`,o)}}function v(e,o,t){let r={type:"agent:deploy-status",payload:{commandId:e.commandId,deploymentId:e.deploymentId,status:o,...t}};i(r)}async function j(e){try{v(e,"building"),await k(e.projectDir),v(e,"deploying");let r=await N(e.projectCode,e.orgCode,3001,3002);await O(r).catch(()=>{});let n=`https://${r}`;v(e,"live",{demoUrl:n}),console.log(`Deploy complete: ${n}`)}catch(o){let t=o instanceof Error?o.message:"Unknown deploy error";console.error("Deploy failed:",t),v(e,"failed",{errorMessage:t})}}function H(e){switch(e.type){case"server:auth-result":e.payload.success||(console.error("Authentication failed:",e.payload.error),process.exit(1));break;case"server:command":console.log(`Received command: ${e.payload.action}`,e.payload);break;case"server:doc-sync":console.log(`Doc sync: ${e.payload.action} ${e.payload.fileName}`);break;case"server:claude-start":{let{commandId:o,sessionId:t,projectId:r,message:n,claudeSessionId:a,projectDir:d}=e.payload;console.log(`Starting Claude session ${t} for project ${r}`),R({commandId:o,sessionId:t,projectDir:d,message:n,claudeSessionId:a});break}case"server:claude-stop":{console.log(`Stop requested for session ${e.payload.sessionId}`);break}case"server:deploy-start":{let{commandId:o,deploymentId:t,projectId:r,projectCode:n,orgCode:a,projectDir:d}=e.payload;console.log(`Starting deployment for project ${n}`),j({commandId:o,deploymentId:t,projectId:r,projectCode:n,orgCode:a,projectDir:d});break}default:console.warn("Unknown message type:",e.type)}}import{spawn as Y}from"child_process";async function U(){return new Promise(e=>{let o=Y(s.claudePath,["auth","status"],{env:{...process.env},stdio:["pipe","pipe","pipe"]}),t="";o.stdout.on("data",r=>{t+=r.toString()}),o.stderr.on("data",r=>{t+=r.toString()}),o.on("close",()=>{try{let r=JSON.parse(t.trim());e({loggedIn:r.loggedIn===!0,email:r.email,subscriptionType:r.subscriptionType})}catch{e({loggedIn:!1})}}),o.on("error",()=>{e({loggedIn:!1})})})}function L(e){$({loggedIn:e.loggedIn,email:e.email}),i({type:"agent:claude-status",payload:e})}var l=null,S=null,g=null,m=!1,I=0,B={loggedIn:!1};function $(e){B=e}var Z=6e4;function i(e){l?.readyState===b.OPEN&&l.send(JSON.stringify(e))}function ee(){_(),S=setInterval(()=>{let e={type:"agent:heartbeat",payload:{cpuUsage:0,memoryUsage:Math.round(process.memoryUsage().heapUsed/1024/1024),activeProjects:0,claude:B}};i(e)},s.heartbeatIntervalMs)}function _(){S&&(clearInterval(S),S=null)}function oe(){if(g)return;let e=Math.min(s.reconnectIntervalMs*Math.pow(2,I),Z);I++,console.log(`Reconnecting in ${e/1e3}s (attempt ${I})...`),g=setTimeout(()=>{g=null,C()},e)}function C(){m||l?.readyState===b.OPEN||(m=!0,console.log(`Connecting to ${s.serverUrl}...`),l=new b(s.serverUrl),l.on("open",()=>{m=!1,console.log("Connected to Ptah Server");let e={type:"agent:auth",payload:{token:s.token}};i(e)}),l.on("message",e=>{try{let o=JSON.parse(e.toString());H(o),o.type==="server:auth-result"&&o.payload?.success&&(console.log(`Authenticated as VPS ${o.payload.vpsId}`),I=0,ee(),U().then(t=>{$({loggedIn:t.loggedIn,email:t.email}),L(t)}))}catch(o){console.error("Failed to parse message:",o)}}),l.on("close",()=>{m=!1,_(),console.log("Disconnected from Ptah Server"),oe()}),l.on("error",e=>{m=!1,console.error("WebSocket error:",e.message)}))}function P(){g&&(clearTimeout(g),g=null),_(),l&&(l.close(),l=null)}import te from"fastify";var A=te({logger:!1});A.get("/health",async()=>({status:"ok",timestamp:new Date().toISOString(),uptime:process.uptime()}));async function G(){await A.listen({port:s.healthPort,host:"0.0.0.0"}),console.log(`Agent health server on http://0.0.0.0:${s.healthPort}`)}async function M(){await A.close()}async function re(){console.log("Starting Ptah Agent..."),console.log(` Server: ${s.serverUrl}`),console.log(` Health port: ${s.healthPort}`),console.log(` Projects dir: ${s.projectsDir}`),await G(),C()}process.on("SIGINT",async()=>{console.log(`
|
|
29
29
|
Shutting down...`),P(),await M(),process.exit(0)});process.on("SIGTERM",async()=>{console.log(`
|
|
30
30
|
Shutting down...`),P(),await M(),process.exit(0)});re().catch(e=>{console.error("Failed to start agent:",e),process.exit(1)});
|