@prismicio/mock/src/value/number.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 type { MockValueStateConfig, MockValueConfig } from "../types"

export type MockNumberValueConfig<
	Model extends prismic.CustomTypeModelNumberField = prismic.CustomTypeModelNumberField,
	State extends prismic.FieldState = prismic.FieldState,
> = MockValueConfig<Model> & MockValueStateConfig<State>

export type MockNumberValue<State extends prismic.FieldState = prismic.FieldState> =
	prismic.NumberField<State>

export const number = <
	Model extends prismic.CustomTypeModelNumberField = prismic.CustomTypeModelNumberField,
	State extends prismic.FieldState = "filled",
>(
	config: MockNumberValueConfig<Model, State>,
): MockNumberValue<State> => {
	const faker = config.faker || createFaker(config.seed)

	return (config.state === "empty" ? null : faker.range(1, 100)) as MockNumberValue<State>
}