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
|
module Fog
module AWS
class Compute
class Real
require 'fog/aws/parsers/compute/basic'
# Modify image attributes
#
# ==== Parameters
# * image_id<~String> - Id of machine image to modify
# * attributes<~Hash>:
# * 'Add.Group'<~Array> - One or more groups to grant launch permission to
# * 'Add.UserId'<~Array> - One or more account ids to grant launch permission to
# * 'Description.Value'<String> - New description for image
# * 'ProductCode'<~Array> - One or more product codes to add to image (these can not be removed)
# * 'Remove.Group'<~Array> - One or more groups to revoke launch permission from
# * 'Remove.UserId'<~Array> - One or more account ids to revoke launch permission from
#
# {Amazon API Reference}[http://docs.amazonwebservices.com/AWSEC2/latest/APIReference/ApiReference-query-ModifyImageAttribute.html]
#
def modify_image_attribute(image_id, attributes)
raise ArgumentError.new("image_id is required") unless image_id
params = {}
params.merge!(Fog::AWS.indexed_param('LaunchPermission.Add.%d.Group', attributes['Add.Group'] || []))
params.merge!(Fog::AWS.indexed_param('LaunchPermission.Add.%d.UserId', attributes['Add.UserId'] || []))
params.merge!(Fog::AWS.indexed_param('LaunchPermission.Remove.%d.Group', attributes['Remove.Group'] || []))
params.merge!(Fog::AWS.indexed_param('LaunchPermission.Remove.%d.UserId', attributes['Remove.UserId'] || []))
params.merge!(Fog::AWS.indexed_param('ProductCode', attributes['ProductCode'] || []))
request({
'Action' => 'ModifyImageAttribute',
'ImageId' => image_id,
:idempotent => true,
:parser => Fog::Parsers::AWS::Compute::Basic.new
}.merge!(params))
end
end
class Mock
def modify_image_attribute(image_id, attributes)
raise ArgumentError.new("image_id is required") unless image_id
unless self.data[:images][image_id]
raise Fog::AWS::Compute::NotFound.new("The AMI ID '#{image_id}' does not exist")
end
(attributes['Add.UserId'] || []).each do |user_id|
if image_launch_permissions = self.data[:image_launch_permissions][image_id]
image_launch_permissions[:users].push(user_id)
end
end
(attributes['Remove.UserId'] || []).each do |user_id|
if image_launch_permissions = self.data[:image_launch_permissions][image_id]
image_launch_permissions[:users].delete(user_id)
end
end
response = Excon::Response.new
response.status = 200
response.body = {
'return' => true,
'requestId' => Fog::AWS::Mock.request_id
}
response
end
end
end
end
end
|
