import type { Discardable } from '@serenity-js/core';
import type { BrowserCapabilities } from '@serenity-js/web';
import type * as playwright from 'playwright-core';

import type { ExtraBrowserContextOptions } from '../../ExtraBrowserContextOptions.js';
import { PlaywrightBrowsingSession } from './PlaywrightBrowsingSession.js';
import type { PlaywrightPage } from './PlaywrightPage.js';

/**
 *  @group Models
 */
export class PlaywrightBrowsingSessionWithBrowser
    extends PlaywrightBrowsingSession
    implements Discardable
{

    constructor(
        protected readonly browser: playwright.Browser,
        protected readonly browserContextOptions: playwright.BrowserContextOptions,
        extraBrowserContextOptions: ExtraBrowserContextOptions,
        selectors: playwright.Selectors,
    ) {
        super(extraBrowserContextOptions, selectors);
    }

    protected override async createBrowserContext(): Promise<playwright.BrowserContext> {
        // todo: move to initialise?
        return this.browser.newContext(this.browserContextOptions);
    }

    protected override async registerCurrentPage(): Promise<PlaywrightPage> {
        const context = await this.browserContext();

        await context.newPage();

        // calling context.newPage() triggers a callback registered via browserContext(),
        // which wraps playwright.Page in PlaywrightPage and adds it to the list of pages
        // returned by this.allPages()

        const allPages = await this.allPages()

        return allPages.at(-1);
    }

    override async browserCapabilities(): Promise<BrowserCapabilities> {
        return {
            browserName: (this.browser as any)._initializer.name,   // todo: raise a PR to Playwright to expose this information
            platformName: process.platform,                         // todo: get the actual platform from Playwright
            browserVersion: this.browser.version()
        }
    }

    async discard(): Promise<void> {
        await this.closeAllPages();

        const context = await this.browserContext();
        await context.close();
    }
}