@workermill/agent 0.8.13 → 0.9.0
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/cli.js +17 -17
- package/dist/index.js +12 -12
- package/package.json +1 -1
package/dist/cli.js
CHANGED
|
@@ -1,24 +1,24 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
|
-
var
|
|
3
|
-
`)[0],n=Ue(o,"..","..","bin","bash.exe");if(We(n))return n}catch{}let e=[Ue(process.env.ProgramFiles||"C:\\Program Files","Git","bin","bash.exe"),"C:\\Program Files\\Git\\bin\\bash.exe","C:\\Program Files (x86)\\Git\\bin\\bash.exe",Ue(bo(),"scoop","apps","git","current","bin","bash.exe")];for(let o of e)if(We(o))return o;return null}function Pt(){if(Ie)try{return de("winget install Anthropic.ClaudeCode --accept-package-agreements --accept-source-agreements",{stdio:"inherit",timeout:18e4}),!0}catch{return!1}else{if(process.platform==="darwin")try{return de("brew install --cask claude-code",{stdio:"inherit",timeout:18e4}),!0}catch{}try{return de("curl -fsSL https://claude.ai/install.sh | bash",{stdio:"inherit",timeout:12e4}),!0}catch{return!1}}}async function so(){console.log(),console.log(u.bold.cyan(" WorkerMill Remote Agent Setup")),console.log(u.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(),console.log(" Run AI workers locally with your Claude Max subscription."),console.log(" Workers execute on your machine, logs stream to the cloud dashboard."),console.log();let e=Math.round(Ct()/(1024*1024*1024)),o=_t().length;console.log(u.dim(" System")),console.log(` ${u.dim("RAM:")} ${e} GB ${u.dim("CPUs:")} ${o}`),e<8?(console.log(),console.log(u.red(" \u2717 Insufficient RAM")),console.log(u.yellow(" WorkerMill requires at least 8 GB of RAM (16 GB recommended).")),console.log(u.yellow(` Your system has ${e} GB.`)),console.log(),process.exit(1)):e<16?(console.log(u.yellow(` \u26A0 ${e} GB RAM is below the recommended 16 GB.`)),console.log(u.yellow(" Workers may run slowly or be killed by the OS under memory pressure."))):console.log(u.green(" \u2713 System meets requirements")),console.log();let n=_e("Checking Docker...").start();if(io("docker")){let l=ro("docker");try{let A=de("docker info --format {{.OSType}}",{encoding:"utf-8",timeout:15e3}).trim();Ie&&A==="windows"&&(n.fail("Docker is running in Windows containers mode"),console.log(),console.log(u.yellow(" WorkerMill workers require Linux containers.")),console.log(u.yellow(" Right-click the Docker Desktop icon in the system tray \u2192")),console.log(u.yellow(" 'Switch to Linux containers...'")),console.log(),console.log(" Then re-run: workermill-agent"),process.exit(1))}catch{n.fail("Docker is installed but the daemon is not running"),console.log(),console.log(u.yellow(" Start Docker Desktop, wait for it to fully initialize,")),console.log(u.yellow(" then re-run: workermill-agent")),process.exit(1)}n.succeed(`Docker ${u.dim(l?`(${l})`:"")}`)}else n.fail("Docker is not installed"),console.log(),console.log(u.yellow(" Docker Desktop is required but must be installed manually:")),console.log(` ${u.cyan("https://docs.docker.com/get-docker/")}`),console.log(),console.log(" Install Docker, start it, then re-run: workermill-agent"),process.exit(1);if(Ie){let l=_e("Checking Git for Windows...").start(),A=ko();A?(l.succeed(`Git for Windows ${u.dim(`(${A})`)}`),process.env.CLAUDE_CODE_GIT_BASH_PATH=A):(l.fail("Git for Windows is not installed"),console.log(),console.log(u.yellow(" Claude Code on Windows requires Git for Windows (Git Bash):")),console.log(` ${u.cyan("https://git-scm.com/downloads/win")}`),console.log(),console.log(" Install Git for Windows, then re-run: workermill-agent"),process.exit(1))}let t=_e("Checking Claude CLI...").start(),r=se();if(r){let l=ro(r);t.succeed(`Claude CLI ${u.dim(l?`(${l})`:"")}`)}else if(t.warn("Claude CLI not found \u2014 installing..."),console.log(),Pt()&&(r=se()),r){let A=ro(r);console.log(),console.log(u.green(` \u2713 Claude CLI installed ${u.dim(A?`(${A})`:"")}`))}else console.log(),console.log(u.red(" Claude CLI was installed but could not be found.")),console.log(),Ie?(console.log(u.yellow(" Winget updated your PATH but this shell doesn't have it yet.")),console.log(u.yellow(" Close this terminal, open a new one, and re-run: workermill-agent"))):(console.log(" Try manually:"),console.log(u.cyan(" curl -fsSL https://claude.ai/install.sh | bash")),console.log(" Then re-run: workermill-agent")),process.exit(1);let i=_e("Checking Claude authentication...").start(),s=Ue(bo(),".claude",".credentials.json");if(We(s))i.succeed("Claude authenticated");else{i.warn("Not authenticated \u2014 launching Claude..."),console.log(),console.log(u.dim(" Claude will open and prompt you to authenticate.")),console.log(u.dim(" Sign in with your Claude Max account, then exit Claude (Ctrl+C).")),console.log();let l={...process.env};if(Ie){let A=ko();A&&(l.CLAUDE_CODE_GIT_BASH_PATH=A)}Ao(r,[],{stdio:"inherit",timeout:3e5,env:l}),We(s)?(console.log(),console.log(u.green(" \u2713 Claude authenticated"))):(console.log(),console.log(u.red(" Authentication failed or was cancelled.")),console.log(" Try manually: open a new terminal and run 'claude'"),console.log(" Then re-run: workermill-agent"),process.exit(1))}console.log(u.green(" \u2713")+` Node.js ${u.dim(`(${process.version})`)}`),console.log(),console.log(u.bold("Configuration")),console.log();let{apiUrl:a}=await no.prompt([{type:"input",name:"apiUrl",message:"WorkerMill API URL:",default:"https://workermill.com",validate:l=>l.startsWith("http://")||l.startsWith("https://")?!0:"Must be a valid URL"}]),{apiKey:c}=await no.prompt([{type:"password",name:"apiKey",message:"API Key (from Settings > Integrations):",mask:"*",validate:l=>l.length>0?!0:"API key is required"}]),m=_e("Validating API key...").start(),$="github",g="",h="";try{let l=await Rt.get(`${a.replace(/\/$/,"")}/api/agent/config`,{headers:{"x-api-key":c},timeout:15e3});$=l.data.scmProvider||"github",g=l.data.workerImageUrl||"",h=l.data.ecrRegistry||"",m.succeed(`Connected! SCM provider: ${$}`)}catch(l){l.response?.status===401?m.fail("Invalid API key. Check Settings > Integrations on the dashboard."):m.fail("Failed to connect to WorkerMill API. Check the URL and try again."),process.exit(1)}console.log(u.dim(" SCM tokens are managed via Settings > Integrations on the dashboard.")),console.log();let{agentId:P}=await no.prompt([{type:"input",name:"agentId",message:"Agent name:",default:`agent-${Tt()}`}]);console.log();let d=g||"workermill-worker:local",w=h.length>0,T=_e("Checking AWS CLI...").start();if(io("aws"))try{de("aws sts get-caller-identity",{stdio:"pipe",timeout:15e3}),T.succeed("AWS CLI configured")}catch{T.warn("AWS CLI found but credentials not configured"),console.log(),console.log(u.yellow(" Configure AWS credentials for private ECR image access:")),console.log(u.cyan(" aws configure")),console.log(u.dim(" Contact your WorkerMill admin for AWS access key / secret key.")),console.log(u.dim(" Setup will continue \u2014 you can configure AWS later before starting."))}else T.warn("AWS CLI not found"),console.log(),console.log(u.yellow(" AWS CLI is required for pulling worker images from private ECR.")),console.log(u.cyan(" Install: https://docs.aws.amazon.com/cli/latest/userguide/getting-started-install.html")),console.log(u.dim(" Setup will continue \u2014 install AWS CLI before starting the agent."));let x=!1;if(w){console.log(),console.log(u.dim(" Authenticating with private ECR..."));let l=h.match(/\.ecr\.([a-z0-9-]+)\.amazonaws\.com/),A=l?l[1]:"us-east-1";try{de(`aws ecr get-login-password --region ${A} | docker login --username AWS --password-stdin ${h}`,{stdio:"pipe",timeout:3e4}),console.log(u.green(" \u2713 ECR authenticated")),x=!0}catch{console.log(u.yellow(" \u26A0 ECR authentication failed (AWS credentials may not be configured yet)")),console.log(u.dim(" Worker image pull will be skipped \u2014 authenticate later with:")),console.log(u.dim(` aws ecr get-login-password --region ${A} | docker login --username AWS --password-stdin ${h}`))}}if(x||!w){console.log(),console.log(u.dim(` Pulling worker image: ${d}`)),console.log(u.dim(" This may take a few minutes on first run (~1.1 GB)...")),console.log();let l=Ao("docker",["pull",d],{stdio:"inherit",timeout:6e5});l.status===0?(console.log(),console.log(u.green(" \u2713 Worker image pulled"))):(console.log(),console.log(u.red(" \u2717 Failed to pull worker image.")),l.error&&console.log(u.yellow(` Error: ${l.error.message}`)),console.log(u.dim(" Setup will continue \u2014 you can pull the image later before starting.")))}let W={apiUrl:a.replace(/\/$/,""),apiKey:c,agentId:P,maxWorkers:1,pollIntervalMs:5e3,heartbeatIntervalMs:3e4,tokens:{github:"",bitbucket:"",gitlab:""},workerImage:d,setupCompletedAt:new Date().toISOString()};oo(W),console.log(),console.log(u.green.bold(" Setup complete!")),console.log(),console.log(` Config saved to: ${u.dim(oe())}`),console.log(),console.log(` Start the agent with: ${u.cyan("workermill-agent start")}`),console.log()}import L from"chalk";import{totalmem as In}from"os";import{spawn as An}from"child_process";import{writeFileSync as lt,existsSync as kn,unlinkSync as bn,openSync as En,createWriteStream as Sn}from"fs";import{createRequire as xt}from"module";var vt=xt(import.meta.url),Mt=vt("../package.json"),Y=Mt.version;te();import G from"chalk";import Nt from"axios";var ao=null;function Ke(e,o){ao=Nt.create({baseURL:e,headers:{"Content-Type":"application/json","x-api-key":o},timeout:3e4})}var M=new Proxy({},{get(e,o){if(!ao)throw new Error("API client not initialized. Call initApi() first.");return ao[o]}});import S from"chalk";te();import p from"chalk";import{spawn as tn,execSync as go}from"child_process";import{spawn as Kt}from"child_process";import $e from"chalk";import Ge from"axios";var Ot=16384,Lt=6e5;async function Eo(e,o,n,t,r){let i=r?.maxTokens??Ot,s=r?.temperature??.7,a=r?.timeoutMs??Lt;switch(e){case"anthropic":return Ft(o,n,t,i,s,a);case"openai":return Dt(o,n,t,i,s,a);case"google":return Ut(o,n,t,i,s,a);case"ollama":return Wt(o,n,t,i,s,a);default:throw new Error(`Unsupported AI provider: ${e}`)}}async function Ft(e,o,n,t,r,i){let a=(await Ge.post("https://api.anthropic.com/v1/messages",{model:e,max_tokens:t,temperature:r,messages:[{role:"user",content:o}]},{headers:{"x-api-key":n,"anthropic-version":"2023-06-01","Content-Type":"application/json"},timeout:i})).data?.content;if(Array.isArray(a))return a.filter(c=>c.type==="text").map(c=>c.text).join("");throw new Error("Unexpected Anthropic API response format")}async function Dt(e,o,n,t,r,i){let a=(await Ge.post("https://api.openai.com/v1/chat/completions",{model:e,max_tokens:t,temperature:r,messages:[{role:"user",content:o}]},{headers:{Authorization:`Bearer ${n}`,"Content-Type":"application/json"},timeout:i})).data?.choices?.[0]?.message;if(a?.content)return a.content;throw new Error("Unexpected OpenAI API response format")}async function Ut(e,o,n,t,r,i){let c=(await Ge.post(`https://generativelanguage.googleapis.com/v1beta/models/${e}:generateContent?key=${n}`,{contents:[{parts:[{text:o}]}],generationConfig:{maxOutputTokens:t,temperature:r}},{headers:{"Content-Type":"application/json"},timeout:i})).data?.candidates?.[0]?.content?.parts?.[0]?.text;if(c)return c;throw new Error("Unexpected Google AI API response format")}async function Wt(e,o,n,t,r,i){let s=n||"http://localhost:11434",c=(await Ge.post(`${s}/api/generate`,{model:e,prompt:o,stream:!1,options:{temperature:r}},{headers:{"Content-Type":"application/json"},timeout:i})).data?.response;if(c)return c;throw new Error("Unexpected Ollama API response format")}var he=null;async function To(){if(he)return he;try{let{data:e}=await M.get("/api/agent/critic-prompt");return he={promptTemplate:e.promptTemplate,approvalThreshold:e.approvalThreshold??85,maxTargetFiles:e.maxTargetFiles??15},ne=he.approvalThreshold,Ne=he.maxTargetFiles,he}catch{return console.warn("Failed to fetch critic config from API"),null}}var Ne=15,ne=85;function Co(e){let o=e.indexOf("```json");if(o!==-1){let t=o+7,r=e.indexOf("{",t);if(r!==-1){let i=So(e,r);if(i)return JSON.parse(i)}}let n=e.indexOf('"stories"');if(n!==-1){let r=e.substring(0,n).lastIndexOf("{");if(r!==-1){let i=So(e,r);if(i)return JSON.parse(i)}}throw new Error("Could not find JSON execution plan in output")}function So(e,o){let n=0,t=!1,r=!1;for(let i=o;i<e.length;i++){let s=e[i];if(r){r=!1;continue}if(s==="\\"){t&&(r=!0);continue}if(s==='"'){t=!t;continue}if(!t){if(s==="{")n++;else if(s==="}"&&(n--,n===0))return e.substring(o,i+1)}}return null}function _o(e){let o=0,n=[];for(let t of e.stories)if(!t.targetFiles||!Array.isArray(t.targetFiles))t.targetFiles=[];else if(t.targetFiles.length>Ne){let r=t.targetFiles.slice(Ne);n.push(`${t.id}: ${t.targetFiles.length} files \u2192 ${Ne} (dropped: ${r.join(", ")})`),t.targetFiles=t.targetFiles.slice(0,Ne),o++}return{truncatedCount:o,details:n}}function Ro(e,o){if(e.stories.length<=o)return{droppedCount:0,details:[]};let n=e.stories.length-o,r=e.stories.slice(o).map(s=>`${s.id}: "${s.title}" (${s.persona})`);e.stories=e.stories.slice(0,o);let i=new Set(e.stories.map(s=>s.id));for(let s of e.stories)s.dependencies=s.dependencies.filter(a=>i.has(a));return{droppedCount:n,details:r}}function Po(e){let o=new Map,n=0,t=[];for(let r of e.stories){if(!r.targetFiles||r.targetFiles.length===0)continue;let i=[],s=[];for(let a of r.targetFiles)o.get(a)?s.push(a):(o.set(a,r.id),i.push(a));s.length>0&&(r.targetFiles=i,n+=s.length,t.push(`${r.id}: removed ${s.join(", ")} (owned by ${s.map(a=>o.get(a)).join(", ")})`))}return{resolvedCount:n,details:t}}function xo(e){return"```json\n"+JSON.stringify(e,null,2)+"\n```"}function Gt(e,o){if(!he)return null;let n=JSON.stringify(o,null,2);return he.promptTemplate.replace("{{PRD}}",e).replace("{{PLAN}}",n)}function jt(e){let o=e.trim();if(o.includes("```")){let r=o.match(/```(?:json)?\s*([\s\S]*?)```/);r&&(o=r[1].trim())}let n=o.indexOf("{");n>0&&(o=o.substring(n));let t=JSON.parse(o);return{approved:t.approved,score:Math.max(0,Math.min(100,Math.round(t.score))),risks:t.risks||[],suggestions:t.suggestions,storyFeedback:Array.isArray(t.storyFeedback)?t.storyFeedback:void 0}}function Bt(e,o,n,t,r){return new Promise((i,s)=>{let a=Kt(e,["--print","--model",o,"--permission-mode","bypassPermissions"],{env:t,stdio:["pipe","pipe","pipe"]});a.stdin.write(n),a.stdin.end();let c="",m="";a.stdout.on("data",g=>{let h=g.toString();c+=h;let P=h.split(`
|
|
4
|
-
`).filter(d=>d.trim());for(let d of P){let
|
|
5
|
-
`)}var
|
|
2
|
+
var $t=Object.defineProperty;var wt=(e,o)=>()=>(e&&(o=e(e=0)),o);var yt=(e,o)=>{for(var n in o)$t(e,n,{get:o[n],enumerable:!0})};var Ao={};yt(Ao,{checkPrerequisites:()=>no,findClaudePath:()=>ie,getConfigDir:()=>Tt,getConfigFile:()=>ee,getLogFile:()=>_e,getPidFile:()=>ce,getSystemInfo:()=>Ue,loadConfig:()=>yo,loadConfigFromFile:()=>he,saveConfigToFile:()=>to,validatePrerequisites:()=>Io});import{existsSync as Ne,readFileSync as It,mkdirSync as At,writeFileSync as kt,chmodSync as bt}from"fs";import{execSync as fe}from"child_process";import{hostname as wo,homedir as Se}from"os";import{join as Z}from"path";function Tt(){return Ce}function ee(){return Te}function ce(){return Et}function _e(){return St}function he(){Ne(Te)||(console.error("No config found. Run 'workermill-agent setup' first."),process.exit(1));let e;try{e=It(Te,"utf-8")}catch{console.error("Failed to read config file:",Te),process.exit(1)}let o;try{o=JSON.parse(e)}catch{console.error("Config file is corrupted. Re-run 'workermill-agent setup'."),process.exit(1)}(!o.apiUrl||!o.apiKey)&&(console.error("Config file is missing required fields (apiUrl, apiKey). Re-run 'workermill-agent setup'."),process.exit(1));let n=o.workerImage||"workermill-worker:local";return{apiUrl:o.apiUrl,apiKey:o.apiKey,agentId:o.agentId,maxWorkers:o.maxWorkers||4,pollIntervalMs:o.pollIntervalMs||5e3,heartbeatIntervalMs:o.heartbeatIntervalMs||3e4,githubToken:o.tokens?.github||"",bitbucketToken:o.tokens?.bitbucket||"",gitlabToken:o.tokens?.gitlab||"",workerImage:n}}function to(e){Ne(Ce)||At(Ce,{recursive:!0}),kt(Te,JSON.stringify(e,null,2),"utf-8");try{bt(Te,384)}catch{}}function yo(){let e=process.env.WORKERMILL_API_URL,o=process.env.WORKERMILL_API_KEY;return e||(console.error("WORKERMILL_API_URL is required in .env.remote"),process.exit(1)),o||(console.error("WORKERMILL_API_KEY is required in .env.remote"),console.error("Get your API key from Settings > Integrations on the WorkerMill dashboard."),process.exit(1)),{apiUrl:e.replace(/\/$/,""),apiKey:o,agentId:process.env.AGENT_ID||`agent-${wo()}`,maxWorkers:parseInt(process.env.MAX_WORKERS||"4",10),pollIntervalMs:parseInt(process.env.POLL_INTERVAL_MS||"5000",10),heartbeatIntervalMs:parseInt(process.env.HEARTBEAT_INTERVAL_MS||"30000",10),githubToken:process.env.GITHUB_TOKEN||"",bitbucketToken:process.env.BITBUCKET_TOKEN||"",gitlabToken:process.env.GITLAB_TOKEN||"",workerImage:process.env.WORKER_IMAGE||"workermill-worker:local"}}function ie(){let e=process.platform==="win32",o=e?"where":"which";try{return fe(`${o} claude`,{stdio:"ignore",timeout:1e4}),"claude"}catch{}let n=[];e?n.push(Z(process.env.ProgramFiles||"C:\\Program Files","ClaudeCode","claude.exe"),Z(process.env.LOCALAPPDATA||"","Programs","ClaudeCode","claude.exe"),Z(Se(),"AppData","Local","Programs","ClaudeCode","claude.exe"),Z(Se(),".local","bin","claude.exe")):n.push(Z(Se(),".local","bin","claude"),"/opt/homebrew/bin/claude","/usr/local/bin/claude");for(let t of n)if(t&&Ne(t))return t;return null}function no(e){let o=[],n=e||"workermill-worker:local";try{let c=fe("docker version --format {{.Server.Version}}",{encoding:"utf-8",timeout:1e4}).trim();o.push({name:"Docker",ok:!0,detail:c})}catch{o.push({name:"Docker",ok:!1,detail:"Not running or not installed"})}let t=ie();if(t)try{let c=fe(`"${t}" --version`,{encoding:"utf-8",timeout:1e4}).trim();o.push({name:"Claude CLI",ok:!0,detail:c})}catch{o.push({name:"Claude CLI",ok:!0,detail:t})}else o.push({name:"Claude CLI",ok:!1,detail:"Not installed"});let r=Se(),i=Z(r,".claude",".credentials.json");Ne(i)?o.push({name:"Claude auth",ok:!0,detail:"Credentials found"}):o.push({name:"Claude auth",ok:!1,detail:"Run 'claude' and complete sign-in"});let s=process.version;parseInt(s.slice(1).split(".")[0],10)>=20?o.push({name:"Node.js",ok:!0,detail:s}):o.push({name:"Node.js",ok:!1,detail:`${s} (need >= 20)`});try{fe(`docker image inspect ${n}`,{stdio:"ignore",timeout:1e4}),o.push({name:"Worker image",ok:!0,detail:n})}catch{o.push({name:"Worker image",ok:!1,detail:`'${n}' not found`})}return o}function Io(){try{fe("docker version",{stdio:"ignore"})}catch{console.error("Docker is not available. Please install Docker and ensure it's running."),process.exit(1)}let e=process.env.WORKER_IMAGE||"workermill-worker:local";try{fe(`docker image inspect ${e}`,{stdio:"ignore"})}catch{console.error(`Worker image '${e}' not found.`),console.error(e==="workermill-worker:local"?"Build it with: ./bin/local-workermill build-worker":`Pull it with: docker pull ${e}`),process.exit(1)}ie()||(console.error("Claude CLI is not installed."),console.error("Install it: curl -fsSL https://claude.ai/install.sh | bash"),process.exit(1));let o=Se(),n=Z(o,".claude",".credentials.json");Ne(n)||(console.error("Claude credentials not found."),console.error("Run 'claude' and complete the sign-in flow to authenticate."),process.exit(1))}function Ue(){let e="unknown";try{e=fe("docker version --format {{.Server.Version}}",{encoding:"utf-8",timeout:1e4}).trim()}catch{}let o="unknown",n=ie();if(n)try{o=fe(`"${n}" --version`,{encoding:"utf-8",timeout:1e4}).trim()}catch{}return{hostname:wo(),platform:process.platform,nodeVersion:process.version,dockerVersion:e,claudeVersion:o}}var Ce,Te,Et,St,oe=wt(()=>{"use strict";Ce=Z(Se(),".workermill"),Te=Z(Ce,"config.json"),Et=Z(Ce,"agent.pid"),St=Z(Ce,"agent.log")});import{existsSync as Gn}from"fs";import{Command as jn}from"commander";oe();import u from"chalk";import Re from"ora";import ro from"inquirer";import{execSync as de,spawnSync as ko}from"child_process";import{existsSync as Ke}from"fs";import{hostname as Ct,homedir as Eo,totalmem as _t,cpus as Rt}from"os";import{join as We}from"path";import Pt from"axios";var Ae=process.platform==="win32";function so(e){try{return de(`${Ae?"where":"which"} ${e}`,{stdio:"ignore",timeout:1e4}),!0}catch{return!1}}function io(e){try{return de(`"${e}" --version`,{encoding:"utf-8",timeout:1e4}).trim()}catch{return null}}function bo(){if(!Ae)return null;if(so("git"))try{let o=de("where git",{encoding:"utf-8",timeout:1e4}).trim().split(`
|
|
3
|
+
`)[0],n=We(o,"..","..","bin","bash.exe");if(Ke(n))return n}catch{}let e=[We(process.env.ProgramFiles||"C:\\Program Files","Git","bin","bash.exe"),"C:\\Program Files\\Git\\bin\\bash.exe","C:\\Program Files (x86)\\Git\\bin\\bash.exe",We(Eo(),"scoop","apps","git","current","bin","bash.exe")];for(let o of e)if(Ke(o))return o;return null}function xt(){if(Ae)try{return de("winget install Anthropic.ClaudeCode --accept-package-agreements --accept-source-agreements",{stdio:"inherit",timeout:18e4}),!0}catch{return!1}else{if(process.platform==="darwin")try{return de("brew install --cask claude-code",{stdio:"inherit",timeout:18e4}),!0}catch{}try{return de("curl -fsSL https://claude.ai/install.sh | bash",{stdio:"inherit",timeout:12e4}),!0}catch{return!1}}}async function ao(){console.log(),console.log(u.bold.cyan(" WorkerMill Remote Agent Setup")),console.log(u.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(),console.log(" Run AI workers locally with your Claude Max subscription."),console.log(" Workers execute on your machine, logs stream to the cloud dashboard."),console.log();let e=Math.round(_t()/(1024*1024*1024)),o=Rt().length;console.log(u.dim(" System")),console.log(` ${u.dim("RAM:")} ${e} GB ${u.dim("CPUs:")} ${o}`),e<8?(console.log(),console.log(u.red(" \u2717 Insufficient RAM")),console.log(u.yellow(" WorkerMill requires at least 8 GB of RAM (16 GB recommended).")),console.log(u.yellow(` Your system has ${e} GB.`)),console.log(),process.exit(1)):e<16?(console.log(u.yellow(` \u26A0 ${e} GB RAM is below the recommended 16 GB.`)),console.log(u.yellow(" Workers may run slowly or be killed by the OS under memory pressure."))):console.log(u.green(" \u2713 System meets requirements")),console.log();let n=Re("Checking Docker...").start();if(so("docker")){let l=io("docker");try{let b=de("docker info --format {{.OSType}}",{encoding:"utf-8",timeout:15e3}).trim();Ae&&b==="windows"&&(n.fail("Docker is running in Windows containers mode"),console.log(),console.log(u.yellow(" WorkerMill workers require Linux containers.")),console.log(u.yellow(" Right-click the Docker Desktop icon in the system tray \u2192")),console.log(u.yellow(" 'Switch to Linux containers...'")),console.log(),console.log(" Then re-run: workermill-agent"),process.exit(1))}catch{n.fail("Docker is installed but the daemon is not running"),console.log(),console.log(u.yellow(" Start Docker Desktop, wait for it to fully initialize,")),console.log(u.yellow(" then re-run: workermill-agent")),process.exit(1)}n.succeed(`Docker ${u.dim(l?`(${l})`:"")}`)}else n.fail("Docker is not installed"),console.log(),console.log(u.yellow(" Docker Desktop is required but must be installed manually:")),console.log(` ${u.cyan("https://docs.docker.com/get-docker/")}`),console.log(),console.log(" Install Docker, start it, then re-run: workermill-agent"),process.exit(1);if(Ae){let l=Re("Checking Git for Windows...").start(),b=bo();b?(l.succeed(`Git for Windows ${u.dim(`(${b})`)}`),process.env.CLAUDE_CODE_GIT_BASH_PATH=b):(l.fail("Git for Windows is not installed"),console.log(),console.log(u.yellow(" Claude Code on Windows requires Git for Windows (Git Bash):")),console.log(` ${u.cyan("https://git-scm.com/downloads/win")}`),console.log(),console.log(" Install Git for Windows, then re-run: workermill-agent"),process.exit(1))}let t=Re("Checking Claude CLI...").start(),r=ie();if(r){let l=io(r);t.succeed(`Claude CLI ${u.dim(l?`(${l})`:"")}`)}else if(t.warn("Claude CLI not found \u2014 installing..."),console.log(),xt()&&(r=ie()),r){let b=io(r);console.log(),console.log(u.green(` \u2713 Claude CLI installed ${u.dim(b?`(${b})`:"")}`))}else console.log(),console.log(u.red(" Claude CLI was installed but could not be found.")),console.log(),Ae?(console.log(u.yellow(" Winget updated your PATH but this shell doesn't have it yet.")),console.log(u.yellow(" Close this terminal, open a new one, and re-run: workermill-agent"))):(console.log(" Try manually:"),console.log(u.cyan(" curl -fsSL https://claude.ai/install.sh | bash")),console.log(" Then re-run: workermill-agent")),process.exit(1);let i=Re("Checking Claude authentication...").start(),s=We(Eo(),".claude",".credentials.json");if(Ke(s))i.succeed("Claude authenticated");else{i.warn("Not authenticated \u2014 launching Claude..."),console.log(),console.log(u.dim(" Claude will open and prompt you to authenticate.")),console.log(u.dim(" Sign in with your Claude Max account, then exit Claude (Ctrl+C).")),console.log();let l={...process.env};if(Ae){let b=bo();b&&(l.CLAUDE_CODE_GIT_BASH_PATH=b)}ko(r,[],{stdio:"inherit",timeout:3e5,env:l}),Ke(s)?(console.log(),console.log(u.green(" \u2713 Claude authenticated"))):(console.log(),console.log(u.red(" Authentication failed or was cancelled.")),console.log(" Try manually: open a new terminal and run 'claude'"),console.log(" Then re-run: workermill-agent"),process.exit(1))}console.log(u.green(" \u2713")+` Node.js ${u.dim(`(${process.version})`)}`),console.log(),console.log(u.bold("Configuration")),console.log();let{apiUrl:a}=await ro.prompt([{type:"input",name:"apiUrl",message:"WorkerMill API URL:",default:"https://workermill.com",validate:l=>l.startsWith("http://")||l.startsWith("https://")?!0:"Must be a valid URL"}]),{apiKey:c}=await ro.prompt([{type:"password",name:"apiKey",message:"API Key (from Settings > Integrations):",mask:"*",validate:l=>l.length>0?!0:"API key is required"}]),h=Re("Validating API key...").start(),w="github",m="",g="";try{let l=await Pt.get(`${a.replace(/\/$/,"")}/api/agent/config`,{headers:{"x-api-key":c},timeout:15e3});w=l.data.scmProvider||"github",m=l.data.workerImageUrl||"",g=l.data.ecrRegistry||"",h.succeed(`Connected! SCM provider: ${w}`)}catch(l){l.response?.status===401?h.fail("Invalid API key. Check Settings > Integrations on the dashboard."):h.fail("Failed to connect to WorkerMill API. Check the URL and try again."),process.exit(1)}console.log(u.dim(" SCM tokens are managed via Settings > Integrations on the dashboard.")),console.log();let{agentId:P}=await ro.prompt([{type:"input",name:"agentId",message:"Agent name:",default:`agent-${Ct()}`}]);console.log();let d=m||"workermill-worker:local",$=g.length>0,v=Re("Checking AWS CLI...").start();if(so("aws"))try{de("aws sts get-caller-identity",{stdio:"pipe",timeout:15e3}),v.succeed("AWS CLI configured")}catch{v.warn("AWS CLI found but credentials not configured"),console.log(),console.log(u.yellow(" Configure AWS credentials for private ECR image access:")),console.log(u.cyan(" aws configure")),console.log(u.dim(" Contact your WorkerMill admin for AWS access key / secret key.")),console.log(u.dim(" Setup will continue \u2014 you can configure AWS later before starting."))}else v.warn("AWS CLI not found"),console.log(),console.log(u.yellow(" AWS CLI is required for pulling worker images from private ECR.")),console.log(u.cyan(" Install: https://docs.aws.amazon.com/cli/latest/userguide/getting-started-install.html")),console.log(u.dim(" Setup will continue \u2014 install AWS CLI before starting the agent."));let A=!1;if($){console.log(),console.log(u.dim(" Authenticating with private ECR..."));let l=g.match(/\.ecr\.([a-z0-9-]+)\.amazonaws\.com/),b=l?l[1]:"us-east-1";try{de(`aws ecr get-login-password --region ${b} | docker login --username AWS --password-stdin ${g}`,{stdio:"pipe",timeout:3e4}),console.log(u.green(" \u2713 ECR authenticated")),A=!0}catch{console.log(u.yellow(" \u26A0 ECR authentication failed (AWS credentials may not be configured yet)")),console.log(u.dim(" Worker image pull will be skipped \u2014 authenticate later with:")),console.log(u.dim(` aws ecr get-login-password --region ${b} | docker login --username AWS --password-stdin ${g}`))}}if(A||!$){console.log(),console.log(u.dim(` Pulling worker image: ${d}`)),console.log(u.dim(" This may take a few minutes on first run (~1.1 GB)...")),console.log();let l=ko("docker",["pull",d],{stdio:"inherit",timeout:6e5});l.status===0?(console.log(),console.log(u.green(" \u2713 Worker image pulled"))):(console.log(),console.log(u.red(" \u2717 Failed to pull worker image.")),l.error&&console.log(u.yellow(` Error: ${l.error.message}`)),console.log(u.dim(" Setup will continue \u2014 you can pull the image later before starting.")))}let B={apiUrl:a.replace(/\/$/,""),apiKey:c,agentId:P,maxWorkers:1,pollIntervalMs:5e3,heartbeatIntervalMs:3e4,tokens:{github:"",bitbucket:"",gitlab:""},workerImage:d,setupCompletedAt:new Date().toISOString()};to(B),console.log(),console.log(u.green.bold(" Setup complete!")),console.log(),console.log(` Config saved to: ${u.dim(ee())}`),console.log(),console.log(` Start the agent with: ${u.cyan("workermill-agent start")}`),console.log()}import U from"chalk";import{totalmem as An}from"os";import{spawn as kn}from"child_process";import{writeFileSync as ct,existsSync as bn,unlinkSync as En,openSync as Sn,createWriteStream as Tn}from"fs";import{createRequire as vt}from"module";var Mt=vt(import.meta.url),Nt=Mt("../package.json"),z=Nt.version;oe();import K from"chalk";import Ot from"axios";var lo=null;function Ge(e,o){lo=Ot.create({baseURL:e,headers:{"Content-Type":"application/json","x-api-key":o},timeout:3e4})}var N=new Proxy({},{get(e,o){if(!lo)throw new Error("API client not initialized. Call initApi() first.");return lo[o]}});import S from"chalk";oe();import p from"chalk";import{spawn as nn,execSync as po}from"child_process";import{spawn as Gt}from"child_process";import we from"chalk";import je from"axios";var Lt=16384,Ft=6e5;async function So(e,o,n,t,r){let i=r?.maxTokens??Lt,s=r?.temperature??.7,a=r?.timeoutMs??Ft;switch(e){case"anthropic":return Dt(o,n,t,i,s,a);case"openai":return Ut(o,n,t,i,s,a);case"google":return Wt(o,n,t,i,s,a);case"ollama":return Kt(o,n,t,i,s,a);default:throw new Error(`Unsupported AI provider: ${e}`)}}async function Dt(e,o,n,t,r,i){let a=(await je.post("https://api.anthropic.com/v1/messages",{model:e,max_tokens:t,temperature:r,messages:[{role:"user",content:o}]},{headers:{"x-api-key":n,"anthropic-version":"2023-06-01","Content-Type":"application/json"},timeout:i})).data?.content;if(Array.isArray(a))return a.filter(c=>c.type==="text").map(c=>c.text).join("");throw new Error("Unexpected Anthropic API response format")}async function Ut(e,o,n,t,r,i){let a=(await je.post("https://api.openai.com/v1/chat/completions",{model:e,max_tokens:t,temperature:r,messages:[{role:"user",content:o}]},{headers:{Authorization:`Bearer ${n}`,"Content-Type":"application/json"},timeout:i})).data?.choices?.[0]?.message;if(a?.content)return a.content;throw new Error("Unexpected OpenAI API response format")}async function Wt(e,o,n,t,r,i){let c=(await je.post(`https://generativelanguage.googleapis.com/v1beta/models/${e}:generateContent?key=${n}`,{contents:[{parts:[{text:o}]}],generationConfig:{maxOutputTokens:t,temperature:r}},{headers:{"Content-Type":"application/json"},timeout:i})).data?.candidates?.[0]?.content?.parts?.[0]?.text;if(c)return c;throw new Error("Unexpected Google AI API response format")}async function Kt(e,o,n,t,r,i){let s=n||"http://localhost:11434",c=(await je.post(`${s}/api/generate`,{model:e,prompt:o,stream:!1,options:{temperature:r}},{headers:{"Content-Type":"application/json"},timeout:i})).data?.response;if(c)return c;throw new Error("Unexpected Ollama API response format")}var $e=null;async function Co(e){if($e)return $e;try{let o={};e!==void 0&&(o.maxTargetFiles=String(e));let{data:n}=await N.get("/api/agent/critic-prompt",{params:o});return $e={promptTemplate:n.promptTemplate,approvalThreshold:n.approvalThreshold??85,maxTargetFiles:n.maxTargetFiles??e??5},te=$e.approvalThreshold,Pe=$e.maxTargetFiles,$e}catch{return console.warn("Failed to fetch critic config from API"),null}}var Pe=5,te=85;function _o(e){let o=e.indexOf("```json");if(o!==-1){let t=o+7,r=e.indexOf("{",t);if(r!==-1){let i=To(e,r);if(i)return JSON.parse(i)}}let n=e.indexOf('"stories"');if(n!==-1){let r=e.substring(0,n).lastIndexOf("{");if(r!==-1){let i=To(e,r);if(i)return JSON.parse(i)}}throw new Error("Could not find JSON execution plan in output")}function To(e,o){let n=0,t=!1,r=!1;for(let i=o;i<e.length;i++){let s=e[i];if(r){r=!1;continue}if(s==="\\"){t&&(r=!0);continue}if(s==='"'){t=!t;continue}if(!t){if(s==="{")n++;else if(s==="}"&&(n--,n===0))return e.substring(o,i+1)}}return null}function Ro(e){let o=0,n=[];for(let t of e.stories)if(!t.targetFiles||!Array.isArray(t.targetFiles))t.targetFiles=[];else if(t.targetFiles.length>Pe){let r=t.targetFiles.slice(Pe);n.push(`${t.id}: ${t.targetFiles.length} files \u2192 ${Pe} (dropped: ${r.join(", ")})`),t.targetFiles=t.targetFiles.slice(0,Pe),o++}return{truncatedCount:o,details:n}}function Po(e,o){if(e.stories.length<=o)return{droppedCount:0,details:[]};let n=e.stories.length-o,r=e.stories.slice(o).map(s=>`${s.id}: "${s.title}" (${s.persona})`);e.stories=e.stories.slice(0,o);let i=new Set(e.stories.map(s=>s.id));for(let s of e.stories)s.dependencies=s.dependencies.filter(a=>i.has(a));return{droppedCount:n,details:r}}function xo(e){let o=new Map,n=0,t=[];for(let r of e.stories){if(!r.targetFiles||r.targetFiles.length===0)continue;let i=[],s=[];for(let a of r.targetFiles)o.get(a)?s.push(a):(o.set(a,r.id),i.push(a));s.length>0&&(r.targetFiles=i,n+=s.length,t.push(`${r.id}: removed ${s.join(", ")} (owned by ${s.map(a=>o.get(a)).join(", ")})`))}return{resolvedCount:n,details:t}}function vo(e){return"```json\n"+JSON.stringify(e,null,2)+"\n```"}function jt(e,o){if(!$e)return null;let n=JSON.stringify(o,null,2);return $e.promptTemplate.replace("{{PRD}}",e).replace("{{PLAN}}",n)}function Bt(e){let o=e.trim();if(o.includes("```")){let r=o.match(/```(?:json)?\s*([\s\S]*?)```/);r&&(o=r[1].trim())}let n=o.indexOf("{");n>0&&(o=o.substring(n));let t=JSON.parse(o);return{approved:t.approved,score:Math.max(0,Math.min(100,Math.round(t.score))),risks:t.risks||[],suggestions:t.suggestions,storyFeedback:Array.isArray(t.storyFeedback)?t.storyFeedback:void 0}}function Vt(e,o,n,t,r){return new Promise((i,s)=>{let a=Gt(e,["--print","--model",o,"--permission-mode","bypassPermissions"],{env:t,stdio:["pipe","pipe","pipe"]});a.stdin.write(n),a.stdin.end();let c="",h="";a.stdout.on("data",m=>{let g=m.toString();c+=g;let P=g.split(`
|
|
4
|
+
`).filter(d=>d.trim());for(let d of P){let $=d.trim().length>200?d.trim().substring(0,200)+"\u2026":d.trim();$&&(r&&Oo(r,`${No} [critic] ${$}`,"output"),console.log(`${Oe()} ${we.dim("\u{1F50D}")} ${we.dim($)}`))}}),a.stderr.on("data",m=>{h+=m.toString()});let w=setTimeout(()=>{a.kill("SIGTERM"),s(new Error("Critic CLI timed out after 20 minutes"))},12e5);a.on("exit",m=>{clearTimeout(w),m!==0?s(new Error(`Critic CLI failed (exit ${m}): ${h.substring(0,300)}`)):i(c)}),a.on("error",m=>{clearTimeout(w),s(m)})})}function Mo(e){let o=["","## CRITIC FEEDBACK \u2014 Your previous plan was REJECTED","",`Score: ${e.score}/100 (need >= ${te} to pass)`,""];if(e.risks.length>0){o.push("### Risks Identified:");for(let n of e.risks)o.push(`- ${n}`);o.push("")}if(e.suggestions&&e.suggestions.length>0){o.push("### Required Changes:");for(let n of e.suggestions)o.push(`- ${n}`);o.push("")}if(e.storyFeedback&&e.storyFeedback.length>0){o.push("### Per-Story Feedback:");for(let n of e.storyFeedback)if(o.push(`- **${n.storyId}**: ${n.feedback}`),n.suggestedChanges)for(let t of n.suggestedChanges)o.push(` - ${t}`);o.push("")}return o.push(`**You MUST address ALL feedback above.** Each story must target at most ${Pe} files.`,"Stories MUST NOT overlap on targetFiles.","","**CRITICAL \u2014 OUTPUT FORMAT:** You MUST output a revised plan as a ```json code block containing the full JSON object with `summary`, `stories`, `risks`, and `assumptions` fields. Do NOT just describe what you would change \u2014 output the COMPLETE revised JSON plan. If you do not output a ```json block, the plan will fail to parse and the task will fail.","","**DO NOT re-explore the repository.** You already explored it in the previous attempt. Go directly to outputting the revised ```json plan. Every tool call wastes output budget \u2014 prioritize emitting the JSON plan."),o.join(`
|
|
5
|
+
`)}var No="[\u{1F5FA}\uFE0F planning_agent \u{1F916}]";function Oe(){return we.dim(new Date().toLocaleTimeString())}async function Oo(e,o,n="system",t="info"){try{await N.post("/api/control-center/logs",{taskId:e,type:n,message:o,severity:t})}catch{}}async function Lo(e,o,n,t,r,i,s,a,c){let h=jt(n,t);if(!h)return console.warn(`${Oe()} ${i} ${we.yellow("\u26A0")} Critic config not available \u2014 skipping validation`),null;let w=s||"anthropic";console.log(`${Oe()} ${i} ${we.dim(`Running critic validation (${w})...`)}`),c&&Oo(c,`${No} Running critic validation (${w})...`);try{let m;if(w==="anthropic")m=await Vt(e,o,h,r,c);else{if(!a)throw new Error(`No API key for critic provider "${w}"`);m=await So(w,o,h,a,{maxTokens:4096,temperature:.3,timeoutMs:12e5})}let g=Bt(m),P=g.score>=te?we.green("\u2713"):we.red("\u2717");return console.log(`${Oe()} ${i} ${P} Critic score: ${g.score}/100 (threshold: ${te})`),g}catch(m){let g=m instanceof Error?m.message:String(m);return console.error(`${Oe()} ${i} ${we.yellow("\u26A0")} Critic failed: ${g.substring(0,100)}`),null}}import{generateText as zt,tool as co,stepCountIs as qt}from"ai";import{createOpenAI as Fo}from"@ai-sdk/openai";import{createAnthropic as Yt}from"@ai-sdk/anthropic";import{createGoogleGenerativeAI as Ht}from"@ai-sdk/google";import{z as ye}from"zod";import{execSync as uo}from"child_process";import{readFileSync as Jt,existsSync as Xt}from"fs";function Qt(e,o,n){switch(e){case"anthropic":return Yt({apiKey:n})(o);case"openai":return Fo({apiKey:n})(o);case"google":return Ht({apiKey:n})(o);case"ollama":return Fo({baseURL:n||"http://localhost:11434/v1",apiKey:"ollama"})(o);default:throw new Error(`Unsupported AI provider: ${e}`)}}var Zt=ye.object({pattern:ye.string().describe("Glob pattern like '**/*.ts', 'src/**/*.js', 'package.json'")}),en=ye.object({path:ye.string().describe("File path relative to the working directory"),limit:ye.number().optional().describe("Max number of lines to read (default: 500)")}),on=ye.object({pattern:ye.string().describe("Search pattern (regex supported)"),glob:ye.string().optional().describe("File glob to filter (e.g. '*.ts', '*.py')")});function tn(e){return{glob:co({description:"Find files matching a glob pattern. Returns file paths relative to the working directory.",inputSchema:Zt,execute:async o=>{try{let n=uo(`find . -path './.git' -prune -o -path './node_modules' -prune -o -name '${o.pattern.replace(/\*\*/g,"*")}' -print 2>/dev/null | head -200`,{cwd:e,encoding:"utf-8",timeout:15e3}).trim();return n||uo("find . -path './.git' -prune -o -path './node_modules' -prune -o -type f -print 2>/dev/null | head -500",{cwd:e,encoding:"utf-8",timeout:15e3}).trim()||"No files found"}catch{return"Error running glob search"}}}),read_file:co({description:"Read the contents of a file. Returns the file text.",inputSchema:en,execute:async o=>{try{let n=`${e}/${o.path}`.replace(/\/\//g,"/");if(!Xt(n))return`File not found: ${o.path}`;let t=Jt(n,"utf-8"),r=t.split(`
|
|
6
6
|
`),i=o.limit||500;return r.length>i?r.slice(0,i).join(`
|
|
7
7
|
`)+`
|
|
8
|
-
... (truncated, ${r.length-i} more lines)`:t}catch(n){return`Error reading file: ${n instanceof Error?n.message:String(n)}`}}}),grep:
|
|
9
|
-
`),
|
|
10
|
-
`);
|
|
11
|
-
|
|
8
|
+
... (truncated, ${r.length-i} more lines)`:t}catch(n){return`Error reading file: ${n instanceof Error?n.message:String(n)}`}}}),grep:co({description:"Search for a pattern in files. Returns matching lines with file paths and line numbers.",inputSchema:on,execute:async o=>{try{let n=o.glob?`--include='${o.glob}'`:"";return uo(`grep -rn ${n} --exclude-dir=node_modules --exclude-dir=.git '${o.pattern.replace(/'/g,"'\\''")}' . 2>/dev/null | head -100`,{cwd:e,encoding:"utf-8",timeout:15e3}).trim()||"No matches found"}catch{return"No matches found"}}})}}async function Do(e){let{provider:o,model:n,apiKey:t,prompt:r,systemPrompt:i,workingDir:s,maxTokens:a=16384,temperature:c=.7,timeoutMs:h=6e5,maxSteps:w=15,enableTools:m=!0}=e,g=Qt(o,n,t),P=m&&s?tn(s):void 0,d=new AbortController,$=setTimeout(()=>d.abort(),h);try{return(await zt({model:g,prompt:r,system:i,maxOutputTokens:a,temperature:c,tools:P,stopWhen:P?qt(w):void 0,abortSignal:d.signal})).text}finally{clearTimeout($)}}function rn(e,o){let n=[e.usage,e.message?.usage,e.result?.usage];for(let t of n)if(t&&typeof t=="object"){let r=t;typeof r.input_tokens=="number"&&(o.inputTokens=Math.max(o.inputTokens,r.input_tokens)),typeof r.output_tokens=="number"&&(o.outputTokens=Math.max(o.outputTokens,r.output_tokens)),typeof r.cache_creation_input_tokens=="number"&&(o.cacheCreationTokens=Math.max(o.cacheCreationTokens,r.cache_creation_input_tokens)),typeof r.cache_read_input_tokens=="number"&&(o.cacheReadTokens=Math.max(o.cacheReadTokens,r.cache_read_input_tokens))}}async function Uo(e,o,n,t){if(!(o.inputTokens===0&&o.outputTokens===0))try{await N.post(`/api/tasks/${e}/usage/partial`,{inputTokens:o.inputTokens,outputTokens:o.outputTokens,cacheCreationTokens:o.cacheCreationTokens,cacheReadTokens:o.cacheReadTokens,model:n,mode:t})}catch{}}var ue=3;function I(){return p.dim(new Date().toLocaleTimeString())}var xe=[],Ie=null;async function Ko(){for(;xe.length>0;){let e=xe.splice(0,50);try{await N.post("/api/control-center/logs/batch",{entries:e},{timeout:5e3})}catch{}}}async function R(e,o,n="system",t="info"){xe.length>=200&&xe.shift(),xe.push({taskId:e,message:o,type:n,severity:t}),Ie||(Ie=Ko().finally(()=>{Ie=null}))}async function sn(){Ie&&await Ie,xe.length>0&&(Ie=Ko().finally(()=>{Ie=null}),await Ie)}async function ve(e,o,n,t,r,i){try{await N.post("/api/agent/planning-progress",{taskId:e,phase:o,elapsedSeconds:n,detail:t,charsGenerated:r,toolCallCount:i})}catch{}}var E="[\u{1F5FA}\uFE0F planning_agent \u{1F916}]";function Be(e){let o=Math.floor(e/60),n=e%60;return o>0?`${o}m ${n}s`:`${n}s`}function Wo(e,o){switch(e){case"initializing":return`${E} Starting planning agent...`;case"reading_repo":return`${E} Reading repository structure...`;case"analyzing":return`${E} Analyzing requirements...`;case"generating_plan":return`${E} Planning in progress \u2014 analyzing requirements and decomposing into steps (${Be(o)} elapsed)`;case"validating":return`${E} Validating plan...`;case"complete":return`${E} Planning complete`}}function an(e,o,n,t,r,i,s){let a=p.cyan(r.slice(0,8));return new Promise((c,h)=>{let m=nn(e,["--print","--verbose","--output-format","stream-json","--model",o,"--permission-mode","bypassPermissions"],{cwd:s,env:t,stdio:["pipe","pipe","pipe"]});m.stdin.write(n),m.stdin.end();let g="",P="",d="",$=0,v=0,A={inputTokens:0,outputTokens:0,cacheCreationTokens:0,cacheReadTokens:0},B=o,l="";function b(k=!1){if(!l)return;let G=l.split(`
|
|
9
|
+
`),pe=k?"":G.pop()||"";for(let Y of G)if(Y.trim()){R(r,`${E} ${Y}`,"output");let F=Y.trim().length>160?Y.trim().substring(0,160)+"\u2026":Y.trim();console.log(`${I()} ${a} ${p.dim("\u{1F4AD}")} ${p.dim(F)}`)}l=pe}let T="initializing",O=!1,y={started:!0,reading:!1,analyzing:!1,generating:!1};function L(k){if(k===T)return;T=k;let G=Math.round((Date.now()-i)/1e3),pe=Wo(k,G);R(r,pe),console.log(`${I()} ${a} ${p.dim(pe)}`)}let V=setInterval(()=>b(),500),q=setInterval(()=>{let k=Math.round((Date.now()-i)/1e3);ve(r,T,k,Wo(T,k),$,v)},2e3),x=0,ae=setInterval(()=>{let k=Math.round((Date.now()-i)/1e3);if(T==="initializing"&&k>=5?L("reading_repo"):T==="reading_repo"&&k>=15&&!O&&L("analyzing"),T==="generating_plan"&&k-x>=30){x=k;let G=`${E} Planning in progress \u2014 analyzing requirements and decomposing into steps (${Be(k)} elapsed)`;R(r,G),console.log(`${I()} ${a} ${p.dim(G)}`)}},5e3),X="";m.stdout.on("data",k=>{X+=k.toString();let G=X.split(`
|
|
10
|
+
`);X=G.pop()||"";for(let pe of G){let Y=pe.trim();if(Y)try{let F=JSON.parse(Y);if(F.type==="assistant"&&F.message?.content){let H=F.message.content;if(Array.isArray(H))for(let me of H)me.type==="text"&&me.text?(g+=me.text,$+=me.text.length,l+=me.text,O||(O=!0,v>0&&!y.analyzing&&(L("analyzing"),y.analyzing=!0)),$>500&&!y.generating&&(L("generating_plan"),y.generating=!0,x=Math.round((Date.now()-i)/1e3))):me.type==="tool_use"&&(v++,y.reading||(L("reading_repo"),y.reading=!0));else typeof H=="string"&&H&&(g+=H,$+=H.length,l+=H)}else F.type==="content_block_delta"&&F.delta?.text?(g+=F.delta.text,$+=F.delta.text.length,l+=F.delta.text,O||(O=!0,v>0&&!y.analyzing&&(L("analyzing"),y.analyzing=!0)),$>500&&!y.generating&&(L("generating_plan"),y.generating=!0,x=Math.round((Date.now()-i)/1e3))):F.type==="content_block_start"&&F.content_block?.type==="tool_use"?(v++,y.reading||(L("reading_repo"),y.reading=!0)):F.type==="result"&&F.result&&(P=typeof F.result=="string"?F.result:"");if(rn(F,A),F.type==="result"&&F.total_cost_usd!==void 0&&F.modelUsage&&typeof F.modelUsage=="object"){let H=Object.keys(F.modelUsage);H.length>0&&(B=H[0])}}catch{g+=Y+`
|
|
11
|
+
`,$+=Y.length}}}),m.stderr.on("data",k=>{d+=k.toString()});let re=setInterval(()=>{(A.inputTokens>0||A.outputTokens>0)&&Uo(r,A,B,"greatest").catch(()=>{})},3e4);function le(){clearInterval(ae),clearInterval(q),clearInterval(V),clearInterval(re),b(!0)}let De=setTimeout(()=>{le(),m.kill("SIGTERM"),h(new Error("Claude CLI timed out after 20 minutes"))},12e5);m.on("exit",k=>{clearTimeout(De),le();let G=Math.round((Date.now()-i)/1e3);ve(r,"validating",G,"Validating plan...",$,v),Uo(r,A,B,"greatest").catch(()=>{}),k!==0?h(new Error(`Claude CLI failed (exit ${k}): ${d.substring(0,300)}`)):c(P||g)}),m.on("error",k=>{clearTimeout(De),le(),h(k)})})}function ln(e,o){if(o)switch(e){case"anthropic":return o.anthropicApiKey;case"openai":return o.openaiApiKey;case"google":return o.googleApiKey;case"ollama":return o.ollamaBaseUrl||"http://localhost:11434";default:return}}function cn(e,o,n){switch(n){case"bitbucket":return`https://x-token-auth:${o}@bitbucket.org/${e}.git`;case"gitlab":return`https://oauth2:${o}@gitlab.com/${e}.git`;default:return`https://x-access-token:${o}@github.com/${e}.git`}}async function dn(e,o,n,t){let r=p.cyan(t.slice(0,8)),i=`/tmp/workermill-planning-${t.slice(0,8)}-${Date.now()}`;try{let s=cn(e,o,n);return console.log(`${I()} ${r} ${p.dim("Cloning repo for planner...")}`),po(`git clone --depth 1 --single-branch "${s}" "${i}"`,{stdio:"ignore",timeout:6e4}),console.log(`${I()} ${r} ${p.green("\u2713")} Repo cloned to ${p.dim(i)}`),i}catch(s){let a=s instanceof Error?s.message:String(s);console.error(`${I()} ${r} ${p.yellow("\u26A0")} Clone failed, planner will run without repo access: ${a.substring(0,100)}`);try{po(`rm -rf "${i}"`,{stdio:"ignore"})}catch{}return null}}async function Go(e,o,n){let t=p.cyan(e.id.slice(0,8));console.log(`${I()} ${t} Fetching planning prompt...`),await R(e.id,`${E} Fetching planning prompt from cloud API...`);let r=await N.get("/api/agent/planning-prompt",{params:{taskId:e.id}}),{prompt:i,model:s,provider:a,maxStories:c,maxTargetFiles:h}=r.data,w=typeof c=="number"?c:8,m=s,g=a||"anthropic",P=g==="anthropic",d=process.env.CLAUDE_CLI_PATH||ie()||"claude",$={...process.env};delete $.CLAUDE_CODE_OAUTH_TOKEN;let v=ln(g,n),A=Date.now(),B=e.description||e.summary,l=null;if(e.githubRepo){let q=e.scmProvider||"github",x=q==="bitbucket"?o.bitbucketToken:q==="gitlab"?o.gitlabToken:o.githubToken;x?l=await dn(e.githubRepo,x,q,e.id):console.log(`${I()} ${t} ${p.yellow("\u26A0")} No SCM token for ${q}, planner will run without repo access`)}let b=i,T=null,O=0,y=[],L=0,V=await Co(typeof h=="number"?h:void 0);V||(console.log(`${I()} ${t} ${p.yellow("\u26A0")} Could not fetch critic config \u2014 critic validation will be skipped`),await R(e.id,`${E} \u26A0\uFE0F Could not fetch critic config from API \u2014 critic validation will be skipped`));try{for(let x=1;x<=ue;x++){let ae=ue>1?` (attempt ${x}/${ue})`:"",X=`${g}/${m}`;x>1?(console.log(`${I()} ${t} Running planner${ae} ${p.dim(`(${p.yellow(X)})`)}`),await R(e.id,`${E} Re-planning${ae} using ${X}`)):(console.log(`${I()} ${t} Running planner ${p.dim(`(${p.yellow(X)})`)}`),await R(e.id,`${E} Starting planning agent using ${X}`));let re;try{if(P)re=await an(d,m,b,$,e.id,A,l||void 0);else{if(!v)throw new Error(`No API key available for provider "${g}". Configure it in Settings > Integrations.`);let _=Math.round((Date.now()-A)/1e3);await ve(e.id,"generating_plan",_,"Generating plan via AI SDK...",0,0),re=await Do({provider:g,model:m,apiKey:v,prompt:b,workingDir:l||void 0,enableTools:!!l,maxSteps:10});let M=Math.round((Date.now()-A)/1e3);await ve(e.id,"validating",M,"Validating plan...",re.length,0)}}catch(_){let M=Math.round((Date.now()-A)/1e3),Q=_ instanceof Error?_.message:String(_);return console.error(`${I()} ${t} ${p.red("\u2717")} Failed after ${M}s: ${Q.substring(0,100)}`),await R(e.id,`${E} Planning failed after ${Be(M)}: ${Q.substring(0,200)}`,"error","error"),!1}let le=Math.round((Date.now()-A)/1e3),De=P?"Claude CLI":`${g} API`;console.log(`${I()} ${t} ${p.green("\u2713")} ${De} done ${p.dim(`(${le}s, ${re.length} chars)`)}`);let k;try{k=_o(re)}catch(_){let M=_ instanceof Error?_.message:String(_);return console.error(`${I()} ${t} ${p.red("\u2717")} Plan parse failed: ${M.substring(0,100)}`),await R(e.id,`${E} Failed to parse execution plan from Claude output: ${M.substring(0,200)}`,"error","error"),await un(e.id,re,o.agentId,t,le)}let{truncatedCount:G,details:pe}=Ro(k);if(G>0){L+=G;let _=`${E} File cap applied: ${G} stories truncated to max ${V?.maxTargetFiles??5} targetFiles`;console.log(`${I()} ${t} ${p.yellow("\u26A0")} ${_}`),await R(e.id,_);for(let M of pe)console.log(`${I()} ${t} ${p.dim(M)}`)}let{droppedCount:Y,details:F}=Po(k,w);if(Y>0){let _=`${E} Story cap applied: ${Y} stories dropped (max ${w})`;console.log(`${I()} ${t} ${p.yellow("\u26A0")} ${_}`),await R(e.id,_);for(let M of F)console.log(`${I()} ${t} ${p.dim(M)}`)}let{resolvedCount:H,details:me}=xo(k);if(H>0){let _=`${E} File overlap resolved: ${H} shared file(s) de-duped across stories`;console.log(`${I()} ${t} ${p.yellow("\u26A0")} ${_}`),await R(e.id,_);for(let M of me)console.log(`${I()} ${t} ${p.dim(M)}`)}console.log(`${I()} ${t} Plan: ${p.bold(k.stories.length)} stories (max ${w})`),await R(e.id,`${E} Plan generated: ${k.stories.length} stories (${Be(le)}). Running critic validation...`);let C=await Lo(d,m,B,k,$,t,g,v,e.id);if(C&&C.score>O?(T=k,O=C.score):!C&&!T&&(T=k),C&&y.push({iteration:x,score:C.score,approved:C.approved||C.score>=te,risks:C.risks,suggestions:C.suggestions,filesCapApplied:G>0?G:void 0}),!C){let _=`${E} \u26A0\uFE0F CRITIC BYPASSED \u2014 Critic validation failed (timeout/parse error). Posting plan WITHOUT quality gate.`;console.log(`${I()} ${t} ${p.yellow("\u26A0")} ${_}`),await R(e.id,_,"error","warning");let M=Date.now()-A;return await go(e.id,k,o.agentId,t,le,void 0,void 0,y,L,M,x)}if(C.approved||C.score>=te){let _=`${E} Critic approved (score: ${C.score}/100)`;if(console.log(`${I()} ${t} ${p.green("\u2713")} ${_}`),await R(e.id,_),C.risks.length>0){let Q=`${E} Critic risks (non-blocking): ${C.risks.join("; ")}`;console.log(`${I()} ${t} ${p.dim(Q)}`),await R(e.id,Q)}let M=Date.now()-A;return await go(e.id,k,o.agentId,t,le,C.score,C.risks,y,L,M,x)}if(x<ue){let _=Mo(C);b=b+`
|
|
12
12
|
|
|
13
|
-
`+_;let v=`${E} Critic rejected (score: ${I.score}/100, threshold: ${ne}). Re-planning with feedback...`;if(console.log(`${k()} ${t} ${p.yellow("\u26A0")} ${v}`),await R(e.id,v),I.risks.length>0){let Z=`${E} Critic risks: ${I.risks.join("; ")}`;console.log(`${k()} ${t} ${p.dim(Z)}`),await R(e.id,Z)}if(I.suggestions&&I.suggestions.length>0){let Z=`${E} Critic suggestions: ${I.suggestions.join("; ")}`;console.log(`${k()} ${t} ${p.dim(Z)}`),await R(e.id,Z)}}else{let _=`${E} Critic rejected after ${ue} iterations (best score: ${b}/100, threshold: ${ne})`;if(console.error(`${k()} ${t} ${p.red("\u2717")} ${_}`),await R(e.id,_,"error","error"),I.risks.length>0){let v=`${E} Final risks: ${I.risks.join("; ")}`;console.error(`${k()} ${t} ${v}`),await R(e.id,v,"error","error")}if(I.suggestions&&I.suggestions.length>0){let v=`${E} Suggestions: ${I.suggestions.join("; ")}`;console.error(`${k()} ${t} ${v}`),await R(e.id,v,"error","error")}}}let D=50;if(A&&b>=D){let j=Math.round((Date.now()-T)/1e3),B=`${E} Best-plan fallback: posting plan with score ${b}/100 (below ${ne} threshold, above ${D} minimum)`;console.log(`${k()} ${t} ${p.yellow("\u26A0")} ${B}`),await R(e.id,B);let ie=Date.now()-T;if(await uo(e.id,A,o.agentId,t,j,b,[`Best-plan fallback: critic rejected after ${ue} iterations`],K,y,ie,ue))return!0;console.log(`${k()} ${t} ${p.yellow("\u26A0")} ${E} Fallback post rejected by server, reporting plan-failed`),await R(e.id,`${E} Fallback plan rejected by server \u2014 reporting failure`)}try{let j=A&&b>=D?`Best-plan fallback rejected by server after ${ue} iterations (best score: ${b}/100)`:`Critic rejected after ${ue} iterations (best score: ${b}/100, threshold: ${ne}, fallback minimum: ${D})`;await M.post("/api/agent/plan-failed",{taskId:e.id,agentId:o.agentId,reason:j,criticHistory:K})}catch{}return!1}finally{if(await rn(),W)try{go(`rm -rf "${W}"`,{stdio:"ignore"})}catch{}}}async function uo(e,o,n,t,r,i,s,a,c,m,$){let g=xo(o);try{let P=(await M.post("/api/agent/plan-result",{taskId:e,rawOutput:g,agentId:n,criticScore:i,criticRisks:s,criticHistory:a,criticIterations:$,fileCapTruncations:c,planningDurationMs:m})).data.storyCount;return console.log(`${k()} ${t} ${p.green("\u2713")} Plan validated: ${p.bold(P)} stories \u2192 ${p.green("queued")}`),await R(e,`${E} Plan validated: ${P} stories. Task queued for execution.`),await Pe(e,"complete",r,"Planning complete",0,0),!0}catch(h){let P=h,d=P.response?.data?.error||P.response?.data?.detail||String(h),w=P.response?.status?` (${P.response.status})`:"";return console.error(`${k()} ${t} ${p.red("\u2717")} Server validation failed${w}: ${d.substring(0,100)}`),await R(e,`${E} Server-side plan validation failed${w}: ${d.substring(0,200)}`,"error","error"),!1}}async function dn(e,o,n,t,r){try{let s=(await M.post("/api/agent/plan-result",{taskId:e,rawOutput:o,agentId:n})).data.storyCount;return console.log(`${k()} ${t} ${p.green("\u2713")} Plan validated (server-side): ${p.bold(s)} stories \u2192 ${p.green("queued")}`),await R(e,`${E} Plan validated: ${s} stories. Task queued for execution.`),await Pe(e,"complete",r,"Planning complete",0,0),!0}catch(i){let a=i.response?.data?.detail||String(i);return console.error(`${k()} ${t} ${p.red("\u2717")} Validation failed: ${a.substring(0,100)}`),await R(e,`${E} Plan validation failed: ${a.substring(0,200)}`,"error","error"),!1}}import f from"chalk";import{spawn as jo,execSync as ze}from"child_process";import*as Be from"path";import*as ae from"fs";import*as qe from"os";function O(){return f.dim(new Date().toLocaleTimeString())}var un=[[/(:\/\/[^:/?#]+:)[^@]+(@)/g,"$1***$2"],[/\b(ghp_|gho_|ghs_|github_pat_)[A-Za-z0-9_]+/g,"$1***"],[/\bglpat-[A-Za-z0-9\-_]+/g,"glpat-***"],[/\b(AKIA)[A-Z0-9]{16}\b/g,"$1***"],[/(Bearer\s+)[A-Za-z0-9._\-]+/gi,"$1***"],[/(x-api-key:\s*)[^\s,'"]+/gi,"$1***"]];function Ve(e){let o=e;for(let[n,t]of un)o=o.replace(n,t);return o}var H=new Map;function gn(){if(process.env.WSL_DISTRO_NAME||process.env.WSL_INTEROP)return!0;try{return ae.readFileSync("/proc/version","utf-8").toLowerCase().includes("microsoft")}catch{return!1}}var Ye=gn(),Bo=Ye||process.platform==="darwin"||process.platform==="win32";function Vo(){if(!Ye)return null;try{let e=ze("hostname -I",{encoding:"utf-8"}).trim().split(/\s+/)[0];if(e&&/^\d+\.\d+\.\d+\.\d+$/.test(e))return e}catch{}return null}function zo(e){if(!Ye)return e;let o=e.match(/^\/mnt\/([a-zA-Z])\/(.*)$/);return o?`${o[1].toUpperCase()}:/${o[2]}`:e}function qo(){let e=Be.join(qe.homedir(),".claude");if(ae.existsSync(e))return e;if(Ye){let o="/mnt/c/Users";if(ae.existsSync(o))try{for(let n of ae.readdirSync(o)){if(["Public","Default","Default User","All Users"].includes(n))continue;let t=Be.join(o,n,".claude");if(ae.existsSync(t))return t}}catch{}}return null}var Go=0;function Yo(e){let o=e.match(/^(\d+\.dkr\.ecr\.[a-z0-9-]+\.amazonaws\.com)/);return o?o[1]:null}function pn(e){let o=e.match(/\.ecr\.([a-z0-9-]+)\.amazonaws\.com/);return o?o[1]:"us-east-1"}function Ho(e){if(Date.now()<Go)return!0;let o=pn(e);try{return ze(`aws ecr get-login-password --region ${o} | docker login --username AWS --password-stdin ${e}`,{stdio:"pipe",timeout:3e4}),Go=Date.now()+660*60*1e3,!0}catch{return!1}}function Jo(e,o,n){if(n?.scmToken)return n.scmToken;switch(e){case"bitbucket":return o.bitbucketToken;case"gitlab":return o.gitlabToken;default:return o.githubToken}}function mn(e){let o=e.jiraFields;if(!o)return!1;let n=o.labels;if(Array.isArray(n)&&n.some(i=>typeof i=="string"&&i.toLowerCase()==="self-review"))return!0;let r=o.issue?.labels;return!!(Array.isArray(r)&&r.some(i=>typeof i=="string"?i.toLowerCase()==="self-review":i&&typeof i=="object"&&"name"in i?(i.name||"").toLowerCase()==="self-review":!1))}async function Xo(e,o,n,t){let r=f.cyan(e.id.slice(0,8));if(H.has(e.id)){console.log(`${O()} ${r} ${f.dim("Already running, skipping")}`);return}let i=`workermill-${e.id.slice(0,8)}`,a=["run","--rm",...!o.workerImage.includes("/")?[]:["--pull","always"],"--name",i],c=Math.round(qe.totalmem()/(1024*1024*1024));if(c<=16?a.push("--memory","6g","--memory-swap","10g","--cpus","2"):(c<=32,a.push("--memory","6g","--memory-swap","12g","--cpus","4")),Bo){let y=Vo();a.push(`--add-host=host.docker.internal:${y||"host-gateway"}`)}else a.push("--network","host");let m=e.workerProvider||"anthropic",$=qo();if(!$&&m==="anthropic"){console.error(`${O()} ${r} ${f.red("\u2717")} Claude credentials not found. Run 'claude' and complete the sign-in flow.`);return}if($){let y=Be.join($,".credentials.json");try{ae.chmodSync(y,438)}catch{}let F=zo($);a.push("-v",`${F}:/home/worker/.claude`)}else console.log(`${O()} ${r} ${f.dim("Skipping Claude mount (non-Anthropic worker)")}`);let g=o.apiUrl.replace(/localhost|127\.0\.0\.1/,"host.docker.internal"),h=e.scmProvider||"github",P=Jo(h,o,t),d=t?.githubToken||o.githubToken,w=h==="bitbucket"&&t?.scmToken||o.bitbucketToken,T=h==="gitlab"&&t?.scmToken||o.gitlabToken,x={NODE_OPTIONS:"--max-old-space-size=3072",EPIC_MODE:"true",EXECUTION_MODE:"local",TASK_ID:e.id,ORG_ID:e.orgId||"",JIRA_ISSUE_KEY:e.jiraIssueKey||"",JIRA_SUMMARY:e.summary||"",JIRA_DESCRIPTION:e.description||"",TASK_SUMMARY:e.summary||"",TASK_DESCRIPTION:e.description||"",WORKER_PERSONA:e.workerPersona||"",RETRY_NUMBER:String(e.retryCount??0),TICKET_KEY:e.jiraIssueKey||"",API_BASE_URL:g,ORG_API_KEY:o.apiKey,SCM_PROVIDER:h,SCM_TOKEN:P,SCM_BASE_URL:t?.scmBaseUrl||"",GITHUB_TOKEN:d,GH_TOKEN:d,GITHUB_REVIEWER_TOKEN:t?.githubReviewerToken||"",BITBUCKET_TOKEN:w,BITBUCKET_USERNAME:t?.bitbucketUsername||"x-token-auth",GITLAB_TOKEN:T,TARGET_REPO:e.githubRepo||"",GITHUB_REPO:e.githubRepo||"",WORKER_MODEL:e.workerModel||String(n.defaultWorkerModel||""),CLAUDE_MODEL:e.workerModel||String(n.defaultWorkerModel||""),JIRA_BASE_URL:t?.jiraBaseUrl||"",JIRA_EMAIL:t?.jiraEmail||"",JIRA_API_TOKEN:t?.jiraApiToken||"",TICKET_SYSTEM:t?.issueTrackerProvider||"jira",LINEAR_API_KEY:t?.linearApiKey||"",AWS_ACCESS_KEY_ID:t?.customerAwsAccessKeyId||"",AWS_SECRET_ACCESS_KEY:t?.customerAwsSecretAccessKey||"",AWS_DEFAULT_REGION:t?.customerAwsRegion||"",AWS_REGION:t?.customerAwsRegion||"",CUSTOMER_AWS_ROLE_ARN:t?.customerAwsRoleArn||"",CUSTOMER_AWS_EXTERNAL_ID:t?.customerAwsExternalId||"",CUSTOMER_AWS_REGION:t?.customerAwsRegion||"",MANAGER_PROVIDER:t?.managerProvider||"anthropic",MANAGER_MODEL:t?.managerModelId||"",BITBUCKET_EMAIL:t?.bitbucketEmail||"",DEPLOYMENT_ENABLED:e.deploymentEnabled||e.parentTaskId?"true":"false",PRD_CHILD_TASK:e.parentTaskId?"true":"false",IMPROVEMENT_ENABLED:e.improvementEnabled?"true":"false",QUALITY_GATE_BYPASS:e.qualityGateBypass?"true":"false",STANDARD_SDK_MODE:e.standardSdkMode?"true":"false",MAX_REVIEW_REVISIONS:String(n.maxReviewRevisions??3),CODEBASE_INDEXING_ENABLED:n.codebaseIndexingEnabled===!0?"true":"false",EXISTING_PR_URL:e.githubPrUrl||"",EXISTING_PR_NUMBER:e.githubPrNumber?String(e.githubPrNumber):"",PARENT_TASK_ID:e.parentTaskId||e.id,PARENT_JIRA_KEY:e.jiraIssueKey&&/-S\d+$/.test(e.jiraIssueKey)&&e.jiraFields?.parentJiraKey||"",TARGET_BRANCH:e.jiraFields?.targetBranch||"",STORY_BRANCH:e.jiraFields?.storyBranch||"",TASK_NOTES:e.taskNotes||"",TARGET_FILES:JSON.stringify(e.jiraFields?.targetFiles||[]),REFERENCE_FILES:JSON.stringify(e.jiraFields?.referenceFiles||[]),PIPELINE_VERSION:e.jiraFields?.pipelineVersion||e.pipelineVersion||"",V2_STEP_INPUT:e.jiraFields?.v2StepInput?JSON.stringify(e.jiraFields.v2StepInput):"",EXECUTION_MODE_SETTING:e.jiraFields?.executionMode||"autonomous",ANTHROPIC_API_KEY:$?"":t?.anthropicApiKey||process.env.ANTHROPIC_API_KEY||"",WORKER_PROVIDER:e.workerProvider||"anthropic",OPENAI_API_KEY:t?.openaiApiKey||"",GOOGLE_API_KEY:t?.googleApiKey||"",GOOGLE_GENERATIVE_AI_API_KEY:t?.googleApiKey||"",OLLAMA_HOST:t?.ollamaBaseUrl||"",OLLAMA_CONTEXT_WINDOW:t?.ollamaContextWindow?String(t.ollamaContextWindow):"",VLLM_BASE_URL:t?.vllmBaseUrl||"",BLOCKER_MAX_AUTO_RETRIES:String(n.blockerMaxAutoRetries??3),BLOCKER_AUTO_RETRY_ENABLED:n.blockerAutoRetryEnabled!==!1?"true":"false",PUSH_AFTER_COMMIT:n.pushAfterCommit!==!1?"true":"false",GRACEFUL_SHUTDOWN_ENABLED:n.gracefulShutdownEnabled!==!1?"true":"false",MAX_PARALLEL_EXPERTS:String(n.maxParallelExperts??4),REVIEW_ENABLED:e.skipManagerReview===!1?"true":"false",SELF_REVIEW_ENABLED:mn(e)||n.selfReviewEnabled!==!1?"true":"false"};for(let[y,F]of Object.entries(x))F!==""&&a.push("-e",`${y}=${F}`);let W=o.workerImage,l=Yo(W);l&&Ho(l),a.push(W);let A=e.skipManagerReview===!1;console.log(`${O()} ${r} ${f.dim("Starting container")} ${f.yellow(i)}`),console.log(`${O()} ${r} ${f.dim(` skipManagerReview=${e.skipManagerReview} \u2192 REVIEW_ENABLED=${A}`)}`),console.log(`${O()} ${r} ${f.dim(` model=${e.workerModel} repo=${e.githubRepo}`)}`),console.log(`${O()} ${r} ${f.dim(` totalRamGB=${c} docker args:`)} ${a.slice(0,10).join(" ")}`);let b=jo("docker",a,{stdio:["ignore","pipe","pipe"],detached:!1});if(!b.pid){console.error(`${O()} ${r} ${f.red("\u2717")} Failed to spawn container`);return}let K={taskId:e.id,containerName:i,process:b,startedAt:new Date,status:"running",resultEmitted:!1};H.set(e.id,K),b.stdout?.on("data",y=>{let F=y.toString().split(`
|
|
14
|
-
`).filter(
|
|
15
|
-
`).filter(
|
|
16
|
-
`).filter(
|
|
17
|
-
`).filter(b=>b.trim());for(let b of A)console.log(`${O()} ${t} ${f.magenta("MGR")} ${f.red(Ve(b))}`)}),x.on("exit",l=>{W.status=l===0?"completed":"failed";let A=Math.round((Date.now()-W.startedAt.getTime())/1e3),b=l===0?f.green("\u2713"):f.red("\u2717"),K=l===0?f.green("completed"):f.red(`failed (exit ${l})`);console.log(`${O()} ${t} ${f.magenta("MGR")} ${b} Manager ${e.managerAction} ${K} ${f.dim(`(${A}s)`)}`),setTimeout(()=>H.delete(r),6e4)}),x.on("error",l=>{W.status="failed",console.error(`${O()} ${t} ${f.magenta("MGR")} ${f.red("\u2717")} Container error: ${l.message}`)})}import{spawn as tt}from"child_process";import Le from"chalk";async function xe(){return console.log(Le.cyan(" Updating @workermill/agent...")),new Promise(e=>{let o=tt("npm",["install","-g","@workermill/agent@latest"],{stdio:"inherit",shell:!0});o.on("error",n=>{console.error(Le.red(` Update failed: ${n.message}`)),e(!1)}),o.on("close",n=>{n===0?(console.log(Le.green(" Update successful.")),e(!0)):(console.error(Le.red(` Update failed with exit code ${n}`)),e(!1))})})}function He(){console.log(Le.cyan(" Restarting agent...")),tt(process.argv[0],process.argv.slice(1),{stdio:"inherit",detached:!0}).unref(),process.exit(0)}var Ae=new Set,ke=new Set,Je=null,nt=!1,Xe=!1;function U(){return S.dim(new Date().toLocaleTimeString())}async function fn(){if(Je)return Je;try{return Je=(await M.get("/api/agent/config")).data,Je}catch{return console.error(`${U()} ${S.red("\u2717")} Failed to fetch org config`),{}}}async function rt(e){if(!Xe)try{let o=await M.get("/api/agent/poll",{params:{agentId:e.agentId}}),n=o.data.tasks;if(n.length===0)return;for(let r of n)r.status==="planning"&&!Ae.has(r.id)?await hn(r,e):r.status==="queued"&&await $n(r,e);let t=o.data.managerTasks;if(t&&t.length>0)for(let r of t)ke.has(r.id)||await wn(r,e)}catch(o){let n=o,t=Ae.size>0||po()>0||ke.size>0;n.response?.status===401?t||console.error(`${U()} ${S.red("\u2717")} Authentication failed. Check your API key.`):t||console.warn(`${U()} ${S.yellow("\u26A0")} Poll error: ${n.message||String(o)}`)}}async function hn(e,o){let n,t;try{let s=await M.post("/api/agent/claim",{taskId:e.id,agentId:o.agentId});if(!s.data.claimed)return;n=s.data.credentials,t=s.data.task}catch{return}let r=S.cyan(e.id.slice(0,8));if(t&&(t.retryCount??0)>0&&t.executionPlanV2!=null){console.log(),console.log(`${U()} ${S.magenta("\u25C6 RESUME")} ${r} ${e.summary.substring(0,60)}`),console.log(`${U()} ${r} Retry #${t.retryCount} with existing plan \u2014 skipping planning`),Ae.add(e.id);try{await M.post("/api/agent/resume-plan",{taskId:e.id,agentId:o.agentId}),console.log(`${U()} ${r} ${S.green("\u2713")} Resumed with existing plan \u2192 ${S.green("queued")}`)}catch(s){let a=s,c=a.response?.data?.error||a.message||String(s);console.error(`${U()} ${r} ${S.red("\u2717")} Resume failed: ${c}`)}Ae.delete(e.id);return}console.log(),console.log(`${U()} ${S.magenta("\u25C6 PLANNING")} ${r} ${e.summary.substring(0,60)}`),Ae.add(e.id),Ko(e,o,n).then(s=>{console.log(s?`${U()} ${S.green("\u2713")} Planning complete for ${r}`:`${U()} ${S.red("\u2717")} Planning failed for ${r}`)}).catch(s=>console.error(`${U()} ${S.red("\u2717")} Planning error for ${r}:`,s.message||s)).finally(()=>Ae.delete(e.id))}async function $n(e,o){if(po()>=o.maxWorkers)return;let t;try{if(t=(await M.post("/api/agent/claim",{taskId:e.id,agentId:o.agentId})).data,!t.claimed)return}catch{return}try{await M.post("/api/agent/started",{taskId:e.id,agentId:o.agentId})}catch{let m=S.cyan(e.id.slice(0,8));console.warn(`${U()} ${S.yellow("\u26A0")} Failed to report started for ${m}`)}let r=S.cyan(e.id.slice(0,8));console.log(),console.log(`${U()} ${S.blue("\u25B6 EXECUTING")} ${r} ${e.summary.substring(0,60)}`);let i=await fn(),s=t.task||{id:e.id,summary:e.summary,description:e.description,jiraIssueKey:e.jiraIssueKey,workerModel:e.workerModel,workerProvider:e.workerProvider,githubRepo:e.githubRepo,scmProvider:e.scmProvider,skipManagerReview:e.skipManagerReview,deploymentEnabled:e.deploymentEnabled,improvementEnabled:e.improvementEnabled,qualityGateBypass:e.qualityGateBypass,standardSdkMode:e.standardSdkMode,parentTaskId:e.parentTaskId,taskNotes:e.taskNotes,githubPrUrl:e.githubPrUrl,githubPrNumber:e.githubPrNumber,executionPlanV2:e.executionPlanV2,jiraFields:e.jiraFields||{}},a=t.credentials;Xo(s,o,i,a).catch(c=>console.error(`${U()} ${S.red("\u2717")} Spawn failed for ${r}:`,c.message||c))}async function wn(e,o){let n=S.cyan(e.id.slice(0,8));ke.add(e.id);try{let t=await M.post("/api/agent/claim-manager",{taskId:e.id,agentId:o.agentId,action:e.managerAction});if(!t.data.claimed){ke.delete(e.id);return}let r=t.data.task,i=t.data.credentials||{},s=e.managerAction==="analyze_logs"?S.yellow("LOG ANALYSIS"):S.yellow("PR REVIEW");console.log(),console.log(`${U()} ${S.magenta("\u25C6 MANAGER")} ${s} ${n} ${e.summary.substring(0,60)}`);let a={id:e.id,summary:r?.summary||e.summary,description:r?.description||e.description,jiraIssueKey:r?.jiraIssueKey||e.jiraIssueKey,githubRepo:r?.githubRepo||e.githubRepo,scmProvider:r?.scmProvider||e.scmProvider,githubPrUrl:r?.githubPrUrl||e.githubPrUrl,githubPrNumber:r?.githubPrNumber||e.githubPrNumber,managerAction:e.managerAction};ot(a,o,i).then(()=>{console.log(`${U()} ${n} ${S.magenta("MGR")} ${S.green("\u2713")} Manager ${e.managerAction} dispatched`)}).catch(c=>{console.error(`${U()} ${n} ${S.magenta("MGR")} ${S.red("\u2717")} Manager spawn failed:`,c.message||c)}).finally(()=>ke.delete(e.id))}catch(t){ke.delete(e.id);let r=t;console.error(`${U()} ${n} ${S.red("\u2717")} Failed to claim manager task:`,r.message||String(t))}}var Qe=null,Ze=null;function it(){Qe&&(clearInterval(Qe),Qe=null),Ze&&(clearInterval(Ze),Ze=null)}function st(e){console.log(` ${S.dim("Polling every")} ${e.pollIntervalMs/1e3}s ${S.dim("\xB7 waiting for tasks...")}`),rt(e),Qe=setInterval(()=>rt(e),e.pollIntervalMs)}function at(e){Ze=setInterval(async()=>{let o=Qo(),n=Array.from(Ae),t=Array.from(ke),r=[...o,...n,...t];try{let i=await M.post("/api/agent/heartbeat",{agentId:e.agentId,activeTasks:r,agentVersion:Y}),s=i.data?.cancelledTasks;if(s&&s.length>0)for(let $ of s)Zo($);let{updateAvailable:a,updateRequired:c,latestVersion:m}=i.data??{};c&&!Xe?(Xe=!0,console.log(`${U()} ${S.red("\u26A0 Agent update required")} (current: ${Y}, required: ${m})`),console.log(`${U()} ${S.yellow("Refusing new tasks until updated.")}`),await xe()?He():(console.log(`${U()} ${S.red("Auto-update failed.")} Run: npm install -g @workermill/agent@latest`),Xe=!1)):a&&!nt&&!c&&(nt=!0,console.log(`${U()} ${S.yellow(`Update available: ${m}`)} (current: ${Y}). Run: workermill-agent update`))}catch{}},e.heartbeatIntervalMs)}te();async function mo(e){console.log(),console.log(G.bold.cyan(" WorkerMill Remote Agent")),console.log(G.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(),Ke(e.apiUrl,e.apiKey);try{let o=await M.get("/api/agent/config"),n=o.data.maxConcurrentWorkers;n&&typeof n=="number"&&(e.maxWorkers=n),o.data.workerImageUrl&&(e.workerImage=o.data.workerImageUrl),console.log(` ${G.green("\u25CF")} Connected to ${G.cyan(e.apiUrl)}`),console.log(` ${G.dim("Agent:")} ${e.agentId}`),console.log(` ${G.dim("Workers:")} ${G.yellow(String(e.maxWorkers))} parallel`),console.log(` ${G.dim("Image:")} ${e.workerImage}`),console.log(` ${G.dim("SCM:")} ${o.data.scmProvider}`),console.log(` ${G.dim("Model:")} ${G.yellow(o.data.defaultWorkerModel)}`),console.log()}catch(o){let n=o;throw n.response?.status===401?new Error("Authentication failed. Check your API key."):new Error(`Failed to connect to WorkerMill API: ${n.message||String(o)}`)}try{let o=await M.post("/api/agent/register",{agentId:e.agentId,maxWorkers:e.maxWorkers,agentVersion:Y}),{updateAvailable:n,updateRequired:t,latestVersion:r}=o.data;t?(console.log(G.red(` \u26A0 Agent update required (current: ${Y}, required: ${r})`)),await xe()?He():console.log(G.yellow(" Auto-update failed. Run: npm install -g @workermill/agent@latest"))):n&&console.log(G.yellow(` Update available: ${r} (current: ${Y}). Run: workermill-agent update`))}catch{}return st(e),at(e),console.log(G.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(` ${G.green("\u25CF")} Agent is running. ${G.dim("Press Ctrl+C to stop.")}`),console.log(),async()=>{console.log(),console.log(G.dim(" Shutting down...")),it();try{await M.post("/api/agent/deregister",{agentId:e.agentId})}catch{}await et(),console.log(` ${G.red("\u25CF")} Agent stopped.`)}}var yn=typeof process<"u"&&process.argv[1]&&(process.argv[1].endsWith("/index.ts")||process.argv[1].endsWith("/index.js"));if(yn){try{await import("dotenv/config")}catch{}let{loadConfig:e,validatePrerequisites:o}=await Promise.resolve().then(()=>(te(),Io)),n=e();o(),console.log(G.dim(" Prerequisites validated."));let t=await mo(n);process.on("SIGINT",async()=>{await t(),process.exit(0)}),process.on("SIGTERM",async()=>{await t(),process.exit(0)})}async function fo(e){kn(oe())||(console.log(L.red("No configuration found.")),console.log(`Run ${L.cyan("workermill-agent setup")} first.`),process.exit(1));let o=fe(),t=to(o.workerImage).filter(d=>!d.ok),r=t.find(d=>d.name==="Worker image"),i=new Set(["Claude CLI","Claude auth"]),s=t.filter(d=>d.name!=="Worker image"&&!i.has(d.name)),a=t.filter(d=>i.has(d.name));if(s.length>0){console.log(L.red("Prerequisites check failed:"));for(let d of s)console.log(L.red(` \u2717 ${d.name}: ${d.detail}`));process.exit(1)}if(a.length>0)for(let d of a)console.log(L.yellow(` \u26A0 ${d.name}: ${d.detail} (required for Anthropic provider)`));if(r){console.log(L.yellow(` Worker image not found locally. Pulling ${o.workerImage}...`));let{spawnSync:d}=await import("child_process");d("docker",["pull",o.workerImage],{stdio:"inherit",timeout:6e5}).status!==0&&(console.log(L.red(" Failed to pull worker image.")),process.exit(1)),console.log(L.green(" \u2713 Worker image pulled"))}if(e.detach){let d=Ce(),w=ce();console.log(L.dim("Starting agent in background...")),console.log(L.dim(` Logs: ${d}`)),console.log(L.dim(` PID: ${w}`));let T=En(d,"a"),x=An("workermill-agent",["start"],{detached:!0,stdio:["ignore",T,T],shell:!0});x.pid?(lt(w,String(x.pid),"utf-8"),x.unref(),console.log(L.green(`Agent started (PID: ${x.pid})`)),console.log(`Check status with: ${L.cyan("workermill-agent status")}`)):(console.log(L.red("Failed to start agent in background.")),process.exit(1));return}let c=Ce(),m=Sn(c,{flags:"a"}),$=process.stdout.write.bind(process.stdout),g=process.stderr.write.bind(process.stderr);process.stdout.write=(d,...w)=>(m.write(d),$(d,...w)),process.stderr.write=(d,...w)=>(m.write(d),g(d,...w)),console.log(),console.log(L.bold.cyan(" WorkerMill Remote Agent")),console.log(L.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log();let h=Math.round(In()/(1024*1024*1024));h<8?(console.log(L.red(` \u2717 Insufficient RAM: ${h} GB (minimum 8 GB, recommended 16 GB)`)),process.exit(1)):h<16&&console.log(L.yellow(` \u26A0 RAM: ${h} GB (below recommended 16 GB \u2014 workers may be slow)`));let P=De();console.log(L.dim(` Agent: ${o.agentId}`)),console.log(L.dim(` Version: ${Y}`)),console.log(L.dim(` Image: ${o.workerImage}`)),console.log();try{let d=await mo(o);lt(ce(),String(process.pid),"utf-8");let w=!1,T=async()=>{w&&(console.log(L.red(`
|
|
18
|
-
Force exit.`)),process.exit(1))
|
|
19
|
-
Cleanup timed out. Force exit.`)),process.exit(1)},1e4);
|
|
20
|
-
`);console.log(
|
|
13
|
+
`+_;let M=`${E} Critic rejected (score: ${C.score}/100, threshold: ${te}). Re-planning with feedback...`;if(console.log(`${I()} ${t} ${p.yellow("\u26A0")} ${M}`),await R(e.id,M),C.risks.length>0){let Q=`${E} Critic risks: ${C.risks.join("; ")}`;console.log(`${I()} ${t} ${p.dim(Q)}`),await R(e.id,Q)}if(C.suggestions&&C.suggestions.length>0){let Q=`${E} Critic suggestions: ${C.suggestions.join("; ")}`;console.log(`${I()} ${t} ${p.dim(Q)}`),await R(e.id,Q)}}else{let _=`${E} Critic rejected after ${ue} iterations (best score: ${O}/100, threshold: ${te})`;if(console.error(`${I()} ${t} ${p.red("\u2717")} ${_}`),await R(e.id,_,"error","error"),C.risks.length>0){let M=`${E} Final risks: ${C.risks.join("; ")}`;console.error(`${I()} ${t} ${M}`),await R(e.id,M,"error","error")}if(C.suggestions&&C.suggestions.length>0){let M=`${E} Suggestions: ${C.suggestions.join("; ")}`;console.error(`${I()} ${t} ${M}`),await R(e.id,M,"error","error")}}}let q=50;if(T&&O>=q){let x=Math.round((Date.now()-A)/1e3),ae=`${E} Best-plan fallback: posting plan with score ${O}/100 (below ${te} threshold, above ${q} minimum)`;console.log(`${I()} ${t} ${p.yellow("\u26A0")} ${ae}`),await R(e.id,ae);let X=Date.now()-A;if(await go(e.id,T,o.agentId,t,x,O,[`Best-plan fallback: critic rejected after ${ue} iterations`],y,L,X,ue))return!0;console.log(`${I()} ${t} ${p.yellow("\u26A0")} ${E} Fallback post rejected by server, reporting plan-failed`),await R(e.id,`${E} Fallback plan rejected by server \u2014 reporting failure`)}try{let x=T&&O>=q?`Best-plan fallback rejected by server after ${ue} iterations (best score: ${O}/100)`:`Critic rejected after ${ue} iterations (best score: ${O}/100, threshold: ${te}, fallback minimum: ${q})`;await N.post("/api/agent/plan-failed",{taskId:e.id,agentId:o.agentId,reason:x,criticHistory:y})}catch{}return!1}finally{if(await sn(),l)try{po(`rm -rf "${l}"`,{stdio:"ignore"})}catch{}}}async function go(e,o,n,t,r,i,s,a,c,h,w){let m=vo(o);try{let P=(await N.post("/api/agent/plan-result",{taskId:e,rawOutput:m,agentId:n,criticScore:i,criticRisks:s,criticHistory:a,criticIterations:w,fileCapTruncations:c,planningDurationMs:h})).data.storyCount;return console.log(`${I()} ${t} ${p.green("\u2713")} Plan validated: ${p.bold(P)} stories \u2192 ${p.green("queued")}`),await R(e,`${E} Plan validated: ${P} stories. Task queued for execution.`),await ve(e,"complete",r,"Planning complete",0,0),!0}catch(g){let P=g,d=P.response?.data?.error||P.response?.data?.detail||String(g),$=P.response?.status?` (${P.response.status})`:"";return console.error(`${I()} ${t} ${p.red("\u2717")} Server validation failed${$}: ${d.substring(0,100)}`),await R(e,`${E} Server-side plan validation failed${$}: ${d.substring(0,200)}`,"error","error"),!1}}async function un(e,o,n,t,r){try{let s=(await N.post("/api/agent/plan-result",{taskId:e,rawOutput:o,agentId:n})).data.storyCount;return console.log(`${I()} ${t} ${p.green("\u2713")} Plan validated (server-side): ${p.bold(s)} stories \u2192 ${p.green("queued")}`),await R(e,`${E} Plan validated: ${s} stories. Task queued for execution.`),await ve(e,"complete",r,"Planning complete",0,0),!0}catch(i){let a=i.response?.data?.detail||String(i);return console.error(`${I()} ${t} ${p.red("\u2717")} Validation failed: ${a.substring(0,100)}`),await R(e,`${E} Plan validation failed: ${a.substring(0,200)}`,"error","error"),!1}}import f from"chalk";import{spawn as Bo,execSync as qe}from"child_process";import*as Ve from"path";import*as se from"fs";import*as Ye from"os";function D(){return f.dim(new Date().toLocaleTimeString())}var gn=[[/(:\/\/[^:/?#]+:)[^@]+(@)/g,"$1***$2"],[/\b(ghp_|gho_|ghs_|github_pat_)[A-Za-z0-9_]+/g,"$1***"],[/\bglpat-[A-Za-z0-9\-_]+/g,"glpat-***"],[/\b(AKIA)[A-Z0-9]{16}\b/g,"$1***"],[/(Bearer\s+)[A-Za-z0-9._\-]+/gi,"$1***"],[/(x-api-key:\s*)[^\s,'"]+/gi,"$1***"]];function ze(e){let o=e;for(let[n,t]of gn)o=o.replace(n,t);return o}var J=new Map;function pn(){if(process.env.WSL_DISTRO_NAME||process.env.WSL_INTEROP)return!0;try{return se.readFileSync("/proc/version","utf-8").toLowerCase().includes("microsoft")}catch{return!1}}var He=pn(),Vo=He||process.platform==="darwin"||process.platform==="win32";function zo(){if(!He)return null;try{let e=qe("hostname -I",{encoding:"utf-8"}).trim().split(/\s+/)[0];if(e&&/^\d+\.\d+\.\d+\.\d+$/.test(e))return e}catch{}return null}function qo(e){if(!He)return e;let o=e.match(/^\/mnt\/([a-zA-Z])\/(.*)$/);return o?`${o[1].toUpperCase()}:/${o[2]}`:e}function Yo(){let e=Ve.join(Ye.homedir(),".claude");if(se.existsSync(e))return e;if(He){let o="/mnt/c/Users";if(se.existsSync(o))try{for(let n of se.readdirSync(o)){if(["Public","Default","Default User","All Users"].includes(n))continue;let t=Ve.join(o,n,".claude");if(se.existsSync(t))return t}}catch{}}return null}var jo=0;function Ho(e){let o=e.match(/^(\d+\.dkr\.ecr\.[a-z0-9-]+\.amazonaws\.com)/);return o?o[1]:null}function mn(e){let o=e.match(/\.ecr\.([a-z0-9-]+)\.amazonaws\.com/);return o?o[1]:"us-east-1"}function Jo(e){if(Date.now()<jo)return!0;let o=mn(e);try{return qe(`aws ecr get-login-password --region ${o} | docker login --username AWS --password-stdin ${e}`,{stdio:"pipe",timeout:3e4}),jo=Date.now()+660*60*1e3,!0}catch{return!1}}function Xo(e,o,n){if(n?.scmToken)return n.scmToken;switch(e){case"bitbucket":return o.bitbucketToken;case"gitlab":return o.gitlabToken;default:return o.githubToken}}function fn(e){let o=e.jiraFields;if(!o)return!1;let n=o.labels;if(Array.isArray(n)&&n.some(i=>typeof i=="string"&&i.toLowerCase()==="self-review"))return!0;let r=o.issue?.labels;return!!(Array.isArray(r)&&r.some(i=>typeof i=="string"?i.toLowerCase()==="self-review":i&&typeof i=="object"&&"name"in i?(i.name||"").toLowerCase()==="self-review":!1))}async function Qo(e,o,n,t){let r=f.cyan(e.id.slice(0,8));if(J.has(e.id)){console.log(`${D()} ${r} ${f.dim("Already running, skipping")}`);return}let i=`workermill-${e.id.slice(0,8)}`,a=["run","--rm",...!o.workerImage.includes("/")?[]:["--pull","always"],"--name",i],c=Math.round(Ye.totalmem()/(1024*1024*1024));if(c<=16?a.push("--memory","6g","--memory-swap","10g","--cpus","2"):(c<=32,a.push("--memory","6g","--memory-swap","12g","--cpus","4")),Vo){let y=zo();a.push(`--add-host=host.docker.internal:${y||"host-gateway"}`)}else a.push("--network","host");let h=e.workerProvider||"anthropic",w=Yo();if(!w&&h==="anthropic"){console.error(`${D()} ${r} ${f.red("\u2717")} Claude credentials not found. Run 'claude' and complete the sign-in flow.`);return}if(w){let y=Ve.join(w,".credentials.json");try{se.chmodSync(y,438)}catch{}let L=qo(w);a.push("-v",`${L}:/home/worker/.claude`)}else console.log(`${D()} ${r} ${f.dim("Skipping Claude mount (non-Anthropic worker)")}`);let m=o.apiUrl.replace(/localhost|127\.0\.0\.1/,"host.docker.internal"),g=e.scmProvider||"github",P=Xo(g,o,t),d=t?.githubToken||o.githubToken,$=g==="bitbucket"&&t?.scmToken||o.bitbucketToken,v=g==="gitlab"&&t?.scmToken||o.gitlabToken,A={NODE_OPTIONS:"--max-old-space-size=3072",EPIC_MODE:"true",EXECUTION_MODE:"local",TASK_ID:e.id,ORG_ID:e.orgId||"",JIRA_ISSUE_KEY:e.jiraIssueKey||"",JIRA_SUMMARY:e.summary||"",JIRA_DESCRIPTION:e.description||"",TASK_SUMMARY:e.summary||"",TASK_DESCRIPTION:e.description||"",WORKER_PERSONA:e.workerPersona||"",RETRY_NUMBER:String(e.retryCount??0),TICKET_KEY:e.jiraIssueKey||"",API_BASE_URL:m,ORG_API_KEY:o.apiKey,SCM_PROVIDER:g,SCM_TOKEN:P,SCM_BASE_URL:t?.scmBaseUrl||"",GITHUB_TOKEN:d,GH_TOKEN:d,GITHUB_REVIEWER_TOKEN:t?.githubReviewerToken||"",BITBUCKET_TOKEN:$,BITBUCKET_USERNAME:t?.bitbucketUsername||"x-token-auth",GITLAB_TOKEN:v,TARGET_REPO:e.githubRepo||"",GITHUB_REPO:e.githubRepo||"",WORKER_MODEL:e.workerModel||String(n.defaultWorkerModel||""),CLAUDE_MODEL:e.workerModel||String(n.defaultWorkerModel||""),JIRA_BASE_URL:t?.jiraBaseUrl||"",JIRA_EMAIL:t?.jiraEmail||"",JIRA_API_TOKEN:t?.jiraApiToken||"",TICKET_SYSTEM:t?.issueTrackerProvider||"jira",LINEAR_API_KEY:t?.linearApiKey||"",AWS_ACCESS_KEY_ID:t?.customerAwsAccessKeyId||"",AWS_SECRET_ACCESS_KEY:t?.customerAwsSecretAccessKey||"",AWS_DEFAULT_REGION:t?.customerAwsRegion||"",AWS_REGION:t?.customerAwsRegion||"",CUSTOMER_AWS_ROLE_ARN:t?.customerAwsRoleArn||"",CUSTOMER_AWS_EXTERNAL_ID:t?.customerAwsExternalId||"",CUSTOMER_AWS_REGION:t?.customerAwsRegion||"",MANAGER_PROVIDER:t?.managerProvider||"anthropic",MANAGER_MODEL:t?.managerModelId||"",BITBUCKET_EMAIL:t?.bitbucketEmail||"",DEPLOYMENT_ENABLED:e.deploymentEnabled||e.parentTaskId?"true":"false",PRD_CHILD_TASK:e.parentTaskId?"true":"false",IMPROVEMENT_ENABLED:e.improvementEnabled?"true":"false",QUALITY_GATE_BYPASS:e.qualityGateBypass?"true":"false",STANDARD_SDK_MODE:e.standardSdkMode?"true":"false",MAX_REVIEW_REVISIONS:String(n.maxReviewRevisions??3),CODEBASE_INDEXING_ENABLED:n.codebaseIndexingEnabled===!0?"true":"false",EXISTING_PR_URL:e.githubPrUrl||"",EXISTING_PR_NUMBER:e.githubPrNumber?String(e.githubPrNumber):"",PARENT_TASK_ID:e.parentTaskId||e.id,PARENT_JIRA_KEY:e.jiraIssueKey&&/-S\d+$/.test(e.jiraIssueKey)&&e.jiraFields?.parentJiraKey||"",TARGET_BRANCH:e.jiraFields?.targetBranch||"",STORY_BRANCH:e.jiraFields?.storyBranch||"",TASK_NOTES:e.taskNotes||"",TARGET_FILES:JSON.stringify(e.jiraFields?.targetFiles||[]),REFERENCE_FILES:JSON.stringify(e.jiraFields?.referenceFiles||[]),PIPELINE_VERSION:e.jiraFields?.pipelineVersion||e.pipelineVersion||"",V2_STEP_INPUT:e.jiraFields?.v2StepInput?JSON.stringify(e.jiraFields.v2StepInput):"",EXECUTION_MODE_SETTING:e.jiraFields?.executionMode||"autonomous",ANTHROPIC_API_KEY:w?"":t?.anthropicApiKey||process.env.ANTHROPIC_API_KEY||"",WORKER_PROVIDER:e.workerProvider||"anthropic",OPENAI_API_KEY:t?.openaiApiKey||"",GOOGLE_API_KEY:t?.googleApiKey||"",GOOGLE_GENERATIVE_AI_API_KEY:t?.googleApiKey||"",OLLAMA_HOST:t?.ollamaBaseUrl||"",OLLAMA_CONTEXT_WINDOW:t?.ollamaContextWindow?String(t.ollamaContextWindow):"",VLLM_BASE_URL:t?.vllmBaseUrl||"",BLOCKER_MAX_AUTO_RETRIES:String(n.blockerMaxAutoRetries??3),BLOCKER_AUTO_RETRY_ENABLED:n.blockerAutoRetryEnabled!==!1?"true":"false",PUSH_AFTER_COMMIT:n.pushAfterCommit!==!1?"true":"false",GRACEFUL_SHUTDOWN_ENABLED:n.gracefulShutdownEnabled!==!1?"true":"false",MAX_PARALLEL_EXPERTS:String(n.maxParallelExperts??4),REVIEW_ENABLED:e.skipManagerReview===!1?"true":"false",SELF_REVIEW_ENABLED:fn(e)||n.selfReviewEnabled!==!1?"true":"false"};for(let[y,L]of Object.entries(A))L!==""&&a.push("-e",`${y}=${L}`);let B=o.workerImage,l=Ho(B);l&&Jo(l),a.push(B);let b=e.skipManagerReview===!1;console.log(`${D()} ${r} ${f.dim("Starting container")} ${f.yellow(i)}`),console.log(`${D()} ${r} ${f.dim(` skipManagerReview=${e.skipManagerReview} \u2192 REVIEW_ENABLED=${b}`)}`),console.log(`${D()} ${r} ${f.dim(` model=${e.workerModel} repo=${e.githubRepo}`)}`),console.log(`${D()} ${r} ${f.dim(` totalRamGB=${c} docker args:`)} ${a.slice(0,10).join(" ")}`);let T=Bo("docker",a,{stdio:["ignore","pipe","pipe"],detached:!1});if(!T.pid){console.error(`${D()} ${r} ${f.red("\u2717")} Failed to spawn container`);return}let O={taskId:e.id,containerName:i,process:T,startedAt:new Date,status:"running",resultEmitted:!1};J.set(e.id,O),T.stdout?.on("data",y=>{let L=y.toString().split(`
|
|
14
|
+
`).filter(V=>V.trim());for(let V of L)console.log(`${D()} ${r} ${f.dim(ze(V))}`),V.includes("::result::")&&(O.resultEmitted=!0)}),T.stderr?.on("data",y=>{let L=y.toString().split(`
|
|
15
|
+
`).filter(V=>V.trim());for(let V of L)console.log(`${D()} ${r} ${f.red(ze(V))}`)}),T.on("exit",y=>{O.status=y===0?"completed":"failed";let L=Math.round((Date.now()-O.startedAt.getTime())/1e3),V=y===0?f.green("\u2713"):f.red("\u2717"),q=y===0?f.green("completed"):f.red(`failed (exit ${y})`);console.log(`${D()} ${r} ${V} Container ${q} ${f.dim(`(${L}s)`)}`),O.resultEmitted||(console.log(`${D()} ${r} ${f.yellow("\u26A0")} No ::result:: marker seen \u2014 posting fallback completion in 15s`),setTimeout(async()=>{try{let x=y===0?"completed":"failed",ae=y!==0?`Worker container exited with code ${y} without reporting completion`:void 0;(await(await fetch(`${o.apiUrl}/api/tasks/${e.id}/worker-complete`,{method:"POST",headers:{"Content-Type":"application/json","x-api-key":o.apiKey},body:JSON.stringify({exitCode:y??1,result:x,errorMessage:ae})})).json()).status==="ignored"?console.log(`${D()} ${r} ${f.dim("Fallback completion ignored (task already transitioned)")}`):console.log(`${D()} ${r} ${f.yellow("\u26A0")} Fallback completion applied: ${x}`)}catch(x){console.error(`${D()} ${r} ${f.red("\u2717")} Fallback completion failed:`,x instanceof Error?x.message:x)}},15e3)),setTimeout(()=>J.delete(e.id),9e4)}),T.on("error",y=>{O.status="failed",console.error(`${D()} ${r} ${f.red("\u2717")} Container error: ${y.message}`)})}function mo(){return Array.from(J.values()).filter(e=>e.status==="running").length}function Zo(){return Array.from(J.values()).filter(e=>e.status==="running").map(e=>e.taskId)}function et(e){let o=J.get(e);if(!o||o.status!=="running")return;let n=f.cyan(e.slice(0,8));console.log(`${D()} ${n} ${f.red("\u25A0")} Stopping container (cancelled by dashboard)`);try{qe(`docker stop ${o.containerName}`,{stdio:"ignore",timeout:15e3}),o.status="completed"}catch{}J.delete(e)}async function ot(){console.log(`${D()} ${f.dim(`Stopping ${J.size} containers...`)}`);for(let[,e]of J)if(e.status==="running")try{qe(`docker stop ${e.containerName}`,{stdio:"ignore",timeout:15e3}),e.status="completed"}catch{}J.clear()}async function tt(e,o,n){let t=f.cyan(e.id.slice(0,8)),r=`manager-${e.id}`;if(J.has(r))return;let i=`wm-manager-${e.id.slice(0,8)}-${Date.now()}`,s=Yo(),a=["run","--rm","--name",i,"--memory=4g","--cpus=2"];if(Vo){let l=zo();a.push(`--add-host=host.docker.internal:${l||"host-gateway"}`)}else a.push("--network","host");if(s){let l=qo(s);a.push("-v",`${l}:/home/worker/.claude`)}let c=o.apiUrl.replace(/localhost|127\.0\.0\.1/,"host.docker.internal"),h=e.scmProvider||"github",w=Xo(h,o,n),m=n?.githubToken||o.githubToken,g=h==="bitbucket"&&n?.scmToken||o.bitbucketToken,P=h==="gitlab"&&n?.scmToken||o.gitlabToken,d={NODE_OPTIONS:"--max-old-space-size=3072",TASK_ID:e.id,MANAGER_ACTION:e.managerAction,JIRA_ISSUE_KEY:e.jiraIssueKey||"",JIRA_SUMMARY:e.summary||"",JIRA_DESCRIPTION:e.description||"",GITHUB_REPO:e.githubRepo||"",PR_URL:e.githubPrUrl||"",PR_NUMBER:e.githubPrNumber?String(e.githubPrNumber):"",API_BASE_URL:c,ORG_API_KEY:o.apiKey,SCM_PROVIDER:h,SCM_TOKEN:w,GITHUB_TOKEN:m,GH_TOKEN:m,BITBUCKET_TOKEN:g,BITBUCKET_USERNAME:n?.bitbucketUsername||"x-token-auth",GITLAB_TOKEN:P,MANAGER_PROVIDER:n?.managerProvider||"anthropic",MANAGER_MODEL:n?.managerModelId||"",ANTHROPIC_API_KEY:s?"":n?.anthropicApiKey||process.env.ANTHROPIC_API_KEY||"",OPENAI_API_KEY:n?.openaiApiKey||"",GOOGLE_API_KEY:n?.googleApiKey||"",JIRA_BASE_URL:n?.jiraBaseUrl||"",JIRA_EMAIL:n?.jiraEmail||"",JIRA_API_TOKEN:n?.jiraApiToken||"",TICKET_SYSTEM:n?.issueTrackerProvider||"jira",LINEAR_API_KEY:n?.linearApiKey||""};for(let[l,b]of Object.entries(d))b!==""&&a.push("-e",`${l}=${b}`);let $=o.workerImage,v=Ho($);v&&Jo(v),a.push("--entrypoint","/bin/bash"),a.push($),a.push("/app/manager-entrypoint.sh"),console.log(`${D()} ${t} ${f.magenta("\u25C6 MANAGER")} Starting ${e.managerAction} container ${f.yellow(i)}`);let A=Bo("docker",a,{stdio:["ignore","pipe","pipe"],detached:!1});if(!A.pid){console.error(`${D()} ${t} ${f.red("\u2717")} Failed to spawn manager container`);return}let B={taskId:r,containerName:i,process:A,startedAt:new Date,status:"running",resultEmitted:!1};J.set(r,B),A.stdout?.on("data",l=>{let b=l.toString().split(`
|
|
16
|
+
`).filter(T=>T.trim());for(let T of b)console.log(`${D()} ${t} ${f.magenta("MGR")} ${f.dim(ze(T))}`)}),A.stderr?.on("data",l=>{let b=l.toString().split(`
|
|
17
|
+
`).filter(T=>T.trim());for(let T of b)console.log(`${D()} ${t} ${f.magenta("MGR")} ${f.red(ze(T))}`)}),A.on("exit",l=>{B.status=l===0?"completed":"failed";let b=Math.round((Date.now()-B.startedAt.getTime())/1e3),T=l===0?f.green("\u2713"):f.red("\u2717"),O=l===0?f.green("completed"):f.red(`failed (exit ${l})`);console.log(`${D()} ${t} ${f.magenta("MGR")} ${T} Manager ${e.managerAction} ${O} ${f.dim(`(${b}s)`)}`),setTimeout(()=>J.delete(r),6e4)}),A.on("error",l=>{B.status="failed",console.error(`${D()} ${t} ${f.magenta("MGR")} ${f.red("\u2717")} Container error: ${l.message}`)})}import{spawn as nt}from"child_process";import Le from"chalk";async function Me(){return console.log(Le.cyan(" Updating @workermill/agent...")),new Promise(e=>{let o=nt("npm",["install","-g","@workermill/agent@latest"],{stdio:"inherit",shell:!0});o.on("error",n=>{console.error(Le.red(` Update failed: ${n.message}`)),e(!1)}),o.on("close",n=>{n===0?(console.log(Le.green(" Update successful.")),e(!0)):(console.error(Le.red(` Update failed with exit code ${n}`)),e(!1))})})}function Je(){console.log(Le.cyan(" Restarting agent...")),nt(process.argv[0],process.argv.slice(1),{stdio:"inherit",detached:!0}).unref(),process.exit(0)}var ke=new Set,be=new Set,Xe=null,rt=!1,Qe=!1;function W(){return S.dim(new Date().toLocaleTimeString())}async function hn(){if(Xe)return Xe;try{return Xe=(await N.get("/api/agent/config")).data,Xe}catch{return console.error(`${W()} ${S.red("\u2717")} Failed to fetch org config`),{}}}async function it(e){if(!Qe)try{let o=await N.get("/api/agent/poll",{params:{agentId:e.agentId}}),n=o.data.tasks;if(n.length===0)return;for(let r of n)r.status==="planning"&&!ke.has(r.id)?await $n(r,e):r.status==="queued"&&await wn(r,e);let t=o.data.managerTasks;if(t&&t.length>0)for(let r of t)be.has(r.id)||await yn(r,e)}catch(o){let n=o,t=ke.size>0||mo()>0||be.size>0;n.response?.status===401?t||console.error(`${W()} ${S.red("\u2717")} Authentication failed. Check your API key.`):t||console.warn(`${W()} ${S.yellow("\u26A0")} Poll error: ${n.message||String(o)}`)}}async function $n(e,o){let n,t;try{let s=await N.post("/api/agent/claim",{taskId:e.id,agentId:o.agentId});if(!s.data.claimed)return;n=s.data.credentials,t=s.data.task}catch{return}let r=S.cyan(e.id.slice(0,8));if(t&&(t.retryCount??0)>0&&t.executionPlanV2!=null){console.log(),console.log(`${W()} ${S.magenta("\u25C6 RESUME")} ${r} ${e.summary.substring(0,60)}`),console.log(`${W()} ${r} Retry #${t.retryCount} with existing plan \u2014 skipping planning`),ke.add(e.id);try{await N.post("/api/agent/resume-plan",{taskId:e.id,agentId:o.agentId}),console.log(`${W()} ${r} ${S.green("\u2713")} Resumed with existing plan \u2192 ${S.green("queued")}`)}catch(s){let a=s,c=a.response?.data?.error||a.message||String(s);console.error(`${W()} ${r} ${S.red("\u2717")} Resume failed: ${c}`)}ke.delete(e.id);return}console.log(),console.log(`${W()} ${S.magenta("\u25C6 PLANNING")} ${r} ${e.summary.substring(0,60)}`),ke.add(e.id),Go(e,o,n).then(s=>{console.log(s?`${W()} ${S.green("\u2713")} Planning complete for ${r}`:`${W()} ${S.red("\u2717")} Planning failed for ${r}`)}).catch(s=>console.error(`${W()} ${S.red("\u2717")} Planning error for ${r}:`,s.message||s)).finally(()=>ke.delete(e.id))}async function wn(e,o){if(mo()>=o.maxWorkers)return;let t;try{if(t=(await N.post("/api/agent/claim",{taskId:e.id,agentId:o.agentId})).data,!t.claimed)return}catch{return}try{await N.post("/api/agent/started",{taskId:e.id,agentId:o.agentId})}catch{let h=S.cyan(e.id.slice(0,8));console.warn(`${W()} ${S.yellow("\u26A0")} Failed to report started for ${h}`)}let r=S.cyan(e.id.slice(0,8));console.log(),console.log(`${W()} ${S.blue("\u25B6 EXECUTING")} ${r} ${e.summary.substring(0,60)}`);let i=await hn(),s=t.task||{id:e.id,summary:e.summary,description:e.description,jiraIssueKey:e.jiraIssueKey,workerModel:e.workerModel,workerProvider:e.workerProvider,githubRepo:e.githubRepo,scmProvider:e.scmProvider,skipManagerReview:e.skipManagerReview,deploymentEnabled:e.deploymentEnabled,improvementEnabled:e.improvementEnabled,qualityGateBypass:e.qualityGateBypass,standardSdkMode:e.standardSdkMode,parentTaskId:e.parentTaskId,taskNotes:e.taskNotes,githubPrUrl:e.githubPrUrl,githubPrNumber:e.githubPrNumber,executionPlanV2:e.executionPlanV2,jiraFields:e.jiraFields||{}},a=t.credentials;Qo(s,o,i,a).catch(c=>console.error(`${W()} ${S.red("\u2717")} Spawn failed for ${r}:`,c.message||c))}async function yn(e,o){let n=S.cyan(e.id.slice(0,8));be.add(e.id);try{let t=await N.post("/api/agent/claim-manager",{taskId:e.id,agentId:o.agentId,action:e.managerAction});if(!t.data.claimed){be.delete(e.id);return}let r=t.data.task,i=t.data.credentials||{},s=e.managerAction==="analyze_logs"?S.yellow("LOG ANALYSIS"):S.yellow("PR REVIEW");console.log(),console.log(`${W()} ${S.magenta("\u25C6 MANAGER")} ${s} ${n} ${e.summary.substring(0,60)}`);let a={id:e.id,summary:r?.summary||e.summary,description:r?.description||e.description,jiraIssueKey:r?.jiraIssueKey||e.jiraIssueKey,githubRepo:r?.githubRepo||e.githubRepo,scmProvider:r?.scmProvider||e.scmProvider,githubPrUrl:r?.githubPrUrl||e.githubPrUrl,githubPrNumber:r?.githubPrNumber||e.githubPrNumber,managerAction:e.managerAction};tt(a,o,i).then(()=>{console.log(`${W()} ${n} ${S.magenta("MGR")} ${S.green("\u2713")} Manager ${e.managerAction} dispatched`)}).catch(c=>{console.error(`${W()} ${n} ${S.magenta("MGR")} ${S.red("\u2717")} Manager spawn failed:`,c.message||c)}).finally(()=>be.delete(e.id))}catch(t){be.delete(e.id);let r=t;console.error(`${W()} ${n} ${S.red("\u2717")} Failed to claim manager task:`,r.message||String(t))}}var Ze=null,eo=null;function st(){Ze&&(clearInterval(Ze),Ze=null),eo&&(clearInterval(eo),eo=null)}function at(e){console.log(` ${S.dim("Polling every")} ${e.pollIntervalMs/1e3}s ${S.dim("\xB7 waiting for tasks...")}`),it(e),Ze=setInterval(()=>it(e),e.pollIntervalMs)}function lt(e){eo=setInterval(async()=>{let o=Zo(),n=Array.from(ke),t=Array.from(be),r=[...o,...n,...t];try{let i=await N.post("/api/agent/heartbeat",{agentId:e.agentId,activeTasks:r,agentVersion:z}),s=i.data?.cancelledTasks;if(s&&s.length>0)for(let w of s)et(w);let{updateAvailable:a,updateRequired:c,latestVersion:h}=i.data??{};c&&!Qe?(Qe=!0,console.log(`${W()} ${S.red("\u26A0 Agent update required")} (current: ${z}, required: ${h})`),console.log(`${W()} ${S.yellow("Refusing new tasks until updated.")}`),await Me()?Je():(console.log(`${W()} ${S.red("Auto-update failed.")} Run: npm install -g @workermill/agent@latest`),Qe=!1)):a&&!rt&&!c&&(rt=!0,console.log(`${W()} ${S.yellow(`Update available: ${h}`)} (current: ${z}). Run: workermill-agent update`))}catch{}},e.heartbeatIntervalMs)}oe();async function fo(e){console.log(),console.log(K.bold.cyan(" WorkerMill Remote Agent")),console.log(K.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(),Ge(e.apiUrl,e.apiKey);try{let o=await N.get("/api/agent/config"),n=o.data.maxConcurrentWorkers;n&&typeof n=="number"&&(e.maxWorkers=n),o.data.workerImageUrl&&(e.workerImage=o.data.workerImageUrl),console.log(` ${K.green("\u25CF")} Connected to ${K.cyan(e.apiUrl)}`),console.log(` ${K.dim("Agent:")} ${e.agentId}`),console.log(` ${K.dim("Workers:")} ${K.yellow(String(e.maxWorkers))} parallel`),console.log(` ${K.dim("Image:")} ${e.workerImage}`),console.log(` ${K.dim("SCM:")} ${o.data.scmProvider}`),console.log(` ${K.dim("Model:")} ${K.yellow(o.data.defaultWorkerModel)}`),console.log()}catch(o){let n=o;throw n.response?.status===401?new Error("Authentication failed. Check your API key."):new Error(`Failed to connect to WorkerMill API: ${n.message||String(o)}`)}try{let o=await N.post("/api/agent/register",{agentId:e.agentId,maxWorkers:e.maxWorkers,agentVersion:z}),{updateAvailable:n,updateRequired:t,latestVersion:r}=o.data;t?(console.log(K.red(` \u26A0 Agent update required (current: ${z}, required: ${r})`)),await Me()?Je():console.log(K.yellow(" Auto-update failed. Run: npm install -g @workermill/agent@latest"))):n&&console.log(K.yellow(` Update available: ${r} (current: ${z}). Run: workermill-agent update`))}catch{}return at(e),lt(e),console.log(K.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(` ${K.green("\u25CF")} Agent is running. ${K.dim("Press Ctrl+C to stop.")}`),console.log(),async()=>{console.log(),console.log(K.dim(" Shutting down...")),st();try{await N.post("/api/agent/deregister",{agentId:e.agentId})}catch{}await ot(),console.log(` ${K.red("\u25CF")} Agent stopped.`)}}var In=typeof process<"u"&&process.argv[1]&&(process.argv[1].endsWith("/index.ts")||process.argv[1].endsWith("/index.js"));if(In){try{await import("dotenv/config")}catch{}let{loadConfig:e,validatePrerequisites:o}=await Promise.resolve().then(()=>(oe(),Ao)),n=e();o(),console.log(K.dim(" Prerequisites validated."));let t=await fo(n);process.on("SIGINT",async()=>{await t(),process.exit(0)}),process.on("SIGTERM",async()=>{await t(),process.exit(0)})}async function ho(e){bn(ee())||(console.log(U.red("No configuration found.")),console.log(`Run ${U.cyan("workermill-agent setup")} first.`),process.exit(1));let o=he(),t=no(o.workerImage).filter(d=>!d.ok),r=t.find(d=>d.name==="Worker image"),i=new Set(["Claude CLI","Claude auth"]),s=t.filter(d=>d.name!=="Worker image"&&!i.has(d.name)),a=t.filter(d=>i.has(d.name));if(s.length>0){console.log(U.red("Prerequisites check failed:"));for(let d of s)console.log(U.red(` \u2717 ${d.name}: ${d.detail}`));process.exit(1)}if(a.length>0)for(let d of a)console.log(U.yellow(` \u26A0 ${d.name}: ${d.detail} (required for Anthropic provider)`));if(r){console.log(U.yellow(` Worker image not found locally. Pulling ${o.workerImage}...`));let{spawnSync:d}=await import("child_process");d("docker",["pull",o.workerImage],{stdio:"inherit",timeout:6e5}).status!==0&&(console.log(U.red(" Failed to pull worker image.")),process.exit(1)),console.log(U.green(" \u2713 Worker image pulled"))}if(e.detach){let d=_e(),$=ce();console.log(U.dim("Starting agent in background...")),console.log(U.dim(` Logs: ${d}`)),console.log(U.dim(` PID: ${$}`));let v=Sn(d,"a"),A=kn("workermill-agent",["start"],{detached:!0,stdio:["ignore",v,v],shell:!0});A.pid?(ct($,String(A.pid),"utf-8"),A.unref(),console.log(U.green(`Agent started (PID: ${A.pid})`)),console.log(`Check status with: ${U.cyan("workermill-agent status")}`)):(console.log(U.red("Failed to start agent in background.")),process.exit(1));return}let c=_e(),h=Tn(c,{flags:"a"}),w=process.stdout.write.bind(process.stdout),m=process.stderr.write.bind(process.stderr);process.stdout.write=(d,...$)=>(h.write(d),w(d,...$)),process.stderr.write=(d,...$)=>(h.write(d),m(d,...$)),console.log(),console.log(U.bold.cyan(" WorkerMill Remote Agent")),console.log(U.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log();let g=Math.round(An()/(1024*1024*1024));g<8?(console.log(U.red(` \u2717 Insufficient RAM: ${g} GB (minimum 8 GB, recommended 16 GB)`)),process.exit(1)):g<16&&console.log(U.yellow(` \u26A0 RAM: ${g} GB (below recommended 16 GB \u2014 workers may be slow)`));let P=Ue();console.log(U.dim(` Agent: ${o.agentId}`)),console.log(U.dim(` Version: ${z}`)),console.log(U.dim(` Image: ${o.workerImage}`)),console.log();try{let d=await fo(o);ct(ce(),String(process.pid),"utf-8");let $=!1,v=async()=>{$&&(console.log(U.red(`
|
|
18
|
+
Force exit.`)),process.exit(1)),$=!0;let A=setTimeout(()=>{console.log(U.red(`
|
|
19
|
+
Cleanup timed out. Force exit.`)),process.exit(1)},1e4);A.unref(),await d(),clearTimeout(A);try{En(ce())}catch{}process.exit(0)};process.on("SIGINT",v),process.on("SIGTERM",v)}catch(d){console.log(U.red(`Failed to start: ${d instanceof Error?d.message:String(d)}`)),process.exit(1)}}oe();import Ee from"chalk";import{existsSync as Cn,readFileSync as _n,unlinkSync as $o}from"fs";async function dt(){let e=ce();if(!Cn(e)){console.log(Ee.yellow("No running agent found (no PID file).")),console.log(Ee.dim(`Expected PID file at: ${e}`));return}let o=_n(e,"utf-8").trim(),n=parseInt(o,10);if(isNaN(n)){console.log(Ee.red(`Invalid PID in ${e}: ${o}`)),$o(e);return}try{process.kill(n,0)}catch{console.log(Ee.yellow(`Agent (PID ${n}) is not running. Cleaning up PID file.`)),$o(e);return}console.log(Ee.dim(`Stopping agent (PID ${n})...`));try{process.kill(n,"SIGTERM")}catch(r){console.log(Ee.red(`Failed to stop agent: ${r instanceof Error?r.message:String(r)}`));return}let t=Date.now();for(;Date.now()-t<15e3;)try{process.kill(n,0),await new Promise(r=>setTimeout(r,500))}catch{break}try{$o(e)}catch{}console.log(Ee.green("Agent stopped."))}oe();import j from"chalk";import{existsSync as ut,readFileSync as Rn}from"fs";import{execSync as Pn}from"child_process";import xn from"axios";async function gt(){if(console.log(),console.log(j.bold("WorkerMill Remote Agent Status")),console.log(j.dim("\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(),!ut(ee())){console.log(j.yellow(" Not configured. Run 'workermill-agent setup' first."));return}let e=he(),o=ce(),n=!1,t=null;if(ut(o)){let i=Rn(o,"utf-8").trim();if(t=parseInt(i,10),!isNaN(t))try{process.kill(t,0),n=!0}catch{n=!1}}let r=n?j.green("\u25CF online"):j.red("\u25CF offline");console.log(` Status: ${r}${t?j.dim(` (PID ${t})`):""}`),console.log(` Agent ID: ${e.agentId}`),console.log(` API URL: ${e.apiUrl}`),console.log(` Max workers: ${e.maxWorkers}`),console.log(` Image: ${e.workerImage}`),console.log(),console.log(j.bold(" Active Containers"));try{let i=Pn('docker ps --filter "name=workermill-" --format "{{.Names}} {{.Status}} {{.RunningFor}}"',{encoding:"utf-8",timeout:1e4}).trim();if(i){let s=i.split(`
|
|
20
|
+
`);console.log(j.dim(` Found ${s.length} container(s):`));for(let a of s){let[c,h,w]=a.split(" ");console.log(` ${j.cyan(c)} ${h} ${j.dim(w||"")}`)}}else console.log(j.dim(" No active containers"))}catch{console.log(j.dim(" Could not query Docker"))}console.log(),console.log(j.bold(" API Connectivity"));try{let i=await xn.get(`${e.apiUrl}/api/agent/config`,{headers:{"x-api-key":e.apiKey},timeout:1e4});console.log(` ${j.green("\u2713")} Connected to ${e.apiUrl}`),console.log(j.dim(` SCM: ${i.data.scmProvider}, Model: ${i.data.defaultWorkerModel}`))}catch(i){i.response?.status===401?console.log(` ${j.red("\u2717")} Authentication failed (invalid API key)`):console.log(` ${j.red("\u2717")} Cannot reach ${e.apiUrl}`)}console.log()}oe();import oo from"chalk";import{existsSync as vn,readFileSync as Mn,watchFile as Nn,statSync as pt,openSync as On,readSync as Ln,closeSync as Fn}from"fs";async function mt(e){let o=_e();if(!vn(o)){console.log(oo.yellow("No log file found.")),console.log(oo.dim(`Expected at: ${o}`)),console.log(oo.dim("Start the agent with --detach to generate logs."));return}let n=parseInt(e.lines||"50",10),r=Mn(o,"utf-8").split(`
|
|
21
21
|
`),i=Math.max(0,r.length-n),s=r.slice(i).join(`
|
|
22
22
|
`);s.trim()&&(process.stdout.write(s),s.endsWith(`
|
|
23
23
|
`)||process.stdout.write(`
|
|
24
|
-
`)),console.log(
|
|
24
|
+
`)),console.log(oo.dim(`\u2500\u2500 Following ${o} (Ctrl+C to stop) \u2500\u2500`));let a=pt(o).size;Nn(o,{interval:500},()=>{try{let c=pt(o).size;if(c<=a){a=c;return}let h=On(o,"r"),w=Buffer.alloc(c-a);Ln(h,w,0,w.length,a),Fn(h),process.stdout.write(w.toString()),a=c}catch{}}),await new Promise(()=>{})}oe();import ne from"chalk";import{spawnSync as Dn,execSync as Un}from"child_process";import{existsSync as Wn}from"fs";function Kn(e){let o=e.match(/^(\d+\.dkr\.ecr\.[a-z0-9-]+\.amazonaws\.com)/);return o?o[1]:null}async function ft(){Wn(ee())||(console.log(ne.red(" No config found. Run 'workermill-agent setup' first.")),process.exit(1));let e=he(),o=e.workerImage;try{Ge(e.apiUrl,e.apiKey);let{data:r}=await N.get("/api/agent/config");r.workerImageUrl&&(o=r.workerImageUrl)}catch{console.log(ne.yellow(" \u26A0 Could not fetch server config \u2014 using local image setting"))}let n=Kn(o);if(n){console.log(ne.dim(" Authenticating to private ECR..."));let r=n.match(/\.ecr\.([a-z0-9-]+)\.amazonaws\.com/),i=r?r[1]:"us-east-1";try{Un(`aws ecr get-login-password --region ${i} | docker login --username AWS --password-stdin ${n}`,{stdio:"pipe",timeout:3e4})}catch{console.log(ne.red(" \u2717 ECR authentication failed.")),console.log(ne.yellow(" Ensure AWS CLI is configured: aws configure")),process.exit(1)}}console.log(),console.log(ne.dim(` Pulling worker image: ${o}`)),console.log(ne.dim(" This may take a few minutes...")),console.log();let t=Dn("docker",["pull",o],{stdio:"inherit",timeout:6e5});console.log(),t.status===0?console.log(ne.green(" \u2713 Worker image updated")):(console.log(ne.red(" \u2717 Failed to pull worker image.")),t.error&&console.log(ne.yellow(` Error: ${t.error.message}`)),console.log(ne.yellow(" Is Docker Desktop running?")),process.exit(1))}import Fe from"chalk";async function ht(){console.log(),console.log(Fe.bold.cyan(" WorkerMill Agent Update")),console.log(Fe.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(` ${Fe.dim("Current version:")} ${z}`),console.log(),await Me()?(console.log(),console.log(Fe.green(" Update complete. If the agent is running, restart it to use the new version."))):(console.log(),console.error(Fe.red(" Update failed. Try running manually: npm install -g @workermill/agent@latest")),process.exitCode=1)}oe();var ge=new jn;ge.name("workermill-agent").description("WorkerMill Remote Agent - Run AI workers locally with your Claude Max subscription").version(z);ge.command("setup").description("Interactive setup wizard - configure API key, validate prerequisites, pull worker image").action(ao);ge.command("start").description("Start the agent, polling the cloud API for tasks").option("--detach","Run in background (daemon mode)").action(ho);ge.command("stop").description("Stop the running agent").action(dt);ge.command("status").description("Show agent status, active containers, and API connectivity").action(gt);ge.command("logs").description("Live tail of agent logs (like tail -f)").option("-n, --lines <count>","Number of lines to show initially","50").action(mt);ge.command("pull").description("Pull the latest worker Docker image").action(ft);ge.command("update").description("Update the agent to the latest version").action(ht);process.argv.length<=2?Gn(ee())?ho({detach:!1}):ao():ge.parse();
|
package/dist/index.js
CHANGED
|
@@ -1,15 +1,15 @@
|
|
|
1
|
-
var Mt=Object.defineProperty;var Ot=(e,t)=>()=>(e&&(t=e(e=0)),t);var Nt=(e,t)=>{for(var r in t)Mt(e,r,{get:t[r],enumerable:!0})};var Je={};Nt(Je,{checkPrerequisites:()=>Jt,findClaudePath:()=>ce,getConfigDir:()=>Bt,getConfigFile:()=>Wt,getLogFile:()=>Yt,getPidFile:()=>Vt,getSystemInfo:()=>ze,loadConfig:()=>Ve,loadConfigFromFile:()=>We,saveConfigToFile:()=>zt,validatePrerequisites:()=>Ye});import{existsSync as ye,readFileSync as Ut,mkdirSync as Dt,writeFileSync as Ft,chmodSync as Kt}from"fs";import{execSync as re}from"child_process";import{hostname as Be,homedir as pe}from"os";import{join as q}from"path";function Bt(){return me}function Wt(){return ge}function Vt(){return jt}function Yt(){return Gt}function We(){ye(ge)||(console.error("No config found. Run 'workermill-agent setup' first."),process.exit(1));let e;try{e=Ut(ge,"utf-8")}catch{console.error("Failed to read config file:",ge),process.exit(1)}let t;try{t=JSON.parse(e)}catch{console.error("Config file is corrupted. Re-run 'workermill-agent setup'."),process.exit(1)}(!t.apiUrl||!t.apiKey)&&(console.error("Config file is missing required fields (apiUrl, apiKey). Re-run 'workermill-agent setup'."),process.exit(1));let r=t.workerImage||"workermill-worker:local";return{apiUrl:t.apiUrl,apiKey:t.apiKey,agentId:t.agentId,maxWorkers:t.maxWorkers||4,pollIntervalMs:t.pollIntervalMs||5e3,heartbeatIntervalMs:t.heartbeatIntervalMs||3e4,githubToken:t.tokens?.github||"",bitbucketToken:t.tokens?.bitbucket||"",gitlabToken:t.tokens?.gitlab||"",workerImage:r}}function zt(e){ye(me)||Dt(me,{recursive:!0}),Ft(ge,JSON.stringify(e,null,2),"utf-8");try{Kt(ge,384)}catch{}}function Ve(){let e=process.env.WORKERMILL_API_URL,t=process.env.WORKERMILL_API_KEY;return e||(console.error("WORKERMILL_API_URL is required in .env.remote"),process.exit(1)),t||(console.error("WORKERMILL_API_KEY is required in .env.remote"),console.error("Get your API key from Settings > Integrations on the WorkerMill dashboard."),process.exit(1)),{apiUrl:e.replace(/\/$/,""),apiKey:t,agentId:process.env.AGENT_ID||`agent-${Be()}`,maxWorkers:parseInt(process.env.MAX_WORKERS||"4",10),pollIntervalMs:parseInt(process.env.POLL_INTERVAL_MS||"5000",10),heartbeatIntervalMs:parseInt(process.env.HEARTBEAT_INTERVAL_MS||"30000",10),githubToken:process.env.GITHUB_TOKEN||"",bitbucketToken:process.env.BITBUCKET_TOKEN||"",gitlabToken:process.env.GITLAB_TOKEN||"",workerImage:process.env.WORKER_IMAGE||"workermill-worker:local"}}function ce(){let e=process.platform==="win32",t=e?"where":"which";try{return re(`${t} claude`,{stdio:"ignore",timeout:1e4}),"claude"}catch{}let r=[];e?r.push(q(process.env.ProgramFiles||"C:\\Program Files","ClaudeCode","claude.exe"),q(process.env.LOCALAPPDATA||"","Programs","ClaudeCode","claude.exe"),q(pe(),"AppData","Local","Programs","ClaudeCode","claude.exe"),q(pe(),".local","bin","claude.exe")):r.push(q(pe(),".local","bin","claude"),"/opt/homebrew/bin/claude","/usr/local/bin/claude");for(let o of r)if(o&&ye(o))return o;return null}function Jt(e){let t=[],r=e||"workermill-worker:local";try{let l=re("docker version --format {{.Server.Version}}",{encoding:"utf-8",timeout:1e4}).trim();t.push({name:"Docker",ok:!0,detail:l})}catch{t.push({name:"Docker",ok:!1,detail:"Not running or not installed"})}let o=ce();if(o)try{let l=re(`"${o}" --version`,{encoding:"utf-8",timeout:1e4}).trim();t.push({name:"Claude CLI",ok:!0,detail:l})}catch{t.push({name:"Claude CLI",ok:!0,detail:o})}else t.push({name:"Claude CLI",ok:!1,detail:"Not installed"});let n=pe(),i=q(n,".claude",".credentials.json");ye(i)?t.push({name:"Claude auth",ok:!0,detail:"Credentials found"}):t.push({name:"Claude auth",ok:!1,detail:"Run 'claude' and complete sign-in"});let s=process.version;parseInt(s.slice(1).split(".")[0],10)>=20?t.push({name:"Node.js",ok:!0,detail:s}):t.push({name:"Node.js",ok:!1,detail:`${s} (need >= 20)`});try{re(`docker image inspect ${r}`,{stdio:"ignore",timeout:1e4}),t.push({name:"Worker image",ok:!0,detail:r})}catch{t.push({name:"Worker image",ok:!1,detail:`'${r}' not found`})}return t}function Ye(){try{re("docker version",{stdio:"ignore"})}catch{console.error("Docker is not available. Please install Docker and ensure it's running."),process.exit(1)}let e=process.env.WORKER_IMAGE||"workermill-worker:local";try{re(`docker image inspect ${e}`,{stdio:"ignore"})}catch{console.error(`Worker image '${e}' not found.`),console.error(e==="workermill-worker:local"?"Build it with: ./bin/local-workermill build-worker":`Pull it with: docker pull ${e}`),process.exit(1)}ce()||(console.error("Claude CLI is not installed."),console.error("Install it: curl -fsSL https://claude.ai/install.sh | bash"),process.exit(1));let t=pe(),r=q(t,".claude",".credentials.json");ye(r)||(console.error("Claude credentials not found."),console.error("Run 'claude' and complete the sign-in flow to authenticate."),process.exit(1))}function ze(){let e="unknown";try{e=re("docker version --format {{.Server.Version}}",{encoding:"utf-8",timeout:1e4}).trim()}catch{}let t="unknown",r=ce();if(r)try{t=re(`"${r}" --version`,{encoding:"utf-8",timeout:1e4}).trim()}catch{}return{hostname:Be(),platform:process.platform,nodeVersion:process.version,dockerVersion:e,claudeVersion:t}}var me,ge,jt,Gt,Te=Ot(()=>{"use strict";me=q(pe(),".workermill"),ge=q(me,"config.json"),jt=q(me,"agent.pid"),Gt=q(me,"agent.log")});import D from"chalk";import Lt from"axios";var Le=null;function Ge(e,t){Le=Lt.create({baseURL:e,headers:{"Content-Type":"application/json","x-api-key":t},timeout:3e4})}var P=new Proxy({},{get(e,t){if(!Le)throw new Error("API client not initialized. Call initApi() first.");return Le[t]}});import h from"chalk";Te();import c from"chalk";import{spawn as ho,execSync as Ke}from"child_process";import{spawn as to}from"child_process";import ie from"chalk";import be from"axios";var Ht=16384,qt=6e5;async function He(e,t,r,o,n){let i=n?.maxTokens??Ht,s=n?.temperature??.7,a=n?.timeoutMs??qt;switch(e){case"anthropic":return Xt(t,r,o,i,s,a);case"openai":return Qt(t,r,o,i,s,a);case"google":return Zt(t,r,o,i,s,a);case"ollama":return eo(t,r,o,i,s,a);default:throw new Error(`Unsupported AI provider: ${e}`)}}async function Xt(e,t,r,o,n,i){let a=(await be.post("https://api.anthropic.com/v1/messages",{model:e,max_tokens:o,temperature:n,messages:[{role:"user",content:t}]},{headers:{"x-api-key":r,"anthropic-version":"2023-06-01","Content-Type":"application/json"},timeout:i})).data?.content;if(Array.isArray(a))return a.filter(l=>l.type==="text").map(l=>l.text).join("");throw new Error("Unexpected Anthropic API response format")}async function Qt(e,t,r,o,n,i){let a=(await be.post("https://api.openai.com/v1/chat/completions",{model:e,max_tokens:o,temperature:n,messages:[{role:"user",content:t}]},{headers:{Authorization:`Bearer ${r}`,"Content-Type":"application/json"},timeout:i})).data?.choices?.[0]?.message;if(a?.content)return a.content;throw new Error("Unexpected OpenAI API response format")}async function Zt(e,t,r,o,n,i){let l=(await be.post(`https://generativelanguage.googleapis.com/v1beta/models/${e}:generateContent?key=${r}`,{contents:[{parts:[{text:t}]}],generationConfig:{maxOutputTokens:o,temperature:n}},{headers:{"Content-Type":"application/json"},timeout:i})).data?.candidates?.[0]?.content?.parts?.[0]?.text;if(l)return l;throw new Error("Unexpected Google AI API response format")}async function eo(e,t,r,o,n,i){let s=r||"http://localhost:11434",l=(await be.post(`${s}/api/generate`,{model:e,prompt:t,stream:!1,options:{temperature:n}},{headers:{"Content-Type":"application/json"},timeout:i})).data?.response;if(l)return l;throw new Error("Unexpected Ollama API response format")}var ne=null;async function Xe(){if(ne)return ne;try{let{data:e}=await P.get("/api/agent/critic-prompt");return ne={promptTemplate:e.promptTemplate,approvalThreshold:e.approvalThreshold??85,maxTargetFiles:e.maxTargetFiles??15},X=ne.approvalThreshold,we=ne.maxTargetFiles,ne}catch{return console.warn("Failed to fetch critic config from API"),null}}var we=15,X=85;function Qe(e){let t=e.indexOf("```json");if(t!==-1){let o=t+7,n=e.indexOf("{",o);if(n!==-1){let i=qe(e,n);if(i)return JSON.parse(i)}}let r=e.indexOf('"stories"');if(r!==-1){let n=e.substring(0,r).lastIndexOf("{");if(n!==-1){let i=qe(e,n);if(i)return JSON.parse(i)}}throw new Error("Could not find JSON execution plan in output")}function qe(e,t){let r=0,o=!1,n=!1;for(let i=t;i<e.length;i++){let s=e[i];if(n){n=!1;continue}if(s==="\\"){o&&(n=!0);continue}if(s==='"'){o=!o;continue}if(!o){if(s==="{")r++;else if(s==="}"&&(r--,r===0))return e.substring(t,i+1)}}return null}function Ze(e){let t=0,r=[];for(let o of e.stories)if(!o.targetFiles||!Array.isArray(o.targetFiles))o.targetFiles=[];else if(o.targetFiles.length>we){let n=o.targetFiles.slice(we);r.push(`${o.id}: ${o.targetFiles.length} files \u2192 ${we} (dropped: ${n.join(", ")})`),o.targetFiles=o.targetFiles.slice(0,we),t++}return{truncatedCount:t,details:r}}function et(e,t){if(e.stories.length<=t)return{droppedCount:0,details:[]};let r=e.stories.length-t,n=e.stories.slice(t).map(s=>`${s.id}: "${s.title}" (${s.persona})`);e.stories=e.stories.slice(0,t);let i=new Set(e.stories.map(s=>s.id));for(let s of e.stories)s.dependencies=s.dependencies.filter(a=>i.has(a));return{droppedCount:r,details:n}}function tt(e){let t=new Map,r=0,o=[];for(let n of e.stories){if(!n.targetFiles||n.targetFiles.length===0)continue;let i=[],s=[];for(let a of n.targetFiles)t.get(a)?s.push(a):(t.set(a,n.id),i.push(a));s.length>0&&(n.targetFiles=i,r+=s.length,o.push(`${n.id}: removed ${s.join(", ")} (owned by ${s.map(a=>t.get(a)).join(", ")})`))}return{resolvedCount:r,details:o}}function ot(e){return"```json\n"+JSON.stringify(e,null,2)+"\n```"}function oo(e,t){if(!ne)return null;let r=JSON.stringify(t,null,2);return ne.promptTemplate.replace("{{PRD}}",e).replace("{{PLAN}}",r)}function ro(e){let t=e.trim();if(t.includes("```")){let n=t.match(/```(?:json)?\s*([\s\S]*?)```/);n&&(t=n[1].trim())}let r=t.indexOf("{");r>0&&(t=t.substring(r));let o=JSON.parse(t);return{approved:o.approved,score:Math.max(0,Math.min(100,Math.round(o.score))),risks:o.risks||[],suggestions:o.suggestions,storyFeedback:Array.isArray(o.storyFeedback)?o.storyFeedback:void 0}}function no(e,t,r,o,n){return new Promise((i,s)=>{let a=to(e,["--print","--model",t,"--permission-mode","bypassPermissions"],{env:o,stdio:["pipe","pipe","pipe"]});a.stdin.write(r),a.stdin.end();let l="",w="";a.stdout.on("data",d=>{let E=d.toString();l+=E;let k=E.split(`
|
|
2
|
-
`).filter(
|
|
3
|
-
`)}var
|
|
1
|
+
var Ot=Object.defineProperty;var Nt=(e,t)=>()=>(e&&(t=e(e=0)),t);var Lt=(e,t)=>{for(var r in t)Ot(e,r,{get:t[r],enumerable:!0})};var He={};Lt(He,{checkPrerequisites:()=>Ht,findClaudePath:()=>ue,getConfigDir:()=>Wt,getConfigFile:()=>Vt,getLogFile:()=>zt,getPidFile:()=>Yt,getSystemInfo:()=>Je,loadConfig:()=>Ye,loadConfigFromFile:()=>Ve,saveConfigToFile:()=>Jt,validatePrerequisites:()=>ze});import{existsSync as we,readFileSync as Dt,mkdirSync as Ft,writeFileSync as Kt,chmodSync as jt}from"fs";import{execSync as ne}from"child_process";import{hostname as We,homedir as ge}from"os";import{join as H}from"path";function Wt(){return fe}function Vt(){return me}function Yt(){return Gt}function zt(){return Bt}function Ve(){we(me)||(console.error("No config found. Run 'workermill-agent setup' first."),process.exit(1));let e;try{e=Dt(me,"utf-8")}catch{console.error("Failed to read config file:",me),process.exit(1)}let t;try{t=JSON.parse(e)}catch{console.error("Config file is corrupted. Re-run 'workermill-agent setup'."),process.exit(1)}(!t.apiUrl||!t.apiKey)&&(console.error("Config file is missing required fields (apiUrl, apiKey). Re-run 'workermill-agent setup'."),process.exit(1));let r=t.workerImage||"workermill-worker:local";return{apiUrl:t.apiUrl,apiKey:t.apiKey,agentId:t.agentId,maxWorkers:t.maxWorkers||4,pollIntervalMs:t.pollIntervalMs||5e3,heartbeatIntervalMs:t.heartbeatIntervalMs||3e4,githubToken:t.tokens?.github||"",bitbucketToken:t.tokens?.bitbucket||"",gitlabToken:t.tokens?.gitlab||"",workerImage:r}}function Jt(e){we(fe)||Ft(fe,{recursive:!0}),Kt(me,JSON.stringify(e,null,2),"utf-8");try{jt(me,384)}catch{}}function Ye(){let e=process.env.WORKERMILL_API_URL,t=process.env.WORKERMILL_API_KEY;return e||(console.error("WORKERMILL_API_URL is required in .env.remote"),process.exit(1)),t||(console.error("WORKERMILL_API_KEY is required in .env.remote"),console.error("Get your API key from Settings > Integrations on the WorkerMill dashboard."),process.exit(1)),{apiUrl:e.replace(/\/$/,""),apiKey:t,agentId:process.env.AGENT_ID||`agent-${We()}`,maxWorkers:parseInt(process.env.MAX_WORKERS||"4",10),pollIntervalMs:parseInt(process.env.POLL_INTERVAL_MS||"5000",10),heartbeatIntervalMs:parseInt(process.env.HEARTBEAT_INTERVAL_MS||"30000",10),githubToken:process.env.GITHUB_TOKEN||"",bitbucketToken:process.env.BITBUCKET_TOKEN||"",gitlabToken:process.env.GITLAB_TOKEN||"",workerImage:process.env.WORKER_IMAGE||"workermill-worker:local"}}function ue(){let e=process.platform==="win32",t=e?"where":"which";try{return ne(`${t} claude`,{stdio:"ignore",timeout:1e4}),"claude"}catch{}let r=[];e?r.push(H(process.env.ProgramFiles||"C:\\Program Files","ClaudeCode","claude.exe"),H(process.env.LOCALAPPDATA||"","Programs","ClaudeCode","claude.exe"),H(ge(),"AppData","Local","Programs","ClaudeCode","claude.exe"),H(ge(),".local","bin","claude.exe")):r.push(H(ge(),".local","bin","claude"),"/opt/homebrew/bin/claude","/usr/local/bin/claude");for(let o of r)if(o&&we(o))return o;return null}function Ht(e){let t=[],r=e||"workermill-worker:local";try{let l=ne("docker version --format {{.Server.Version}}",{encoding:"utf-8",timeout:1e4}).trim();t.push({name:"Docker",ok:!0,detail:l})}catch{t.push({name:"Docker",ok:!1,detail:"Not running or not installed"})}let o=ue();if(o)try{let l=ne(`"${o}" --version`,{encoding:"utf-8",timeout:1e4}).trim();t.push({name:"Claude CLI",ok:!0,detail:l})}catch{t.push({name:"Claude CLI",ok:!0,detail:o})}else t.push({name:"Claude CLI",ok:!1,detail:"Not installed"});let n=ge(),i=H(n,".claude",".credentials.json");we(i)?t.push({name:"Claude auth",ok:!0,detail:"Credentials found"}):t.push({name:"Claude auth",ok:!1,detail:"Run 'claude' and complete sign-in"});let s=process.version;parseInt(s.slice(1).split(".")[0],10)>=20?t.push({name:"Node.js",ok:!0,detail:s}):t.push({name:"Node.js",ok:!1,detail:`${s} (need >= 20)`});try{ne(`docker image inspect ${r}`,{stdio:"ignore",timeout:1e4}),t.push({name:"Worker image",ok:!0,detail:r})}catch{t.push({name:"Worker image",ok:!1,detail:`'${r}' not found`})}return t}function ze(){try{ne("docker version",{stdio:"ignore"})}catch{console.error("Docker is not available. Please install Docker and ensure it's running."),process.exit(1)}let e=process.env.WORKER_IMAGE||"workermill-worker:local";try{ne(`docker image inspect ${e}`,{stdio:"ignore"})}catch{console.error(`Worker image '${e}' not found.`),console.error(e==="workermill-worker:local"?"Build it with: ./bin/local-workermill build-worker":`Pull it with: docker pull ${e}`),process.exit(1)}ue()||(console.error("Claude CLI is not installed."),console.error("Install it: curl -fsSL https://claude.ai/install.sh | bash"),process.exit(1));let t=ge(),r=H(t,".claude",".credentials.json");we(r)||(console.error("Claude credentials not found."),console.error("Run 'claude' and complete the sign-in flow to authenticate."),process.exit(1))}function Je(){let e="unknown";try{e=ne("docker version --format {{.Server.Version}}",{encoding:"utf-8",timeout:1e4}).trim()}catch{}let t="unknown",r=ue();if(r)try{t=ne(`"${r}" --version`,{encoding:"utf-8",timeout:1e4}).trim()}catch{}return{hostname:We(),platform:process.platform,nodeVersion:process.version,dockerVersion:e,claudeVersion:t}}var fe,me,Gt,Bt,be=Nt(()=>{"use strict";fe=H(ge(),".workermill"),me=H(fe,"config.json"),Gt=H(fe,"agent.pid"),Bt=H(fe,"agent.log")});import U from"chalk";import Ut from"axios";var Ue=null;function Be(e,t){Ue=Ut.create({baseURL:e,headers:{"Content-Type":"application/json","x-api-key":t},timeout:3e4})}var M=new Proxy({},{get(e,t){if(!Ue)throw new Error("API client not initialized. Call initApi() first.");return Ue[t]}});import y from"chalk";be();import c from"chalk";import{spawn as yo,execSync as je}from"child_process";import{spawn as oo}from"child_process";import se from"chalk";import Ae from"axios";var qt=16384,Xt=6e5;async function qe(e,t,r,o,n){let i=n?.maxTokens??qt,s=n?.temperature??.7,a=n?.timeoutMs??Xt;switch(e){case"anthropic":return Qt(t,r,o,i,s,a);case"openai":return Zt(t,r,o,i,s,a);case"google":return eo(t,r,o,i,s,a);case"ollama":return to(t,r,o,i,s,a);default:throw new Error(`Unsupported AI provider: ${e}`)}}async function Qt(e,t,r,o,n,i){let a=(await Ae.post("https://api.anthropic.com/v1/messages",{model:e,max_tokens:o,temperature:n,messages:[{role:"user",content:t}]},{headers:{"x-api-key":r,"anthropic-version":"2023-06-01","Content-Type":"application/json"},timeout:i})).data?.content;if(Array.isArray(a))return a.filter(l=>l.type==="text").map(l=>l.text).join("");throw new Error("Unexpected Anthropic API response format")}async function Zt(e,t,r,o,n,i){let a=(await Ae.post("https://api.openai.com/v1/chat/completions",{model:e,max_tokens:o,temperature:n,messages:[{role:"user",content:t}]},{headers:{Authorization:`Bearer ${r}`,"Content-Type":"application/json"},timeout:i})).data?.choices?.[0]?.message;if(a?.content)return a.content;throw new Error("Unexpected OpenAI API response format")}async function eo(e,t,r,o,n,i){let l=(await Ae.post(`https://generativelanguage.googleapis.com/v1beta/models/${e}:generateContent?key=${r}`,{contents:[{parts:[{text:t}]}],generationConfig:{maxOutputTokens:o,temperature:n}},{headers:{"Content-Type":"application/json"},timeout:i})).data?.candidates?.[0]?.content?.parts?.[0]?.text;if(l)return l;throw new Error("Unexpected Google AI API response format")}async function to(e,t,r,o,n,i){let s=r||"http://localhost:11434",l=(await Ae.post(`${s}/api/generate`,{model:e,prompt:t,stream:!1,options:{temperature:n}},{headers:{"Content-Type":"application/json"},timeout:i})).data?.response;if(l)return l;throw new Error("Unexpected Ollama API response format")}var ie=null;async function Qe(e){if(ie)return ie;try{let t={};e!==void 0&&(t.maxTargetFiles=String(e));let{data:r}=await M.get("/api/agent/critic-prompt",{params:t});return ie={promptTemplate:r.promptTemplate,approvalThreshold:r.approvalThreshold??85,maxTargetFiles:r.maxTargetFiles??e??5},q=ie.approvalThreshold,$e=ie.maxTargetFiles,ie}catch{return console.warn("Failed to fetch critic config from API"),null}}var $e=5,q=85;function Ze(e){let t=e.indexOf("```json");if(t!==-1){let o=t+7,n=e.indexOf("{",o);if(n!==-1){let i=Xe(e,n);if(i)return JSON.parse(i)}}let r=e.indexOf('"stories"');if(r!==-1){let n=e.substring(0,r).lastIndexOf("{");if(n!==-1){let i=Xe(e,n);if(i)return JSON.parse(i)}}throw new Error("Could not find JSON execution plan in output")}function Xe(e,t){let r=0,o=!1,n=!1;for(let i=t;i<e.length;i++){let s=e[i];if(n){n=!1;continue}if(s==="\\"){o&&(n=!0);continue}if(s==='"'){o=!o;continue}if(!o){if(s==="{")r++;else if(s==="}"&&(r--,r===0))return e.substring(t,i+1)}}return null}function et(e){let t=0,r=[];for(let o of e.stories)if(!o.targetFiles||!Array.isArray(o.targetFiles))o.targetFiles=[];else if(o.targetFiles.length>$e){let n=o.targetFiles.slice($e);r.push(`${o.id}: ${o.targetFiles.length} files \u2192 ${$e} (dropped: ${n.join(", ")})`),o.targetFiles=o.targetFiles.slice(0,$e),t++}return{truncatedCount:t,details:r}}function tt(e,t){if(e.stories.length<=t)return{droppedCount:0,details:[]};let r=e.stories.length-t,n=e.stories.slice(t).map(s=>`${s.id}: "${s.title}" (${s.persona})`);e.stories=e.stories.slice(0,t);let i=new Set(e.stories.map(s=>s.id));for(let s of e.stories)s.dependencies=s.dependencies.filter(a=>i.has(a));return{droppedCount:r,details:n}}function ot(e){let t=new Map,r=0,o=[];for(let n of e.stories){if(!n.targetFiles||n.targetFiles.length===0)continue;let i=[],s=[];for(let a of n.targetFiles)t.get(a)?s.push(a):(t.set(a,n.id),i.push(a));s.length>0&&(n.targetFiles=i,r+=s.length,o.push(`${n.id}: removed ${s.join(", ")} (owned by ${s.map(a=>t.get(a)).join(", ")})`))}return{resolvedCount:r,details:o}}function rt(e){return"```json\n"+JSON.stringify(e,null,2)+"\n```"}function ro(e,t){if(!ie)return null;let r=JSON.stringify(t,null,2);return ie.promptTemplate.replace("{{PRD}}",e).replace("{{PLAN}}",r)}function no(e){let t=e.trim();if(t.includes("```")){let n=t.match(/```(?:json)?\s*([\s\S]*?)```/);n&&(t=n[1].trim())}let r=t.indexOf("{");r>0&&(t=t.substring(r));let o=JSON.parse(t);return{approved:o.approved,score:Math.max(0,Math.min(100,Math.round(o.score))),risks:o.risks||[],suggestions:o.suggestions,storyFeedback:Array.isArray(o.storyFeedback)?o.storyFeedback:void 0}}function io(e,t,r,o,n){return new Promise((i,s)=>{let a=oo(e,["--print","--model",t,"--permission-mode","bypassPermissions"],{env:o,stdio:["pipe","pipe","pipe"]});a.stdin.write(r),a.stdin.end();let l="",I="";a.stdout.on("data",p=>{let m=p.toString();l+=m;let k=m.split(`
|
|
2
|
+
`).filter(N=>N.trim());for(let N of k){let R=N.trim().length>200?N.trim().substring(0,200)+"\u2026":N.trim();R&&(n&&st(n,`${it} [critic] ${R}`,"output"),console.log(`${Ee()} ${se.dim("\u{1F50D}")} ${se.dim(R)}`))}}),a.stderr.on("data",p=>{I+=p.toString()});let b=setTimeout(()=>{a.kill("SIGTERM"),s(new Error("Critic CLI timed out after 20 minutes"))},12e5);a.on("exit",p=>{clearTimeout(b),p!==0?s(new Error(`Critic CLI failed (exit ${p}): ${I.substring(0,300)}`)):i(l)}),a.on("error",p=>{clearTimeout(b),s(p)})})}function nt(e){let t=["","## CRITIC FEEDBACK \u2014 Your previous plan was REJECTED","",`Score: ${e.score}/100 (need >= ${q} to pass)`,""];if(e.risks.length>0){t.push("### Risks Identified:");for(let r of e.risks)t.push(`- ${r}`);t.push("")}if(e.suggestions&&e.suggestions.length>0){t.push("### Required Changes:");for(let r of e.suggestions)t.push(`- ${r}`);t.push("")}if(e.storyFeedback&&e.storyFeedback.length>0){t.push("### Per-Story Feedback:");for(let r of e.storyFeedback)if(t.push(`- **${r.storyId}**: ${r.feedback}`),r.suggestedChanges)for(let o of r.suggestedChanges)t.push(` - ${o}`);t.push("")}return t.push(`**You MUST address ALL feedback above.** Each story must target at most ${$e} files.`,"Stories MUST NOT overlap on targetFiles.","","**CRITICAL \u2014 OUTPUT FORMAT:** You MUST output a revised plan as a ```json code block containing the full JSON object with `summary`, `stories`, `risks`, and `assumptions` fields. Do NOT just describe what you would change \u2014 output the COMPLETE revised JSON plan. If you do not output a ```json block, the plan will fail to parse and the task will fail.","","**DO NOT re-explore the repository.** You already explored it in the previous attempt. Go directly to outputting the revised ```json plan. Every tool call wastes output budget \u2014 prioritize emitting the JSON plan."),t.join(`
|
|
3
|
+
`)}var it="[\u{1F5FA}\uFE0F planning_agent \u{1F916}]";function Ee(){return se.dim(new Date().toLocaleTimeString())}async function st(e,t,r="system",o="info"){try{await M.post("/api/control-center/logs",{taskId:e,type:r,message:t,severity:o})}catch{}}async function at(e,t,r,o,n,i,s,a,l){let I=ro(r,o);if(!I)return console.warn(`${Ee()} ${i} ${se.yellow("\u26A0")} Critic config not available \u2014 skipping validation`),null;let b=s||"anthropic";console.log(`${Ee()} ${i} ${se.dim(`Running critic validation (${b})...`)}`),l&&st(l,`${it} Running critic validation (${b})...`);try{let p;if(b==="anthropic")p=await io(e,t,I,n,l);else{if(!a)throw new Error(`No API key for critic provider "${b}"`);p=await qe(b,t,I,a,{maxTokens:4096,temperature:.3,timeoutMs:12e5})}let m=no(p),k=m.score>=q?se.green("\u2713"):se.red("\u2717");return console.log(`${Ee()} ${i} ${k} Critic score: ${m.score}/100 (threshold: ${q})`),m}catch(p){let m=p instanceof Error?p.message:String(p);return console.error(`${Ee()} ${i} ${se.yellow("\u26A0")} Critic failed: ${m.substring(0,100)}`),null}}import{generateText as so,tool as De,stepCountIs as ao}from"ai";import{createOpenAI as lt}from"@ai-sdk/openai";import{createAnthropic as lo}from"@ai-sdk/anthropic";import{createGoogleGenerativeAI as co}from"@ai-sdk/google";import{z as ae}from"zod";import{execSync as Fe}from"child_process";import{readFileSync as uo,existsSync as po}from"fs";function go(e,t,r){switch(e){case"anthropic":return lo({apiKey:r})(t);case"openai":return lt({apiKey:r})(t);case"google":return co({apiKey:r})(t);case"ollama":return lt({baseURL:r||"http://localhost:11434/v1",apiKey:"ollama"})(t);default:throw new Error(`Unsupported AI provider: ${e}`)}}var mo=ae.object({pattern:ae.string().describe("Glob pattern like '**/*.ts', 'src/**/*.js', 'package.json'")}),fo=ae.object({path:ae.string().describe("File path relative to the working directory"),limit:ae.number().optional().describe("Max number of lines to read (default: 500)")}),$o=ae.object({pattern:ae.string().describe("Search pattern (regex supported)"),glob:ae.string().optional().describe("File glob to filter (e.g. '*.ts', '*.py')")});function ho(e){return{glob:De({description:"Find files matching a glob pattern. Returns file paths relative to the working directory.",inputSchema:mo,execute:async t=>{try{let r=Fe(`find . -path './.git' -prune -o -path './node_modules' -prune -o -name '${t.pattern.replace(/\*\*/g,"*")}' -print 2>/dev/null | head -200`,{cwd:e,encoding:"utf-8",timeout:15e3}).trim();return r||Fe("find . -path './.git' -prune -o -path './node_modules' -prune -o -type f -print 2>/dev/null | head -500",{cwd:e,encoding:"utf-8",timeout:15e3}).trim()||"No files found"}catch{return"Error running glob search"}}}),read_file:De({description:"Read the contents of a file. Returns the file text.",inputSchema:fo,execute:async t=>{try{let r=`${e}/${t.path}`.replace(/\/\//g,"/");if(!po(r))return`File not found: ${t.path}`;let o=uo(r,"utf-8"),n=o.split(`
|
|
4
4
|
`),i=t.limit||500;return n.length>i?n.slice(0,i).join(`
|
|
5
5
|
`)+`
|
|
6
|
-
... (truncated, ${n.length-i} more lines)`:o}catch(r){return`Error reading file: ${r instanceof Error?r.message:String(r)}`}}}),grep:
|
|
7
|
-
`),
|
|
8
|
-
`);
|
|
9
|
-
`,
|
|
6
|
+
... (truncated, ${n.length-i} more lines)`:o}catch(r){return`Error reading file: ${r instanceof Error?r.message:String(r)}`}}}),grep:De({description:"Search for a pattern in files. Returns matching lines with file paths and line numbers.",inputSchema:$o,execute:async t=>{try{let r=t.glob?`--include='${t.glob}'`:"";return Fe(`grep -rn ${r} --exclude-dir=node_modules --exclude-dir=.git '${t.pattern.replace(/'/g,"'\\''")}' . 2>/dev/null | head -100`,{cwd:e,encoding:"utf-8",timeout:15e3}).trim()||"No matches found"}catch{return"No matches found"}}})}}async function ct(e){let{provider:t,model:r,apiKey:o,prompt:n,systemPrompt:i,workingDir:s,maxTokens:a=16384,temperature:l=.7,timeoutMs:I=6e5,maxSteps:b=15,enableTools:p=!0}=e,m=go(t,r,o),k=p&&s?ho(s):void 0,N=new AbortController,R=setTimeout(()=>N.abort(),I);try{return(await so({model:m,prompt:n,system:i,maxOutputTokens:a,temperature:l,tools:k,stopWhen:k?ao(b):void 0,abortSignal:N.signal})).text}finally{clearTimeout(R)}}function wo(e,t){let r=[e.usage,e.message?.usage,e.result?.usage];for(let o of r)if(o&&typeof o=="object"){let n=o;typeof n.input_tokens=="number"&&(t.inputTokens=Math.max(t.inputTokens,n.input_tokens)),typeof n.output_tokens=="number"&&(t.outputTokens=Math.max(t.outputTokens,n.output_tokens)),typeof n.cache_creation_input_tokens=="number"&&(t.cacheCreationTokens=Math.max(t.cacheCreationTokens,n.cache_creation_input_tokens)),typeof n.cache_read_input_tokens=="number"&&(t.cacheReadTokens=Math.max(t.cacheReadTokens,n.cache_read_input_tokens))}}async function ut(e,t,r,o){if(!(t.inputTokens===0&&t.outputTokens===0))try{await M.post(`/api/tasks/${e}/usage/partial`,{inputTokens:t.inputTokens,outputTokens:t.outputTokens,cacheCreationTokens:t.cacheCreationTokens,cacheReadTokens:t.cacheReadTokens,model:r,mode:o})}catch{}}var te=3;function g(){return c.dim(new Date().toLocaleTimeString())}var he=[],le=null;async function pt(){for(;he.length>0;){let e=he.splice(0,50);try{await M.post("/api/control-center/logs/batch",{entries:e},{timeout:5e3})}catch{}}}async function A(e,t,r="system",o="info"){he.length>=200&&he.shift(),he.push({taskId:e,message:t,type:r,severity:o}),le||(le=pt().finally(()=>{le=null}))}async function Eo(){le&&await le,he.length>0&&(le=pt().finally(()=>{le=null}),await le)}async function ye(e,t,r,o,n,i){try{await M.post("/api/agent/planning-progress",{taskId:e,phase:t,elapsedSeconds:r,detail:o,charsGenerated:n,toolCallCount:i})}catch{}}var $="[\u{1F5FA}\uFE0F planning_agent \u{1F916}]";function _e(e){let t=Math.floor(e/60),r=e%60;return t>0?`${t}m ${r}s`:`${r}s`}function dt(e,t){switch(e){case"initializing":return`${$} Starting planning agent...`;case"reading_repo":return`${$} Reading repository structure...`;case"analyzing":return`${$} Analyzing requirements...`;case"generating_plan":return`${$} Planning in progress \u2014 analyzing requirements and decomposing into steps (${_e(t)} elapsed)`;case"validating":return`${$} Validating plan...`;case"complete":return`${$} Planning complete`}}function Io(e,t,r,o,n,i,s){let a=c.cyan(n.slice(0,8));return new Promise((l,I)=>{let p=yo(e,["--print","--verbose","--output-format","stream-json","--model",t,"--permission-mode","bypassPermissions"],{cwd:s,env:o,stdio:["pipe","pipe","pipe"]});p.stdin.write(r),p.stdin.end();let m="",k="",N="",R=0,K=0,P={inputTokens:0,outputTokens:0,cacheCreationTokens:0,cacheReadTokens:0},G=t,h="";function D(f=!1){if(!h)return;let F=h.split(`
|
|
7
|
+
`),oe=f?"":F.pop()||"";for(let W of F)if(W.trim()){A(n,`${$} ${W}`,"output");let x=W.trim().length>160?W.trim().substring(0,160)+"\u2026":W.trim();console.log(`${g()} ${a} ${c.dim("\u{1F4AD}")} ${c.dim(x)}`)}h=oe}let w="initializing",v=!1,d={started:!0,reading:!1,analyzing:!1,generating:!1};function C(f){if(f===w)return;w=f;let F=Math.round((Date.now()-i)/1e3),oe=dt(f,F);A(n,oe),console.log(`${g()} ${a} ${c.dim(oe)}`)}let j=setInterval(()=>D(),500),B=setInterval(()=>{let f=Math.round((Date.now()-i)/1e3);ye(n,w,f,dt(w,f),R,K)},2e3),_=0,Z=setInterval(()=>{let f=Math.round((Date.now()-i)/1e3);if(w==="initializing"&&f>=5?C("reading_repo"):w==="reading_repo"&&f>=15&&!v&&C("analyzing"),w==="generating_plan"&&f-_>=30){_=f;let F=`${$} Planning in progress \u2014 analyzing requirements and decomposing into steps (${_e(f)} elapsed)`;A(n,F),console.log(`${g()} ${a} ${c.dim(F)}`)}},5e3),z="";p.stdout.on("data",f=>{z+=f.toString();let F=z.split(`
|
|
8
|
+
`);z=F.pop()||"";for(let oe of F){let W=oe.trim();if(W)try{let x=JSON.parse(W);if(x.type==="assistant"&&x.message?.content){let V=x.message.content;if(Array.isArray(V))for(let re of V)re.type==="text"&&re.text?(m+=re.text,R+=re.text.length,h+=re.text,v||(v=!0,K>0&&!d.analyzing&&(C("analyzing"),d.analyzing=!0)),R>500&&!d.generating&&(C("generating_plan"),d.generating=!0,_=Math.round((Date.now()-i)/1e3))):re.type==="tool_use"&&(K++,d.reading||(C("reading_repo"),d.reading=!0));else typeof V=="string"&&V&&(m+=V,R+=V.length,h+=V)}else x.type==="content_block_delta"&&x.delta?.text?(m+=x.delta.text,R+=x.delta.text.length,h+=x.delta.text,v||(v=!0,K>0&&!d.analyzing&&(C("analyzing"),d.analyzing=!0)),R>500&&!d.generating&&(C("generating_plan"),d.generating=!0,_=Math.round((Date.now()-i)/1e3))):x.type==="content_block_start"&&x.content_block?.type==="tool_use"?(K++,d.reading||(C("reading_repo"),d.reading=!0)):x.type==="result"&&x.result&&(k=typeof x.result=="string"?x.result:"");if(wo(x,P),x.type==="result"&&x.total_cost_usd!==void 0&&x.modelUsage&&typeof x.modelUsage=="object"){let V=Object.keys(x.modelUsage);V.length>0&&(G=V[0])}}catch{m+=W+`
|
|
9
|
+
`,R+=W.length}}}),p.stderr.on("data",f=>{N+=f.toString()});let X=setInterval(()=>{(P.inputTokens>0||P.outputTokens>0)&&ut(n,P,G,"greatest").catch(()=>{})},3e4);function ee(){clearInterval(Z),clearInterval(B),clearInterval(j),clearInterval(X),D(!0)}let Te=setTimeout(()=>{ee(),p.kill("SIGTERM"),I(new Error("Claude CLI timed out after 20 minutes"))},12e5);p.on("exit",f=>{clearTimeout(Te),ee();let F=Math.round((Date.now()-i)/1e3);ye(n,"validating",F,"Validating plan...",R,K),ut(n,P,G,"greatest").catch(()=>{}),f!==0?I(new Error(`Claude CLI failed (exit ${f}): ${N.substring(0,300)}`)):l(k||m)}),p.on("error",f=>{clearTimeout(Te),ee(),I(f)})})}function To(e,t){if(t)switch(e){case"anthropic":return t.anthropicApiKey;case"openai":return t.openaiApiKey;case"google":return t.googleApiKey;case"ollama":return t.ollamaBaseUrl||"http://localhost:11434";default:return}}function bo(e,t,r){switch(r){case"bitbucket":return`https://x-token-auth:${t}@bitbucket.org/${e}.git`;case"gitlab":return`https://oauth2:${t}@gitlab.com/${e}.git`;default:return`https://x-access-token:${t}@github.com/${e}.git`}}async function Ao(e,t,r,o){let n=c.cyan(o.slice(0,8)),i=`/tmp/workermill-planning-${o.slice(0,8)}-${Date.now()}`;try{let s=bo(e,t,r);return console.log(`${g()} ${n} ${c.dim("Cloning repo for planner...")}`),je(`git clone --depth 1 --single-branch "${s}" "${i}"`,{stdio:"ignore",timeout:6e4}),console.log(`${g()} ${n} ${c.green("\u2713")} Repo cloned to ${c.dim(i)}`),i}catch(s){let a=s instanceof Error?s.message:String(s);console.error(`${g()} ${n} ${c.yellow("\u26A0")} Clone failed, planner will run without repo access: ${a.substring(0,100)}`);try{je(`rm -rf "${i}"`,{stdio:"ignore"})}catch{}return null}}async function gt(e,t,r){let o=c.cyan(e.id.slice(0,8));console.log(`${g()} ${o} Fetching planning prompt...`),await A(e.id,`${$} Fetching planning prompt from cloud API...`);let n=await M.get("/api/agent/planning-prompt",{params:{taskId:e.id}}),{prompt:i,model:s,provider:a,maxStories:l,maxTargetFiles:I}=n.data,b=typeof l=="number"?l:8,p=s,m=a||"anthropic",k=m==="anthropic",N=process.env.CLAUDE_CLI_PATH||ue()||"claude",R={...process.env};delete R.CLAUDE_CODE_OAUTH_TOKEN;let K=To(m,r),P=Date.now(),G=e.description||e.summary,h=null;if(e.githubRepo){let B=e.scmProvider||"github",_=B==="bitbucket"?t.bitbucketToken:B==="gitlab"?t.gitlabToken:t.githubToken;_?h=await Ao(e.githubRepo,_,B,e.id):console.log(`${g()} ${o} ${c.yellow("\u26A0")} No SCM token for ${B}, planner will run without repo access`)}let D=i,w=null,v=0,d=[],C=0,j=await Qe(typeof I=="number"?I:void 0);j||(console.log(`${g()} ${o} ${c.yellow("\u26A0")} Could not fetch critic config \u2014 critic validation will be skipped`),await A(e.id,`${$} \u26A0\uFE0F Could not fetch critic config from API \u2014 critic validation will be skipped`));try{for(let _=1;_<=te;_++){let Z=te>1?` (attempt ${_}/${te})`:"",z=`${m}/${p}`;_>1?(console.log(`${g()} ${o} Running planner${Z} ${c.dim(`(${c.yellow(z)})`)}`),await A(e.id,`${$} Re-planning${Z} using ${z}`)):(console.log(`${g()} ${o} Running planner ${c.dim(`(${c.yellow(z)})`)}`),await A(e.id,`${$} Starting planning agent using ${z}`));let X;try{if(k)X=await Io(N,p,D,R,e.id,P,h||void 0);else{if(!K)throw new Error(`No API key available for provider "${m}". Configure it in Settings > Integrations.`);let T=Math.round((Date.now()-P)/1e3);await ye(e.id,"generating_plan",T,"Generating plan via AI SDK...",0,0),X=await ct({provider:m,model:p,apiKey:K,prompt:D,workingDir:h||void 0,enableTools:!!h,maxSteps:10});let S=Math.round((Date.now()-P)/1e3);await ye(e.id,"validating",S,"Validating plan...",X.length,0)}}catch(T){let S=Math.round((Date.now()-P)/1e3),J=T instanceof Error?T.message:String(T);return console.error(`${g()} ${o} ${c.red("\u2717")} Failed after ${S}s: ${J.substring(0,100)}`),await A(e.id,`${$} Planning failed after ${_e(S)}: ${J.substring(0,200)}`,"error","error"),!1}let ee=Math.round((Date.now()-P)/1e3),Te=k?"Claude CLI":`${m} API`;console.log(`${g()} ${o} ${c.green("\u2713")} ${Te} done ${c.dim(`(${ee}s, ${X.length} chars)`)}`);let f;try{f=Ze(X)}catch(T){let S=T instanceof Error?T.message:String(T);return console.error(`${g()} ${o} ${c.red("\u2717")} Plan parse failed: ${S.substring(0,100)}`),await A(e.id,`${$} Failed to parse execution plan from Claude output: ${S.substring(0,200)}`,"error","error"),await _o(e.id,X,t.agentId,o,ee)}let{truncatedCount:F,details:oe}=et(f);if(F>0){C+=F;let T=`${$} File cap applied: ${F} stories truncated to max ${j?.maxTargetFiles??5} targetFiles`;console.log(`${g()} ${o} ${c.yellow("\u26A0")} ${T}`),await A(e.id,T);for(let S of oe)console.log(`${g()} ${o} ${c.dim(S)}`)}let{droppedCount:W,details:x}=tt(f,b);if(W>0){let T=`${$} Story cap applied: ${W} stories dropped (max ${b})`;console.log(`${g()} ${o} ${c.yellow("\u26A0")} ${T}`),await A(e.id,T);for(let S of x)console.log(`${g()} ${o} ${c.dim(S)}`)}let{resolvedCount:V,details:re}=ot(f);if(V>0){let T=`${$} File overlap resolved: ${V} shared file(s) de-duped across stories`;console.log(`${g()} ${o} ${c.yellow("\u26A0")} ${T}`),await A(e.id,T);for(let S of re)console.log(`${g()} ${o} ${c.dim(S)}`)}console.log(`${g()} ${o} Plan: ${c.bold(f.stories.length)} stories (max ${b})`),await A(e.id,`${$} Plan generated: ${f.stories.length} stories (${_e(ee)}). Running critic validation...`);let E=await at(N,p,G,f,R,o,m,K,e.id);if(E&&E.score>v?(w=f,v=E.score):!E&&!w&&(w=f),E&&d.push({iteration:_,score:E.score,approved:E.approved||E.score>=q,risks:E.risks,suggestions:E.suggestions,filesCapApplied:F>0?F:void 0}),!E){let T=`${$} \u26A0\uFE0F CRITIC BYPASSED \u2014 Critic validation failed (timeout/parse error). Posting plan WITHOUT quality gate.`;console.log(`${g()} ${o} ${c.yellow("\u26A0")} ${T}`),await A(e.id,T,"error","warning");let S=Date.now()-P;return await Ke(e.id,f,t.agentId,o,ee,void 0,void 0,d,C,S,_)}if(E.approved||E.score>=q){let T=`${$} Critic approved (score: ${E.score}/100)`;if(console.log(`${g()} ${o} ${c.green("\u2713")} ${T}`),await A(e.id,T),E.risks.length>0){let J=`${$} Critic risks (non-blocking): ${E.risks.join("; ")}`;console.log(`${g()} ${o} ${c.dim(J)}`),await A(e.id,J)}let S=Date.now()-P;return await Ke(e.id,f,t.agentId,o,ee,E.score,E.risks,d,C,S,_)}if(_<te){let T=nt(E);D=D+`
|
|
10
10
|
|
|
11
|
-
`+b;let R=`${$} Critic rejected (score: ${g.score}/100, threshold: ${X}). Re-planning with feedback...`;if(console.log(`${m()} ${o} ${c.yellow("\u26A0")} ${R}`),await A(e.id,R),g.risks.length>0){let H=`${$} Critic risks: ${g.risks.join("; ")}`;console.log(`${m()} ${o} ${c.dim(H)}`),await A(e.id,H)}if(g.suggestions&&g.suggestions.length>0){let H=`${$} Critic suggestions: ${g.suggestions.join("; ")}`;console.log(`${m()} ${o} ${c.dim(H)}`),await A(e.id,H)}}else{let b=`${$} Critic rejected after ${te} iterations (best score: ${f}/100, threshold: ${X})`;if(console.error(`${m()} ${o} ${c.red("\u2717")} ${b}`),await A(e.id,b,"error","error"),g.risks.length>0){let R=`${$} Final risks: ${g.risks.join("; ")}`;console.error(`${m()} ${o} ${R}`),await A(e.id,R,"error","error")}if(g.suggestions&&g.suggestions.length>0){let R=`${$} Suggestions: ${g.suggestions.join("; ")}`;console.error(`${m()} ${o} ${R}`),await A(e.id,R,"error","error")}}}let O=50;if(L&&f>=O){let F=Math.round((Date.now()-x)/1e3),j=`${$} Best-plan fallback: posting plan with score ${f}/100 (below ${X} threshold, above ${O} minimum)`;console.log(`${m()} ${o} ${c.yellow("\u26A0")} ${j}`),await A(e.id,j);let Q=Date.now()-x;if(await Fe(e.id,L,t.agentId,o,F,f,[`Best-plan fallback: critic rejected after ${te} iterations`],U,p,Q,te))return!0;console.log(`${m()} ${o} ${c.yellow("\u26A0")} ${$} Fallback post rejected by server, reporting plan-failed`),await A(e.id,`${$} Fallback plan rejected by server \u2014 reporting failure`)}try{let F=L&&f>=O?`Best-plan fallback rejected by server after ${te} iterations (best score: ${f}/100)`:`Critic rejected after ${te} iterations (best score: ${f}/100, threshold: ${X}, fallback minimum: ${O})`;await P.post("/api/agent/plan-failed",{taskId:e.id,agentId:t.agentId,reason:F,criticHistory:U})}catch{}return!1}finally{if(await wo(),K)try{Ke(`rm -rf "${K}"`,{stdio:"ignore"})}catch{}}}async function Fe(e,t,r,o,n,i,s,a,l,w,I){let d=ot(t);try{let k=(await P.post("/api/agent/plan-result",{taskId:e,rawOutput:d,agentId:r,criticScore:i,criticRisks:s,criticHistory:a,criticIterations:I,fileCapTruncations:l,planningDurationMs:w})).data.storyCount;return console.log(`${m()} ${o} ${c.green("\u2713")} Plan validated: ${c.bold(k)} stories \u2192 ${c.green("queued")}`),await A(e,`${$} Plan validated: ${k} stories. Task queued for execution.`),await $e(e,"complete",n,"Planning complete",0,0),!0}catch(E){let k=E,C=k.response?.data?.error||k.response?.data?.detail||String(E),_=k.response?.status?` (${k.response.status})`:"";return console.error(`${m()} ${o} ${c.red("\u2717")} Server validation failed${_}: ${C.substring(0,100)}`),await A(e,`${$} Server-side plan validation failed${_}: ${C.substring(0,200)}`,"error","error"),!1}}async function Ao(e,t,r,o,n){try{let s=(await P.post("/api/agent/plan-result",{taskId:e,rawOutput:t,agentId:r})).data.storyCount;return console.log(`${m()} ${o} ${c.green("\u2713")} Plan validated (server-side): ${c.bold(s)} stories \u2192 ${c.green("queued")}`),await A(e,`${$} Plan validated: ${s} stories. Task queued for execution.`),await $e(e,"complete",n,"Planning complete",0,0),!0}catch(i){let a=i.response?.data?.detail||String(i);return console.error(`${m()} ${o} ${c.red("\u2717")} Validation failed: ${a.substring(0,100)}`),await A(e,`${$} Plan validation failed: ${a.substring(0,200)}`,"error","error"),!1}}import u from"chalk";import{spawn as mt,execSync as Se}from"child_process";import*as _e from"path";import*as Z from"fs";import*as ke from"os";function v(){return u.dim(new Date().toLocaleTimeString())}var _o=[[/(:\/\/[^:/?#]+:)[^@]+(@)/g,"$1***$2"],[/\b(ghp_|gho_|ghs_|github_pat_)[A-Za-z0-9_]+/g,"$1***"],[/\bglpat-[A-Za-z0-9\-_]+/g,"glpat-***"],[/\b(AKIA)[A-Z0-9]{16}\b/g,"$1***"],[/(Bearer\s+)[A-Za-z0-9._\-]+/gi,"$1***"],[/(x-api-key:\s*)[^\s,'"]+/gi,"$1***"]];function Re(e){let t=e;for(let[r,o]of _o)t=t.replace(r,o);return t}var V=new Map;function Ro(){if(process.env.WSL_DISTRO_NAME||process.env.WSL_INTEROP)return!0;try{return Z.readFileSync("/proc/version","utf-8").toLowerCase().includes("microsoft")}catch{return!1}}var Pe=Ro(),ft=Pe||process.platform==="darwin"||process.platform==="win32";function $t(){if(!Pe)return null;try{let e=Se("hostname -I",{encoding:"utf-8"}).trim().split(/\s+/)[0];if(e&&/^\d+\.\d+\.\d+\.\d+$/.test(e))return e}catch{}return null}function ht(e){if(!Pe)return e;let t=e.match(/^\/mnt\/([a-zA-Z])\/(.*)$/);return t?`${t[1].toUpperCase()}:/${t[2]}`:e}function yt(){let e=_e.join(ke.homedir(),".claude");if(Z.existsSync(e))return e;if(Pe){let t="/mnt/c/Users";if(Z.existsSync(t))try{for(let r of Z.readdirSync(t)){if(["Public","Default","Default User","All Users"].includes(r))continue;let o=_e.join(t,r,".claude");if(Z.existsSync(o))return o}}catch{}}return null}var gt=0;function wt(e){let t=e.match(/^(\d+\.dkr\.ecr\.[a-z0-9-]+\.amazonaws\.com)/);return t?t[1]:null}function So(e){let t=e.match(/\.ecr\.([a-z0-9-]+)\.amazonaws\.com/);return t?t[1]:"us-east-1"}function Et(e){if(Date.now()<gt)return!0;let t=So(e);try{return Se(`aws ecr get-login-password --region ${t} | docker login --username AWS --password-stdin ${e}`,{stdio:"pipe",timeout:3e4}),gt=Date.now()+660*60*1e3,!0}catch{return!1}}function It(e,t,r){if(r?.scmToken)return r.scmToken;switch(e){case"bitbucket":return t.bitbucketToken;case"gitlab":return t.gitlabToken;default:return t.githubToken}}function ko(e){let t=e.jiraFields;if(!t)return!1;let r=t.labels;if(Array.isArray(r)&&r.some(i=>typeof i=="string"&&i.toLowerCase()==="self-review"))return!0;let n=t.issue?.labels;return!!(Array.isArray(n)&&n.some(i=>typeof i=="string"?i.toLowerCase()==="self-review":i&&typeof i=="object"&&"name"in i?(i.name||"").toLowerCase()==="self-review":!1))}async function Tt(e,t,r,o){let n=u.cyan(e.id.slice(0,8));if(V.has(e.id)){console.log(`${v()} ${n} ${u.dim("Already running, skipping")}`);return}let i=`workermill-${e.id.slice(0,8)}`,a=["run","--rm",...!t.workerImage.includes("/")?[]:["--pull","always"],"--name",i],l=Math.round(ke.totalmem()/(1024*1024*1024));if(l<=16?a.push("--memory","6g","--memory-swap","10g","--cpus","2"):(l<=32,a.push("--memory","6g","--memory-swap","12g","--cpus","4")),ft){let p=$t();a.push(`--add-host=host.docker.internal:${p||"host-gateway"}`)}else a.push("--network","host");let w=e.workerProvider||"anthropic",I=yt();if(!I&&w==="anthropic"){console.error(`${v()} ${n} ${u.red("\u2717")} Claude credentials not found. Run 'claude' and complete the sign-in flow.`);return}if(I){let p=_e.join(I,".credentials.json");try{Z.chmodSync(p,438)}catch{}let M=ht(I);a.push("-v",`${M}:/home/worker/.claude`)}else console.log(`${v()} ${n} ${u.dim("Skipping Claude mount (non-Anthropic worker)")}`);let d=t.apiUrl.replace(/localhost|127\.0\.0\.1/,"host.docker.internal"),E=e.scmProvider||"github",k=It(E,t,o),C=o?.githubToken||t.githubToken,_=E==="bitbucket"&&o?.scmToken||t.bitbucketToken,x=E==="gitlab"&&o?.scmToken||t.gitlabToken,G={NODE_OPTIONS:"--max-old-space-size=3072",EPIC_MODE:"true",EXECUTION_MODE:"local",TASK_ID:e.id,ORG_ID:e.orgId||"",JIRA_ISSUE_KEY:e.jiraIssueKey||"",JIRA_SUMMARY:e.summary||"",JIRA_DESCRIPTION:e.description||"",TASK_SUMMARY:e.summary||"",TASK_DESCRIPTION:e.description||"",WORKER_PERSONA:e.workerPersona||"",RETRY_NUMBER:String(e.retryCount??0),TICKET_KEY:e.jiraIssueKey||"",API_BASE_URL:d,ORG_API_KEY:t.apiKey,SCM_PROVIDER:E,SCM_TOKEN:k,SCM_BASE_URL:o?.scmBaseUrl||"",GITHUB_TOKEN:C,GH_TOKEN:C,GITHUB_REVIEWER_TOKEN:o?.githubReviewerToken||"",BITBUCKET_TOKEN:_,BITBUCKET_USERNAME:o?.bitbucketUsername||"x-token-auth",GITLAB_TOKEN:x,TARGET_REPO:e.githubRepo||"",GITHUB_REPO:e.githubRepo||"",WORKER_MODEL:e.workerModel||String(r.defaultWorkerModel||""),CLAUDE_MODEL:e.workerModel||String(r.defaultWorkerModel||""),JIRA_BASE_URL:o?.jiraBaseUrl||"",JIRA_EMAIL:o?.jiraEmail||"",JIRA_API_TOKEN:o?.jiraApiToken||"",TICKET_SYSTEM:o?.issueTrackerProvider||"jira",LINEAR_API_KEY:o?.linearApiKey||"",AWS_ACCESS_KEY_ID:o?.customerAwsAccessKeyId||"",AWS_SECRET_ACCESS_KEY:o?.customerAwsSecretAccessKey||"",AWS_DEFAULT_REGION:o?.customerAwsRegion||"",AWS_REGION:o?.customerAwsRegion||"",CUSTOMER_AWS_ROLE_ARN:o?.customerAwsRoleArn||"",CUSTOMER_AWS_EXTERNAL_ID:o?.customerAwsExternalId||"",CUSTOMER_AWS_REGION:o?.customerAwsRegion||"",MANAGER_PROVIDER:o?.managerProvider||"anthropic",MANAGER_MODEL:o?.managerModelId||"",BITBUCKET_EMAIL:o?.bitbucketEmail||"",DEPLOYMENT_ENABLED:e.deploymentEnabled||e.parentTaskId?"true":"false",PRD_CHILD_TASK:e.parentTaskId?"true":"false",IMPROVEMENT_ENABLED:e.improvementEnabled?"true":"false",QUALITY_GATE_BYPASS:e.qualityGateBypass?"true":"false",STANDARD_SDK_MODE:e.standardSdkMode?"true":"false",MAX_REVIEW_REVISIONS:String(r.maxReviewRevisions??3),CODEBASE_INDEXING_ENABLED:r.codebaseIndexingEnabled===!0?"true":"false",EXISTING_PR_URL:e.githubPrUrl||"",EXISTING_PR_NUMBER:e.githubPrNumber?String(e.githubPrNumber):"",PARENT_TASK_ID:e.parentTaskId||e.id,PARENT_JIRA_KEY:e.jiraIssueKey&&/-S\d+$/.test(e.jiraIssueKey)&&e.jiraFields?.parentJiraKey||"",TARGET_BRANCH:e.jiraFields?.targetBranch||"",STORY_BRANCH:e.jiraFields?.storyBranch||"",TASK_NOTES:e.taskNotes||"",TARGET_FILES:JSON.stringify(e.jiraFields?.targetFiles||[]),REFERENCE_FILES:JSON.stringify(e.jiraFields?.referenceFiles||[]),PIPELINE_VERSION:e.jiraFields?.pipelineVersion||e.pipelineVersion||"",V2_STEP_INPUT:e.jiraFields?.v2StepInput?JSON.stringify(e.jiraFields.v2StepInput):"",EXECUTION_MODE_SETTING:e.jiraFields?.executionMode||"autonomous",ANTHROPIC_API_KEY:I?"":o?.anthropicApiKey||process.env.ANTHROPIC_API_KEY||"",WORKER_PROVIDER:e.workerProvider||"anthropic",OPENAI_API_KEY:o?.openaiApiKey||"",GOOGLE_API_KEY:o?.googleApiKey||"",GOOGLE_GENERATIVE_AI_API_KEY:o?.googleApiKey||"",OLLAMA_HOST:o?.ollamaBaseUrl||"",OLLAMA_CONTEXT_WINDOW:o?.ollamaContextWindow?String(o.ollamaContextWindow):"",VLLM_BASE_URL:o?.vllmBaseUrl||"",BLOCKER_MAX_AUTO_RETRIES:String(r.blockerMaxAutoRetries??3),BLOCKER_AUTO_RETRY_ENABLED:r.blockerAutoRetryEnabled!==!1?"true":"false",PUSH_AFTER_COMMIT:r.pushAfterCommit!==!1?"true":"false",GRACEFUL_SHUTDOWN_ENABLED:r.gracefulShutdownEnabled!==!1?"true":"false",MAX_PARALLEL_EXPERTS:String(r.maxParallelExperts??4),REVIEW_ENABLED:e.skipManagerReview===!1?"true":"false",SELF_REVIEW_ENABLED:ko(e)||r.selfReviewEnabled!==!1?"true":"false"};for(let[p,M]of Object.entries(G))M!==""&&a.push("-e",`${p}=${M}`);let K=t.workerImage,y=wt(K);y&&Et(y),a.push(K);let L=e.skipManagerReview===!1;console.log(`${v()} ${n} ${u.dim("Starting container")} ${u.yellow(i)}`),console.log(`${v()} ${n} ${u.dim(` skipManagerReview=${e.skipManagerReview} \u2192 REVIEW_ENABLED=${L}`)}`),console.log(`${v()} ${n} ${u.dim(` model=${e.workerModel} repo=${e.githubRepo}`)}`),console.log(`${v()} ${n} ${u.dim(` totalRamGB=${l} docker args:`)} ${a.slice(0,10).join(" ")}`);let f=mt("docker",a,{stdio:["ignore","pipe","pipe"],detached:!1});if(!f.pid){console.error(`${v()} ${n} ${u.red("\u2717")} Failed to spawn container`);return}let U={taskId:e.id,containerName:i,process:f,startedAt:new Date,status:"running",resultEmitted:!1};V.set(e.id,U),f.stdout?.on("data",p=>{let M=p.toString().split(`
|
|
12
|
-
`).filter(
|
|
13
|
-
`).filter(
|
|
14
|
-
`).filter(
|
|
15
|
-
`).filter(
|
|
11
|
+
`+T;let S=`${$} Critic rejected (score: ${E.score}/100, threshold: ${q}). Re-planning with feedback...`;if(console.log(`${g()} ${o} ${c.yellow("\u26A0")} ${S}`),await A(e.id,S),E.risks.length>0){let J=`${$} Critic risks: ${E.risks.join("; ")}`;console.log(`${g()} ${o} ${c.dim(J)}`),await A(e.id,J)}if(E.suggestions&&E.suggestions.length>0){let J=`${$} Critic suggestions: ${E.suggestions.join("; ")}`;console.log(`${g()} ${o} ${c.dim(J)}`),await A(e.id,J)}}else{let T=`${$} Critic rejected after ${te} iterations (best score: ${v}/100, threshold: ${q})`;if(console.error(`${g()} ${o} ${c.red("\u2717")} ${T}`),await A(e.id,T,"error","error"),E.risks.length>0){let S=`${$} Final risks: ${E.risks.join("; ")}`;console.error(`${g()} ${o} ${S}`),await A(e.id,S,"error","error")}if(E.suggestions&&E.suggestions.length>0){let S=`${$} Suggestions: ${E.suggestions.join("; ")}`;console.error(`${g()} ${o} ${S}`),await A(e.id,S,"error","error")}}}let B=50;if(w&&v>=B){let _=Math.round((Date.now()-P)/1e3),Z=`${$} Best-plan fallback: posting plan with score ${v}/100 (below ${q} threshold, above ${B} minimum)`;console.log(`${g()} ${o} ${c.yellow("\u26A0")} ${Z}`),await A(e.id,Z);let z=Date.now()-P;if(await Ke(e.id,w,t.agentId,o,_,v,[`Best-plan fallback: critic rejected after ${te} iterations`],d,C,z,te))return!0;console.log(`${g()} ${o} ${c.yellow("\u26A0")} ${$} Fallback post rejected by server, reporting plan-failed`),await A(e.id,`${$} Fallback plan rejected by server \u2014 reporting failure`)}try{let _=w&&v>=B?`Best-plan fallback rejected by server after ${te} iterations (best score: ${v}/100)`:`Critic rejected after ${te} iterations (best score: ${v}/100, threshold: ${q}, fallback minimum: ${B})`;await M.post("/api/agent/plan-failed",{taskId:e.id,agentId:t.agentId,reason:_,criticHistory:d})}catch{}return!1}finally{if(await Eo(),h)try{je(`rm -rf "${h}"`,{stdio:"ignore"})}catch{}}}async function Ke(e,t,r,o,n,i,s,a,l,I,b){let p=rt(t);try{let k=(await M.post("/api/agent/plan-result",{taskId:e,rawOutput:p,agentId:r,criticScore:i,criticRisks:s,criticHistory:a,criticIterations:b,fileCapTruncations:l,planningDurationMs:I})).data.storyCount;return console.log(`${g()} ${o} ${c.green("\u2713")} Plan validated: ${c.bold(k)} stories \u2192 ${c.green("queued")}`),await A(e,`${$} Plan validated: ${k} stories. Task queued for execution.`),await ye(e,"complete",n,"Planning complete",0,0),!0}catch(m){let k=m,N=k.response?.data?.error||k.response?.data?.detail||String(m),R=k.response?.status?` (${k.response.status})`:"";return console.error(`${g()} ${o} ${c.red("\u2717")} Server validation failed${R}: ${N.substring(0,100)}`),await A(e,`${$} Server-side plan validation failed${R}: ${N.substring(0,200)}`,"error","error"),!1}}async function _o(e,t,r,o,n){try{let s=(await M.post("/api/agent/plan-result",{taskId:e,rawOutput:t,agentId:r})).data.storyCount;return console.log(`${g()} ${o} ${c.green("\u2713")} Plan validated (server-side): ${c.bold(s)} stories \u2192 ${c.green("queued")}`),await A(e,`${$} Plan validated: ${s} stories. Task queued for execution.`),await ye(e,"complete",n,"Planning complete",0,0),!0}catch(i){let a=i.response?.data?.detail||String(i);return console.error(`${g()} ${o} ${c.red("\u2717")} Validation failed: ${a.substring(0,100)}`),await A(e,`${$} Plan validation failed: ${a.substring(0,200)}`,"error","error"),!1}}import u from"chalk";import{spawn as ft,execSync as ke}from"child_process";import*as Re from"path";import*as Q from"fs";import*as Pe from"os";function O(){return u.dim(new Date().toLocaleTimeString())}var Ro=[[/(:\/\/[^:/?#]+:)[^@]+(@)/g,"$1***$2"],[/\b(ghp_|gho_|ghs_|github_pat_)[A-Za-z0-9_]+/g,"$1***"],[/\bglpat-[A-Za-z0-9\-_]+/g,"glpat-***"],[/\b(AKIA)[A-Z0-9]{16}\b/g,"$1***"],[/(Bearer\s+)[A-Za-z0-9._\-]+/gi,"$1***"],[/(x-api-key:\s*)[^\s,'"]+/gi,"$1***"]];function Se(e){let t=e;for(let[r,o]of Ro)t=t.replace(r,o);return t}var Y=new Map;function So(){if(process.env.WSL_DISTRO_NAME||process.env.WSL_INTEROP)return!0;try{return Q.readFileSync("/proc/version","utf-8").toLowerCase().includes("microsoft")}catch{return!1}}var ve=So(),$t=ve||process.platform==="darwin"||process.platform==="win32";function ht(){if(!ve)return null;try{let e=ke("hostname -I",{encoding:"utf-8"}).trim().split(/\s+/)[0];if(e&&/^\d+\.\d+\.\d+\.\d+$/.test(e))return e}catch{}return null}function yt(e){if(!ve)return e;let t=e.match(/^\/mnt\/([a-zA-Z])\/(.*)$/);return t?`${t[1].toUpperCase()}:/${t[2]}`:e}function wt(){let e=Re.join(Pe.homedir(),".claude");if(Q.existsSync(e))return e;if(ve){let t="/mnt/c/Users";if(Q.existsSync(t))try{for(let r of Q.readdirSync(t)){if(["Public","Default","Default User","All Users"].includes(r))continue;let o=Re.join(t,r,".claude");if(Q.existsSync(o))return o}}catch{}}return null}var mt=0;function Et(e){let t=e.match(/^(\d+\.dkr\.ecr\.[a-z0-9-]+\.amazonaws\.com)/);return t?t[1]:null}function ko(e){let t=e.match(/\.ecr\.([a-z0-9-]+)\.amazonaws\.com/);return t?t[1]:"us-east-1"}function It(e){if(Date.now()<mt)return!0;let t=ko(e);try{return ke(`aws ecr get-login-password --region ${t} | docker login --username AWS --password-stdin ${e}`,{stdio:"pipe",timeout:3e4}),mt=Date.now()+660*60*1e3,!0}catch{return!1}}function Tt(e,t,r){if(r?.scmToken)return r.scmToken;switch(e){case"bitbucket":return t.bitbucketToken;case"gitlab":return t.gitlabToken;default:return t.githubToken}}function Po(e){let t=e.jiraFields;if(!t)return!1;let r=t.labels;if(Array.isArray(r)&&r.some(i=>typeof i=="string"&&i.toLowerCase()==="self-review"))return!0;let n=t.issue?.labels;return!!(Array.isArray(n)&&n.some(i=>typeof i=="string"?i.toLowerCase()==="self-review":i&&typeof i=="object"&&"name"in i?(i.name||"").toLowerCase()==="self-review":!1))}async function bt(e,t,r,o){let n=u.cyan(e.id.slice(0,8));if(Y.has(e.id)){console.log(`${O()} ${n} ${u.dim("Already running, skipping")}`);return}let i=`workermill-${e.id.slice(0,8)}`,a=["run","--rm",...!t.workerImage.includes("/")?[]:["--pull","always"],"--name",i],l=Math.round(Pe.totalmem()/(1024*1024*1024));if(l<=16?a.push("--memory","6g","--memory-swap","10g","--cpus","2"):(l<=32,a.push("--memory","6g","--memory-swap","12g","--cpus","4")),$t){let d=ht();a.push(`--add-host=host.docker.internal:${d||"host-gateway"}`)}else a.push("--network","host");let I=e.workerProvider||"anthropic",b=wt();if(!b&&I==="anthropic"){console.error(`${O()} ${n} ${u.red("\u2717")} Claude credentials not found. Run 'claude' and complete the sign-in flow.`);return}if(b){let d=Re.join(b,".credentials.json");try{Q.chmodSync(d,438)}catch{}let C=yt(b);a.push("-v",`${C}:/home/worker/.claude`)}else console.log(`${O()} ${n} ${u.dim("Skipping Claude mount (non-Anthropic worker)")}`);let p=t.apiUrl.replace(/localhost|127\.0\.0\.1/,"host.docker.internal"),m=e.scmProvider||"github",k=Tt(m,t,o),N=o?.githubToken||t.githubToken,R=m==="bitbucket"&&o?.scmToken||t.bitbucketToken,K=m==="gitlab"&&o?.scmToken||t.gitlabToken,P={NODE_OPTIONS:"--max-old-space-size=3072",EPIC_MODE:"true",EXECUTION_MODE:"local",TASK_ID:e.id,ORG_ID:e.orgId||"",JIRA_ISSUE_KEY:e.jiraIssueKey||"",JIRA_SUMMARY:e.summary||"",JIRA_DESCRIPTION:e.description||"",TASK_SUMMARY:e.summary||"",TASK_DESCRIPTION:e.description||"",WORKER_PERSONA:e.workerPersona||"",RETRY_NUMBER:String(e.retryCount??0),TICKET_KEY:e.jiraIssueKey||"",API_BASE_URL:p,ORG_API_KEY:t.apiKey,SCM_PROVIDER:m,SCM_TOKEN:k,SCM_BASE_URL:o?.scmBaseUrl||"",GITHUB_TOKEN:N,GH_TOKEN:N,GITHUB_REVIEWER_TOKEN:o?.githubReviewerToken||"",BITBUCKET_TOKEN:R,BITBUCKET_USERNAME:o?.bitbucketUsername||"x-token-auth",GITLAB_TOKEN:K,TARGET_REPO:e.githubRepo||"",GITHUB_REPO:e.githubRepo||"",WORKER_MODEL:e.workerModel||String(r.defaultWorkerModel||""),CLAUDE_MODEL:e.workerModel||String(r.defaultWorkerModel||""),JIRA_BASE_URL:o?.jiraBaseUrl||"",JIRA_EMAIL:o?.jiraEmail||"",JIRA_API_TOKEN:o?.jiraApiToken||"",TICKET_SYSTEM:o?.issueTrackerProvider||"jira",LINEAR_API_KEY:o?.linearApiKey||"",AWS_ACCESS_KEY_ID:o?.customerAwsAccessKeyId||"",AWS_SECRET_ACCESS_KEY:o?.customerAwsSecretAccessKey||"",AWS_DEFAULT_REGION:o?.customerAwsRegion||"",AWS_REGION:o?.customerAwsRegion||"",CUSTOMER_AWS_ROLE_ARN:o?.customerAwsRoleArn||"",CUSTOMER_AWS_EXTERNAL_ID:o?.customerAwsExternalId||"",CUSTOMER_AWS_REGION:o?.customerAwsRegion||"",MANAGER_PROVIDER:o?.managerProvider||"anthropic",MANAGER_MODEL:o?.managerModelId||"",BITBUCKET_EMAIL:o?.bitbucketEmail||"",DEPLOYMENT_ENABLED:e.deploymentEnabled||e.parentTaskId?"true":"false",PRD_CHILD_TASK:e.parentTaskId?"true":"false",IMPROVEMENT_ENABLED:e.improvementEnabled?"true":"false",QUALITY_GATE_BYPASS:e.qualityGateBypass?"true":"false",STANDARD_SDK_MODE:e.standardSdkMode?"true":"false",MAX_REVIEW_REVISIONS:String(r.maxReviewRevisions??3),CODEBASE_INDEXING_ENABLED:r.codebaseIndexingEnabled===!0?"true":"false",EXISTING_PR_URL:e.githubPrUrl||"",EXISTING_PR_NUMBER:e.githubPrNumber?String(e.githubPrNumber):"",PARENT_TASK_ID:e.parentTaskId||e.id,PARENT_JIRA_KEY:e.jiraIssueKey&&/-S\d+$/.test(e.jiraIssueKey)&&e.jiraFields?.parentJiraKey||"",TARGET_BRANCH:e.jiraFields?.targetBranch||"",STORY_BRANCH:e.jiraFields?.storyBranch||"",TASK_NOTES:e.taskNotes||"",TARGET_FILES:JSON.stringify(e.jiraFields?.targetFiles||[]),REFERENCE_FILES:JSON.stringify(e.jiraFields?.referenceFiles||[]),PIPELINE_VERSION:e.jiraFields?.pipelineVersion||e.pipelineVersion||"",V2_STEP_INPUT:e.jiraFields?.v2StepInput?JSON.stringify(e.jiraFields.v2StepInput):"",EXECUTION_MODE_SETTING:e.jiraFields?.executionMode||"autonomous",ANTHROPIC_API_KEY:b?"":o?.anthropicApiKey||process.env.ANTHROPIC_API_KEY||"",WORKER_PROVIDER:e.workerProvider||"anthropic",OPENAI_API_KEY:o?.openaiApiKey||"",GOOGLE_API_KEY:o?.googleApiKey||"",GOOGLE_GENERATIVE_AI_API_KEY:o?.googleApiKey||"",OLLAMA_HOST:o?.ollamaBaseUrl||"",OLLAMA_CONTEXT_WINDOW:o?.ollamaContextWindow?String(o.ollamaContextWindow):"",VLLM_BASE_URL:o?.vllmBaseUrl||"",BLOCKER_MAX_AUTO_RETRIES:String(r.blockerMaxAutoRetries??3),BLOCKER_AUTO_RETRY_ENABLED:r.blockerAutoRetryEnabled!==!1?"true":"false",PUSH_AFTER_COMMIT:r.pushAfterCommit!==!1?"true":"false",GRACEFUL_SHUTDOWN_ENABLED:r.gracefulShutdownEnabled!==!1?"true":"false",MAX_PARALLEL_EXPERTS:String(r.maxParallelExperts??4),REVIEW_ENABLED:e.skipManagerReview===!1?"true":"false",SELF_REVIEW_ENABLED:Po(e)||r.selfReviewEnabled!==!1?"true":"false"};for(let[d,C]of Object.entries(P))C!==""&&a.push("-e",`${d}=${C}`);let G=t.workerImage,h=Et(G);h&&It(h),a.push(G);let D=e.skipManagerReview===!1;console.log(`${O()} ${n} ${u.dim("Starting container")} ${u.yellow(i)}`),console.log(`${O()} ${n} ${u.dim(` skipManagerReview=${e.skipManagerReview} \u2192 REVIEW_ENABLED=${D}`)}`),console.log(`${O()} ${n} ${u.dim(` model=${e.workerModel} repo=${e.githubRepo}`)}`),console.log(`${O()} ${n} ${u.dim(` totalRamGB=${l} docker args:`)} ${a.slice(0,10).join(" ")}`);let w=ft("docker",a,{stdio:["ignore","pipe","pipe"],detached:!1});if(!w.pid){console.error(`${O()} ${n} ${u.red("\u2717")} Failed to spawn container`);return}let v={taskId:e.id,containerName:i,process:w,startedAt:new Date,status:"running",resultEmitted:!1};Y.set(e.id,v),w.stdout?.on("data",d=>{let C=d.toString().split(`
|
|
12
|
+
`).filter(j=>j.trim());for(let j of C)console.log(`${O()} ${n} ${u.dim(Se(j))}`),j.includes("::result::")&&(v.resultEmitted=!0)}),w.stderr?.on("data",d=>{let C=d.toString().split(`
|
|
13
|
+
`).filter(j=>j.trim());for(let j of C)console.log(`${O()} ${n} ${u.red(Se(j))}`)}),w.on("exit",d=>{v.status=d===0?"completed":"failed";let C=Math.round((Date.now()-v.startedAt.getTime())/1e3),j=d===0?u.green("\u2713"):u.red("\u2717"),B=d===0?u.green("completed"):u.red(`failed (exit ${d})`);console.log(`${O()} ${n} ${j} Container ${B} ${u.dim(`(${C}s)`)}`),v.resultEmitted||(console.log(`${O()} ${n} ${u.yellow("\u26A0")} No ::result:: marker seen \u2014 posting fallback completion in 15s`),setTimeout(async()=>{try{let _=d===0?"completed":"failed",Z=d!==0?`Worker container exited with code ${d} without reporting completion`:void 0;(await(await fetch(`${t.apiUrl}/api/tasks/${e.id}/worker-complete`,{method:"POST",headers:{"Content-Type":"application/json","x-api-key":t.apiKey},body:JSON.stringify({exitCode:d??1,result:_,errorMessage:Z})})).json()).status==="ignored"?console.log(`${O()} ${n} ${u.dim("Fallback completion ignored (task already transitioned)")}`):console.log(`${O()} ${n} ${u.yellow("\u26A0")} Fallback completion applied: ${_}`)}catch(_){console.error(`${O()} ${n} ${u.red("\u2717")} Fallback completion failed:`,_ instanceof Error?_.message:_)}},15e3)),setTimeout(()=>Y.delete(e.id),9e4)}),w.on("error",d=>{v.status="failed",console.error(`${O()} ${n} ${u.red("\u2717")} Container error: ${d.message}`)})}function Ge(){return Array.from(Y.values()).filter(e=>e.status==="running").length}function At(){return Array.from(Y.values()).filter(e=>e.status==="running").map(e=>e.taskId)}function _t(e){let t=Y.get(e);if(!t||t.status!=="running")return;let r=u.cyan(e.slice(0,8));console.log(`${O()} ${r} ${u.red("\u25A0")} Stopping container (cancelled by dashboard)`);try{ke(`docker stop ${t.containerName}`,{stdio:"ignore",timeout:15e3}),t.status="completed"}catch{}Y.delete(e)}async function Rt(){console.log(`${O()} ${u.dim(`Stopping ${Y.size} containers...`)}`);for(let[,e]of Y)if(e.status==="running")try{ke(`docker stop ${e.containerName}`,{stdio:"ignore",timeout:15e3}),e.status="completed"}catch{}Y.clear()}async function St(e,t,r){let o=u.cyan(e.id.slice(0,8)),n=`manager-${e.id}`;if(Y.has(n))return;let i=`wm-manager-${e.id.slice(0,8)}-${Date.now()}`,s=wt(),a=["run","--rm","--name",i,"--memory=4g","--cpus=2"];if($t){let h=ht();a.push(`--add-host=host.docker.internal:${h||"host-gateway"}`)}else a.push("--network","host");if(s){let h=yt(s);a.push("-v",`${h}:/home/worker/.claude`)}let l=t.apiUrl.replace(/localhost|127\.0\.0\.1/,"host.docker.internal"),I=e.scmProvider||"github",b=Tt(I,t,r),p=r?.githubToken||t.githubToken,m=I==="bitbucket"&&r?.scmToken||t.bitbucketToken,k=I==="gitlab"&&r?.scmToken||t.gitlabToken,N={NODE_OPTIONS:"--max-old-space-size=3072",TASK_ID:e.id,MANAGER_ACTION:e.managerAction,JIRA_ISSUE_KEY:e.jiraIssueKey||"",JIRA_SUMMARY:e.summary||"",JIRA_DESCRIPTION:e.description||"",GITHUB_REPO:e.githubRepo||"",PR_URL:e.githubPrUrl||"",PR_NUMBER:e.githubPrNumber?String(e.githubPrNumber):"",API_BASE_URL:l,ORG_API_KEY:t.apiKey,SCM_PROVIDER:I,SCM_TOKEN:b,GITHUB_TOKEN:p,GH_TOKEN:p,BITBUCKET_TOKEN:m,BITBUCKET_USERNAME:r?.bitbucketUsername||"x-token-auth",GITLAB_TOKEN:k,MANAGER_PROVIDER:r?.managerProvider||"anthropic",MANAGER_MODEL:r?.managerModelId||"",ANTHROPIC_API_KEY:s?"":r?.anthropicApiKey||process.env.ANTHROPIC_API_KEY||"",OPENAI_API_KEY:r?.openaiApiKey||"",GOOGLE_API_KEY:r?.googleApiKey||"",JIRA_BASE_URL:r?.jiraBaseUrl||"",JIRA_EMAIL:r?.jiraEmail||"",JIRA_API_TOKEN:r?.jiraApiToken||"",TICKET_SYSTEM:r?.issueTrackerProvider||"jira",LINEAR_API_KEY:r?.linearApiKey||""};for(let[h,D]of Object.entries(N))D!==""&&a.push("-e",`${h}=${D}`);let R=t.workerImage,K=Et(R);K&&It(K),a.push("--entrypoint","/bin/bash"),a.push(R),a.push("/app/manager-entrypoint.sh"),console.log(`${O()} ${o} ${u.magenta("\u25C6 MANAGER")} Starting ${e.managerAction} container ${u.yellow(i)}`);let P=ft("docker",a,{stdio:["ignore","pipe","pipe"],detached:!1});if(!P.pid){console.error(`${O()} ${o} ${u.red("\u2717")} Failed to spawn manager container`);return}let G={taskId:n,containerName:i,process:P,startedAt:new Date,status:"running",resultEmitted:!1};Y.set(n,G),P.stdout?.on("data",h=>{let D=h.toString().split(`
|
|
14
|
+
`).filter(w=>w.trim());for(let w of D)console.log(`${O()} ${o} ${u.magenta("MGR")} ${u.dim(Se(w))}`)}),P.stderr?.on("data",h=>{let D=h.toString().split(`
|
|
15
|
+
`).filter(w=>w.trim());for(let w of D)console.log(`${O()} ${o} ${u.magenta("MGR")} ${u.red(Se(w))}`)}),P.on("exit",h=>{G.status=h===0?"completed":"failed";let D=Math.round((Date.now()-G.startedAt.getTime())/1e3),w=h===0?u.green("\u2713"):u.red("\u2717"),v=h===0?u.green("completed"):u.red(`failed (exit ${h})`);console.log(`${O()} ${o} ${u.magenta("MGR")} ${w} Manager ${e.managerAction} ${v} ${u.dim(`(${D}s)`)}`),setTimeout(()=>Y.delete(n),6e4)}),P.on("error",h=>{G.status="failed",console.error(`${O()} ${o} ${u.magenta("MGR")} ${u.red("\u2717")} Container error: ${h.message}`)})}import{createRequire as vo}from"module";var Co=vo(import.meta.url),xo=Co("../package.json"),ce=xo.version;import{spawn as kt}from"child_process";import Ie from"chalk";async function Ce(){return console.log(Ie.cyan(" Updating @workermill/agent...")),new Promise(e=>{let t=kt("npm",["install","-g","@workermill/agent@latest"],{stdio:"inherit",shell:!0});t.on("error",r=>{console.error(Ie.red(` Update failed: ${r.message}`)),e(!1)}),t.on("close",r=>{r===0?(console.log(Ie.green(" Update successful.")),e(!0)):(console.error(Ie.red(` Update failed with exit code ${r}`)),e(!1))})})}function xe(){console.log(Ie.cyan(" Restarting agent...")),kt(process.argv[0],process.argv.slice(1),{stdio:"inherit",detached:!0}).unref(),process.exit(0)}var de=new Set,pe=new Set,Me=null,Pt=!1,Oe=!1;function L(){return y.dim(new Date().toLocaleTimeString())}async function Mo(){if(Me)return Me;try{return Me=(await M.get("/api/agent/config")).data,Me}catch{return console.error(`${L()} ${y.red("\u2717")} Failed to fetch org config`),{}}}async function vt(e){if(!Oe)try{let t=await M.get("/api/agent/poll",{params:{agentId:e.agentId}}),r=t.data.tasks;if(r.length===0)return;for(let n of r)n.status==="planning"&&!de.has(n.id)?await Oo(n,e):n.status==="queued"&&await No(n,e);let o=t.data.managerTasks;if(o&&o.length>0)for(let n of o)pe.has(n.id)||await Lo(n,e)}catch(t){let r=t,o=de.size>0||Ge()>0||pe.size>0;r.response?.status===401?o||console.error(`${L()} ${y.red("\u2717")} Authentication failed. Check your API key.`):o||console.warn(`${L()} ${y.yellow("\u26A0")} Poll error: ${r.message||String(t)}`)}}async function Oo(e,t){let r,o;try{let s=await M.post("/api/agent/claim",{taskId:e.id,agentId:t.agentId});if(!s.data.claimed)return;r=s.data.credentials,o=s.data.task}catch{return}let n=y.cyan(e.id.slice(0,8));if(o&&(o.retryCount??0)>0&&o.executionPlanV2!=null){console.log(),console.log(`${L()} ${y.magenta("\u25C6 RESUME")} ${n} ${e.summary.substring(0,60)}`),console.log(`${L()} ${n} Retry #${o.retryCount} with existing plan \u2014 skipping planning`),de.add(e.id);try{await M.post("/api/agent/resume-plan",{taskId:e.id,agentId:t.agentId}),console.log(`${L()} ${n} ${y.green("\u2713")} Resumed with existing plan \u2192 ${y.green("queued")}`)}catch(s){let a=s,l=a.response?.data?.error||a.message||String(s);console.error(`${L()} ${n} ${y.red("\u2717")} Resume failed: ${l}`)}de.delete(e.id);return}console.log(),console.log(`${L()} ${y.magenta("\u25C6 PLANNING")} ${n} ${e.summary.substring(0,60)}`),de.add(e.id),gt(e,t,r).then(s=>{console.log(s?`${L()} ${y.green("\u2713")} Planning complete for ${n}`:`${L()} ${y.red("\u2717")} Planning failed for ${n}`)}).catch(s=>console.error(`${L()} ${y.red("\u2717")} Planning error for ${n}:`,s.message||s)).finally(()=>de.delete(e.id))}async function No(e,t){if(Ge()>=t.maxWorkers)return;let o;try{if(o=(await M.post("/api/agent/claim",{taskId:e.id,agentId:t.agentId})).data,!o.claimed)return}catch{return}try{await M.post("/api/agent/started",{taskId:e.id,agentId:t.agentId})}catch{let I=y.cyan(e.id.slice(0,8));console.warn(`${L()} ${y.yellow("\u26A0")} Failed to report started for ${I}`)}let n=y.cyan(e.id.slice(0,8));console.log(),console.log(`${L()} ${y.blue("\u25B6 EXECUTING")} ${n} ${e.summary.substring(0,60)}`);let i=await Mo(),s=o.task||{id:e.id,summary:e.summary,description:e.description,jiraIssueKey:e.jiraIssueKey,workerModel:e.workerModel,workerProvider:e.workerProvider,githubRepo:e.githubRepo,scmProvider:e.scmProvider,skipManagerReview:e.skipManagerReview,deploymentEnabled:e.deploymentEnabled,improvementEnabled:e.improvementEnabled,qualityGateBypass:e.qualityGateBypass,standardSdkMode:e.standardSdkMode,parentTaskId:e.parentTaskId,taskNotes:e.taskNotes,githubPrUrl:e.githubPrUrl,githubPrNumber:e.githubPrNumber,executionPlanV2:e.executionPlanV2,jiraFields:e.jiraFields||{}},a=o.credentials;bt(s,t,i,a).catch(l=>console.error(`${L()} ${y.red("\u2717")} Spawn failed for ${n}:`,l.message||l))}async function Lo(e,t){let r=y.cyan(e.id.slice(0,8));pe.add(e.id);try{let o=await M.post("/api/agent/claim-manager",{taskId:e.id,agentId:t.agentId,action:e.managerAction});if(!o.data.claimed){pe.delete(e.id);return}let n=o.data.task,i=o.data.credentials||{},s=e.managerAction==="analyze_logs"?y.yellow("LOG ANALYSIS"):y.yellow("PR REVIEW");console.log(),console.log(`${L()} ${y.magenta("\u25C6 MANAGER")} ${s} ${r} ${e.summary.substring(0,60)}`);let a={id:e.id,summary:n?.summary||e.summary,description:n?.description||e.description,jiraIssueKey:n?.jiraIssueKey||e.jiraIssueKey,githubRepo:n?.githubRepo||e.githubRepo,scmProvider:n?.scmProvider||e.scmProvider,githubPrUrl:n?.githubPrUrl||e.githubPrUrl,githubPrNumber:n?.githubPrNumber||e.githubPrNumber,managerAction:e.managerAction};St(a,t,i).then(()=>{console.log(`${L()} ${r} ${y.magenta("MGR")} ${y.green("\u2713")} Manager ${e.managerAction} dispatched`)}).catch(l=>{console.error(`${L()} ${r} ${y.magenta("MGR")} ${y.red("\u2717")} Manager spawn failed:`,l.message||l)}).finally(()=>pe.delete(e.id))}catch(o){pe.delete(e.id);let n=o;console.error(`${L()} ${r} ${y.red("\u2717")} Failed to claim manager task:`,n.message||String(o))}}var Ne=null,Le=null;function Ct(){Ne&&(clearInterval(Ne),Ne=null),Le&&(clearInterval(Le),Le=null)}function xt(e){console.log(` ${y.dim("Polling every")} ${e.pollIntervalMs/1e3}s ${y.dim("\xB7 waiting for tasks...")}`),vt(e),Ne=setInterval(()=>vt(e),e.pollIntervalMs)}function Mt(e){Le=setInterval(async()=>{let t=At(),r=Array.from(de),o=Array.from(pe),n=[...t,...r,...o];try{let i=await M.post("/api/agent/heartbeat",{agentId:e.agentId,activeTasks:n,agentVersion:ce}),s=i.data?.cancelledTasks;if(s&&s.length>0)for(let b of s)_t(b);let{updateAvailable:a,updateRequired:l,latestVersion:I}=i.data??{};l&&!Oe?(Oe=!0,console.log(`${L()} ${y.red("\u26A0 Agent update required")} (current: ${ce}, required: ${I})`),console.log(`${L()} ${y.yellow("Refusing new tasks until updated.")}`),await Ce()?xe():(console.log(`${L()} ${y.red("Auto-update failed.")} Run: npm install -g @workermill/agent@latest`),Oe=!1)):a&&!Pt&&!l&&(Pt=!0,console.log(`${L()} ${y.yellow(`Update available: ${I}`)} (current: ${ce}). Run: workermill-agent update`))}catch{}},e.heartbeatIntervalMs)}be();async function Uo(e){console.log(),console.log(U.bold.cyan(" WorkerMill Remote Agent")),console.log(U.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(),Be(e.apiUrl,e.apiKey);try{let t=await M.get("/api/agent/config"),r=t.data.maxConcurrentWorkers;r&&typeof r=="number"&&(e.maxWorkers=r),t.data.workerImageUrl&&(e.workerImage=t.data.workerImageUrl),console.log(` ${U.green("\u25CF")} Connected to ${U.cyan(e.apiUrl)}`),console.log(` ${U.dim("Agent:")} ${e.agentId}`),console.log(` ${U.dim("Workers:")} ${U.yellow(String(e.maxWorkers))} parallel`),console.log(` ${U.dim("Image:")} ${e.workerImage}`),console.log(` ${U.dim("SCM:")} ${t.data.scmProvider}`),console.log(` ${U.dim("Model:")} ${U.yellow(t.data.defaultWorkerModel)}`),console.log()}catch(t){let r=t;throw r.response?.status===401?new Error("Authentication failed. Check your API key."):new Error(`Failed to connect to WorkerMill API: ${r.message||String(t)}`)}try{let t=await M.post("/api/agent/register",{agentId:e.agentId,maxWorkers:e.maxWorkers,agentVersion:ce}),{updateAvailable:r,updateRequired:o,latestVersion:n}=t.data;o?(console.log(U.red(` \u26A0 Agent update required (current: ${ce}, required: ${n})`)),await Ce()?xe():console.log(U.yellow(" Auto-update failed. Run: npm install -g @workermill/agent@latest"))):r&&console.log(U.yellow(` Update available: ${n} (current: ${ce}). Run: workermill-agent update`))}catch{}return xt(e),Mt(e),console.log(U.dim(" \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500")),console.log(` ${U.green("\u25CF")} Agent is running. ${U.dim("Press Ctrl+C to stop.")}`),console.log(),async()=>{console.log(),console.log(U.dim(" Shutting down...")),Ct();try{await M.post("/api/agent/deregister",{agentId:e.agentId})}catch{}await Rt(),console.log(` ${U.red("\u25CF")} Agent stopped.`)}}var Do=typeof process<"u"&&process.argv[1]&&(process.argv[1].endsWith("/index.ts")||process.argv[1].endsWith("/index.js"));if(Do){try{await import("dotenv/config")}catch{}let{loadConfig:e,validatePrerequisites:t}=await Promise.resolve().then(()=>(be(),He)),r=e();t(),console.log(U.dim(" Prerequisites validated."));let o=await Uo(r);process.on("SIGINT",async()=>{await o(),process.exit(0)}),process.on("SIGTERM",async()=>{await o(),process.exit(0)})}export{ue as findClaudePath,Je as getSystemInfo,Ye as loadConfig,Ve as loadConfigFromFile,Uo as startAgent,ze as validatePrerequisites};
|