deploy.sh

#!/bin/bash

# ─────────────────────────────────────────────
#  Discord Bot – Deploy / Management Menu
#  Supports: Docker mode  |  Python direct mode
# ─────────────────────────────────────────────

APP_NAME="discord-bot"
COMPOSE_FILE="docker-compose.yml"
PID_FILE="/tmp/${APP_NAME}.pid"
LOG_FILE="./bot.log"
PYTHON_CMD="python3"
MAIN_SCRIPT="main.py"

# ── Colours ────────────────────────────────────
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
CYAN='\033[0;36m'
BOLD='\033[1m'
NC='\033[0m' # No Colour

# ── Helpers ────────────────────────────────────
print_banner() {
    clear
    echo -e "${CYAN}${BOLD}"
    echo "  ██████╗ ██╗███████╗ ██████╗ ██████╗ ██████╗ ██████╗  ██████╗ ████████╗"
    echo "  ██╔══██╗██║██╔════╝██╔════╝██╔═══██╗██╔══██╗██╔══██╗██╔═══██╗╚══██╔══╝"
    echo "  ██║  ██║██║███████╗██║     ██║   ██║██████╔╝██║  ██║██║   ██║   ██║   "
    echo "  ██║  ██║██║╚════██║██║     ██║   ██║██╔══██╗██║  ██║██║   ██║   ██║   "
    echo "  ██████╔╝██║███████║╚██████╗╚██████╔╝██║  ██║██████╔╝╚██████╔╝   ██║   "
    echo "  ╚═════╝ ╚═╝╚══════╝ ╚═════╝ ╚═════╝ ╚═╝  ╚═╝╚═════╝  ╚═════╝    ╚═╝  "
    echo -e "${NC}"
    echo -e "  ${BOLD}Discord Music Bot – Management Menu${NC}"
    echo -e "  Mode: ${YELLOW}${DEPLOY_MODE^^}${NC}"
    echo "  ─────────────────────────────────────────"
}

print_status() {
    local status
    status=$(get_status)
    if [[ "$status" == "running" ]]; then
        echo -e "  Bot Status : ${GREEN}${BOLD}RUNNING${NC}"
    else
        echo -e "  Bot Status : ${RED}${BOLD}STOPPED${NC}"
    fi
    echo "  ─────────────────────────────────────────"
}

info()    { echo -e "  ${CYAN}[INFO]${NC}  $*"; }
success() { echo -e "  ${GREEN}[OK]${NC}    $*"; }
warn()    { echo -e "  ${YELLOW}[WARN]${NC}  $*"; }
error()   { echo -e "  ${RED}[ERROR]${NC} $*"; }

pause() {
    echo ""
    read -rp "  Press [Enter] to return to menu..."
}

# ── Status detection ───────────────────────────
get_status() {
    if [[ "$DEPLOY_MODE" == "docker" ]]; then
        if docker ps --format '{{.Names}}' 2>/dev/null | grep -q "^${APP_NAME}$"; then
            echo "running"
        else
            echo "stopped"
        fi
    else
        if [[ -f "$PID_FILE" ]]; then
            local pid
            pid=$(cat "$PID_FILE")
            if kill -0 "$pid" 2>/dev/null; then
                echo "running"
            else
                rm -f "$PID_FILE"
                echo "stopped"
            fi
        else
            echo "stopped"
        fi
    fi
}

# ══════════════════════════════════════════════
#  DOCKER FUNCTIONS
# ══════════════════════════════════════════════
docker_start() {
    info "Building Docker image..."
    if [[ -f "$COMPOSE_FILE" ]]; then
        docker compose up -d --build
    else
        docker build -t "$APP_NAME" . && \
        docker run -d \
            --name "$APP_NAME" \
            --restart unless-stopped \
            --env-file .env \
            "$APP_NAME"
    fi

    if [[ $? -eq 0 ]]; then
        success "Bot started successfully (Docker)."
    else
        error "Failed to start bot."
    fi
}

