import { ReactElement, ReactNode } from "react";
import { MDXRemote } from "remote-mdx/rsc";
import { cn } from "@/lib/utils";
import remarkGfm from "remark-gfm";
import {
    Table,
    TableBody,
    TableCell,
    TableHead,
    TableHeader,
    TableRow,
} from "@/components/ui/table";

/**
 * The MDX components to style.
 */
const components: any = {
    h1: ({ children }: { children: ReactNode }) => (
        <Heading size="text-4xl">{children}</Heading>
    ),
    h2: ({ children }: { children: ReactNode }) => (
        <Heading size="text-3xl">{children}</Heading>
    ),
    h3: ({ children }: { children: ReactNode }) => (
        <Heading size="text-2xl">{children}</Heading>
    ),
    h4: ({ children }: { children: ReactNode }) => (
        <Heading size="text-xl">{children}</Heading>
    ),
    h5: ({ children }: { children: ReactNode }) => (
        <Heading size="text-lg">{children}</Heading>
    ),
    h6: ({ children }: { children: ReactNode }) => (
        <Heading size="text-md">{children}</Heading>
    ),
    a: ({ href, children }: { href: string; children: ReactNode }) => (
        <a
            className="text-minecraft-green-4 cursor-pointer hover:opacity-85 transition-all transform-gpu"
            href={href}
        >
            {children}
        </a>
    ),
    p: ({ children }: { children: ReactNode }) => (
        <p className="leading-4 text-zinc-300/80">{children}</p>
    ),
    ul: ({ children }: { children: ReactNode }) => (
        <ul className="list-disc list-inside">{children}</ul>
    ),

    // Tables
    table: ({ children }: { children: any }) => (
        <Table>
            <TableHeader>
                <TableRow>
                    {children?.[0].props.children.props.children}
                </TableRow>
            </TableHeader>
            <TableBody>{children?.[1].props.children}</TableBody>
        </Table>
    ),
    th: ({ children }: { children: ReactNode }) => (
        <TableHead>{children}</TableHead>
    ),
    td: ({ children }: { children: ReactNode }) => (
        <TableCell>{children}</TableCell>
    ),
};

/**
 * The custom render for MDX.
 *
 * @param props the props for the MDX
 * @return the custom mdx
 */
export const CustomMDX = (props: any): ReactElement => (
    <MDXRemote
        {...props}
        components={{
            ...components,
            ...(props.components || {}),
        }}
        options={{
            mdxOptions: {
                remarkPlugins: [remarkGfm],
                remarkRehypeOptions: {
                    passThrough: ["link"],
                },
            },
        }}
    />
);

/**
 * A heading component.
 *
 * @param size the size of the heading.
 * @param children the children within the heading
 * @return the heading jsx
 */
const Heading = ({
    size,
    children,
}: {
    size: string;
    children: ReactNode;
}): ReactElement => (
    <h1 className={cn("pt-2.5 font-bold", size)}>{children}</h1>
);