import { type BrowserPage, commands } from '@vitest/browser/context'
import type { ImageSnapshotIdOptions } from '../../shared/types.ts'
import { ctx } from '../ctx.ts'
import { toTaskId } from '../task_id.ts'

export interface HasImageSnapshotAction {
	hasImageSnapshot(this: BrowserPage, options?: ImageSnapshotIdOptions | undefined): Promise<boolean>
}

/**
 * Check if the snapshot image exists.
 */
export function hasImageSnapshot(this: BrowserPage, options?: ImageSnapshotIdOptions | undefined) {
	const test = ctx.getCurrentTest()
	if (!test) {
		throw new Error('`hasImageSnapshot()` must be called in a test.')
	}

	if (test.concurrent) {
		throw new Error(
			'`hasImageSnapshot()` cannot be called in a concurrent test because ' +
				"concurrent tests run at the same time in the same iframe and affect each other's environment. ",
		)
	}
	const taskId = toTaskId(test)
	const isAutoSnapshot = !!test.meta.vis?.isAutoSnapshot
	if (options?.customizeSnapshotId) {
		return commands
			.imageSnapshotNextIndex(taskId)
			.then((index) =>
				commands.hasImageSnapshot(
					taskId,
					options.customizeSnapshotId!({ id: taskId, index, isAutoSnapshot }),
					isAutoSnapshot,
				),
			)
	}

	return commands.hasImageSnapshot(taskId, undefined, isAutoSnapshot)
}