<script lang="ts">
	import * as Calendar from '$lib/components/ui/calendar/index.js';
	import * as Select from '$lib/components/ui/select/index.js';
	import { cn } from '$lib/utils/style';
	import { CalendarDate, DateFormatter, getLocalTimeZone, today } from '@internationalized/date';
	import { Calendar as CalendarPrimitive, type WithoutChildrenOrChild } from 'bits-ui';

	let {
		value = $bindable(),
		placeholder = $bindable()
	}: WithoutChildrenOrChild<CalendarPrimitive.RootProps> & {
		value: CalendarDate | undefined;
	} = $props();

	const currentDate = today(getLocalTimeZone());

	const monthFmt = new DateFormatter('en-US', {
		month: 'long'
	});

	const monthOptions = Array.from({ length: 12 }, (_, i) => {
		const month = currentDate.set({ month: i + 1 });
		return {
			value: month.month,
			label: monthFmt.format(month.toDate(getLocalTimeZone()))
		};
	});

	const yearOptions = Array.from({ length: 100 }, (_, i) => ({
		label: String(new Date().getFullYear() + i),
		value: new Date().getFullYear() + i
	}));

	const defaultYear = $derived(
		placeholder ? { value: placeholder.year, label: String(placeholder.year) } : undefined
	);

	const defaultMonth = $derived(
		placeholder
			? {
					value: placeholder.month,
					label: monthFmt.format(placeholder.toDate(getLocalTimeZone()))
				}
			: undefined
	);

	const monthLabel = $derived(
		monthOptions.find((m) => m.value === defaultMonth?.value)?.label ?? 'Select a month'
	);
</script>

<CalendarPrimitive.Root
	type="single"
	weekdayFormat="short"
	class={cn('rounded-md border p-3')}
	bind:value
	bind:placeholder
>
	{#snippet children({ months, weekdays })}
		<Calendar.Header class="flex w-full items-center justify-between gap-2">
			<Select.Root
				type="single"
				value={`${defaultMonth?.value}`}
				onValueChange={(v) => {
					if (!placeholder) return;
					if (v === `${placeholder.month}`) return;
					placeholder = placeholder.set({ month: Number.parseInt(v) });
				}}
			>
				<Select.Trigger aria-label="Select month" class="w-[60%]">
					{monthLabel}
				</Select.Trigger>
				<Select.Content class="max-h-[200px] overflow-y-auto">
					{#each monthOptions as { value, label } (value)}
						<Select.Item value={`${value}`} {label} />
					{/each}
				</Select.Content>
			</Select.Root>
			<Select.Root
				type="single"
				value={`${defaultYear?.value}`}
				onValueChange={(v) => {
					if (!v || !placeholder) return;
					if (v === `${placeholder?.year}`) return;
					placeholder = placeholder.set({ year: Number.parseInt(v) });
				}}
			>
				<Select.Trigger aria-label="Select year" class="w-[40%]">
					{defaultYear?.label ?? 'Select year'}
				</Select.Trigger>
				<Select.Content class="max-h-[200px] overflow-y-auto">
					{#each yearOptions as { value, label } (value)}
						<Select.Item value={`${value}`} {label} />
					{/each}
				</Select.Content>
			</Select.Root>
		</Calendar.Header>
		<Calendar.Months>
			{#each months as month (month)}
				<Calendar.Grid>
					<Calendar.GridHead>
						<Calendar.GridRow class="flex">
							{#each weekdays as weekday (weekday)}
								<Calendar.HeadCell>
									{weekday.slice(0, 2)}
								</Calendar.HeadCell>
							{/each}
						</Calendar.GridRow>
					</Calendar.GridHead>
					<Calendar.GridBody>
						{#each month.weeks as weekDates (weekDates)}
							<Calendar.GridRow class="mt-2 w-full">
								{#each weekDates as date (date)}
									<Calendar.Cell class="select-none" {date} month={month.value}>
										<Calendar.Day />
									</Calendar.Cell>
								{/each}
							</Calendar.GridRow>
						{/each}
					</Calendar.GridBody>
				</Calendar.Grid>
			{/each}
		</Calendar.Months>
	{/snippet}
</CalendarPrimitive.Root>
