#!/bin/bash
# Скрипт для получения cookies YouTube через yt-dlp

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
COOKIES_FILE="$SCRIPT_DIR/youtube_cookies.txt"

echo "Получение cookies YouTube из браузера..."
echo ""
echo "Выберите браузер:"
echo "1) Chrome"
echo "2) Firefox"
echo "3) Edge"
echo "4) Opera"
read -p "Введите номер (1-4): " browser

case "$browser" in
    1) BROWSER="chrome" ;;
    2) BROWSER="firefox" ;;
    3) BROWSER="edge" ;;
    4) BROWSER="opera" ;;
    *)
        echo "Неверный выбор"
        exit 1
        ;;
esac

echo "Получаю cookies из $BROWSER..."
echo "Файл cookies будет сохранен в: $COOKIES_FILE"

# Проверяем наличие yt-dlp
if ! command -v yt-dlp &> /dev/null; then
    echo ""
    echo "❌ yt-dlp не найден в системе"
    echo ""
    echo "Установите yt-dlp одним из способов:"
    echo ""
    echo "1. Через pacman (Arch Linux):"
    echo "   sudo pacman -S yt-dlp"
    echo ""
    echo "2. Через pip:"
    echo "   pip install yt-dlp"
    echo "   или"
    echo "   pip3 install yt-dlp"
    echo ""
    echo "3. Через pipx (рекомендуется):"
    echo "   pipx install yt-dlp"
    echo ""
    exit 1
fi

yt-dlp --cookies-from-browser "$BROWSER" --cookies "$COOKIES_FILE" --no-download https://www.youtube.com 2>&1 | head -10

if [ -f "$COOKIES_FILE" ]; then
    echo ""
    echo "✅ Cookies успешно сохранены в $COOKIES_FILE"
    echo ""
    echo "Теперь перезапустите youtube-downloader:"
    echo "  docker compose -f youtube-downloader/docker-compose.yml restart"
else
    echo ""
    echo "❌ Ошибка: файл cookies не был создан"
    exit 1
fi