Class: Puppet::Indirector::Request

Inherits:
Object
  • Object
show all
Includes:
Util::PsychSupport
Defined in:
lib/puppet/indirector/request.rb

Overview

This class encapsulates all of the information you need to make an Indirection call, and as a result also handles REST calls. It’s somewhat analogous to an HTTP Request object, except tuned for our Indirector.

Constant Summary collapse

OPTION_ATTRIBUTES =

trusted_information is specifically left out because we can’t serialize it and keep it “trusted”

[:ip, :node, :authenticated, :ignore_terminus, :ignore_cache, :instance, :environment]

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Util::PsychSupport

#encode_with, #init_with

Constructor Details

#initialize(indirection_name, method, key, instance, options = {}) ⇒ Request

Returns a new instance of Request.



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
 
# File 'lib/puppet/indirector/request.rb', line 57

def initialize(indirection_name, method, key, instance, options = {})
  @instance = instance
  options ||= {}

  self.indirection_name = indirection_name
  self.method = method

  options = options.inject({}) { |hash, ary| hash[ary[0].to_sym] = ary[1]; hash }

  set_attributes(options)

  @options = options

  if key
    # If the request key is a URI, then we need to treat it specially,
    # because it rewrites the key.  We could otherwise strip server/port/etc
    # info out in the REST class, but it seemed bad design for the REST
    # class to rewrite the key.

    if key.to_s =~ /^\w+:\// and not Puppet::Util.absolute_path?(key.to_s) # it's a URI
      set_uri_key(key)
    else
      @key = key
    end
  end

  @key = @instance.name if ! @key and @instance
end

Instance Attribute Details

#authenticated ⇒ Object

Returns the value of attribute authenticated.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def authenticated
  @authenticated
end

#ignore_cache ⇒ Object

Returns the value of attribute ignore_cache.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def ignore_cache
  @ignore_cache
end

#ignore_terminus ⇒ Object

Returns the value of attribute ignore_terminus.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def ignore_terminus
  @ignore_terminus
end

#indirection_name ⇒ Object

Returns the value of attribute indirection_name.



17
18
19
 
# File 'lib/puppet/indirector/request.rb', line 17

def indirection_name
  @indirection_name
end

#instance ⇒ Object

Returns the value of attribute instance.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def instance
  @instance
end

#ip ⇒ Object

Returns the value of attribute ip.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def ip
  @ip
end

#key ⇒ Object

Returns the value of attribute key.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def key
  @key
end

#method ⇒ Object

Returns the value of attribute method.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def method
  @method
end

#node ⇒ Object

Returns the value of attribute node.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def node
  @node
end

#options ⇒ Object

Returns the value of attribute options.



13
14
15
 
# File 'lib/puppet/indirector/request.rb', line 13

def options
  @options
end

#port ⇒ Object

Returns the value of attribute port.



15
16
17
 
# File 'lib/puppet/indirector/request.rb', line 15

def port
  @port
end

#protocol ⇒ Object

Returns the value of attribute protocol.



15
16
17
 
# File 'lib/puppet/indirector/request.rb', line 15

def protocol
  @protocol
end

#server ⇒ Object

Returns the value of attribute server.



15
16
17
 
# File 'lib/puppet/indirector/request.rb', line 15

def server
  @server
end

#uri ⇒ Object

Returns the value of attribute uri.



15
16
17
 
# File 'lib/puppet/indirector/request.rb', line 15

def uri
  @uri
end

Instance Method Details

#authenticated? ⇒ Boolean

Is this an authenticated request?

Returns:

  • (Boolean) 


24
25
26
27
 
# File 'lib/puppet/indirector/request.rb', line 24

def authenticated?
  # Double negative, so we just get true or false
  ! ! authenticated
end

#description ⇒ Object 



173
174
175
 
# File 'lib/puppet/indirector/request.rb', line 173

def description
  return(uri ? uri : "/#{indirection_name}/#{key}")
end

#do_request(srv_service = :puppet, default_server = nil, default_port = nil, &block) ⇒ Object 



177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
 
# File 'lib/puppet/indirector/request.rb', line 177

def do_request(srv_service=:puppet, default_server=nil, default_port=nil, &block)
  # We were given a specific server to use, so just use that one.
  # This happens if someone does something like specifying a file
  # source using a puppet:// URI with a specific server.
  return yield(self) if !self.server.nil?

  if Puppet.settings[:use_srv_records]
    Puppet::Network::Resolver.each_srv_record(Puppet.settings[:srv_domain], srv_service) do |srv_server, srv_port|
      begin
        self.server = srv_server
        self.port   = srv_port
        return yield(self)
      rescue SystemCallError => e
        Puppet.warning _("Error connecting to %{srv_server}:%{srv_port}: %{message}") % { srv_server: srv_server, srv_port: srv_port, message: e.message }
      end
    end
  end

  # ... Fall back onto the default server.
  begin
    bound_server = Puppet.lookup(:server)
  rescue
    if primary_server = Puppet.settings[:server_list][0]
      bound_server = primary_server[0]
    else
      bound_server = Puppet.settings[:server]
    end
  end

  begin
    bound_port = Puppet.lookup(:serverport)
  rescue
    if primary_server = Puppet.settings[:server_list][0]
      bound_port = primary_server[1]
    else
      bound_port = Puppet.settings[:masterport]
    end
  end
  self.server = default_server || bound_server
  self.port   = default_port || bound_port

  Puppet.debug "No more servers left, falling back to #{self.server}:#{self.port}" if Puppet.settings[:use_srv_records]

  return yield(self)
