sdis1819-t7g05

import java.io.IOException;

import java.io.File;

import java.net.DatagramPacket;

import java.net.InetAddress;

import java.net.MulticastSocket;

import java.util.Arrays;

/**

 * MCListener

 */

public class MCListener implements Runnable {

    private static final int PACKET_SIZE = 256;

    private MulticastSocket socket;

    private InetAddress address;

    private int port;

    private int serverId;

    private Peer peer;

    MCListener(InetAddress address, int port, int serverId, Peer peer) throws IOException {

        this.address = address;

        this.port = port;

        this.serverId = serverId;

        this.peer = peer;

        socket = new MulticastSocket(port);

        socket.joinGroup(this.address);

    }

    @Override

    public void run() {

        print("MC-Listener started: Address-"+this.address.getHostName()+" Port-"+this.port);

        while (true) {

            byte[] buf = new byte[PACKET_SIZE];

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

            try {

                socket.receive(packet);

                parseMessage(Arrays.copyOfRange(buf, 0, packet.getLength()));

            } catch (IOException e) {

                e.printStackTrace();

            }

        }

    }

    private void parseMessage(byte[] data) {

        String message = new String(data);

        String[] args = message.split(" ");

        switch (args[0]) {

            //STORED <Version> <SenderId> <FileId> <ChunkNo> <CRLF><CRLF>

            case "STORED":

                String fileId = args[3];

                String version = args[1];

                int chunkNo = Integer.parseInt(args[4]);

                new Thread(new StoredTask(this.peer, fileId, version, chunkNo, this)).start();

                print("starting "+args[0]+" protocol");

                break;

            //GETCHUNK <Version> <SenderId> <FileId> <ChunkNo> <CRLF><CRLF>

            case "GETCHUNK":

                print("received GETCHUNK message");

                if (Integer.parseInt(args[2]) == this.peer.getSenderId() ||

                    !this.peer.chunksMap.containsKey(args[3]) || 

                    !this.peer.chunksMap.get(args[3]).getChunksMap().containsKey(Integer.parseInt(args[4]))) {

                    return;

                }

                new Thread(new GetchunkTask(this.peer, args, this)).start();

                break;

            //DELETE <Version> <SenderId> <FileId> <CRLF><CRLF>

            case "DELETE":

                print("received DELETE message");

                if (!this.peer.chunksMap.containsKey(args[3])) {

                    return;

                }

                this.peer.chunksMap.remove(args[3]); //remove ficheiro da hashmap

                        this.peer.repDegMap.remove(args[3]);

                DeleteFile fileToDelete = new DeleteFile("Peer-" + this.peer.getSenderId() + File.separator + "Chunks" + File.separator + args[3], args[3]);

                        this.peer.deleteMap.put(args[3], fileToDelete);

                        for(File chunk : fileToDelete.getFile().listFiles()){

                            this.peer.removeStorage(chunk.length());

                            if(!chunk.delete())

                                    print("Error deleting chunk");

                }

                if(!fileToDelete.getFile().delete())

                                print("Error deleting chunks directory");

                break;

            //REMOVED <Version> <SenderId> <FileId> <ChunkNo> <CRLF><CRLF>

            case "REMOVED":

                print("received REMOVED message");

                print("SenderID: "+this.peer.getSenderId());

                print("args[2]: "+Integer.parseInt(args[2]));

                if (this.peer.getSenderId() == Integer.parseInt(args[2]) || !this.peer.chunksMap.containsKey(args[3]) || !this.peer.chunksMap.get(args[3]).getChunksMap().containsKey(Integer.parseInt(args[4])))

                    return;

                new Thread(new RemovedTask(this.peer, args, this)).start();

                break;

            default:

                break;

        }

    }

    public void print(String message) {

        System.out.println("[PEER-"+this.serverId+"-MC] "+message);

    }

}