import { mergeProps } from 'zag-ripple';
import { trackSplit, track } from 'ripple';
import { useDatePickerContext } from './use-date-picker-context';
import { useDatePickerTablePropsContext } from './use-date-picker-table-props-context';
import { ark } from '../factory';
import type { HTMLProps, MaybeTracked, PolymorphicProps } from '../../types';

export interface DatePickerTableHeadBaseProps extends PolymorphicProps<'thead'> {}
export interface DatePickerTableHeadProps extends HTMLProps<
  'thead'
>, DatePickerTableHeadBaseProps {}

export component DatePickerTableHead(props: MaybeTracked<DatePickerTableHeadProps>) {
  const [children, localProps] = trackSplit(props, ['children']);
  const datePicker = useDatePickerContext();
  const tableProps = useDatePickerTablePropsContext();

  let mergedProps = track(
    () => mergeProps(@datePicker.getTableHeadProps(@tableProps), @localProps),
  );

  <ark.thead {...@mergedProps}>
    <@children />
  </ark.thead>
}