bambuddy/frontend/src/components/CameraTile.tsx

import { useEffect, useMemo, useRef, useState } from 'react';
import { useTranslation } from 'react-i18next';
import { AlertTriangle, VideoOff, WifiOff } from 'lucide-react';
import { getAuthToken, withStreamToken } from '../api/client';
import { formatDuration } from '../utils/date';
import { useStreamImgRef } from '../hooks/useStreamImgRef';

export type CameraTileMode = 'live' | 'snapshot' | 'paused';
export type CameraTileStatusMode = 'off' | 'compact' | 'full';

interface CameraTileProps {
  printerId: number;
  printerName: string;
  cameraRotation?: number;
  mode: CameraTileMode;
  snapshotIntervalMs: number;
  connected: boolean;
  onClick?: () => void;
  // Optional status overlay — wired by CameraWall from the shared
  // ['printerStatus', id] query. All optional so existing tests don't break.
  statusMode?: CameraTileStatusMode;
  printerState?: string | null;
  progress?: number | null;
  remainingMin?: number | null;
  layerNum?: number | null;
  totalLayers?: number | null;
  printName?: string | null;
  hmsErrorCount?: number;
}

// Tiles render lighter than EmbeddedCameraViewer's full window: lower fps,
// no drag/resize/zoom shell, and snapshot fallback when off-cap. The server
// still does the MJPEG fan-out, so per-tile cost is one TLS pull on the wire.
const LIVE_FPS = 8;
const LIVE_RENEWAL_MS = 60_000;
// Between closing the old stream and opening the new one: a USB camera serves
// one reader, and the old ffmpeg must let go of the device first.
const RENEWAL_GAP_MS = 200;
// A renewal that failed is tried once more after this.
const RENEWAL_RETRY_MS = 5_000;
// A renewal with no frame by then has failed too: a snapshot camera that is
// down keeps its stream open without one, and the img fires no event at all.
const RENEWAL_FIRST_FRAME_MS = 15_000;

type StatusBucket = 'printing' | 'paused' | 'finished' | 'error' | 'idle';

function classifyState(state: string | null | undefined, hmsErrorCount: number): StatusBucket {
  if (hmsErrorCount > 0) return 'error';
  switch (state) {
    case 'RUNNING':
      return 'printing';
    case 'PAUSE':
      return 'paused';
    case 'FINISH':
    case 'FAILED':
      return 'finished';
    default:
      return 'idle';
  }
}

const BUCKET_CHIP_CLASS: Record<StatusBucket, string> = {
  printing: 'bg-bambu-green/85 text-black',
  paused: 'bg-amber-500/85 text-black',
  finished: 'bg-sky-500/80 text-white',
  error: 'bg-red-500/85 text-white',
  idle: 'bg-bambu-dark-tertiary/80 text-bambu-gray',
};

