sdis1819-t2g03

package protocols;

import server.Peer;

import utils.Utils;

import static chunk.FolderManager.fileSplit;

import java.io.File;

import java.util.HashMap;

import java.io.IOException;

import java.nio.file.Files;

import java.util.ArrayList;

import java.io.FileNotFoundException;

import java.nio.file.attribute.BasicFileAttributes;

import java.security.NoSuchAlgorithmException;

import chunk.Chunk;

import chunk.FileManager;

import chunk.FolderManager;

import handlers.BackupChunkHandler;

public class BackupChunk implements Runnable {

        private String fileID;

        private Peer peer;

        private String version;

        private byte[] data;

        private int replication;

        private File file;

        public BackupChunk(String version, File file, int replication, Peer peer) {

                this.version = version;

                this.file = file;

                this.replication = replication;

                this.peer = peer;

        }

        public Peer getPeer() {

                return peer;

        }

        public String getVersion() {

                return version;

        }

        @Override

        public void run() {

                try {

                        data = FolderManager.loadFile(file);

                } catch (IOException e) {

                        e.printStackTrace();

                }

                try {

                        fileID = hashFileID(file);

                } catch (NoSuchAlgorithmException e) {

                        e.printStackTrace();

                }

                ArrayList<Chunk> chunks;

                chunks = fileSplit(data, fileID, replication);

                HashMap<String, Chunk> chunksInfo = new HashMap<>();

                peer.getPeerInformation().startChunkReplication(fileID, chunks.size());

                for (Chunk chunk : chunks) {

                        new Thread(new BackupChunkHandler(this, chunk)).start();

                        chunksInfo.put(Integer.toString(chunk.getChunkNo()), new Chunk(chunk.getChunkNo(), chunk.getReplication()));

                }

                peer.addFileToFileManager(file.getPath(), new FileManager(file, fileID, replication, chunksInfo));

                peer.getPeerInformation().resetChunkReplication(fileID);

                System.out.println("Finished backupInitiator!");

        }

        private String hashFileID(File file) throws NoSuchAlgorithmException {

                BasicFileAttributes attributes;

                try {

                        attributes = Files.readAttributes(file.toPath(), BasicFileAttributes.class);

                } catch (IOException e) {

                        System.out.println("Couldn't read file's metadata: " + e.getMessage());

                        return null;

                }

                String fileId = file.getName() + attributes.lastModifiedTime() + attributes.size();

                return Utils.hash(fileId);

        }

}