@prismicio/mock/src/model/link.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 MockLinkModel<
	AllowTargetBlank extends boolean = boolean,
	AllowText extends boolean = boolean,
	Repeat extends boolean = boolean,
> = prismic.CustomTypeModelLinkField & {
	config: AllowTargetBlank extends true
		? { allowTargetBlank: true }
		: { allowTargetBlank?: undefined }
} & {
	config: AllowText extends true ? { allowText: true } : { allowText?: undefined }
} & {
	config: Repeat extends true ? { repeat: true } : { repeat?: undefined }
}

export type MockLinkModelConfig<
	AllowTargetBlank extends boolean = boolean,
	AllowText extends boolean = boolean,
	Repeat extends boolean = boolean,
> = {
	allowTargetBlank?: AllowTargetBlank
	allowText?: AllowText
	repeat?: Repeat
} & MockModelConfig

export const link = <
	AllowTargetBlank extends boolean = boolean,
	AllowText extends boolean = boolean,
	Repeat extends boolean = boolean,
>(
	config: MockLinkModelConfig<AllowTargetBlank, AllowText, Repeat>,
): MockLinkModel<AllowTargetBlank, AllowText, Repeat> => {
	const faker = config.faker || createFaker(config.seed)

	return {
		type: prismic.CustomTypeModelFieldType.Link,
		config: {
			label: capitalCase(faker.word()),
			placeholder: sentenceCase(faker.words(3)),
			select: null,
			allowTargetBlank:
				("allowTargetBlank" in config ? config.allowTargetBlank : faker.boolean()) || undefined,
			allowText: ("allowText" in config ? config.allowText : faker.boolean()) || undefined,
			repeat: ("repeat" in config ? config.repeat : faker.boolean()) || undefined,
		},
	} as MockLinkModel<AllowTargetBlank, AllowText, Repeat>
}