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", this::handleBackup);

        protocols.put("DELETE", this::handleDelete);

    }

    @Override

    public void run() {

            try {

            Registry registry = LocateRegistry.getRegistry(null);

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

        } catch (Exception e) {

                System.out.println("Error when opening RMI stub");

            e.printStackTrace();

        }

        protocols.get(sub_protocol).run();

    }

    private void handleBackup() {

        File file = new File(this.opnd_1);

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

        try {

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

        } catch (RemoteException e) {

                e.printStackTrace();

        }

    }

    private void handleDelete() {

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

        try {

            initiatorPeer.delete(this.opnd_1);

        } catch (RemoteException e) {

                e.printStackTrace();

        }

    }

}