setup.sh

#!/bin/bash
# =============================================================================
# setup.sh - Master orchestrator for workstation-setup
# =============================================================================
#
# Usage:
#   ./setup.sh              # Full installation (all components)
#   ./setup.sh --terminal   # Terminal setup only
#   ./setup.sh --skip-godot # Full installation EXCEPT Godot
#   ./setup.sh --help
#
# Note: This script overwrites existing configs without backup.
# The repo (assets/) is the single source of truth.
#
# =============================================================================

set -e

# Resolve paths
SETUP_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
SCRIPTS_DIR="$SETUP_DIR/scripts"

# Source common utilities
source "$SCRIPTS_DIR/common.sh"

# =============================================================================
# Configuration & Defaults
# =============================================================================

# Standard components (Run by default)
INSTALL_TERMINAL=true
INSTALL_AGENT=true
INSTALL_QDRANT=true
INSTALL_GODOT=true
INSTALL_APPS=true
INSTALL_OBS=true
INSTALL_EASYEFFECTS=true
INSTALL_DNS=true
INSTALL_VSCODE=true
INSTALL_REMOTE=true

# Optional components (Do not run by default)
INSTALL_ONEDRIVE=false
INSTALL_VIETNAMESE=false

# =============================================================================
# Argument Parsing
# =============================================================================

show_help() {
    echo "Terminal Custom Setup Script"
    echo ""
    echo "Usage: $0 [OPTIONS]"
    echo ""
    echo "Modes:"
    echo "  Default (no args)   Run full setup (all components)."
    echo "  Exclusive           If any component flag is used, ONLY that component runs."
    echo ""
    echo "Component Flags (Triggers Exclusive Mode):"
    echo "  --full              Run full setup (same as default)"
    echo "  --terminal          Terminal setup (shell, dotfiles, fonts, power tools)"
    echo "  --agent             Agent system setup (Agent Global Hub, Claude integration) ⭐ NEW"
    echo "  --qdrant            Qdrant setup"
    echo "  --godot             Godot setup"
    echo "  --apps              Additional apps (Chrome, Dropbox, Flatpaks)"
    echo "  --obs               OBS Studio setup"
    echo "  --easyeffects       EasyEffects audio setup"
    echo "  --dns               DNS setup (Cloudflare Block Malware)"
    echo "  --onedrive          OneDrive setup (supports multiple accounts)"
    echo "  --vietnamese        Vietnamese input setup (fcitx5-unikey)"

    echo "  --vscode            Visual Studio Code setup"
    echo "  --remote            Remote access setup (Tailscale, Mosh, SSH, WoL)"

    echo ""
    echo "Skip Flags (For Default Mode):"
    echo "  --skip-terminal     Skip terminal setup"
    echo "  --skip-agent        Skip agent system setup"
    echo "  --skip-qdrant       Skip Qdrant"
    echo "  --skip-godot        Skip Godot"
    echo "  --skip-apps         Skip Apps"
    echo "  --skip-obs          Skip OBS Studio"
    echo "  --skip-easyeffects  Skip EasyEffects"
    echo "  --skip-dns          Skip DNS setup"

    echo "  --skip-vscode       Skip VS Code setup"

    echo ""
    echo "Examples:"
    echo "  $0                  # Full installation"
    echo "  $0 --terminal       # ONLY run terminal setup"
    echo "  $0 --skip-godot     # Full setup EXCEPT Godot"
    echo ""
    echo "Note: Running setup will overwrite existing configs (no backup)."
    echo "The repo (assets/) is the single source of truth."
    exit 0
}

# Detect Exclusive Mode
# If any positive component flag is present, switch to exclusive mode.
EXCLUSIVE_MODE=false
for arg in "$@"; do
    case $arg in
        --terminal|--agent|--qdrant|--godot|--apps|--obs|--easyeffects|--dns|--onedrive|--vietnamese|--vscode|--remote)
            EXCLUSIVE_MODE=true
            break
            ;;
    esac
done

if $EXCLUSIVE_MODE; then
    # In exclusive mode, disable all standard components by default.
    # Only explicitly requested components will be enabled in the loop below.
    INSTALL_TERMINAL=false
    INSTALL_AGENT=false
    INSTALL_QDRANT=false
    INSTALL_GODOT=false
    INSTALL_APPS=false
    INSTALL_OBS=false
    INSTALL_EASYEFFECTS=false
    INSTALL_DNS=false
    INSTALL_ONEDRIVE=false
    INSTALL_VIETNAMESE=false

    INSTALL_VSCODE=false
    INSTALL_REMOTE=false

fi

