File: body.rb

package info (click to toggle)
ruby-aws-sdk-core 3.242.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky
  • size: 2,420 kB
  • sloc: ruby: 18,795; makefile: 4
file content (105 lines) | stat: -rw-r--r-- 3,214 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
# frozen_string_literal: true

module Aws
  module Rest
    module Request
      class Body

        include Seahorse::Model::Shapes

        # @param [Class] serializer_class
        # @param [Seahorse::Model::ShapeRef] rules
        def initialize(serializer_class, rules)
          @serializer_class = serializer_class
          @rules = rules
        end

        # @param [Seahorse::Client::Http::Request] http_req
        # @param [Hash] params
        def apply(http_req, params)
          body = build_body(params)

          # for rest-json, ensure we send at least an empty object
          # don't send an empty object for streaming? case.
          if body.nil? &&
             json_builder? &&
             modeled_body? &&
             !streaming?
            body = '{}'
          end
          http_req.body = body
        end

        private

        # operation is modeled for body when it is modeled for a payload
        # either with payload trait or normal members.
        def modeled_body?
          return true if @rules[:payload]
          @rules.shape.members.each do |member|
            _name, shape = member
            return true if shape.location.nil?
          end
          false
        end

        def build_body(params)
          if streaming?
            params[@rules[:payload]]
          elsif @rules[:payload]
            params = params[@rules[:payload]]
            if params
              if xml_builder? &&
                 @rules.shape.member?(@rules[:payload_member].location_name)
                # serializing payload member name for rest-xml is as follows:
                # 1. Use the member locationName if the member value doesn't match the member's name (default)
                # 2. Use the value of the locationName on the member's target if present
                # 3. Use the shape name of the member's target
                serialize(@rules[:payload_member], params, location_name: payload_location_name)
              else
                serialize(@rules[:payload_member], params)
              end
            end
          else
            params = body_params(params)
            serialize(@rules, params) unless params.empty?
          end
        end

        def payload_location_name
          @rules[:payload_member].shape['locationName'] ||
            @rules[:payload_member].shape.name
        end

        def streaming?
          @rules[:payload] && (
            BlobShape === @rules[:payload_member].shape ||
            StringShape === @rules[:payload_member].shape
          )
        end

        def xml_builder?
          @serializer_class == Xml::Builder
        end

        def json_builder?
          @serializer_class == Json::Builder
        end

        def serialize(rules, params, location_name: nil)
          @serializer_class.new(rules, location_name: location_name).serialize(params)
        end

        def body_params(params)
          @rules.shape.members.inject({}) do |hash, (member_name, member_ref)|
            if !member_ref.location && params.key?(member_name)
              hash[member_name] = params[member_name]
            end
            hash
          end
        end

      end
    end
  end
end