first_setup.py

"""
Первичная установка Starvell Cardinal
"""

import os
import sys
import configparser
import hashlib
import time
from pathlib import Path

try:
    from colorama import Fore, Style, init, Back
    init(autoreset=True)
    HAS_COLOR = True
except ImportError:
    # Если colorama не установлена
    class Fore:
        CYAN = MAGENTA = RED = BLUE = GREEN = YELLOW = WHITE = LIGHTBLUE_EX = LIGHTGREEN_EX = ""
    class Style:
        BRIGHT = RESET_ALL = DIM = ""
    class Back:
        BLACK = ""
    HAS_COLOR = False


# ═══════════════════════════════════════════════════════════
# 🎨 Стилизация
# ═══════════════════════════════════════════════════════════

def clear():
    """Очистить экран"""
    os.system('cls' if os.name == 'nt' else 'clear')


def print_logo():
    """Красивый ASCII логотип"""
    logo = f"""
{Fore.CYAN}{Style.BRIGHT}
    ███████╗████████╗ █████╗ ██████╗ ██╗   ██╗███████╗██╗     ██╗     
    ██╔════╝╚══██╔══╝██╔══██╗██╔══██╗██║   ██║██╔════╝██║     ██║     
    ███████╗   ██║   ███████║██████╔╝██║   ██║█████╗  ██║     ██║     
    ╚════██║   ██║   ██╔══██║██╔══██╗╚██╗ ██╔╝██╔══╝  ██║     ██║     
    ███████║   ██║   ██║  ██║██║  ██║ ╚████╔╝ ███████╗███████╗███████╗
    ╚══════╝   ╚═╝   ╚═╝  ╚═╝╚═╝  ╚═╝  ╚═══╝  ╚══════╝╚══════╝╚══════╝
{Style.RESET_ALL}"""
    print(logo)


def print_header(step=None, total=4):
    """Красивый заголовок с индикатором прогресса"""
    clear()
    print_logo()
    
    if step:
        # Индикатор прогресса
        progress = "█" * step + "░" * (total - step)
        percentage = int((step / total) * 100)
        
        print(f"\n{Fore.CYAN}╔{'═' * 70}╗")
        print(f"║{' ' * 24}{Fore.WHITE}МАСТЕР УСТАНОВКИ{Fore.CYAN}{' ' * 30}║")
        print(f"║{' ' * 70}║")
        print(f"║  {Fore.LIGHTBLUE_EX}Прогресс: {Fore.GREEN}{progress}{Fore.CYAN} {percentage}% {Fore.WHITE}(Шаг {step} из {total}){Fore.CYAN}{' ' * (70 - 45 - len(str(step)) - len(str(total)))}║")
        print(f"╚{'═' * 70}╝{Style.RESET_ALL}\n")
    else:
        print(f"\n{Fore.CYAN}╔{'═' * 70}╗")
        print(f"║{' ' * 20}{Fore.WHITE}{Style.BRIGHT}ДОБРО ПОЖАЛОВАТЬ В STARVELL CARDINAL{Fore.CYAN}{Style.NORMAL}{' ' * 15}║")
        print(f"║{' ' * 22}{Fore.WHITE}Made by @knowtake{Fore.CYAN}{' ' * 25}║")
        print(f"╚{'═' * 70}╝{Style.RESET_ALL}\n")


def print_box(title, lines, color=Fore.CYAN):
    """Красивая рамка с текстом"""
    width = 68
    print(f"\n{color}┌─ {Fore.WHITE}{Style.BRIGHT}{title}{Style.RESET_ALL}")
    for line in lines:
        print(f"{color}│ {Fore.WHITE}{line}{Style.RESET_ALL}")
    print(f"{color}└{'─' * width}{Style.RESET_ALL}\n")


def print_info(text, icon="ℹ"):
    """Информационное сообщение"""
    print(f"{Fore.LIGHTBLUE_EX}{icon}  {Fore.WHITE}{text}{Style.RESET_ALL}")


