forge-pipeline 0.1.0

package/lib/prompts.sh

@@ -0,0 +1,554 @@
+#!/usr/bin/env bash
+# ─────────────────────────────────────────────────────────────
+# Forge – prompt generation and template substitution
+# Handles prompt assembly for all agent roles.
+# ─────────────────────────────────────────────────────────────
+
+[[ -n "${_PROMPTS_SH_LOADED:-}" ]] && return
+_PROMPTS_SH_LOADED=1
+
+source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/utils.sh"
+
+FORGE_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
+
+# ── Anti-attribution rules (appended to every agent prompt) ──
+
+ANTI_ATTRIBUTION_RULES='
+---
+
+## STRICT RULES — ATTRIBUTION AND COMMIT MESSAGES
+
+These rules are NON-NEGOTIABLE and override any other instructions:
+
+1. **NEVER mention AI, Claude, Anthropic, LLM, GPT, or any AI assistant** in:
+   - Commit messages
+   - Code comments
+   - Documentation or generated files
+   - Variable names, function names, or string literals
+   - PR titles, PR descriptions, or branch names
+
+2. **Commit messages must look human-written** using conventional commits:
+   - `feat: add user authentication endpoints`
+   - `fix: resolve race condition in cart service`
+   - `refactor: extract validation logic into shared utils`
+   - `test: add integration tests for payment flow`
+   - `docs: update API documentation`
+
+3. **Do NOT add any attribution headers** such as:
+   - "Co-Authored-By" with any AI name
+   - "Generated by" anything
+   - "Created with" any tool name
+   - Any similar attribution to automated tools
+
+4. **Do NOT reference this pipeline** in any output code, comments, or commits. No mentions of "forge pipeline", "autonomous agents", or "automated coding".
+
+5. **Code comments should be natural** — explain the "why", not reference how the code was produced.
+'
+
+# ── Augment tools integration ────────────────────────────────
+
+get_augment_tools_flag() {
+  local config_file="${FORGE_ROOT}/.forge/config.json"
+  if [[ ! -f "$config_file" ]]; then
+    echo ""
+    return
+  fi
+
+  local augment_available
+  augment_available=$(jq -r '(.augment_available // .AUGMENT_AVAILABLE) // empty' "$config_file" 2>/dev/null)
+
+  if [[ "$augment_available" == "true" ]]; then
+    local tool_name
+    tool_name=$(jq -r '(.augment_tool_name // .AUGMENT_TOOL_NAME) // empty' "$config_file" 2>/dev/null)
+    if [[ -n "$tool_name" ]]; then
+      echo "$tool_name"
+      return
+    fi
+  fi
+
+  echo ""
+}
+
+# ── Build allowed tools per role ─────────────────────────────
+
+build_allowed_tools() {
+  local role="$1"
+  local augment_tool
+  augment_tool=$(get_augment_tools_flag)
+
+  local tools=""
+
+  case "$role" in
+    worker)
+      tools="Read,Write,Bash"
+      ;;
+    lead)
+      tools="Read,Write,Bash"
+      ;;
+    auditor_senior)
+      tools="Read,Bash(find),Bash(cat),Bash(grep),Bash(wc),Bash(ls)"
+      ;;
+    auditor_junior)
+      tools="Read,Bash(find),Bash(cat),Bash(grep),Bash(ls)"
+      ;;
+    architect)
+      tools="Read,Write,Bash(find),Bash(cat),Bash(ls),Bash(head)"
+      ;;
+    challenger)
+      tools="Read,Write,Bash(find),Bash(cat),Bash(ls)"
+      ;;
+    coordinator)
+      tools="Read,Write,Bash(find),Bash(cat),Bash(ls)"
+      ;;
+    integration)
+      tools="Read,Write,Bash"
+      ;;
+    fix)
+      tools="Read,Write,Bash"
+      ;;
+    *)
+      log_warn "Unknown role '$role' – defaulting to Read,Bash"
+      tools="Read,Bash"
+      ;;
+  esac
+
+  if [[ -n "$augment_tool" ]]; then
+    tools="${tools},${augment_tool}"
+  fi
+
+  echo "$tools"
+}
+
+# ── Architect prompt ─────────────────────────────────────────
+
+generate_architect_prompt() {
+  local prompt=""
+
+  # Base template
+  prompt+=$(cat "${FORGE_DIR}/prompts/architect_spec.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # User requirement
+  prompt+="## USER REQUIREMENT"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/requirement.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Existing repo structure
+  prompt+="## EXISTING REPO STRUCTURE"$'\n\n'
+  prompt+='```'$'\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/repo-tree.txt" 2>/dev/null)
+  prompt+=$'\n```\n\n'
+
+  # Existing key files
+  prompt+="## EXISTING KEY FILES"$'\n\n'
+  prompt+='```'$'\n'
+  local key_files
+  key_files=$(cd "${FORGE_ROOT}" && cat README.md package.json pyproject.toml Cargo.toml go.mod 2>/dev/null || echo "New project")
+  prompt+="$key_files"
+  prompt+=$'\n```\n\n'
+
+  # Instruction
+  prompt+="## INSTRUCTION"$'\n\n'
+  prompt+="Write your spec to ${FORGE_ROOT}/.forge/spec/spec.md"$'\n'
+
+  prompt+="$ANTI_ATTRIBUTION_RULES"
+
+  echo "$prompt"
+}
+
+# ── Architect revision prompt ────────────────────────────────
+
+generate_architect_revision_prompt() {
+  local prompt=""
+
+  # Base template
+  prompt+=$(cat "${FORGE_DIR}/prompts/architect_spec.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  prompt+="**NOTE: This is a REVISION pass. Review the challenger's feedback and update your spec accordingly.**"$'\n\n'
+
+  # Current spec
+  prompt+="## YOUR CURRENT SPEC"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/spec.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Challenger's review
+  prompt+="## CHALLENGER'S REVIEW"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/review.json" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # User requirement
+  prompt+="## USER REQUIREMENT"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/requirement.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Existing repo structure
+  prompt+="## EXISTING REPO STRUCTURE"$'\n\n'
+  prompt+='```'$'\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/repo-tree.txt" 2>/dev/null)
+  prompt+=$'\n```\n\n'
+
+  # Instruction
+  prompt+="## INSTRUCTION"$'\n\n'
+  prompt+="Update your spec at ${FORGE_ROOT}/.forge/spec/spec.md addressing the challenger's feedback."$'\n'
+
+  prompt+="$ANTI_ATTRIBUTION_RULES"
+
+  echo "$prompt"
+}
+
+# ── Challenger prompt ────────────────────────────────────────
+
+generate_challenger_prompt() {
+  local prompt=""
+
+  # Base template
+  prompt+=$(cat "${FORGE_DIR}/prompts/challenger_review.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Original user requirement
+  prompt+="## ORIGINAL USER REQUIREMENT"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/requirement.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Architect's spec
+  prompt+="## ARCHITECT'S SPEC"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/spec.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Existing repo structure
+  prompt+="## EXISTING REPO STRUCTURE"$'\n\n'
+  prompt+='```'$'\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/repo-tree.txt" 2>/dev/null)
+  prompt+=$'\n```\n\n'
+
+  # Instruction
+  prompt+="## INSTRUCTION"$'\n\n'
+  prompt+="Write your review to ${FORGE_ROOT}/.forge/spec/review.json"$'\n'
+
+  prompt+="$ANTI_ATTRIBUTION_RULES"
+
+  echo "$prompt"
+}
+
+# ── Coordinator prompt ───────────────────────────────────────
+
+generate_coordinator_prompt() {
+  local prompt=""
+
+  # Base template
+  prompt+=$(cat "${FORGE_DIR}/prompts/coordinator_plan.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Approved technical spec
+  prompt+="## APPROVED TECHNICAL SPEC"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/spec.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Original user requirement
+  prompt+="## ORIGINAL USER REQUIREMENT"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/requirement.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Repo structure
+  prompt+="## REPO STRUCTURE"$'\n\n'
+  prompt+='```'$'\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/repo-tree.txt" 2>/dev/null)
+  prompt+=$'\n```\n\n'
+
+  # Existing key files
+  prompt+="## EXISTING KEY FILES"$'\n\n'
+  prompt+='```'$'\n'
+  local key_files
+  key_files=$(cd "${FORGE_ROOT}" && cat README.md package.json pyproject.toml Cargo.toml go.mod 2>/dev/null || echo "New project")
+  prompt+="$key_files"
+  prompt+=$'\n```\n\n'
+
+  # JSON schema for plan.json
+  prompt+="## PLAN JSON SCHEMA"$'\n\n'
+  prompt+='The plan.json file MUST follow this exact schema:'$'\n\n'
+  prompt+='```json'$'\n'
+  prompt+='{
+  "project_type": "greenfield | incremental",
+  "leads": [
+    {
+      "name": "lead-<module-name>",
+      "module": "<module-name>",
+      "description": "<what this lead is responsible for>",
+      "dependencies": ["<other lead names this depends on, if any>"],
+      "workers": [
+        {
+          "name": "worker-<task-name>",
+          "task": "<concise task description>",
+          "files": ["<list of files this worker will create or modify>"],
+          "spec_excerpt": "<relevant portion of the spec for this task>"
+        }
+      ]
+    }
+  ]
+}'
+  prompt+=$'\n```\n\n'
+
+  # Instruction
+  prompt+="## INSTRUCTION"$'\n\n'
+  prompt+="Write your implementation plan to ${FORGE_ROOT}/.forge/plan.json"$'\n'
+
+  prompt+="$ANTI_ATTRIBUTION_RULES"
+
+  echo "$prompt"
+}
+
+# ── Lead prompt ──────────────────────────────────────────────
+
+generate_lead_prompt() {
+  local lead_name="$1"
+  local lead_json="$2"
+
+  local module_name
+  module_name=$(echo "$lead_json" | jq -r '.module // .name' 2>/dev/null)
+
+  local workers_json
+  workers_json=$(echo "$lead_json" | jq -c '.workers // []' 2>/dev/null)
+
+  local prompt=""
+
+  # Base template
+  local template
+  template=$(cat "${FORGE_DIR}/prompts/lead.md" 2>/dev/null)
+
+  if [[ -n "$template" ]]; then
+    # Perform substitutions
+    template="${template//\{MODULE_NAME\}/$module_name}"
+    template="${template//\{WORKER_DEFINITIONS_FROM_PLAN\}/$workers_json}"
+    prompt+="$template"
+  else
+    prompt+="# Lead Agent: ${lead_name}"$'\n'
+    prompt+="Module: ${module_name}"$'\n\n'
+  fi
+
+  prompt+=$'\n\n'
+
+  # Spec content for reference
+  prompt+="## TECHNICAL SPEC"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/spec.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Worker definitions
+  prompt+="## WORKER DEFINITIONS"$'\n\n'
+  prompt+='```json'$'\n'
+  prompt+="$workers_json"
+  prompt+=$'\n```\n\n'
+
+  # Workflow instructions
+  prompt+="## DETAILED WORKFLOW INSTRUCTIONS"$'\n\n'
+  prompt+="You are the lead for module **${module_name}**. You are running as a Claude Code instance inside a git worktree at \`${FORGE_ROOT}/.forge/worktrees/${lead_name}\`. Follow these steps exactly:"$'\n\n'
+
+  prompt+="### Step 1: Create worktrees for each worker"$'\n'
+  prompt+="For each worker in your workers list, create a git worktree branching off YOUR lead branch:"$'\n'
+  prompt+='```bash'$'\n'
+  prompt+="cd ${FORGE_ROOT}"$'\n'
+  prompt+="git worktree add ${FORGE_ROOT}/.forge/worktrees/<worker-name> -b forge/<worker-name> forge/${lead_name}"$'\n'
+  prompt+='```'$'\n\n'
+
+  prompt+="### Step 2: Write task spec files"$'\n'
+  prompt+="For each worker, write a detailed task specification as a markdown file. Include the task description, file list, and relevant spec excerpts from the worker definitions above."$'\n'
+  prompt+='```bash'$'\n'
+  prompt+="mkdir -p ${FORGE_ROOT}/.forge/tasks"$'\n'
+  prompt+="# Write detailed task to ${FORGE_ROOT}/.forge/tasks/<worker-name>.md"$'\n'
+  prompt+='```'$'\n'
+  prompt+="The task file should contain everything the worker needs: what to build, which files to create/modify, relevant API contracts, data models, conventions to follow, and a reminder to commit changes when done."$'\n\n'
+
+  prompt+="### Step 3: Spawn workers via tmux"$'\n'
+  prompt+="Launch each worker as a separate Claude Code instance in its own tmux session. IMPORTANT: Use \`forge-\` prefix for session names."$'\n'
+  prompt+='```bash'$'\n'
+  prompt+="# For each worker:"$'\n'
+  prompt+="touch ${FORGE_ROOT}/.forge/status/<worker-name>.expected"$'\n'
+  prompt+="tmux new-session -d -s \"forge-<worker-name>\" \\"$'\n'
+  prompt+="  \"cd ${FORGE_ROOT}/.forge/worktrees/<worker-name> && claude -p \\\"\\\$(cat ${FORGE_ROOT}/.forge/tasks/<worker-name>.md)\\\" --allowedTools 'Read,Write,Bash' 2>&1 | tee ${FORGE_ROOT}/.forge/logs/<worker-name>.log; echo done > ${FORGE_ROOT}/.forge/status/<worker-name>.done\""$'\n'
+  prompt+='```'$'\n\n'
+
+  prompt+="### Step 4: Poll for worker completion"$'\n'
+  prompt+="Monitor workers by checking for .done files. Poll every 30 seconds:"$'\n'
+  prompt+='```bash'$'\n'
+  prompt+="# Loop until all workers are done"$'\n'
+  prompt+="while true; do"$'\n'
+  prompt+="  done_count=\$(ls ${FORGE_ROOT}/.forge/status/worker-*.done 2>/dev/null | wc -l | tr -d ' ')"$'\n'
+  prompt+="  expected_count=\$(ls ${FORGE_ROOT}/.forge/status/worker-*.expected 2>/dev/null | wc -l | tr -d ' ')"$'\n'
+  prompt+="  echo \"Workers: \${done_count}/\${expected_count} done\""$'\n'
+  prompt+="  [ \"\$done_count\" -ge \"\$expected_count\" ] && break"$'\n'
+  prompt+="  sleep 30"$'\n'
+  prompt+="done"$'\n'
+  prompt+='```'$'\n'
+  prompt+="Also check if any tmux session died without creating a .done file — if so, the worker crashed. You may respawn it (max 2 retries)."$'\n\n'
+
+  prompt+="### Step 5: Review each worker's output"$'\n'
+  prompt+="For each completed worker, review its changes:"$'\n'
+  prompt+='```bash'$'\n'
+  prompt+="cd ${FORGE_ROOT}/.forge/worktrees/<worker-name>"$'\n'
+  prompt+="git log --oneline forge/${lead_name}..HEAD"$'\n'
+  prompt+="git diff forge/${lead_name}..HEAD"$'\n'
+  prompt+='```'$'\n'
+  prompt+="Check: Does the code match the task spec? Is it well-structured? Are there obvious bugs?"$'\n'
+  prompt+="If a worker's output is poor, write fix instructions and respawn (max 2 retries per worker)."$'\n\n'
+
+  prompt+="### Step 6: Merge all worker branches into your lead branch"$'\n'
+  prompt+="After all workers pass review, merge them sequentially:"$'\n'
+  prompt+='```bash'$'\n'
+  prompt+="cd ${FORGE_ROOT}/.forge/worktrees/${lead_name}"$'\n'
+  prompt+="git merge --no-ff forge/<worker-name> -m \"Merge <worker-name> implementation\""$'\n'
+  prompt+="# Repeat for each worker"$'\n'
+  prompt+='```'$'\n'
+  prompt+="Resolve any merge conflicts if they occur (they should be rare since workers own exclusive files)."$'\n\n'
+
+  prompt+="### Step 7: Write review and signal completion"$'\n'
+  prompt+='```bash'$'\n'
+  prompt+="# Write a review summary"$'\n'
+  prompt+="cat > ${FORGE_ROOT}/.forge/reviews/${lead_name}-review.json << 'EOF'"$'\n'
+  prompt+='{"lead": "'"${lead_name}"'", "module": "'"${module_name}"'", "status": "complete", "workers_merged": [...], "issues": [...]}'$'\n'
+  prompt+="EOF"$'\n\n'
+  prompt+="# Signal completion"$'\n'
+  prompt+="echo \"done\" > ${FORGE_ROOT}/.forge/status/${lead_name}.done"$'\n'
+  prompt+='```'$'\n\n'
+
+  prompt+="## FORGE_ROOT"$'\n\n'
+  prompt+="All absolute paths use: \`${FORGE_ROOT}\`"$'\n'
+  prompt+="Your worktree is at: \`${FORGE_ROOT}/.forge/worktrees/${lead_name}\`"$'\n'
+  prompt+="Your branch is: \`forge/${lead_name}\`"$'\n'
+
+  prompt+="$ANTI_ATTRIBUTION_RULES"
+
+  echo "$prompt"
+}
+
+# ── Worker prompt ────────────────────────────────────────────
+
+generate_worker_prompt() {
+  local worker_name="$1"
+  local task_description="$2"
+  local file_list="$3"
+  local spec_excerpt="$4"
+
+  local prompt=""
+
+  # Base template
+  local template
+  template=$(cat "${FORGE_DIR}/prompts/worker.md" 2>/dev/null)
+
+  if [[ -n "$template" ]]; then
+    template="${template//\{TASK_DESCRIPTION\}/$task_description}"
+    template="${template//\{FILE_LIST\}/$file_list}"
+    template="${template//\{SPEC_EXCERPT_RELEVANT_TO_THIS_TASK\}/$spec_excerpt}"
+    prompt+="$template"
+  else
+    prompt+="# Worker Agent: ${worker_name}"$'\n\n'
+    prompt+="## TASK DESCRIPTION"$'\n\n'
+    prompt+="$task_description"$'\n\n'
+    prompt+="## FILES TO WORK ON"$'\n\n'
+    prompt+="$file_list"$'\n\n'
+    prompt+="## RELEVANT SPEC EXCERPT"$'\n\n'
+    prompt+="$spec_excerpt"$'\n\n'
+  fi
+
+  prompt+=$'\n\n'
+
+  # Completion instruction
+  prompt+="## COMPLETION"$'\n\n'
+  prompt+="When done, commit all changes and create the completion marker:"$'\n'
+  prompt+='```bash'$'\n'
+  prompt+="touch ${FORGE_ROOT}/.forge/status/${worker_name}.done"$'\n'
+  prompt+='```'$'\n'
+
+  prompt+="$ANTI_ATTRIBUTION_RULES"
+
+  echo "$prompt"
+}
+
+# ── Senior auditor prompt ───────────────────────────────────
+
+generate_senior_auditor_prompt() {
+  local prompt=""
+
+  # Base template
+  prompt+=$(cat "${FORGE_DIR}/prompts/senior_auditor.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Context
+  prompt+="## CONTEXT"$'\n\n'
+  prompt+="You are auditing the fully integrated codebase after all modules have been merged."$'\n'
+  prompt+="Review the entire project for correctness, consistency, and completeness."$'\n\n'
+
+  # FORGE_ROOT
+  prompt+="## FORGE_ROOT"$'\n\n'
+  prompt+="All absolute paths use: ${FORGE_ROOT}"$'\n\n'
+
+  # Spec for reference
+  prompt+="## APPROVED SPEC"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/spec.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Plan for reference
+  prompt+="## IMPLEMENTATION PLAN"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/plan.json" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Instruction
+  prompt+="## INSTRUCTION"$'\n\n'
+  prompt+="Write the consolidated audit report to ${FORGE_ROOT}/.forge/audit/consolidated-audit.json"$'\n'
+  prompt+="Write the audit plan to ${FORGE_ROOT}/.forge/audit/senior-audit-plan.json"$'\n'
+  prompt+="Each junior auditor should write their findings to ${FORGE_ROOT}/.forge/audit/junior-{name}.json"$'\n'
+
+  prompt+="$ANTI_ATTRIBUTION_RULES"
+
+  echo "$prompt"
+}
+
+# ── Final report prompt ─────────────────────────────────────
+
+generate_final_report_prompt() {
+  local user_prompt="$1"
+
+  local prompt=""
+
+  # Base template
+  prompt+=$(cat "${FORGE_DIR}/prompts/coordinator_final.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Original task
+  prompt+="## ORIGINAL TASK"$'\n\n'
+  prompt+="$user_prompt"$'\n\n'
+
+  # Approved spec
+  prompt+="## APPROVED SPEC"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/spec/spec.md" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Plan
+  prompt+="## IMPLEMENTATION PLAN"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/plan.json" 2>/dev/null)
+  prompt+=$'\n\n'
+
+  # Lead reviews
+  prompt+="## LEAD REVIEWS"$'\n\n'
+  local lead_reviews
+  lead_reviews=$(cat "${FORGE_ROOT}"/.forge/reviews/*.json 2>/dev/null || echo "No lead reviews available")
+  prompt+="$lead_reviews"$'\n\n'
+
+  # Audit results
+  prompt+="## AUDIT RESULTS"$'\n\n'
+  prompt+=$(cat "${FORGE_ROOT}/.forge/audit/consolidated-audit.json" 2>/dev/null || echo "No audit results available")
+  prompt+=$'\n\n'
+
+  # Fix reports
+  prompt+="## FIX REPORTS"$'\n\n'
+  local fix_reports
+  fix_reports=$(cat "${FORGE_ROOT}"/.forge/fixes/*.json 2>/dev/null || echo "No fix reports available")
+  prompt+="$fix_reports"$'\n\n'
+
+  # Instruction
+  prompt+="## INSTRUCTION"$'\n\n'
+  prompt+="Write the final project report to ${FORGE_ROOT}/.forge/DONE.md"$'\n'
+
+  prompt+="$ANTI_ATTRIBUTION_RULES"
+
+  echo "$prompt"
+}