main.py

import os
import sys
import asyncio
import platform
import subprocess

CONFIG_FILE = "config.py"

def is_interactive():
    return sys.stdin.isatty()

# ============================================================
# 1️⃣ SETUP WIZARD — запускается только если интерактивный шелл
# ============================================================
def setup_wizard():
    print("\n" + "="*40)
    print(" 🛠  SETUP WIZARD")
    print("="*40)

    use_discord = input("1. Включить Discord бота? (y/n): ").lower().strip() == 'y'
    discord_token = input("   -> DISCORD_TOKEN: ").strip() if use_discord else ""
    discord_owner = input("   -> Discord Owner Username: ").strip() if use_discord else ""

    use_telegram = input("\n2. Включить Telegram бота? (y/n): ").lower().strip() == 'y'
    telegram_token = input("   -> TELEGRAM_TOKEN: ").strip() if use_telegram else ""
    telegram_owner = input("   -> Telegram Owner ID: ").strip() if use_telegram else ""
    webhook_url = ""
    if use_telegram:
        webhook_url = input("   -> Telegram Webhook URL (пусто = polling): ").strip()

    with open(CONFIG_FILE, "w", encoding="utf-8") as f:
        f.write(f'''# Auto-generated config
ENABLE_DISCORD = {use_discord}
DISCORD_TOKEN = "{discord_token}"
DISCORD_OWNER = "{discord_owner}"

ENABLE_TELEGRAM = {use_telegram}
TELEGRAM_TOKEN = "{telegram_token}"
TELEGRAM_OWNER = "{telegram_owner}"
TELEGRAM_WEBHOOK_URL = "{webhook_url}"
''')

    print("\n✅ Config created! Перезапусти программу.\n")
    sys.exit(0)


# ============================================================
# 2️⃣ Проверка наличия конфига
# ============================================================
if not os.path.exists(CONFIG_FILE):
    if is_interactive():
        setup_wizard()
    else:
        print("[CONFIG] Файл config.py не найден. Создайте его вручную или запустите интерактивно.")
        print("Пример:\nENABLE_DISCORD = False\nENABLE_TELEGRAM = True\nTELEGRAM_TOKEN = 'yourtoken'\n")
        sys.exit(1)

import config


# ============================================================
# 3️⃣ Установка зависимостей
# ============================================================
REQUIRED = ["psutil"]
if getattr(config, "ENABLE_DISCORD", False): REQUIRED.append("discord.py")
if getattr(config, "ENABLE_TELEGRAM", False): REQUIRED.append("python-telegram-bot==20.*")

for lib in REQUIRED:
    try:
        __import__(lib.split("==")[0].replace("-", "_"))
    except ImportError:
        subprocess.check_call([sys.executable, "-m", "pip", "install", lib])


# ============================================================
# 4️⃣ Запуск соответствующего бота
# ============================================================
async def main():
    if config.ENABLE_DISCORD and not config.ENABLE_TELEGRAM:
        from bot_discord import run_discord
        await run_discord()

    elif config.ENABLE_TELEGRAM and not config.ENABLE_DISCORD:
        from bot_telegram import run_telegram
        await run_telegram()

    elif config.ENABLE_DISCORD and config.ENABLE_TELEGRAM:
        print("⚠️ Один процесс — один бот. Включи только один в config.py")
        sys.exit(1)
    else:
        print("❌ Ни один бот не включен в config.py")


if __name__ == "__main__":
    if platform.system() == "Windows":
        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
    asyncio.run(main())