import * as ScrollAreaPrimitive from "@radix-ui/react-scroll-area";
import * as React from "react";

import { cn } from "~/lib/utils";

function ScrollArea({
    className,
    children,
    scrollbarSize,
    viewportRef,
    ...props
}: React.ComponentProps<typeof ScrollAreaPrimitive.Root> & {
    scrollbarSize?: "default" | "narrow" | "none";
    viewportRef?: React.Ref<HTMLDivElement>;
}) {
    return (
        <ScrollAreaPrimitive.Root
            data-slot="scroll-area"
            className={cn("relative", className)}
            {...props}
        >
            <ScrollAreaPrimitive.Viewport
                ref={viewportRef}
                data-slot="scroll-area-viewport"
                className="focus-visible:ring-ring/50 size-full rounded-[inherit] transition-[color,box-shadow] outline-none focus-visible:ring-[3px] focus-visible:outline-1"
            >
                {children}
            </ScrollAreaPrimitive.Viewport>
            <ScrollBar size={scrollbarSize} />
            <ScrollAreaPrimitive.Corner />
        </ScrollAreaPrimitive.Root>
    );
}

function ScrollBar({
    className,
    orientation = "vertical",
    size = "default",
    ...props
}: React.ComponentProps<typeof ScrollAreaPrimitive.ScrollAreaScrollbar> & {
    size?: "default" | "narrow" | "none";
}) {
    const classes = {
        vertical: {
            className: "h-full border-l border-l-transparent",
            size: {
                default: "w-2.5",
                narrow: "w-1.5",
                none: "hidden",
            },
        },
        horizontal: {
            className: "flex-col border-t border-t-transparent",
            size: {
                default: "h-2.5",
                narrow: "h-1.5",
                none: "hidden",
            },
        },
    };

    return (
        <ScrollAreaPrimitive.ScrollAreaScrollbar
            data-slot="scroll-area-scrollbar"
            orientation={orientation}
            className={cn(
                "flex touch-none p-px transition-colors select-none",
                classes[orientation].className,
                classes[orientation].size[size],
                className,
            )}
            {...props}
        >
            <ScrollAreaPrimitive.ScrollAreaThumb
                data-slot="scroll-area-thumb"
                className="bg-border relative flex-1 rounded-full"
            />
        </ScrollAreaPrimitive.ScrollAreaScrollbar>
    );
}

export { ScrollArea, ScrollBar };