sdis1819-t7g06

package Protocols;

import Message.*;

import Peers.Peer;

import Storage.Chunk;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.nio.file.Files;

import java.nio.file.Paths;

import java.util.ArrayList;

import java.util.concurrent.TimeUnit;

import static Global.Globals.*;

public class RestoreProtocol implements Runnable {

    private final Peer peer;

    private final String file;

    public RestoreProtocol(Peer peer, String filepath) {

        this.peer = peer;

        this.file = filepath;

    }

    ArrayList<Message> createMessages(float size, String fileId) {

        ArrayList<Message> messages = new ArrayList<Message>();

        int chunks = (int) (size / MAX_CHUNK_SIZE);

        for (int i = 0; i <= chunks; i++) {

            Message mes = new GetChunkMsg(peer.getVersionProtocol(), peer.getId(), fileId, i);

            messages.add(mes);

        }

        return messages;

    }

    @Override

    public void run() {

        String[] name = file.split("/");

        if (!peer.getControl().getBackedUpFiles().containsKey(name[name.length - 1])) {

            System.out.println("wrong in ");

            return;

        }

        float size;

        try {

            size = Files.size(Paths.get(file));

        } catch (IOException e) {

            e.printStackTrace();

            System.out.println("Do not exists");

            return;

        }

        ArrayList<Message> messages = createMessages(size, peer.getControl().getBackedUpFiles().get(name[name.length - 1]));

        do {

            for (Message i : messages) {

              peer.getThreadPool().schedule(() -> {

                    peer.getMc_channel().send(i);

                }, 0, TimeUnit.MILLISECONDS);

            }

        } while (!allChanksreicived(messages));

        ByteArrayOutputStream my_stream = new ByteArrayOutputStream();

        byte[] fileInfo;

        try {

            for (Chunk i : peer.getControl().getRestoredChunks().get(messages.get(0).getFileID())) {

                my_stream.write(i.getInfo());

            }

            fileInfo = my_stream.toByteArray();

            peer.getStorage().saveFile(peer.getControl().getBackedUpFiles().get(name[name.length - 1]), fileInfo);

        } catch (Exception e) {

            e.printStackTrace();

        }

        System.out.println("Restored");

    }

    boolean allChanksreicived(ArrayList<Message> messages) {

        try {

            if (peer.getControl().getRestoredChunks().get(messages.get(0).getFileID()) == null) {

                return false;

            }

            return messages.size() == peer.getControl().getRestoredChunks().get(messages.get(0).getFileID()).size();

        } catch (Exception e) {

            e.printStackTrace();

        }

        return false;

    }

}