sdis1819-t4g03

package threads;

import java.io.*;

import java.util.*;

import server.Server;

public class ManageRestored implements Runnable {

    private String fileName;

    public ManageRestored(String fileName) {

        this.fileName = fileName;

    }

    @Override

    public void run() {

        if (!Server.getStorage().getRemainingChunks().containsValue("false")) {

            if (restoreFile())

                System.out.println("> File restored!\n");

            else System.out.println("ERROR: File not restored.\n");

        } else System.out.println("ERROR: File not restored, chunks missing.\n");

        Server.getStorage().getRemainingChunks().clear();

    }

    private boolean restoreFile() {

        String filePath = "database/" + Server.getServerId() + "/" + this.fileName;

        File file = new File(filePath);

        byte[] body;

        try {

            FileOutputStream fos = new FileOutputStream(file, true);

            if (!file.exists()) {

                file.getParentFile().mkdirs();

                file.createNewFile();

            }

            List<String> sortedChunkKeys = new ArrayList<>(Server.getStorage().getRemainingChunks().keySet());

            sortedChunkKeys.sort((o1, o2) -> {

                int chunkNr1 = Integer.valueOf(o1.split(".")[1]);

                int chunkNr2 = Integer.valueOf(o2.split(".")[1]);

                return Integer.compare(chunkNr1, chunkNr2);

            });

            for (String key : sortedChunkKeys) {

                String chunkPath = Server.getServerId() + "/" + key;

                File chunkFile = new File(chunkPath);

                if (!chunkFile.exists()) {

                    return false;

                }

                body = new byte[(int) chunkFile.length()];

                FileInputStream in = new FileInputStream(chunkFile);

                in.read(body);

                fos.write(body);

                chunkFile.delete();

                in.close();

            }

            fos.close();

            return true;

        } catch (IOException e) {

            e.printStackTrace();

        }

        return false;

    }

}