# encoding: utf-8
# Copyright 2014 Aerospike, Inc.
#
# Portions may be licensed to Aerospike, Inc. under one or more contributor
# license agreements.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
# use this file except in compliance with the License. You may obtain a copy of
# the License at http:#www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations under
# the License.

require 'aerospike/record'

require 'aerospike/command/single_command'
require 'aerospike/utils/epoc'
require 'aerospike/value/value'

module Aerospike

  private

  class ReadCommand < SingleCommand

    attr_reader :record

    def initialize(cluster, policy, key, bin_names)
      super(cluster, key)

      @bin_names = bin_names
      @policy = policy

      self
    end

    def write_buffer
      set_read(@key, @bin_names)
    end

    def parse_result
      # Read header.
      begin
        @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
      rescue => e
        Aerospike.logger.error("parse result error: #{e}")
        raise e
      end

      # A number of these are commented out because we just don't care enough to read
      # that section of the header. If we do care, uncomment and check!
      sz = @data_buffer.read_int64(0)
      header_length = @data_buffer.read(8).ord
      result_code = @data_buffer.read(13).ord & 0xFF
      generation = @data_buffer.read_int32(14)
      expiration = Aerospike.TTL(@data_buffer.read_int32(18))
      field_count = @data_buffer.read_int16(26) # almost certainly 0
      op_count = @data_buffer.read_int16(28)
      receive_size = (sz & 0xFFFFFFFFFFFF) - header_length

      # Read remaining message bytes.
      if receive_size > 0
        size_buffer_sz(receive_size)

        begin
          @conn.read(@data_buffer, receive_size)
        rescue => e
          Aerospike.logger.warn("parse result error: #{e}")
          raise e
        end

      end

      if result_code != 0
        return nil if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
        return nil if result_code == Aerospike::ResultCode::LARGE_ITEM_NOT_FOUND
        if result_code == Aerospike::ResultCode::UDF_BAD_RESPONSE
          begin
            @record = parse_record(op_count, field_count, generation, expiration)
            handle_udf_error(result_code)
          rescue => e
            Aerospike.logger.warn("UDF execution error: #{e}")
            raise e
          end

        end

        raise Aerospike::Exceptions::Aerospike.new(result_code)
      end

      if op_count == 0
        # data Bin was not returned.
        @record = Record.new(@node, @key, nil, nil, generation, expiration)
        return
      end

      @record = parse_record(op_count, field_count, generation, expiration)
    end

    def handle_udf_error(result_code)
      ret = @record.bins['FAILURE']
      raise Aerospike::Exceptions::Aerospike.new(result_code, ret) if ret
      raise Aerospike::Exceptions::Aerospike.new(result_code)
    end

    def parse_record(op_count, field_count, generation, expiration)
      bins = nil
      duplicates = nil
      receive_offset = 0

      # There can be fields in the response (setname etc).
      # But for now, ignore them. Expose them to the API if needed in the future.
      if field_count != 0
        # Just skip over all the fields
        for i in 0...field_count
          field_size = @data_buffer.read_int32(receive_offset)
          receive_offset += (4 + field_size)
        end
      end

      for i in 0...op_count
        op_size = @data_buffer.read_int32(receive_offset)
        particle_type = @data_buffer.read(receive_offset+5).ord
        version = @data_buffer.read(receive_offset+6).ord
        name_size = @data_buffer.read(receive_offset+7).ord
        name = @data_buffer.read(receive_offset+8, name_size).force_encoding('utf-8')
        receive_offset += 4 + 4 + name_size


        particle_bytes_size = op_size - (4 + name_size)
        value = Aerospike.bytes_to_particle(particle_type, @data_buffer, receive_offset, particle_bytes_size)
        receive_offset += particle_bytes_size

        vmap = {}

        if version > 0 || duplicates != nil
          unless duplicates
            duplicates = []
            duplicates << bins
            bins = nil

            for j in 0..version-1
              duplicates << nil
            end
          else
            for j in duplicates.length..version
              duplicates << nil
            end
          end

          vmap = duplicates[version]
          unless vmap
            vmap = {}
            duplicates[version] = vmap
          end
        else
          bins = {} unless bins
          vmap = bins
        end
        vmap[name] = value
      end

      # Remove nil duplicates just in case there were holes in the version number space.
      duplicates.compact! if duplicates

      Record.new(@node, @key, bins, duplicates, generation, expiration)
    end

  end # class

end # module