import * as React from "react"
import { Command } from "cmdk"
import { CommandSeparator, CommandShortcut } from "@/components/ui/command"
import { LaIcon } from "@/components/custom/la-icon"
import { CommandItemType, CommandAction } from "./command-data"
import { HTMLLikeElement, renderHTMLLikeElement } from "@/lib/utils"

export interface CommandItemProps extends Omit<CommandItemType, "action"> {
	action: CommandAction
	handleAction: (action: CommandAction, payload?: any) => void
}

const HTMLLikeRenderer: React.FC<{ content: HTMLLikeElement | string }> = React.memo(({ content }) => {
	return <>{renderHTMLLikeElement(content)}</>
})

HTMLLikeRenderer.displayName = "HTMLLikeRenderer"

export const CommandItem: React.FC<CommandItemProps> = React.memo(
	({ icon, label, action, payload, shortcut, handleAction }) => (
		<Command.Item onSelect={() => handleAction(action, payload)}>
			{icon && <LaIcon name={icon} />}
			<HTMLLikeRenderer content={label} />
			{shortcut && <CommandShortcut>{shortcut}</CommandShortcut>}
		</Command.Item>
	)
)

CommandItem.displayName = "CommandItem"

export interface CommandGroupProps {
	heading?: string
	items: CommandItemType[]
	handleAction: (action: CommandAction, payload?: any) => void
	isLastGroup: boolean
}

export const CommandGroup: React.FC<CommandGroupProps> = React.memo(({ heading, items, handleAction, isLastGroup }) => {
	return (
		<>
			{heading ? (
				<Command.Group heading={heading}>
					{items.map((item, index) => (
						<CommandItem key={`${heading}-${item.label}-${index}`} {...item} handleAction={handleAction} />
					))}
				</Command.Group>
			) : (
				items.map((item, index) => (
					<CommandItem key={`item-${item.label}-${index}`} {...item} handleAction={handleAction} />
				))
			)}
			{!isLastGroup && <CommandSeparator className="my-1.5" />}
		</>
	)
})

CommandGroup.displayName = "CommandGroup"
