@z_ptah/agent 0.0.2 → 0.0.3
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
2
|
import"dotenv/config";function j(e){let o=process.env[e];if(!o)throw new Error(`Missing required environment variable: ${e}`);return o}var i={serverUrl:j("PTAH_SERVER_URL"),token:j("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"};import E from"ws";import{spawn as ee}from"child_process";function H(e){let{commandId:o,sessionId:t,projectDir:n,message:r,claudeSessionId:s}=e,p=["-p",r,"--output-format","stream-json","--allowedTools","Read,Edit,Write,Glob,Grep","--directory",n];s&&p.push("--resume",s),console.log(`Starting Claude for session ${t} in ${n}`);let g=ee("claude",p,{cwd:n,env:{...process.env},stdio:["pipe","pipe","pipe"]}),y="";g.stdout.on("data",c=>{let d=c.toString();y+=d,a({type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:d,isComplete:!1}})}),g.stderr.on("data",c=>{let d=c.toString(),v={type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:`[stderr] ${d}`,isComplete:!1}};a(v)}),g.on("close",c=>{console.log(`Claude exited with code ${c} for session ${t}`),_(t);let d;try{let Y=y.split(`
|
|
3
|
-
`).filter(Boolean);for(let Z of Y){let U=JSON.parse(Z);typeof U.session_id=="string"&&(d=U.session_id)}}catch{}let v={type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:c===0?"":`Process exited with code ${c}`,isComplete:!0,claudeSessionId:d??s}};a(v)}),g.on("error",c=>{console.error(`Claude spawn error for session ${t}:`,c),_(t);let d={type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:`Error spawning Claude: ${c.message}`,isComplete:!0}};a(d)})}var
|
|
3
|
+
`).filter(Boolean);for(let Z of Y){let U=JSON.parse(Z);typeof U.session_id=="string"&&(d=U.session_id)}}catch{}let v={type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:c===0?"":`Process exited with code ${c}`,isComplete:!0,claudeSessionId:d??s}};a(v)}),g.on("error",c=>{console.error(`Claude spawn error for session ${t}:`,c),_(t);let d={type:"agent:claude-output",payload:{commandId:o,sessionId:t,output:`Error spawning Claude: ${c.message}`,isComplete:!0}};a(d)})}var M=new Map;function B(e){if(M.has(e.sessionId)){console.warn(`Session ${e.sessionId} already has an active Claude process`);return}M.set(e.sessionId,{commandId:e.commandId}),H(e)}function _(e){M.delete(e)}import{spawn as oe}from"child_process";async function G(e){return new Promise((o,t)=>{console.log(`Building Docker in ${e}...`);let n=oe("docker",["compose","up","-d","--build"],{cwd:e,stdio:"inherit"});n.on("close",r=>{r===0?(console.log("Docker build successful"),o()):t(new Error(`Docker build failed with code ${r}`))}),n.on("error",r=>{t(new Error(`Docker build error: ${r.message}`))})})}import{writeFile as te}from"fs/promises";import{spawn as ne}from"child_process";function S(e,o){return new Promise((t,n)=>{let r=ne(e,o,{stdio:"inherit"});r.on("close",s=>{s===0?t():n(new Error(`${e} failed with code ${s}`))}),r.on("error",n)})}async function W(e,o,t,n){let r=/^[a-z0-9][a-z0-9-]*$/;if(!r.test(e)||!r.test(o))throw new Error("Invalid project or org code for Nginx config");if(t<1024||t>65535||n<1024||n>65535)throw new Error("Invalid port number");let s=`${e}.${o}.mybptah.com`,p=`
|
|
4
4
|
server {
|
|
5
5
|
listen 80;
|
|
6
6
|
server_name ${s};
|
|
@@ -25,7 +25,7 @@ server {
|
|
|
25
25
|
proxy_cache_bypass $http_upgrade;
|
|
26
26
|
}
|
|
27
27
|
}
|
|
28
|
-
`,g=`/etc/nginx/sites-available/${s}`,y=`/etc/nginx/sites-enabled/${s}`;await te(g,p);try{await S("ln",["-sf",g,y])}catch{}return await S("nginx",["-t"]),await S("nginx",["-s","reload"]),console.log(`Nginx configured for ${s}`),s}async function J(e){try{await S("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 w(e,o,t){let n={type:"agent:deploy-status",payload:{commandId:e.commandId,deploymentId:e.deploymentId,status:o,...t}};a(n)}async function V(e){try{w(e,"building"),await G(e.projectDir),w(e,"deploying");let n=await W(e.projectCode,e.orgCode,3001,3002);await J(n).catch(()=>{});let r=`https://${n}`;w(e,"live",{demoUrl:r}),console.log(`Deploy complete: ${r}`)}catch(o){let t=o instanceof Error?o.message:"Unknown deploy error";console.error("Deploy failed:",t),w(e,"failed",{errorMessage:t})}}import{spawn as x}from"child_process";var re=5*60*1e3,u=null,m=null,
|
|
29
|
-
`)}function C(){u&&(console.log("Cancelling Claude login..."),
|
|
28
|
+
`,g=`/etc/nginx/sites-available/${s}`,y=`/etc/nginx/sites-enabled/${s}`;await te(g,p);try{await S("ln",["-sf",g,y])}catch{}return await S("nginx",["-t"]),await S("nginx",["-s","reload"]),console.log(`Nginx configured for ${s}`),s}async function J(e){try{await S("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 w(e,o,t){let n={type:"agent:deploy-status",payload:{commandId:e.commandId,deploymentId:e.deploymentId,status:o,...t}};a(n)}async function V(e){try{w(e,"building"),await G(e.projectDir),w(e,"deploying");let n=await W(e.projectCode,e.orgCode,3001,3002);await J(n).catch(()=>{});let r=`https://${n}`;w(e,"live",{demoUrl:r}),console.log(`Deploy complete: ${r}`)}catch(o){let t=o instanceof Error?o.message:"Unknown deploy error";console.error("Deploy failed:",t),w(e,"failed",{errorMessage:t})}}import{spawn as x}from"child_process";var re=5*60*1e3,u=null,m=null,P=!1;function T(){m&&(clearTimeout(m),m=null),u&&(u.kill(),u=null)}function I(e){if(P)return;P=!0,a({type:"agent:claude-login-result",payload:e})}function q(){if(u){console.warn("Login already in progress, ignoring");return}console.log("Starting Claude auth login..."),P=!1;let e=x("script",["-qfc","claude auth login","/dev/null"],{env:{...process.env,TERM:"dumb"},stdio:["pipe","pipe","pipe"]});u=e;let o=!1;m=setTimeout(()=>{console.warn("Claude login timed out"),T(),I({success:!1,error:"Login timed out after 5 minutes"})},re),e.stdout.on("data",t=>{let n=t.toString();if(console.log(`[claude login stdout] ${n.trim()}`),!o){let r=n.match(/(https?:\/\/[^\s]+)/);if(r){o=!0;let s={type:"agent:claude-login-url",payload:{url:r[1]}};a(s)}}}),e.stderr.on("data",t=>{let n=t.toString();if(console.log(`[claude login stderr] ${n.trim()}`),!o){let r=n.match(/(https?:\/\/[^\s]+)/);if(r){o=!0;let s={type:"agent:claude-login-url",payload:{url:r[1]}};a(s)}}}),e.on("close",async t=>{console.log(`Claude auth login exited with code ${t}`),u=null,m&&(clearTimeout(m),m=null);let n=await A();n.loggedIn?(await se(["superpowers@claude-plugins-official"]),I({success:!0,email:n.email,subscriptionType:n.subscriptionType})):I({success:!1,error:"Login failed or was cancelled"})}),e.on("error",t=>{console.error("Claude auth login spawn error:",t),T(),I({success:!1,error:`Spawn error: ${t.message}`})})}function F(e){if(!u){console.warn("No login process running, ignoring input");return}if(e.startsWith("http://localhost")){if(!/^http:\/\/localhost(:\d+)?(\/[^\s]*)?$/.test(e)){console.warn("Invalid localhost URL, rejecting:",e);return}console.log("Curling localhost redirect URL:",e);let t=x("curl",["-s",e]);t.on("close",n=>{console.log(`Curl exited with code ${n}`)}),t.on("error",n=>{console.error("Curl error:",n.message)});return}console.log("Writing code to login process stdin"),u.stdin?.write(e+`
|
|
29
|
+
`)}function C(){u&&(console.log("Cancelling Claude login..."),T())}async function A(){return new Promise(e=>{let o=x("claude",["auth","status"],{env:{...process.env},stdio:["pipe","pipe","pipe"]}),t="";o.stdout.on("data",n=>{t+=n.toString()}),o.stderr.on("data",n=>{t+=n.toString()}),o.on("close",()=>{try{let n=JSON.parse(t.trim());e({loggedIn:n.loggedIn===!0,email:n.email,subscriptionType:n.subscriptionType})}catch{e({loggedIn:!1})}}),o.on("error",()=>{e({loggedIn:!1})})})}async function se(e){for(let o of e)console.log(`Installing plugin: ${o}`),await new Promise(t=>{let n=x("claude",["plugin","install",o,"--scope","user"],{env:{...process.env},stdio:["pipe","pipe","pipe"]});n.on("close",r=>{r===0?console.log(`Plugin ${o} installed successfully`):console.warn(`Plugin ${o} install exited with code ${r}`),t()}),n.on("error",r=>{console.warn(`Plugin ${o} install error: ${r.message}`),t()})})}function X(e){R({loggedIn:e.loggedIn,email:e.email}),a({type:"agent:claude-status",payload:e})}function z(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:n,message:r,claudeSessionId:s,projectDir:p}=e.payload;console.log(`Starting Claude session ${t} for project ${n}`),B({commandId:o,sessionId:t,projectDir:p,message:r,claudeSessionId:s});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:n,projectCode:r,orgCode:s,projectDir:p}=e.payload;console.log(`Starting deployment for project ${r}`),V({commandId:o,deploymentId:t,projectId:n,projectCode:r,orgCode:s,projectDir:p});break}case"server:claude-login-start":{console.log("Received claude-login-start command"),q();break}case"server:claude-login-input":{let{value:o}=e.payload;console.log("Received claude-login-input"),F(o);break}case"server:claude-login-cancel":{console.log("Received claude-login-cancel"),C();break}default:console.warn("Unknown message type:",e.type)}}var l=null,$=null,f=null,h=!1,b=0,K={loggedIn:!1};function R(e){K=e}var ae=6e4;function a(e){l?.readyState===E.OPEN&&l.send(JSON.stringify(e))}function ie(){k(),$=setInterval(()=>{let e={type:"agent:heartbeat",payload:{cpuUsage:0,memoryUsage:Math.round(process.memoryUsage().heapUsed/1024/1024),activeProjects:0,claude:K}};a(e)},i.heartbeatIntervalMs)}function k(){$&&(clearInterval($),$=null)}function le(){if(f)return;let e=Math.min(i.reconnectIntervalMs*Math.pow(2,b),ae);b++,console.log(`Reconnecting in ${e/1e3}s (attempt ${b})...`),f=setTimeout(()=>{f=null,N()},e)}function N(){h||l?.readyState===E.OPEN||(h=!0,console.log(`Connecting to ${i.serverUrl}...`),l=new E(i.serverUrl),l.on("open",()=>{h=!1,console.log("Connected to Ptah Server");let e={type:"agent:auth",payload:{token:i.token}};a(e)}),l.on("message",e=>{try{let o=JSON.parse(e.toString());z(o),o.type==="server:auth-result"&&o.payload?.success&&(console.log(`Authenticated as VPS ${o.payload.vpsId}`),b=0,ie(),A().then(t=>{R({loggedIn:t.loggedIn,email:t.email}),X(t)}))}catch(o){console.error("Failed to parse message:",o)}}),l.on("close",()=>{h=!1,k(),C(),console.log("Disconnected from Ptah Server"),le()}),l.on("error",e=>{h=!1,console.error("WebSocket error:",e.message)}))}function D(){f&&(clearTimeout(f),f=null),k(),l&&(l.close(),l=null)}import ce from"fastify";var O=ce({logger:!1});O.get("/health",async()=>({status:"ok",timestamp:new Date().toISOString(),uptime:process.uptime()}));async function Q(){await O.listen({port:i.healthPort,host:"0.0.0.0"}),console.log(`Agent health server on http://0.0.0.0:${i.healthPort}`)}async function L(){await O.close()}async function de(){console.log("Starting Ptah Agent..."),console.log(` Server: ${i.serverUrl}`),console.log(` Health port: ${i.healthPort}`),console.log(` Projects dir: ${i.projectsDir}`),await Q(),N()}process.on("SIGINT",async()=>{console.log(`
|
|
30
30
|
Shutting down...`),D(),await L(),process.exit(0)});process.on("SIGTERM",async()=>{console.log(`
|
|
31
31
|
Shutting down...`),D(),await L(),process.exit(0)});de().catch(e=>{console.error("Failed to start agent:",e),process.exit(1)});
|