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

import {
    getBrowserDescription,
    getBrowserCapabilities,
    isBrowserstackCapability,
    getParentSuiteName,
    isBrowserstackSession,
    patchConsoleLogs,
    isTrue
} from './util.js'
import type { BrowserstackConfig, BrowserstackOptions, MultiRemoteAction } from './types.js'
import type { Pickle, Feature, ITestCaseHookParameter, CucumberHook } from './cucumber-types.js'
import InsightsHandler from './insights-handler.js'
import TestReporter from './reporter.js'
import { DEFAULT_OPTIONS, PERF_MEASUREMENT_ENV } from './constants.js'
import CrashReporter from './crash-reporter.js'
import AccessibilityHandler from './accessibility-handler.js'
import { BStackLogger } from './bstackLogger.js'
import PercyHandler from './Percy/Percy-Handler.js'
import Listener from './testOps/listener.js'
import { saveWorkerData } from './data-store.js'
import UsageStats from './testOps/usageStats.js'
import { shouldProcessEventForTesthub } from './testHub/utils.js'
import AiHandler from './ai-handler.js'
import PerformanceTester from './instrumentation/performance/performance-tester.js'
import * as PERFORMANCE_SDK_EVENTS from './instrumentation/performance/constants.js'

import { _fetch as fetch } from './fetchWrapper.js'

export default class BrowserstackService implements Services.ServiceInstance {
    private _sessionBaseUrl = 'https://api.browserstack.com/automate/sessions'
    private _failReasons: string[] = []
    private _hookFailReasons: string[] = []
    private _pureTestFailReasons: string[] = []
    private _scenariosThatRan: string[] = []
    private _failureStatuses: string[] = ['failed', 'ambiguous', 'undefined', 'unknown']
    private _browser?: WebdriverIO.Browser
    private _suiteTitle?: string
    private _suiteFile?: string
    private _fullTitle?: string
    private _options: BrowserstackConfig & BrowserstackOptions
    private _specsRan: boolean = false
    private _observability
    private _currentTest?: Frameworks.Test | ITestCaseHookParameter
    private _insightsHandler?: InsightsHandler
    private _accessibility
    private _accessibilityHandler?: AccessibilityHandler
    private _percy
    private _percyCaptureMode: string | undefined = undefined
    private _percyHandler?: PercyHandler
    private _turboScale

    constructor (
        options: BrowserstackConfig & Options.Testrunner,
        private _caps: Capabilities.ResolvedTestrunnerCapabilities,
        private _config: Options.Testrunner
    ) {
        this._options = { ...DEFAULT_OPTIONS, ...options }
        // added to maintain backward compatibility with webdriverIO v5
        if (!this._config) {
            this._config = this._options
        }

        this._observability = this._options.testObservability
        this._accessibility = this._options.accessibility
        this._percy = isTrue(process.env.BROWSERSTACK_PERCY)
        this._percyCaptureMode = process.env.BROWSERSTACK_PERCY_CAPTURE_MODE
        this._turboScale = this._options.turboScale

        PerformanceTester.startMonitoring('performance-report-service.csv')
        if (shouldProcessEventForTesthub('')) {
            this._config.reporters?.push(TestReporter)
        }

        if (process.env.BROWSERSTACK_TURBOSCALE) {
            this._turboScale = process.env.BROWSERSTACK_TURBOSCALE === 'true'
        }
        process.env.BROWSERSTACK_TURBOSCALE_INTERNAL = String(this._turboScale)

        // Cucumber specific
        const strict = Boolean(this._config.cucumberOpts && this._config.cucumberOpts.strict)
        // See https://github.com/cucumber/cucumber-js/blob/master/src/runtime/index.ts#L136
        if (strict) {
            this._failureStatuses.push('pending')
        }

        if (process.env.WDIO_WORKER_ID === process.env.BEST_PLATFORM_CID) {
            process.env.PERCY_SNAPSHOT = 'true'
        }
    }

