sdis1819-t7g07

package peer;

import channels.McChannel;

import channels.MdbChannel;

import protocols.BackupProtocol;

import protocols.RestoreProtocol;

import utils.Utils;

import java.io.File;

import java.io.IOException;

import java.net.*;

import java.security.NoSuchAlgorithmException;

import utils.*;

public class Peer {

    private static int senderId; //id of the server that has sent the message

    private static InetAddress mcChannel;

    private static InetAddress mdbChannel;

    private static InetAddress mdrChannel;

    private static int mcPort;

    private static int mdbPort;

    private static int mdrPort;

    //BACKUP, DELETE, RECLAIM, RESTORE;

    public static void main(String[] args) throws IOException, NoSuchAlgorithmException, InterruptedException {

        if(!parseArgs(args)) return;

        joinMulticastGroup();

        startThreads();

        getRequests();

    }

    private static void getRequests() throws IOException, NoSuchAlgorithmException, InterruptedException {

        DatagramSocket request = new DatagramSocket();

        int portNumber = request.getLocalPort();

        System.out.println("Socket Peer opened in port number: " + portNumber);

        while(true){

            byte[] buffer = new byte[Utils.BUFFER_SIZE];

            DatagramPacket packet = new DatagramPacket(buffer, buffer.length);

            request.receive(packet);

            InetAddress clientAddress = packet.getAddress();

            String requestString = new String(packet.getData(), 0, packet.getLength());

            System.out.println();

            System.out.println("---------------------------------------------------------------------------");

            System.out.println("Request from TestApp: " + requestString);

            System.out.println();

            String[] requestArgs = requestString.split("\\s+");

            switch(requestArgs[0]){

                    case "BACKUP":

                        new BackupProtocol(requestArgs);

                        break;

                    case "RESTORE":

                        new RestoreProtocol(requestArgs);

                        break;

                    default:

                        break;

            }

            int port = packet.getPort();

            String answer = "Got your request";

            buffer = answer.getBytes();

            packet = new DatagramPacket(buffer, 0, buffer.length, clientAddress, port);

            request.send(packet);

        }

    }

    public static void startThreads(){ //only subscribed mc channel

        Utils.mcChannel = new McChannel();

        new Thread(Utils.mcChannel).start();

        Utils.mdbChannel = new MdbChannel();

        new Thread(Utils.mdbChannel).start();

        Utils.mdrChannel = new channels.MdrChannel();

        new Thread(Utils.mdrChannel).start();

    }

    public static void joinMulticastGroup() throws IOException{

        //mc channel

        Utils.mcSocket = new MulticastSocket(getMcPort());

        Utils.mcSocket.joinGroup(getMcAddress());

        //mdb channel

        Utils.mdbSocket = new MulticastSocket(getMdbPort());

        Utils.mdbSocket.joinGroup(getMdbAddress());

        //mdr channel

        Utils.mdrSocket = new MulticastSocket(getMdrPort());

        Utils.mdrSocket.joinGroup(getMdrAddress());

    }

    public static boolean parseArgs(String[] args) throws UnknownHostException{

        if(args == null || args.length != 4){

            System.out.println("Usage: java peer.peer <SenderId> <McAddress>:<McPortNumber> <MdbAddress>:<MdbPortNumber> <MdrAddress>:<MdrPortNumber>");

            return false;

        }

        //sets sender id

        setSenderId(Integer.parseInt(args[0]));

        //sets address and port for channel MC

        String[] mcPart = args[1].split(":");

        setMcAddress(InetAddress.getByName(mcPart[0]));

        setMcPort(Integer.parseInt(mcPart[1]));

        //sets address and port for channel MDB

        String[] mdbPart = args[2].split(":");

        setMdbAddress(InetAddress.getByName(mdbPart[0]));

        setMdbPort(Integer.parseInt(mdbPart[1]));

        //sets address and port for channel MDR

        String[] mdrPart = args[3].split(":");

        setMdrAddress(InetAddress.getByName(mdrPart[0]));

        setMdrPort(Integer.parseInt(mdrPart[1]));

        return true;

    }

    //getters and setters

    public static InetAddress getMcAddress(){

        return mcChannel;

    }

    public static InetAddress getMdbAddress(){

        return mdbChannel;

    }

    public static InetAddress getMdrAddress(){

        return mdrChannel;

    }

    public static int getMcPort(){

        return mcPort;

    }

    public static int getMdbPort(){

        return mdbPort;

    }

    public static int getMdrPort(){

        return mdrPort;

    }

    public static void setMcAddress(InetAddress mcAddress){

        mcChannel = mcAddress;

    }

    public static void setMdbAddress(InetAddress mdbAddress){

        mdbChannel = mdbAddress;

    }

    public static void setMdrAddress(InetAddress mdrAddress){

        mdrChannel = mdrAddress;

    }

    public static void setMcPort(int port){

        mcPort = port;

    }

    public static void setMdbPort(int port){

        mdbPort = port;

    }

    public static void setMdrPort(int port){

        mdrPort = port;

    }

    public static void setSenderId(int id){

        senderId = id;

    }

    public static int getSenderId(){

        return senderId;

    }

}