Frame.java

package org.httpkit.server;

import org.httpkit.HttpUtils;

import java.nio.ByteBuffer;

public abstract class Frame {
    public final byte[] data;

    public Frame(byte data[]) {
        this.data = data;
    }

    public static class BinaryFrame extends Frame {
        public BinaryFrame(byte[] data) {
            super(data);
        }
    }

    public static class CloseFrame extends Frame {

        public final static short CLOSE_NORMAL = 1000;
        // indicates that an endpoint is "going away", such as a server going down
        // or a browser having navigated away from a page.
        public final static short CLOSE_AWAY = 1001;
        //        public final static short CLOSE_PROTOCOL_ERROR = 1002;
//        public final static short CLOSE_NOT_IMPL = 1003;
        // This is a generic status code that can be returned when there is no other
        // more suitable status code (e.g., 1003 or 1009)
//        public final static short CLOSE_VIOLATES_POLICY = 1008;
        public final static short CLOSE_MESG_BIG = 1009;
//        public final static short CLOSE_SEVER_ERROR = 1011;

//        private static byte[] bytes(short code) {
//            return ByteBuffer.allocate(2).putShort(code).array();
//        }

//        public static final CloseFrame NORMAL = new CloseFrame(bytes(CLOSE_NORMAL));
//        public static final CloseFrame AWAY = new CloseFrame(bytes(CLOSE_AWAY));
//        public static final CloseFrame MESG_BIG = new CloseFrame(bytes(CLOSE_MESG_BIG));
//        public static final CloseFrame SERVER_ERROR = new CloseFrame(bytes(CLOSE_MESG_BIG));

        public CloseFrame(byte[] data) {
            super(data);
        }

        public int getStatus() {
            if (data.length >= 2) {
                return ByteBuffer.wrap(data, 0, 2).getShort();
            }
            return CLOSE_NORMAL;
        }
    }

    public static class PingFrame extends Frame {
        public PingFrame(byte[] data) {
            super(data);
        }
    }

    public static class PongFrame extends Frame {
        public PongFrame(byte[] data) {
            super(data);
        }
    }

    public static class TextFrame extends Frame {
        private final String msg;

        public TextFrame(byte[] data) {
            super(data);
            this.msg = new String(data, HttpUtils.UTF_8);
        }

        public String getText() {
            return msg;
        }
    }
}