src/components/control-tray/ControlTray.tsx

import cn from "classnames";
import React, { memo, ReactNode, RefObject, useEffect, useRef, useState } from "react";
import { useLiveAPIContext } from "../../contexts/LiveAPIContext";
// Hooks for webcam/screen capture if still used with new UI
import { useScreenCapture } from "../../hooks/use-screen-capture";
import { useWebcam } from "../../hooks/use-webcam";
import { AudioRecorder } from "../../lib/audio-recorder";
import { isIOS } from "../../lib/platform";
// import AudioPulse from "../audio-pulse/AudioPulse"; // Original pulse, new design has different aesthetic
// import "./control-tray.scss"; // Comment out or remove if styles conflict

// --- SVG Icon Definitions (inline for simplicity) ---
const SvgPauseIcon = () => <svg width="37" height="37" viewBox="0 0 37 37" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M15.9872 29.6198V8.28342C15.9872 6.25781 15.132 5.44757 12.9713 5.44757H7.52469C5.36404 5.44757 4.50879 6.25781 4.50879 8.28342V29.6198C4.50879 31.6454 5.36404 32.4556 7.52469 32.4556H12.9713C15.132 32.4556 15.9872 31.6454 15.9872 29.6198Z" fill="#BE123C"/><path opacity="0.4" d="M31.5175 29.6198V8.28342C31.5175 6.25781 30.6622 5.44757 28.5016 5.44757H23.055C20.9093 5.44757 20.0391 6.25781 20.0391 8.28342V29.6198C20.0391 31.6454 20.8943 32.4556 23.055 32.4556H28.5016C30.6622 32.4556 31.5175 31.6454 31.5175 29.6198Z" fill="#BE123C"/></svg>;
const SvgMicrophoneIcon = () => <svg width="69" height="68" viewBox="0 0 69 68" fill="none" xmlns="http://www.w3.org/2000/svg"><path opacity="0.4" d="M49.9479 27.1824C49.0803 27.1824 48.3907 27.872 48.3907 28.7396V32.2544C48.3907 40.1293 41.984 46.5361 34.109 46.5361C26.234 46.5361 19.8273 40.1293 19.8273 32.2544V28.7173C19.8273 27.8497 19.1377 27.1601 18.2701 27.1601C17.4025 27.1601 16.7129 27.8497 16.7129 28.7173V32.2321C16.7129 41.2861 23.6758 48.7384 32.5518 49.5393V54.2776C32.5518 55.1452 33.2414 55.8348 34.109 55.8348C34.9766 55.8348 35.6662 55.1452 35.6662 54.2776V49.5393C44.52 48.7607 51.5051 41.2861 51.5051 32.2321V28.7173C51.4829 27.872 50.7933 27.1824 49.9479 27.1824Z" fill="#BE123C"/><path d="M34.1099 11.3434C28.682 11.3434 24.2773 15.7481 24.2773 21.176V32.5658C24.2773 37.9938 28.682 42.3984 34.1099 42.3984C39.5379 42.3984 43.9425 37.9938 43.9425 32.5658V21.176C43.9425 15.7481 39.5379 11.3434 34.1099 11.3434ZM37.0241 26.8042C36.8684 27.3826 36.3567 27.7608 35.7784 27.7608C35.6671 27.7608 35.5559 27.7385 35.4447 27.7163C34.5771 27.4716 33.665 27.4716 32.7974 27.7163C32.0856 27.9165 31.396 27.4938 31.218 26.8042C31.0178 26.1146 31.4404 25.4027 32.1301 25.2247C33.4426 24.8688 34.8218 24.8688 36.1343 25.2247C36.8017 25.4027 37.2021 26.1146 37.0241 26.8042ZM38.2031 22.4885C38.0029 23.0224 37.5135 23.3339 36.9796 23.3339C36.8239 23.3339 36.6904 23.3116 36.5347 23.2671C34.9775 22.6887 33.2423 22.6887 31.6852 23.2671C31.0178 23.5118 30.2614 23.1559 30.0167 22.4885C29.772 21.8212 30.128 21.0648 30.7953 20.8423C32.9309 20.0637 35.289 20.0637 37.4245 20.8423C38.0919 21.087 38.4478 21.8212 38.2031 22.4885Z" fill="#BE123C"/></svg>;
const SvgCameraIcon = () => <svg width="40" height="40" viewBox="0 0 40 40" fill="none" xmlns="http://www.w3.org/2000/svg"><path opacity="0.4" d="M21.8118 5.52235H11.9265C6.29183 5.52235 4.10059 7.7136 4.10059 13.3482V26.5286C4.10059 30.318 6.16002 34.3545 11.9265 34.3545H21.8118C27.4464 34.3545 29.6376 32.1633 29.6376 26.5286V13.3482C29.6376 7.7136 27.4464 5.52235 21.8118 5.52235Z" fill="#2252A0"/><path d="M19.3406 18.9169C21.0512 18.9169 22.438 17.5302 22.438 15.8195C22.438 14.1089 21.0512 12.7222 19.3406 12.7222C17.6299 12.7222 16.2432 14.1089 16.2432 15.8195C16.2432 17.5302 17.6299 18.9169 19.3406 18.9169Z" fill="#2252A0"/><path d="M36.0629 10.3332C35.3874 9.98721 33.9705 9.5918 32.0429 10.9428L29.6045 12.6562C29.621 12.8869 29.6374 13.1011 29.6374 13.3482V26.5286C29.6374 26.7758 29.6045 26.9899 29.6045 27.2206L32.0429 28.934C33.0643 29.659 33.954 29.8896 34.6625 29.8896C35.2721 29.8896 35.7499 29.7249 36.0629 29.5601C36.7384 29.2141 37.8752 28.275 37.8752 25.919V13.9743C37.8752 11.6183 36.7384 10.6792 36.0629 10.3332Z" fill="#2252A0"/></svg>;
const SvgStopCamIcon = () => <svg width="40" height="40" viewBox="0 0 40 40" fill="none" xmlns="http://www.w3.org/2000/svg"><path opacity="0.4" d="M29.0785 10.8076L6.91966 32.9665C4.61316 31.5002 3.70703 28.8807 3.70703 26.36V13.18C3.70703 7.54555 5.89821 5.35437 11.5327 5.35437H21.4177C26.1789 5.35437 28.4854 6.9195 29.0785 10.8076Z" fill="#2252A0"/><path opacity="0.4" d="M29.2427 15.2394V26.36C29.2427 26.4918 29.2262 26.5907 29.2262 26.706C29.2262 26.8213 29.2097 26.9366 29.2097 27.052H29.2262C29.045 32.1757 26.8208 34.1856 21.417 34.1856H11.532C11.1202 34.1856 10.7412 34.1692 10.3623 34.1197L29.2427 15.2394Z" fill="#2252A0"/><path opacity="0.4" d="M29.21 27.052C29.21 26.9366 29.2264 26.8213 29.2264 26.706C29.2429 26.8213 29.2429 26.9366 29.2264 27.052H29.21Z" fill="#2252A0"/><path opacity="0.4" d="M29.2264 12.4716C29.2429 12.5869 29.2429 12.7187 29.2264 12.834C29.2264 12.7187 29.21 12.6034 29.21 12.488L29.2264 12.4716Z" fill="#2252A0"/><path d="M37.4804 13.8061V25.734C37.4804 28.0899 36.3436 29.029 35.6682 29.3749C35.3551 29.5397 34.8774 29.7209 34.2678 29.7209C33.5594 29.7209 32.6697 29.4903 31.6483 28.7654L29.2264 27.052H29.21C29.21 26.9366 29.2264 26.8213 29.2264 26.706C29.2264 26.5907 29.2429 26.4918 29.2429 26.36V15.2394L34.6302 9.85205C35.0751 9.885 35.421 10.0168 35.6682 10.1651C36.3436 10.5111 37.4804 11.4501 37.4804 13.8061Z" fill="#2252A0"/><path d="M35.8666 3.67393C35.3723 3.17968 34.565 3.17968 34.0708 3.67393L3.6744 34.0868C3.18015 34.581 3.18015 35.3883 3.6744 35.8826C3.92152 36.1132 4.23455 36.245 4.56405 36.245C4.89355 36.245 5.20657 36.1132 5.4537 35.8661L35.8666 5.45323C36.3773 4.95898 36.3773 4.16818 35.8666 3.67393Z" fill="#2252A0"/></svg>;
const SvgSwitchCameraIcon = () => <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" strokeWidth="2" strokeLinecap="round" strokeLinejoin="round" className="w-[22px] h-[22px] group-hover:rotate-[360deg] transition-transform duration-300 ease-in-out"><path d="M11 19H4a2 2 0 0 1-2-2V7a2 2 0 0 1 2-2h5"/><path d="M13 5h7a2 2 0 0 1 2 2v10a2 2 0 0 1-2 2h-5"/><path d="m17 12-3-3 3-3"/><path d="m7 12 3 3-3 3"/></svg>;
const SvgHumanIcon = () => <svg width="70" height="70" viewBox="0 0 88 89" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M75.1481 81.6361H12.9259C9.66667 81.6361 7 78.9721 7 75.7161V58.5112C7 57.5862 7 57.1052 7.44444 56.2172C8.85185 52.9612 13 50.2232 19.4815 47.8922C24.1111 56.6982 33.3704 62.6921 44 62.6921C54.6296 62.6921 63.9259 56.6982 68.5185 47.8922C75 50.1862 79.1852 52.9982 80.5556 56.2172C81 56.6612 81 57.6232 81 58.5112V75.7161C81 78.9721 78.3333 81.6361 75.0741 81.6361H75.1481Z" stroke="#FCFCFC" strokeWidth="6.42146" strokeLinecap="round" strokeLinejoin="round"/><path d="M44.0371 50.1862C33.8519 50.1862 25.5186 41.8612 25.5186 31.6863V26.1363C25.5186 15.9613 33.8519 7.63635 44.0371 7.63635C54.2223 7.63635 62.5556 15.9613 62.5556 26.1363V31.6863C62.5556 41.8612 54.2223 50.1862 44.0371 50.1862Z" stroke="#FCFCFC" strokeWidth="6.42146" strokeLinecap="round" strokeLinejoin="round"/></svg>;
// --- End SVG Icon Definitions ---