docker_stop() {
    info "Stopping Docker container..."
    if [[ -f "$COMPOSE_FILE" ]]; then
        docker compose down
    else
        docker stop "$APP_NAME" && docker rm "$APP_NAME"
    fi

    if [[ $? -eq 0 ]]; then
        success "Bot stopped."
    else
        error "Failed to stop bot."
    fi
}

docker_restart() {
    info "Restarting Docker container..."
    if [[ -f "$COMPOSE_FILE" ]]; then
        docker compose restart
    else
        docker restart "$APP_NAME"
    fi

    if [[ $? -eq 0 ]]; then
        success "Bot restarted."
    else
        error "Failed to restart bot."
    fi
}

docker_logs() {
    info "Showing logs (Ctrl+C to exit)..."
    echo ""
    if [[ -f "$COMPOSE_FILE" ]]; then
        docker compose logs -f --tail=50
    else
        docker logs -f --tail=50 "$APP_NAME"
    fi
}

# ══════════════════════════════════════════════
#  PYTHON DIRECT FUNCTIONS
# ══════════════════════════════════════════════
python_start() {
    if [[ ! -f ".env" ]]; then
        error ".env file not found. Create it from .env.example first."
        return 1
    fi

    info "Activating virtual environment (if present)..."
    [[ -f "venv/bin/activate" ]] && source venv/bin/activate

    info "Installing / updating dependencies..."
    pip install -q -r requirements.txt

    info "Starting bot in background..."
    nohup $PYTHON_CMD "$MAIN_SCRIPT" >> "$LOG_FILE" 2>&1 &
    echo $! > "$PID_FILE"
    success "Bot started (PID: $(cat $PID_FILE)). Logs → ${LOG_FILE}"
}

python_stop() {
    if [[ ! -f "$PID_FILE" ]]; then
        warn "No PID file found. Bot may not be running."
        return 1
    fi

    local pid
    pid=$(cat "$PID_FILE")
    info "Stopping bot (PID: ${pid})..."
    kill "$pid" 2>/dev/null && rm -f "$PID_FILE"

    if [[ $? -eq 0 ]]; then
        success "Bot stopped."
    else
        error "Could not stop process ${pid}."
    fi
}

python_restart() {
    python_stop
    sleep 1
    python_start
}

python_logs() {
    if [[ ! -f "$LOG_FILE" ]]; then
        warn "Log file not found: ${LOG_FILE}"
        return 1
    fi
    info "Showing logs (Ctrl+C to exit)..."
    echo ""
    tail -f "$LOG_FILE"
}

# ══════════════════════════════════════════════
#  MENU ACTIONS (dispatch by mode)
# ══════════════════════════════════════════════
action_start() {
    local status
    status=$(get_status)
    if [[ "$status" == "running" ]]; then
        warn "Bot is already running."
    else
        if [[ "$DEPLOY_MODE" == "docker" ]]; then docker_start; else python_start; fi
    fi
    pause
}

action_stop() {
    local status
    status=$(get_status)
    if [[ "$status" == "stopped" ]]; then
        warn "Bot is already stopped."
    else
        if [[ "$DEPLOY_MODE" == "docker" ]]; then docker_stop; else python_stop; fi
    fi
    pause
}

action_restart() {
    if [[ "$DEPLOY_MODE" == "docker" ]]; then docker_restart; else python_restart; fi
    pause
}

action_logs() {
    if [[ "$DEPLOY_MODE" == "docker" ]]; then docker_logs; else python_logs; fi
    pause
}

action_status() {
    echo ""
    local status
    status=$(get_status)
    if [[ "$status" == "running" ]]; then
        success "Bot is ${GREEN}${BOLD}RUNNING${NC}."
        if [[ "$DEPLOY_MODE" == "docker" ]]; then
            echo ""
            docker ps --filter "name=${APP_NAME}" --format "  Container: {{.Names}}\n  Image:     {{.Image}}\n  Uptime:    {{.Status}}"
        else
            echo -e "  PID: $(cat $PID_FILE 2>/dev/null || echo 'N/A')"
        fi
    else
        warn "Bot is ${RED}${BOLD}STOPPED${NC}."
    fi
    pause
}

