sdis1819-t7g07

package channels;

import peer.Peer;

import utils.FileController;

import utils.Utils;

import java.io.*;

import java.net.DatagramPacket;

public class MdrChannel implements Runnable{

    public void run(){

        while(true){

            byte[] buf = new byte[64000 + 1000];

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

            try{

                Utils.mdrSocket.receive(receivePacket);

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

                String[] receivedArray = received.split("\\r\\n\\r\\n");

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

                byte[] bodyByteArray = getArrayFromOffset(receivePacket.getData(), offsetOfBody, receivePacket.getLength());

                FileController receivedFile = new FileController(receivedArray, bodyByteArray);

                if(Peer.getSenderId() == Utils.restorePeer){ //save to init peer

                    saveFile(receivedFile);

                }

            } catch (IOException e) {

                e.printStackTrace();

            }

        }

    }

    private byte[] getArrayFromOffset(byte[] data, int offsetOfBody, int length) {

        int size = length - offsetOfBody;

        byte[] toRet = new byte[size];

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

            toRet[i - offsetOfBody] = data[i];

        }

        return toRet;

    }

    private void saveFile(FileController receivedFile) throws IOException {

        File theDir = new File("storage/Restored");

        // if the directory does not exist, create it

        if (!theDir.exists()) {

            boolean result = false;

            try{

                theDir.mkdir();

                result = true;

            }

            catch(SecurityException se){

            }

            if(result) {

            }

        }

        File dir = new File("storage/Restored/Peer" + Peer.getSenderId());

        // if the directory does not exist, create it

        if (!dir.exists()) {

            boolean result = false;

            try{

                dir.mkdir();

                result = true;

            }

            catch(SecurityException se){

            }

            if(result) {

            }

        }

        File f = new File("storage/Restored/Peer" + Peer.getSenderId() + "/" + receivedFile.getChunkNo());

        if(!f.exists()) { //if file is not saved yet by other peer...

            f.createNewFile();

        }

        OutputStream out = new FileOutputStream(f);

        out.write(receivedFile.getBody(), 0, receivedFile.getBody().length);

        out.close();

        Utils.chunkReceived = true;

    }

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

        DatagramPacket packet = new DatagramPacket(buf, buf.length, Peer.getMdrAddress(), Peer.getMdrPort());

        Utils.mdrSocket.send(packet);

    }

}