import type {
	ContentRelationshipField,
	EmptyContentRelationshipField,
	FilledContentRelationshipField,
} from "../value/contentRelationship"
import type { PrismicDocument } from "../value/document"

import type { PrismicMigrationDocument } from "./Document"

type ValueOrThunk<T> = T | (() => Promise<T> | T)

/**
 * A content relationship field in a migration.
 */
export type MigrationContentRelationship<
	TDocuments extends PrismicDocument = PrismicDocument,
> =
	| ValueOrThunk<TDocuments | PrismicMigrationDocument<TDocuments> | undefined>
	| (Pick<ContentRelationshipField, "link_type"> & {
			id: ValueOrThunk<
				TDocuments | PrismicMigrationDocument<TDocuments> | undefined
			>
	  })

/**
 * The minimum amount of information needed to represent a content relationship
 * field with the migration API.
 */
export type MigrationContentRelationshipField =
	| Pick<FilledContentRelationshipField, "link_type" | "id">
	| EmptyContentRelationshipField