// Styles for ControlTray specifically (Tailwind will handle most, this is for custom parts)
const controlTrayStyles = `
    .footer-controls {
        width: 100%; display: flex; gap: 1rem; position: absolute;
        bottom: 0; padding: 2rem 3rem; align-items: center;
        z-index: 20; /* Ensure it's above video but below modals */
    }
    .footer-controls.layout-default { justify-content: space-between; }
    .footer-controls.layout-with-small-logo { justify-content: space-around; }

    .control-button {
        height: 80px; width: 80px; border-radius: 9999px; /* rounded-full */
        padding: 0; display: flex; align-items: center; justify-content: center;
        border-width: 1px; /* border */
        /* border-color will come from var(--border) via Tailwind's border-border */
        box-shadow: 0 4px 6px -1px rgba(0,0,0,0.1), 0 2px 4px -1px rgba(0,0,0,0.06); /* shadow-md */
        cursor: pointer;
        transition: transform 0.2s ease-out, box-shadow 0.2s ease-out;
    }
    .control-button:hover {
        transform: scale(1.05);
        box-shadow: 0 10px 15px -3px rgba(0,0,0,0.1), 0 4px 6px -2px rgba(0,0,0,0.05); /* shadow-lg */
    }
    .cam-button-color { background-color: #E0ECFF; } /* Light blue */
    .mic-button-color { background-color: #fecdd3; } /* Light red */
    .dark .cam-button-color { background-color: #223355; } /* Darker blue */
    .dark .mic-button-color { background-color: #5C2129; } /* Darker red */

    .switch-camera-button-container {
        position: absolute; bottom: calc(100% + 0.65rem); left: 50%;
        z-index: 5; opacity: 0;
        transform: translateY(15px) scale(0.7) translateX(-50%);
        pointer-events: none;
        transition: opacity 0.35s cubic-bezier(0.68, -0.55, 0.27, 1.55), transform 0.35s cubic-bezier(0.68, -0.55, 0.27, 1.55);
        transform-origin: center bottom;
    }
    .switch-camera-button-container.visible {
        opacity: 1; transform: translateY(0) scale(1) translateX(-50%); pointer-events: auto;
    }
    .switch-camera-button-content {
        width: 48px; height: 48px;
        background-color: var(--background); /* Uses themed background */
        border: 1px solid var(--border); /* Uses themed border */
        border-radius: 9999px; /* rounded-full */
        display: flex; align-items: center; justify-content: center;
        box-shadow: 0 5px 10px rgba(0,0,0,0.12), 0 2px 4px rgba(0,0,0,0.08);
        cursor: pointer; transform-origin: center;
        transition: transform 0.2s ease-out, box-shadow 0.2s ease-out;
    }
    .switch-camera-button-content:hover {
        transform: scale(1.12) rotate(-6deg);
        box-shadow: 0 7px 15px rgba(0,0,0,0.18), 0 3px 6px rgba(0,0,0,0.12);
    }
    .switch-camera-button-content:active { transform: scale(1.03) rotate(0deg); }
    .switch-camera-button-content svg {
      /* stroke is set by var(--foreground) or Tailwind text color */
      /* width/height set on SvgSwitchCameraIcon component */
    }
    #small-logo-container.flex { display: flex; align-items: center; justify-content: center; }

    /* Ping animation for logos */
    @keyframes ping { 75%, 100% { transform: scale(2); opacity: 0; } }
    .animate-ping { animation: ping 1s cubic-bezier(0, 0, 0.2, 1) infinite; }
`;