export function CameraTile({
  printerId,
  printerName,
  cameraRotation = 0,
  mode,
  snapshotIntervalMs,
  connected,
  onClick,
  statusMode = 'off',
  printerState = null,
  progress = null,
  remainingMin = null,
  layerNum = null,
  totalLayers = null,
  printName = null,
  hmsErrorCount = 0,
}: CameraTileProps) {
  const { t } = useTranslation();
  const [bust, setBust] = useState(0);
  const [errored, setErrored] = useState(false);
  // The old stream is closed and the new one not yet due: the img renders with
  // no src (removing it closes the connection), so no re-render reopens it early.
  const [reopening, setReopening] = useState(false);
  const lastModeRef = useRef<CameraTileMode>(mode);
  // Every img this tile drops (leaving live mode, a renewal, unmounting) has
  // its stream closed; the /camera/stop sent on leaving live mode is skipped
  // while a dropped img still holds it.
  const imgRef = useRef<HTMLImageElement | null>(null);
  const attachImg = useStreamImgRef(imgRef);
  // 'renewed': the current img is a renewal not yet showing a frame; 'retried':
  // it is the one retry of a renewal that failed.
  const renewalRef = useRef<'none' | 'renewed' | 'retried'>('none');
  const renewalTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);
  const firstFrameTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);

  // Tell the backend to release its MJPEG transcoder when this tile stops
  // being live — either by unmounting or by transitioning to snapshot/paused.
  // EmbeddedCameraViewer uses the same /camera/stop with keepalive on unmount.
  useEffect(() => {
    const wasLive = lastModeRef.current === 'live';
    const isLive = mode === 'live';
    lastModeRef.current = mode;
    if (wasLive && !isLive) {
      const headers: Record<string, string> = {};
      const token = getAuthToken();
      if (token) headers['Authorization'] = `Bearer ${token}`;
      fetch(`/api/v1/printers/${printerId}/camera/stop`, {
        method: 'POST',
        keepalive: true,
        headers,
      }).catch(() => {});
    }
    setErrored(false);
    setBust((b) => b + 1);
  }, [mode, printerId]);

  useEffect(() => {
    return () => {
      if (lastModeRef.current === 'live') {
        const headers: Record<string, string> = {};
        const token = getAuthToken();
        if (token) headers['Authorization'] = `Bearer ${token}`;
        fetch(`/api/v1/printers/${printerId}/camera/stop`, {
          method: 'POST',
          keepalive: true,
          headers,
        }).catch(() => {});
      }
    };
  }, [printerId]);

  useEffect(() => {
    if (mode !== 'snapshot') return;
    const interval = setInterval(() => setBust((b) => b + 1), snapshotIntervalMs);
    return () => clearInterval(interval);
  }, [mode, snapshotIntervalMs]);

  // A stream the server ended (its re-check, the camera dropping) fires no img
  // event: the tile would keep its last frame. Renew it, closing the old
  // connection first (a detached img keeps it open). No /camera/stop: other
  // viewers of the printer share the stream. Not in a hidden page, where the
  // stream token is not refreshed. A tile showing no signal is left alone:
  // each failed camera img makes the app re-mint the stream token and
  // reconnect every camera on the page.
  //
  // Built once: they only use refs and state setters, and call each other.
  const { clearFirstFrameTimer, streamFailed, reopenAfter } = useMemo(() => {
    const clearFirstFrameTimer = () => {
      if (firstFrameTimerRef.current !== null) clearTimeout(firstFrameTimerRef.current);
      firstFrameTimerRef.current = null;
    };

    // The stream failed (an error, or a renewal with no frame in time): a
    // renewal is retried once, anything else shows no signal.
    const streamFailed = () => {
      clearFirstFrameTimer();
      if (renewalRef.current === 'renewed') {
        renewalRef.current = 'retried';
        reopenAfter(RENEWAL_RETRY_MS);
        return;
      }
      renewalRef.current = 'none';
      setErrored(true);
    };

    const reopenAfter = (ms: number) => {
      // Closed now, not at the next render: React may batch this with the reopen.
      imgRef.current?.removeAttribute('src');
      setReopening(true);
      if (renewalTimerRef.current !== null) clearTimeout(renewalTimerRef.current);
      renewalTimerRef.current = setTimeout(() => {
        renewalTimerRef.current = null;
        setReopening(false);
        setBust((b) => b + 1);
        clearFirstFrameTimer();
        // A hidden page does not load a lazy img: that time does not count.
        const deadline = () => {
          if (document.hidden) {
            firstFrameTimerRef.current = setTimeout(deadline, RENEWAL_FIRST_FRAME_MS);
            return;
          }
          streamFailed();
        };
        firstFrameTimerRef.current = setTimeout(deadline, RENEWAL_FIRST_FRAME_MS);
      }, ms);
    };

    return { clearFirstFrameTimer, streamFailed, reopenAfter };
  }, []);

  useEffect(() => {
    renewalRef.current = 'none';
    if (mode !== 'live' || !connected) return;
    const interval = setInterval(() => {
      if (document.hidden || !imgRef.current || renewalTimerRef.current !== null) return;
      renewalRef.current = 'renewed';
      reopenAfter(RENEWAL_GAP_MS);
    }, LIVE_RENEWAL_MS);
    return () => {
      clearInterval(interval);
      if (renewalTimerRef.current !== null) clearTimeout(renewalTimerRef.current);
      renewalTimerRef.current = null;
      clearFirstFrameTimer();
      setReopening(false);
    };
  }, [mode, connected, reopenAfter, clearFirstFrameTimer]);

  const liveUrl = withStreamToken(
    `/api/v1/printers/${printerId}/camera/stream?fps=${LIVE_FPS}&t=${bust}`,
  );
  const snapshotUrl = withStreamToken(
    `/api/v1/printers/${printerId}/camera/snapshot?t=${bust}`,
  );

  // A kiosk wall passes no onClick — there is no pointer at a TV, and the page
  // is authenticated by a token that cannot open the single-camera view. Render
  // the tile as plain, non-focusable content rather than a button that looks
  // clickable and then does nothing.
  const interactive = onClick != null;

  const transform = cameraRotation ? `rotate(${cameraRotation}deg)` : undefined;

  const bucket = classifyState(printerState, hmsErrorCount);
  // Hide chip for idle to keep cold walls clean; always show when something
  // is happening (printing/paused/finished/error).
  const showChip = connected && statusMode !== 'off' && bucket !== 'idle';
  const isPrintingOrPaused = bucket === 'printing' || bucket === 'paused';
  const showInfoStrip = connected && statusMode === 'full' && isPrintingOrPaused;
  const fileLabel = printName ?? null;
  const progressPct = progress != null ? Math.round(progress) : null;
  const hasLayers = layerNum != null && totalLayers != null && totalLayers > 0;
  const hasRemaining = remainingMin != null && remainingMin > 0;

  const rootClass = `group relative aspect-video w-full overflow-hidden rounded-lg border border-bambu-dark-tertiary bg-black text-left ${
    interactive ? 'focus:outline-none focus:ring-2 focus:ring-bambu-green' : 'cursor-default'
  }`;

  const content = (
    <>
      {!connected || mode === 'paused' ? (
        <div className="absolute inset-0 flex items-center justify-center bg-bambu-dark/60">
          {connected ? (
            <VideoOff className="h-8 w-8 text-bambu-gray/70" aria-hidden="true" />
          ) : (
            <WifiOff className="h-8 w-8 text-bambu-gray/70" aria-hidden="true" />
          )}
        </div>
      ) : errored ? (
        <div className="absolute inset-0 flex flex-col items-center justify-center gap-1 bg-black/80 text-bambu-gray">
          <VideoOff className="h-7 w-7" aria-hidden="true" />
          <span className="text-xs">{t('printers.camWall.noSignal')}</span>
        </div>
      ) : (
        <img
          ref={attachImg}
          key={`${mode}-${bust}`}
          src={reopening ? undefined : mode === 'live' ? liveUrl : snapshotUrl}
          alt={printerName}
          draggable={false}
          loading="lazy"
          className="h-full w-full select-none object-contain"
          style={{ transform }}
          onLoad={() => {
            clearFirstFrameTimer();
            renewalRef.current = 'none';
          }}
          onError={(e) => {
            // An img with an empty or missing src is closed, not failed.
            if (!e.currentTarget.getAttribute('src')) return;
            streamFailed();
          }}
        />
      )}

      {/* Status chip (top-left) */}
      {showChip && (
        <span
          className={`absolute left-2 top-2 flex items-center gap-1 rounded px-1.5 py-0.5 text-[10px] font-semibold uppercase tracking-wide ${BUCKET_CHIP_CLASS[bucket]}`}
        >
          {hmsErrorCount > 0 && (
            <AlertTriangle
              className="h-3 w-3"
              aria-hidden="true"
            />
          )}
          <span>{t(`printers.status.${bucket}`)}</span>
        </span>
      )}

      {/* Mode indicator (top-right) */}
      <span
        className={`absolute right-2 top-2 rounded px-1.5 py-0.5 text-[10px] font-semibold uppercase tracking-wide ${
          mode === 'live'
            ? 'bg-red-500/80 text-white'
            : mode === 'snapshot'
              ? 'bg-amber-500/70 text-black'
              : 'bg-bambu-dark-tertiary/70 text-bambu-gray'
        }`}
      >
        {mode === 'live'
          ? t('printers.camWall.live')
          : mode === 'snapshot'
            ? t('printers.camWall.snap')
            : t('printers.camWall.off')}
      </span>

      {/* Bottom overlay: name + (when full) print info */}
      <div className="absolute inset-x-0 bottom-0 bg-gradient-to-t from-black/85 via-black/55 to-transparent px-2 pb-1.5 pt-3 text-white">
        {showInfoStrip && (
          <div className="mb-0.5 space-y-0.5 text-[11px] leading-tight text-white/90">
            {fileLabel && (
              <div className="truncate" title={fileLabel}>
                {fileLabel}
              </div>
            )}
            <div className="flex flex-wrap items-center gap-x-2 gap-y-0.5 text-bambu-gray">
              {progressPct != null && (
                <span className="font-semibold text-white">{progressPct}%</span>
              )}
              {hasLayers && (
                <span>
                  {t('printers.camWall.layer', {
                    cur: layerNum,
                    total: totalLayers,
                  })}
                </span>
              )}
              {hasRemaining && (
                <span>
                  {t('printers.camWall.timeLeft', {
                    time: formatDuration((remainingMin ?? 0) * 60),
                  })}
                </span>
              )}
            </div>
          </div>
        )}
        <span className="block truncate text-xs font-medium">{printerName}</span>
      </div>
    </>
  );

  if (!interactive) {
    return (
      <div className={rootClass} title={printerName}>
        {content}
      </div>
    );
  }

  return (
    <button type="button" onClick={onClick} className={rootClass} title={printerName}>
      {content}
    </button>
  );
}