import React, { useState, useEffect, useRef } from 'react';
import { Card, CardContent, CardHeader, CardTitle } from '@/components/ui/card';
import { Button } from '@/components/ui/button';
import { Input } from '@/components/ui/input';
import { Badge } from '@/components/ui/badge';
import { Terminal, Bluetooth, Cpu, ShieldAlert, Play, Link, Server, Activity } from 'lucide-react';

interface HardwareComponent {
  id: string;
  component: string;
  driver: string;
  status: 'Installed' | 'Missing' | 'Injected';
  capability: string;
}

const MOCK_HARDWARE_STACK: Record<string, HardwareComponent[]> = {
  'Google': [
    { id: '1', component: 'Pixel Tensor SoC', driver: 'android.hardware.usb.gadget', status: 'Installed', capability: 'ADB over USB/BLE' },
    { id: '2', component: 'Broadcom BCM4389', driver: 'bt_bcm.ko', status: 'Installed', capability: 'BLE 5.3 / Classic' },
    { id: '3', component: 'Remote Shell Daemon', driver: 'adb_remote.apk', status: 'Missing', capability: 'Root Shell Execution' },
  ],
  'Apple': [
    { id: '1', component: 'Apple Silicon', driver: 'com.apple.driver.usb', status: 'Installed', capability: 'Serial over BLE' },
    { id: '2', component: 'CoreBluetooth', driver: 'IOBluetoothFamily', status: 'Installed', capability: 'GATT Server' },
    { id: '3', component: 'MobileDevice Service', driver: 'usbmuxd', status: 'Missing', capability: 'Remote Debugging' },
  ],
  'Microsoft': [
    { id: '1', component: 'x86_64 Architecture', driver: 'BTHUSB.SYS', status: 'Installed', capability: 'Win32 BT Stack' },
    { id: '2', component: 'PowerShell Host', driver: 'System.Management.Automation.dll', status: 'Installed', capability: 'PS Remoting' },
    { id: '3', component: 'Remote Execution Bridge', driver: 'win32_bt_bridge.dll', status: 'Missing', capability: 'System Level Execution' },
  ],
  'Unknown': [
    { id: '1', component: 'Generic BT Controller', driver: 'btusb', status: 'Installed', capability: 'Basic RFCOMM' },
  ]
};

