import React from 'react';
import type { Ref } from 'react';

import { NestedNavigation as CoreNestedNavigation } from '@shoptet/ui-core-web';
import type { NestedNavigationOwnProps as CoreNestedNavigationOwnProps } from '@shoptet/ui-core-web';
import type { Exact, SafeOmit } from '@shoptet/utils';

import { NestedNavigationContent } from './NestedNavigationContent';
import { NestedNavigationHeader } from './NestedNavigationHeader';
import { NestedNavigationItem } from './NestedNavigationItem';
import { NestedNavigationList } from './NestedNavigationList';

export interface NestedNavigationOwnProps<TKey extends string = string> extends CoreNestedNavigationOwnProps<TKey> {
  ref?: Ref<HTMLDivElement>;
  /**
   * Whether to render the standard header above the levels. Disable to compose
   * `NestedNavigation.Header` inside each `Level` instead.
   * @default true
   */
  renderHeader?: boolean;
}

export interface NestedNavigationInheritedDivProps extends SafeOmit<
  React.HTMLAttributes<HTMLDivElement>,
  'children' | 'className' | 'title'
> {}

export interface NestedNavigationProps<TKey extends string = string>
  extends NestedNavigationOwnProps<TKey>, NestedNavigationInheritedDivProps {}

export const NestedNavigation = Object.assign(
  function NestedNavigation<TKey extends string = string>({
    stack,
    defaultStack,
    onStackChange,
    onClose,
    children,
    ref,
    renderHeader = true,
    ...rest
  }: NestedNavigationProps<TKey>) {
    rest satisfies Exact<NestedNavigationInheritedDivProps, typeof rest>;

    return (
      <CoreNestedNavigation
        {...rest}
        ref={ref}
        stack={stack}
        defaultStack={defaultStack}
        onStackChange={onStackChange}
        onClose={onClose}
        className='nestedNavigation'
      >
        {renderHeader && <NestedNavigationHeader />}
        {children}
      </CoreNestedNavigation>
    );
  },
  {
    Content: NestedNavigationContent,
    Header: NestedNavigationHeader,
    Item: NestedNavigationItem,
    Level: CoreNestedNavigation.Level,
    List: NestedNavigationList,
  }
);