import { test as playwrightTestFn } from '@playwright/test'
import {
  fixtures,
  Fixtures,
  AuthenticatedTronApp,
} from './fixtures/fixtureSetup'
export { expect, Page, BrowserContext, TestInfo } from '@playwright/test'

// Our custom decorated Zoo test object. Makes it easier to add fixtures, and
// switch between web and electron if needed.
const pwTestFnWithFixtures = playwrightTestFn.extend<Fixtures>(fixtures)

export function test(desc, objOrFn, fnMaybe) {
  const hasTestConf = typeof objOrFn === 'object'
  const fn = hasTestConf ? fnMaybe : objOrFn

  return pwTestFnWithFixtures(
    desc,
    hasTestConf ? objOrFn : {},
    async (
      { page, context, cmdBar, editor, toolbar, scene, homePage },
      testInfo
    ) => {
      // To switch to web, change this to AuthenticatedApp from fixtureSetup.ts
      const tronApp = new AuthenticatedTronApp(context, page, testInfo)

      const fixtures: Fixtures = { cmdBar, editor, toolbar, scene, homePage }
      await tronApp.initialise({ fixtures })

      // We need to patch this because addInitScript will bind too late in our
      // electron tests, never running. We need to call reload() after each call
      // to guarantee it runs.
      const oldContextAddInitScript = tronApp.context.addInitScript
      tronApp.context.addInitScript = async function (a, b) {
        await oldContextAddInitScript.apply(this, [a, b])
        await tronApp.page.reload()
      }

      // No idea why we mix and match page and context's addInitScript but we do
      const oldPageAddInitScript = tronApp.page.addInitScript
      tronApp.page.addInitScript = async function (a, b) {
        await oldPageAddInitScript.apply(this, [a, b])
        await tronApp.page.reload()
      }

      // Create a consistent way to resize the page across electron and web.
      tronApp.page.setBodyDimensions = async function (dims: {
        width: number
        height: number
      }) {
        return this.evaluate((dims) => {
          window.document.body.style.width = dims.width + 'px'
          window.document.body.style.height = dims.height + 'px'
        }, dims)
      }

      await fn(
        {
          context: tronApp.context,
          page: tronApp.page,
          ...fixtures,
        },
        testInfo
      )

      testInfo.tronApp = tronApp
    }
  )
}

test.describe = pwTestFnWithFixtures.describe
test.beforeEach = pwTestFnWithFixtures.beforeEach
test.afterEach = pwTestFnWithFixtures.afterEach
test.step = pwTestFnWithFixtures.step
test.skip = pwTestFnWithFixtures.skip
test.setTimeout = pwTestFnWithFixtures.setTimeout