SDIS1819-T4G10

import java.io.File;

import java.io.IOException;

import java.net.DatagramPacket;

import java.net.InetAddress;

import java.net.MulticastSocket;

import java.net.UnknownHostException;

import java.rmi.registry.LocateRegistry;

import java.rmi.registry.Registry;

public class TestApp {

        private static String peer_ap;

        private static String sub_protocol;

        private static String file_path;

        private static double disk_space = 0;

        private static int ReplicationDeg = 0;

    public static void main(String[] args) throws UnknownHostException { //<peer_ap> <sub_protocol> <opnd_1> <opnd_2>

            peer_ap = args[0];

        sub_protocol = args[1];

        try {

            // Getting the registry

            Registry registry = LocateRegistry.getRegistry(null);

            // Looking up the registry for the remote object

            RMI stub = (RMI) registry.lookup(peer_ap);

                System.out.println("PROTOCOL:  "+sub_protocol);

            if(sub_protocol.equals("BACKUP"))

            {

                    if(args.length != 4) {

                            System.out.println("Invalid arguments for Backup protocol");

                            System.out.println("java TestApp <peer_ap> BACKUP <file_path> <replication_degree>");

                            return;

                    }

                    ReplicationDeg = Integer.parseInt(args[3]);

                file_path = args[2];

                File f = new File(file_path);

                if(!f.exists()) {

                    System.out.println("ERROR: File path specified doesn't lead to a file.");

                    return;

                }

                stub.operation(sub_protocol, file_path, ReplicationDeg, disk_space);

            }

            else if(sub_protocol.equals("RESTORE"))

            {

                    if(args.length != 3) {

                            System.out.println("Invalid arguments for Restore protocol");

                            System.out.println("java TestApp <peer_ap> RESTORE <file_path>");

                            return;

                    }

                file_path = args[2];

                stub.operation(sub_protocol, file_path, 0, 0);

            }

            else if(sub_protocol.equals("DELETE"))

            {

                    if(args.length != 3) {

                            System.out.println("Invalid arguments for Delete protocol");

                            System.out.println("java TestApp <peer_ap> DELETE <file_path> ");

                            return;

                    }

                file_path = args[2];

                stub.operation(sub_protocol, file_path, ReplicationDeg, disk_space);

            }

            else if(sub_protocol.equals("RECLAIM"))

            {

                    if(args.length != 3) {

                            System.out.println("Invalid arguments for Reclaim protocol");

                            System.out.println("java TestApp <peer_ap> RECLAIM <space>");

                            return;

                    }

                disk_space = Double.parseDouble(args[2]);

                stub.operation(sub_protocol, file_path, ReplicationDeg, disk_space);

            }

            else if(sub_protocol.equals("STATE"))

            {

                    if(args.length != 2) {

                            System.out.println("Invalid arguments for State protocol");

                            System.out.println("java TestApp <peer_ap> STATE");

                            return;

                    }

                stub.operation(sub_protocol, file_path, ReplicationDeg, disk_space);

            }

           } catch (Exception e) {

                   System.err.println("TestApp Exception: " + e.toString());

                      e.printStackTrace();

           }

        System.out.println("PROTOCOL EXECUTED");

    }

}