#!/bin/bash
# data-platform startup check hook
# Checks for common issues at session start
# All output MUST have [data-platform] prefix

PREFIX="[data-platform]"

# Check if MCP venv exists - check cache first, then local
CACHE_VENV="$HOME/.cache/claude-mcp-venvs/leo-claude-mktplace/data-platform/.venv/bin/python"
PLUGIN_ROOT="${CLAUDE_PLUGIN_ROOT:-$(dirname "$(dirname "$(realpath "$0")")")}"
MARKETPLACE_ROOT="$(dirname "$(dirname "$PLUGIN_ROOT")")"
LOCAL_VENV="$MARKETPLACE_ROOT/mcp-servers/data-platform/.venv/bin/python"

# Check cache first (preferred), then local symlink
CACHE_VENV_DIR="$HOME/.cache/claude-mcp-venvs/leo-claude-mktplace/data-platform/.venv"
LOCAL_VENV_DIR="$MARKETPLACE_ROOT/mcp-servers/data-platform/.venv"

if [[ -f "$CACHE_VENV" ]]; then
    VENV_PATH="$CACHE_VENV"
    # Auto-create symlink in installed marketplace if missing
    if [[ ! -e "$LOCAL_VENV_DIR" && -d "$CACHE_VENV_DIR" ]]; then
        mkdir -p "$(dirname "$LOCAL_VENV_DIR")" 2>/dev/null
        ln -sf "$CACHE_VENV_DIR" "$LOCAL_VENV_DIR" 2>/dev/null
    fi
elif [[ -f "$LOCAL_VENV" ]]; then
    VENV_PATH="$LOCAL_VENV"
else
    echo "$PREFIX MCP venv missing - run /initial-setup or setup.sh"
    exit 0
fi

# Check PostgreSQL configuration (optional - just warn if configured but failing)
POSTGRES_CONFIG="$HOME/.config/claude/postgres.env"
if [[ -f "$POSTGRES_CONFIG" ]]; then
    source "$POSTGRES_CONFIG"
    if [[ -n "${POSTGRES_URL:-}" ]]; then
        # Quick connection test (5 second timeout)
        RESULT=$("$VENV_PATH" -c "
import asyncio
import sys
async def test():
    try:
        import asyncpg
        conn = await asyncpg.connect('$POSTGRES_URL', timeout=5)
        await conn.close()
        return 'OK'
    except Exception as e:
        return f'FAIL: {e}'
print(asyncio.run(test()))
" 2>/dev/null || echo "FAIL: asyncpg not installed")

        if [[ "$RESULT" == "OK" ]]; then
            # PostgreSQL OK - say nothing
            :
        elif [[ "$RESULT" == *"FAIL"* ]]; then
            echo "$PREFIX PostgreSQL connection failed - check POSTGRES_URL"
        fi
    fi
fi

# Check dbt project (if in a project with dbt_project.yml)
if [[ -f "dbt_project.yml" ]] || [[ -f "transform/dbt_project.yml" ]]; then
    if ! command -v dbt &> /dev/null; then
        echo "$PREFIX dbt CLI not found - dbt tools unavailable"
    fi
fi

# All checks passed - say nothing
exit 0