sdis1819-t7g06

package Message;

import java.io.BufferedReader;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.io.InputStreamReader;

import java.net.DatagramPacket;

import static Global.Globals.*;

public class Message {

    protected MessageType type;

    protected String version;

    protected int senderId;

    protected String fileID;

    protected Integer chunkNo;

    protected Integer repDegree;

    protected byte[] body;

    /**

     * @param type      Message.Message type

     * @param version

     * @param senderId

     * @param fileID

     * @param chunkNo

     * @param repDegree

     * @param body

     */

    public Message(MessageType type, String version, int senderId, String fileID, Integer chunkNo, Integer repDegree, byte[] body) {

        this.type = type;

        this.version = version;

        this.senderId = senderId;

        this.fileID = fileID;

        this.chunkNo = chunkNo;

        this.repDegree = repDegree;

        this.body = body;

    }

    /**

     * Translate UDP datagram packet

     *

     * @param message Message.Message that comes fr

     *                om UDP Datagram

     */

    public Message(DatagramPacket message) {

        if (message.getData().length == 0 || message.getLength() == 0) {

            return;

        }

        int messageLength = message.getLength();

        byte[] data = message.getData();

        byte[] header = new byte[messageLength];

        System.arraycopy(data, 0, header, 0, messageLength);

        ByteArrayInputStream getHeader = new ByteArrayInputStream(header);

        BufferedReader buffer = new BufferedReader(new InputStreamReader(getHeader));

        try {

            String head = buffer.readLine();

            int size = head.length() + 4;

            this.body = new byte[messageLength - size];

            System.arraycopy(data, size, this.body, 0, messageLength - size);

            String[] l = head.split(" ");

            ms(l);

            this.version = l[1];

            this.senderId = Integer.parseInt(l[2]);

            this.fileID = l[3];

        } catch (IOException e) {

            //e.printStackTrace();

        }

    }

    void ms(String[] header) {

        switch (header[0]) {

            case "PUTCHUNK":

                this.type = MessageType.PUTCHUNK;

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

                this.repDegree = Integer.parseInt(header[5]);

                break;

            case "STORED":

                this.type = MessageType.STORED;

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

                this.repDegree = null;

                break;

            case "REMOVED":

                this.type = MessageType.REMOVED;

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

                this.repDegree = null;

                break;

            case "GETCHUNK":

                this.type = MessageType.GETCHUNK;

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

                this.repDegree = null;

                break;

            case "DELETE":

                this.type = MessageType.DELETE;

                this.chunkNo = null;

                this.repDegree = null;

                break;

            case "CHUNK":

                this.type = MessageType.CHUNK;

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

                this.repDegree = null;

                break;

            default:

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

                return;

        }

    }

    public byte[] messagePacket() {

        try {

            String message = "";

            String type = typeToString(this.type);

            String senderId = Integer.toString(this.senderId);

            message += type + " " + this.version + " " + senderId + " " + this.fileID;

            String chunkNo;

            if (this.chunkNo != null) {

                chunkNo = (this.chunkNo).toString();

                message += " " + chunkNo;

            }

            if (this.repDegree != null)

                message += " " + this.repDegree;

            message += " " + CRLF + CRLF;

            byte[] messageByte = message.getBytes();

            byte[] result;

            if (this.body != null) {

                result = new byte[messageByte.length + this.body.length];

                System.arraycopy(messageByte, 0, result, 0, messageByte.length);

                System.arraycopy(this.body, 0, result, messageByte.length, this.body.length);

                return result;

            }

            return messageByte;

        } catch (Exception e) {

           // e.printStackTrace();

        }

        return null;

    }

    String typeToString(MessageType type) {

        switch (type) {

            case PUTCHUNK:

                return "PUTCHUNK";

            case STORED:

                return "STORED";

            case REMOVED:

                return "REMOVED";

            case GETCHUNK:

                return "GETCHUNK";

            case DELETE:

                return "DELETE";

            case CHUNK:

                return "CHUNK";

            default:

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

                return "NULL";

        }

    }

    public int getSenderId() {

        return senderId;

    }

    public Integer getChunkNo() {

        return chunkNo;

    }

    public Integer getRepDegree() {

        return repDegree;

    }

    public byte[] getBody() {

        return body;

    }

    public String getVersion() {

        return version;

    }

    public String getFileID() {

        return fileID;

    }

    public MessageType getType() {

        return type;

    }

}