#!/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; }
# Resolve symlinks to find the actual repository location
SOURCE="${BASH_SOURCE[0]}"
while [ -h "$SOURCE" ]; do
    DIR="$( cd -P "$( dirname "$SOURCE" )" && pwd )"
    SOURCE="$(readlink "$SOURCE")"
    [[ $SOURCE != /* ]] && SOURCE="$DIR/$SOURCE"
done
SCRIPT_DIR="$( cd -P "$( dirname "$SOURCE" )" && pwd )"

# --- 1. BOOTSTRAP ---
# Only bootstrap if we are not in a git repo and the target hidden folder does not exist
if [ ! -d "$SCRIPT_DIR/.git" ] && [ ! -d "$HOME/.opencortex" ] && [[ ! "$(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}"
    echo -e "${YELLOW}--- Installing System Dependencies ---${NC}"
    if command_exists apt-get; then
        sudo apt-get update && sudo apt-get install -y sbcl emacs-nox rlwrap netcat-openbsd curl git socat libssl-dev libncurses5-dev libffi-dev zlib1g-dev libsqlite3-dev
    fi
    if [ ! -d "$HOME/quicklisp" ]; then
        curl -O https://beta.quicklisp.org/quicklisp.lisp
        sbcl --non-interactive --load quicklisp.lisp --eval "(quicklisp-quickstart:install)" --eval "(ql-util:without-prompting (ql:add-to-init-file))"
        rm quicklisp.lisp
    fi
    cd "$SCRIPT_DIR"
    if [ ! -f .env ]; then
        cp .env.example .env
        
        echo -e "\n${YELLOW}--- Identity Configuration ---${NC}"
        echo "Let's personalize your OpenCortex experience."
        read -p "Your Name [User]: " user_name < /dev/tty
        user_name=${user_name:-User}
        sed -i "s|MEMEX_USER=.*|MEMEX_USER=\"$user_name\"|" .env
        
        read -p "Agent Name [OpenCortex]: " agent_name < /dev/tty
        agent_name=${agent_name:-OpenCortex}
        sed -i "s|MEMEX_ASSISTANT=.*|MEMEX_ASSISTANT=\"$agent_name\"|" .env
        
        echo -e "\n${YELLOW}--- LLM Configuration ---${NC}"
        echo "You can enter your LLM API keys now, or press Enter to skip and configure them later."
        read -p "Gemini API Key: " gemini_key < /dev/tty
        [ -n "$gemini_key" ] && sed -i "s|GEMINI_API_KEY=.*|GEMINI_API_KEY=\"$gemini_key\"|" .env
        read -p "Anthropic API Key: " anthropic_key < /dev/tty
        [ -n "$anthropic_key" ] && sed -i "s|ANTHROPIC_API_KEY=.*|ANTHROPIC_API_KEY=\"$anthropic_key\"|" .env
        read -p "OpenAI API Key: " openai_key < /dev/tty
        [ -n "$openai_key" ] && sed -i "s|OPENAI_API_KEY=.*|OPENAI_API_KEY=\"$openai_key\"|" .env
        read -p "OpenRouter API Key: " openrouter_key < /dev/tty
        [ -n "$openrouter_key" ] && sed -i "s|OPENROUTER_API_KEY=.*|OPENROUTER_API_KEY=\"$openrouter_key\"|" .env

        echo -e "\n${YELLOW}--- Memex Folder Structure ---${NC}"
        echo "Enter the absolute paths for your existing folder structure (press Enter to accept default)."
        read -p "Memex Root [$HOME/memex]: " memex_dir < /dev/tty
        memex_dir=${memex_dir:-$HOME/memex}
        sed -i "s|MEMEX_DIR=.*|MEMEX_DIR=\"$memex_dir\"|" .env
        sed -i "s|\"/memex/|\"$memex_dir/|g" .env
        sed -i "s|SKILLS_DIR=.*|SKILLS_DIR=\"$SCRIPT_DIR/skills\"|" .env
        sed -i "s|ZETTELKASTEN_DIR=.*|ZETTELKASTEN_DIR=\"$memex_dir/notes\"|" .env

        
        read -p "Inbox Directory [$memex_dir/inbox]: " inbox_dir < /dev/tty
        inbox_dir=${inbox_dir:-$memex_dir/inbox}
        sed -i "s|INBOX_DIR=.*|INBOX_DIR=\"$inbox_dir\"|" .env
        
        read -p "Daily Directory [$memex_dir/daily]: " daily_dir < /dev/tty
        daily_dir=${daily_dir:-$memex_dir/daily}
        sed -i "s|DAILY_DIR=.*|DAILY_DIR=\"$daily_dir\"|" .env
        
        read -p "Projects Directory [$memex_dir/projects]: " proj_dir < /dev/tty
        proj_dir=${proj_dir:-$memex_dir/projects}
        sed -i "s|PROJECTS_DIR=.*|PROJECTS_DIR=\"$proj_dir\"|" .env
    fi
        # Ensure the directories actually exist
        mkdir -p "$memex_dir"
        mkdir -p "$inbox_dir"
        mkdir -p "$daily_dir"
        mkdir -p "$proj_dir"
        mkdir -p "$memex_dir/notes"
        mkdir -p "$memex_dir/areas"
        mkdir -p "$memex_dir/resources"
        mkdir -p "$memex_dir/archives"
        mkdir -p "$memex_dir/system"

    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 "${YELLOW}--- Compiling and Loading OpenCortex (this may take a minute) ---${NC}"
    sbcl --non-interactive \
         --eval "(load (merge-pathnames \"quicklisp/setup.lisp\" (user-homedir-pathname)))" \
         --eval "(push (truename \"$SCRIPT_DIR/\") asdf:*central-registry*)" \
         --eval "(ql:quickload :opencortex)"
    
    if [ $? -ne 0 ]; then
        echo -e "${RED}✗ Compilation or Loading failed.${NC}"
        exit 1
    fi

    echo -e "${YELLOW}--- Finalizing: Awakening the Brain as a background daemon ---${NC}"
    # Nuke any existing brain logs
    > "$SCRIPT_DIR/brain.log"
    "$SCRIPT_DIR/opencortex.sh" --boot > "$SCRIPT_DIR/brain.log" 2>&1 &
    
    local success=false
    for i in {1..30}; do
        if nc -z localhost $PORT 2>/dev/null; then
            success=true
            break
        fi
        sleep 2
        echo -n "."
    done
    
    if [ "$success" = true ]; then
        echo -e "\n${GREEN}✓ Brain is alive and responsive on port $PORT.${NC}"
        echo -e "${GREEN}✓ Setup complete. You can now run 'opencortex tui'.${NC}"
    else
        echo -e "\n${RED}✗ Brain failed to wake up.${NC}"
        echo -e "${YELLOW}Full Log Path: $(realpath "$SCRIPT_DIR/brain.log")${NC}"
        echo -e "${YELLOW}--- LOG START ---${NC}"
        cat "$SCRIPT_DIR/brain.log"
        echo -e "${YELLOW}--- LOG END ---${NC}"
        # Kill the background process if it exists
        pkill -f "sbcl.*opencortex" || true
        exit 1
    fi
}

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 (merge-pathnames \"quicklisp/setup.lisp\" (user-homedir-pathname)))" \
         --eval "(setf *debugger-hook* (lambda (c h) (declare (ignore h)) (format *error-output* \"FATAL LISP ERROR: ~a~%\" c) (uiop:print-backtrace :stream *error-output*) (uiop:quit 1)))" \
         --eval "(push (truename \"$SCRIPT_DIR/\") asdf:*central-registry*)" \
         --eval "(format t \"--- Quickloading OpenCortex ---~%\")" \
         --eval "(ql:quickload :opencortex)" \
         --eval "(opencortex:main)"
fi

# --- 4. INTERACT ---
if [[ "$1" == "tui" ]]; then
    # Ensure daemon is running
    if ! (nc -z $HOST $PORT 2>/dev/null || (command_exists socat && socat - TCP:$HOST:$PORT,connect-timeout=1 2>/dev/null)); then
        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 nc -z $HOST $PORT 2>/dev/null || (command_exists socat && socat - TCP:$HOST:$PORT,connect-timeout=1 2>/dev/null); then break; fi
            echo -n "."
        done
        echo ""
    fi
    
    # Launch TUI
    echo -e "${BLUE}Launching Croatoan TUI...${NC}"
    exec sbcl --non-interactive \
         --load ~/quicklisp/setup.lisp \
         --eval "(push (truename \"$SCRIPT_DIR/\") asdf:*central-registry*)" \
         --eval "(ql:quickload :opencortex/tui :silent t)" \
         --eval "(opencortex.tui:main)"
fi

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