export type ControlTrayProps = {
  videoRef: RefObject<HTMLVideoElement>;
  children?: ReactNode; // Kept for extensibility, not used by new HTML structure
  supportsVideo: boolean;
  onVideoStreamChange?: (stream: MediaStream | null) => void;
};

// React component for the logo (replaces createLogoHTML)
const LogoDisplay = ({ isMini, isHumanActive, isAiActive = false /* Not used in HTML */ }) => {
    if (!isHumanActive && !isAiActive) return null; // Don't render if no one is active

    const size = isMini ? 80 : 200;
    const iconSize = isMini ? 35 : 70;
    const insetBase = isMini
        ? { ping: 10, outer: 0, mid: 5, inner: 12, icon: 22 }
        : { ping: 40, outer: 0, mid: 20, inner: 50, icon: 65 }; // Values from HTML's JS

    const bgColorBase = isHumanActive ? 'blue' : (isAiActive ? 'green' : 'gray');
    // Tailwind classes for dynamic background colors
    const pingColor = `bg-${bgColorBase}-200`;
    const outerRingColor = `bg-${bgColorBase}-200`;
    const midRingColor = `bg-${bgColorBase}-300`;
    const innerRingColor = `bg-${bgColorBase}-400`;

    return (
        <div className="relative" style={{ width: `${size}px`, height: `${size}px` }}>
            <div className={`absolute rounded-full opacity-50 animate-ping ${pingColor}`} style={{ inset: `${insetBase.ping}px` }}></div>
            <div className={`absolute inset-0 rounded-full opacity-50 ${outerRingColor}`} style={{inset: `${insetBase.outer}px`}}></div>
            <div className={`absolute rounded-full opacity-50 ${midRingColor}`} style={{ inset: `${insetBase.mid}px` }}></div>
            <div className={`absolute rounded-full opacity-50 ${innerRingColor}`} style={{ inset: `${insetBase.inner}px` }}></div>
            <div className="z-10 absolute" style={{ inset: `${insetBase.icon}px`, display: 'flex', alignItems: 'center', justifyContent: 'center' }}>
                {isHumanActive && <SvgHumanIcon />}
                {/* Add AI icon if isAiActive is true and you have one */}
            </div>
        </div>
    );
};


