node-server/websocket.js

const TranscriptClient = require("./transcription-client");

// TODO remove x seconds after host left (incase reconnect)
const initializeWebSocket = (io) => {
  io.on("connection", (socket) => {
    console.log(`connection made (${socket.id})`);
    const transcriptClient = new TranscriptClient();

    transcriptClient.on("translation", (result) => {
      console.log(result)
      io.to(socket.id).emit("translation", result)
    })

    socket.on('configure_stream', ({language}) => {
      transcriptClient.startTranscriptionStream("en-US")
    })

    socket.on('incoming_audio', (data) => {
      transcriptClient.send(data)
    })

    socket.on("disconnect", () => {
      transcriptClient.endTranscriptionStream()
    });
  });

  return io;
};

module.exports = initializeWebSocket;