action_install_deps() {
    info "Installing system dependencies (FFmpeg, Python3, pip)..."
    sudo apt-get update -qq
    sudo apt-get install -y ffmpeg python3 python3-pip python3-venv

    if [[ ! -d "venv" ]]; then
        info "Creating virtual environment..."
        python3 -m venv venv
    fi

    source venv/bin/activate
    info "Installing Python packages..."
    pip install -q -r requirements.txt
    success "All dependencies installed."
    pause
}

action_switch_mode() {
    echo ""
    echo -e "  Current mode: ${YELLOW}${DEPLOY_MODE^^}${NC}"
    echo ""
    echo "  [1] Docker"
    echo "  [2] Python (direct)"
    echo ""
    read -rp "  Select mode: " mode_choice
    case "$mode_choice" in
        1) DEPLOY_MODE="docker"  && success "Switched to Docker mode." ;;
        2) DEPLOY_MODE="python"  && success "Switched to Python mode." ;;
        *) warn "Invalid choice." ;;
    esac
    pause
}

# ══════════════════════════════════════════════
#  MODE SELECTION (first run)
# ══════════════════════════════════════════════
select_mode() {
    clear
    echo -e "${CYAN}${BOLD}"
    echo "  ╔══════════════════════════════════════╗"
    echo "  ║      Discord Bot – Deploy Setup      ║"
    echo "  ╚══════════════════════════════════════╝"
    echo -e "${NC}"
    echo "  How do you want to run the bot?"
    echo ""
    echo "  [1] Docker  (recommended – isolated container)"
    echo "  [2] Python  (direct – runs in current environment)"
    echo ""
    read -rp "  Choose [1/2]: " choice

    case "$choice" in
        1)
            if ! command -v docker &>/dev/null; then
                error "Docker is not installed. Install it first: https://docs.docker.com/engine/install/ubuntu/"
                exit 1
            fi
            DEPLOY_MODE="docker"
            ;;
        2)
            DEPLOY_MODE="python"
            ;;
        *)
            error "Invalid choice. Exiting."
            exit 1
            ;;
    esac
}

# ══════════════════════════════════════════════
#  MAIN MENU LOOP
# ══════════════════════════════════════════════
main_menu() {
    while true; do
        print_banner
        print_status
        echo ""
        echo -e "  ${BOLD}[1]${NC} Start"
        echo -e "  ${BOLD}[2]${NC} Stop"
        echo -e "  ${BOLD}[3]${NC} Restart"
        echo -e "  ${BOLD}[4]${NC} View Logs"
        echo -e "  ${BOLD}[5]${NC} Status"
        echo -e "  ${BOLD}[6]${NC} Install Dependencies"
        echo -e "  ${BOLD}[7]${NC} Switch Mode (current: ${YELLOW}${DEPLOY_MODE^^}${NC})"
        echo -e "  ${BOLD}[0]${NC} Exit"
        echo ""
        read -rp "  Choose an option: " opt

        case "$opt" in
            1) action_start ;;
            2) action_stop ;;
            3) action_restart ;;
            4) action_logs ;;
            5) action_status ;;
            6) action_install_deps ;;
            7) action_switch_mode ;;
            0) echo -e "\n  ${CYAN}Goodbye!${NC}\n"; exit 0 ;;
            *) warn "Invalid option."; pause ;;
        esac
    done
}

# ══════════════════════════════════════════════
#  ENTRY POINT
# ══════════════════════════════════════════════

# Allow non-interactive mode via CLI args: ./deploy.sh start|stop|restart
if [[ $# -gt 0 ]]; then
    DEPLOY_MODE="${DEPLOY_MODE:-python}"
    case "$1" in
        start)   action_start ;;
        stop)    action_stop ;;
        restart) action_restart ;;
        status)  action_status ;;
        logs)    action_logs ;;
        *)
            echo "Usage: $0 [start|stop|restart|status|logs]"
            echo "       $0         (interactive menu)"
            exit 1
            ;;
    esac
    exit 0
fi

select_mode
main_menu