reflector/www/app/transcripts/useWebRTC.ts

import { useEffect, useState } from "react";
import Peer from "simple-peer";
import {
  DefaultApi,
  V1TranscriptRecordWebrtcRequest,
} from "../api/apis/DefaultApi";
import { useError } from "../(errors)/errorContext";

const useWebRTC = (
  stream: MediaStream | null,
  transcriptId: string | null,
  api: DefaultApi,
): Peer => {
  const [peer, setPeer] = useState<Peer | null>(null);
  const { setError } = useError();

  useEffect(() => {
    if (!stream || !transcriptId) {
      return;
    }

    let p: Peer;

    try {
      p = new Peer({ initiator: true, stream: stream });
    } catch (error) {
      setError(`Failed to create WebRTC Peer: ${error.message}`);
      return;
    }

    p.on("error", (err) => {
      setError(`WebRTC error: ${err.message}`);
    });

    p.on("signal", (data: any) => {
      if ("sdp" in data) {
        const requestParameters: V1TranscriptRecordWebrtcRequest = {
          transcriptId: transcriptId,
          rtcOffer: {
            sdp: data.sdp,
            type: data.type,
          },
        };

        api
          .v1TranscriptRecordWebrtc(requestParameters)
          .then((answer) => {
            try {
              p.signal(answer);
            } catch (error) {
              setError(`Failed to signal answer: ${error.message}`);
            }
          })
          .catch((err) => {
            const errorString =
              "WebRTC signaling error: " +
              (err.response || err.message || "Unknown error");
            setError(errorString);
            console.error(errorString);
          });
      }
    });

    p.on("connect", () => {
      console.log("WebRTC connected");
      setPeer(p);
    });

    return () => {
      p.destroy();
    };
  }, [stream, transcriptId]);

  return peer;
};

export default useWebRTC;