import React, { useState, useMemo } from 'react';
import { InvestigationCase, FinancialHop } from '../../types/forensics';
import { ReverseNetworkGraph } from '../ReverseNetworkGraph';
import { generateRelationalDatasets } from '../../services/relationalDataGenerator';
import { 
  CreditCard, 
  DollarSign, 
  ArrowRight, 
  AlertTriangle, 
  ShieldAlert, 
  Layers, 
  RefreshCw, 
  CheckCircle2,
  ExternalLink,
  GitFork
} from 'lucide-react';

interface FinancialGraphTabProps {
  currentCase: InvestigationCase;
  onNavigateToWorkbook?: () => void;
}

export const FinancialGraphTab: React.FC<FinancialGraphTabProps> = ({
  currentCase,
  onNavigateToWorkbook,
}) => {
  const ledger = currentCase.financialLedger;
  const [selectedHop, setSelectedHop] = useState<FinancialHop | null>(
    ledger.hops.length > 0 ? ledger.hops[0] : null
  );
  const [graphMode, setGraphMode] = useState<'REVERSE_TOPOLOGY' | 'LINEAR_PIPELINE'>('REVERSE_TOPOLOGY');

  // Generate Topological Graph
  const { graphNodes, graphEdges } = useMemo(() => {
    return generateRelationalDatasets([currentCase], currentCase);
  }, [currentCase]);

  const getHopTypeBadge = (type: string) => {
    switch (type) {
      case 'CRYPTO_MIXER':
        return <span className="bg-rose-950 text-rose-300 border border-rose-800 px-1.5 py-0.5 rounded text-[10px] font-mono font-bold">CRYPTO MIXER</span>;
      case 'SMURFING_DEPOSIT':
        return <span className="bg-amber-950 text-amber-300 border border-amber-800 px-1.5 py-0.5 rounded text-[10px] font-mono font-bold">BSA SMURFING</span>;
      case 'HAWALA_CORRIDOR':
        return <span className="bg-purple-950 text-purple-300 border border-purple-800 px-1.5 py-0.5 rounded text-[10px] font-mono font-bold">HAWALA WIRE</span>;
      case 'MERCHANT_CASH_OUT':
        return <span className="bg-emerald-950 text-emerald-300 border border-emerald-800 px-1.5 py-0.5 rounded text-[10px] font-mono font-bold">CASH-OUT ENDPOINT</span>;
      default:
        return <span className="bg-blue-950 text-blue-300 border border-blue-800 px-1.5 py-0.5 rounded text-[10px] font-mono">LAYERED NODE</span>;
    }
  };

  return (
    <div className="space-y-4 p-4 text-slate-100">
      {/* Top Banner: Financial Laundering & Structuring Summary */}
      <div className="bg-slate-900/80 border border-slate-800 rounded-xl p-4 shadow-md space-y-2">
        <div className="flex items-center justify-between">
          <div className="flex items-center gap-2">
            <CreditCard className="w-5 h-5 text-emerald-400" />
            <h3 className="text-sm font-bold uppercase tracking-wider font-mono">
              Terror Financing & Anti-Money Laundering (AML) Multi-Hop Graph
            </h3>
          </div>
          <div className="flex items-center gap-2">
            {ledger.detectedStructuring && (
              <span className="text-[11px] font-mono px-2 py-0.5 rounded bg-rose-950 text-rose-300 border border-rose-800 font-bold">
                BSA &lt;$10K SMURFING DETECTED
              </span>
            )}
            <span className="text-[11px] font-mono px-2 py-0.5 rounded bg-emerald-950 text-emerald-300 border border-emerald-800 font-bold">
              ${ledger.totalVolumeUsd.toLocaleString()} USD INTERCEPTED
            </span>
          </div>
        </div>

        <p className="text-xs text-slate-300 leading-relaxed font-mono">
          {ledger.structuringSummary}
        </p>

        {ledger.sanctionsHits.length > 0 && (
          <div className="bg-rose-950/60 border border-rose-800 p-2.5 rounded-lg flex items-center gap-2 text-xs font-mono text-rose-200">
            <ShieldAlert className="w-4 h-4 text-rose-400 shrink-0" />
            <span>
              <strong>OFAC / Watchlist Alert:</strong> Matching entities identified on global sanctions rosters: {ledger.sanctionsHits.join(', ')}
            </span>
          </div>
        )}
      </div>

      {/* View Switcher: Reverse Network Graph vs Linear Pipeline */}
      <div className="flex items-center justify-between bg-slate-900/90 border border-slate-800 rounded-xl p-2.5">
        <div className="flex items-center gap-2">
          <button
            onClick={() => setGraphMode('REVERSE_TOPOLOGY')}
            className={`flex items-center gap-1.5 px-3 py-1.5 rounded-lg text-xs font-mono font-bold transition-all cursor-pointer ${
              graphMode === 'REVERSE_TOPOLOGY'
                ? 'bg-cyan-950 border border-cyan-500 text-cyan-300 shadow'
                : 'bg-slate-950 text-slate-400 hover:text-slate-200'
            }`}
          >
            <GitFork className="w-3.5 h-3.5 rotate-180" />
            <span>Reverse Network Topology (Depth Gradient & Traversal)</span>
          </button>

          <button
            onClick={() => setGraphMode('LINEAR_PIPELINE')}
            className={`flex items-center gap-1.5 px-3 py-1.5 rounded-lg text-xs font-mono font-bold transition-all cursor-pointer ${
              graphMode === 'LINEAR_PIPELINE'
                ? 'bg-cyan-950 border border-cyan-500 text-cyan-300 shadow'
                : 'bg-slate-950 text-slate-400 hover:text-slate-200'
            }`}
          >
            <Layers className="w-3.5 h-3.5" />
            <span>Linear Hop Pipeline</span>
          </button>
        </div>

        {onNavigateToWorkbook && (
          <button
            onClick={onNavigateToWorkbook}
            className="flex items-center gap-1 text-xs font-mono text-cyan-400 hover:text-cyan-300 bg-slate-950 border border-slate-800 hover:border-cyan-800 px-3 py-1.5 rounded-lg transition-all cursor-pointer"
          >
            <span>Open Relational SQL Workbook</span>
            <ExternalLink className="w-3.5 h-3.5" />
          </button>
        )}
      </div>

      {/* Primary Graph View */}
      {graphMode === 'REVERSE_TOPOLOGY' ? (
        <ReverseNetworkGraph
          nodes={graphNodes}
          edges={graphEdges}
          onSelectNodeInDatabase={() => {
            if (onNavigateToWorkbook) onNavigateToWorkbook();
          }}
        />
      ) : (
        /* Linear Hop Flow Container */
        <div className="bg-slate-900/80 border border-slate-800 rounded-xl p-4 shadow-md space-y-3">
          <div className="flex items-center justify-between border-b border-slate-800 pb-2">
            <div className="flex items-center gap-2 text-xs font-bold text-slate-200">
              <Layers className="w-4 h-4 text-cyan-400" />
              <span>TRANSACTION TOPOLOGY & ANONYMIZATION HOPS</span>
            </div>
            <span className="text-[11px] text-slate-400 font-mono">Click a node or hop below to inspect payload</span>
          </div>

          {ledger.hops.length === 0 ? (
            <div className="p-8 text-center text-slate-500 text-xs font-mono">
              No on-chain or fiat hops logged for this docket.
            </div>
          ) : (
            <div className="space-y-4">
              <div className="flex flex-wrap items-center justify-start gap-2 p-3 bg-slate-950 rounded-lg border border-slate-800/80 overflow-x-auto">
                {ledger.hops.map((hop, index) => {
                  const isSelected = selectedHop?.id === hop.id;
                  return (
                    <React.Fragment key={hop.id}>
                      <div
                        onClick={() => setSelectedHop(hop)}
                        className={`p-3 rounded-lg border cursor-pointer transition-all min-w-[200px] flex-1 shrink-0 ${
                          isSelected
                            ? 'bg-slate-800 border-cyan-500 shadow-md shadow-cyan-950/50'
                            : 'bg-slate-900 border-slate-800 hover:border-slate-700'
                        }`}
                      >
                        <div className="flex items-center justify-between text-[10px] font-mono text-slate-400 mb-1">
                          <span>HOP {index + 1}</span>
                          {getHopTypeBadge(hop.hopType)}
                        </div>

                        <div className="text-xs font-bold text-slate-100 mb-1">
                          ${hop.amountUsd.toLocaleString()} <span className="text-cyan-400 font-mono text-[10px] font-normal">{hop.asset}</span>
                        </div>

                        <div className="text-[10px] font-mono text-slate-400 truncate">
                          {hop.riskTag}
                        </div>

                        <div className="text-[9px] text-slate-500 font-mono mt-1">
                          {hop.timestamp}
                        </div>
                      </div>

                      {index < ledger.hops.length - 1 && (
                        <ArrowRight className="w-4 h-4 text-cyan-500 shrink-0 hidden md:block" />
                      )}
                    </React.Fragment>
                  );
                })}
              </div>

              {selectedHop && (
                <div className="bg-slate-950 p-4 rounded-xl border border-cyan-900/60 space-y-3">
                  <div className="flex items-center justify-between border-b border-slate-800 pb-2">
                    <div className="flex items-center gap-2">
                      <span className="text-xs font-mono font-bold text-cyan-300">{selectedHop.id} DETAILS</span>
                      {getHopTypeBadge(selectedHop.hopType)}
                    </div>
                    <span className="text-xs font-mono text-emerald-400 font-bold">
                      ${selectedHop.amountUsd.toLocaleString()} USD ({selectedHop.asset})
                    </span>
                  </div>

                  <div className="grid grid-cols-1 md:grid-cols-2 gap-3 text-xs font-mono">
                    <div className="bg-slate-900 p-2.5 rounded border border-slate-800">
                      <span className="text-slate-500 block text-[10px]">ORIGIN NODE / SOURCE</span>
                      <span className="text-slate-200 font-bold break-all">{selectedHop.from}</span>
                    </div>

                    <div className="bg-slate-900 p-2.5 rounded border border-slate-800">
                      <span className="text-slate-500 block text-[10px]">DESTINATION NODE / TARGET</span>
                      <span className="text-slate-200 font-bold break-all">{selectedHop.to}</span>
                    </div>
                  </div>

                  <div className="bg-slate-900 p-3 rounded border border-slate-800 space-y-1 text-xs font-mono">
                    <div className="text-[10px] text-slate-400 font-bold">AML RISK & SANCTIONS TAG:</div>
                    <div className="text-rose-400 font-bold">{selectedHop.riskTag}</div>
                    <div className="text-slate-400 text-[11px]">
                      Recorded Timestamp: {selectedHop.timestamp} UTC • Sanctions Flag: {selectedHop.flaggedSanction ? 'STRICT COMPLIANCE ALERT' : 'Standard Monitoring'}
                    </div>
                  </div>
                </div>
              )}
            </div>
          )}
        </div>
      )}

      {/* Comprehensive Ledger Table */}
      <div className="bg-slate-900/80 border border-slate-800 rounded-xl p-4 shadow-md space-y-3">
        <div className="flex items-center justify-between">
          <div className="flex items-center gap-2 text-xs font-bold text-slate-200">
            <DollarSign className="w-4 h-4 text-emerald-400" />
            <span>FULL FINANCIAL FORENSICS LEDGER ({ledger.hops.length} RECORDS)</span>
          </div>
        </div>

        <div className="overflow-x-auto">
          <table className="w-full text-left text-xs font-mono border-collapse">
            <thead>
              <tr className="bg-slate-950 text-slate-400 border-b border-slate-800">
                <th className="p-2.5">Hop ID</th>
                <th className="p-2.5">Timestamp</th>
                <th className="p-2.5">Source</th>
                <th className="p-2.5">Destination</th>
                <th className="p-2.5">Amount & Asset</th>
                <th className="p-2.5">Classification</th>
              </tr>
            </thead>
            <tbody className="divide-y divide-slate-800/60">
              {ledger.hops.map((hop) => (
                <tr key={hop.id} className="hover:bg-slate-800/40 text-slate-300">
                  <td className="p-2.5 font-bold text-cyan-400">{hop.id}</td>
                  <td className="p-2.5 text-slate-400">{hop.timestamp}</td>
                  <td className="p-2.5 text-slate-300 max-w-[150px] truncate">{hop.from}</td>
                  <td className="p-2.5 text-slate-300 max-w-[150px] truncate">{hop.to}</td>
                  <td className="p-2.5 font-bold text-emerald-300">${hop.amountUsd.toLocaleString()} ({hop.asset})</td>
                  <td className="p-2.5">{getHopTypeBadge(hop.hopType)}</td>
                </tr>
              ))}
            </tbody>
          </table>
        </div>
      </div>
    </div>
  );
};