from urllib.parse import urlencode
from playwright.async_api import async_playwright

async def main():
    query_params = urlencode({
        "timeout": 60000,
    })
    cdp_url = f"ws://localhost:3000?{query_params}"

    async with async_playwright() as p:
        print("Starting up the first browser and logging in...")
        # Connect over CDP
        browser1 = await p.chromium.connect_over_cdp(cdp_url)

        # Use the default context exposed by CDP and open a page
        context1 = browser1.contexts[0]
        page1 = await context1.new_page()

        # Create a raw CDP session for Browserless-specific commands
        cdp_session = await context1.new_cdp_session(page1)

        await page1.goto(
            "https://practicetestautomation.com/practice-test-login/",
            wait_until="networkidle",
            timeout=60_000,
        )
        await page1.fill("#username", "student")
        await page1.fill("#password", "Password123")
        await page1.click("#submit")
    
        # Ask Browserless to keep the session alive and give us a reconnect URL
        resp = await cdp_session.send("Browserless.reconnect", {"timeout": 60_000})
        browser_ws_endpoint = resp["browserWSEndpoint"]
        print(
            "Login successful! Enabling reconnection, the browser will remain active "
            "for 60 seconds after disconnect. browserWSEndpoint:", browser_ws_endpoint
        )

        # Detach from the first connection WITHOUT killing the remote browser.
        # (Browserless intercepts close after `Browserless.reconnect` and keeps it alive.)
        await browser1.close()

        # Reconnect using the returned endpoint
        browser2 = await p.chromium.connect_over_cdp(f"{browser_ws_endpoint}?{query_params}")

        # Grab the existing page (skip about:blank)
        context2 = browser2.contexts[0]
        pages = context2.pages
        page2 = next((pg for pg in pages if not pg.url.startswith("about:blank")), pages[0])

        title = await page2.title()
        await browser2.close()
        print("Second browser closed - Demo complete! Page 2 title:", title)

if __name__ == "__main__":
    asyncio.run(main())