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

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

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

import { createFaker } from "../lib/createFaker";

import { MockValueStateConfig, MockValueConfig } from "../types";

import * as modelGen from "../model";

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>;
	}
};