end

#encode_params(params) ⇒ Object 



140
141
142
143
144
 
# File 'lib/puppet/indirector/request.rb', line 140

def encode_params(params)
  params.collect do |key, value|
    "#{key}=#{Puppet::Util.uri_query_encode(value.to_s)}"
  end.join("&")
end

#environment ⇒ Object 



29
30
31
32
33
 
# File 'lib/puppet/indirector/request.rb', line 29

def environment
  # If environment has not been set directly, we should use the application's
  # current environment
  @environment ||= Puppet.lookup(:current_environment)
end

#environment=(env) ⇒ Object 



35
36
37
38
39
40
41
42
 
# File 'lib/puppet/indirector/request.rb', line 35

def environment=(env)
  @environment =
  if env.is_a?(Puppet::Node::Environment)
    env
  else
    Puppet.lookup(:environments).get!(env)
  end
end

#expand_into_parameters(data) ⇒ Object 



111
112
113
114
115
116
117
118
119
120
121
122
123
124
 
# File 'lib/puppet/indirector/request.rb', line 111

def expand_into_parameters(data)
  data.inject([]) do |params, key_value|
    key, value = key_value

    expanded_value = case value
                     when Array
                       value.collect { |val| [key, val] }
                     else
                       [key_value]
                     end

    params.concat(expand_primitive_types_into_parameters(expanded_value))
  end
end

#expand_primitive_types_into_parameters(data) ⇒ Object 



126
127
128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/puppet/indirector/request.rb', line 126

def expand_primitive_types_into_parameters(data)
  data.inject([]) do |params, key_value|
    key, value = key_value
    case value
    when nil
      params
    when true, false, String, Symbol, Integer, Float
      params << [key, value]
    else
      raise ArgumentError, _("HTTP REST queries cannot handle values of type '%{klass}'") % { klass: value.class }
    end
  end
end

#ignore_cache? ⇒ Boolean

LAK:NOTE This is a messy interface to the cache, and it’s only used by the Configurer class. I decided it was better to implement it now and refactor later, when we have a better design, than to spend another month coming up with a design now that might not be any better.

Returns:

  • (Boolean) 


49
50
51
 
# File 'lib/puppet/indirector/request.rb', line 49

def ignore_cache?
  ignore_cache
end

#ignore_terminus? ⇒ Boolean

Returns:

  • (Boolean) 


53
54
55
 
# File 'lib/puppet/indirector/request.rb', line 53

def ignore_terminus?
  ignore_terminus
end

#indirection ⇒ Object

Look up the indirection based on the name provided.



87
88
89
 
# File 'lib/puppet/indirector/request.rb', line 87

def indirection
  Puppet::Indirector::Indirection.instance(indirection_name)
end

#initialize_from_hash(hash) ⇒ Object 



146
147
148
149
150
151
152
 
# File 'lib/puppet/indirector/request.rb', line 146

def initialize_from_hash(hash)
  @indirection_name = hash['indirection_name'].to_sym
  @method = hash['method'].to_sym
  @key = hash['key']
  @instance = hash['instance']
  @options = hash['options']
end

#model ⇒ Object

Raises:

  • (ArgumentError) 


95
96
97
98
 
# File 'lib/puppet/indirector/request.rb', line 95

def model
  raise ArgumentError, _("Could not find indirection '%{indirection}'") % { indirection: indirection_name } unless i = indirection
  i.model
end

#plural? ⇒ Boolean

Are we trying to interact with multiple resources, or just one?

Returns:

  • (Boolean) 


101
102
103
 
# File 'lib/puppet/indirector/request.rb', line 101

def plural?
  method == :search
end

#query_string ⇒ Object

Create the query string, if options are present.



106
107
108
109
 
# File 'lib/puppet/indirector/request.rb', line 106

def query_string
  return "" if options.nil? || options.empty?
  encode_params(expand_into_parameters(options.to_a))
end

#remote? ⇒ Boolean

Returns:

  • (Boolean) 


223
224
225
 
# File 'lib/puppet/indirector/request.rb', line 223

def remote?
  self.node or self.ip
end

#to_data_hash ⇒ Object 



154
155
156
157
158
159
160
 
# File 'lib/puppet/indirector/request.rb', line 154

def to_data_hash
  { 'indirection_name' => @indirection_name.to_s,
    'method' => @method.to_s,
    'key' => @key,
    'instance' => @instance,
    'options' => @options }
end

#to_hash ⇒ Object 



162
163
164
165
166
167
168
169
170
171
 
# File 'lib/puppet/indirector/request.rb', line 162

def to_hash
  result = options.dup

  OPTION_ATTRIBUTES.each do |attribute|
    if value = send(attribute)
      result[attribute] = value
    end
  end
  result
end