import base64
import json
import os
import sys
import requests


def get_vpn_servers():
    # Получаем ссылку из секретов Gitea
    sub_url = os.getenv("VPN_SUB_URL")

    if not sub_url:
        print("Ошибка: Переменная окружения VPN_SUB_URL не задана!")
        sys.exit(1)

    # 1. Сначала пробуем сделать запрос по исходной ссылке
    # Имитируем заголовки приложения Happ
    headers = {
        "User-Agent": "Happ/2.1.0 (iPhone; iOS 15.5; Scale/3.00)",
        "Accept": "*/*",
    }

    try:
        print("Отправка запроса к серверу VPN...")
        response = requests.get(sub_url, headers=headers, timeout=15)

        if response.status_code != 200:
            print(f"Ошибка сервера: Статус {response.status_code}")
            sys.exit(1)

        raw_content = response.text.strip()

        # 2. Проверяем, не JSON ли это (на случай, если в секрет сразу зашит appConfig URL)
        try:
            data = json.loads(raw_content)
            print("Обнаружен формат JSON (панель Happ API).")
            servers = data.get("data", {}).get("server_user", data)

            with open("servers.json", "w", encoding="utf-8") as f:
                json.dump(servers, f, indent=2, ensure_ascii=False)
            print("Данные сохранены в файл servers.json")
            return
        except json.JSONDecodeError:
            # Если не JSON, идем дальше
            pass

        # 3. Проверяем, не Base64 ли это (стандартная подписка)
        try:
            # Добавляем padding, если строка обрезана (стандартный трюк для base64 в Python)
            padded_content = raw_content + "=" * (4 - len(raw_content) % 4)
            decoded_bytes = base64.b64decode(padded_content)
            decoded_text = decoded_bytes.decode("utf-8", errors="ignore").strip()

            # Если внутри есть vless/ss/ssr/trojan/vmess ссылки
            if any(
                proto in decoded_text
                for proto in ["vless://", "vmess://", "ss://", "trojan://"]
            ):
                print(
                    "Обнаружена Base64 подписка. Успешно декодировано в список ссылок."
                )

                # Вариант А: Сохраняем как простой текстовый список серверов
                with open("servers.txt", "w", encoding="utf-8") as f:
                    f.write(decoded_text)
                print("Ссылки сохранены в файл servers.txt")

                # Вариант Б: Для удобства сохраняем этот же список в JSON массив
                links_list = [
                    line.strip()
                    for line in decoded_text.splitlines()
                    if line.strip()
                ]
                with open("servers.json", "w", encoding="utf-8") as f:
                    json.dump(links_list, f, indent=2, ensure_ascii=False)
                print("Ссылки также структурированы в servers.json")
                return
        except Exception as b64_err:
            print(f"Попытка декодировать Base64 не удалась: {b64_err}")

        # 4. Если это и не JSON, и не Base64, возможно это уже готовый сырой текст со ссылками
        if any(
            proto in raw_content
            for proto in ["vless://", "vmess://", "ss://", "trojan://"]
        ):
            print("Сервер вернул сырой текст со ссылками.")
            with open("servers.txt", "w", encoding="utf-8") as f:
                f.write(raw_content)

            links_list = [
                line.strip() for line in raw_content.splitlines() if line.strip()
            ]
            with open("servers.json", "w", encoding="utf-8") as f:
                json.dump(links_list, f, indent=2, ensure_ascii=False)
            print("Данные сохранены в servers.txt и servers.json")
            return

        # 5. Если вообще ничего не подошло, сохраняем как есть для анализа
        print(
            "Неизвестный формат ответа. Сохраняем raw-текст в raw_response.txt для отладки."
        )
        with open("raw_response.txt", "w", encoding="utf-8") as f:
            f.write(raw_content)
        sys.exit(1)

    except Exception as e:
        print(f"Произошла критическая ошибка: {e}")
        sys.exit(1)


if __name__ == "__main__":
    get_vpn_servers()