@prismicio/mock/src/model/linkToMedia.ts

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

import * as prismic from "@prismicio/client"

import { capitalCase, sentenceCase } from "../lib/changeCase"
import { createFaker } from "../lib/createFaker"
import type { MockModelConfig } from "../types"

type MockLinkToMediaModel<AllowText extends boolean = boolean> =
	prismic.CustomTypeModelLinkToMediaField & {
		config: AllowText extends true ? { allowText: true } : { allowText?: undefined }
	}

export type MockLinkToMediaModelConfig<AllowText extends boolean = boolean> = MockModelConfig & {
	allowText?: AllowText
}

export const linkToMedia = <AllowText extends boolean = boolean>(
	config: MockLinkToMediaModelConfig<AllowText>,
): MockLinkToMediaModel<AllowText> => {
	const faker = config.faker || createFaker(config.seed)

	return {
		type: prismic.CustomTypeModelFieldType.Link,
		config: {
			label: capitalCase(faker.word()),
			placeholder: sentenceCase(faker.words(3)),
			select: prismic.CustomTypeModelLinkSelectType.Media,
			allowText: ("allowText" in config ? config.allowText : faker.boolean()) || undefined,
		},
	} as MockLinkToMediaModel<AllowText>
}