setup.py

"""
setup.py — automatic bearer token collection for Dawn Validator BOT.

Script:
- reads emails.txt and captcha_key.txt
- simulates the login and token retrieval process
- saves tokens to tokens.json

Actual HTTP automation can be implemented inside the fetch_token_for_email function.
"""

from __future__ import annotations

import hashlib
import time
from typing import Dict

from config_manager import (
    read_emails,
    read_captcha_key,
    save_tokens,
)
from ui import print_title, info, warn, error, pause


def fetch_token_for_email(email: str, captcha_key: str) -> str:
    """
    Stub for actual bearer token retrieval.

    Implement here:
    - automated login via HTTP/browser
    - captcha bypass with 2captcha
    - bearer token extraction from responses
    """
    seed = f"{email}:{captcha_key}:{time.time()}".encode("utf-8")
    return hashlib.sha256(seed).hexdigest()


def run_setup() -> None:
    print_title("Automatic Token Setup (setup.py)")
    emails = read_emails()
    if not emails:
        error("No emails found in emails.txt — configure accounts before running Setup.")
        pause()
        return

    captcha_key = read_captcha_key()
    if not captcha_key:
        error("Captcha key not found — configure captcha_key.txt before running Setup.")
        pause()
        return

    warn(f"Automatic token collection will be performed for {len(emails)} accounts.")

    tokens: Dict[str, str] = {}
    for idx, email in enumerate(emails, start=1):
        info(f"[{idx}/{len(emails)}] Retrieving token for {email} ...")
        token = fetch_token_for_email(email, captcha_key)
        tokens[email] = token
        time.sleep(0.3)

    save_tokens(tokens)
    info("Setup complete. You can now start the main bot.")
    pause()


if __name__ == "__main__":
    run_setup()