import logger from '@wdio/logger'
import type { Capabilities, Services } from '@wdio/types'

import { setPort } from './client.js'
import { CUSTOM_CAP } from './constants.js'
import type { SharedStoreServiceCapabilities } from './types.js'

const log = logger('@wdio/shared-store-service')

let server: SharedStoreServer

export default class SharedStoreLauncher implements Services.HookFunctions {
    private _app?: PolkaInstance

    async onPrepare (_: never, capabilities: Capabilities.TestrunnerCapabilities) {
        /**
         * import during runtime to avoid unnecessary dependency loading
         */
        server = (await import('./server.js')) as SharedStoreServer
        const { port, app } = await server.startServer()
        this._app = app
        setPort(port)

        const capsList = Array.isArray(capabilities)
            ? capabilities
            : Object.values(capabilities).map((multiremoteOption) => multiremoteOption.capabilities)

        const caps: Partial<SharedStoreServiceCapabilities>[] = capsList.flatMap((c) => {
            const multiremote = c as Capabilities.RequestedMultiremoteCapabilities
            if (!multiremote.browserName && multiremote[Object.keys(multiremote)[0]].capabilities) {
                return Object.values(multiremote).map((options) =>
                    (options.capabilities as Capabilities.W3CCapabilities)?.alwaysMatch ||
                    (options.capabilities as WebdriverIO.Capabilities)
                )
            }

            const w3cCaps = c as Capabilities.W3CCapabilities
            return w3cCaps.alwaysMatch || c as WebdriverIO.Capabilities
        })
        caps.forEach((c) => { c[CUSTOM_CAP] = port })
        log.info(`Started shared server on port ${port}`)
    }

    async onComplete () {
        return new Promise<void>((resolve) => {
            if (this._app && this._app.server.close) {
                this._app.server.close(() => resolve())
            }
            return resolve()
        })
    }
}