sdis1819-t7g07

import java.io.IOException;

import java.net.*;

import java.util.Scanner;

import static utils.Utils.BUFFER_SIZE;

//TODO: parse args for reclaim and delete not fully functional

//grupo 11

public class TestApp {

    private static String hostname = "localhost";

    private static int portNumber;

    private static String protocolRequest;

    private static String filePath;

    private static int replicationDegree;

    public static void main(String[] args) throws SocketException, UnknownHostException {

        mainMenu();

        }

    private static void mainMenu() throws SocketException, UnknownHostException {

        System.out.println("1- Backup a file");

        System.out.println("2- Restore a file");

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

        Scanner in = new Scanner(System.in);

        int option = in.nextInt();

        switch(option){

            case 0:

                System.out.println("exiting application...");

                break;

            case 1:

                backupMenu();

                break;

            case 2:

                restoreMenu();

                break;

        }

    }

    private static void restoreMenu() throws SocketException, UnknownHostException {

        String[] args = new String[4];

        System.out.println("Please insert the peer port number which you want to send the request");

        Scanner in = new Scanner(System.in);

        String portnumber = in.nextLine();

        args[0] = portnumber;

        args[1] = "RESTORE";

        System.out.println("Specify the file name you want to restore ");

        String filename = in.nextLine();

        args[2] = filename;

        if(!parseArgs(args)) return;

        sendRequest(args);

    }

    private static void backupMenu() throws SocketException, UnknownHostException {

        String[] args = new String[4];

        System.out.println("Please insert the peer port number which you want to send the request");

        Scanner in = new Scanner(System.in);

        String portnumber = in.nextLine();

        args[0] = portnumber;

        args[1] = "BACKUP";

        System.out.println("Specify the file name you want to backup (must be inside test folder) ");

        String filename = in.nextLine();

        args[2] = filename;

        System.out.println("Now enter the replication degree");

        String replicD = in.nextLine();

        args[3] = replicD;

        if(!parseArgs(args)) return;

        sendRequest(args);

    }

    private static void sendRequest(String[] args) throws SocketException, UnknownHostException {

        DatagramSocket socket = new DatagramSocket();

        byte[] buffer;

        String message = "";

        for(int i = 1; i < args.length; i++){

            if(i == args.length -1){

                message += args[i];

                break;

            }

            message += args[i] + " ";

        }

        buffer = message.getBytes();

        InetAddress address = InetAddress.getByName(hostname);

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

        try {

            socket.send(packet);

        } catch (IOException e) {

            e.printStackTrace();

        }

        buffer = new byte[BUFFER_SIZE];

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

        //

        try {

            socket.receive(packet);

        } catch (IOException e) {

            e.printStackTrace();

        }

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

        System.out.println("Peer said: " + answer);

        socket.close();

        //after request sent and done, come back to main menu

        mainMenu();

    }

    private static boolean parseArgs(String[] args) {

        if(args == null || args.length < 3) {

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

            return false;

        }

        String[] peer_ap = args[0].split(":");

        if(peer_ap.length == 1) { //user just gave the port number

            portNumber = Integer.parseInt(peer_ap[0]);

        }

        else if(peer_ap.length == 2) {

            hostname = peer_ap[0];

            portNumber = Integer.parseInt(peer_ap[1]);

        }

        else {

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

            return false;

        }

        switch(args[1]) {

            case "BACKUP":

                protocolRequest = "BACKUP";

                filePath = args[2];

                if(args[3] == null){

                    System.out.println("In backup protocol, you have to specify also the replication degree");

                    return false;

                }

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

                break;

            case "RESTORE":

                protocolRequest = "RESTORE";

                filePath = args[2];

                break;

            case "DELETE":

                protocolRequest = "DELETE";

                filePath = args[2];

                break;

            case "RECLAIM":

                protocolRequest = "RECLAIM";

                break;

                default: return false;

        }

        return true;

    }

}