sdis1819-t2g11

import java.io.BufferedReader;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.io.InputStreamReader;

public class Message   {

    public enum MessageType {

        PUTCHUNK, STORED, GETCHUNK, CHUNK, DELETE, REMOVED

    }

    public static String CRLF = "\r\n";

    public static String SPACE = " ";

    // Header

    private MessageType type = null;

    private Integer senderID = null;

    private String protocolVersion = null;

    private String fileID = null;

    private Integer chunkNo = null;

    private byte[] message;

    private Integer replicationDegree = null;

    // Body

    private byte[] body = null;

    public byte[] getBody() {

        return body;

    }

    public void setBody(byte[] body){

        this.body=body;

    }

    public void setMessageType(MessageType mgs){

        this.type=mgs;

    }

    public void setSenderId(Integer id){

        this.senderID=id;

    }

    public int getReplicationDegree(){

        return replicationDegree;

    }

    public void setReplicationDegree(int rep){

        this.replicationDegree=rep;

    }

    public String getProtocolVersion() {

        return protocolVersion;

    }

    public void setProtocolVersion(String pv) {

        this.protocolVersion=pv;

    }

    public Integer getSenderID() {

        return senderID;

    }

    public String getFileID() {

        return fileID;

    }

    public void setFileID(String fileId) {

        this.fileID=fileId;

    }

    public Integer getChunkNo() {

        return chunkNo;

    }

    public void setChunkNo(Integer no) {

        this.chunkNo= no;

    }

    public MessageType getType(){

        return this.type;

    }

    public byte[] getMessage(){

        return message;

    }

    // Processes a message from DatagramPacket (handles received messages)

    public Message(byte[] message, int length) throws Exception{

        this.message=message;

        ByteArrayInputStream arrayInputStream = new ByteArrayInputStream(message);

        BufferedReader reader = new BufferedReader(

                new InputStreamReader(arrayInputStream));

        String header = "";

        try {

            header = reader.readLine();

        } catch (IOException e) {

            e.printStackTrace();

        }

        if(header == "" ) {

            System.out.println("[ERROR] Message header badly formed!");

            return;

        }

        String[] headerSplit = header.split("\\s+");

        this.protocolVersion = headerSplit[1]; //tem de ser igual a versao do cliente

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

        this.fileID = headerSplit[3];

        switch (headerSplit[0]){

            case "PUTCHUNK":

                if (headerSplit.length != 6) {

                    System.out.println("[ERROR] No. of header fields does not follow <PUTCHUNK> message type.");

                    return;

                }

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

                this.type = MessageType.PUTCHUNK;

                this.replicationDegree = Integer.parseInt(headerSplit[5]);

                break;

            case "STORED":

                if (headerSplit.length != 5) {

                    System.out.println("[ERROR] No. of header fields does not follow <STORE> message type.");

                    return;

                }

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

                this.type = MessageType.STORED;

                break;

            case "GETCHUNK":

                if (headerSplit.length != 5) {

                    System.out.println("[ERROR] No. of header fields does not follow <GETCHUNK> message type.");

                    return;

                }

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

                this.type = MessageType.GETCHUNK;

                break;

            case "CHUNK":

                if (headerSplit.length != 5) {

                    System.out.println("[ERROR] No. of header fields does not follow <CHUNK> message type.");

                    return;

                }

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

                this.type = MessageType.CHUNK;

                break;

            case "DELETE":

                if (headerSplit.length != 4) {

                    System.out.println("[ERROR] No. of header fields does not follow <DELETE> message type.");

                    return;

                }

                this.type = MessageType.DELETE;

                break;

            case "REMOVED":

                if (headerSplit.length != 5) {

                    System.out.println("[ERROR] No. of header fields does not follow <REMOVED> message type.");

                    return;

                }

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

                this.type = MessageType.REMOVED;

                break;

            default:

                System.out.println("[ERROR] Not a valid message type.");

        }

        if(this.type == MessageType.PUTCHUNK || this.type == MessageType.CHUNK){

            int bodyLength = length - header.length() - 4;

            this.body = new byte[bodyLength];

            System.arraycopy(message, header.length()+4, this.body, 0, bodyLength);

        }

    }

    public String msgTypeToString(MessageType msgType){

        switch(msgType){

            case PUTCHUNK:

                return "PUTCHUNK";

            case STORED:

                return "STORED";

            case GETCHUNK:

                return "GETCHUNK";

            case CHUNK:

                return "CHUNK";

            case DELETE:

                return "DELETE";

            case REMOVED:

                return "REMOVED";

            default:

                break;

        }

        return "";

    }

    @Override

    public String toString(){

        StringBuilder string = new StringBuilder();

        string.append(msgTypeToString(this.type)).append(SPACE).append(this.protocolVersion).append(SPACE).append(this.senderID).append(SPACE).append(this.fileID).append(SPACE);

        if (this.replicationDegree != null){

            string.append(this.replicationDegree).append(SPACE);

        }

        if (this.chunkNo != null){

            string.append(this.chunkNo);

        }

        string.append(CRLF+CRLF);

        if(this.body != null)

            string.append(new String(this.body));

        return string.toString();

    }

}