@z_ptah/agent 0.0.21 → 0.0.23
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 +9 -8
- package/package.json +1 -1
package/dist/index.js
CHANGED
|
@@ -1,8 +1,8 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
|
-
import"dotenv/config";import{execSync as
|
|
3
|
-
`,"utf-8")}function
|
|
4
|
-
`))if(n.trim())try{
|
|
5
|
-
`);S=
|
|
2
|
+
import"dotenv/config";import{execSync as ce}from"child_process";function H(e){let t=process.env[e];if(!t)throw new Error(`Missing required environment variable: ${e}`);return t}function le(){if(process.env.CLAUDE_PATH)return process.env.CLAUDE_PATH;try{return ce("which claude",{encoding:"utf-8"}).trim()}catch{return"/usr/bin/claude"}}var i={serverUrl:H("PTAH_SERVER_URL"),token:H("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:le()};import E from"ws";import{spawn as Ae}from"child_process";import{mkdirSync as De}from"fs";import{randomUUID as M}from"crypto";import{randomUUID as h}from"crypto";function L(e,t,o){let s=e.trim();if(!s)return[];let r;try{r=JSON.parse(s)}catch{return[]}let n=[],c=new Date().toISOString();if(r.type==="content_block_delta"){let a=r.delta;return a?.type==="text_delta"&&a.text&&n.push({kind:"text_delta",id:h(),sessionId:t,commandId:o,timestamp:c,role:"assistant",content:a.text}),a?.type==="thinking_delta"&&a.thinking&&n.push({kind:"thinking",id:h(),sessionId:t,commandId:o,timestamp:c,role:"assistant",content:a.thinking}),n}if(r.type==="content_block_start"){let a=r.content_block;return a?.type==="tool_use"&&n.push({kind:"tool_use",id:h(),sessionId:t,commandId:o,timestamp:c,role:"assistant",toolName:a.name||"tool",toolId:a.id||void 0,toolInput:{}}),n}if(r.type==="content_block_stop"||r.type==="message_start"||r.type==="message_delta"||r.type==="message_stop")return n;if(r.type==="assistant"&&r.message){let a=r.message;if(Array.isArray(a.content))for(let l of a.content){let f=de(l.type);if(!f)continue;let m={kind:f,id:h(),sessionId:t,commandId:o,timestamp:c,role:"assistant"};switch(f){case"text":m.content=l.text||"";break;case"tool_use":m.toolName=l.name||"tool",m.toolId=l.id||void 0,m.toolInput=l.input??{};break;case"thinking":m.content=l.thinking||"";break}n.push(m)}}return r.type==="tool_result"&&n.push({kind:"tool_result",id:h(),sessionId:t,commandId:o,timestamp:c,role:"assistant",toolId:r.tool_use_id||void 0,content:typeof r.content=="string"?r.content:JSON.stringify(r.content??""),isError:r.is_error||!1}),r.type==="result"&&n.push({kind:"complete",id:h(),sessionId:t,commandId:o,timestamp:c,role:"assistant",content:r.result||"",metadata:{sessionId:r.session_id,cost:r.cost_usd??r.cost,totalCost:r.total_cost_usd,duration:r.duration_ms,numTurns:r.num_turns}}),n}function de(e){switch(e){case"text":return"text";case"tool_use":return"tool_use";case"thinking":return"thinking";default:return null}}import{mkdirSync as pe,appendFileSync as ue,existsSync as ge,readFileSync as me}from"fs";import{join as fe,dirname as ye}from"path";function F(e,t){return fe(i.projectsDir,e,"chat-history",`${t}.jsonl`)}function x(e,t){let o=F(e,t.sessionId);pe(ye(o),{recursive:!0}),ue(o,JSON.stringify(t)+`
|
|
3
|
+
`,"utf-8")}function J(e,t){let o=F(e,t);if(!ge(o))return[];let s=me(o,"utf-8"),r=[];for(let n of s.split(`
|
|
4
|
+
`))if(n.trim())try{r.push(JSON.parse(n))}catch{}return r}import{readdirSync as ke,statSync as Ie,readFileSync as _e,writeFileSync as be,existsSync as $e}from"fs";import{join as C}from"path";import{readFileSync as he,existsSync as ve}from"fs";import{basename as Se}from"path";var xe={".xlsx":"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",".xls":"application/vnd.ms-excel",".pdf":"application/pdf",".docx":"application/vnd.openxmlformats-officedocument.wordprocessingml.document",".doc":"application/msword",".csv":"text/csv",".json":"application/json",".md":"text/markdown",".txt":"text/plain",".zip":"application/zip"};function we(e){let t=e.substring(e.lastIndexOf(".")).toLowerCase();return xe[t]||"application/octet-stream"}async function B(e,t){if(!ve(t))return console.warn(`[uploader] File not found: ${t}`),!1;let o=Se(t),s=we(o),r=he(t),n=i.serverUrl.replace(/^wss:/,"https:").replace(/^ws:/,"http:").replace(/\/ws\/agent$/,""),c=new FormData;c.append("projectId",e),c.append("file",new Blob([r],{type:s}),o);try{let a=await fetch(`${n}/agent/documents/upload`,{method:"POST",headers:{Authorization:`Bearer ${i.token}`},body:c});if(!a.ok){let l=await a.text();return console.error(`[uploader] Upload failed (${a.status}): ${l}`),!1}return console.log(`[uploader] Uploaded ${o} for project ${e}`),!0}catch(a){return console.error("[uploader] Upload error:",a),!1}}var Ce=new Set([".xlsx",".xls",".csv",".pdf",".docx",".doc"]),q=".ptah-uploaded.json";function Me(e){let t=C(e,q);if(!$e(t))return{};try{return JSON.parse(_e(t,"utf-8"))}catch{return{}}}function Pe(e,t){be(C(e,q),JSON.stringify(t,null,2))}async function G(e,t){let o=Me(t),s=!1,r;try{r=ke(t)}catch{return}for(let n of r){let c=n.substring(n.lastIndexOf(".")).toLowerCase();if(!Ce.has(c))continue;let a=C(t,n);try{let l=Ie(a);if(!l.isFile())continue;let f=o[n];if(f&&f>=l.mtimeMs)continue;console.log(`[auto-upload] Uploading ${n}...`),await B(e,a)&&(o[n]=l.mtimeMs,s=!0)}catch(l){console.error(`[auto-upload] Failed to process ${n}:`,l)}}s&&Pe(t,o)}function V(e){let{commandId:t,sessionId:o,projectId:s,projectDir:r,message:n,claudeSessionId:c,systemPrompt:a}=e,f=["-p",n,"--output-format","stream-json","--verbose","--allowedTools","Bash,Read,Edit,Write,Glob,Grep","--append-system-prompt",a||"Your text responses are rendered as markdown in a web UI. Use markdown formatting (tables, code blocks, lists, headings) for readability. You may freely create/edit files as needed \u2014 but always include a summary of what you did in your text response so the user can see it in the UI."];c&&f.push("--resume",c),De(r,{recursive:!0}),console.log(`Starting Claude for session ${o} in ${r}`);let m={kind:"text",id:M(),sessionId:o,commandId:t,timestamp:new Date().toISOString(),role:"user",content:n};x(s,m),d({type:"agent:claude-message",payload:m});let k=Ae(i.claudePath,f,{cwd:r,env:{...process.env},stdio:["pipe","pipe","pipe"]}),S="",ie;function z(p){if(!p.trim())return;try{let y=JSON.parse(p);y.session_id&&(ie=y.session_id)}catch{return}let u=L(p,o,t);for(let y of u)y.kind!=="text_delta"&&x(s,y),d({type:"agent:claude-message",payload:y})}k.stdout.on("data",p=>{S+=p.toString();let u=S.split(`
|
|
5
|
+
`);S=u.pop()||"";for(let y of u)z(y)}),k.stderr.on("data",p=>{}),k.on("close",p=>{if(S.trim()&&z(S),console.log(`Claude exited with code ${p} for session ${o}`),P(o),p===0&&G(s,r).catch(u=>console.error("[auto-upload] Error:",u)),p!==0){let u={kind:"error",id:M(),sessionId:o,commandId:t,timestamp:new Date().toISOString(),role:"assistant",content:`Process exited with code ${p}`};x(s,u),d({type:"agent:claude-message",payload:u})}}),k.on("error",p=>{console.error(`Claude spawn error for session ${o}:`,p),P(o);let u={kind:"error",id:M(),sessionId:o,commandId:t,timestamp:new Date().toISOString(),role:"assistant",content:`Error spawning Claude: ${p.message}`};x(s,u),d({type:"agent:claude-message",payload:u})})}var A=new Map;function K(e){if(A.has(e.sessionId)){console.warn(`Session ${e.sessionId} already has an active Claude process`);return}A.set(e.sessionId,{commandId:e.commandId}),V(e)}function P(e){A.delete(e)}import{spawn as Ne}from"child_process";async function W(e){return new Promise((t,o)=>{console.log(`Building Docker in ${e}...`);let s=Ne("docker",["compose","up","-d","--build"],{cwd:e,stdio:"inherit"});s.on("close",r=>{r===0?(console.log("Docker build successful"),t()):o(new Error(`Docker build failed with code ${r}`))}),s.on("error",r=>{o(new Error(`Docker build error: ${r.message}`))})})}import{writeFile as Ee}from"fs/promises";import{spawn as Re}from"child_process";function I(e,t){return new Promise((o,s)=>{let r=Re(e,t,{stdio:"inherit"});r.on("close",n=>{n===0?o():s(new Error(`${e} failed with code ${n}`))}),r.on("error",s)})}async function X(e,t,o,s){let r=/^[a-z0-9][a-z0-9-]*$/;if(!r.test(e)||!r.test(t))throw new Error("Invalid project or org code for Nginx config");if(o<1024||o>65535||s<1024||s>65535)throw new Error("Invalid port number");let n=`${e}.${t}.mybptah.com`,c=`
|
|
6
6
|
server {
|
|
7
7
|
listen 80;
|
|
8
8
|
server_name ${n};
|
|
@@ -18,7 +18,7 @@ server {
|
|
|
18
18
|
}
|
|
19
19
|
|
|
20
20
|
location / {
|
|
21
|
-
proxy_pass http://127.0.0.1:${
|
|
21
|
+
proxy_pass http://127.0.0.1:${s};
|
|
22
22
|
proxy_http_version 1.1;
|
|
23
23
|
proxy_set_header Upgrade $http_upgrade;
|
|
24
24
|
proxy_set_header Connection 'upgrade';
|
|
@@ -27,6 +27,7 @@ server {
|
|
|
27
27
|
proxy_cache_bypass $http_upgrade;
|
|
28
28
|
}
|
|
29
29
|
}
|
|
30
|
-
`,a=`/etc/nginx/sites-available/${n}`,
|
|
31
|
-
|
|
32
|
-
Shutting down...`),
|
|
30
|
+
`,a=`/etc/nginx/sites-available/${n}`,l=`/etc/nginx/sites-enabled/${n}`;await Ee(a,c);try{await I("ln",["-sf",a,l])}catch{}return await I("nginx",["-t"]),await I("nginx",["-s","reload"]),console.log(`Nginx configured for ${n}`),n}async function Y(e){try{await I("certbot",["--nginx","-d",e,"--non-interactive","--agree-tos","--email","ssl@mybptah.com"]),console.log(`SSL configured for ${e}`)}catch(t){console.error(`SSL setup failed for ${e}:`,t)}}function _(e,t,o){let s={type:"agent:deploy-status",payload:{commandId:e.commandId,deploymentId:e.deploymentId,status:t,...o}};d(s)}async function Q(e){try{_(e,"building"),await W(e.projectDir),_(e,"deploying");let s=await X(e.projectCode,e.orgCode,3001,3002);await Y(s).catch(()=>{});let r=`https://${s}`;_(e,"live",{demoUrl:r}),console.log(`Deploy complete: ${r}`)}catch(t){let o=t instanceof Error?t.message:"Unknown deploy error";console.error("Deploy failed:",o),_(e,"failed",{errorMessage:o})}}import{execSync as D}from"child_process";function Z(){try{return D(`docker ps -a --format '{"id":"{{.ID}}","name":"{{.Names}}","image":"{{.Image}}","status":"{{.Status}}","state":"{{.State}}","ports":"{{.Ports}}","createdAt":"{{.CreatedAt}}"}'`,{encoding:"utf-8",timeout:1e4}).trim().split(`
|
|
31
|
+
`).filter(t=>t.trim()).map(t=>{let o=JSON.parse(t),s=o.name.match(/^ptah-([a-f0-9-]+)/);return{...o,projectId:s?.[1]||void 0}})}catch(e){return console.error("[docker] Failed to list containers:",e),[]}}function ee(e){try{return D(`docker stop ${e}`,{encoding:"utf-8",timeout:3e4}),{success:!0}}catch(t){return{success:!1,error:t.message}}}function te(e){try{return D(`docker rm -f ${e}`,{encoding:"utf-8",timeout:3e4}),{success:!0}}catch(t){return{success:!1,error:t.message}}}function oe(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:t,sessionId:o,projectId:s,message:r,claudeSessionId:n,projectDir:c,systemPrompt:a}=e.payload;console.log(`Starting Claude session ${o} for project ${s}`),K({commandId:t,sessionId:o,projectId:s,projectDir:c,message:r,claudeSessionId:n,systemPrompt:a});break}case"server:claude-stop":{console.log(`Stop requested for session ${e.payload.sessionId}`);break}case"server:chat-history-request":{let{sessionId:t,projectId:o}=e.payload;console.log(`Chat history request for session ${t}`);let s=J(o,t);d({type:"agent:chat-history",payload:{sessionId:t,messages:s}});break}case"server:deploy-start":{let{commandId:t,deploymentId:o,projectId:s,projectCode:r,orgCode:n,projectDir:c}=e.payload;console.log(`Starting deployment for project ${r}`),Q({commandId:t,deploymentId:o,projectId:s,projectCode:r,orgCode:n,projectDir:c});break}case"server:docker-list":{let{requestId:t}=e.payload;console.log(`Docker list request: ${t}`);let o=Z();d({type:"agent:docker-list",payload:{requestId:t,containers:o}});break}case"server:docker-stop":{let{requestId:t,containerId:o}=e.payload;console.log(`Docker stop: ${o}`);let s=ee(o);d({type:"agent:docker-action",payload:{requestId:t,...s}});break}case"server:docker-remove":{let{requestId:t,containerId:o}=e.payload;console.log(`Docker remove: ${o}`);let s=te(o);d({type:"agent:docker-action",payload:{requestId:t,...s}});break}default:console.warn("Unknown message type:",e.type)}}import{spawn as Te}from"child_process";async function re(){return new Promise(e=>{let t=Te(i.claudePath,["auth","status"],{env:{...process.env},stdio:["pipe","pipe","pipe"]}),o="";t.stdout.on("data",s=>{o+=s.toString()}),t.stderr.on("data",s=>{o+=s.toString()}),t.on("close",()=>{try{let s=JSON.parse(o.trim());e({loggedIn:s.loggedIn===!0,email:s.email,subscriptionType:s.subscriptionType})}catch{e({loggedIn:!1})}}),t.on("error",()=>{e({loggedIn:!1})})})}function se(e){N({loggedIn:e.loggedIn,email:e.email}),d({type:"agent:claude-status",payload:e})}var g=null,b=null,v=null,w=!1,$=0,ne={loggedIn:!1};function N(e){ne=e}var Oe=6e4;function d(e){g?.readyState===E.OPEN&&g.send(JSON.stringify(e))}function Ue(){R(),b=setInterval(()=>{let e={type:"agent:heartbeat",payload:{cpuUsage:0,memoryUsage:Math.round(process.memoryUsage().heapUsed/1024/1024),activeProjects:0,claude:ne}};d(e)},i.heartbeatIntervalMs)}function R(){b&&(clearInterval(b),b=null)}function je(){if(v)return;let e=Math.min(i.reconnectIntervalMs*Math.pow(2,$),Oe);$++,console.log(`Reconnecting in ${e/1e3}s (attempt ${$})...`),v=setTimeout(()=>{v=null,T()},e)}function T(){w||g?.readyState===E.OPEN||(w=!0,console.log(`Connecting to ${i.serverUrl}...`),g=new E(i.serverUrl),g.on("open",()=>{w=!1,console.log("Connected to Ptah Server");let e={type:"agent:auth",payload:{token:i.token}};d(e)}),g.on("message",e=>{try{let t=JSON.parse(e.toString());oe(t),t.type==="server:auth-result"&&t.payload?.success&&(console.log(`Authenticated as VPS ${t.payload.vpsId}`),$=0,Ue(),re().then(o=>{N({loggedIn:o.loggedIn,email:o.email}),se(o)}))}catch(t){console.error("Failed to parse message:",t)}}),g.on("close",()=>{w=!1,R(),console.log("Disconnected from Ptah Server"),je()}),g.on("error",e=>{w=!1,console.error("WebSocket error:",e.message)}))}function O(){v&&(clearTimeout(v),v=null),R(),g&&(g.close(),g=null)}import ze from"fastify";var U=ze({logger:!1});U.get("/health",async()=>({status:"ok",timestamp:new Date().toISOString(),uptime:process.uptime()}));async function ae(){await U.listen({port:i.healthPort,host:"0.0.0.0"}),console.log(`Agent health server on http://0.0.0.0:${i.healthPort}`)}async function j(){await U.close()}async function He(){console.log("Starting Ptah Agent..."),console.log(` Server: ${i.serverUrl}`),console.log(` Health port: ${i.healthPort}`),console.log(` Projects dir: ${i.projectsDir}`),await ae(),T()}process.on("SIGINT",async()=>{console.log(`
|
|
32
|
+
Shutting down...`),O(),await j(),process.exit(0)});process.on("SIGTERM",async()=>{console.log(`
|
|
33
|
+
Shutting down...`),O(),await j(),process.exit(0)});He().catch(e=>{console.error("Failed to start agent:",e),process.exit(1)});
|