#
# Copyright (c) 2006- Facebook
# Distributed under the Apache Software License
#
# See accompanying file LICENSE or visit the Thrift site at:
# http://developers.facebook.com/thrift/
#
# Author: Mark Slee <mcslee@facebook.com>
#

# this require is to make generated struct definitions happy
require 'set'

module Thrift
  class ProtocolException < Exception

    UNKNOWN = 0
    INVALID_DATA = 1
    NEGATIVE_SIZE = 2
    SIZE_LIMIT = 3
    BAD_VERSION = 4

    attr_reader :type

    def initialize(type=UNKNOWN, message=nil)
      super(message)
      @type = type
    end

  end
  deprecate_class! :TProtocolException => ProtocolException

  class Protocol

    attr_reader :trans

    def initialize(trans)
      @trans = trans
    end

    def native?
      puts "wrong method is being called!"
      false
    end

    def write_message_begin(name, type, seqid); nil; end
    deprecate! :writeMessageBegin => :write_message_begin

    def write_message_end; nil; end
    deprecate! :writeMessageEnd => :write_message_end

    def write_struct_begin(name); nil; end
    deprecate! :writeStructBegin => :write_struct_begin

    def write_struct_end; nil; end
    deprecate! :writeStructEnd => :write_struct_end

    def write_field_begin(name, type, id); nil; end
    deprecate! :writeFieldBegin => :write_field_begin

    def write_field_end; nil; end
    deprecate! :writeFieldEnd => :write_field_end

    def write_field_stop; nil; end
    deprecate! :writeFieldStop => :write_field_stop

    def write_map_begin(ktype, vtype, size); nil; end
    deprecate! :writeMapBegin => :write_map_begin

    def write_map_end; nil; end
    deprecate! :writeMapEnd => :write_map_end

    def write_list_begin(etype, size); nil; end
    deprecate! :writeListBegin => :write_list_begin

    def write_list_end; nil; end
    deprecate! :writeListEnd => :write_list_end

    def write_set_begin(etype, size); nil; end
    deprecate! :writeSetBegin => :write_set_begin

    def write_set_end; nil; end
    deprecate! :writeSetEnd => :write_set_end

    def write_bool(bool); nil; end
    deprecate! :writeBool => :write_bool

    def write_byte(byte); nil; end
    deprecate! :writeByte => :write_byte

    def write_i16(i16); nil; end
    deprecate! :writeI16 => :write_i16

    def write_i32(i32); nil; end
    deprecate! :writeI32 => :write_i32

    def write_i64(i64); nil; end
    deprecate! :writeI64 => :write_i64

    def write_double(dub); nil; end
    deprecate! :writeDouble => :write_double

    def write_string(str); nil; end
    deprecate! :writeString => :write_string

    def read_message_begin; nil; end
    deprecate! :readMessageBegin => :read_message_begin

    def read_message_end; nil; end
    deprecate! :readMessageEnd => :read_message_end

    def read_struct_begin; nil; end
    deprecate! :readStructBegin => :read_struct_begin

    def read_struct_end; nil; end
    deprecate! :readStructEnd => :read_struct_end

    def read_field_begin; nil; end
    deprecate! :readFieldBegin => :read_field_begin

    def read_field_end; nil; end
    deprecate! :readFieldEnd => :read_field_end

    def read_map_begin; nil; end
    deprecate! :readMapBegin => :read_map_begin

    def read_map_end; nil; end
    deprecate! :readMapEnd => :read_map_end

    def read_list_begin; nil; end
    deprecate! :readListBegin => :read_list_begin

    def read_list_end; nil; end
    deprecate! :readListEnd => :read_list_end

    def read_set_begin; nil; end
    deprecate! :readSetBegin => :read_set_begin

    def read_set_end; nil; end
    deprecate! :readSetEnd => :read_set_end

    def read_bool; nil; end
    deprecate! :readBool => :read_bool

    def read_byte; nil; end
    deprecate! :readByte => :read_byte

    def read_i16; nil; end
    deprecate! :readI16 => :read_i16

    def read_i32; nil; end
    deprecate! :readI32 => :read_i32

    def read_i64; nil; end
    deprecate! :readI64 => :read_i64

    def read_double; nil; end
    deprecate! :readDouble => :read_double

    def read_string; nil; end
    deprecate! :readString => :read_string

    def write_field(name, type, fid, value)
      write_field_begin(name, type, fid)
      write_type(type, value)
      write_field_end
    end

    def write_type(type, value)
      case type
      when Types::BOOL
        write_bool(value)
      when Types::BYTE
        write_byte(value)
      when Types::DOUBLE
        write_double(value)
      when Types::I16
        write_i16(value)
      when Types::I32
        write_i32(value)
      when Types::I64
        write_i64(value)
      when Types::STRING
        write_string(value)
      when Types::STRUCT
        value.write(self)
      else
        raise NotImplementedError
      end
    end

    def read_type(type)
      case type
      when Types::BOOL
        read_bool
      when Types::BYTE
        read_byte
      when Types::DOUBLE
        read_double
      when Types::I16
        read_i16
      when Types::I32
        read_i32
      when Types::I64
        read_i64
      when Types::STRING
        read_string
      else
        raise NotImplementedError
      end
    end

    def skip(type)
      case type
      when Types::STOP
        nil
      when Types::BOOL
        read_bool
      when Types::BYTE
        read_byte
      when Types::I16
        read_i16
      when Types::I32
        read_i32
      when Types::I64
        read_i64
      when Types::DOUBLE
        read_double
      when Types::STRING
        read_string
      when Types::STRUCT
        read_struct_begin
        while true
          name, type, id = read_field_begin
          break if type == Types::STOP
          skip(type)
          read_field_end
        end
        read_struct_end
      when Types::MAP
        ktype, vtype, size = read_map_begin
        size.times do
          skip(ktype)
          skip(vtype)
        end
        read_map_end
      when Types::SET
        etype, size = read_set_begin
        size.times do
          skip(etype)
        end
        read_set_end
      when Types::LIST
        etype, size = read_list_begin
        size.times do
          skip(etype)
        end
        read_list_end
      end
    end

  end
  deprecate_class! :TProtocol => Protocol

  class ProtocolFactory
    def get_protocol(trans); nil; end
    deprecate! :getProtocol => :get_protocol
  end
  deprecate_class! :TProtocolFactory => ProtocolFactory
end