YTChatHub/client/app/dashboard/page.tsx

'use client';

import { useCallback, useEffect, useMemo, useState, useRef } from 'react';
import type { ChatMessage } from '@shared/chat';

const BACKEND_URL = process.env.NEXT_PUBLIC_BACKEND_URL ?? 'http://localhost:4100';
const POLL_INTERVAL = 2500;

// Global singleton to prevent multiple SSE connections across all renders/remounts
let globalSSEConnection: EventSource | null = null;
let globalConnectionListeners: Set<(payload: any) => void> = new Set();

export default function DashboardPage() {
  const { messages, refresh, error: pollError } = useChatMessages();
  const { selection, status: overlayStatus } = useOverlaySelection();
  const { connected, liveId, connect, disconnect, connecting } = useConnection();

  const handleSelect = useCallback(
    async (message: ChatMessage) => {
      try {
        await fetch(`${BACKEND_URL}/overlay/selection`, {
          method: 'POST',
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify({ id: message.id })
        });
      } catch (error) {
        console.error('Failed to select message', error);
      }
    },
    []
  );

  const handleClear = useCallback(async () => {
    try {
      await fetch(`${BACKEND_URL}/overlay/selection`, { method: 'DELETE' });
    } catch (error) {
      console.error('Failed to clear selection', error);
    }
  }, []);

  useEffect(() => {
    const timer = setInterval(() => {
      refresh();
    }, POLL_INTERVAL);

    return () => clearInterval(timer);
  }, [refresh]);

  const statusHint = useMemo(() => {
    if (pollError) return 'Backend unreachable';
    if (overlayStatus === 'connecting') return 'Connecting to overlay…';
    if (overlayStatus === 'error') return 'Overlay stream disconnected';
    return 'Live';
  }, [pollError, overlayStatus]);

  const superChats = useMemo(() => messages.filter(m => m.superChat), [messages]);
  const newMembers = useMemo(() => messages.filter(m => m.membershipGift), [messages]);
  const regularMessages = useMemo(() => messages.filter(m => !m.superChat && !m.membershipGift), [messages]);

  return (
    <main className="dashboard">
      <header className="dashboard__header">
        <div className="dashboard__title">
          <h1>🎬 Live Chat Monitor</h1>
          <p className="muted">Select a message to display on your OBS overlay</p>
        </div>
        <div className="dashboard__connect">
          <ConnectionControl
            inline
            connected={connected}
            liveId={liveId}
            connecting={connecting}
            onConnect={connect}
            onDisconnect={disconnect}
          />
        </div>
        <div className="dashboard__status">
          <span className={`status status--${overlayStatus}`}>{statusHint}</span>
          <span className="message-count">{messages.length} messages</span>
        </div>
      </header>

      <section className="dashboard__grid">
        <div className="panel panel--chat">
          <div className="panel__header">
            <h2>💬 Live Chat</h2>
            {selection && (
              <button className="btn-clear" onClick={handleClear}>
                Clear Selection
              </button>
            )}
          </div>
          <div className="chatList">
            {regularMessages.map((message) => (
              <ChatItem
                key={message.id}
                message={message}
                isSelected={selection?.id === message.id}
                onSelect={() => handleSelect(message)}
              />
            ))}
            {regularMessages.length === 0 && (
              <div className="chatList__empty">
                <p>⏳ Waiting for chat messages...</p>
              </div>
            )}
          </div>
        </div>

        <div className="panel panel--super">
          <div className="panel__header">
            <h2>💰 Super Chats</h2>
            <span className="badge badge--count">{superChats.length}</span>
          </div>
          <div className="chatList">
            {superChats.map((message) => (
              <ChatItem
                key={message.id}
                message={message}
                isSelected={selection?.id === message.id}
                onSelect={() => handleSelect(message)}
              />
            ))}
            {superChats.length === 0 && (
              <div className="chatList__empty">
                <p>No super chats yet</p>
              </div>
            )}
          </div>
        </div>

        <div className="panel panel--members">
          <div className="panel__header">
            <h2>⭐ New Members</h2>
            <span className="badge badge--count">{newMembers.length}</span>
          </div>
          <div className="chatList">
            {newMembers.map((message) => (
              <MemberItem
                key={message.id}
                message={message}
                isSelected={selection?.id === message.id}
                onSelect={() => handleSelect(message)}
              />
            ))}
            {newMembers.length === 0 && (
              <div className="chatList__empty">
                <p>No new members yet</p>
              </div>
            )}
          </div>
        </div>
      </section>
    </main>
  );
}