def print_success(text, icon="✓"):
    """Сообщение об успехе"""
    print(f"{Fore.GREEN}{icon}  {Fore.WHITE}{text}{Style.RESET_ALL}")


def print_error(text, icon="✗"):
    """Сообщение об ошибке"""
    print(f"{Fore.RED}{icon}  {Fore.WHITE}{text}{Style.RESET_ALL}")


def print_warning(text, icon="⚠"):
    """Предупреждение"""
    print(f"{Fore.YELLOW}{icon}  {Fore.WHITE}{text}{Style.RESET_ALL}")


def ask(prompt, default="", secret=False):
    """Запросить ввод с подсказкой"""
    if default:
        full_prompt = f"{Fore.CYAN}❯ {Fore.WHITE}{prompt} {Fore.GREEN}[{default}]{Fore.CYAN}: {Style.RESET_ALL}"
    else:
        full_prompt = f"{Fore.CYAN}❯ {Fore.WHITE}{prompt}{Fore.CYAN}: {Style.RESET_ALL}"
    
    if secret:
        import getpass
        value = getpass.getpass(full_prompt)
    else:
        value = input(full_prompt).strip()
    
    return value if value else default


def ask_yes_no(prompt, default=True):
    """Запросить да/нет"""
    default_str = f"{Fore.GREEN}Y{Fore.WHITE}/{Fore.RED}n" if default else f"{Fore.WHITE}y{Fore.RED}/N"
    
    while True:
        full_prompt = f"{Fore.CYAN}❯ {Fore.WHITE}{prompt} ({default_str}{Fore.CYAN}): {Style.RESET_ALL}"
        value = input(full_prompt).strip().lower()
        
        if not value:
            return default
        
        if value in ['y', 'yes', 'д', 'да', '1']:
            return True
        if value in ['n', 'no', 'н', 'нет', '0']:
            return False
        
        print_error("Введите Y (да) или N (нет)")


def animate_dots(text, duration=1):
    """Анимация точек"""
    print(f"{Fore.CYAN}{text}", end="", flush=True)
    for _ in range(3):
        time.sleep(duration / 3)
        print(".", end="", flush=True)
    print(f" {Fore.GREEN}✓{Style.RESET_ALL}")
    time.sleep(0.3)


# ═══════════════════════════════════════════════════════════
# 🔧 Установка
# ═══════════════════════════════════════════════════════════

