sdis1819-t2g03

package client;

import java.io.File;

import java.rmi.RemoteException;

import java.rmi.registry.LocateRegistry;

import java.rmi.registry.Registry;

import java.util.HashMap;

import java.util.Map;

import server.InitiatorPeer;

public class TestApp implements Runnable {

        private InitiatorPeer initiatorPeer;

        private String peer_ap;

        private String sub_protocol;

        private String opnd_1;

        private String opnd_2;

        private Map<String, Runnable> protocols;

        public static void main(String[] args) {

                if (args.length < 2 || args.length > 4) {

                        System.out.println("Usage: <java TestApp <peer_ap> <sub_protocol> <opnd_1> <opnd_2> >");

                        System.out.println("<peer_ap> - peer access point");

                        System.out.println("<sub_protocol> - BACKUP, RESTORE, DELETE, RECLAIM");

                        System.out.println("<opnd_1> - path name of the file to backup/restore/delete");

                        System.out.println("<opnd_2> - replication degree");

                        System.exit(1);

                }

                String peer_ap = args[0];

                String sub_protocol = args[1];

                String opnd_1;

                String opnd_2;

                if(args.length > 2) 

                        opnd_1 = args[2];

                else

                        opnd_1 = null;

                if(args.length > 3) 

                        opnd_2 = args[3];

                else

                        opnd_2 = null;

                TestApp app = new TestApp(peer_ap, sub_protocol, opnd_1, opnd_2);

                new Thread(app).start();

        }

        public TestApp(String peer_ap, String sub_protocol, String opnd_1, String opnd_2) {

                protocols = new HashMap<>();

                this.peer_ap = peer_ap;

                this.sub_protocol = sub_protocol;

                this.opnd_1 = opnd_1;

                this.opnd_2 = opnd_2;

                protocols.put("BACKUP", () -> {

                        try {

                                backup();

                        } catch (NumberFormatException e) {

                                e.printStackTrace();

                        } catch (RemoteException e) {

                                e.printStackTrace();

                        }

                });

                protocols.put("DELETE", () -> {

                        try {

                                delete();

                        } catch (RemoteException e) {

                                e.printStackTrace();

                        }

                });

        }

        @Override

        public void run() {

                try {

                        Registry registry = LocateRegistry.getRegistry(null);

                        initiatorPeer = (InitiatorPeer) registry.lookup(peer_ap);

                } catch (Exception e) {

                        e.printStackTrace();

                }

                protocols.get(sub_protocol).run();

        }

        private void backup() throws NumberFormatException, RemoteException {

                File file = new File(this.opnd_1);

                System.out.println("--> BACKUP :: Saving chunks at " + file.getAbsolutePath() + "\"");

                initiatorPeer.backup(file, Integer.parseInt(this.opnd_2));

        }

        private void delete() throws RemoteException {

                System.out.println("--> DELETE :: delete file: " + opnd_1 + "\"");

                initiatorPeer.delete(this.opnd_1);

        }

}