import os

from playwright.sync_api import sync_playwright, TimeoutError as PlaywrightTimeoutError

from utils.logger import setup_logger

logger = setup_logger()


def get_token_from_local_storage(
    email: str, password: str, record_video: bool = False
) -> str:
    logger.info("Starting Playwright flow")

    with sync_playwright() as p:
        logger.info("Playwright server started")

        browser = p.chromium.launch(
            headless=True,
            args=[
                "--no-sandbox",
                "--disable-dev-shm-usage",
                "--disable-gpu",
                "--single-process",
                "--no-zygote",
            ],
        )
        logger.info("Chromium launched successfully")

        video_dir = None
        if record_video:
            video_dir = os.path.join(os.path.dirname(__file__), "videos")
            os.makedirs(video_dir, exist_ok=True)

        if record_video:
            context = browser.new_context(
                record_video_dir=video_dir,
                record_video_size={"width": 1280, "height": 720},
            )
        else:
            context = browser.new_context()
        page = context.new_page()
        logger.info("Page created")

        try:
            logger.info("Navigating to site...")
            page.goto("https://pashub.net/", timeout=30000)

            logger.info("Filling login form...")

            email_input = page.locator("#email")
            email_input.wait_for(state="visible", timeout=10000)
            email_input.fill(email)

            password_input = page.locator("#password")
            password_input.wait_for(state="visible", timeout=10000)
            password_input.focus()
            password_input.fill(password)

            logger.info("Submitting login...")
            page.wait_for_selector("#btn-login", state="visible", timeout=10000)
            with page.expect_navigation(timeout=15000):
                page.click("#btn-login")

            page.wait_for_timeout(3000)

            if "login" in page.url.lower():
                raise Exception("Login failed (still on login page)")

            logger.info(f"Login likely successful. URL: {page.url}")

            token = page.evaluate(
                """() => {
                    return localStorage.getItem('token');
                }"""
            )

            if not token:
                raise Exception("Login succeeded but no token found")

            return token

        except PlaywrightTimeoutError as e:
            raise Exception(f"Timeout during login flow: {str(e)}")

        except Exception as e:
            raise Exception(f"Unexpected error: {str(e)}")

        finally:
            context.close()
            browser.close()

            if record_video and video_dir:
                logger.info(f"Video(s) saved in: {video_dir}")