module Vertica
  module Messages

    class Message
      def self.message_id(message_id)
        self.send(:define_method, :message_id) { message_id }
      end

      def message_string(msg)
        msg = msg.join if msg.is_a?(Array)
        bytesize = msg.respond_to?(:bytesize) ? 4 + msg.bytesize : 4 + msg.size
        message_size = [bytesize].pack('N')
        message_id ? "#{message_id}#{message_size}#{msg}" : "#{message_size}#{msg}"
      end
    end

    class BackendMessage < Message
      MessageIdMap = {}

      def self.factory(type, data)
        #puts "factory reading message #{type} #{size} #{type.class}"
        if klass = MessageIdMap[type]           #explicitly use the char value, for 1.9 compat
          klass.new data
        else
          Messages::Unknown.new type, data
        end
      end

      def self.message_id(message_id)
        super
        MessageIdMap[message_id] = self          #explicitly use the char value, for 1.9 compat
      end

      def initialize(data)
      end
    end

    class FrontendMessage < Message
      def to_bytes
        message_string ''
      end
    end

  end
end