"use client"

import { ArrowRightIcon, MinusIcon, PackageIcon, PlusIcon } from "lucide-react"
import { createContext, type HTMLAttributes, type ReactNode, useContext } from "react"
import { Badge } from "@/components/ui/badge"
import { cn } from "@/lib/utils"

type ChangeType = "major" | "minor" | "patch" | "added" | "removed"

interface PackageInfoContextType {
  name: string
  currentVersion?: string
  newVersion?: string
  changeType?: ChangeType
}

const PackageInfoContext = createContext<PackageInfoContextType>({
  name: "",
})

export type PackageInfoProps = HTMLAttributes<HTMLDivElement> & {
  name: string
  currentVersion?: string
  newVersion?: string
  changeType?: ChangeType
}

export const PackageInfo = ({
  name,
  currentVersion,
  newVersion,
  changeType,
  className,
  children,
  ...props
}: PackageInfoProps) => (
  <PackageInfoContext.Provider value={{ name, currentVersion, newVersion, changeType }}>
    <div className={cn("rounded-lg border bg-background p-4", className)} {...props}>
      {children ?? (
        <>
          <PackageInfoHeader>
            <PackageInfoName />
            {changeType && <PackageInfoChangeType />}
          </PackageInfoHeader>
          {(currentVersion || newVersion) && <PackageInfoVersion />}
        </>
      )}
    </div>
  </PackageInfoContext.Provider>
)

export type PackageInfoHeaderProps = HTMLAttributes<HTMLDivElement>

export const PackageInfoHeader = ({ className, children, ...props }: PackageInfoHeaderProps) => (
  <div className={cn("flex items-center justify-between gap-2", className)} {...props}>
    {children}
  </div>
)

export type PackageInfoNameProps = HTMLAttributes<HTMLDivElement>

export const PackageInfoName = ({ className, children, ...props }: PackageInfoNameProps) => {
  const { name } = useContext(PackageInfoContext)

  return (
    <div className={cn("flex items-center gap-2", className)} {...props}>
      <PackageIcon className="size-4 text-muted-foreground" />
      <span className="font-medium font-mono text-sm">{children ?? name}</span>
    </div>
  )
}

const changeTypeStyles: Record<ChangeType, string> = {
  major: "bg-red-100 text-red-700 dark:bg-red-900/30 dark:text-red-400",
  minor: "bg-yellow-100 text-yellow-700 dark:bg-yellow-900/30 dark:text-yellow-400",
  patch: "bg-green-100 text-green-700 dark:bg-green-900/30 dark:text-green-400",
  added: "bg-blue-100 text-blue-700 dark:bg-blue-900/30 dark:text-blue-400",
  removed: "bg-gray-100 text-gray-700 dark:bg-gray-900/30 dark:text-gray-400",
}

const changeTypeIcons: Record<ChangeType, ReactNode> = {
  major: <ArrowRightIcon className="size-3" />,
  minor: <ArrowRightIcon className="size-3" />,
  patch: <ArrowRightIcon className="size-3" />,
  added: <PlusIcon className="size-3" />,
  removed: <MinusIcon className="size-3" />,
}

export type PackageInfoChangeTypeProps = HTMLAttributes<HTMLDivElement>

export const PackageInfoChangeType = ({
  className,
  children,
  ...props
}: PackageInfoChangeTypeProps) => {
  const { changeType } = useContext(PackageInfoContext)

  if (!changeType) {
    return null
  }

  return (
    <Badge
      className={cn("gap-1 text-xs capitalize", changeTypeStyles[changeType], className)}
      variant="secondary"
      {...props}
    >
      {changeTypeIcons[changeType]}
      {children ?? changeType}
    </Badge>
  )
}

export type PackageInfoVersionProps = HTMLAttributes<HTMLDivElement>

export const PackageInfoVersion = ({ className, children, ...props }: PackageInfoVersionProps) => {
  const { currentVersion, newVersion } = useContext(PackageInfoContext)

  if (!(currentVersion || newVersion)) {
    return null
  }

  return (
    <div
      className={cn(
        "mt-2 flex items-center gap-2 font-mono text-muted-foreground text-sm",
        className,
      )}
      {...props}
    >
      {children ?? (
        <>
          {currentVersion && <span>{currentVersion}</span>}
          {currentVersion && newVersion && <ArrowRightIcon className="size-3" />}
          {newVersion && <span className="font-medium text-foreground">{newVersion}</span>}
        </>
      )}
    </div>
  )
}

export type PackageInfoDescriptionProps = HTMLAttributes<HTMLParagraphElement>

export const PackageInfoDescription = ({
  className,
  children,
  ...props
}: PackageInfoDescriptionProps) => (
  <p className={cn("mt-2 text-muted-foreground text-sm", className)} {...props}>
    {children}
  </p>
)

export type PackageInfoContentProps = HTMLAttributes<HTMLDivElement>

export const PackageInfoContent = ({ className, children, ...props }: PackageInfoContentProps) => (
  <div className={cn("mt-3 border-t pt-3", className)} {...props}>
    {children}
  </div>
)

export type PackageInfoDependenciesProps = HTMLAttributes<HTMLDivElement>

export const PackageInfoDependencies = ({
  className,
  children,
  ...props
}: PackageInfoDependenciesProps) => (
  <div className={cn("space-y-2", className)} {...props}>
    <span className="font-medium text-muted-foreground text-xs uppercase tracking-wide">
      Dependencies
    </span>
    <div className="space-y-1">{children}</div>
  </div>
)

export type PackageInfoDependencyProps = HTMLAttributes<HTMLDivElement> & {
  name: string
  version?: string
}

export const PackageInfoDependency = ({
  name,
  version,
  className,
  children,
  ...props
}: PackageInfoDependencyProps) => (
  <div className={cn("flex items-center justify-between text-sm", className)} {...props}>
    {children ?? (
      <>
        <span className="font-mono text-muted-foreground">{name}</span>
        {version && <span className="font-mono text-xs">{version}</span>}
      </>
    )}
  </div>
)

/** Demo component for preview */
export default function PackageInfoDemo() {
  return (
    <div className="flex w-full max-w-md flex-col gap-4 p-4">
      <PackageInfo name="react" currentVersion="18.2.0" newVersion="19.0.0" changeType="major">
        <PackageInfoHeader>
          <PackageInfoName />
          <PackageInfoChangeType />
        </PackageInfoHeader>
        <PackageInfoVersion />
        <PackageInfoDescription>
          A JavaScript library for building user interfaces
        </PackageInfoDescription>
        <PackageInfoContent>
          <PackageInfoDependencies>
            <PackageInfoDependency name="loose-envify" version="^1.1.0" />
            <PackageInfoDependency name="scheduler" version="^0.23.0" />
          </PackageInfoDependencies>
        </PackageInfoContent>
      </PackageInfo>

      <PackageInfo name="@tanstack/react-query" changeType="added">
        <PackageInfoHeader>
          <PackageInfoName />
          <PackageInfoChangeType />
        </PackageInfoHeader>
        <PackageInfoDescription>Powerful asynchronous state management</PackageInfoDescription>
      </PackageInfo>

      <PackageInfo name="lodash" currentVersion="4.17.21" changeType="removed">
        <PackageInfoHeader>
          <PackageInfoName />
          <PackageInfoChangeType />
        </PackageInfoHeader>
        <PackageInfoVersion />
      </PackageInfo>
    </div>
  )
}