/**
 * Copyright Microsoft Corporation. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { config, fixtures as baseFixtures } from '@playwright/test-runner';
import type { Browser, BrowserContext, BrowserContextOptions, BrowserType, LaunchOptions, Page } from '../index';
import * as path from 'path';

// Worker fixture declarations -------------------------------------------------
// ... these live as long as the worker process.

type PlaywrightWorkerFixtures = {
  // Playwright library.
  playwright: typeof import('../index');
  // Browser type (Chromium / WebKit / Firefox)
  browserType: BrowserType<Browser>;
  // Default browserType.launch() options.
  defaultBrowserOptions: LaunchOptions;
  // Browser instance, shared for the worker.
  browser: Browser;
  // True iff browserName is Chromium
  isChromium: boolean;
  // True iff browserName is Firefox
  isFirefox: boolean;
  // True iff browserName is WebKit
  isWebKit: boolean;
  // True iff running on Windows.
  isWindows: boolean;
  // True iff running on Mac.
  isMac: boolean;
  // True iff running on Linux.
  isLinux: boolean;
};


// Test fixture definitions, those are created for each test ------------------

type PlaywrightTestFixtures = {
  // Default browser.newContext() options.
  defaultContextOptions: BrowserContextOptions;
  // Factory for creating a context with given additional options.
  contextFactory: (options?: BrowserContextOptions) => Promise<BrowserContext>;
  // Context instance for test.
  context: BrowserContext;
  // Page instance for test.
  page: Page;
};

export const fixtures = baseFixtures
    .defineParameter('browserName', 'Browser type name', (process.env.BROWSER || 'chromium') as 'chromium' | 'firefox' | 'webkit')
    .defineParameter('headful', 'Whether to run tests headless or headful', process.env.HEADFUL ? true : false)
    .defineParameter('platform', 'Operating system', process.platform as ('win32' | 'linux' | 'darwin'))
    .defineParameter('screenshotOnFailure', 'Generate screenshot on failure', false)
    .defineParameter('slowMo', 'Slows down Playwright operations by the specified amount of milliseconds', 0)
    .defineParameter('trace', 'Whether to record the execution trace', !!process.env.TRACING || false)
    .defineWorkerFixtures<PlaywrightWorkerFixtures>({
      defaultBrowserOptions: async ({ headful, slowMo }, runTest) => {
        await runTest({
          handleSIGINT: false,
          slowMo,
          headless: !headful,
        });
      },

      playwright: async ({ }, runTest) => {
        const playwright = require('../index');
        await runTest(playwright);
      },

      browserType: async ({ playwright, browserName }, runTest) => {
        const browserType = (playwright as any)[browserName];
        await runTest(browserType);
      },

      browser: async ({ browserType, defaultBrowserOptions }, runTest) => {
        const browser = await browserType.launch(defaultBrowserOptions);
        await runTest(browser);
        await browser.close();
      },

      isChromium: async ({ browserName }, runTest) => {
        await runTest(browserName === 'chromium');
      },

      isFirefox: async ({ browserName }, runTest) => {
        await runTest(browserName === 'firefox');
      },

      isWebKit: async ({ browserName }, runTest) => {
        await runTest(browserName === 'webkit');
      },

      isWindows: async ({ platform }, test) => {
        await test(platform === 'win32');
      },

      isMac: async ({ platform }, test) => {
        await test(platform === 'darwin');
      },

      isLinux: async ({ platform }, test) => {
        await test(platform === 'linux');
      },
    })
    .defineTestFixtures<PlaywrightTestFixtures>({
      defaultContextOptions: async ({ trace, testInfo }, runTest) => {
        if (trace || testInfo.retry) {
          await runTest({
            _traceResourcesPath: path.join(config.outputDir, 'trace-resources'),
            _tracePath: testInfo.outputPath('playwright.trace'),
            videosPath: testInfo.outputPath(''),
          } as any);
        } else {
          await runTest({});
        }
      },

      contextFactory: async ({ browser, defaultContextOptions, testInfo, screenshotOnFailure }, runTest) => {
        const contexts: BrowserContext[] = [];
        async function contextFactory(options: BrowserContextOptions = {}) {
          const context = await browser.newContext({ ...defaultContextOptions, ...options });
          contexts.push(context);
          return context;
        }
        await runTest(contextFactory);

        if (screenshotOnFailure && (testInfo.status !== testInfo.expectedStatus)) {
          let ordinal = 0;
          for (const context of contexts) {
            for (const page of context.pages())
              await page.screenshot({ timeout: 5000, path: testInfo.outputPath + `-test-failed-${++ordinal}.png` });
          }
        }
        for (const context of contexts)
          await context.close();
      },

      context: async ({ contextFactory }, runTest) => {
        const context = await contextFactory();
        await runTest(context);
      // Context factory is taking care of closing the context,
      // so that it could capture a screenshot on failure.
      },

      page: async ({ context }, runTest) => {
      // Always create page off context so that they matched.
        await runTest(await context.newPage());
      // Context fixture is taking care of closing the page.
      },
    })
    .overrideTestFixtures({
      testParametersPathSegment: async ({ browserName, platform }, runTest) => {
        await runTest(browserName + '-' + platform);
      }
    });

// If browser is not specified, we are running tests against all three browsers.
fixtures.generateParametrizedTests(
    'browserName',
    process.env.BROWSER ? [process.env.BROWSER] as any : ['chromium', 'webkit', 'firefox']);