@prismicio/mock/src/lib/buildEmbedField.ts

Generate mock Prismic documents, fields, Slices, and models for development and testing environments

import type * as prismic from "@prismicio/client"

import { createFaker, type Faker } from "../lib/createFaker"
import type { Seed } from "../types"

type BuildEmbedFieldConfig<Data extends prismic.AnyOEmbed = prismic.AnyOEmbed> = {
	url?: string
	html?: string
	data: Data
} & (
	| {
			seed: Seed
			faker?: never
	  }
	| {
			faker: Faker
			seed?: never
	  }
)

export const buildEmbedField = <Data extends prismic.AnyOEmbed = prismic.AnyOEmbed>(
	config: BuildEmbedFieldConfig<Data>,
): prismic.EmbedField<Data, "filled"> => {
	const faker = config.faker || createFaker(config.seed)

	return {
		embed_url: config.url ?? faker.url(),
		html: `<div>embed html</div>`,
		...config.data,
	} as prismic.EmbedField<Data, "filled">
}