sdis1819-t7g07

package channels;

import peer.Peer;

import utils.FileController;

import utils.Utils;

import java.io.File;

import java.io.IOException;

import java.net.DatagramPacket;

import java.util.ArrayList;

import java.util.Random;

public class MdbChannel implements Runnable{

    public void run() {

        while(true){

            byte[] buf = new byte[65000]; //header + body

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

            try {

                //received chunck

                Utils.mdbSocket.receive(receivedPacket);

                String received = new String(receivedPacket.getData(),0,receivedPacket.getLength());

                String[] receivedArray = received.split("\\r\\n\\r\\n"); //separate header from body

                int offsetOfBody = receivedArray[0].length() + 4;

                String receivedChunkData = received.substring(offsetOfBody, received.length());

                int size = received.length() - offsetOfBody;

                byte[] toRet = new byte[size];

                for(int i = offsetOfBody; i < received.length(); i++) {

                    toRet[i - offsetOfBody] = receivedPacket.getData()[i];

                }

                FileController receivedFile = new FileController(receivedArray, toRet);

                //addToChunkDatabase

                if(!Utils.storedChunks.containsKey(receivedFile.getHashedFileId() + ":" + receivedFile.getChunkNo())){

                    Utils.storedChunks.put(receivedFile.getHashedFileId() + ":" + receivedFile.getChunkNo(),new ArrayList<Integer>());

                    Utils.storedChunks.get(receivedFile.getHashedFileId() + ":" +  receivedFile.getChunkNo()).add(Integer.parseInt(receivedFile.getReplicationDeg()));

                }

                //saveFileIfSentFromAnotherPeer

                if(! (receivedFile.getSenderId() == Peer.getSenderId())) {

                    //System.out.println("MDB: Received CMD - " + receivedFile.getStringInformation());

                    if(!checkIfFileExists(receivedFile.getHashedFileId() + receivedFile.getChunkNo()))

                        receivedFile.saveToDisk(Peer.getSenderId(),receivedFile.getChunkNo());

                    try {

                        Thread.sleep(new Random().nextInt(400));

                    } catch (InterruptedException e) {

                        e.printStackTrace();

                    }

                    sendStoredMessage(receivedFile);

                }

            }

            catch(IOException e){

                e.getCause();

            }

        }

    }

    private void sendStoredMessage(FileController receivedFile) throws IOException {

        String message = ("STORED " + receivedFile.getVersion() + " " +

                Peer.getSenderId() + " " + receivedFile.getHashedFileId()

                + " " + receivedFile.getChunkNo() + Utils.crlf);

        byte[] buf = new byte[256];

        buf = message.getBytes();

        DatagramPacket packet = new DatagramPacket(buf,buf.length, Peer.getMcAddress(),Peer.getMcPort());

        Utils.mcSocket.send(packet);

    }

    public static void sendMessage(byte[] buf) throws IOException{

        DatagramPacket packet = new DatagramPacket(buf,buf.length, Peer.getMdbAddress(),Peer.getMdbPort());

        Utils.mdbSocket.send(packet);

    }

    private boolean checkIfFileExists(String string) {

        File f = new File("src/test/" + Peer.getSenderId() + "/" + string);

        if(f.exists() && !f.isDirectory())

            return true;

        return false;

    }

}