#!/bin/bash

# Pocket TTS Discord Bot - Linux Setup Script
# This script helps set up the bot and install it as a systemd service

set -e

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

echo -e "${GREEN}========================================${NC}"
echo -e "${GREEN}  Pocket TTS Discord Bot - Linux Setup${NC}"
echo -e "${GREEN}========================================${NC}"
echo

# Get the directory where this script is located
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
USERNAME="$(whoami)"

# Check if running as root
if [ "$EUID" -eq 0 ]; then
    echo -e "${RED}Please do not run this script as root.${NC}"
    echo "Run it as the user who will own the bot."
    exit 1
fi

# Function to check if a command exists
command_exists() {
    command -v "$1" >/dev/null 2>&1
}

echo -e "${YELLOW}Step 1: Checking system dependencies...${NC}"

# Check for Python 3.10+
if command_exists python3; then
    PYTHON_VERSION=$(python3 -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')
    PYTHON_MAJOR=$(echo $PYTHON_VERSION | cut -d. -f1)
    PYTHON_MINOR=$(echo $PYTHON_VERSION | cut -d. -f2)
    
    if [ "$PYTHON_MAJOR" -ge 3 ] && [ "$PYTHON_MINOR" -ge 10 ]; then
        echo -e "  ${GREEN}✓${NC} Python $PYTHON_VERSION found"
    else
        echo -e "  ${RED}✗${NC} Python 3.10+ required, found $PYTHON_VERSION"
        echo "  Please install Python 3.10 or later"
        exit 1
    fi
else
    echo -e "  ${RED}✗${NC} Python 3 not found"
    echo "  Please install Python 3.10 or later"
    exit 1
fi

# Check for FFmpeg
if command_exists ffmpeg; then
    FFMPEG_VERSION=$(ffmpeg -version 2>&1 | head -n1 | cut -d' ' -f3)
    echo -e "  ${GREEN}✓${NC} FFmpeg found ($FFMPEG_VERSION)"
else
    echo -e "  ${RED}✗${NC} FFmpeg not found"
    echo ""
    echo "  Please install FFmpeg:"
    echo "    Ubuntu/Debian: sudo apt install ffmpeg"
    echo "    Fedora: sudo dnf install ffmpeg"
    echo "    Arch: sudo pacman -S ffmpeg"
    exit 1
fi

# Check for pip
if command_exists pip3; then
    echo -e "  ${GREEN}✓${NC} pip3 found"
else
    echo -e "  ${RED}✗${NC} pip3 not found"
    echo "  Please install python3-pip"
    exit 1
fi

echo
echo -e "${YELLOW}Step 2: Setting up virtual environment...${NC}"

cd "$SCRIPT_DIR"

if [ -d "venv" ]; then
    echo "  Virtual environment already exists"
else
    echo "  Creating virtual environment..."
    python3 -m venv venv
    echo -e "  ${GREEN}✓${NC} Virtual environment created"
fi

echo "  Activating virtual environment..."
source venv/bin/activate

echo "  Installing dependencies..."
pip install --upgrade pip -q
pip install -r requirements.txt -q
echo -e "  ${GREEN}✓${NC} Dependencies installed"

echo
echo -e "${YELLOW}Step 3: Checking configuration...${NC}"

# Check for .env file
if [ -f ".env" ]; then
    echo -e "  ${GREEN}✓${NC} .env file found"
else
    echo -e "  ${YELLOW}!${NC} .env file not found"
    echo "  Creating .env template..."
    cat > .env << 'EOF'
# Discord Bot Configuration
DISCORD_TOKEN=your_bot_token_here
TEXT_CHANNEL_ID=your_channel_id_here

# Voice Configuration
VOICES_DIR=./voices
# DEFAULT_VOICE=estinien
EOF
    echo -e "  ${YELLOW}!${NC} Please edit .env with your Discord token and channel ID"
fi

# Check for voices directory
if [ -d "voices" ]; then
    VOICE_COUNT=$(find voices -name "*.wav" 2>/dev/null | wc -l)
    echo -e "  ${GREEN}✓${NC} voices directory found ($VOICE_COUNT voice files)"
    if [ "$VOICE_COUNT" -eq 0 ]; then
        echo -e "  ${YELLOW}!${NC} No voice files found. Add .wav files to the voices directory."
    fi
else
    echo "  Creating voices directory..."
    mkdir -p voices
    echo -e "  ${YELLOW}!${NC} Add voice .wav files to the voices directory"
fi

echo
echo -e "${YELLOW}Step 4: Setting up systemd service...${NC}"

read -p "Do you want to install the bot as a systemd service? (y/n) " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
    # Create the service file with correct paths
    SERVICE_FILE="/tmp/pockettts.service"
    
    cat > "$SERVICE_FILE" << EOF
[Unit]
Description=Pocket TTS Discord Bot
After=network-online.target
Wants=network-online.target

[Service]
User=$USERNAME
Group=$USERNAME
WorkingDirectory=$SCRIPT_DIR
ExecStart=$SCRIPT_DIR/venv/bin/python bot.py
Restart=on-failure
RestartSec=10
TimeoutStopSec=30
StandardOutput=journal
StandardError=journal
SyslogIdentifier=pockettts

# Security hardening
NoNewPrivileges=true
ProtectSystem=strict
ProtectHome=read-only
ReadWritePaths=$SCRIPT_DIR/voices
PrivateTmp=true

[Install]
WantedBy=multi-user.target
EOF

    echo "  Installing systemd service (requires sudo)..."
    sudo cp "$SERVICE_FILE" /etc/systemd/system/pockettts.service
    sudo systemctl daemon-reload
    echo -e "  ${GREEN}✓${NC} Service installed"
    
    read -p "Do you want to enable the service to start on boot? (y/n) " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        sudo systemctl enable pockettts
        echo -e "  ${GREEN}✓${NC} Service enabled for boot"
    fi
    
    read -p "Do you want to start the service now? (y/n) " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        sudo systemctl start pockettts
        echo -e "  ${GREEN}✓${NC} Service started"
        sleep 2
        echo
        echo "  Service status:"
        sudo systemctl status pockettts --no-pager || true
    fi
fi

echo
echo -e "${GREEN}========================================${NC}"
echo -e "${GREEN}  Setup Complete!${NC}"
echo -e "${GREEN}========================================${NC}"
echo
echo "Useful commands:"
echo "  Start bot:      sudo systemctl start pockettts"
echo "  Stop bot:       sudo systemctl stop pockettts"
echo "  Restart bot:    sudo systemctl restart pockettts"
echo "  View status:    sudo systemctl status pockettts"
echo "  View logs:      journalctl -u pockettts -f"
echo "  Disable boot:   sudo systemctl disable pockettts"
echo
echo "To run the bot manually (without systemd):"
echo "  cd $SCRIPT_DIR"
echo "  source venv/bin/activate"
echo "  python bot.py"
echo