package alice.datasegment;

import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.NetworkInterface;
import java.net.SocketAddress;
import java.net.StandardProtocolFamily;
import java.net.StandardSocketOptions;
import java.nio.channels.DatagramChannel;

import org.apache.log4j.Logger;

import alice.daemon.IncomingUdpConnection;
import alice.daemon.MulticastConnection;
import alice.daemon.OutboundTcpConnection;

public class MulticastDataSegmentManager extends RemoteDataSegmentManager {

    public MulticastDataSegmentManager(String connectionKey ,final String MCASTADDR, final int port, final String nis) {
        logger = Logger.getLogger(connectionKey);
        InetAddress mAddr;
        try {
            mAddr = InetAddress.getByName(MCASTADDR);

            DatagramChannel dcr =  createDatagramChannel(mAddr, port, nis);
            dcr.bind(new InetSocketAddress(port));
            SocketAddress sAddrr = new InetSocketAddress(mAddr,port);
            MulticastConnection receiver = new MulticastConnection(dcr, sAddrr);

            DatagramChannel dcs =  createDatagramChannel(mAddr, port, nis);
            SocketAddress sAddrs = new InetSocketAddress(mAddr,port);
            connection = new MulticastConnection(dcs, sAddrs); // sender

            IncomingUdpConnection in = new IncomingUdpConnection((MulticastConnection) connection, receiver, this);
            in.setName("multicast-IncomingUdp");
            in.start();
            OutboundTcpConnection out = new OutboundTcpConnection(connection); // OutboundUdpConnection sender
            out.setName(connectionKey+"OutboundUdp");
            out.start();

        } catch (Exception e) {
            e.printStackTrace();
        }

    }

    private DatagramChannel createDatagramChannel(InetAddress group, int port, String nis) {
        DatagramChannel dc = null;
        NetworkInterface ni;
        try {
            ni = NetworkInterface.getByName(nis);
            if (ni==null) {
                System.err.println("Can't open network interface "+nis);
                throw new IOException();
            }
            if (!ni.supportsMulticast()) {
                System.err.println("Network interface does not support multicast"+nis);
                throw new IOException();
            }

            dc = DatagramChannel.open(StandardProtocolFamily.INET);
            dc.setOption(StandardSocketOptions.SO_REUSEADDR, true);
            dc.setOption(StandardSocketOptions.IP_MULTICAST_IF, ni);
            dc.join(group, ni);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return dc;
    }
}