Compass/tests/e2e/utils/contextManager.ts

import {Browser} from '@playwright/test'

import {App} from '../web/pages/app'

export class ContextManager {
  private contexts: Map<string, App> = new Map()

  constructor(private browser: Browser) {}

  async createContext(customName?: string): Promise<App> {
    const name = customName ?? crypto.randomUUID().slice(0, 6)
    const existing = this.contexts.get(name)
    // Return the existing one instead of closing it?
    if (existing) await existing.page.context().close()

    const context = await this.browser.newContext()
    const page = await context.newPage()
    const app = new App(page)
    this.contexts.set(name, app)
    return app
  }

  getContext(name: string): App | undefined {
    return this.contexts.get(name)
  }

  async closeAll(): Promise<void> {
    for (const app of this.contexts.values()) {
      await app.page.context().close()
    }
    this.contexts.clear()
  }
}