Class: Chef::UserV1

Inherits:
Object
  • Object
show all
Includes:
Mixin::ApiVersionRequestHandling, Mixin::FromFile, Mixin::ParamsValidate
Defined in:
lib/chef/user_v1.rb

Constant Summary collapse

SUPPORTED_API_VERSIONS =
[0, 1]

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Mixin::ApiVersionRequestHandling

#reregister_only_v0_supported_error_msg, #server_client_api_version_intersection

Methods included from Mixin::ParamsValidate

#lazy, #set_or_return, #validate

Methods included from Mixin::FromFile

#class_from_file, #from_file

Constructor Details

#initializeUserV1


44
45
46
47
48
49
50
51
52
53
54
55
# File 'lib/chef/user_v1.rb', line 44

def initialize
  @username = nil
  @display_name = nil
  @first_name = nil
  @middle_name = nil
  @last_name = nil
  @email = nil
  @password = nil
  @public_key = nil
  @private_key = nil
  @create_key = nil
end

Class Method Details

.from_hash(user_hash) ⇒ Object

Class Methods


258
259
260
261
262
263
264
265
266
267
268
269
270
271
# File 'lib/chef/user_v1.rb', line 258

def self.from_hash(user_hash)
  user = Chef::UserV1.new
  user.username user_hash["username"]
  user.display_name user_hash["display_name"] if user_hash.key?("display_name")
  user.first_name user_hash["first_name"] if user_hash.key?("first_name")
  user.middle_name user_hash["middle_name"] if user_hash.key?("middle_name")
  user.last_name user_hash["last_name"] if user_hash.key?("last_name")
  user.email user_hash["email"] if user_hash.key?("email")
  user.password user_hash["password"] if user_hash.key?("password")
  user.public_key user_hash["public_key"] if user_hash.key?("public_key")
  user.private_key user_hash["private_key"] if user_hash.key?("private_key")
  user.create_key user_hash["create_key"] if user_hash.key?("create_key")
  user
end

.from_json(json) ⇒ Object


273
274
275
# File 'lib/chef/user_v1.rb', line 273

def self.from_json(json)
  Chef::UserV1.from_hash(Chef::JSONCompat.from_json(json))
end

.list(inflate = false) ⇒ Object


277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
# File 'lib/chef/user_v1.rb', line 277

def self.list(inflate = false)
  response = Chef::ServerAPI.new(Chef::Config[:chef_server_url]).get("users")
  users = if response.is_a?(Array)
            # EC 11 / CS 12 V0, V1
            #   GET /organizations/<org>/users
            transform_list_response(response)
          else
            # OSC 11
            #  GET /users
            # EC 11 / CS 12 V0, V1
            #  GET /users
            response # OSC
          end

  if inflate
    users.inject({}) do |user_map, (name, _url)|
      user_map[name] = Chef::UserV1.load(name)
      user_map
    end
  else
    users
  end
end

.load(username) ⇒ Object


301
302
303
304
305
# File 'lib/chef/user_v1.rb', line 301

def self.load(username)
  # will default to the current API version (Chef::Authenticator::DEFAULT_SERVER_API_VERSION)
  response = Chef::ServerAPI.new(Chef::Config[:chef_server_url]).get("users/#{username}")
  Chef::UserV1.from_hash(response)
end

Instance Method Details

#chef_root_rest_v0Object


57
58
59
# File 'lib/chef/user_v1.rb', line 57

def chef_root_rest_v0
  @chef_root_rest_v0 ||= Chef::ServerAPI.new(Chef::Config[:chef_server_root], { :api_version => "0" })
end

#chef_root_rest_v1Object


61
62
63
# File 'lib/chef/user_v1.rb', line 61

def chef_root_rest_v1
  @chef_root_rest_v1 ||= Chef::ServerAPI.new(Chef::Config[:chef_server_root], { :api_version => "1" })
end

#createObject


140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
# File 'lib/chef/user_v1.rb', line 140

