#!/bin/bash
# Setup agents for NODE2 - update node_id for agents that should be on NODE2

set -e

echo "🔧 Setting up agents for NODE2..."

# List of agent IDs that should be on NODE2
# You can modify this list based on your needs
NODE2_AGENTS=(
    "ag_atlas"
    "ag_builder"
    "ag_greeter"
    "ag_oracle"
)

NODE_ID="node-2-macbook-m4max"

# Update node_id for each agent
for agent_id in "${NODE2_AGENTS[@]}"; do
    echo "  Updating $agent_id to node_id=$NODE_ID..."
    ssh root@144.76.224.179 "docker exec daarion-postgres psql -U postgres -d daarion -c \"UPDATE agents SET node_id = '$NODE_ID' WHERE id = '$agent_id' AND (node_id IS NULL OR node_id = '');\""
done

echo "✅ Agents updated for NODE2"
echo ""
echo "Verifying..."
ssh root@144.76.224.179 "docker exec daarion-postgres psql -U postgres -d daarion -c \"SELECT id, display_name, node_id, status FROM agents WHERE node_id = '$NODE_ID' ORDER BY display_name;\""