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