def create
  # try v1, fail back to v0 if v1 not supported
  begin
    payload = {
      :username => @username,
      :display_name => @display_name,
      :first_name => @first_name,
      :last_name => @last_name,
      :email => @email,
      :password => @password,
    }
    payload[:public_key] = @public_key unless @public_key.nil?
    payload[:create_key] = @create_key unless @create_key.nil?
    payload[:middle_name] = @middle_name unless @middle_name.nil?
    raise Chef::Exceptions::InvalidUserAttribute, "You cannot set both public_key and create_key for create." if !@create_key.nil? && !@public_key.nil?
    new_user = chef_root_rest_v1.post("users", payload)

    # get the private_key out of the chef_key hash if it exists
    if new_user["chef_key"]
      if new_user["chef_key"]["private_key"]
        new_user["private_key"] = new_user["chef_key"]["private_key"]
      end
      new_user["public_key"] = new_user["chef_key"]["public_key"]
      new_user.delete("chef_key")
    end
  rescue Net::HTTPServerException => e
    # rescue API V0 if 406 and the server supports V0
    supported_versions = server_client_api_version_intersection(e, SUPPORTED_API_VERSIONS)
    raise e unless supported_versions && supported_versions.include?(0)
    payload = {
      :username => @username,
      :display_name => @display_name,
      :first_name => @first_name,
      :last_name => @last_name,
      :email => @email,
      :password => @password,
    }
    payload[:middle_name] = @middle_name unless @middle_name.nil?
    payload[:public_key] = @public_key unless @public_key.nil?
    # under API V0, the server will create a key pair if public_key isn't passed
    new_user = chef_root_rest_v0.post("users", payload)
  end

  Chef::UserV1.from_hash(to_hash.merge(new_user))
end

#create_key(arg = nil) ⇒ Object


95
96
97
98
# File 'lib/chef/user_v1.rb', line 95

def create_key(arg = nil)
  set_or_return(:create_key, arg,
                :kind_of => [TrueClass, FalseClass])
end

#destroyObject


135
136
137
138
# File 'lib/chef/user_v1.rb', line 135

def destroy
  # will default to the current API version (Chef::Authenticator::DEFAULT_SERVER_API_VERSION)
  Chef::ServerAPI.new(Chef::Config[:chef_server_url]).delete("users/#{@username}")
end

#display_name(arg = nil) ⇒ Object


70
71
72
73
# File 'lib/chef/user_v1.rb', line 70

def display_name(arg = nil)
  set_or_return(:display_name,
                arg, :kind_of => String)
end

#email(arg = nil) ⇒ Object


90
91
92
93
# File 'lib/chef/user_v1.rb', line 90

def email(arg = nil)
  set_or_return(:email,
                arg, :kind_of => String)
end

#first_name(arg = nil) ⇒ Object


75
76
77
78
# File 'lib/chef/user_v1.rb', line 75

def first_name(arg = nil)
  set_or_return(:first_name,
                arg, :kind_of => String)
end

#last_name(arg = nil) ⇒ Object


85
86
87
88
# File 'lib/chef/user_v1.rb', line 85

def last_name(arg = nil)
  set_or_return(:last_name,
                arg, :kind_of => String)
end

#middle_name(arg = nil) ⇒ Object


80
81
82
83
# File 'lib/chef/user_v1.rb', line 80

def middle_name(arg = nil)
  set_or_return(:middle_name,
                arg, :kind_of => String)
end

#password(arg = nil) ⇒ Object


110
111
112
113
# File 'lib/chef/user_v1.rb', line 110

def password(arg = nil)
  set_or_return(:password,
                arg, :kind_of => String)
end

#private_key(arg = nil) ⇒ Object


105
106
107
108
# File 'lib/chef/user_v1.rb', line 105

def private_key(arg = nil)
  set_or_return(:private_key,
                arg, :kind_of => String)
end

#public_key(arg = nil) ⇒ Object


100
101
102
103
# File 'lib/chef/user_v1.rb', line 100

def public_key(arg = nil)
  set_or_return(:public_key,
                arg, :kind_of => String)
end

#reregisterObject

