import * as React from 'react';
import { ChevronLeftIcon, ChevronRightIcon, MoreHorizontalIcon } from 'lucide-react';

import { cn } from '../../shared/utils';
import { Button, buttonVariants } from '../button';

/**
 * Page navigation controls for paginated data sets.
 *
 * @description
 * Composes horizontal pagination with numbered links, ellipses, and
 * Previous/Next controls. Follows ARIA navigation landmark patterns.
 *
 * @ai-rules
 * 1. Compose in order: `Pagination > PaginationContent > PaginationItem > (PaginationLink | PaginationNext | PaginationPrevious)`.
 * 2. In SPAs, set `href="#"` on `<PaginationLink>` and handle page change via `onClick` to prevent native page reloads.
 * 3. Mark the currently active page with `isActive={true}` on `<PaginationLink>`.
 */
interface PaginationProps extends React.ComponentProps<'nav'> {
  /** Aria-label for the navigation landmark. @default "pagination" */
  navLabel?: string;
}

function Pagination({ className, navLabel = 'pagination', ...props }: PaginationProps) {
  return (
    <nav
      role="navigation"
      aria-label={navLabel}
      data-slot="pagination"
      className={cn('mx-auto flex w-full justify-center', className)}
      {...props}
    />
  );
}

function PaginationContent({ className, ...props }: React.ComponentProps<'ul'>) {
  return (
    <ul
      data-slot="pagination-content"
      className={cn('flex flex-row items-center gap-1', className)}
      {...props}
    />
  );
}

function PaginationItem({ ...props }: React.ComponentProps<'li'>) {
  return <li data-slot="pagination-item" {...props} />;
}

type PaginationLinkProps = {
  isActive?: boolean;
  disabled?: boolean;
} & Pick<React.ComponentProps<typeof Button>, 'size'> &
  React.ComponentProps<'a'>;

const PaginationLink = React.forwardRef<React.ElementRef<'a'>, PaginationLinkProps>(
  ({ className, isActive, disabled, size = 'icon', href, ...props }, ref) => (
    <a
      ref={ref}
      aria-current={isActive ? 'page' : undefined}
      aria-disabled={disabled || undefined}
      tabIndex={disabled ? -1 : undefined}
      data-slot="pagination-link"
      data-active={isActive}
      href={disabled ? undefined : (href ?? '#')}
      className={cn(
        buttonVariants({
          variant: isActive ? 'outline' : 'ghost',
          size,
        }),
        disabled && 'pointer-events-none opacity-50',
        className
      )}
      {...props}
    />
  )
);
PaginationLink.displayName = 'PaginationLink';

interface PaginationPreviousProps extends React.ComponentProps<typeof PaginationLink> {
  /** Visible text next to the icon. @default "Previous" */
  label?: string;
  /** Screen-reader label. @default "Go to previous page" */
  ariaLabel?: string;
}

function PaginationPrevious({
  className,
  disabled,
  label = 'Previous',
  ariaLabel = 'Go to previous page',
  ...props
}: PaginationPreviousProps) {
  return (
    <PaginationLink
      aria-label={ariaLabel}
      size="default"
      disabled={disabled}
      className={cn('gap-1 px-2.5 sm:pl-2.5', className)}
      {...props}
    >
      <ChevronLeftIcon />
      <span className="hidden sm:block">{label}</span>
    </PaginationLink>
  );
}

interface PaginationNextProps extends React.ComponentProps<typeof PaginationLink> {
  /** Visible text next to the icon. @default "Next" */
  label?: string;
  /** Screen-reader label. @default "Go to next page" */
  ariaLabel?: string;
}

function PaginationNext({
  className,
  disabled,
  label = 'Next',
  ariaLabel = 'Go to next page',
  ...props
}: PaginationNextProps) {
  return (
    <PaginationLink
      aria-label={ariaLabel}
      size="default"
      disabled={disabled}
      className={cn('gap-1 px-2.5 sm:pr-2.5', className)}
      {...props}
    >
      <span className="hidden sm:block">{label}</span>
      <ChevronRightIcon />
    </PaginationLink>
  );
}

function PaginationEllipsis({ className, ...props }: React.ComponentProps<'span'>) {
  return (
    <span
      aria-hidden
      data-slot="pagination-ellipsis"
      className={cn('flex size-9 items-center justify-center', className)}
      {...props}
    >
      <MoreHorizontalIcon className="size-4" />
      <span className="sr-only">More pages</span>
    </span>
  );
}

export {
  Pagination,
  PaginationContent,
  PaginationLink,
  PaginationItem,
  PaginationPrevious,
  PaginationNext,
  PaginationEllipsis,
};