import type { BrowserContext } from 'playwright';
import { RenderOptions } from './render';

export interface State {
  tests: Record<string, () => void>;
  currentSuite: string;
  currentTest: string;
  retryMap: Record<string, number>;
  __filename?: string;
  activeTest?: string;
  pendingExpects: Record<string, number>;
  seen: Set<string>;
  options: RenderOptions;
  isGlobalSetupTeardownRegistered: boolean;
  debugging: Set<string>;
  passedTests: Set<string>;
  exposeGlobals: Record<string, any>;
  bridge?: Function;
  pause?: Function;
  pauseAtEveryStep?: boolean;
  /**
   * Tests run way faster if we keep the window alive. Note: Only do this when not
   * in video mode else the video memory wreaks havoc.
   */
  browserContextInstance?: BrowserContext & {
    headless?: boolean;
  };
  nextIndex: number;
  debugPort?: number;
  redirectUrl?: string;
  /** What to do when everything is really done, like shutdown docker container */
  afterAllsDone: Array<() => Promise<void>>;
  artifacts: Array<{
    test: string;
    type: 'snapshot' | 'diff' | 'received' | 'trace' | 'video';
    path: string;
    confirmed: boolean;
  }>;
  isCi: boolean;
  artifactsJson?: string;
  bootstrappedAt: string;
  getState: () => ReturnType<typeof expect.getState>;
  testPath?: string;
  /**
   * When a test is being run we need to track which element to render, this is only useful for
   * the browser side of things
   */
  browserState?: {
    retryAttempt: number;
    // eslint-disable-next-line @typescript-eslint/no-explicit-any
    renderElement: { __type: 'renderElement'; value: any };
  };
}

export const state: State = {
  tests: {},
  retryMap: {},
  options: {},
  seen: new Set(),
  currentSuite: '',
  currentTest: '',
  isGlobalSetupTeardownRegistered: false,
  pendingExpects: {},
  exposeGlobals: {},
  debugging: new Set(),
  passedTests: new Set(),
  nextIndex: 0,
  afterAllsDone: [],
  artifacts: [],
  isCi: false,
  bootstrappedAt: '',
  getState: () => expect?.getState?.() ?? {},
};