export function MessengerPipe() {
  const [macId, setMacId] = useState('');
  const [vendor, setVendor] = useState('Unknown');
  const [connectionState, setConnectionState] = useState<'disconnected' | 'scanning' | 'connecting' | 'connected'>('disconnected');
  const [terminalLogs, setTerminalLogs] = useState<string[]>([
    `[SYSTEM] Remote Execution Environment Initialized.`,
    `[SYSTEM] Awaiting target MAC ID for Bluetooth/ADB bridging...`
  ]);
  const [hardwareStack, setHardwareStack] = useState<HardwareComponent[]>([]);
  
  const terminalRef = useRef<HTMLDivElement>(null);

  // Auto-scroll terminal
  useEffect(() => {
    if (terminalRef.current) {
      terminalRef.current.scrollTop = terminalRef.current.scrollHeight;
    }
  }, [terminalLogs]);

  // AI Debugger Loop
  useEffect(() => {
    let interval: NodeJS.Timeout;
    if (connectionState === 'connected') {
      interval = setInterval(() => {
        const latency = (Math.random() * 15 + 5).toFixed(2);
        const packetLoss = Math.random() > 0.95 ? '0.1%' : '0.0%';
        addLog(`[AI-DEBUG] Latency: ${latency}ms | Packet Loss: ${packetLoss} | Link: STABLE | ADB Sync: OK`);
      }, 5000);
    }
    return () => clearInterval(interval);
  }, [connectionState]);

  const addLog = (msg: string) => {
    setTerminalLogs(prev => [...prev, `[${new Date().toISOString().split('T')[1].slice(0,-1)}] ${msg}`]);
  };

  const lookupVendor = (mac: string) => {
    const prefix = mac.toUpperCase().replace(/[^A-F0-9]/g, '').substring(0, 6);
    // Mock OUI lookup
    if (['001A11', 'F88A5E', '3C5AB4'].includes(prefix)) return 'Google';
    if (['002500', '001CB3', 'F0D1A9'].includes(prefix)) return 'Apple';
    if (['0050F2', '281878', '7C1E52'].includes(prefix)) return 'Microsoft';
    return 'Unknown';
  };

  const handleConnect = () => {
    if (!macId.match(/^([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2})$/)) {
      addLog(`[ERROR] Invalid MAC Address format. Use XX:XX:XX:XX:XX:XX`);
      return;
    }

    setConnectionState('scanning');
    addLog(`[SCAN] Initiating targeted Bluetooth scan for MAC: ${macId}`);
    
    setTimeout(() => {
      const detectedVendor = lookupVendor(macId);
      setVendor(detectedVendor);
      addLog(`[SCAN] Device found. Vendor matched: ${detectedVendor}`);
      setConnectionState('connecting');
      
      addLog(`[BRIDGE] Compiling OS-specific execution repositories...`);
      
      setTimeout(() => {
        if (detectedVendor === 'Google') {
          addLog(`[BRIDGE] Compiling Android SDK & ADB binaries...`);
          addLog(`[BRIDGE] Translating connection to ADB over BLE...`);
        } else if (detectedVendor === 'Microsoft') {
          addLog(`[BRIDGE] Compiling Win32 BT drivers & PowerShell host...`);
          addLog(`[BRIDGE] Translating connection to PS Remoting...`);
        } else if (detectedVendor === 'Apple') {
          addLog(`[BRIDGE] Compiling MobileDevice framework...`);
          addLog(`[BRIDGE] Establishing serial over GATT...`);
        }

        const stack = MOCK_HARDWARE_STACK[detectedVendor] || MOCK_HARDWARE_STACK['Unknown'];
        setHardwareStack(stack);
        
        addLog(`[SYSTEM] Hardware stack mapped. Identifying missing drivers...`);
        
        setTimeout(() => {
          // Simulate driver installation
          const updatedStack = stack.map(comp => 
            comp.status === 'Missing' ? { ...comp, status: 'Injected' as const } : comp
          );
          setHardwareStack(updatedStack);
          
          addLog(`[SYSTEM] Missing drivers injected dynamically into target memory.`);
          addLog(`[SUCCESS] Remote execution bridge established. Awaiting deployment commands.`);
          setConnectionState('connected');
        }, 2000);

      }, 1500);
    }, 1500);
  };

  const handleDeploy = () => {
    if (connectionState !== 'connected') return;
    
    addLog(`[USER] Initiating payload deployment to ${macId}...`);
    addLog(`[SYSTEM] Requesting remote execution confirmation...`);
    
    setTimeout(() => {
      addLog(`[SYSTEM] Target authorized connection.`);
      if (vendor === 'Google') {
        addLog(`[EXEC] adb -s ${macId} shell "nohup ./remote_daemon &"`);
      } else if (vendor === 'Microsoft') {
        addLog(`[EXEC] Invoke-Command -ComputerName ${macId} -ScriptBlock { Start-Process remote_daemon.exe }`);
      } else {
        addLog(`[EXEC] ./deploy_payload.sh --target ${macId}`);
      }
      addLog(`[SUCCESS] Remote execution active. Telemetry pipe open.`);
    }, 1000);
  };

  return (
    <div className="flex flex-col h-full space-y-4 max-w-6xl mx-auto w-full">
      <div className="flex items-center justify-between">
        <h2 className="text-2xl font-bold text-primary uppercase tracking-widest flex items-center gap-3">
          <Server className="w-6 h-6" />
          Messenger Pipe & Remote Exec
        </h2>
        <div className="flex items-center gap-2">
          <span className="text-xs font-mono text-muted-foreground uppercase">AI Debugger:</span>
          <Badge variant="outline" className={`text-[10px] ${connectionState === 'connected' ? 'text-green-400 border-green-400/50' : 'text-muted-foreground'}`}>
            {connectionState === 'connected' ? 'ACTIVE' : 'STANDBY'}
          </Badge>
        </div>
      </div>

      <div className="grid grid-cols-1 lg:grid-cols-3 gap-4 flex-1 min-h-0">
        
        {/* Left Column: Terminal & Controls */}
        <div className="col-span-1 lg:col-span-2 flex flex-col space-y-4">
          <Card className="bg-[#0d0d0d] border-[#222]">
            <CardHeader className="p-4 border-b border-[#222]">
              <CardTitle className="text-xs uppercase tracking-widest text-muted-foreground flex items-center gap-2">
                <Bluetooth className="w-4 h-4" />
                Target Acquisition
              </CardTitle>
            </CardHeader>
            <CardContent className="p-4 flex gap-4 items-end">
              <div className="flex-1 space-y-2">
                <label className="text-[10px] uppercase font-bold text-muted-foreground">Target MAC ID</label>
                <Input 
                  value={macId}
                  onChange={(e) => setMacId(e.target.value)}
                  placeholder="00:1A:11:XX:XX:XX"
                  className="bg-[#111] border-[#333] font-mono text-xs uppercase"
                  disabled={connectionState !== 'disconnected'}
                />
              </div>
              <Button 
                onClick={handleConnect}
                disabled={connectionState !== 'disconnected' || !macId}
                className="w-32 bg-primary/20 text-primary border border-primary/50 hover:bg-primary/30 h-9"
              >
                <Link className="w-4 h-4 mr-2" />
                Connect
              </Button>
              <Button 
                onClick={handleDeploy}
                disabled={connectionState !== 'connected'}
                className="w-32 bg-red-500/20 text-red-400 border border-red-500/50 hover:bg-red-500/30 h-9"
              >
                <Play className="w-4 h-4 mr-2" />
                Deploy
              </Button>
            </CardContent>
          </Card>

          <Card className="bg-[#0a0a0a] border-[#222] flex-1 flex flex-col overflow-hidden">
            <CardHeader className="p-2 border-b border-[#222] bg-[#111]">
              <CardTitle className="text-[10px] uppercase tracking-widest text-muted-foreground flex items-center gap-2 px-2">
                <Terminal className="w-3 h-3" />
                Execution Terminal (AI Monitored)
              </CardTitle>
            </CardHeader>
            <CardContent className="p-0 flex-1 relative">
              <div 
                ref={terminalRef}
                className="absolute inset-0 p-4 font-mono text-[11px] overflow-y-auto space-y-1"
              >
                {terminalLogs.map((log, i) => (
                  <div key={i} className={`
                    ${log.includes('[ERROR]') ? 'text-red-400' : ''}
                    ${log.includes('[SUCCESS]') ? 'text-green-400' : ''}
                    ${log.includes('[AI-DEBUG]') ? 'text-cyan-400 opacity-80' : ''}
                    ${log.includes('[EXEC]') ? 'text-yellow-400 font-bold' : ''}
                    ${!log.match(/\[(ERROR|SUCCESS|AI-DEBUG|EXEC)\]/) ? 'text-gray-300' : ''}
                  `}>
                    {log}
                  </div>
                ))}
                {connectionState === 'scanning' && (
                  <div className="text-gray-400 animate-pulse">Scanning...</div>
                )}
                {connectionState === 'connecting' && (
                  <div className="text-gray-400 animate-pulse">Establishing bridge...</div>
                )}
              </div>
            </CardContent>
          </Card>
        </div>

        {/* Right Column: Hardware Stack */}
        <div className="col-span-1 flex flex-col space-y-4">
          <Card className="bg-[#0d0d0d] border-[#222] flex-1 flex flex-col">
            <CardHeader className="p-4 border-b border-[#222]">
              <CardTitle className="text-xs uppercase tracking-widest text-muted-foreground flex items-center gap-2">
                <Cpu className="w-4 h-4" />
                Remote Hardware Stack
              </CardTitle>
            </CardHeader>
            <CardContent className="p-4 flex-1 overflow-y-auto space-y-4">
              <div className="flex items-center justify-between p-3 bg-[#111] border border-[#333] rounded-lg">
                <div className="flex flex-col">
                  <span className="text-[10px] uppercase text-muted-foreground">Detected Vendor</span>
                  <span className="text-sm font-bold text-primary">{vendor}</span>
                </div>
                <Activity className={`w-5 h-5 ${connectionState === 'connected' ? 'text-green-400' : 'text-muted-foreground'}`} />
              </div>

              <div className="space-y-2">
                <span className="text-[10px] uppercase font-bold text-muted-foreground tracking-widest">Component Capabilities</span>
                
                {hardwareStack.length === 0 ? (
                  <div className="text-[10px] text-muted-foreground italic p-4 text-center border border-dashed border-[#333] rounded">
                    Awaiting connection to map hardware stack...
                  </div>
                ) : (
                  <div className="space-y-2">
                    {hardwareStack.map((comp) => (
                      <div key={comp.id} className="p-3 bg-[#1a1a1a] border border-[#222] rounded space-y-2">
                        <div className="flex justify-between items-start">
                          <span className="text-xs font-bold text-gray-200">{comp.component}</span>
                          <span className={`text-[9px] px-1.5 py-0.5 rounded uppercase font-bold ${
                            comp.status === 'Installed' ? 'bg-green-500/20 text-green-400' :
                            comp.status === 'Injected' ? 'bg-yellow-500/20 text-yellow-400' :
                            'bg-red-500/20 text-red-400'
                          }`}>
                            {comp.status}
                          </span>
                        </div>
                        <div className="flex flex-col gap-1">
                          <div className="flex justify-between text-[10px]">
                            <span className="text-muted-foreground">Driver:</span>
                            <span className="font-mono text-gray-400">{comp.driver}</span>
                          </div>
                          <div className="flex justify-between text-[10px]">
                            <span className="text-muted-foreground">Capability:</span>
                            <span className="text-cyan-400">{comp.capability}</span>
                          </div>
                        </div>
                      </div>
                    ))}
                  </div>
                )}
              </div>
            </CardContent>
          </Card>
        </div>

      </div>
    </div>
  );
}