'use client'

import { CH } from '@code-hike/mdx/components'
import dynamic from 'next/dynamic'
import { ArrowUpRight, Triangle } from 'lucide-react'
import {
  Badge,
  cn,
  Collapsible_Shadcn_,
  CollapsibleContent_Shadcn_,
  CollapsibleTrigger_Shadcn_,
  Heading,
  Image,
  Mermaid,
} from 'ui'
import { Admonition } from 'ui-patterns/admonition'
import type { PropsWithChildren } from 'react'
import type { ImageProps } from 'ui/src/components/Image/Image'

const Avatar = dynamic(() => import('~/components/Avatar'))
const Chart = dynamic(() => import('~/components/Charts/PGCharts'))
const CodeBlock = dynamic(() => import('~/components/CodeBlock/CodeBlock'))
const Tabs = dynamic(() => import('~/components/Tabs/Tabs'), { ssr: false })
const TabPanel = dynamic(
  () => import('~/components/Tabs/Tabs').then((mod) => ({ default: mod.TabPanel })),
  { ssr: false }
)
const NamedCodeBlock = dynamic(
  () =>
    import('~/components/CodeTabs').then((mod) => ({
      default: mod.NamedCodeBlock,
    })),
  {
    ssr: false,
  }
)
const ImageFadeStack = dynamic(() => import('~/components/ImageFadeStack'))
const ImageGrid = dynamic(() => import('~/components/ImageGrid'))
const InlineCodeTag = dynamic(() => import('~/components/InlineCode'))
const Quote = dynamic(() => import('~/components/Quote'))

// import all components used in blog articles here
// to do: move this into a helper/utils, it is used elsewhere

const ignoreClass = 'ignore-on-export'

const LinkComponent = (props: PropsWithChildren<HTMLAnchorElement>) => (
  <a
    href={props.href}
    target={props.target}
    className={cn('inline relative [&_p]:inline', props.target === '_blank' && 'mr-4')}
  >
    {props.children}{' '}
    {props.target === '_blank' && <ArrowUpRight className="absolute -right-3.5 w-3 top-0" />}
  </a>
)

const BlogCollapsible = ({
  title,
  containerClassName,
  ...props
}: {
  title: string
  containerClassName?: string
}) => {
  return (
    <Collapsible_Shadcn_ className={containerClassName}>
      <CollapsibleTrigger_Shadcn_
        className="
        data-[state=open]:text
        hover:text-foreground-light
        flex items-center gap-3
        [&>svg]:fill-current
        [&>svg]:rotate-90
        [&>svg]:transition-transform
        [&>svg]:data-[state='open']:rotate-180
        [&>svg]:data-[state='open']:text
        "
      >
        <Triangle size={10} />
        <span>{title}</span>
      </CollapsibleTrigger_Shadcn_>
      <CollapsibleContent_Shadcn_ {...props} />
    </Collapsible_Shadcn_>
  )
}

export default function mdxComponents(type?: 'blog' | 'lp' | undefined) {
  const components = {
    CodeBlock,
    Tabs,
    TabPanel,
    NamedCodeBlock,
    CH,
    h1: (props: any) => <Heading {...props} tag="h1" />,
    h2: (props: any) => <Heading {...props} tag="h2" />,
    h3: (props: any) => <Heading {...props} tag="h3" />,
    h4: (props: any) => <Heading {...props} tag="h4" />,
    h5: (props: any) => <Heading {...props} tag="h5" />,
    h6: (props: any) => <Heading {...props} tag="h6" />,
    Badge,
    Quote,
    Avatar,
    PGChart: (props: any) => {
      return <Chart {...props} />
    },
    pre: (props: any) => {
      if (props.className !== ignoreClass) {
        const childProps = props.children?.props
        // Detect mermaid code blocks and render with Mermaid component
        if (childProps?.className === 'language-mermaid') {
          return <Mermaid chart={childProps.children} />
        }
        return <CodeBlock {...childProps} />
      } else {
        return <code {...props} />
      }
    },
    ImageGrid,
    ImageFadeStack,
    img: (props: any) => {
      if (props.className !== ignoreClass) {
        return (
          <Image
            fill
            className={cn(
              'm-0 object-cover',
              type === 'blog' ? 'rounded-md border' : '',
              props.wide && 'wide',
              props.className
            )}
            {...props}
          />
        )
      }
      // biome-ignore lint/a11y/useAltText: provided in props
      return <img {...props} />
    },
    Img: ({ zoomable = true, className, ...props }: ImageProps & { wide?: boolean }) => (
      <Image
        fill
        containerClassName={cn(props.wide && 'wide')}
        className={cn('m-0 object-cover', type === 'blog' ? 'rounded-md border' : '', className)}
        zoomable={zoomable}
        {...props}
      />
    ),
    Link: LinkComponent,
    code: (props: any) => <InlineCodeTag>{props.children}</InlineCodeTag>,
    BlogCollapsible: (props: any) => <BlogCollapsible {...props} />,
    Subtitle: (props: any) => (
      <p className={cn('-mt-6 text-foreground-lighter text-lg', props.className)} {...props} />
    ),
    Admonition,
    Mermaid,
  }

  return components as any
}