vibe-forge 0.3.12 → 0.8.1

package/bin/forge-daemon.sh

@@ -1,775 +1,477 @@
-#!/usr/bin/env bash
-#
-# Vibe Forge - Background Daemon
-# Monitors task folders and routes files automatically
-#
-# Usage:
-#   forge-daemon.sh start   - Start the daemon
-#   forge-daemon.sh stop    - Stop the daemon
-#   forge-daemon.sh status  - Check daemon status
-#
-
-SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
-FORGE_ROOT="$(dirname "$SCRIPT_DIR")"
-
-# =============================================================================
-# Load Shared Libraries
-# =============================================================================
-
-# shellcheck source=lib/colors.sh
-source "$SCRIPT_DIR/lib/colors.sh"
-# shellcheck source=lib/constants.sh
-source "$SCRIPT_DIR/lib/constants.sh"
-# shellcheck source=lib/config.sh
-source "$SCRIPT_DIR/lib/config.sh"
-# shellcheck source=lib/database.sh
-source "$SCRIPT_DIR/lib/database.sh"
-
-# =============================================================================
-# Daemon Configuration
-# =============================================================================
-
-CONFIG_FILE="$FORGE_ROOT/.forge/config.json"
-FORGE_DB="$FORGE_ROOT/.forge/forge.db"
-PID_FILE="$FORGE_ROOT/.forge/daemon.pid"
-LOG_FILE="$FORGE_ROOT/.forge/daemon.log"
-NOTIFY_FILE="$FORGE_ROOT/.forge/notifications.log"
-NOTIFIED_FILE="$FORGE_ROOT/.forge/notified-tasks.txt"
-STATE_FILE="$FORGE_ROOT/$CONTEXT_DIR/forge-state.yaml"
-LOCK_FILE="$FORGE_ROOT/.forge/daemon.lock"
-
-# Log file rotation settings
-MAX_LOG_SIZE=1048576  # 1MB
-MAX_NOTIFY_ENTRIES=1000
-
-# Load terminal type from config (safe parsing)
-TERMINAL_TYPE="manual"
-if [[ -f "$CONFIG_FILE" ]]; then
-    TERMINAL_TYPE=$(json_get_string "$CONFIG_FILE" "terminal_type") || TERMINAL_TYPE="manual"
-fi
-
-# =============================================================================
-# Utility Functions
-# =============================================================================
-
-# Rotate log file if it gets too large
-rotate_log_if_needed() {
-    local log="$1"
-    if [[ -f "$log" ]]; then
-        local size
-        size=$(stat -f%z "$log" 2>/dev/null || stat --format=%s "$log" 2>/dev/null || echo 0)
-        if [[ "$size" -gt "$MAX_LOG_SIZE" ]]; then
-            mv "$log" "${log}.old"
-            touch "$log"
-        fi
-    fi
-}
-
-# Trim notification entries to prevent unbounded growth
-trim_notified_file() {
-    if [[ -f "$NOTIFIED_FILE" ]]; then
-        local count
-        count=$(wc -l < "$NOTIFIED_FILE" 2>/dev/null || echo 0)
-        if [[ "$count" -gt "$MAX_NOTIFY_ENTRIES" ]]; then
-            # Keep last 500 entries
-            tail -500 "$NOTIFIED_FILE" > "${NOTIFIED_FILE}.tmp"
-            mv "${NOTIFIED_FILE}.tmp" "$NOTIFIED_FILE"
-        fi
-    fi
-}
-
-# Safe file move with symlink protection
-safe_move_task() {
-    local src="$1"
-    local dest_dir="$2"
-
-    # SECURITY: Skip symlinks to prevent symlink attacks
-    if [[ -L "$src" ]]; then
-        echo "[$(date -Iseconds)] WARNING: Skipping symlink: $src" >> "$LOG_FILE"
-        return 1
-    fi
-
-    # SECURITY: Verify source is a regular file
-    if [[ ! -f "$src" ]]; then
-        return 1
-    fi
-
-    # SECURITY: Verify destination is within FORGE_ROOT
-    local real_dest
-    real_dest=$(cd "$dest_dir" 2>/dev/null && pwd)
-    local forge_root_real
-    forge_root_real=$(cd "$FORGE_ROOT" 2>/dev/null && pwd)
-
-    if [[ "$real_dest" != "$forge_root_real"/* ]]; then
-        echo "[$(date -Iseconds)] ERROR: Destination outside FORGE_ROOT: $dest_dir" >> "$LOG_FILE"
-        return 1
-    fi
-
-    local filename
-    filename=$(basename "$src")
-    mv "$src" "$dest_dir/$filename"
-}
-
-# =============================================================================
-# Notification Functions
-# =============================================================================
-
-notify() {
-    local message="$1"
-    local urgency="${2:-normal}"  # normal or urgent
-    local timestamp
-    timestamp=$(date -Iseconds)
-
-    # Log to notifications file
-    echo "[$timestamp] $message" >> "$NOTIFY_FILE"
-
-    # Log to main log
-    echo "[$timestamp] NOTIFY: $message" >> "$LOG_FILE"
-
-    # Terminal bell (works in most terminals)
-    printf '\a'
-
-    # System toast notification for urgent messages
-    if [[ "$urgency" == "urgent" ]]; then
-        send_system_notification "$message"
-    fi
-}
-
-# Send system-level notification (platform-specific)
-send_system_notification() {
-    local message="$1"
-    local title="Vibe Forge"
-
-    case "$(uname -s)" in
-        MINGW*|MSYS*|CYGWIN*)
-            # Windows: Use PowerShell toast notification
-            powershell.exe -NoProfile -Command "
-                \$null = [Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType = WindowsRuntime]
-                \$template = [Windows.UI.Notifications.ToastNotificationManager]::GetTemplateContent([Windows.UI.Notifications.ToastTemplateType]::ToastText02)
-                \$textNodes = \$template.GetElementsByTagName('text')
-                \$textNodes.Item(0).AppendChild(\$template.CreateTextNode('$title')) | Out-Null
-                \$textNodes.Item(1).AppendChild(\$template.CreateTextNode('$message')) | Out-Null
-                \$toast = [Windows.UI.Notifications.ToastNotification]::new(\$template)
-                [Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier('Vibe Forge').Show(\$toast)
-            " 2>/dev/null &
-            ;;
-        Darwin)
-            # macOS: Use osascript
-            osascript -e "display notification \"$message\" with title \"$title\"" 2>/dev/null &
-            ;;
-        Linux)
-            # Linux: Use notify-send if available
-            if command -v notify-send &>/dev/null; then
-                notify-send "$title" "$message" 2>/dev/null &
-            fi
-            ;;
-    esac
-}
-
-check_new_pending_tasks() {
-    # Create notified file if it doesn't exist
-    touch "$NOTIFIED_FILE"
-
-    # Check for new pending tasks
-    for task in "$FORGE_ROOT/$TASKS_PENDING"/*.md; do
-        if [[ -f "$task" && ! -L "$task" ]]; then
-            local filename
-            filename=$(basename "$task")
-
-            # Check if we've already notified about this task
-            if ! grep -qF "$filename" "$NOTIFIED_FILE" 2>/dev/null; then
-                # Extract task info from frontmatter safely
-                local task_id task_title assigned_to
-
-                # Use head to limit read, tr to sanitize, and strip ANSI escape sequences
-                task_id=$(grep -m1 "^id:" "$task" 2>/dev/null | cut -d':' -f2 | tr -d ' "' | tr -d '\033' | sed 's/\[[0-9;]*m//g' | head -c 100)
-                task_title=$(grep -m1 "^title:" "$task" 2>/dev/null | cut -d':' -f2- | tr -d '"' | tr -d '\033' | sed 's/\[[0-9;]*m//g' | head -c 200)
-                assigned_to=$(grep -m1 "^assigned_to:" "$task" 2>/dev/null | cut -d':' -f2 | tr -d ' "' | tr -d '\033' | sed 's/\[[0-9;]*m//g' | head -c 50)
-
-                # Use filename as fallback
-                task_id="${task_id:-$filename}"
-                task_title="${task_title:-New task}"
-
-                # Notify
-                if [[ -n "$assigned_to" ]]; then
-                    notify "New task for $assigned_to: $task_title ($task_id)"
-                else
-                    notify "New pending task: $task_title ($task_id)"
-                fi
-
-                # Mark as notified (atomic append)
-                echo "$filename" >> "$NOTIFIED_FILE"
-            fi
-        fi
-    done
-
-    # Also check needs-changes for tasks that need rework
-    for task in "$FORGE_ROOT/$TASKS_NEEDS_CHANGES"/*.md; do
-        if [[ -f "$task" && ! -L "$task" ]]; then
-            local filename
-            filename=$(basename "$task")
-            local notified_key="needs-changes:$filename"
-
-            if ! grep -qF "$notified_key" "$NOTIFIED_FILE" 2>/dev/null; then
-                local task_id assigned_to
-                task_id=$(grep -m1 "^id:" "$task" 2>/dev/null | cut -d':' -f2 | tr -d ' "' | head -c 100)
-                assigned_to=$(grep -m1 "^assigned_to:" "$task" 2>/dev/null | cut -d':' -f2 | tr -d ' "' | head -c 50)
-
-                task_id="${task_id:-$filename}"
-
-                if [[ -n "$assigned_to" ]]; then
-                    notify "Task needs changes ($assigned_to): $task_id"
-                else
-                    notify "Task needs changes: $task_id"
-                fi
-
-                echo "$notified_key" >> "$NOTIFIED_FILE"
-            fi
-        fi
-    done
-}
-
-check_attention_needed() {
-    # Check for workers needing attention (urgent notifications)
-    if [[ ! -d "$FORGE_ROOT/$TASKS_ATTENTION" ]]; then
-        return 0
-    fi
-
-    for attention_file in "$FORGE_ROOT/$TASKS_ATTENTION"/*.md; do
-        if [[ -f "$attention_file" && ! -L "$attention_file" ]]; then
-            local filename
-            filename=$(basename "$attention_file")
-            local notified_key="attention:$filename"
-
-            if ! grep -qF "$notified_key" "$NOTIFIED_FILE" 2>/dev/null; then
-                # Extract attention info
-                local agent issue
-                agent=$(grep -m1 "^agent:" "$attention_file" 2>/dev/null | cut -d':' -f2 | tr -d ' "' | head -c 50)
-                issue=$(grep -m1 "^##" "$attention_file" 2>/dev/null | sed 's/^## *//' | head -c 200)
-
-                agent="${agent:-Unknown}"
-                issue="${issue:-Needs attention}"
-
-                # Ring bell multiple times for attention
-                printf '\a\a\a'
-
-                # Send urgent notification with toast
-                notify "🔔 $agent needs help: $issue" "urgent"
-
-                echo "$notified_key" >> "$NOTIFIED_FILE"
-            fi
-        fi
-    done
-}
-
-# Sync agent status from JSON files to SQLite (with mtime filtering)
-sync_agent_status_to_db() {
-    local status_dir="$FORGE_ROOT/$AGENT_STATUS_DIR"
-
-    if [[ ! -d "$status_dir" ]]; then
-        return 0
-    fi
-
-    for status_file in "$status_dir"/*.json; do
-        if [[ -f "$status_file" && ! -L "$status_file" ]]; then
-            # Get file modification time
-            local file_mtime
-            file_mtime=$(stat -c %Y "$status_file" 2>/dev/null || stat -f %m "$status_file" 2>/dev/null || echo "0")
-
-            # Get agent name from filename
-            local agent_name
-            agent_name=$(basename "$status_file" .json)
-
-            # Check if file has changed since last read
-            local stored_mtime
-            stored_mtime=$(db_get_agent_mtime "$agent_name")
-
-            if [[ "$file_mtime" -gt "$stored_mtime" ]]; then
-                # File changed - parse and update DB
-                local agent status task message updated
-                agent=$(jq -r '.agent // "unknown"' "$status_file" 2>/dev/null)
-                status=$(jq -r '.status // "unknown"' "$status_file" 2>/dev/null)
-                task=$(jq -r '.task // ""' "$status_file" 2>/dev/null)
-                message=$(jq -r '.message // ""' "$status_file" 2>/dev/null | head -c 80)
-                updated=$(jq -r '.updated // ""' "$status_file" 2>/dev/null)
-
-                # Upsert to database
-                db_upsert_agent_status "$agent" "$status" "$task" "$message" "$updated" "$file_mtime"
-
-                echo "[$(date -Iseconds)] Synced status for $agent: $status" >> "$LOG_FILE"
-            fi
-        fi
-    done
-}
-
-# Build worker status from SQLite (for YAML output)
-build_worker_status() {
-    local now_epoch
-    now_epoch=$(date +%s)
-    local stale_threshold=300  # 5 minutes
-
-    # Check if we have any agent status in DB
-    local agent_count
-    agent_count=$(sqlite3 "$FORGE_DB" "SELECT COUNT(*) FROM agent_status;" 2>/dev/null || echo "0")
-
-    if [[ "$agent_count" -eq 0 ]]; then
-        return 0
-    fi
-
-    echo "workers:"
-
-    # Read from database
-    while IFS='|' read -r agent status task message updated; do
-        local stale_marker=""
-
-        # Check if stale
-        if [[ -n "$updated" ]]; then
-            local updated_epoch age
-            updated_epoch=$(date -d "$updated" +%s 2>/dev/null || date -j -f "%Y-%m-%dT%H:%M:%S" "${updated%Z}" +%s 2>/dev/null || echo "0")
-            age=$((now_epoch - updated_epoch))
-            if [[ "$age" -gt "$stale_threshold" ]]; then
-                stale_marker=" (stale)"
-            fi
-        fi
-
-        echo "  - agent: $agent"
-        echo "    status: $status$stale_marker"
-        if [[ -n "$task" ]]; then
-            echo "    task: $task"
-        fi
-        if [[ -n "$message" ]]; then
-            echo "    message: \"$message\""
-        fi
-        echo "    updated: $updated"
-    done < <(db_get_all_agent_statuses)
-}
-
-# =============================================================================
-# Daemon Functions
-# =============================================================================
-
-update_state() {
-    # Count tasks in each folder (using find with -maxdepth for safety)
-    local pending in_progress completed review approved needs_changes merged attention
-    pending=$(find "$FORGE_ROOT/$TASKS_PENDING" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-    in_progress=$(find "$FORGE_ROOT/$TASKS_IN_PROGRESS" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-    completed=$(find "$FORGE_ROOT/$TASKS_COMPLETED" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-    review=$(find "$FORGE_ROOT/$TASKS_REVIEW" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-    approved=$(find "$FORGE_ROOT/$TASKS_APPROVED" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-    needs_changes=$(find "$FORGE_ROOT/$TASKS_NEEDS_CHANGES" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-    merged=$(find "$FORGE_ROOT/$TASKS_MERGED" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-    attention=$(find "$FORGE_ROOT/$TASKS_ATTENTION" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-
-    local blocked=0
-
-    # Build attention details if any workers need help
-    local attention_details=""
-    if [[ "$attention" -gt 0 ]]; then
-        attention_details=$(build_attention_details)
-    fi
-
-    # Build worker status from agent-status files
-    local worker_status=""
-    if [[ -d "$FORGE_ROOT/$AGENT_STATUS_DIR" ]]; then
-        worker_status=$(build_worker_status)
-    fi
-
-    # Write state file atomically (write to temp, then move)
-    local temp_state="${STATE_FILE}.tmp.$$"
-    cat > "$temp_state" << EOF
-# Vibe Forge State
-# Auto-updated by forge-daemon
-# Last updated: $(date -Iseconds)
-
-forge:
-  status: active
-  daemon_pid: $$
-
-tasks:
-  pending: $pending
-  in_progress: $in_progress
-  completed: $completed
-  in_review: $review
-  approved: $approved
-  needs_changes: $needs_changes
-  merged: $merged
-  blocked: $blocked
-  attention_needed: $attention
-
-$attention_details
-$worker_status
-last_updated: $(date -Iseconds)
-EOF
-    mv "$temp_state" "$STATE_FILE"
-}
-
-build_attention_details() {
-    echo "attention:"
-    for attention_file in "$FORGE_ROOT/$TASKS_ATTENTION"/*.md; do
-        if [[ -f "$attention_file" && ! -L "$attention_file" ]]; then
-            local agent created issue
-            agent=$(grep -m1 "^agent:" "$attention_file" 2>/dev/null | cut -d':' -f2 | tr -d ' "' | head -c 50)
-            created=$(grep -m1 "^created:" "$attention_file" 2>/dev/null | cut -d':' -f2- | tr -d ' ' | head -c 30)
-            # Get the issue line (first ## heading content or fallback)
-            issue=$(sed -n '/^## Issue/,/^##/p' "$attention_file" 2>/dev/null | grep -v "^##" | head -1 | tr -d '\n' | head -c 100)
-            issue="${issue:-Needs attention}"
-
-            echo "  - agent: $agent"
-            echo "    since: $created"
-            echo "    issue: \"$issue\""
-        fi
-    done
-}
-
-route_completed_to_review() {
-    # Move completed tasks to review queue
-    for task in "$FORGE_ROOT/$TASKS_COMPLETED"/*.md; do
-        if [[ -f "$task" && ! -L "$task" ]]; then
-            local filename
-            filename=$(basename "$task")
-            echo "[$(date -Iseconds)] Routing $filename to review" >> "$LOG_FILE"
-            safe_move_task "$task" "$FORGE_ROOT/$TASKS_REVIEW"
-        fi
-    done
-}
-
-route_approved_to_merged() {
-    # Move approved tasks to merged archive
-    for task in "$FORGE_ROOT/$TASKS_APPROVED"/*.md; do
-        if [[ -f "$task" && ! -L "$task" ]]; then
-            local filename
-            filename=$(basename "$task")
-            echo "[$(date -Iseconds)] Archiving $filename to merged" >> "$LOG_FILE"
-            safe_move_task "$task" "$FORGE_ROOT/$TASKS_MERGED"
-        fi
-    done
-}
-
-# Determine daemon state based on activity (for adaptive polling)
-determine_daemon_state() {
-    # Check if there are in-progress tasks
-    local in_progress_count
-    in_progress_count=$(find "$FORGE_ROOT/$TASKS_IN_PROGRESS" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-
-    # Check if there are active workers
-    local active_workers
-    active_workers=$(db_count_active_workers 2>/dev/null || echo "0")
-
-    if [[ "$in_progress_count" -gt 0 ]] || [[ "$active_workers" -gt 0 ]]; then
-        echo "active"
-    else
-        echo "idle"
-    fi
-}
-
-# Get current poll interval in seconds (from DB, with fallback)
-get_poll_interval() {
-    local interval_ms
-    interval_ms=$(db_get_poll_interval_ms 2>/dev/null || echo "30000")
-    # Convert ms to seconds (bash integer division)
-    echo $((interval_ms / 1000))
-}
-
-daemon_loop() {
-    echo "[$(date -Iseconds)] Forge daemon started (PID: $$)" >> "$LOG_FILE"
-
-    # Create lock file
-    echo $$ > "$LOCK_FILE"
-
-    # Initialize database
-    db_init
-    echo "[$(date -Iseconds)] Database initialized at $FORGE_DB" >> "$LOG_FILE"
-
-    # Cleanup on exit
-    trap 'rm -f "$LOCK_FILE"; echo "[$(date -Iseconds)] Daemon exiting" >> "$LOG_FILE"' EXIT
-
-    local iteration=0
-    local current_state="idle"
-    local poll_interval=2  # Start fast, adjust based on activity
-
-    while true; do
-        # Increment iteration counter
-        ((iteration++)) || true
-
-        # Sync agent status from JSON files to SQLite (with mtime filtering)
-        sync_agent_status_to_db
-
-        # Check for new tasks and notify
-        check_new_pending_tasks
-
-        # Check for workers needing attention (urgent)
-        check_attention_needed
-
-        # Route tasks
-        route_completed_to_review
-        route_approved_to_merged
-
-        # Update state file
-        update_state
-
-        # Adaptive polling: check activity and adjust interval
-        local new_state
-        new_state=$(determine_daemon_state)
-        if [[ "$new_state" != "$current_state" ]]; then
-            current_state="$new_state"
-            db_set_daemon_state "$current_state"
-            poll_interval=$(get_poll_interval)
-            echo "[$(date -Iseconds)] State changed to $current_state, poll interval: ${poll_interval}s" >> "$LOG_FILE"
-        fi
-
-        # Periodic maintenance (every 100 iterations)
-        if [[ $((iteration % 100)) -eq 0 ]]; then
-            rotate_log_if_needed "$LOG_FILE"
-            rotate_log_if_needed "$NOTIFY_FILE"
-            trim_notified_file
-            # Cleanup stale agent status (30+ minutes old)
-            db_cleanup_stale_agents 30
-            # Prune history older than 7 days
-            db_prune_history 7
-        fi
-
-        sleep "$poll_interval"
-    done
-}
-
-# =============================================================================
-# Commands
-# =============================================================================
-
-cmd_start() {
-    # Check if already running
-    if [[ -f "$PID_FILE" ]]; then
-        local pid
-        pid=$(cat "$PID_FILE")
-        if kill -0 "$pid" 2>/dev/null; then
-            echo "Daemon already running (PID: $pid)"
-            return 0
-        else
-            # Stale PID file
-            rm -f "$PID_FILE"
-        fi
-    fi
-
-    # Check for lock file (another instance check)
-    if [[ -f "$LOCK_FILE" ]]; then
-        local lock_pid
-        lock_pid=$(cat "$LOCK_FILE" 2>/dev/null)
-        if kill -0 "$lock_pid" 2>/dev/null; then
-            echo "Another daemon instance is running (PID: $lock_pid)"
-            return 1
-        else
-            rm -f "$LOCK_FILE"
-        fi
-    fi
-
-    # Create directories if needed (with secure permissions)
-    mkdir -p "$FORGE_ROOT/.forge"
-    chmod 700 "$FORGE_ROOT/.forge"
-
-    mkdir -p "$FORGE_ROOT/$TASKS_PENDING"
-    mkdir -p "$FORGE_ROOT/$TASKS_IN_PROGRESS"
-    mkdir -p "$FORGE_ROOT/$TASKS_COMPLETED"
-    mkdir -p "$FORGE_ROOT/$TASKS_REVIEW"
-    mkdir -p "$FORGE_ROOT/$TASKS_APPROVED"
-    mkdir -p "$FORGE_ROOT/$TASKS_NEEDS_CHANGES"
-    mkdir -p "$FORGE_ROOT/$TASKS_MERGED"
-    mkdir -p "$FORGE_ROOT/$TASKS_ATTENTION"
-    mkdir -p "$FORGE_ROOT/$AGENT_STATUS_DIR"
-
-    # Start daemon in background
-    daemon_loop &
-    local pid=$!
-    echo "$pid" > "$PID_FILE"
-
-    log_success "Forge daemon started (PID: $pid)"
-    echo "   Log: $LOG_FILE"
-}
-
-cmd_stop() {
-    if [[ ! -f "$PID_FILE" ]]; then
-        echo "Daemon not running"
-        return 0
-    fi
-
-    local pid
-    pid=$(cat "$PID_FILE")
-
-    if kill -0 "$pid" 2>/dev/null; then
-        kill "$pid"
-        rm -f "$PID_FILE" "$LOCK_FILE"
-        echo "[$(date -Iseconds)] Forge daemon stopped" >> "$LOG_FILE"
-        log_success "Daemon stopped"
-    else
-        rm -f "$PID_FILE" "$LOCK_FILE"
-        echo "Daemon was not running (stale PID file removed)"
-    fi
-
-    # Update state to show inactive
-    if [[ -f "$STATE_FILE" ]]; then
-        # Use sed carefully - different syntax on macOS vs Linux
-        if [[ "$(uname)" == "Darwin" ]]; then
-            sed -i '' 's/status: active/status: stopped/' "$STATE_FILE" 2>/dev/null || true
-        else
-            sed -i 's/status: active/status: stopped/' "$STATE_FILE" 2>/dev/null || true
-        fi
-    fi
-}
-
-cmd_status() {
-    echo ""
-    log_header "🔥 Forge Daemon Status"
-
-    if [[ -f "$PID_FILE" ]]; then
-        local pid
-        pid=$(cat "$PID_FILE")
-        if kill -0 "$pid" 2>/dev/null; then
-            log_success "Running (PID: $pid)"
-        else
-            log_warn "Stopped (stale PID file)"
-        fi
-    else
-        echo "Status: Stopped"
-    fi
-
-    echo ""
-
-    if [[ -f "$STATE_FILE" ]]; then
-        echo "Task Counts:"
-        grep -E "pending:|in_progress:|completed:|in_review:|approved:|needs_changes:|merged:|attention_needed:" "$STATE_FILE" | sed 's/^/  /'
-    fi
-
-    echo ""
-
-    # Show attention-needed workers prominently
-    local attention_count
-    attention_count=$(find "$FORGE_ROOT/$TASKS_ATTENTION" -maxdepth 1 -name "*.md" -type f 2>/dev/null | wc -l)
-    if [[ "$attention_count" -gt 0 ]]; then
-        echo -e "${RED}🔔 ATTENTION NEEDED:${NC}"
-        for attention_file in "$FORGE_ROOT/$TASKS_ATTENTION"/*.md; do
-            if [[ -f "$attention_file" && ! -L "$attention_file" ]]; then
-                local agent issue
-                agent=$(grep -m1 "^agent:" "$attention_file" 2>/dev/null | cut -d':' -f2 | tr -d ' "' | head -c 50)
-                issue=$(sed -n '/^## Issue/,/^##/p' "$attention_file" 2>/dev/null | grep -v "^##" | head -1 | head -c 80)
-                echo -e "  ${YELLOW}$agent${NC}: $issue"
-            fi
-        done
-        echo ""
-    fi
-
-    # Show worker status
-    if [[ -d "$FORGE_ROOT/$AGENT_STATUS_DIR" ]]; then
-        local status_count
-        status_count=$(find "$FORGE_ROOT/$AGENT_STATUS_DIR" -maxdepth 1 -name "*.json" -type f 2>/dev/null | wc -l)
-        if [[ "$status_count" -gt 0 ]]; then
-            echo "Active Workers:"
-            local now_epoch stale_threshold
-            now_epoch=$(date +%s)
-            stale_threshold=300
-            for status_file in "$FORGE_ROOT/$AGENT_STATUS_DIR"/*.json; do
-                if [[ -f "$status_file" && ! -L "$status_file" ]]; then
-                    local agent status task updated stale_marker icon
-                    agent=$(jq -r '.agent // "unknown"' "$status_file" 2>/dev/null)
-                    status=$(jq -r '.status // "unknown"' "$status_file" 2>/dev/null)
-                    task=$(jq -r '.task // ""' "$status_file" 2>/dev/null)
-                    updated=$(jq -r '.updated // ""' "$status_file" 2>/dev/null)
-
-                    # Check staleness
-                    stale_marker=""
-                    if [[ -n "$updated" ]]; then
-                        local updated_epoch age
-                        updated_epoch=$(date -d "$updated" +%s 2>/dev/null || date -j -f "%Y-%m-%dT%H:%M:%S" "${updated%Z}" +%s 2>/dev/null || echo "0")
-                        age=$((now_epoch - updated_epoch))
-                        if [[ "$age" -gt "$stale_threshold" ]]; then
-                            stale_marker=" ${YELLOW}(stale)${NC}"
-                        fi
-                    fi
-
-                    # Status icon
-                    case "$status" in
-                        "working") icon="🔨" ;;
-                        "idle") icon="💤" ;;
-                        "blocked") icon="🚫" ;;
-                        "testing") icon="🧪" ;;
-                        "reviewing") icon="👁️" ;;
-                        *) icon="❓" ;;
-                    esac
-
-                    if [[ -n "$task" ]]; then
-                        echo -e "  $icon ${CYAN}$agent${NC}: $status ($task)$stale_marker"
-                    else
-                        echo -e "  $icon ${CYAN}$agent${NC}: $status$stale_marker"
-                    fi
-                fi
-            done
-            echo ""
-        fi
-    fi
-
-    # Show recent notifications
-    if [[ -f "$NOTIFY_FILE" ]]; then
-        local notify_count
-        notify_count=$(wc -l < "$NOTIFY_FILE" 2>/dev/null || echo "0")
-        if [[ "$notify_count" -gt 0 ]]; then
-            echo "Recent Notifications (last 5):"
-            tail -5 "$NOTIFY_FILE" | sed 's/^/  /'
-            echo ""
-        fi
-    fi
-
-    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
-}
-
-cmd_notifications() {
-    echo ""
-    log_header "🔔 Forge Notifications"
-
-    if [[ -f "$NOTIFY_FILE" ]]; then
-        local count="${1:-10}"
-        echo "Last $count notifications:"
-        echo ""
-        tail -"$count" "$NOTIFY_FILE"
-    else
-        echo "No notifications yet."
-    fi
-
-    echo ""
-    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
-}
-
-cmd_clear_notifications() {
-    rm -f "$NOTIFY_FILE" "$NOTIFIED_FILE"
-    log_success "Notifications cleared"
-}
-
-# =============================================================================
-# Main
-# =============================================================================
-
-main() {
-    local command="${1:-status}"
-
-    case "$command" in
-        "start")
-            cmd_start
-            ;;
-        "stop")
-            cmd_stop
-            ;;
-        "status")
-            cmd_status
-            ;;
-        "notifications"|"notify")
-            shift
-            cmd_notifications "$@"
-            ;;
-        "clear")
-            cmd_clear_notifications
-            ;;
-        *)
-            echo "Usage: forge-daemon.sh [start|stop|status|notifications|clear]"
-            exit 1
-            ;;
-    esac
-}
-
-main "$@"
+#!/usr/bin/env bash
+#
+# Vibe Forge - Background Daemon
+# Monitors task folders and routes files automatically
+#
+# Usage:
+#   forge-daemon.sh start   - Start the daemon
+#   forge-daemon.sh stop    - Stop the daemon
+#   forge-daemon.sh status  - Check daemon status
+#
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+FORGE_ROOT="$(dirname "$SCRIPT_DIR")"
+
+# =============================================================================
+# Load Shared Libraries
+# =============================================================================
+
+# shellcheck source=lib/colors.sh
+source "$SCRIPT_DIR/lib/colors.sh"
+# shellcheck source=lib/constants.sh
+source "$SCRIPT_DIR/lib/constants.sh"
+# shellcheck source=lib/config.sh
+source "$SCRIPT_DIR/lib/config.sh"
+# shellcheck source=lib/json.sh
+source "$SCRIPT_DIR/lib/json.sh"
+# shellcheck source=lib/database.sh
+source "$SCRIPT_DIR/lib/database.sh"
+# shellcheck source=lib/util.sh
+source "$SCRIPT_DIR/lib/util.sh"
+
+# Daemon modules (routing has no deps; sync before state)
+# shellcheck source=lib/daemon/routing.sh
+source "$SCRIPT_DIR/lib/daemon/routing.sh"
+# shellcheck source=lib/daemon/notifications.sh
+source "$SCRIPT_DIR/lib/daemon/notifications.sh"
+# shellcheck source=lib/daemon/sync.sh
+source "$SCRIPT_DIR/lib/daemon/sync.sh"
+# shellcheck source=lib/daemon/state.sh
+source "$SCRIPT_DIR/lib/daemon/state.sh"
+# shellcheck source=lib/daemon/display.sh
+source "$SCRIPT_DIR/lib/daemon/display.sh"
+# shellcheck source=lib/daemon/budgets.sh
+source "$SCRIPT_DIR/lib/daemon/budgets.sh"
+# shellcheck source=lib/daemon/dependencies.sh
+source "$SCRIPT_DIR/lib/daemon/dependencies.sh"
+
+# =============================================================================
+# Daemon Configuration
+# =============================================================================
+
+CONFIG_FILE="$FORGE_ROOT/.forge/config.json"
+FORGE_DB="$FORGE_ROOT/.forge/forge.db"
+PID_FILE="$FORGE_ROOT/.forge/daemon.pid"
+LOG_FILE="$FORGE_ROOT/.forge/daemon.log"
+NOTIFY_FILE="$FORGE_ROOT/.forge/notifications.log"
+NOTIFIED_FILE="$FORGE_ROOT/.forge/notified-tasks.txt"
+STATE_FILE="$FORGE_ROOT/$CONTEXT_DIR/forge-state.yaml"
+LOCK_FILE="$FORGE_ROOT/.forge/daemon.lock"
+DASHBOARD_PID_FILE="$FORGE_ROOT/.forge/dashboard.pid"
+
+# Log file rotation settings (values defined in constants.sh)
+# MAX_LOG_SIZE, MAX_NOTIFY_ENTRIES are loaded from constants.sh
+
+# Load config (safe parsing via json_get_string)
+TERMINAL_TYPE="manual"
+DASHBOARD_ENABLED="false"
+DASHBOARD_VOICE="false"
+DASHBOARD_PORT="2800"
+if [[ -f "$CONFIG_FILE" ]]; then
+    TERMINAL_TYPE=$(json_get_string "$CONFIG_FILE" "terminal_type")     || TERMINAL_TYPE="manual"
+    DASHBOARD_ENABLED=$(json_get_string "$CONFIG_FILE" "dashboard_enabled") || DASHBOARD_ENABLED="false"
+    DASHBOARD_VOICE=$(json_get_string "$CONFIG_FILE" "dashboard_voice")   || DASHBOARD_VOICE="false"
+    DASHBOARD_PORT=$(json_get_string "$CONFIG_FILE" "dashboard_port")    || DASHBOARD_PORT="2800"
+fi
+
+# =============================================================================
+# Utility Functions (daemon-local)
+# =============================================================================
+
+# Rotate log file if it gets too large
+rotate_log_if_needed() {
+    local log="$1"
+    if [[ -f "$log" ]]; then
+        local size
+        size=$(stat -f%z "$log" 2>/dev/null || stat --format=%s "$log" 2>/dev/null || echo 0)
+        if [[ "$size" -gt "$MAX_LOG_SIZE" ]]; then
+            mv "$log" "${log}.old"
+            touch "$log"
+        fi
+    fi
+}
+
+# Trim notification entries to prevent unbounded growth
+trim_notified_file() {
+    if [[ -f "$NOTIFIED_FILE" ]]; then
+        local count
+        count=$(wc -l < "$NOTIFIED_FILE" 2>/dev/null || echo 0)
+        if [[ "$count" -gt "$MAX_NOTIFY_ENTRIES" ]]; then
+            # Keep last 500 entries
+            tail -500 "$NOTIFIED_FILE" > "${NOTIFIED_FILE}.tmp"
+            mv "${NOTIFIED_FILE}.tmp" "$NOTIFIED_FILE"
+        fi
+    fi
+}
+
+# =============================================================================
+# Daemon Loop
+# =============================================================================
+
+daemon_loop() {
+    echo "[$(date -Iseconds)] Forge daemon started (PID: $$)" >> "$LOG_FILE"
+
+    # Atomic lock file acquisition (noclobber = O_EXCL, prevents TOCTOU race on systems without flock)
+    if ! (set -o noclobber; echo $$ > "$LOCK_FILE") 2>/dev/null; then
+        echo "[$(date -Iseconds)] Lock file already held by another daemon instance — exiting" >> "$LOG_FILE"
+        exit 1
+    fi
+
+    # Initialize database (will attempt sqlite3 install if missing)
+    if ! db_init; then
+        echo "[$(date -Iseconds)] FATAL: Database initialization failed (sqlite3 missing?)" >> "$LOG_FILE"
+        rm -f "$LOCK_FILE"
+        exit 1
+    fi
+    echo "[$(date -Iseconds)] Database initialized at $FORGE_DB" >> "$LOG_FILE"
+
+    # Cleanup on exit
+    trap 'rm -f "$LOCK_FILE"; echo "[$(date -Iseconds)] Daemon exiting" >> "$LOG_FILE"' EXIT
+
+    local iteration=0
+    local current_state="idle"
+    local poll_interval=2  # Start fast, adjust based on activity
+
+    while true; do
+        # Increment iteration counter
+        ((iteration++)) || true
+
+        # Sync agent status from JSON files to SQLite (with mtime filtering)
+        sync_agent_status_to_db
+
+        # Check for new tasks and notify
+        check_new_pending_tasks
+
+        # Check for workers needing attention (urgent)
+        check_attention_needed
+
+        # Check for Heimdall escalations (lab worker policy violations)
+        check_heimdall_escalations "$FORGE_ROOT/_vibe-chain-output/worker-inbox"
+
+        # Check token budget warnings for long-running agents (T2-F1)
+        check_token_budgets
+
+        # Check task dependencies (T2-H2)
+        check_task_dependencies
+
+        # Route tasks
+        route_completed_to_review
+        route_approved_to_merged
+
+        # Update state file
+        update_state
+
+        # Adaptive polling: check activity and adjust interval
+        local new_state
+        new_state=$(determine_daemon_state)
+        if [[ "$new_state" != "$current_state" ]]; then
+            current_state="$new_state"
+            db_set_daemon_state "$current_state"
+            poll_interval=$(get_poll_interval)
+            echo "[$(date -Iseconds)] State changed to $current_state, poll interval: ${poll_interval}s" >> "$LOG_FILE"
+        fi
+
+        # Periodic maintenance (every MAINTENANCE_INTERVAL iterations)
+        if [[ $((iteration % MAINTENANCE_INTERVAL)) -eq 0 ]]; then
+            rotate_log_if_needed "$LOG_FILE"
+            rotate_log_if_needed "$NOTIFY_FILE"
+            trim_notified_file
+            # Cleanup stale agent status
+            db_cleanup_stale_agents "$STALE_CLEANUP_MINUTES"
+            # Prune old history
+            db_prune_history "$HISTORY_PRUNE_DAYS"
+        fi
+
+        sleep "$poll_interval"
+    done
+}
+
+# =============================================================================
+# Watchdog
+# =============================================================================
+
+# WATCHDOG_MAX_RESTARTS -- how many times the watchdog will restart a crashed
+# daemon_loop before giving up.  Override via environment if needed.
+WATCHDOG_MAX_RESTARTS="${WATCHDOG_MAX_RESTARTS:-5}"
+
+# watchdog_loop
+# Wraps daemon_loop with automatic restart on unexpected exit.
+# Intentional stops (cmd_stop) are detected via the .stopping sentinel file,
+# which cmd_stop creates before sending SIGTERM to the watchdog PID.
+watchdog_loop() {
+    local restart_count=0
+    local backoff=5  # seconds; doubles each restart up to 60s
+
+    while [[ $restart_count -lt $WATCHDOG_MAX_RESTARTS ]]; do
+        daemon_loop
+        local exit_code=$?
+
+        # Intentional stop: cmd_stop touches .stopping before killing us
+        if [[ -f "$FORGE_ROOT/.forge/daemon.stopping" ]]; then
+            rm -f "$FORGE_ROOT/.forge/daemon.stopping"
+            echo "[$(date -Iseconds)] WATCHDOG: daemon stopped intentionally" >> "$LOG_FILE"
+            return 0
+        fi
+
+        restart_count=$((restart_count + 1))
+        echo "[$(date -Iseconds)] WATCHDOG: daemon exited unexpectedly (code $exit_code), restarting in ${backoff}s (attempt $restart_count/$WATCHDOG_MAX_RESTARTS)" >> "$LOG_FILE"
+
+        sleep "$backoff"
+        backoff=$((backoff * 2))
+        [[ $backoff -gt 60 ]] && backoff=60
+    done
+
+    echo "[$(date -Iseconds)] WATCHDOG: daemon exceeded $WATCHDOG_MAX_RESTARTS restart attempts — giving up" >> "$LOG_FILE"
+}
+
+# =============================================================================
+# Dashboard Commands
+# =============================================================================
+
+cmd_dashboard_start() {
+    if ! command -v node &>/dev/null; then
+        log_warn "Node.js not found — cannot start dashboard"
+        return 1
+    fi
+
+    # Check already running
+    if [[ -f "$DASHBOARD_PID_FILE" ]]; then
+        local pid
+        pid=$(cat "$DASHBOARD_PID_FILE")
+        if kill -0 "$pid" 2>/dev/null; then
+            echo "Dashboard already running (PID: $pid)"
+            echo "   URL: http://localhost:${DASHBOARD_PORT}"
+            return 0
+        else
+            rm -f "$DASHBOARD_PID_FILE"
+        fi
+    fi
+
+    DASHBOARD_PORT="$DASHBOARD_PORT" node "$SCRIPT_DIR/dashboard/server.js" \
+        >> "$FORGE_ROOT/.forge/dashboard.log" 2>&1 &
+    local pid=$!
+    echo "$pid" > "$DASHBOARD_PID_FILE"
+
+    log_success "Dashboard started (PID: $pid)"
+    echo "   URL:  http://localhost:${DASHBOARD_PORT}"
+    echo "   Log:  $FORGE_ROOT/.forge/dashboard.log"
+
+    # Open browser (platform-specific, best-effort)
+    case "$(uname -s)" in
+        MINGW*|MSYS*|CYGWIN*)
+            cmd.exe /c start "" "http://localhost:${DASHBOARD_PORT}" 2>/dev/null & ;;
+        Darwin)
+            open "http://localhost:${DASHBOARD_PORT}" 2>/dev/null & ;;
+        Linux)
+            xdg-open "http://localhost:${DASHBOARD_PORT}" 2>/dev/null & ;;
+    esac
+}
+
+cmd_dashboard_stop() {
+    if [[ ! -f "$DASHBOARD_PID_FILE" ]]; then
+        echo "Dashboard not running"
+        return 0
+    fi
+
+    local pid
+    pid=$(cat "$DASHBOARD_PID_FILE")
+    if kill -0 "$pid" 2>/dev/null; then
+        kill "$pid"
+        rm -f "$DASHBOARD_PID_FILE"
+        log_success "Dashboard stopped"
+    else
+        rm -f "$DASHBOARD_PID_FILE"
+        echo "Dashboard was not running (stale PID file removed)"
+    fi
+}
+
+# =============================================================================
+# Commands
+# =============================================================================
+
+cmd_start() {
+    # Create directories if needed (with secure permissions)
+    mkdir -p "$FORGE_ROOT/.forge"
+    chmod 700 "$FORGE_ROOT/.forge"
+
+    # SECURITY: Use flock for atomic lock acquisition to prevent TOCTOU race
+    # This prevents multiple daemon instances from starting simultaneously
+    local lock_fd=200
+    local startup_lock="$FORGE_ROOT/.forge/startup.lock"
+
+    # Try to acquire exclusive lock (non-blocking)
+    if command -v flock &>/dev/null; then
+        # flock available (Linux, some Git Bash installations)
+        exec 200>"$startup_lock"
+        if ! flock -n 200; then
+            echo "Another daemon startup is in progress"
+            return 1
+        fi
+        # Lock acquired - will be released when subshell exits or fd closes
+    fi
+    # If flock not available, fall back to PID-based check (less secure but functional)
+
+    # Check if already running
+    if [[ -f "$PID_FILE" ]]; then
+        local pid
+        pid=$(cat "$PID_FILE")
+        if kill -0 "$pid" 2>/dev/null; then
+            echo "Daemon already running (PID: $pid)"
+            return 0
+        else
+            # Stale PID file
+            rm -f "$PID_FILE"
+        fi
+    fi
+
+    # Check for lock file (another instance check - defense in depth)
+    if [[ -f "$LOCK_FILE" ]]; then
+        local lock_pid
+        lock_pid=$(cat "$LOCK_FILE" 2>/dev/null)
+        if kill -0 "$lock_pid" 2>/dev/null; then
+            echo "Another daemon instance is running (PID: $lock_pid)"
+            return 1
+        else
+            rm -f "$LOCK_FILE"
+        fi
+    fi
+
+    mkdir -p "$FORGE_ROOT/$TASKS_PENDING"
+    mkdir -p "$FORGE_ROOT/$TASKS_IN_PROGRESS"
+    mkdir -p "$FORGE_ROOT/$TASKS_COMPLETED"
+    mkdir -p "$FORGE_ROOT/$TASKS_REVIEW"
+    mkdir -p "$FORGE_ROOT/$TASKS_APPROVED"
+    mkdir -p "$FORGE_ROOT/$TASKS_NEEDS_CHANGES"
+    mkdir -p "$FORGE_ROOT/$TASKS_MERGED"
+    mkdir -p "$FORGE_ROOT/$TASKS_ATTENTION"
+    mkdir -p "$FORGE_ROOT/$AGENT_STATUS_DIR"
+
+    # Start daemon under watchdog (auto-restarts on unexpected exit)
+    watchdog_loop &
+    local pid=$!
+    echo "$pid" > "$PID_FILE"
+
+    log_success "Forge daemon started (PID: $pid)"
+    echo "   Log: $LOG_FILE"
+
+    # Auto-launch dashboard if voice or dashboard is enabled in config
+    if [[ "$DASHBOARD_VOICE" == "true" ]] || [[ "$DASHBOARD_ENABLED" == "true" ]]; then
+        cmd_dashboard_start
+    fi
+
+    # Note: flock is automatically released when the fd is closed (script exits)
+}
+
+cmd_stop() {
+    if [[ ! -f "$PID_FILE" ]]; then
+        echo "Daemon not running"
+        return 0
+    fi
+
+    local pid
+    pid=$(cat "$PID_FILE")
+
+    if kill -0 "$pid" 2>/dev/null; then
+        # Signal the watchdog that this is an intentional stop before killing it
+        touch "$FORGE_ROOT/.forge/daemon.stopping"
+        kill "$pid"
+        rm -f "$PID_FILE" "$LOCK_FILE"
+        echo "[$(date -Iseconds)] Forge daemon stopped" >> "$LOG_FILE"
+        log_success "Daemon stopped"
+    else
+        rm -f "$PID_FILE" "$LOCK_FILE"
+        echo "Daemon was not running (stale PID file removed)"
+    fi
+
+    # Stop dashboard if running
+    cmd_dashboard_stop
+
+    # Update state to show inactive
+    if [[ -f "$STATE_FILE" ]]; then
+        # Use cross-platform sed helper
+        sed_inplace 's/status: active/status: stopped/' "$STATE_FILE" 2>/dev/null || true
+    fi
+}
+
+cmd_status() {
+    echo ""
+    log_header "🔥 Forge Daemon Status"
+
+    display_daemon_status
+    echo ""
+
+    display_task_counts
+    echo ""
+
+    display_attention_needed
+    display_worker_status
+    display_recent_notifications
+
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+}
+
+cmd_notifications() {
+    echo ""
+    log_header "🔔 Forge Notifications"
+
+    if [[ -f "$NOTIFY_FILE" ]]; then
+        local count="${1:-10}"
+        echo "Last $count notifications:"
+        echo ""
+        tail -"$count" "$NOTIFY_FILE"
+    else
+        echo "No notifications yet."
+    fi
+
+    echo ""
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+}
+
+cmd_clear_notifications() {
+    rm -f "$NOTIFY_FILE" "$NOTIFIED_FILE"
+    log_success "Notifications cleared"
+}
+
+# =============================================================================
+# Main
+# =============================================================================
+
+main() {
+    local command="${1:-status}"
+
+    case "$command" in
+        "start")
+            cmd_start
+            ;;
+        "stop")
+            cmd_stop
+            ;;
+        "status")
+            cmd_status
+            ;;
+        "notifications"|"notify")
+            shift
+            cmd_notifications "$@"
+            ;;
+        "clear")
+            cmd_clear_notifications
+            ;;
+        "dashboard")
+            shift
+            case "${1:-start}" in
+                "start") cmd_dashboard_start ;;
+                "stop")  cmd_dashboard_stop ;;
+                *)
+                    echo "Usage: forge-daemon.sh dashboard [start|stop]"
+                    exit $EXIT_INVALID_ARGUMENT
+                    ;;
+            esac
+            ;;
+        *)
+            echo "Usage: forge-daemon.sh [start|stop|status|notifications|clear|dashboard]"
+            exit $EXIT_INVALID_ARGUMENT
+            ;;
+    esac
+}
+
+main "$@"