const PORT = process.env.PORT || 3000;
const io = require('socket.io')(PORT, {
  cors: {
    origin: "*",
    methods: ["GET", "POST"]
  }
});

const peerSocketMap = new Map();
const socketPeerMap = new Map();

io.on('connection', socket => {
  console.log(`New client connected, socket ID: ${socket.id}`);

  const emitToTarget = (eventType, data, recipient) => {
    const targetSocketId = peerSocketMap.get(recipient);
    const sender = socketPeerMap.get(socket.id);
    if (sender && targetSocketId) {
      io.to(targetSocketId).emit(eventType, {...data, sender});
    }
  };

  socket.on('offer', data => emitToTarget('offer', data, data.recipient));
  socket.on('answer', data => emitToTarget('answer', data, data.recipient));
  socket.on('candidate', data => emitToTarget('candidate', data, data.recipient));

  socket.on('hello', peerId => {
    console.log(`Received hello from ${peerId}`);
    peerSocketMap.set(peerId, socket.id);
    socketPeerMap.set(socket.id, peerId);
    socket.broadcast.emit('hello', peerId);
  });

  socket.on('disconnect', () => {
    peerSocketMap.delete(socketPeerMap.get(socket.id));
    socketPeerMap.delete(socket.id);
    console.log(`Client disconnected, socket ID: ${socket.id}`);
  });
});

console.log(`Signaling server running on port ${PORT}`);

const Ministun = require("ministun");

const stunConfig = {
	udp4: true,
	udp6: true,
	port: 3478,
	log: console.log,
	err: console.err,
	sw: true
};

const server = new Ministun(stunConfig);

console.log(`STUN server running on port ${stunConfig.port}`)