function ControlTray({
  videoRef,
  onVideoStreamChange = () => {},
  supportsVideo,
}: ControlTrayProps) {
  const webcam = useWebcam();
  const screenCapture = useScreenCapture(); // Kept if screen share button is re-added
  
  const [isCamActive, setIsCamActive] = useState(false); // New state for camera (visual)
  const [isMicActive, setIsMicActive] = useState(false); // New state for mic (visual)

  const [currentFacingMode, setCurrentFacingMode] = useState<'user' | 'environment'>('user');
  const [isSwitchingCamera, setIsSwitchingCamera] = useState(false);
  
  const [audioRecorder] = useState(() => new AudioRecorder());
  // const [inVolume, setInVolume] = useState(0); // For original AudioPulse
  
  const { client, connected, connect, disconnect, volume, setConfig } = useLiveAPIContext();

  // Connect to API when component mounts or when mic/cam are first activated
  useEffect(() => {
    // This is a placeholder for connection logic.
    // The original app had an explicit connect button.
    // You might want to connect automatically, or when mic/cam is first toggled.
    if (!connected && (isMicActive || isCamActive)) {
        // connect(); // Example: connect when mic/cam active
    }
  }, [isMicActive, isCamActive, connected, connect]);


  // Audio recording based on isMicActive AND connected
  useEffect(() => {
    const onData = (base64: string) => {
      if (client && connected && isMicActive) { // Check isMicActive
        client.sendRealtimeInput([{ mimeType: "audio/pcm;rate=16000", data: base64 }]);
      }
    };
    // const onVolume = setInVolume; // For original AudioPulse

    if (connected && isMicActive && audioRecorder) {
      audioRecorder.on("data", onData)/*.on("volume", onVolume)*/.start();
    } else if (audioRecorder) {
      audioRecorder.stop();
    }
    return () => {
      if (audioRecorder) {
        audioRecorder.off("data", onData)/*.off("volume", onVolume)*/.stop();
      }
    };
  }, [connected, client, isMicActive, audioRecorder]);

  // Video stream management
  useEffect(() => {
    const videoElement = videoRef.current;
    if (videoElement) {
      if (isCamActive && webcam.stream) {
        videoElement.srcObject = webcam.stream;
        videoElement.play().catch(e => console.warn("Video play failed:", e));
        onVideoStreamChange(webcam.stream);
      } else {
        videoElement.srcObject = null;
        onVideoStreamChange(null);
      }
    }
  }, [isCamActive, webcam.stream, videoRef, onVideoStreamChange]);


  // Video frame sending (simplified, adapt frame rate as needed)
  useEffect(() => {
    let frameSenderId: number;
    const sendFrame = () => {
      if (connected && isCamActive && webcam.stream && videoRef.current && client) {
        const video = videoRef.current;
        const canvas = document.createElement('canvas'); // Temporary canvas
        const scale = 0.25;
        canvas.width = video.videoWidth * scale;
        canvas.height = video.videoHeight * scale;
        if (canvas.width > 0 && canvas.height > 0) {
          const ctx = canvas.getContext("2d");
          if (ctx) {
            ctx.drawImage(video, 0, 0, canvas.width, canvas.height);
            const base64 = canvas.toDataURL("image/jpeg", 0.8);
            const data = base64.slice(base64.indexOf(",") + 1);
            client.sendRealtimeInput([{ mimeType: "image/jpeg", data }]);
          }
        }
      }
      frameSenderId = window.setTimeout(sendFrame, 1000 / 2); // ~2 FPS, adjust as needed
    };

    if (isCamActive && connected) {
      frameSenderId = window.setTimeout(sendFrame, 200); // Initial delay
    }
    return () => clearTimeout(frameSenderId);
  }, [connected, isCamActive, webcam.stream, client, videoRef]);


  const handleToggleMic = async () => {
    if (!connected) { // Auto-connect if not connected
        try {
            await connect();
        } catch (err) {
            console.error("Failed to connect on mic toggle:", err);
            return; // Don't toggle mic if connection fails
        }
    }
    // After ensuring connection (or if already connected)
    setIsMicActive(prev => !prev);
    // The audioRecorder useEffect will handle starting/stopping based on isMicActive & connected
  };

  const handleToggleCam = async () => {
    if (isSwitchingCamera) return;
    const newCamState = !isCamActive;

    if (newCamState) { // Turning camera ON
        if (!connected) { // Auto-connect if not connected
            try {
                await connect();
            } catch (err) {
                console.error("Failed to connect on cam toggle:", err);
                return; 
            }
        }
      try {
        await webcam.start({ video: { facingMode: currentFacingMode }, audio: false });
        setIsCamActive(true);
      } catch (error) {
        console.error("Error starting webcam:", error);
        setIsCamActive(false); // Ensure it's off if start failed
      }
    } else { // Turning camera OFF
      webcam.stop();
      setIsCamActive(false);
      // Optional: Disconnect if both mic and cam are off
      // if (!isMicActive && !newCamState && connected) {
      //   disconnect();
      // }
    }
  };

  const handleSwitchCamera = async () => {
    if (!isCamActive || isSwitchingCamera) return;
    setIsSwitchingCamera(true);
    const newFacingMode = currentFacingMode === 'user' ? 'environment' : 'user';
    try {
      await webcam.stop(); // Stop current stream
      await webcam.start({ video: { facingMode: { exact: newFacingMode } }, audio: false });
      setCurrentFacingMode(newFacingMode);
      // onVideoStreamChange already handled by webcam.stream effect
    } catch (error) {
      console.error("Error switching camera:", error);
      // Try to restore previous mode or stop
      try {
        await webcam.start({ video: { facingMode: currentFacingMode }, audio: false });
      } catch (restoreError) {
        console.error("Error restoring camera:", restoreError);
        setIsCamActive(false); // Turn off cam if all fails
        webcam.stop();
      }
    } finally {
      setIsSwitchingCamera(false);
    }
  };

  // Determine layout for footer controls
  const footerLayoutClass = isCamActive ? 'layout-with-small-logo' : 'layout-default';

  // Update large logo visibility in App.tsx (this is tricky, App.tsx owns that element)
  // For simplicity, we'll just log. A better way is via context or Zustand.
  useEffect(() => {
    const largeLogoEl = document.getElementById('large-logo-container');
    if (largeLogoEl) {
        if (!isCamActive && isMicActive) {
            largeLogoEl.innerHTML = ''; // Clear previous
            const logoDiv = document.createElement('div');
            // This is a hacky way to render the LogoDisplay component's HTML.
            // ReactDOM.render(<LogoDisplay isMini={false} isHumanActive={true} />, logoDiv); // This won't work directly here.
            // For now, just show/hide. Proper rendering needs state lift or context.
            // You might need to pass isMicActive up to App.tsx to render the large logo there.
            largeLogoEl.classList.remove('hidden');
            largeLogoEl.classList.add('flex');
            // Ideally, render <LogoDisplay isMini={false} isHumanActive={true} /> inside it from App.tsx
        } else {
            largeLogoEl.classList.add('hidden');
            largeLogoEl.classList.remove('flex');
            largeLogoEl.innerHTML = '';
        }
    }
  }, [isCamActive, isMicActive]);


  return (
    <>
      <style>{controlTrayStyles}</style>
      <div id="footer-controls" className={`footer-controls ${footerLayoutClass}`}>
        {/* Cam Button Wrapper (Left) */}
        <div id="cam-button-wrapper" className="relative flex justify-center">
          <button
            id="cam-button"
            className="control-button cam-button-color"
            onClick={handleToggleCam}
            disabled={isSwitchingCamera || !supportsVideo}
          >
            {isCamActive ? <SvgStopCamIcon /> : <SvgCameraIcon />}
          </button>
          <div
            id="switch-camera-button-container"
            className={cn("switch-camera-button-container", { visible: isCamActive && !isSwitchingCamera })}
          >
            <button
              id="switch-camera-button"
              aria-label="Switch Camera"
              className="switch-camera-button-content group" // Added group for icon hover
              onClick={handleSwitchCamera}
              disabled={!isCamActive || isSwitchingCamera}
            >
              <SvgSwitchCameraIcon />
            </button>
          </div>
        </div>

        {/* Small Logo Container (Center, if cam active) */}
        {isCamActive && (
            <div id="small-logo-container" className="flex">
                 <LogoDisplay isMini={true} isHumanActive={true} />
            </div>
        )}


        {/* Mic Button (Right) */}
        <button
          id="mic-button"
          className="control-button mic-button-color"
          onClick={handleToggleMic}
          disabled={isSwitchingCamera} // Mic can be used even if video not supported
        >
          {isMicActive ? <SvgPauseIcon /> : <SvgMicrophoneIcon />}
        </button>
      </div>
    </>
  );
}

export default memo(ControlTray);