#!/bin/bash
# TikTok Worker - Ubuntu Setup Script
# Pokreni: chmod +x setup_ubuntu.sh && ./setup_ubuntu.sh

echo "=========================================="
echo "  TikTok Worker - Ubuntu Setup"
echo "=========================================="

# Instaliraj Python ako ne postoji
if ! command -v python3 &> /dev/null; then
    echo "[*] Instaliram Python3..."
    sudo apt update
    sudo apt install -y python3 python3-pip python3-venv
fi

# Kreiraj virtual environment
echo "[*] Kreiram virtual environment..."
python3 -m venv venv
source venv/bin/activate

# Instaliraj dependencies
echo "[*] Instaliram Python dependencies..."
pip install --upgrade pip
pip install -r requirements.txt

# Instaliraj Playwright browser
echo "[*] Instaliram Chromium za Playwright..."
playwright install chromium
playwright install-deps chromium

# Kreiraj systemd service (opciono)
echo ""
echo "=========================================="
echo "  Setup završen!"
echo "=========================================="
echo ""
echo "Za pokretanje:"
echo "  source venv/bin/activate"
echo "  python app.py"
echo ""
echo "Otvori: http://tvoj-server:5000"
echo ""

# Pitaj za systemd service
read -p "Instaliraj kao systemd service? (y/n): " answer
if [ "$answer" = "y" ]; then
    SERVICE_FILE="/etc/systemd/system/tiktok-worker.service"
    CURRENT_DIR=$(pwd)
    CURRENT_USER=$(whoami)
    
    sudo bash -c "cat > $SERVICE_FILE << EOF
[Unit]
Description=TikTok Worker Dashboard
After=network.target

[Service]
Type=simple
User=$CURRENT_USER
WorkingDirectory=$CURRENT_DIR
ExecStart=$CURRENT_DIR/venv/bin/python app.py
Restart=always
RestartSec=10

[Install]
WantedBy=multi-user.target
EOF"
    
    sudo systemctl daemon-reload
    sudo systemctl enable tiktok-worker
    sudo systemctl start tiktok-worker
    
    echo ""
    echo "[✓] Service instaliran!"
    echo "    Status: sudo systemctl status tiktok-worker"
    echo "    Logs:   sudo journalctl -u tiktok-worker -f"
fi