/* eslint-disable @typescript-eslint/no-explicit-any */
import React, { useEffect, useState } from "react";

type CountProps = {
  mindmap: any;
};

export const MindmapCount = ({ mindmap }: CountProps) => {
  const [counts, setCounts] = useState({ words: 0, nodes: 0 });

  useEffect(() => {
    if (!mindmap) return;

    const calculateCounts = (data: any) => {
      let nodes = 0;
      let textStr = "";

      const walk = (nodeData: any) => {
        if (!nodeData) return;
        nodes++;
        // simple-mind-map node data structure: { data: { text: ... }, children: [...] }
        const text = String(nodeData.data?.text || "");
        textStr += text;
        
        if (nodeData.children && Array.isArray(nodeData.children)) {
          nodeData.children.forEach(walk);
        }
      };

      walk(data);

      // Remove HTML tags for word count
      const tempDiv = document.createElement("div");
      tempDiv.innerHTML = textStr;
      const words = tempDiv.textContent?.length || 0;

      setCounts({ words, nodes });
    };

    const onDataChange = (data: any) => {
      calculateCounts(data);
    };

    mindmap.on("data_change", onDataChange);
    
    // Initial calculation if data is available
    // mindmap.getData() might return full data object
    try {
        const initialData = mindmap.getData();
        if (initialData) calculateCounts(initialData);
    } catch (e) {
        console.warn("Failed to get initial mindmap data for count", e);
    }

    return () => {
      mindmap.off("data_change", onDataChange);
    };
  }, [mindmap]);

  return (
    <div className="absolute bottom-4 left-4 flex items-center gap-4 rounded-md border border-gray-200 bg-white/90 px-3 py-1 text-xs text-gray-600 shadow-sm backdrop-blur-sm select-none z-10">
      <div className="flex items-center gap-1">
        <span>字数</span>
        <span className="font-medium text-gray-900">{counts.words}</span>
      </div>
      <div className="flex items-center gap-1">
        <span>节点</span>
        <span className="font-medium text-gray-900">{counts.nodes}</span>
      </div>
    </div>
  );
};
