spectra

#!/usr/bin/env bash
################################################################################
# SPECTRA - Unified CLI, Bootstrap & Archiving Launcher
################################################################################
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
LAUNCH_SCRIPT="$SCRIPT_DIR/scripts/launch/spectra-launch.sh"
INSTALL_SCRIPT="$SCRIPT_DIR/scripts/install/install-spectra.sh"
REPAIR_SCRIPT="$SCRIPT_DIR/scripts/install/repair-installation.sh"
SETUP_SCRIPT="$SCRIPT_DIR/scripts/setup/setup_env.sh"

# ANSI Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
BOLD='\033[1m'
NC='\033[0m'

print_status()  { echo -e "${BLUE}▶${NC} $1"; }
print_success() { echo -e "${GREEN}✓${NC} $1"; }
print_error()   { echo -e "${RED}✗${NC} $1" >&2; }
print_warning() { echo -e "${YELLOW}⚠${NC} $1"; }

print_banner() {
    echo -e "${CYAN}"
    cat << 'EOF'
╔═══════════════════════════════════════════════════════════════════════════╗
║                          ███████╗██████╗ ███████╗ ██████╗████████╗██████╗  ║
║                          ██╔════╝██╔══██╗██╔════╝██╔════╝╚══██╔══╝██╔══██╗ ║
║                          ███████╗██████╔╝█████╗  ██║        ██║   ██████╔╗ ║
║                          ╚════██║██╔═══╝ ██╔══╝  ██║        ██║   ██╔══██║ ║
║                          ███████║██║     ███████╗╚██████╗   ██║   ██║  ██║ ║
║                          ╚══════╝╚═╝     ╚══════╝ ╚═════╝   ╚═╝   ╚═╝  ╚═╝ ║
╚═══════════════════════════════════════════════════════════════════════════╝
              Telegram Network Discovery & Archiving System [CLI]
EOF
    echo -e "${NC}"
}

# Resolve Python environment
resolve_python() {
    if [[ -n "${VIRTUAL_ENV:-}" && -x "${VIRTUAL_ENV}/bin/python" ]]; then
        echo "${VIRTUAL_ENV}/bin/python"
    elif [[ -x "$SCRIPT_DIR/.venv/bin/python" ]]; then
        echo "$SCRIPT_DIR/.venv/bin/python"
    elif [[ -x "$SCRIPT_DIR/spectra_venv/bin/python" ]]; then
        echo "$SCRIPT_DIR/spectra_venv/bin/python"
    elif command -v python3 >/dev/null 2>&1; then
        echo "python3"
    else
        print_error "Python 3 is required but not found in PATH."
        exit 1
    fi
}

is_configured() {
    [[ -f "$SCRIPT_DIR/data/config/spectra_config.json" || -f "$SCRIPT_DIR/spectra_config.json" ]]
}

run_bootstrap() {
    print_banner
    print_status "Starting SPECTRA bootstrap & environment setup..."
    if [[ -x "$LAUNCH_SCRIPT" ]]; then
        exec "$LAUNCH_SCRIPT"
    elif [[ -f "$SETUP_SCRIPT" ]]; then
        bash "$SETUP_SCRIPT"
    else
        local PY_BIN
        PY_BIN="$(resolve_python)"
        print_status "Installing SPECTRA package in editable mode..."
        "$PY_BIN" -m pip install -e "$SCRIPT_DIR"
        print_success "SPECTRA installation complete."
    fi
}

show_help() {
    cat << 'EOF'
SPECTRA - Telegram Network Discovery & Archiving System [CLI-First]

Usage: ./spectra [COMMAND] [OPTIONS]

Core Operational Commands:
  discover            Run CaaS-aware semantic discovery (--seed @channel)
  profile             Extract dossiers, pricing, and aliases (--profile @channel)
  process-queue       Process the automated worker intelligence queue
  archive             Archive messages and media from Telegram channels
  semantic-tui        Launch the interactive Semantic Intelligence TUI
  tui                 Launch the legacy full-terminal interface

Environment & Management:
  bootstrap           Auto-setup environment, dependencies, and venv
  install             Install/update dependencies and SPECTRA package
  repair              Repair virtualenv or missing packages
  help, -h, --help    Show this help message

Examples:
  ./spectra                                # Launch TUI (or auto-bootstrap if unconfigured)
  ./spectra discover --seed @target        # Discover connected Telegram channels
  ./spectra profile --target @target       # Extract threat actor profile & economics
  ./spectra process-queue --batch-size 250 # Process batch intelligence queue
  ./spectra bootstrap                      # Install dependencies & configure
EOF
}

# Main Dispatcher
COMMAND="${1:-}"

case "$COMMAND" in
    ""|"tui"|"launch"|"run")
        if is_configured; then
            PY="$(resolve_python)"
            exec "$PY" -m tgarchive.ui.tui_caas "$@"
        else
            run_bootstrap
        fi
        ;;
    "bootstrap"|"setup")
        run_bootstrap
        ;;
    "install")
        if [[ -f "$INSTALL_SCRIPT" ]]; then
            exec bash "$INSTALL_SCRIPT"
        else
            run_bootstrap
        fi
        ;;
    "repair")
        if [[ -f "$REPAIR_SCRIPT" ]]; then
            exec bash "$REPAIR_SCRIPT"
        else
            run_bootstrap
        fi
        ;;
    "semantic-tui")
        shift || true
        PY="$(resolve_python)"
        exec "$PY" -m tgarchive.ui.tui_caas "$@"
        ;;
    "discover"|"semantic-discovery")
        shift || true
        PY="$(resolve_python)"
        exec "$PY" -m tgarchive.osint.caas.discovery_ops "$@"
        ;;
    "profile")
        shift || true
        PY="$(resolve_python)"
        exec "$PY" -m tgarchive.osint.caas.cli profile "$@"
        ;;
    "process-queue")
        shift || true
        PY="$(resolve_python)"
        exec "$PY" -m tgarchive.osint.caas.cli process-queue "$@"
        ;;
    "help"|"-h"|"--help")
        show_help
        ;;
    *)
        # Forward any other subcommand directly to tgarchive CLI
        PY="$(resolve_python)"
        exec "$PY" -m tgarchive "$@"
        ;;
esac