import asyncio

from playwright.async_api import async_playwright
from ppadb.client import Client


# This function works
def forward(port):
    """Creates a port forwarding for one device"""
    client = Client()
    device = client.devices()[0]
    device.forward(local=f"tcp:{port}", remote="localabstract:chrome_devtools_remote")


# This function blocks on the goto()
async def run(port):
    """Connects to a USB device over CDP (via the port forwarding) and visit an URL"""
    async with async_playwright() as pw:
        browser = await pw.chromium.connect_over_cdp(f"http://localhost:{port}")

    # Infinite wait
    await browser.contexts[0].pages[0].goto("https://google.com")


if __name__ == '__main__':
    forward_port = 9001

    # Create port forwarding
    forward(forward_port)

    # Run the async function
    asyncio.run(run(forward_port))