main.py

import logging
import time

from bot.adb import ADBClient
from bot.config import parse_args
from bot.vision import load_template
from bot.drops import run_drop_loop
from bot.quests import run_quest_loop

logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s | %(levelname)s | %(message)s",
)
logger = logging.getLogger(__name__)


def prompt_mode() -> tuple[bool, bool]:
    """Prompt for mode selection. Returns (quest_mode, epic_mode)."""
    print("Soundmap Bot")
    print("------------")
    print("1. Drop farming")
    print("2. Quest automation")
    while True:
        choice = input("Select mode (1/2): ").strip()
        if choice == "1":
            return False, False
        if choice == "2":
            break

    print("")
    print("Quest Mode")
    print("----------")
    print("1. Normal (collect songs)")
    print("2. Epic hunting (quicksell non-epics, stop on epic)")
    while True:
        sub = input("Select quest mode (1/2): ").strip()
        if sub == "1":
            return True, False
        if sub == "2":
            return True, True


def main() -> int:
    parsed = parse_args()
    cfg = parsed.cfg
    if parsed.quest_mode:
        quest_mode = True
        epic_mode = cfg.epic_mode
    else:
        quest_mode, epic_mode = prompt_mode()

    adb = ADBClient(cfg.adb_path)
    start_time = time.monotonic()

    logger.info("Smart Drop Bot starting (Scrcpy/ADB Mode)...")

    template_gray = load_template(
        path=cfg.template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! The bot will run but cannot check for coins.",
        invalid_message="Could not load '%s'. Coin detection disabled.",
    )
    open_button_template_gray = load_template(
        path=parsed.open_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Open button detection disabled.",
        invalid_message="Could not load '%s'. Open button detection disabled.",
    )
    collect_button_template_gray = load_template(
        path=parsed.collect_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Collect button detection disabled.",
        invalid_message="Could not load '%s'. Collect button detection disabled.",
    )
    plus_button_template_gray = load_template(
        path=parsed.plus_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Quest + button detection disabled.",
        invalid_message="Could not load '%s'. Quest + button detection disabled.",
    )
    reroll_button_template_gray = load_template(
        path=parsed.reroll_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Quest reroll button detection disabled.",
        invalid_message="Could not load '%s'. Quest reroll button detection disabled.",
    )
    confirm_button_template_gray = load_template(
        path=parsed.confirm_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Quest confirm button detection disabled.",
        invalid_message="Could not load '%s'. Quest confirm button detection disabled.",
    )
    no_songs_template_gray = load_template(
        path=parsed.no_songs_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Quest no-songs detection disabled.",
        invalid_message="Could not load '%s'. Quest no-songs detection disabled.",
    )
    reroll_alt_template_gray = load_template(
        path=parsed.reroll_alt_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Alt reroll button detection disabled.",
        invalid_message="Could not load '%s'. Alt reroll button detection disabled.",
    )
    reroll_confirm_template_gray = load_template(
        path=parsed.reroll_confirm_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Reroll confirmation popup detection disabled.",
        invalid_message="Could not load '%s'. Reroll confirmation popup detection disabled.",
    )
    lets_go_template_gray = load_template(
        path=parsed.lets_go_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! 'Let's Go!' button detection disabled.",
        invalid_message="Could not load '%s'. 'Let's Go!' button detection disabled.",
    )
    epic_tag_template_gray = load_template(
        path=parsed.epic_tag_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Epic tag detection disabled.",
        invalid_message="Could not load '%s'. Epic tag detection disabled.",
    )
    rare_tag_template_gray = load_template(
        path=parsed.rare_tag_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Rare tag detection disabled — absence-of-rare logic will not work.",
        invalid_message="Could not load '%s'. Rare tag detection disabled.",
    )
    common_tag_template_gray = load_template(
        path=parsed.common_tag_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Common tag detection disabled — absence-of-common logic will not work.",
        invalid_message="Could not load '%s'. Common tag detection disabled.",
    )
    sell_button_template_gray = load_template(
        path=parsed.sell_button_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Sell button detection disabled.",
        invalid_message="Could not load '%s'. Sell button detection disabled.",
    )
    sell_confirm_template_gray = load_template(
        path=parsed.sell_confirm_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Sell confirm button detection disabled.",
        invalid_message="Could not load '%s'. Sell confirm button detection disabled.",
    )
    quest_update_ok_template_gray = load_template(
        path=parsed.quest_update_ok_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Quest updated popup dismissal disabled.",
        invalid_message="Could not load '%s'. Quest updated popup dismissal disabled.",
    )
    lucky_quest_continue_template_gray = load_template(
        path=parsed.lucky_quest_continue_template_file,
        scale=cfg.scale,
        missing_message="'%s' not found! Lucky quest popup dismissal disabled.",
        invalid_message="Could not load '%s'. Lucky quest popup dismissal disabled.",
    )

    logger.info("Bot is live. Press Ctrl+C in this terminal to stop.")

    if quest_mode:
        if plus_button_template_gray is None:
            logger.error("Quest mode requires a valid plus button template. Stopping.")
            return 1
        return run_quest_loop(
            adb=adb,
            cfg=cfg,
            epic_mode=epic_mode,
            plus_button_template_gray=plus_button_template_gray,
            reroll_button_template_gray=reroll_button_template_gray,
            confirm_button_template_gray=confirm_button_template_gray,
            no_songs_template_gray=no_songs_template_gray,
            lets_go_template_gray=lets_go_template_gray,
            collect_button_template_gray=collect_button_template_gray,
            reroll_confirm_template_gray=reroll_confirm_template_gray,
            reroll_alt_template_gray=reroll_alt_template_gray,
            epic_tag_template_gray=epic_tag_template_gray,
            rare_tag_template_gray=rare_tag_template_gray,
            common_tag_template_gray=common_tag_template_gray,
            sell_button_template_gray=sell_button_template_gray,
            sell_confirm_template_gray=sell_confirm_template_gray,
            quest_update_ok_template_gray=quest_update_ok_template_gray,
            lucky_quest_continue_template_gray=lucky_quest_continue_template_gray,
            start_time=start_time,
        )
    else:
        return run_drop_loop(
            adb=adb,
            cfg=cfg,
            template_gray=template_gray,
            open_button_template_gray=open_button_template_gray,
            collect_button_template_gray=collect_button_template_gray,
            start_time=start_time,
        )


if __name__ == "__main__":
    raise SystemExit(main())