@action-llama/action-llama 0.10.0 → 0.10.2

package/dist/cli/commands/doctor.js

@@ -1,16 +1,14 @@
 import { resolve } from "path";
 import { existsSync } from "fs";
-import { confirm } from "@inquirer/prompts";
-import { execFileSync } from "child_process";
-import { STSClient, GetCallerIdentityCommand } from "@aws-sdk/client-sts";
-import { IAMClient, CreateRoleCommand, PutRolePolicyCommand, PutUserPolicyCommand, GetRoleCommand } from "@aws-sdk/client-iam";
-import { ECRClient, SetRepositoryPolicyCommand } from "@aws-sdk/client-ecr";
 import { discoverAgents, loadAgentConfig, loadGlobalConfig } from "../../shared/config.js";
 import { resolveCredential } from "../../credentials/registry.js";
 import { promptCredential } from "../../credentials/prompter.js";
 import { parseCredentialRef, credentialExists, writeCredentialFields } from "../../shared/credentials.js";
 import { createLocalBackend, createBackendFromCloudConfig } from "../../shared/remote.js";
-import { AWS_CONSTANTS } from "../../shared/aws-constants.js";
+import { ConfigError, CredentialError } from "../../shared/errors.js";
+import { reconcileCloudIam, validateEcsRoles, reconcileLambdaRoles } from "./cloud-iam.js";
+// Re-export for backward compatibility (used by cloud-setup.ts and scheduler/index.ts)
+export { reconcileCloudIam, validateEcsRoles, reconcileLambdaRoles } from "./cloud-iam.js";
 // Webhook secret credential types — these support multiple named instances
 const WEBHOOK_SECRET_TYPES = {
     github: "github_webhook_secret",
@@ -20,7 +18,7 @@ export async function execute(opts) {
     const projectPath = resolve(opts.project);
     // Guard: refuse to run if the project path looks like an agent directory
     if (existsSync(resolve(projectPath, "agent-config.toml")) || existsSync(resolve(projectPath, "ACTIONS.md"))) {
-        throw new Error(`"${projectPath}" looks like an agent directory, not a project directory. ` +
+        throw new ConfigError(`"${projectPath}" looks like an agent directory, not a project directory. ` +
             `Run 'al doctor' from the project root (the parent directory).`);
     }
     const agents = discoverAgents(projectPath);
@@ -52,86 +50,16 @@ export async function execute(opts) {
         console.log("No credentials required by any agent.");
     }
     else if (opts.checkOnly) {
-        // --- Non-interactive credential check (headless mode) ---
-        let okCount = 0;
-        const missing = [];
-        if (opts.cloud) {
-            // Check cloud backend
-            const globalConfig = loadGlobalConfig(projectPath);
-            const cloudConfig = globalConfig.cloud;
-            if (!cloudConfig) {
-                throw new Error("No [cloud] section found in config.toml. " +
-                    "Run 'al cloud setup' to configure a cloud provider first.");
-            }
-            const remote = await createBackendFromCloudConfig(cloudConfig);
-            console.log(`Checking ${credentialRefs.size} credential(s) in ${cloudConfig.provider}...`);
-            for (const ref of credentialRefs) {
-                const { type, instance } = parseCredentialRef(ref);
-                const def = resolveCredential(type);
-                if (await remote.exists(type, instance)) {
-                    console.log(`  [ok] ${def.label} (${ref})`);
-                    okCount++;
-                }
-                else {
-                    console.log(`  [MISSING] ${def.label} (${ref})`);
-                    missing.push(ref);
-                }
-            }
-            if (missing.length > 0) {
-                throw new Error(`${missing.length} credential(s) missing from ${cloudConfig.provider}: ${missing.join(", ")}.\n` +
-                    `Push them with 'al doctor -c' first.`);
-            }
-            console.log(`${okCount} credential(s) verified in ${cloudConfig.provider}.`);
-        }
-        else {
-            // Check local filesystem (no prompts)
-            console.log(`Checking ${credentialRefs.size} credential(s)...`);
-            for (const ref of credentialRefs) {
-                const { type, instance } = parseCredentialRef(ref);
-                const def = resolveCredential(type);
-                if (await credentialExists(type, instance)) {
-                    console.log(`  [ok] ${def.label} (${ref})`);
-                    okCount++;
-                }
-                else {
-                    console.log(`  [MISSING] ${def.label} (${ref})`);
-                    missing.push(ref);
-                }
-            }
-            if (missing.length > 0) {
-                throw new Error(`${missing.length} credential(s) missing: ${missing.join(", ")}.\n` +
-                    `Run 'al doctor' interactively to configure them.`);
-            }
-            console.log(`${okCount} credential(s) verified.`);
-        }
+        await checkCredentials(credentialRefs, projectPath, opts.cloud);
     }
     else {
-        // --- Interactive credential check (used by `al doctor` and `al doctor -c`) ---
-        console.log(`\nChecking ${credentialRefs.size} credential(s)...\n`);
-        let okCount = 0;
-        let promptedCount = 0;
-        for (const ref of credentialRefs) {
-            const { type, instance } = parseCredentialRef(ref);
-            const def = resolveCredential(type);
-            if (await credentialExists(type, instance)) {
-                console.log(`  [ok] ${def.label} (${ref})`);
-                okCount++;
-                continue;
-            }
-            const result = await promptCredential(def, instance);
-            if (result && Object.keys(result.values).length > 0) {
-                await writeCredentialFields(type, instance, result.values);
-                promptedCount++;
-            }
-        }
-        console.log(`\nDone. ${okCount} already present, ${promptedCount} configured.`);
+        await promptCredentials(credentialRefs);
     }
     // --- Cloud mode: push creds + reconcile IAM (interactive only) ---
     if (opts.cloud && !opts.checkOnly) {
-        const globalConfig = loadGlobalConfig(projectPath);
         const cloudConfig = globalConfig.cloud;
         if (!cloudConfig) {
-            throw new Error("No [cloud] section found in config.toml. " +
+            throw new ConfigError("No [cloud] section found in config.toml. " +
                 "Run 'al cloud setup' to configure a cloud provider first.");
         }
         // Push local creds to cloud
@@ -166,580 +94,77 @@ export async function execute(opts) {
         }
     }
 }
-// --- Cloud IAM reconciliation ---
-export async function reconcileCloudIam(projectPath, cloud) {
-    if (cloud.provider === "cloud-run") {
-        await reconcileGcp(projectPath, cloud);
-    }
-    else if (cloud.provider === "ecs") {
-        await reconcileAws(projectPath, cloud);
-    }
-    else {
-        throw new Error(`Unknown cloud provider: "${cloud.provider}"`);
-    }
-}
-async function reconcileGcp(projectPath, cloud) {
-    const { gcpProject, secretPrefix: configPrefix } = cloud;
-    if (!gcpProject) {
-        throw new Error("cloud.gcpProject is required in config.toml");
-    }
-    const secretPrefix = configPrefix || AWS_CONSTANTS.DEFAULT_SECRET_PREFIX;
-    // Verify gcloud is available and authenticated
-    try {
-        gcloud(["auth", "print-access-token"], gcpProject);
-    }
-    catch (err) {
-        throw new Error("gcloud CLI is not authenticated. Run 'gcloud auth login' first.\n" +
-            `Original error: ${err.message}`);
-    }
-    const agents = discoverAgents(projectPath);
-    if (agents.length === 0) {
-        console.log("No agents found. Create agents first.");
-        return;
-    }
-    // Pre-flight: check if any secrets exist in GSM with this prefix
-    const preflight = listGsmSecretCount(gcpProject, secretPrefix);
-    if (preflight === 0) {
-        console.log(`\nWarning: No secrets found in GSM with prefix "${secretPrefix}".\n` +
-            `IAM bindings are created against existing secrets, so you should push credentials first.\n`);
-        const proceed = await confirm({
-            message: "Continue anyway? (Service accounts will be created but no secrets will be bound)",
-            default: false,
-        });
-        if (!proceed) {
-            console.log("Aborted. Push credentials first, then re-run.");
-            return;
-        }
-    }
-    console.log(`\nSetting up Cloud Run service accounts for ${agents.length} agent(s)...\n`);
-    for (const name of agents) {
-        const config = loadAgentConfig(projectPath, name);
-        const saName = AWS_CONSTANTS.serviceAccountName(name);
-        const saEmail = AWS_CONSTANTS.serviceAccountEmail(name, gcpProject);
-        console.log(`  Agent: ${name}`);
-        console.log(`    SA: ${saEmail}`);
-        // 1. Create service account (idempotent)
-        try {
-            gcloud([
-                "iam", "service-accounts", "create", saName,
-                "--display-name", `Action Llama agent: ${name}`,
-                "--project", gcpProject,
-            ], gcpProject);
-            console.log(`    Created service account`);
-        }
-        catch (err) {
-            if (err.message?.includes("already exists")) {
-                console.log(`    Service account already exists`);
-            }
-            else {
-                throw err;
-            }
-        }
-        // 2. Collect all secret names this agent needs
-        const credRefs = [...new Set(config.credentials)];
-        if (config.model.authType !== "pi_auth" && !credRefs.includes("anthropic_key:default")) {
-            credRefs.push("anthropic_key:default");
+// --- Credential check (headless / non-interactive) ---
+async function checkCredentials(credentialRefs, projectPath, cloud) {
+    let okCount = 0;
+    const missing = [];
+    if (cloud) {
+        const globalConfig = loadGlobalConfig(projectPath);
+        const cloudConfig = globalConfig.cloud;
+        if (!cloudConfig) {
+            throw new ConfigError("No [cloud] section found in config.toml. " +
+                "Run 'al cloud setup' to configure a cloud provider first.");
         }
-        const secretNames = [];
-        for (const ref of credRefs) {
+        const remote = await createBackendFromCloudConfig(cloudConfig);
+        console.log(`Checking ${credentialRefs.size} credential(s) in ${cloudConfig.provider}...`);
+        for (const ref of credentialRefs) {
             const { type, instance } = parseCredentialRef(ref);
-            const fields = listGsmFields(gcpProject, secretPrefix, type, instance);
-            for (const field of fields) {
-                secretNames.push(`${secretPrefix}--${type}--${instance}--${field}`);
-            }
-        }
-        // 3. Grant secretmanager.secretAccessor on each secret
-        let boundCount = 0;
-        for (const secretName of secretNames) {
-            try {
-                gcloud([
-                    "secrets", "add-iam-policy-binding", secretName,
-                    "--member", `serviceAccount:${saEmail}`,
-                    "--role", "roles/secretmanager.secretAccessor",
-                    "--project", gcpProject,
-                ], gcpProject);
-                boundCount++;
-            }
-            catch (err) {
-                if (err.message?.includes("already exists") || err.message?.includes("already has")) {
-                    boundCount++;
-                }
-                else {
-                    console.log(`    Warning: failed to bind ${secretName}: ${err.message}`);
-                }
-            }
-        }
-        console.log(`    Bound ${boundCount} secret(s)`);
-        // 4. Grant the SA permission to act as itself (for Cloud Run job execution)
-        try {
-            gcloud([
-                "iam", "service-accounts", "add-iam-policy-binding", saEmail,
-                "--member", `serviceAccount:${saEmail}`,
-                "--role", "roles/iam.serviceAccountUser",
-                "--project", gcpProject,
-            ], gcpProject);
-        }
-        catch {
-            // May already be bound
-        }
-        console.log("");
-    }
-    console.log("Done. Each agent now has an isolated service account with access to only its declared secrets.");
-}
-async function reconcileAws(projectPath, cloud) {
-    const { awsRegion, ecrRepository, awsSecretPrefix } = cloud;
-    if (!awsRegion) {
-        throw new Error("cloud.awsRegion is required in config.toml");
-    }
-    if (!ecrRepository) {
-        throw new Error("cloud.ecrRepository is required in config.toml");
-    }
-    const secretPrefix = awsSecretPrefix || AWS_CONSTANTS.DEFAULT_SECRET_PREFIX;
-    // Extract account ID from ECR repo URI
-    const accountMatch = ecrRepository.match(/^(\d+)\.dkr\.ecr\./);
-    if (!accountMatch) {
-        throw new Error(`Cannot extract AWS account ID from cloud.ecrRepository: "${ecrRepository}". ` +
-            `Expected format: 123456789012.dkr.ecr.<region>.amazonaws.com/<repo>`);
-    }
-    const accountId = accountMatch[1];
-    // Verify AWS credentials are valid
-    const stsClient = new STSClient({ region: awsRegion });
-    try {
-        await stsClient.send(new GetCallerIdentityCommand({}));
-    }
-    catch (err) {
-        throw new Error("AWS CLI is not authenticated. Run 'aws configure' or set AWS_ACCESS_KEY_ID/AWS_SECRET_ACCESS_KEY.\n" +
-            `Original error: ${err.message}`);
-    }
-    const agents = discoverAgents(projectPath);
-    if (agents.length === 0) {
-        console.log("No agents found. Create agents first.");
-        return;
-    }
-    // Trust policy for ECS tasks
-    const trustPolicy = JSON.stringify({
-        Version: "2012-10-17",
-        Statement: [{
-                Effect: "Allow",
-                Principal: { Service: "ecs-tasks.amazonaws.com" },
-                Action: "sts:AssumeRole",
-            }],
-    });
-    const iamClient = new IAMClient({ region: awsRegion });
-    // Ensure execution role has Secrets Manager access (ECS uses this role to inject secrets)
-    if (cloud.executionRoleArn) {
-        const executionRoleName = cloud.executionRoleArn.split("/").pop();
-        try {
-            await iamClient.send(new PutRolePolicyCommand({
-                RoleName: executionRoleName,
-                PolicyName: "ActionLlamaExecution",
-                PolicyDocument: JSON.stringify({
-                    Version: "2012-10-17",
-                    Statement: [
-                        {
-                            Effect: "Allow",
-                            Action: "secretsmanager:GetSecretValue",
-                            Resource: `arn:aws:secretsmanager:${awsRegion}:${accountId}:secret:${secretPrefix}/*`,
-                        },
-                        {
-                            Effect: "Allow",
-                            Action: "logs:CreateLogGroup",
-                            Resource: `arn:aws:logs:${awsRegion}:${accountId}:log-group:${AWS_CONSTANTS.LOG_GROUP}*`,
-                        },
-                    ],
-                }),
-            }));
-            console.log(`Execution role (${executionRoleName}): Secrets Manager + CloudWatch policy applied`);
-        }
-        catch (err) {
-            throw new Error(`Failed to attach ActionLlamaExecution policy to ${executionRoleName}: ${err.message}\n` +
-                `The execution role needs secretsmanager:GetSecretValue and logs:CreateLogGroup permissions.\n` +
-                `Either grant your IAM user iam:PutRolePolicy on this role, or attach the policy manually in the AWS Console.`);
-        }
-    }
-    // Ensure ECR repository policy grants Lambda pull access
-    await ensureLambdaEcrPolicy(awsRegion, ecrRepository);
-    console.log(`\nSetting up ECS task roles for ${agents.length} agent(s)...\n`);
-    for (const name of agents) {
-        const config = loadAgentConfig(projectPath, name);
-        const roleName = AWS_CONSTANTS.taskRoleName(name);
-        console.log(`  Agent: ${name}`);
-        console.log(`    Role: ${roleName}`);
-        // 1. Create IAM role (idempotent)
-        try {
-            await iamClient.send(new CreateRoleCommand({
-                RoleName: roleName,
-                AssumeRolePolicyDocument: trustPolicy,
-            }));
-            console.log(`    Created IAM role`);
-        }
-        catch (err) {
-            if (err.name === "EntityAlreadyExistsException") {
-                console.log(`    IAM role already exists`);
+            const def = resolveCredential(type);
+            if (await remote.exists(type, instance)) {
+                console.log(`  [ok] ${def.label} (${ref})`);
+                okCount++;
             }
             else {
-                throw err;
+                console.log(`  [MISSING] ${def.label} (${ref})`);
+                missing.push(ref);
             }
         }
-        // 2. Collect secret ARNs this agent needs
-        const credRefs = [...new Set(config.credentials)];
-        if (config.model.authType !== "pi_auth" && !credRefs.includes("anthropic_key:default")) {
-            credRefs.push("anthropic_key:default");
-        }
-        const secretArns = [];
-        for (const ref of credRefs) {
-            const { type, instance } = parseCredentialRef(ref);
-            secretArns.push(`arn:aws:secretsmanager:${awsRegion}:${accountId}:secret:${secretPrefix}/${type}/${instance}/*`);
-        }
-        // 3. Put inline policy for Secrets Manager access
-        if (secretArns.length > 0) {
-            const policy = JSON.stringify({
-                Version: "2012-10-17",
-                Statement: [{
-                        Effect: "Allow",
-                        Action: "secretsmanager:GetSecretValue",
-                        Resource: secretArns,
-                    }],
-            });
-            try {
-                await iamClient.send(new PutRolePolicyCommand({
-                    RoleName: roleName,
-                    PolicyName: "SecretsAccess",
-                    PolicyDocument: policy,
-                }));
-                console.log(`    Bound ${secretArns.length} secret path(s)`);
-            }
-            catch (err) {
-                console.log(`    Warning: failed to put policy: ${err.message}`);
-            }
-        }
-        else {
-            console.log(`    No secrets to bind`);
-        }
-        console.log("");
-    }
-    // Grant iam:PassRole on ECS task roles + execution role to the calling identity
-    const taskRoleArns = agents.map((name) => `arn:aws:iam::${accountId}:role/${AWS_CONSTANTS.taskRoleName(name)}`);
-    if (cloud.executionRoleArn) {
-        taskRoleArns.push(cloud.executionRoleArn);
-    }
-    if (taskRoleArns.length > 0) {
-        await grantPassRole(awsRegion, iamClient, taskRoleArns, "ActionLlamaEcsPassRole");
-    }
-    console.log("Done. Each agent now has an isolated IAM task role with access to only its declared secrets.");
-    console.log(`\nTask roles follow the convention: al-{agentName}-task-role`);
-    console.log("The ECS runtime will use them automatically at launch time.");
-}
-// --- Helpers ---
-/**
- * Grant iam:PassRole on the given role ARNs to the calling IAM user.
- * Required so the CLI can assign roles to ECS tasks and Lambda functions.
- */
-async function grantPassRole(awsRegion, iamClient, roleArns, policyName) {
-    const stsClient = new STSClient({ region: awsRegion });
-    const identity = await stsClient.send(new GetCallerIdentityCommand({}));
-    const callerArn = identity.Arn;
-    // Extract user name from ARN (arn:aws:iam::ACCOUNT:user/USERNAME)
-    const userMatch = callerArn.match(/:user\/(.+)$/);
-    if (!userMatch) {
-        console.log(`\n  Note: Caller ${callerArn} is not an IAM user — skipping iam:PassRole auto-grant.`);
-        console.log(`  If you get PassRole errors, add this policy to your IAM identity:`);
-        console.log(JSON.stringify({
-            Version: "2012-10-17",
-            Statement: [{
-                    Effect: "Allow",
-                    Action: "iam:PassRole",
-                    Resource: roleArns,
-                }],
-        }, null, 2));
-        return;
-    }
-    const userName = userMatch[1];
-    const policy = JSON.stringify({
-        Version: "2012-10-17",
-        Statement: [{
-                Effect: "Allow",
-                Action: "iam:PassRole",
-                Resource: roleArns,
-            }],
-    });
-    try {
-        await iamClient.send(new PutUserPolicyCommand({
-            UserName: userName,
-            PolicyName: policyName,
-            PolicyDocument: policy,
-        }));
-        console.log(`  Granted iam:PassRole on ${roleArns.length} role(s) to user ${userName}`);
-    }
-    catch (err) {
-        console.log(`  Warning: could not grant iam:PassRole to user ${userName}: ${err.message}`);
-        console.log(`  You may need to manually add iam:PassRole permission for these roles:`);
-        for (const arn of roleArns) {
-            console.log(`    - ${arn}`);
-        }
-    }
-}
-function gcloud(args, _project) {
-    return execFileSync("gcloud", args, {
-        encoding: "utf-8",
-        timeout: 30_000,
-        stdio: ["pipe", "pipe", "pipe"],
-    }).trim();
-}
-function listGsmSecretCount(gcpProject, prefix) {
-    try {
-        const output = gcloud([
-            "secrets", "list",
-            "--filter", `name:${prefix}--`,
-            "--format", "value(name)",
-            "--project", gcpProject,
-        ], gcpProject);
-        if (!output.trim())
-            return 0;
-        return output.trim().split("\n").length;
-    }
-    catch {
-        return 0;
-    }
-}
-function listGsmFields(gcpProject, prefix, type, instance) {
-    const filter = `name:${prefix}--${type}--${instance}--`;
-    try {
-        const output = gcloud([
-            "secrets", "list",
-            "--filter", filter,
-            "--format", "value(name)",
-            "--project", gcpProject,
-        ], gcpProject);
-        if (!output.trim())
-            return [];
-        const fields = [];
-        for (const line of output.split("\n")) {
-            const secretId = line.trim().split("/").pop();
-            const parts = secretId.split("--");
-            if (parts.length === 4 && parts[0] === prefix && parts[1] === type && parts[2] === instance) {
-                fields.push(parts[3]);
-            }
+        if (missing.length > 0) {
+            throw new CredentialError(`${missing.length} credential(s) missing from ${cloudConfig.provider}: ${missing.join(", ")}.\n` +
+                `Push them with 'al doctor -c' first.`);
         }
-        return fields;
-    }
-    catch {
-        return [];
+        console.log(`${okCount} credential(s) verified in ${cloudConfig.provider}.`);
     }
-}
-export async function validateEcsRoles(projectPath, cloud) {
-    const { awsRegion } = cloud;
-    if (!awsRegion) {
-        throw new Error("cloud.awsRegion is required for ECS validation");
-    }
-    const agents = discoverAgents(projectPath);
-    if (agents.length === 0)
-        return;
-    const iamClient = new IAMClient({ region: awsRegion });
-    const missing = [];
-    const hasIncorrectTrust = [];
-    for (const name of agents) {
-        const roleName = AWS_CONSTANTS.taskRoleName(name);
-        try {
-            const role = await iamClient.send(new GetRoleCommand({ RoleName: roleName }));
-            // Check if the role has the correct trust policy for ECS tasks
-            const trustPolicy = JSON.parse(decodeURIComponent(role.Role.AssumeRolePolicyDocument));
-            const hasEcsTrust = trustPolicy.Statement?.some((stmt) => stmt.Effect === "Allow" &&
-                stmt.Principal?.Service === "ecs-tasks.amazonaws.com" &&
-                (stmt.Action === "sts:AssumeRole" || stmt.Action?.includes("sts:AssumeRole")));
-            if (!hasEcsTrust) {
-                hasIncorrectTrust.push(roleName);
-                console.log(`  [TRUST ISSUE] ${roleName} - missing ECS task trust policy`);
-            }
-            else {
-                console.log(`  [ok] ${roleName}`);
-            }
-        }
-        catch (err) {
-            if (err.name === "NoSuchEntityException") {
-                missing.push(roleName);
-                console.log(`  [MISSING] ${roleName}`);
+    else {
+        console.log(`Checking ${credentialRefs.size} credential(s)...`);
+        for (const ref of credentialRefs) {
+            const { type, instance } = parseCredentialRef(ref);
+            const def = resolveCredential(type);
+            if (await credentialExists(type, instance)) {
+                console.log(`  [ok] ${def.label} (${ref})`);
+                okCount++;
             }
             else {
-                console.log(`  [ERROR] ${roleName}: ${err.message}`);
+                console.log(`  [MISSING] ${def.label} (${ref})`);
+                missing.push(ref);
             }
         }
-    }
-    if (missing.length > 0 || hasIncorrectTrust.length > 0) {
-        console.log(`\n⚠️  Found IAM role issues that will cause ECS task failures:`);
         if (missing.length > 0) {
-            console.log(`\n${missing.length} IAM task role(s) are missing:`);
-            missing.forEach(role => console.log(`  - ${role}`));
-            console.log(`\n🔧 Fix: Run 'al doctor -c' to create missing roles automatically.`);
-        }
-        if (hasIncorrectTrust.length > 0) {
-            console.log(`\n${hasIncorrectTrust.length} IAM role(s) have incorrect trust policies:`);
-            hasIncorrectTrust.forEach(role => console.log(`  - ${role}`));
-            console.log(`\n🔧 Fix: Update trust policy to allow ECS tasks to assume the role:`);
-            console.log(`For each role above, run:`);
-            console.log(`  aws iam update-assume-role-policy --role-name ROLE_NAME --policy-document file://ecs-trust.json`);
+            throw new CredentialError(`${missing.length} credential(s) missing: ${missing.join(", ")}.\n` +
+                `Run 'al doctor' interactively to configure them.`);
         }
-        console.log(`\nECS task trust policy (save as ecs-trust.json):`);
-        console.log(JSON.stringify({
-            Version: "2012-10-17",
-            Statement: [{
-                    Effect: "Allow",
-                    Principal: { Service: "ecs-tasks.amazonaws.com" },
-                    Action: "sts:AssumeRole",
-                }],
-        }, null, 2));
-        console.log(`\n💡 Alternatively, re-run the cloud setup to fix all issues:`);
-        console.log(`  al cloud setup`);
-        // Throw error to prevent proceeding with invalid configuration
-        throw new Error(`${missing.length + hasIncorrectTrust.length} IAM task role(s) have issues that will prevent ECS tasks from starting. Fix the roles above before proceeding.`);
-    }
-    else {
-        console.log(`All ${agents.length} IAM task role(s) exist and have correct trust policies.`);
+        console.log(`${okCount} credential(s) verified.`);
     }
 }
-async function ensureLambdaEcrPolicy(awsRegion, ecrRepoUri) {
-    const repoName = ecrRepoUri.split("/").pop();
-    if (!repoName)
-        return;
-    const ecrClient = new ECRClient({ region: awsRegion });
-    const policy = JSON.stringify({
-        Version: "2012-10-17",
-        Statement: [{
-                Sid: "LambdaECRImageRetrievalPolicy",
-                Effect: "Allow",
-                Principal: { Service: "lambda.amazonaws.com" },
-                Action: [
-                    "ecr:BatchGetImage",
-                    "ecr:GetDownloadUrlForLayer",
-                ],
-            }],
-    });
-    try {
-        await ecrClient.send(new SetRepositoryPolicyCommand({
-            repositoryName: repoName,
-            policyText: policy,
-        }));
-        console.log(`ECR repository policy: granted Lambda pull access`);
-    }
-    catch (err) {
-        console.log(`Warning: could not set ECR repository policy for Lambda: ${err.message}`);
-    }
-}
-async function reconcileLambdaRoles(projectPath, cloud) {
-    const { awsRegion, ecrRepository, awsSecretPrefix } = cloud;
-    if (!awsRegion || !ecrRepository)
-        return;
-    const secretPrefix = awsSecretPrefix || AWS_CONSTANTS.DEFAULT_SECRET_PREFIX;
-    const accountMatch = ecrRepository.match(/^(\d+)\.dkr\.ecr\./);
-    if (!accountMatch)
-        return;
-    const accountId = accountMatch[1];
-    const globalConfig = loadGlobalConfig(projectPath);
-    const agents = discoverAgents(projectPath);
-    const iamClient = new IAMClient({ region: awsRegion });
-    // Trust policy for Lambda
-    const trustPolicy = JSON.stringify({
-        Version: "2012-10-17",
-        Statement: [{
-                Effect: "Allow",
-                Principal: { Service: "lambda.amazonaws.com" },
-                Action: "sts:AssumeRole",
-            }],
-    });
-    let created = 0;
-    for (const name of agents) {
-        const config = loadAgentConfig(projectPath, name);
-        const effectiveTimeout = config.timeout ?? globalConfig.local?.timeout ?? 900;
-        // Only create Lambda roles for agents that will route to Lambda
-        if (effectiveTimeout > AWS_CONSTANTS.LAMBDA_MAX_TIMEOUT)
+// --- Interactive credential prompting ---
+async function promptCredentials(credentialRefs) {
+    console.log(`\nChecking ${credentialRefs.size} credential(s)...\n`);
+    let okCount = 0;
+    let promptedCount = 0;
+    for (const ref of credentialRefs) {
+        const { type, instance } = parseCredentialRef(ref);
+        const def = resolveCredential(type);
+        if (await credentialExists(type, instance)) {
+            console.log(`  [ok] ${def.label} (${ref})`);
+            okCount++;
             continue;
-        const roleName = AWS_CONSTANTS.lambdaRoleName(name);
-        // Create role
-        try {
-            await iamClient.send(new CreateRoleCommand({
-                RoleName: roleName,
-                AssumeRolePolicyDocument: trustPolicy,
-            }));
-            console.log(`  Created Lambda role: ${roleName}`);
-            created++;
-        }
-        catch (err) {
-            if (err.name === "EntityAlreadyExistsException") {
-                console.log(`  [ok] ${roleName}`);
-            }
-            else {
-                console.log(`  [ERROR] ${roleName}: ${err.message}`);
-                continue;
-            }
-        }
-        // Add secrets + ECR + logs policy
-        const credRefs = [...new Set(config.credentials)];
-        if (config.model.authType !== "pi_auth" && !credRefs.includes("anthropic_key:default")) {
-            credRefs.push("anthropic_key:default");
         }
-        const secretArns = credRefs.map((ref) => {
-            const { type, instance } = parseCredentialRef(ref);
-            return `arn:aws:secretsmanager:${awsRegion}:${accountId}:secret:${secretPrefix}/${type}/${instance}/*`;
-        });
-        const policy = JSON.stringify({
-            Version: "2012-10-17",
-            Statement: [
-                {
-                    Effect: "Allow",
-                    Action: "secretsmanager:GetSecretValue",
-                    Resource: secretArns,
-                },
-                {
-                    Effect: "Allow",
-                    Action: [
-                        "logs:CreateLogGroup",
-                        "logs:CreateLogStream",
-                        "logs:PutLogEvents",
-                    ],
-                    Resource: `arn:aws:logs:${awsRegion}:${accountId}:*`,
-                },
-                {
-                    Effect: "Allow",
-                    Action: "ecr:GetAuthorizationToken",
-                    Resource: "*",
-                },
-                {
-                    Effect: "Allow",
-                    Action: [
-                        "ecr:BatchGetImage",
-                        "ecr:GetDownloadUrlForLayer",
-                    ],
-                    Resource: `arn:aws:ecr:${awsRegion}:${accountId}:repository/*`,
-                },
-            ],
-        });
-        try {
-            await iamClient.send(new PutRolePolicyCommand({
-                RoleName: roleName,
-                PolicyName: "LambdaExecution",
-                PolicyDocument: policy,
-            }));
+        const result = await promptCredential(def, instance);
+        if (result && Object.keys(result.values).length > 0) {
+            await writeCredentialFields(type, instance, result.values);
+            promptedCount++;
         }
-        catch (err) {
-            console.log(`  Warning: failed to put policy on ${roleName}: ${err.message}`);
-        }
-    }
-    if (created > 0) {
-        console.log(`Created ${created} Lambda execution role(s).`);
-    }
-    else {
-        console.log(`All Lambda roles up to date.`);
-    }
-    // Grant iam:PassRole on Lambda roles to the calling identity
-    const lambdaRoleArns = agents
-        .filter((name) => {
-        const config = loadAgentConfig(projectPath, name);
-        const effectiveTimeout = config.timeout ?? globalConfig.local?.timeout ?? 900;
-        return effectiveTimeout <= AWS_CONSTANTS.LAMBDA_MAX_TIMEOUT;
-    })
-        .map((name) => `arn:aws:iam::${accountId}:role/${AWS_CONSTANTS.lambdaRoleName(name)}`);
-    if (lambdaRoleArns.length > 0) {
-        await grantPassRole(awsRegion, iamClient, lambdaRoleArns, "ActionLlamaLambdaPassRole");
     }
+    console.log(`\nDone. ${okCount} already present, ${promptedCount} configured.`);
 }
 //# sourceMappingURL=doctor.js.map