Note: remove after API v0 no longer supported by client (and knife command).


230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
# File 'lib/chef/user_v1.rb', line 230

def reregister
  begin
    payload = to_hash.merge({ "private_key" => true })
    reregistered_self = chef_root_rest_v0.put("users/#{username}", payload)
    private_key(reregistered_self["private_key"])
  # only V0 supported for reregister
  rescue Net::HTTPServerException => e
    # if there was a 406 related to versioning, give error explaining that
    # only API version 0 is supported for reregister command
    if e.response.code == "406" && e.response["x-ops-server-api-version"]
      version_header = Chef::JSONCompat.from_json(e.response["x-ops-server-api-version"])
      min_version = version_header["min_version"]
      max_version = version_header["max_version"]
      error_msg = reregister_only_v0_supported_error_msg(max_version, min_version)
      raise Chef::Exceptions::OnlyApiVersion0SupportedForAction.new(error_msg)
    else
      raise e
    end
  end
  self
end

#save(new_key = false) ⇒ Object


219
220
221
222
223
224
225
226
227
# File 'lib/chef/user_v1.rb', line 219

def save(new_key = false)
  create
rescue Net::HTTPServerException => e
  if e.response.code == "409"
    update(new_key)
  else
    raise e
  end
end

#to_hashObject


115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
# File 'lib/chef/user_v1.rb', line 115

def to_hash
  result = {
    "username" => @username,
  }
  result["display_name"] = @display_name unless @display_name.nil?
  result["first_name"] = @first_name unless @first_name.nil?
  result["middle_name"] = @middle_name unless @middle_name.nil?
  result["last_name"] = @last_name unless @last_name.nil?
  result["email"] = @email unless @email.nil?
  result["password"] = @password unless @password.nil?
  result["public_key"] = @public_key unless @public_key.nil?
  result["private_key"] = @private_key unless @private_key.nil?
  result["create_key"] = @create_key unless @create_key.nil?
  result
end

#to_json(*a) ⇒ Object


131
132
133
# File 'lib/chef/user_v1.rb', line 131

def to_json(*a)
  Chef::JSONCompat.to_json(to_hash, *a)
end

#to_sObject


252
253
254
# File 'lib/chef/user_v1.rb', line 252

def to_s
  "user[#{@username}]"
end

#update(new_key = false) ⇒ Object


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
# File 'lib/chef/user_v1.rb', line 186

def update(new_key = false)
  begin
    payload = { :username => username }
    payload[:display_name] = display_name unless display_name.nil?
    payload[:first_name] = first_name unless first_name.nil?
    payload[:middle_name] = middle_name unless middle_name.nil?
    payload[:last_name] = last_name unless last_name.nil?
    payload[:email] = email unless email.nil?
    payload[:password] = password unless password.nil?

    # API V1 will fail if these key fields are defined, and try V0 below if relevant 400 is returned
    payload[:public_key] = public_key unless public_key.nil?
    payload[:private_key] = new_key if new_key

    updated_user = chef_root_rest_v1.put("users/#{username}", payload)
  rescue Net::HTTPServerException => e
    if e.response.code == "400"
      # if a 400 is returned but the error message matches the error related to private / public key fields, try V0
      # else, raise the 400
      error = Chef::JSONCompat.from_json(e.response.body)["error"].first
      error_match = /Since Server API v1, all keys must be updated via the keys endpoint/.match(error)
      if error_match.nil?
        raise e
      end
    else # for other types of errors, test for API versioning errors right away
      supported_versions = server_client_api_version_intersection(e, SUPPORTED_API_VERSIONS)
      raise e unless supported_versions && supported_versions.include?(0)
    end
    updated_user = chef_root_rest_v0.put("users/#{username}", payload)
  end
  Chef::UserV1.from_hash(to_hash.merge(updated_user))
end

#username(arg = nil) ⇒ Object


65
66
67
68
# File 'lib/chef/user_v1.rb', line 65

def username(arg = nil)
  set_or_return(:username, arg,
                :regex => /^[a-z0-9\-_]+$/)
end