import { cn } from '@/lib/utils';
import { Package, Truck, Factory, CornerDownRight, Box, ShoppingCart } from 'lucide-react';
import { formatDate } from '@/lib/date';

export interface TraceabilityNode {
    id: string;
    type: 'target_batch' | 'source_batch' | 'production_order' | 'material_batch' | 'goods_receipt' | 'outbound_transaction';
    label: string;
    batch_number?: string;
    date?: string;
    vendor_id?: number | string;
    product_name?: string;
    spec?: string;
    quantity?: number | string;
    unit?: string;
    warehouse_name?: string;
    children?: TraceabilityNode[];
}

interface TreeViewProps {
    data: TraceabilityNode;
}

const getNodeIcon = (type: TraceabilityNode['type']) => {
    switch (type) {
        case 'target_batch':
        case 'source_batch':
            return <Package className="h-5 w-5 text-primary-main" />;
        case 'production_order':
            return <Factory className="h-5 w-5 text-other-warning" />;
        case 'material_batch':
            return <Box className="h-5 w-5 text-primary-main" />;
        case 'goods_receipt':
            return <Truck className="h-5 w-5 text-other-info" />;
        case 'outbound_transaction':
            return <ShoppingCart className="h-5 w-5 text-other-success" />;
        default:
            return <Box className="h-5 w-5 text-gray-400" />;
    }
};

const getNodeColor = (type: TraceabilityNode['type']) => {
    switch (type) {
        case 'target_batch':
        case 'source_batch':
            return 'border-primary-light bg-primary-lightest';
        case 'production_order':
            return 'border-orange-200 bg-orange-50';
        case 'material_batch':
            return 'border-primary-light bg-primary-lightest';
        case 'goods_receipt':
            return 'border-blue-200 bg-blue-50';
        case 'outbound_transaction':
            return 'border-emerald-200 bg-emerald-50';
        default:
            return 'border-gray-200 bg-gray-50';
    }
};

const TreeNode = ({ node, isLast, level = 0 }: { node: TraceabilityNode; isLast: boolean; level?: number }) => {
    const hasChildren = node.children && node.children.length > 0;

    return (
        <div className="relative">
            {/* 連接線 (上層到此節點) */}
            {level > 0 && (
                <div
                    className={cn(
                        "absolute border-l-2 border-gray-200",
                        isLast ? "h-6 top-0" : "h-full top-0"
                    )}
                    style={{ left: '-1.5rem' }}
                />
            )}

            {/* 橫向連接線 */}
            {level > 0 && (
                <div
                    className="absolute border-t-2 border-gray-200 w-6 top-6"
                    style={{ left: '-1.5rem' }}
                />
            )}

            <div className="flex items-start mb-6">
                {level > 0 && (
                    <div className="w-6 h-12 shrink-0 flex items-center justify-end pr-2 text-gray-400">
                        <CornerDownRight className="h-4 w-4" />
                    </div>
                )}

                <div className={cn(
                    "relative flex flex-col p-4 rounded-xl border shadow-sm min-w-[320px] max-w-md",
                    getNodeColor(node.type)
                )}>
                    <div className="flex items-center gap-3">
                        <div className="p-2 bg-white rounded-lg shadow-sm shrink-0">
                            {getNodeIcon(node.type)}
                        </div>
                        <div className="flex-1 min-w-0">
                            <h4 className="font-semibold text-gray-900 truncate">{node.label}</h4>
                            {node.date && (
                                <div className="mt-0.5">
                                    <span className="text-xs text-gray-500">{formatDate(node.date)}</span>
                                </div>
                            )}
                        </div>
                    </div>

                    {(node.product_name || node.warehouse_name || (node.quantity !== undefined && node.quantity !== null)) && (
                        <div className="mt-3 pt-3 border-t border-gray-200/60 flex flex-wrap gap-4">
                            {node.product_name && (
                                <div className="flex-1 min-w-[120px]">
                                    <span className="text-[11px] text-gray-500 font-medium block mb-0.5">品名 / 規格</span>
                                    <span className="text-sm font-medium text-gray-800 leading-tight block truncate" title={`${node.product_name} ${node.spec ? `(${node.spec})` : ''}`}>
                                        {node.product_name}
                                        {node.spec && <span className="text-gray-400 text-xs ml-1">({node.spec})</span>}
                                    </span>
                                </div>
                            )}

                            {node.warehouse_name && (
                                <div className="shrink-0 min-w-[80px]">
                                    <span className="text-[11px] text-gray-500 font-medium block mb-0.5">倉庫</span>
                                    <span className="text-sm font-medium text-gray-800 leading-tight block truncate" title={node.warehouse_name}>
                                        {node.warehouse_name}
                                    </span>
                                </div>
                            )}

                            {node.quantity !== undefined && node.quantity !== null && (
                                <div className="shrink-0 text-right min-w-[60px]">
                                    <span className="text-[11px] text-gray-500 font-medium block mb-0.5">數量</span>
                                    <span className="text-sm font-bold text-gray-900 block truncate">
                                        {Number(node.quantity).toLocaleString()}
                                        <span className="text-[10px] ml-0.5 text-gray-500 font-normal">{node.unit || 'PCS'}</span>
                                    </span>
                                </div>
                            )}
                        </div>
                    )}
                </div>
            </div>

            {hasChildren && (
                <div className="pl-12 relative">
                    {node.children!.map((child, index) => (
                        <TreeNode
                            key={`${child.id} -${index} `}
                            node={child}
                            isLast={index === node.children!.length - 1}
                            level={level + 1}
                        />
                    ))}
                </div>
            )}
        </div>
    );
};

export default function TreeView({ data }: TreeViewProps) {
    if (!data) return null;

    return (
        <div className="p-4 md:p-8 bg-gray-50/50 rounded-2xl border border-gray-100 overflow-x-auto">
            <TreeNode node={data} isLast={true} level={0} />
        </div>
    );
}