timecrafters_configuration_tool_desktop/lib/tacnet/packet.rb

module TAC
  class TACNET
    class Packet
      PROTOCOL_VERSION = "0"
      PROTOCOL_HEADER_SEPERATOR = "|"
      PROTOCOL_HEARTBEAT = "heartbeat"

      PACKET_TYPES = {
        handshake: 0,
        heartbeat: 1,
        dump_config: 2,

        add_group: 3,
        update_group: 4,
        delete_group: 5,

        add_action: 6,
        update_action: 7,
        delete_action: 8,

        add_variable: 9,
        update_variable: 10,
        delete_variable: 11,
      }

      def self.from_stream(message)
        slice = message.split("|", 4)

        if slice.size < 4
          warn "Failed to split packet along first 4 " + PROTOCOL_HEADER_SEPERATOR + ". Raw return: " + Arrays.toString(slice)
          return nil
        end

        if slice.first != PROTOCOL_VERSION
          warn "Incompatible protocol version received, expected: " + PROTOCOL_VERSION + " got: " + slice.first
          return nil
        end

        unless valid_packet_type?(Integer(slice[1]))
          warn "Unknown packet type detected: #{slice[1]}"
          return nil
        end

        version = slice[0]
        type = PACKET_TYPES.key(Integer(slice[1]))
        content_length = Integer(slice[2])
        content = slice[3]

        return Packet.new(version, type, content_length, body)
      end

      def self.create(packet_type, body)
        Packet.new(PROTOCOL_VERSION, packet_type, body.length, body)
      end

      def self.valid_packet_type?(packet_type)
        PACKET_TYPES.values.find { |t| t == packet_type }
      end

      attr_reader :version, :type, :content_length, :body
      def initialize(version, type, content_length, body)
        @version = version
        @type = type
        @content_length = content_length
        @body = body
      end

      def encode_header
        string = ""
        string += PROTOCOL_VERSION
        string += PROTOCOL_HEADER_SEPERATOR
        string += packet_type
        string += PROTOCOL_HEADER_SEPERATOR
        string += content_length
        string += PROTOCOL_HEADER_SEPERATOR

        return string
      end

      def valid?
        true
      end

      def to_s
        "#{encode_header}#{body}"
      end
    end
  end
end