import { Box, Center } from "@mantine/core"
import type React from "react"
import type { EntrySourceType } from "@/app/entries/slice"
import { FeedFavicon } from "@/components/content/FeedFavicon"
import { tss } from "@/tss"
import { UnreadCount } from "./UnreadCount"

interface TreeNodeProps {
    id: string
    type: EntrySourceType
    name: React.ReactNode
    icon: React.ReactNode
    unread: number
    selected: boolean
    expanded?: boolean
    level: number
    hasError: boolean
    hasWarning: boolean
    hasNewEntries: boolean
    onClick: (e: React.MouseEvent, id: string) => void
    onIconClick?: (e: React.MouseEvent, id: string) => void
}

const useStyles = tss
    .withParams<{
        selected: boolean
        hasError: boolean
        hasWarning: boolean
        hasUnread: boolean
    }>()
    .create(({ theme, colorScheme, selected, hasError, hasWarning, hasUnread }) => {
        let backgroundColor = "inherit"
        if (selected) backgroundColor = colorScheme === "dark" ? theme.colors.dark[4] : theme.colors.gray[1]

        let color: string
        if (hasError) {
            color = theme.colors.red[6]
        } else if (hasWarning) {
            color = theme.colors.yellow[6]
        } else if (colorScheme === "dark") {
            color = hasUnread ? theme.colors.dark[0] : theme.colors.dark[3]
        } else {
            color = hasUnread ? theme.black : theme.colors.gray[6]
        }

        return {
            node: {
                display: "flex",
                alignItems: "center",
                cursor: "pointer",
                color,
                backgroundColor,
                "&:hover": {
                    backgroundColor: colorScheme === "dark" ? theme.colors.dark[6] : theme.colors.gray[0],
                },
            },
            nodeText: {
                flexGrow: 1,
                whiteSpace: "nowrap",
                overflow: "hidden",
                textOverflow: "ellipsis",
            },
        }
    })

export function TreeNode(props: Readonly<TreeNodeProps>) {
    const { classes } = useStyles({
        selected: props.selected,
        hasError: props.hasError,
        hasWarning: props.hasWarning,
        hasUnread: props.unread > 0,
    })
    return (
        <Box
            py={1}
            pl={props.level * 20}
            className={`${classes.node} cf-treenode cf-treenode-${props.type}`}
            onClick={(e: React.MouseEvent) => props.onClick(e, props.id)}
            data-id={props.id}
            data-type={props.type}
            data-unread-count={props.unread}
        >
            <Box mr={6} onClick={(e: React.MouseEvent) => props.onIconClick?.(e, props.id)} className="cf-treenode-icon">
                <Center>{typeof props.icon === "string" ? <FeedFavicon url={props.icon} /> : props.icon}</Center>
            </Box>
            <Box className={classes.nodeText}>{props.name}</Box>
            {!props.expanded && (
                <Box className="cf-treenode-unread-count">
                    <UnreadCount unreadCount={props.unread} showIndicator={props.hasNewEntries} />
                </Box>
            )}
        </Box>
    )
}