function useChatMessages() {
  const [messages, setMessages] = useState<ChatMessage[]>([]);
  const [error, setError] = useState<Error | null>(null);

  const refresh = useCallback(async () => {
    try {
      const response = await fetch(`${BACKEND_URL}/chat/messages`);
      if (!response.ok) throw new Error(`Request failed: ${response.status}`);
      const data = await response.json();
      setMessages(Array.isArray(data.messages) ? data.messages : []);
      setError(null);
    } catch (err) {
      setError(err as Error);
    }
  }, []);

  useEffect(() => {
    refresh();
  }, [refresh]);

  return { messages, refresh, error };
}

type OverlayStatus = 'connecting' | 'live' | 'error';

type SelectionPayload = {
  message: ChatMessage | null;
};

function useOverlaySelection() {
  const [selection, setSelection] = useState<ChatMessage | null>(null);
  const [status, setStatus] = useState<OverlayStatus>('connecting');
  const listenerRef = useRef<((payload: any) => void) | null>(null);

  useEffect(() => {
    // Create global connection if it doesn't exist
    if (!globalSSEConnection) {
      console.log('[Dashboard] Creating GLOBAL SSE connection');
      globalSSEConnection = new EventSource(`${BACKEND_URL}/overlay/stream`);
      
      globalSSEConnection.addEventListener('selection', ((event: MessageEvent) => {
        try {
          const payload: SelectionPayload = JSON.parse(event.data);
          // Broadcast to all listeners
          globalConnectionListeners.forEach(listener => listener(payload));
        } catch (error) {
          console.error('Failed to parse selection payload', error);
        }
      }) as EventListener);

      globalSSEConnection.addEventListener('heartbeat', () => {
        // Heartbeat - connection is alive
      });

      globalSSEConnection.onerror = () => {
        console.error('[Dashboard] SSE error');
      };
    }

    // Register this component's listener
    const myListener = (payload: SelectionPayload) => {
      setSelection(payload.message);
      setStatus('live');
    };
    
    listenerRef.current = myListener;
    globalConnectionListeners.add(myListener);

    return () => {
      // Unregister this component's listener
      if (listenerRef.current) {
        globalConnectionListeners.delete(listenerRef.current);
      }
      // Don't close global connection - other components might use it
    };
  }, []);

  return { selection, status };
}

function useConnection() {
  const [connected, setConnected] = useState(false);
  const [liveId, setLiveId] = useState<string | null>(null);
  const [connecting, setConnecting] = useState(false);

  const checkStatus = useCallback(async () => {
    try {
      const response = await fetch(`${BACKEND_URL}/health`);
      const data = await response.json();
      setConnected(data.connected);
      setLiveId(data.liveId);
    } catch (error) {
      console.error('Failed to check connection status', error);
    }
  }, []);

  useEffect(() => {
    checkStatus();
    const interval = setInterval(checkStatus, 5000);
    return () => clearInterval(interval);
  }, [checkStatus]);

  const connect = useCallback(async (liveId: string) => {
    setConnecting(true);
    try {
      const response = await fetch(`${BACKEND_URL}/chat/connect`, {
        method: 'POST',
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ liveId })
      });
      
      if (!response.ok) {
        const error = await response.json();
        throw new Error(error.error || 'Failed to connect');
      }
      
      await checkStatus();
    } catch (error) {
      console.error('Failed to connect', error);
      alert('Failed to connect to YouTube Live chat. Please check the Live ID.');
    } finally {
      setConnecting(false);
    }
  }, [checkStatus]);

  const disconnect = useCallback(async () => {
    try {
      await fetch(`${BACKEND_URL}/chat/disconnect`, { method: 'POST' });
      await checkStatus();
    } catch (error) {
      console.error('Failed to disconnect', error);
    }
  }, [checkStatus]);

  return { connected, liveId, connect, disconnect, connecting };
}