    _updateCaps (fn: (caps: WebdriverIO.Capabilities) => void) {
        const multiRemoteCap = this._caps as Capabilities.RequestedMultiremoteCapabilities

        if (multiRemoteCap.capabilities) {
            return Object.entries(multiRemoteCap).forEach(([, caps]) => fn(caps.capabilities as WebdriverIO.Capabilities))
        }

        return fn(this._caps as WebdriverIO.Capabilities)
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'beforeSession' })
    beforeSession (config: Options.Testrunner) {
        // if no user and key is specified even though a browserstack service was
        // provided set user and key with values so that the session request
        // will fail
        const testObservabilityOptions = this._options.testObservabilityOptions
        if (!config.user && !(testObservabilityOptions && testObservabilityOptions.user)) {
            config.user = 'NotSetUser'
        }

        if (!config.key && !(testObservabilityOptions && testObservabilityOptions.key)) {
            config.key = 'NotSetKey'
        }
        this._config.user = config.user
        this._config.key = config.key
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'before' })
    async before(caps: Capabilities.ResolvedTestrunnerCapabilities, specs: string[], browser: WebdriverIO.Browser) {
        // added to maintain backward compatibility with webdriverIO v5
        this._browser = browser ? browser : globalThis.browser
        PerformanceTester.browser = this._browser

        // Healing Support:
        if (!isBrowserstackSession(this._browser)) {
            try {
                await AiHandler.selfHeal(this._options, caps, this._browser)
            } catch (err) {
                if (this._options.selfHeal === true) {
                    BStackLogger.warn(`Error while setting up self-healing: ${err}. Disabling healing for this session.`)
                }
            }
        }

        // Ensure capabilities are not null in case of multiremote

        if (this._isAppAutomate()) {
            this._sessionBaseUrl = 'https://api-cloud.browserstack.com/app-automate/sessions'
        }

        if (this._turboScale) {
            this._sessionBaseUrl = 'https://api.browserstack.com/automate-turboscale/v1/sessions'
        }

        this._scenariosThatRan = []
        PerformanceTester.scenarioThatRan = this._scenariosThatRan

        if (this._browser) {
            try {
                const sessionId = this._browser.sessionId

                try {
                    this._accessibilityHandler = new AccessibilityHandler(
                        this._browser,
                        this._caps,
                        this._options,
                        this._isAppAutomate(),
                        this._config,
                        this._config.framework,
                        this._accessibility,
                        this._turboScale,
                        this._options.accessibilityOptions
                    )
                    await this._accessibilityHandler.before(sessionId)

                    Listener.setAccessibilityOptions(this._options.accessibilityOptions)
                } catch (err) {
                    BStackLogger.error(`[Accessibility Test Run] Error in service class before function: ${err}`)
                }

                if (shouldProcessEventForTesthub('')) {
                    patchConsoleLogs()

                    this._insightsHandler = new InsightsHandler(
                        this._browser,
                        this._config.framework,
                        this._caps,
                        this._options
                    )
                    await this._insightsHandler.before()
                }

                /**
                 * register command event
                 */
                this._browser.on('command', async (command) => {
                    if (shouldProcessEventForTesthub('')) {
                        this._insightsHandler?.browserCommand(
                            'client:beforeCommand',
                            Object.assign(command, { sessionId }),
                            this._currentTest
                        )
                    }
                    await this._percyHandler?.browserBeforeCommand(
                        Object.assign(command, { sessionId }),
                    )
                })

                /**
                 * register result event
                 */
                this._browser.on('result', (result) => {
                    if (shouldProcessEventForTesthub('')) {
                        this._insightsHandler?.browserCommand(
                            'client:afterCommand',
                            Object.assign(result, { sessionId }),
                            this._currentTest
                        )
                    }
                    this._percyHandler?.browserAfterCommand(
                        Object.assign(result, { sessionId }),
                    )
                })
            } catch (err) {
                BStackLogger.error(`Error in service class before function: ${err}`)
                if (shouldProcessEventForTesthub('')) {
                    CrashReporter.uploadCrashReport(`Error in service class before function: ${err}`, err ? (err as Error).stack as string : 'unknown error')
                }
            }

            if (this._percy) {
                this._percyHandler = new PercyHandler(
                    this._percyCaptureMode,
                    this._browser,
                    this._caps,
                    this._isAppAutomate(),
                    this._config.framework
                )
                this._percyHandler.before()
            }
        }

        return await this._printSessionURL()
    }

    /**
     * Set the default job name at the suite level to make sure we account
     * for the cases where there is a long running `before` function for a
     * suite or one that can fail.
     * Don't do this for Jasmine because `suite.title` is `Jasmine__TopLevel__Suite`
     * and `suite.fullTitle` is `undefined`, so no alternative to use for the job name.
     */
    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'beforeSuite' })
    async beforeSuite (suite: Frameworks.Suite) {
        this._suiteTitle = suite.title
        this._insightsHandler?.setSuiteFile(suite.file)
        this._accessibilityHandler?.setSuiteFile(suite.file)

        if (suite.title && suite.title !== 'Jasmine__TopLevel__Suite') {
            await this._setSessionName(suite.title)
        }
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'beforeHook' })
    async beforeHook (test: Frameworks.Test|CucumberHook, context: unknown) {
        if (this._config.framework !== 'cucumber') {
            this._currentTest = test as Frameworks.Test // not update currentTest when this is called for cucumber step
        }
        await this._insightsHandler?.beforeHook(test, context)
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'afterHook' })
    async afterHook(test: Frameworks.Test | CucumberHook, context: unknown, result: Frameworks.TestResult) {
        // Track hook failures separately
        if (result && !result.passed) {
            const hookError = (result.error && result.error.message) || 'Hook failed'
            this._hookFailReasons.push(hookError)

            // Still add to main failReasons for backward compatibility if ignoreHooksStatus is not enabled
            if (!this._options.testObservabilityOptions?.ignoreHooksStatus) {
                this._failReasons.push(hookError)
            }
        }
        await this._insightsHandler?.afterHook(test, result)
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'beforeTest' })
    async beforeTest (test: Frameworks.Test) {
        this._currentTest = test
        let suiteTitle = this._suiteTitle

        if (test.fullName) {
            // For Jasmine, `suite.title` is `Jasmine__TopLevel__Suite`.
            // This tweak allows us to set the real suite name.
            const testSuiteName = test.fullName.slice(0, test.fullName.indexOf(test.description || '') - 1)
            if (this._suiteTitle === 'Jasmine__TopLevel__Suite') {
                suiteTitle = testSuiteName
            } else if (this._suiteTitle) {
                suiteTitle = getParentSuiteName(this._suiteTitle, testSuiteName)
            }
        }

        await this._setSessionName(suiteTitle, test)
        await this._setAnnotation(`Test: ${test.fullName ?? test.title}`)
        await this._accessibilityHandler?.beforeTest(suiteTitle, test)
        await this._insightsHandler?.beforeTest(test)
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'afterTest' })
    async afterTest(test: Frameworks.Test, context: never, results: Frameworks.TestResult) {
        this._specsRan = true
        const { error, passed } = results
        if (!passed) {
            const testError = (error && error.message) || 'Unknown Error'
            this._failReasons.push(testError)

            // Track this as a pure test failure (not hook-related)
            this._pureTestFailReasons.push(testError)
        }
        await this._accessibilityHandler?.afterTest(this._suiteTitle, test)
        await this._insightsHandler?.afterTest(test, results)
        await this._percyHandler?.afterTest()
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'after' })
    async after (result: number) {
        const { preferScenarioName, setSessionName, setSessionStatus } = this._options
        // For Cucumber: Checks scenarios that ran (i.e. not skipped) on the session
        // Only 1 Scenario ran and option enabled => Redefine session name to Scenario's name
        if (preferScenarioName && this._scenariosThatRan.length === 1){
            this._fullTitle = this._scenariosThatRan.pop()
        }

        // if (setSessionStatus) {
        //     const hasReasons = this._failReasons.length > 0
        //     await this._updateJob({
        //         status: result === 0 && this._specsRan ? 'passed' : 'failed',
        //         ...(setSessionName ? { name: this._fullTitle } : {}),
        //         ...(result === 0 && this._specsRan ?
        //             {} : hasReasons ? { reason: this._failReasons.join('\n') } : {})
        //     })
        // }

        await PerformanceTester.measureWrapper(PERFORMANCE_SDK_EVENTS.AUTOMATE_EVENTS.SESSION_STATUS, async () => {
            if (setSessionStatus) {
                const ignoreHooksStatus = this._options.testObservabilityOptions?.ignoreHooksStatus === true
                let sessionStatus: string
                let failureReason: string | undefined

                if (result === 0 && this._specsRan) {
                    // Test runner reported success and tests ran
                    if (ignoreHooksStatus) {
                        // Only consider pure test failures, ignore hook failures
                        const hasPureTestFailures = this._pureTestFailReasons.length > 0
                        sessionStatus = hasPureTestFailures ? 'failed' : 'passed'
                        failureReason = hasPureTestFailures ? this._pureTestFailReasons.join('\n') : undefined
                    } else {
                        // Default behavior: consider all failures including hooks
                        const hasReasons = this._failReasons.length > 0
                        sessionStatus = hasReasons ? 'failed' : 'passed'
                        failureReason = hasReasons ? this._failReasons.join('\n') : undefined
                    }
                } else if (ignoreHooksStatus && this._specsRan) {
                    // Test runner reported failure but ignoreHooksStatus is enabled
                    // Check if we only have hook failures and no pure test failures
                    const hasPureTestFailures = this._pureTestFailReasons.length > 0
                    const hasOnlyHookFailures = this._failReasons.length === 0 && this._hookFailReasons.length > 0

                    if (hasOnlyHookFailures && !hasPureTestFailures) {
                        // Only hook failures exist - mark as passed when ignoreHooksStatus is true
                        sessionStatus = 'passed'
                        failureReason = undefined
                    } else {
                        // Pure test failures exist - mark as failed
                        sessionStatus = 'failed'
                        failureReason = hasPureTestFailures ? this._pureTestFailReasons.join('\n') : undefined
                    }
                } else {
                    // Default behavior: mark as failed (test runner reported failure or no tests ran)
                    sessionStatus = 'failed'
                    if (ignoreHooksStatus && this._pureTestFailReasons.length > 0) {
                        failureReason = this._pureTestFailReasons.join('\n')
                    } else if (this._failReasons.length > 0) {
                        failureReason = this._failReasons.join('\n')
                    } else {
                        failureReason = undefined
                    }
                }

                await this._updateJob({
                    status: sessionStatus,
                    ...(setSessionName ? { name: this._fullTitle } : {}),
                    ...(failureReason ? { reason: failureReason } : {})
                })
            }
        })()

        await Listener.getInstance().onWorkerEnd()
        await this._percyHandler?.teardown()
        this.saveWorkerData()

        await PerformanceTester.stopAndGenerate('performance-service.html')
        if (process.env[PERF_MEASUREMENT_ENV]) {
            PerformanceTester.calculateTimes([
                'onRunnerStart', 'onSuiteStart', 'onSuiteEnd',
                'onTestStart', 'onTestEnd', 'onTestSkip', 'before',
                'beforeHook', 'afterHook', 'beforeTest', 'afterTest',
                'uploadPending', 'teardown', 'browserCommand'
            ])
        }

        // Override process exit when we have only hook failures and ignoreHooksStatus is true
        const ignoreHooksStatus = this._options.testObservabilityOptions?.ignoreHooksStatus === true
        const hasOnlyHookFailures = this._failReasons.length === 0 && this._hookFailReasons.length > 0
        const shouldOverrideResult = ignoreHooksStatus && this._specsRan && hasOnlyHookFailures

        if (shouldOverrideResult && result !== 0) {
            return
        }
    }

    /**
     * For CucumberJS
     */

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'beforeFeature' })
    async beforeFeature(uri: string, feature: Feature) {
        this._suiteTitle = feature.name
        await this._setSessionName(feature.name)
        await this._setAnnotation(`Feature: ${feature.name}`)
        await this._insightsHandler?.beforeFeature(uri, feature)
    }

    /**
     * Runs before a Cucumber Scenario.
     * @param world world object containing information on pickle and test step
     */
    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'beforeScenario' })
    async beforeScenario (world: ITestCaseHookParameter) {
        this._currentTest = world
        await this._accessibilityHandler?.beforeScenario(world)
        await this._insightsHandler?.beforeScenario(world)
        const scenarioName = world.pickle.name || 'unknown scenario'
        await this._setAnnotation(`Scenario: ${scenarioName}`)
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'afterScenario' })
    async afterScenario (world: ITestCaseHookParameter) {
        this._specsRan = true
        const status = world.result?.status.toLowerCase()
        if (status !== 'skipped') {
            this._scenariosThatRan.push(world.pickle.name || 'unknown pickle name')
        }

        if (status && this._failureStatuses.includes(status)) {
            const exception = (
                (world.result && world.result.message) ||
                (status === 'pending'
                    ? `Some steps/hooks are pending for scenario "${world.pickle.name}"`
                    : 'Unknown Error'
                )
            )

            // For Cucumber with ignoreHooksStatus, check if failure is due to test steps or hooks
            const ignoreHooksStatus = this._options.testObservabilityOptions?.ignoreHooksStatus === true
            if (ignoreHooksStatus && this._insightsHandler) {
                // Check if any test steps failed (excluding hook failures)
                const hasTestStepFailures = this._insightsHandler.hasTestStepFailures(world)
                if (hasTestStepFailures) {
                    // Test steps failed - this is a pure test failure, add to both arrays
                    this._failReasons.push(exception)
                    this._pureTestFailReasons.push(exception)
                }
                // If no test steps failed, this is likely a hook-only failure - don't add to main failure arrays
            } else {
                // Default behavior: treat all scenario failures as test failures
                this._failReasons.push(exception)
                this._pureTestFailReasons.push(exception)
            }
        }

        await this._accessibilityHandler?.afterScenario(world)
        await this._insightsHandler?.afterScenario(world)
        await this._percyHandler?.afterScenario()
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'beforeStep' })
    async beforeStep (step: Frameworks.PickleStep, scenario: Pickle) {
        await this._insightsHandler?.beforeStep(step, scenario)
        await this._setAnnotation(`Step: ${step.keyword}${step.text}`)
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'afterStep' })
    async afterStep (step: Frameworks.PickleStep, scenario: Pickle, result: Frameworks.PickleResult) {
        await this._insightsHandler?.afterStep(step, scenario, result)
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.EVENTS.SDK_HOOK, { hookType: 'onReload' })
    async onReload(oldSessionId: string, newSessionId: string) {
        if (!this._browser) {
            return Promise.resolve()
        }

        const { setSessionName, setSessionStatus } = this._options
        const ignoreHooksStatus = this._options.testObservabilityOptions?.ignoreHooksStatus === true

        let sessionStatus: string
        let failureReason: string | undefined

        if (ignoreHooksStatus) {
            // Only consider pure test failures, ignore hook failures
            const hasPureTestFailures = this._pureTestFailReasons.length > 0
            sessionStatus = hasPureTestFailures ? 'failed' : 'passed'
            failureReason = hasPureTestFailures ? this._pureTestFailReasons.join('\n') : undefined
        } else {
            // Default behavior: consider all failures including hooks
            const hasReasons = this._failReasons.length > 0
            sessionStatus = hasReasons ? 'failed' : 'passed'
            failureReason = hasReasons ? this._failReasons.join('\n') : undefined
        }

        if (!this._browser.isMultiremote) {
            BStackLogger.info(`Update (reloaded) job with sessionId ${oldSessionId}, ${sessionStatus}`)
        } else {
            const browserName = (this._browser as unknown as WebdriverIO.MultiRemoteBrowser).instances.filter(
                (browserName: string) => this._browser && (this._browser as unknown as WebdriverIO.MultiRemoteBrowser).getInstance(browserName).sessionId === newSessionId)[0]
            BStackLogger.info(`Update (reloaded) multiremote job for browser "${browserName}" and sessionId ${oldSessionId}, ${sessionStatus}`)
        }

        BStackLogger.warn(`Session Reloaded: Old Session Id: ${oldSessionId}, New Session Id: ${newSessionId}`)
        await this._insightsHandler?.sendCBTInfo()

        if (setSessionStatus) {
            await this._update(oldSessionId, {
                status: sessionStatus,
                ...(setSessionName ? { name: this._fullTitle } : {}),
                ...(failureReason ? { reason: failureReason } : {})
            })
        }

        this._scenariosThatRan = []
        delete this._fullTitle
        delete this._suiteFile
        this._failReasons = []
        this._hookFailReasons = []
        this._pureTestFailReasons = []
        await this._printSessionURL()
    }

    _isAppAutomate(): boolean {
        const browserDesiredCapabilities = (this._browser?.capabilities ?? {})
        const desiredCapabilities = (this._caps ?? {}) as WebdriverIO.Capabilities
        return (
            !!browserDesiredCapabilities['appium:app'] ||
            !!desiredCapabilities['appium:app'] ||
            !!desiredCapabilities['appium:options']?.app
        )
    }

    _updateJob (requestBody: unknown) {
        return this._multiRemoteAction((sessionId: string, browserName: string) => {
            BStackLogger.info(browserName
                ? `Update multiremote job for browser "${browserName}" and sessionId ${sessionId}`
                : `Update job with sessionId ${sessionId}`
            )
            return this._update(sessionId, requestBody)
        })
    }

    _multiRemoteAction (action: MultiRemoteAction) {
        if (!this._browser) {
            return Promise.resolve()
        }

        if (!this._browser.isMultiremote) {
            return action(this._browser.sessionId)
        }

        const multiremotebrowser = this._browser as unknown as WebdriverIO.MultiRemoteBrowser
        return Promise.all(multiremotebrowser.instances
            .filter((browserName: string) => {
                const cap = getBrowserCapabilities(multiremotebrowser, this._caps, browserName)
                return isBrowserstackCapability(cap)
            })
            .map((browserName: string) => (
                action(multiremotebrowser.getInstance(browserName).sessionId, browserName)
            ))
        )
    }

    _update(sessionId: string, requestBody: unknown) {
        if (!isBrowserstackSession(this._browser)) {
            return Promise.resolve()
        }
        const sessionUrl = `${this._sessionBaseUrl}/${sessionId}.json`
        BStackLogger.debug(`Updating Browserstack session at ${sessionUrl} with request body: `, requestBody)

        const encodedAuth = Buffer.from(`${this._config.user}:${this._config.key}`, 'utf8').toString('base64')
        const headers: Record<string, string> = {
            'Content-Type': 'application/json; charset=utf-8',
            Authorization: `Basic ${encodedAuth}`,
        }

        if (this._turboScale) {
            return fetch(sessionUrl, {
                method: 'PATCH',
                body: JSON.stringify(requestBody),
                headers
            })
        }
        return fetch(sessionUrl, {
            method: 'PUT',
            body: JSON.stringify(requestBody),
            headers
        })
    }

    @PerformanceTester.Measure(PERFORMANCE_SDK_EVENTS.AUTOMATE_EVENTS.PRINT_BUILDLINK)
    async _printSessionURL() {
        if (!this._browser || !isBrowserstackSession(this._browser)) {
            return Promise.resolve()
        }
        await this._multiRemoteAction(async (sessionId, browserName) => {
            const sessionUrl = `${this._sessionBaseUrl}/${sessionId}.json`
            BStackLogger.debug(`Requesting Browserstack session URL at ${sessionUrl}`)

            let browserUrl

            const encodedAuth = Buffer.from(`${this._config.user}:${this._config.key}`, 'utf8').toString('base64')
            const headers: Record<string, string> = {
                'Content-Type': 'application/json; charset=utf-8',
                Authorization: `Basic ${encodedAuth}`,
            }

            if (this._turboScale) {
                const response = await fetch(sessionUrl, {
                    method: 'GET',
                    headers
                })
                const res = response.clone()
                browserUrl = (await res.json()).url
            } else {
                const response = await fetch(sessionUrl, {
                    method: 'GET',
                    headers
                })
                const res = response.clone()
                browserUrl = (await res.json()).automation_session.browser_url
            }

            if (!this._browser) {
                return
            }

            const capabilities = getBrowserCapabilities(this._browser, this._caps, browserName)
            const browserString = getBrowserDescription(capabilities)
            BStackLogger.info(`${browserString} session: ${browserUrl}`)
        })
    }

    private async _setSessionName(suiteTitle: string | undefined, test?: Frameworks.Test) {
        if (!this._options.setSessionName || !suiteTitle) {
            return
        }

        let name = suiteTitle
        if (this._options.sessionNameFormat) {
            name = this._options.sessionNameFormat(
                this._config,
                this._caps,
                suiteTitle,
                test?.title
            )
        } else if (test && !test.fullName) {
            // Mocha
            const pre = this._options.sessionNamePrependTopLevelSuiteTitle ? `${suiteTitle} - ` : ''
            const post = !this._options.sessionNameOmitTestTitle ? ` - ${test.title}` : ''
            name = `${pre}${test.parent}${post}`
        }

        this._percyHandler?._setSessionName(name)

        if (name !== this._fullTitle) {
            this._fullTitle = name
            await this._updateJob({ name })
        }
    }

    private _setAnnotation(data: string) {
        return this._executeCommand('annotate', { data, level: 'info' })
    }

    private async _executeCommand(
        action: string,
        args?: object,
    ) {
        if (!this._browser || !isBrowserstackSession(this._browser)) {
            return Promise.resolve()
        }

        const cmd = { action, ...(args ? { arguments: args } : {}) }
        const script = `browserstack_executor: ${JSON.stringify(cmd)}`

        if (this._browser.isMultiremote) {
            const multiRemoteBrowser = this._browser as unknown as WebdriverIO.MultiRemoteBrowser
            return Promise.all(Object.keys(this._caps).map(async (browserName) => {
                const browser = multiRemoteBrowser.getInstance(browserName)
                return (await browser.executeScript(script, []))
            }))
        }

        return (await this._browser.executeScript(script, []))
    }

    private saveWorkerData() {
        saveWorkerData({
            usageStats: UsageStats.getInstance().getDataToSave()
        })
    }
}