sdis1819-t2g03

package handlers;

import utils.Utils;

import utils.Utils.MessageType;

import java.io.*;

public class Message {

        private int argsNo;

        private int chunkNo;

        private int senderID;

        private String fileID;

        private String version;

        private int replication;

        private MessageType msgType;

        private byte [] body;

        public Message(byte[] information, int length) { 

                String header;

                String headerWithoutStuff;

                String [] splittedHeader;

                header= readHeader(information);

                headerWithoutStuff = header.trim().replaceAll("\\s+", " ");

                splittedHeader = headerWithoutStuff.split("\\s+");

                translateHeader(splittedHeader);

                if (msgType == MessageType.CHUNK)

                        this.body = readBody(information, header.length(), length);

                if(msgType == MessageType.PUTCHUNK) 

                        this.body = readBody(information, header.length(), length);

        }

        public Message(MessageType type, String[] args) {

                this.msgType = type;

                version = args[0];

                senderID = Integer.parseInt(args[1]);

                fileID = args[2];

                if (type != MessageType.DELETE)

                        chunkNo = Integer.parseInt(args[3]);

                if (type == MessageType.PUTCHUNK) {

                        replication = Integer.parseInt(args[4]);

                }

        }

        public Message(MessageType type, String[] args, byte[] information) {

                this(type, args);

                body = information;

        }

        private String readHeader(byte[] information) {

                ByteArrayInputStream stream;

                BufferedReader reader;

                String header = "";

                stream = new ByteArrayInputStream(information);

                reader = new BufferedReader(new InputStreamReader(stream));

                try {

                        header = reader.readLine();

                } catch (IOException e) {

                        e.printStackTrace();

                }

                return header;

        }

        private byte[] readBody(byte[] information, int headerLength, int informationLength) {

                int readBytes;

                ByteArrayInputStream message;

                byte[] bodyContent;

                readBytes = informationLength - headerLength - 4;

                message = new ByteArrayInputStream(information, headerLength + 4, readBytes);

                bodyContent = new byte[readBytes];

                message.read(bodyContent, 0, readBytes);

                return bodyContent;

        }

        private void translateHeader(String[] headerSplit) {

                switch (headerSplit[0]) {

                case "PUTCHUNK": 

                {

                        msgType = MessageType.PUTCHUNK;

                        argsNo = 6;

                } break;

                case "STORED":

                {

                        msgType = MessageType.STORED;

                        argsNo = 5;

                }

                break;

                case "GETCHUNK":

                {        

                        msgType = MessageType.GETCHUNK;

                        argsNo = 5;

                } break;

                case "CHUNK":

                {        msgType = MessageType.CHUNK;

                argsNo = 5;

                } break;

                case "DELETE":

                {        

                        msgType = MessageType.DELETE;

                        argsNo = 4;

                } break;

                case "REMOVED":

                {        

                        msgType = MessageType.REMOVED;

                        argsNo = 5;

                } break;

                default:

                        return;

                }

                if (headerSplit.length != argsNo)

                        return;

                version = headerSplit[1];

                senderID = Integer.parseInt(headerSplit[2]);

                fileID = headerSplit[3];

                if (argsNo > 4)

                        chunkNo = Integer.parseInt(headerSplit[4]);

                if (msgType == MessageType.PUTCHUNK)

                        replication = Integer.parseInt(headerSplit[5]);

        }

        public String headerString() {

                String string;

                if(msgType == MessageType.PUTCHUNK)

                        string = msgType + " " + version + " " + senderID + " " + fileID + " " + chunkNo + " " + replication + " " + Utils.CRLF + Utils.CRLF;

                else if(msgType == MessageType.DELETE)

                        string = msgType + " " + version + " " + senderID + " " + fileID + " " + Utils.CRLF + Utils.CRLF;

                else        

                        string = msgType + " " + version + " " + senderID + " " + fileID + " " + chunkNo + " " + Utils.CRLF + Utils.CRLF;

                return string;

        }

        public byte[] getBytes() throws IOException {

                byte header[]; 

                header = headerString().getBytes();

                if (body == null)

                        return header;

                else {

                        ByteArrayOutputStream out = new ByteArrayOutputStream();

                        out.write(header);

                        out.write(body);

                        return out.toByteArray();

                }

        }

        public String getVersion() {

                return version;

        }

        public int getSenderID() {

                return senderID;

        }

        public String getFileID() {

                return fileID;

        }

        public int getChunkNo() {

                return chunkNo;

        }

        public int getReplicationDegree() {

                return replication;

        }

        public byte[] getBody() {

                return body;

        }

        public MessageType getType() {

                return msgType;

        }

        @Override

        public String toString() {

                String str;

                if(msgType == MessageType.PUTCHUNK)

                {

                        str = "<" + msgType +">" + " " + "<" + version+">" + " " + "<" + senderID+">" + " " +"<" + fileID+">" + " " + "<" +chunkNo+">";

                } 

                else if(msgType == MessageType.DELETE) 

                {

                        str = "<" + msgType +">" + " " + "<" + version+">" + " " + "<" + senderID+">" + " " +"<" + fileID+">";

                }

                else 

                {

                        str = "<" + msgType +">" + " " + "<" + version+">" + " " + "<" + senderID+">" + " " +"<" + fileID+">" + " " + "<" +chunkNo+">";

                }

                return str;

        }

}