import type { Locator, Page } from '@playwright/test'
import { expect, step } from './tla-test'

export class WorkspaceInviteDialog {
	public readonly dialog: Locator
	public readonly acceptButton: Locator

	constructor(public readonly page: Page) {
		this.dialog = this.page.getByText('You have been invited to join')
		this.acceptButton = this.page.getByRole('button', { name: 'Accept invitation' })
	}

	async isVisible() {
		return this.dialog.isVisible()
	}

	async expectIsVisible() {
		await expect(this.dialog).toBeVisible()
	}

	async expectIsNotVisible() {
		await expect(this.dialog).not.toBeVisible()
	}

	@step
	async acceptInvitation() {
		await this.acceptButton.click()
	}
}