import { z } from 'astro/zod';

const defaults = { minHeadingLevel: 2, maxHeadingLevel: 3 };

const TableOfContentsBaseSchema = z
	.union([
		z.object({
			/** The level to start including headings at in the table of contents. Default: 2. */
			minHeadingLevel: z.int().min(1).max(6).optional().default(2),
			/** The level to stop including headings at in the table of contents. Default: 3. */
			maxHeadingLevel: z.int().min(1).max(6).optional().default(3),
		}),
		z.boolean().transform((enabled) => (enabled ? defaults : false)),
	])
	.refine((toc) => (toc ? toc.minHeadingLevel <= toc.maxHeadingLevel : true), {
		error: 'minHeadingLevel must be less than or equal to maxHeadingLevel',
	});

export const UserConfigTableOfContentsSchema = () => TableOfContentsBaseSchema.default(defaults);

export const FrontmatterTableOfContentsSchema = () => TableOfContentsBaseSchema.optional();