import sys
import signal
from time import time_ns
from playwright.sync_api import sync_playwright

from helpers.helper_functions import log_note, get_random_text, login_nextcloud, close_modal, timeout_handler

def main(browser_name: str = "chromium"):
    with sync_playwright() as playwright:
        log_note(f"Launch browser {browser_name}")
        signal.signal(signal.SIGALRM, timeout_handler)
        signal.alarm(10)
        if browser_name == "firefox":
            browser = playwright.firefox.launch(headless=True)
        else:
            # this leverages new headless mode by Chromium: https://developer.chrome.com/articles/new-headless/
            # The mode is however ~40% slower: https://github.com/microsoft/playwright/issues/21216
            browser = playwright.chromium.launch(headless=False,args=["--headless=new"])
        context = browser.new_context()
        page = context.new_page()
        signal.alarm(0) # remove timeout signal
        try:
            page.goto('http://nc/')

            # 1. Create User
            log_note("Create admin user")
            page.locator('#adminlogin').fill('Crash')
            page.locator('#adminpass').fill('Override')
            page.locator('.primary').click()

            # 2. Install all Apps
            log_note("Install recommended apps")
            install_selector = '.button-vue--vue-primary'
            page.locator(install_selector).click()

            # 3. Dashboard
            page.locator('.app-dashboard').wait_for(timeout=240_000)
            log_note("Installation complete")
            browser.close()

        except Exception as e:
            if hasattr(e, 'message'): # only Playwright error class has this member
                log_note(f"Exception occurred: {e.message}")

            # set a timeout. Since the call to page.content() is blocking we need to defer it to the OS
            signal.signal(signal.SIGALRM, timeout_handler)
            signal.alarm(20)
            log_note(f"Page content was: {page.content()}")
            signal.alarm(0) # remove timeout signal
            raise e


if __name__ == '__main__':
    if len(sys.argv) > 1:
        browser_name = sys.argv[1].lower()
        if browser_name not in ["chromium", "firefox"]:
            print("Invalid browser name. Please choose either 'chromium' or 'firefox'.")
            sys.exit(1)
    else:
        browser_name = "chromium"

    main(browser_name)