type ConnectionControlProps = {
  connected: boolean;
  liveId: string | null;
  connecting: boolean;
  onConnect: (liveId: string) => void;
  onDisconnect: () => void;
  inline?: boolean;
};

function ConnectionControl({ connected, liveId, connecting, onConnect, onDisconnect, inline }: ConnectionControlProps) {
  const [inputValue, setInputValue] = useState('');

  const handleSubmit = (e: React.FormEvent) => {
    e.preventDefault();
    if (inputValue.trim()) {
      onConnect(inputValue.trim());
    }
  };

  return (
    <div className={"connectionControl" + (inline ? " connectionControl--inline" : "") }>
      <div className="connectionControl__content">
        {connected ? (
          <div className="connectionControl__connected">
            <div className="connectionControl__info">
              <span className="connectionControl__badge">🟢 Connected</span>
              <span className="connectionControl__liveId">Live ID: {liveId}</span>
            </div>
            <button className="btn-disconnect" onClick={onDisconnect}>
              Disconnect
            </button>
          </div>
        ) : (
          <form className="connectionControl__form" onSubmit={handleSubmit}>
            <div className="connectionControl__input">
              <label htmlFor="liveId">YouTube Live Stream ID or URL</label>
              <input
                id="liveId"
                type="text"
                value={inputValue}
                onChange={(e) => setInputValue(e.target.value)}
                placeholder="e.g., dQw4w9WgXcQ or https://youtube.com/watch?v=..."
                disabled={connecting}
              />
            </div>
            <button
              type="submit"
              className="btn-connect"
              disabled={connecting || !inputValue.trim()}
            >
              {connecting ? 'Connecting...' : 'Connect to Stream'}
            </button>
          </form>
        )}
      </div>
    </div>
  );
}

type ChatItemProps = {
  message: ChatMessage;
  isSelected: boolean;
  onSelect: () => void;
};

function ChatItem({ message, isSelected, onSelect }: ChatItemProps) {
  return (
    <button
      className={isSelected ? 'chatItem chatItem--active' : 'chatItem'}
      onClick={onSelect}
    >
      <div className="chatItem__header">
        {message.authorPhoto && (
          <img src={message.authorPhoto} alt={message.author} className="chatItem__avatar" />
        )}
        <div className="chatItem__meta">
          <div className="chatItem__authorLine">
            <span className="chatItem__author">{message.author}</span>
            {message.badges && message.badges.map((badge, i) => (
              <span key={i} className={`badge badge--${badge.type}`} title={badge.label}>
                {badge.type === 'moderator' && '🛡️'}
                {badge.type === 'member' && '⭐'}
                {badge.type === 'verified' && '✓'}
              </span>
            ))}
          </div>
          <time className="chatItem__time">{new Date(message.publishedAt).toLocaleTimeString()}</time>
        </div>
      </div>
      {message.superChat && (
        <div className="chatItem__superchat" style={{ backgroundColor: message.superChat.color }}>
          💰 {message.superChat.amount}
        </div>
      )}
      <p className="chatItem__text">{message.text}</p>
    </button>
  );
}

function MemberItem({ message, isSelected, onSelect }: ChatItemProps) {
  return (
    <button
      className={isSelected ? 'memberItem memberItem--active' : 'memberItem'}
      onClick={onSelect}
    >
      <div className="memberItem__header">
        {message.authorPhoto && (
          <img src={message.authorPhoto} alt={message.author} className="memberItem__avatar" />
        )}
        <div className="memberItem__info">
          <span className="memberItem__author">{message.author}</span>
          <span className="memberItem__level">
            {message.membershipLevel || 'New member'}
          </span>
          <time className="memberItem__time">{new Date(message.publishedAt).toLocaleTimeString()}</time>
        </div>
      </div>
      {message.text && <p className="memberItem__text">{message.text}</p>}
    </button>
  );
}