microdao-daarion/services/city-service/migrations.py

import asyncpg
import os
import logging

logger = logging.getLogger(__name__)

DATABASE_URL = os.getenv("DATABASE_URL", "postgresql://postgres:postgres@localhost:5432/daarion")

async def run_migrations():
    conn = None
    try:
        conn = await asyncpg.connect(DATABASE_URL)
        
        # Add logo_url and banner_url to microdaos table (previous task)
        await conn.execute("""
            ALTER TABLE microdaos
            ADD COLUMN IF NOT EXISTS logo_url TEXT,
            ADD COLUMN IF NOT EXISTS banner_url TEXT;
        """)
        
        # Add logo_url and banner_url to city_rooms table (previous task)
        await conn.execute("""
            ALTER TABLE city_rooms
            ADD COLUMN IF NOT EXISTS logo_url TEXT,
            ADD COLUMN IF NOT EXISTS banner_url TEXT;
        """)

        # NEW: Add crew_team_key to agents table (TASK 044)
        await conn.execute("""
            ALTER TABLE agents
            ADD COLUMN IF NOT EXISTS crew_team_key TEXT;
        """)
        logger.info("Migration: Added crew_team_key to agents table.")

        # TASK 044: Add room_role, is_public, sort_order to city_rooms table
        await conn.execute("""
            ALTER TABLE city_rooms
            ADD COLUMN IF NOT EXISTS room_role TEXT,
            ADD COLUMN IF NOT EXISTS is_public BOOLEAN DEFAULT TRUE,
            ADD COLUMN IF NOT EXISTS sort_order INTEGER DEFAULT 100;
        """)
        logger.info("Migration: Added room_role, is_public, sort_order to city_rooms table.")

        # TASK C2: Create nodes table with owner_microdao (replaces node_cache for ontology)
        await conn.execute("""
            CREATE TABLE IF NOT EXISTS nodes (
                id TEXT PRIMARY KEY,
                display_name TEXT NOT NULL,
                hostname TEXT,
                owner_microdao_id TEXT REFERENCES microdaos(id),
                node_type TEXT,
                environment TEXT DEFAULT 'unknown',
                cpu_cores INTEGER,
                ram_gb INTEGER,
                gpu_count INTEGER DEFAULT 0,
                disk_gb INTEGER,
                status TEXT DEFAULT 'unknown',
                guardian_agent_id TEXT,
                steward_agent_id TEXT,
                created_at TIMESTAMPTZ DEFAULT NOW(),
                updated_at TIMESTAMPTZ DEFAULT NOW()
            );
        """)
        logger.info("Migration: Created nodes table with owner_microdao.")

        # TASK C3: Insert NODE1 and NODE2 if not exist
        await conn.execute("""
            INSERT INTO nodes (id, display_name, hostname, owner_microdao_id, node_type, environment, status, guardian_agent_id, steward_agent_id)
            VALUES 
                ('node-1-hetzner-gex44', 'Hetzner GEX44 Production', '144.76.224.179', 'dao_daarion', 'compute', 'production', 'online', 'monitor-node1', 'node-steward-node1'),
                ('node-2-macbook-m4max', 'MacBook Pro M4 Max', '192.168.1.33', 'dao_daarion', 'hybrid', 'development', 'online', 'monitor-node2', 'node-steward-node2')
            ON CONFLICT (id) DO UPDATE SET
                display_name = EXCLUDED.display_name,
                hostname = EXCLUDED.hostname,
                owner_microdao_id = EXCLUDED.owner_microdao_id,
                node_type = EXCLUDED.node_type,
                environment = EXCLUDED.environment,
                guardian_agent_id = EXCLUDED.guardian_agent_id,
                steward_agent_id = EXCLUDED.steward_agent_id,
                updated_at = NOW();
        """)
        logger.info("Migration: Inserted/updated NODE1 and NODE2 in nodes table.")

    except Exception as e:
        logger.error(f"Error running migrations: {e}")
        raise
    finally:
        if conn:
            await conn.close()