info.conspire.temur.network.protocol.ClientMessage.java Source code

Introduction

Here is the source code for info.conspire.temur.network.protocol.ClientMessage.java

Source

package info.conspire.temur.network.protocol;

import info.conspire.temur.util.Base64Encoding;
import info.conspire.temur.util.WireEncoding;
import io.netty.buffer.ByteBuf;

/**
 * "THE BEER-WARE LICENSE" (Revision 42):
 * <crowlie@hybridcore.net> wrote this file. As long as you retain this notice you
 * can do whatever you want with this stuff. If we meet some day, and you think
 * this stuff is worth it, you can buy me a beer in return Crowley.
 */
public class ClientMessage {
    private final static int ID_LENGTH = 2;

    private int id;
    private int length;
    private int originIndex = 0;

    private ByteBuf buffer;

    public ClientMessage(int id, int length, ByteBuf buffer) {
        this.length = length;
        this.id = id;
        this.buffer = buffer;
        this.originIndex = buffer.readerIndex();
    }

    public int getID() {
        return id;
    }

    public int getLength() {
        return length;
    }

    public void resetIndex() {
        this.buffer.readerIndex(this.originIndex);
    }

    public void advanceOffset(int i) {
        buffer.readerIndex((this.originIndex + i));
    }

    public byte[] readBytes(int i) {
        return buffer.readBytes(i).array();
    }

    public int readInt() {
        int size = this.remaining();
        int pos = this.buffer.readerIndex();

        if (size > WireEncoding.MAX_INTEGER_BYTE_AMOUNT) {
            size = WireEncoding.MAX_INTEGER_BYTE_AMOUNT;
        }

        int[] dat = WireEncoding.decodeInt(this.buffer.readBytes(size).array());

        this.buffer.readerIndex(pos + dat[1]);
        return dat[0];
    }

    public int readFixedInt() {
        return Base64Encoding.PopInt(this.readBytes(2));
    }

    public int parseFixedInt() {
        try {
            return Integer.parseInt(this.readString());
        } catch (NumberFormatException e) {
            return 0;
        }
    }

    public boolean readBoolean() {
        return (this.buffer.readByte() == WireEncoding.POSITIVE);
    }

    public String readString() {
        return new String(this.readBytes(this.readFixedInt()));
    }

    private int remaining() {
        return (this.length - (this.buffer.readerIndex() - this.originIndex));
    }

    public String getHeader() {
        String resp = "";
        resp += new String(Base64Encoding.encodeInt(this.getID(), ClientMessage.ID_LENGTH));

        return resp;
    }

    public String getBody() {
        int readerIndex = this.buffer.readerIndex();
        this.resetIndex();

        String resp = "";
        resp += new String(this.buffer.readBytes(this.length).array());

        this.buffer.readerIndex(readerIndex);

        return resp;
    }

    public String toString() {
        return this.getHeader() + this.getBody();
    }
}