class Aws::Json::RpcHeadersHandler

def add_headers(context)
  context.http_request.headers['Content-Type'] = content_type(context)
  context.http_request.headers['X-Amz-Target'] = target(context)
  context
end

def call(context)
  @handler.call(add_headers(context)).on(200..599) do |resp|
    context[:request_id] = context.http_response.headers['x-amzn-requestid']
  end
end

def content_type(context)
  CONTENT_TYPE % [context.config.api.metadata('jsonVersion')]
end

def target(context)
  prefix = context.config.api.metadata('targetPrefix')
  "#{prefix}.#{context.operation.name}"
end