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()
    try {
      const page = await context.newPage()
      const app = new App(page, false)
      this.contexts.set(name, app)
      return app
    } catch (error) {
      await context.close()
      throw error
    }
  }

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

  async closeAll(): Promise<void> {
    for (const app of this.contexts.values()) {
      try {
        await app.page.context().close()
      } catch {
        // context may already be closed
      }
    }
    this.contexts.clear()
  }
}