class Aws::RpcV2::Handler

def apply_request_id(context)
  context[:request_id] = context.http_response.headers['x-amzn-requestid']
end

def build_body(context)
  Builder.new(context.operation.input).serialize(context.params)
end

def build_request(context)
  context.http_request.headers['Smithy-Protocol'] = 'rpc-v2-cbor'
  context.http_request.headers['X-Amzn-Query-Mode'] = 'true' if query_compatible?(context)
  context.http_request.http_method = 'POST'
  context.http_request.body = build_body(context)
  build_url(context)
end

def build_url(context)
  base = context.http_request.endpoint
  service_name = context.config.api.metadata['targetPrefix']
  base.path += "/service/#{service_name}/operation/#{context.operation.name}"
end

def call(context)
  build_request(context)
  response = with_metric { @handler.call(context) }
  response.on(200..299) { |resp| resp.data = parse_body(context) }
  response.on(200..599) { |_resp| apply_request_id(context) }
  response
end

def parse_body(context)
  cbor = context.http_response.body_contents
  if (rules = context.operation.output)
    if cbor.is_a?(Array)
      # an array of emitted events
      if cbor[0].respond_to?(:response)
        # initial response exists
        # it must be the first event arrived
        resp_struct = cbor.shift.response
      else
        resp_struct = context.operation.output.shape.struct_class.new
      end
      rules.shape.members.each do |name, ref|
        if ref.eventstream
          resp_struct.send("#{name}=", cbor.to_enum)
        end
      end
      resp_struct
    else
      Parser.new(
        rules,
        query_compatible: query_compatible?(context)
      ).parse(cbor)
    end
  else
    EmptyStructure.new
  end
end

def query_compatible?(context)
  context.config.api.metadata.key?('awsQueryCompatible')
end

def with_metric(&block)
  Aws::Plugins::UserAgent.metric('PROTOCOL_RPC_V2_CBOR', &block)
end