#!/bin/bash
set -e

PORT=9105
HOST=${1:-localhost}
RED='\033[0;31m'; GREEN='\033[0;32m'; BLUE='\033[0;34m'; YELLOW='\033[0;33m'; NC='\033[0m'

command_exists() { command -v "$1" >/dev/null 2>&1; }
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"

# --- 1. BOOTSTRAP ---
if [ ! -d "$SCRIPT_DIR/.git" ] && [[ ! "$(pwd)" =~ "opencortex" ]]; then
    echo -e "${BLUE}=== OpenCortex: Zero-to-One Bootstrapper ===${NC}"
    git clone http://10.10.10.201:3001/amr/opencortex.git opencortex
    cd opencortex && git submodule update --init --recursive
    exec ./opencortex.sh "$@"
fi

# --- 2. SETUP ---
setup_system() {
    echo -e "${BLUE}=== OpenCortex: Initializing System ===${NC}"
    cd "$SCRIPT_DIR"
    [ ! -f .env ] && cp .env.example .env
    mkdir -p src
    for f in literate/*.org; do
        emacs --batch --eval "(require 'org)" --eval "(org-babel-tangle-file \"$f\")" >/dev/null 2>&1 || true
    done
    mkdir -p "$HOME/.local/bin"
    ln -sf "$SCRIPT_DIR/opencortex.sh" "$HOME/.local/bin/opencortex"
    echo -e "${GREEN}✓ Setup complete.${NC}"
}

if [ ! -f "$SCRIPT_DIR/src/package.lisp" ] || [ ! -f "$SCRIPT_DIR/.env" ]; then
    setup_system
fi

# --- 3. BOOT ---
if [[ "$1" == "--boot" ]]; then
    if [ -f "$SCRIPT_DIR/.env" ]; then
        while IFS='=' read -r key value || [ -n "$key" ]; do
          if [[ $key =~ ^[a-zA-Z_][a-zA-Z0-9_]*$ ]]; then
            val=$(echo "$value" | sed 's/^"//;s/"$//')
            export "$key=$val"
          fi
        done < "$SCRIPT_DIR/.env"
    fi
    exec sbcl --non-interactive \
         --eval "(load \"~/quicklisp/setup.lisp\")" \
         --eval "(push \"$SCRIPT_DIR/\" asdf:*central-registry*)" \
         --eval "(ql:quickload :opencortex)" \
         --eval "(opencortex:main)"
fi

# --- 4. INTERACT ---
connect() {
    if command_exists socat && socat - TCP:$HOST:$PORT,connect-timeout=1 2>/dev/null; then
        socat - TCP:$HOST:$PORT
        return 0
    elif command_exists nc && nc -z $HOST $PORT 2>/dev/null; then
        nc $HOST $PORT
        return 0
    fi
    return 1
}

if connect; then exit 0; fi

echo -e "${YELLOW}Brain is offline. Awakening...${NC}"
"$SCRIPT_DIR/opencortex.sh" --boot > "$SCRIPT_DIR/brain.log" 2>&1 &

for i in {1..15}; do
    sleep 2
    if connect; then exit 0; fi
    echo -n "."
done

echo -e "${RED}\n✗ Failed to connect to brain.${NC}"
exit 1