sdis1819-t7g07

package channels;

import peer.Peer;

import utils.Chunk;

import utils.Utils;

import static utils.Utils.BUFFER_SIZE;

import static utils.Utils.chunkReceived;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.net.DatagramPacket;

import java.util.Arrays;

import java.util.Random;

public class McChannel implements Runnable{

    public void run() {

        while (true) {

            byte[] buf = new byte[BUFFER_SIZE];

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

            try {

                Utils.mcSocket.receive(receiveDatagram);

                String receivedMessage = new String(buf, 0, buf.length).trim();

                String[] receivedArgs = receivedMessage.split("\\s+");

                switch (receivedArgs[0]) {

                    case "GETCHUNK":

                        if (!receivedArgs[2].equals((Integer.toString(Peer.getSenderId())))) { //if not init peer send chunk message to other peers

                            Utils.chunkReceived = false;

                            sendChunk(receivedArgs);

                        }

                        break;

                    case "STORED":

                        //System.out.println("STORED <Version> <SenderId> <FileId> <ChunkNo> <CRLF><CRLF>");

                        break;

                }

            } catch (IOException e) {

                System.out.println(e.getMessage());

            } catch (InterruptedException e) {

                e.printStackTrace();

            }

        }

    }

    private void sendChunk(String[] receivedArgs) throws IOException, InterruptedException {

        String chunkyNo = receivedArgs[4];

        String cleanChunkNo = chunkyNo.substring(0, chunkyNo.length() - 4);

        File toCheck = new File("storage/Backup/Peer" + Peer.getSenderId() + "/" + receivedArgs[3] + "/" + cleanChunkNo);

        if(toCheck.exists() && !toCheck.isDirectory()){ //go get the chunk in memory

            FileInputStream stream = new FileInputStream(toCheck);

            byte[] buffer = new byte[64000];

            int sizeRead = stream.read(buffer,0,64000);

            stream.close();

            byte[] realBuffer = Arrays.copyOf(buffer,sizeRead);

            Chunk toSend = new Chunk(realBuffer,"CHUNK",receivedArgs[3], cleanChunkNo);

            sendTheChunkMdr(toSend);

        }

    }

    private void sendTheChunkMdr(Chunk toSend) throws InterruptedException, IOException {

        int delay = new Random().nextInt(400); // waits before sending chunk message

        Thread.sleep(delay);

        if(!Utils.chunkReceived){

            System.out.println("McChannel: " + toSend.getHeaderMsg());

            MdrChannel.sendMessage(toSend.getBuffer());

        }

    }

}