# Parse Flags
for arg in "$@"; do
    case $arg in
        # Component Flags
        --full)
            INSTALL_TERMINAL=true
            INSTALL_AGENT=true
            INSTALL_QDRANT=true
            INSTALL_GODOT=true
            INSTALL_APPS=true
            INSTALL_OBS=true
            INSTALL_EASYEFFECTS=true
            INSTALL_DNS=true
            INSTALL_ONEDRIVE=true
            INSTALL_VIETNAMESE=true
            INSTALL_VSCODE=true
            INSTALL_REMOTE=true

            ;;
        --terminal)           INSTALL_TERMINAL=true ;;
        --agent)              INSTALL_AGENT=true ;;
        --qdrant)             INSTALL_QDRANT=true ;;
        --godot)              INSTALL_GODOT=true ;;
        --apps)               INSTALL_APPS=true ;;
        --obs)                INSTALL_OBS=true ;;
        --easyeffects)        INSTALL_EASYEFFECTS=true ;;
        --dns)                INSTALL_DNS=true ;;
        --onedrive)           INSTALL_ONEDRIVE=true ;;
        --vietnamese)         INSTALL_VIETNAMESE=true ;;
        --vscode)             INSTALL_VSCODE=true ;;
        --remote)             INSTALL_REMOTE=true ;;

        # Skip Flags
        --skip-terminal)      INSTALL_TERMINAL=false ;;
        --skip-agent)         INSTALL_AGENT=false ;;
        --skip-qdrant)        INSTALL_QDRANT=false ;;
        --skip-godot)         INSTALL_GODOT=false ;;
        --skip-apps)          INSTALL_APPS=false ;;
        --skip-obs)           INSTALL_OBS=false ;;
        --skip-easyeffects)   INSTALL_EASYEFFECTS=false ;;
        --skip-dns)           INSTALL_DNS=false ;;
        --skip-vscode)        INSTALL_VSCODE=false ;;
        --skip-remote)        INSTALL_REMOTE=false ;;

        # Other
        --help|-h)            show_help ;;
        *)
            log_error "Unknown option: $arg"
            echo "Use --help for usage information."
            exit 1
            ;;
    esac
done

# =============================================================================
# Pre-flight Checks
# =============================================================================
check_not_root
verify_backup_dir

echo ""
echo -e "${GREEN}${BOLD}╔══════════════════════════════════════════════════════════════╗${NC}"
echo -e "${GREEN}${BOLD}║          Terminal Custom Setup for Nobara/Fedora            ║${NC}"
echo -e "${GREEN}${BOLD}╚══════════════════════════════════════════════════════════════╝${NC}"
echo ""

if $EXCLUSIVE_MODE; then
    log_info "Running in EXCLUSIVE MODE (Selected components only)"
fi

# =============================================================================
# Run Setup Scripts
# =============================================================================

# 1. Terminal Setup (full: core + enhancement)
if $INSTALL_TERMINAL; then
    log_section "Running Terminal Setup..."
    bash "$SCRIPTS_DIR/terminal_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping terminal setup"
fi

# 2. Agent System Setup
if $INSTALL_AGENT; then
    log_section "Running Agent System Setup..."
    bash "$SCRIPTS_DIR/agent_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping agent system setup"
fi

# 3. Qdrant Setup
if $INSTALL_QDRANT; then
    log_section "Running Qdrant Setup..."
    bash "$SCRIPTS_DIR/qdrant_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping Qdrant setup"
fi

# 4. Godot Setup
if $INSTALL_GODOT; then
    log_section "Running Godot Setup..."
    bash "$SCRIPTS_DIR/godot_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping Godot setup"
fi

# 5. Additional Apps Setup
if $INSTALL_APPS; then
    log_section "Running Apps Setup..."
    bash "$SCRIPTS_DIR/apps_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping additional apps setup"
fi

# 6. OBS Studio Setup
if $INSTALL_OBS; then
    log_section "Running OBS Setup..."
    bash "$SCRIPTS_DIR/obs_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping OBS setup"
fi

# 7. EasyEffects Setup
if $INSTALL_EASYEFFECTS; then
    log_section "Running EasyEffects Setup..."
    bash "$SCRIPTS_DIR/easyeffects_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping EasyEffects setup"
fi

# 8. DNS Setup
if $INSTALL_DNS; then
    log_section "Running DNS Setup..."
    bash "$SCRIPTS_DIR/dns_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping DNS setup"
fi

# 9. Vietnamese Input Method
if $INSTALL_VIETNAMESE; then
    log_section "Running Vietnamese Input Setup..."
    bash "$SCRIPTS_DIR/input_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping Vietnamese input setup"
fi

# 10. OneDrive Setup
if $INSTALL_ONEDRIVE; then
    log_section "Running OneDrive Setup..."
    bash "$SCRIPTS_DIR/onedrive_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping OneDrive setup"
fi

# 11. VS Code Setup
if $INSTALL_VSCODE; then
    log_section "Running VS Code Setup..."
    bash "$SCRIPTS_DIR/vscode_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping VS Code setup"
fi

# 12. Remote Access Setup
if $INSTALL_REMOTE; then
    log_section "Running Remote Access Setup..."
    bash "$SCRIPTS_DIR/remote_access_setup.sh"
elif ! $EXCLUSIVE_MODE; then
    log_warn "Skipping remote access setup"
fi

# =============================================================================
# Complete
# =============================================================================
echo ""
echo -e "${GREEN}${BOLD}╔══════════════════════════════════════════════════════════════╗${NC}"
echo -e "${GREEN}${BOLD}║                    Setup Complete!                           ║${NC}"
echo -e "${GREEN}${BOLD}╚══════════════════════════════════════════════════════════════╝${NC}"
echo ""
if $INSTALL_TERMINAL; then
    echo "Please log out and log back in for the shell change to take effect."
    echo "Then open a new terminal to enjoy your restored setup!"
else
    echo "Setup finished. Check component-specific notes above."
fi
echo ""

if $INSTALL_QDRANT; then
    log_info "Qdrant is running at: http://localhost:6333"
fi

if $INSTALL_GODOT; then
    log_info "Run Godot with: godot"
fi

if $INSTALL_TERMINAL; then
    echo ""
    log_info "Power tools installed! New commands available:"
    echo "  - ls  → eza (with icons)"
    echo "  - cat → bat (syntax highlighting)"
    echo "  - cd  → z (smart directory jumping)"
    echo "  - lg  → lazygit"
    echo "  - y   → yazi (file manager)"
    echo ""
    log_info "Install tmux plugins: Press Ctrl+a then I inside tmux"
fi