vibe-forge 0.3.12 → 0.8.1

package/bin/lib/daemon/sync.sh

@@ -0,0 +1,103 @@
+#!/usr/bin/env bash
+#
+# bin/lib/daemon/sync.sh
+#
+# Agent status synchronization functions - JSON files to SQLite
+#
+# Dependencies: json.sh, database.sh, constants.sh
+# Globals required: FORGE_ROOT, FORGE_DB, AGENT_STATUS_DIR, LOG_FILE,
+#                   STALE_STATUS_THRESHOLD
+
+# Prevent double-sourcing
+[[ -n "${_DAEMON_SYNC_LOADED:-}" ]] && return 0
+_DAEMON_SYNC_LOADED=1
+
+# 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=$(json_read "$status_file" "agent" "unknown")
+                status=$(json_read "$status_file" "status" "unknown")
+                task=$(json_read "$status_file" "task" "")
+                message=$(json_read "$status_file" "message" "" | head -c 80)
+                updated=$(json_read "$status_file" "updated" "")
+
+                # Check if status actually changed (T1-D3: record transitions)
+                local old_status
+                old_status=$(sqlite3 "$FORGE_DB" \
+                    "SELECT status FROM agent_status WHERE agent = '$(db_escape "$agent")';" 2>/dev/null || echo "")
+                if [[ "$old_status" != "$status" && -n "$status" ]]; then
+                    db_record_status_history "$agent" "$status" "$task"
+                fi
+
+                # 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=$STALE_STATUS_THRESHOLD
+
+    # 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)
+}