@prismicio/mock/src/value/select.ts

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

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

import { createFaker } from "../lib/createFaker"
import * as modelGen from "../model"
import type { MockValueStateConfig, MockValueConfig } from "../types"

export type MockSelectValueConfig<
	Model extends prismic.CustomTypeModelSelectField = prismic.CustomTypeModelSelectField,
	State extends prismic.FieldState = prismic.FieldState,
> = MockValueConfig<Model> & MockValueStateConfig<State>

export type MockSelectValue<
	Model extends prismic.CustomTypeModelSelectField = prismic.CustomTypeModelSelectField,
	State extends prismic.FieldState = prismic.FieldState,
> = prismic.SelectField<NonNullable<NonNullable<Model["config"]>["options"]>[number], State>

export const select = <
	Model extends prismic.CustomTypeModelSelectField = prismic.CustomTypeModelSelectField,
	State extends prismic.FieldState = "filled",
>(
	config: MockSelectValueConfig<Model, State>,
): MockSelectValue<Model, State> => {
	const faker = config.faker || createFaker(config.seed)

	const model = config.model || modelGen.select({ faker })
	const defaultValue = model.config?.default_value
	const options = model.config?.options || []

	if (config.state === "empty") {
		return null as MockSelectValue<Model, State>
	} else {
		return (
			typeof defaultValue === "string" && faker.boolean()
				? defaultValue
				: faker.randomElement(options)
		) as MockSelectValue<Model, State>
	}
}