def run_setup():
    """Запустить установку"""
    config = configparser.ConfigParser()
    
    # ═══ Приветствие ═══
    print_header()
    
    print(f"{Fore.CYAN}╔{'═' * 70}╗")
    print(f"║{' ' * 70}║")
    print(f"║{' ' * 15}{Fore.WHITE}{Style.BRIGHT}🚀 Мастер быстрой настройки бота 🚀{Fore.CYAN}{Style.NORMAL}{' ' * 18}║")
    print(f"║{' ' * 70}║")
    print(f"║{' ' * 10}{Fore.WHITE}Сейчас мы за 4 простых шага настроим ваш бот{Fore.CYAN}{' ' * 16}║")
    print(f"║{' ' * 15}{Fore.WHITE}Это займёт всего пару минут!{Fore.CYAN}{' ' * 27}║")
    print(f"║{' ' * 70}║")
    print(f"╚{'═' * 70}╝{Style.RESET_ALL}\n")
    
    print_info("Вы можете в любой момент прервать установку нажав Ctrl+C")
    print_info("Все настройки позже можно изменить через бота\n")
    
    input(f"{Fore.GREEN}{Style.BRIGHT}❯ Нажмите Enter для начала...{Style.RESET_ALL}")
    
    # ═══════════════════════════════════════════════════════════
    # ШАГ 1: Telegram Bot Token
    # ═══════════════════════════════════════════════════════════
    print_header(step=1)
    print_header(step=1)
    
    print_box(
        "📱 ШАГ 1: Telegram Bot Token",
        [
            "Для работы бота нужен токен от @BotFather",
            "",
            "Как получить:",
            "  1️⃣  Напишите @BotFather в Telegram",
            "  2️⃣  Отправьте команду /newbot",
            "  3️⃣  Следуйте инструкциям",
            "  4️⃣  Скопируйте полученный токен"
        ]
    )
    
    while True:
        bot_token = ask("Введите токен бота")
        
        if not bot_token or ':' not in bot_token:
            print_error("Неверный формат токена!")
            print_warning("Токен должен выглядеть так: 123456789:ABCdefGHIjklMNOpqrsTUVwxyz\n")
            continue
        
        try:
            parts = bot_token.split(':')
            if not parts[0].isdigit() or len(parts[1]) < 20:
                raise ValueError()
            
            animate_dots("Проверка токена")
            print_success("Токен принят!\n")
            break
        except:
            print_error("Неверный формат токена!\n")
    
    config['Telegram'] = {'token': bot_token}
    
    # ═══════════════════════════════════════════════════════════
    # ШАГ 2: Пароль
    # ═══════════════════════════════════════════════════════════
    print_header(step=2)
    
    print_box(
        "🔐 ШАГ 2: Пароль для бота",
        [
            "Пароль защитит бот от несанкционированного доступа",
            "",
            "Требования:",
            "  ✓ Минимум 8 символов",
            "  ✓ Заглавные и строчные буквы (A-z)",
            "  ✓ Хотя бы одна цифра (0-9)"
        ]
    )
    
    while True:
        password = ask("Придумайте надёжный пароль")
        
        if len(password) < 8:
            print_error("Слишком короткий! Минимум 8 символов\n")
            continue
        
        if password.lower() == password or password.upper() == password:
            print_error("Нужны и заглавные, и строчные буквы!\n")
            continue
        
        if not any(c.isdigit() for c in password):
            print_error("Нужна хотя бы одна цифра!\n")
            continue
        
        # Подтверждение пароля
        password_confirm = ask("Повторите пароль")
        
        if password != password_confirm:
            print_error("Пароли не совпадают! Попробуйте снова\n")
            continue
        
        animate_dots("Сохранение пароля")
        print_success("Пароль принят!\n")
        break
    
    password_hash = hashlib.sha256(password.encode()).hexdigest()
    config['Telegram']['secretkeyhash'] = password_hash
    config['Telegram']['adminIds'] = '[]'
    config['Telegram']['enabled'] = '1'
    
    # ═══════════════════════════════════════════════════════════
    # ШАГ 3: Starvell Session
    # ═══════════════════════════════════════════════════════════
    print_header(step=3)
    
    print_box(
        "🔑 ШАГ 3: Starvell Session Cookie",
        [
            "Session cookie нужен для доступа к API Starvell",
            "",
            "Как получить:",
            "  1️⃣  Откройте starvell.com и войдите в аккаунт",
            "  2️⃣  Нажмите F12 (открыть DevTools)",
            "  3️⃣  Перейдите: Application → Cookies → starvell.com",
            "  4️⃣  Найдите cookie с именем 'session'",
            "  5️⃣  Скопируйте её значение (двойной клик → Ctrl+C)"
        ]
    )
    
    while True:
        session = ask("Введите session cookie")
        
        if not session or len(session) < 10:
            print_error("Cookie не может быть пустой!\n")
            continue
        
        animate_dots("Проверка cookie")
        print_success("Session сохранён!\n")
        break
    
    # User-Agent (опционально)
    print()
    print_info("User-Agent (опционально):")
    print_info("Оставьте пустым для использования стандартного\n")
    
    user_agent = ask("User-Agent (или Enter для пропуска)", "")
    
    if not user_agent:
        user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
        print_success("Используется стандартный User-Agent\n")
    else:
        print_success("User-Agent сохранён\n")
    
    config['Starvell'] = {
        'session_cookie': session,
        'user_agent': user_agent,
        'autoRaise': '0',
        'autoDelivery': '0',
        'autoRestore': '0',
        'locale': 'ru',
        'autoTicket': '0',
        'autoTicketInterval': '3600',
        'autoTicketMaxOrders': '5',
        'autoTicketOrderAge': '48'
    }

    # ═══════════════════════════════════════════════════════════
    # ШАГ 4: Прокси для Telegram (опционально)
    # ═══════════════════════════════════════════════════════════
    print_header(step=4)

    print_box(
        "🌐 ШАГ 4: Прокси для Telegram (необязательно)",
        [
            "Если Telegram заблокирован у вашего провайдера (напр. в РФ),",
            "укажите прокси — бот будет ходить в Telegram через него.",
            "",
            "Поддерживаются: SOCKS5, SOCKS4, HTTP(S).",
            "Рекомендуется SOCKS5.",
            "",
            "Можно пропустить (Enter) и включить позже через бота.",
        ]
    )

    proxy_cfg = {
        'enabled': 'false',
        'type': 'socks5',
        'ip': '',
        'port': '',
        'login': '',
        'password': '',
        'check': 'false',
    }

    use_proxy = ask_yes_no("Использовать прокси для Telegram?", default=False)

    if use_proxy:
        print()
        print_info("Тип прокси: 1) socks5  2) socks4  3) http  4) https")
        ptype_map = {'1': 'socks5', '2': 'socks4', '3': 'http', '4': 'https',
                     'socks5': 'socks5', 'socks4': 'socks4',
                     'http': 'http', 'https': 'https'}
        while True:
            ptype_in = ask("Введите тип прокси", "socks5").strip().lower()
            ptype = ptype_map.get(ptype_in)
            if ptype:
                break
            print_error("Неверный тип! Введите 1-4 или socks5/socks4/http/https\n")

        while True:
            proxy_ip = ask("IP или хост прокси (напр. 127.0.0.1)").strip()
            if proxy_ip:
                break
            print_error("Адрес не может быть пустым!\n")

        while True:
            proxy_port = ask("Порт прокси (напр. 1080)").strip()
            if proxy_port.isdigit() and 0 < int(proxy_port) < 65536:
                break
            print_error("Порт должен быть числом 1-65535!\n")

        print()
        print_info("Авторизация прокси (если нужна). Enter — пропустить.")
        proxy_login = ask("Логин (или Enter)", "").strip()
        proxy_password = ""
        if proxy_login:
            proxy_password = ask("Пароль", "").strip()

        proxy_cfg = {
            'enabled': 'true',
            'type': ptype,
            'ip': proxy_ip,
            'port': proxy_port,
            'login': proxy_login,
            'password': proxy_password,
            'check': 'false',
        }

        animate_dots("Сохранение прокси")
        _auth = f"{proxy_login}:***@" if proxy_login else ""
        print_success(f"Прокси сохранён: {ptype}://{_auth}{proxy_ip}:{proxy_port}\n")
    else:
        print_info("Прокси не используется — подключение к Telegram напрямую.\n")

    config['Proxy'] = proxy_cfg

    
    # ═══════════════════════════════════════════════════════════
    # Сохранение
    # ═══════════════════════════════════════════════════════════
    
    # Дополнительные настройки (по умолчанию)
    config['Notifications'] = {
        'checkInterval': '30',
        'newMessages': '1',
        'newOrders': '1',
        'lotRestore': '1',
        'botStart': '1',
        'lotDeactivate': '1',
        'lotBump': '1',
        'autoTicket': '1'
    }
    
    config['AutoRaise'] = {
        'interval': '3600',
        'gameId': '1',
        'categories': '[10, 11, 12]'
    }
    
    config['AutoUpdate'] = {
        'enabled': '1',
        'auto_install': '1'
    }
    
    config['KeepAlive'] = {
        'enabled': '1',
        'interval': '30'
    }
    
    config['Storage'] = {
        'dir': 'storage'
    }
    
    config['Blacklist'] = {
        'block_delivery': '1',
        'block_response': '1',
        'block_msg_notifications': '1',
        'block_order_notifications': '1'
    }
    
    config['Other'] = {
        'debug': '0',
        'watermark': '🤖'
    }
    
    config['AutoTicket'] = {
        'ticketType': '1',
        'orderUserTypeId': '2',
        'orderTopicId': '501'
    }
    
    # Сохраняем конфигурацию
    print()
    animate_dots("Сохранение конфигурации")
    
    try:
        config_path = Path("configs/_main.cfg")
        config_path.parent.mkdir(parents=True, exist_ok=True)
        
        with open(config_path, 'w', encoding='utf-8') as f:
            config.write(f)
        
        print_success("Конфигурация сохранена!\n")
        time.sleep(0.5)
        
    except Exception as e:
        print_error(f"Ошибка сохранения: {e}\n")
        return False
    
    # ═══════════════════════════════════════════════════════════
    # Финал - Успешное завершение
    # ═══════════════════════════════════════════════════════════
    clear()
    print_logo()
    
    print(f"\n{Fore.GREEN}{Style.BRIGHT}╔{'═' * 70}╗")
    print(f"║{' ' * 70}║")
    print(f"║{' ' * 20}🎉 УСТАНОВКА ЗАВЕРШЕНА УСПЕШНО! 🎉{' ' * 19}║")
    print(f"║{' ' * 70}║")
    print(f"╚{'═' * 70}╝{Style.RESET_ALL}\n")
    
    print_box(
        "📋 Сводка установки",
        [
            f"✓ Bot Token: {bot_token[:20]}...{bot_token[-10:]}",
            f"✓ Пароль: {'*' * 16}",
            f"✓ Session: {session[:25]}...{session[-15:] if len(session) > 40 else ''}",
            f"✓ Конфигурация сохранена: configs/_main.cfg"
        ],
        color=Fore.GREEN
    )
    
    print_box(
        "🚀 Что дальше?",
        [
            "1️⃣  Запустите бота:",
            f"    {Fore.GREEN}python main.py{Fore.WHITE}",
            "",
            "2️⃣  Откройте бот в Telegram и отправьте:",
            f"    {Fore.GREEN}/start{Fore.WHITE}",
            "",
            "3️⃣  Введите пароль для авторизации:",
            f"    {Fore.GREEN}{password}{Fore.WHITE}",
            "",
            "4️⃣  Готово! Управляйте ботом через меню",
            "",
            f"{Fore.YELLOW}💡 Все настройки можно изменить через бота{Fore.WHITE}"
        ],
        color=Fore.CYAN
    )
    
    print(f"{Fore.CYAN}╔{'═' * 70}╗")
    print(f"║{' ' * 18}{Fore.WHITE}Документация: {Fore.GREEN}docs/PLUGINS_API.md{Fore.CYAN}{' ' * 22}║")
    print(f"║{' ' * 20}{Fore.WHITE}Автор: {Fore.GREEN}@knowtake{Fore.CYAN}{' ' * 30}║")
    print(f"║{' ' * 19}{Fore.WHITE}Канал: {Fore.GREEN}@starvellingbot{Fore.CYAN}{' ' * 25}║")
    print(f"╚{'═' * 70}╝{Style.RESET_ALL}\n")
    
    print(f"{Fore.YELLOW}{Style.BRIGHT}Спасибо за установку Starvell Cardinal!{Style.RESET_ALL}\n")
    
    return True


# ═══════════════════════════════════════════════════════════
# 🚀 Запуск
# ═══════════════════════════════════════════════════════════

if __name__ == "__main__":
    try:
        success = run_setup()
        sys.exit(0 if success else 1)
    except KeyboardInterrupt:
        print(f"\n\n{Fore.YELLOW}Установка прервана пользователем{Style.RESET_ALL}\n")
        sys.exit(1)
    except Exception as e:
        print(f"\n\n{Fore.RED}Критическая ошибка: {e}{Style.RESET_ALL}\n")
        sys.exit(1)