Class: Morpheus::Cli::SecurityGroups

Inherits:
Object
  • Object
show all
Includes:
CliCommand
Defined in:
lib/morpheus/cli/security_groups.rb

Instance Attribute Summary

Attributes included from CliCommand

#no_prompt

Class Method Summary collapse

Instance Method Summary collapse

Methods included from CliCommand

#build_common_options, #build_option_type_options, #command_name, #default_subcommand, #establish_remote_appliance_connection, #full_command_usage, #handle_subcommand, included, #interactive?, #my_help_command, #my_terminal, #my_terminal=, #parse_id_list, #parse_list_options, #parse_list_subtitles, #print, #print_error, #puts, #puts_error, #raise_command_error, #run_command_for_each_arg, #subcommand_aliases, #subcommand_usage, #subcommands, #usage, #verify_access_token!

Class Method Details

.load_security_group_file ⇒ Object 



231
232
233
234
235
236
237
238
 
# File 'lib/morpheus/cli/security_groups.rb', line 231

def self.load_security_group_file
  remote_file = security_group_file_path
  if File.exist? remote_file
    return YAML.load_file(remote_file)
  else
    {}
  end
end

.save_security_group(new_config) ⇒ Object 



244
245
246
247
248
249
250
251
252
 
# File 'lib/morpheus/cli/security_groups.rb', line 244

def self.save_security_group(new_config)
  fn = security_group_file_path
  if !Dir.exists?(File.dirname(fn))
    FileUtils.mkdir_p(File.dirname(fn))
  end
  File.open(fn, 'w') {|f| f.write new_config.to_yaml } #Store
  FileUtils.chmod(0600, fn)
  new_config
end

.security_group_file_path ⇒ Object 



240
241
242
 
# File 'lib/morpheus/cli/security_groups.rb', line 240

def self.security_group_file_path
  File.join(Morpheus::Cli.home_directory,"securitygroup")
end

Instance Method Details

#add(args) ⇒ Object 



110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
 
# File 'lib/morpheus/cli/security_groups.rb', line 110

def add(args)
  params = {:securityGroup => {}}
  options = {}
  optparse = Morpheus::Cli::OptionParser.new do |opts|
    opts.banner = subcommand_usage("[name] [options]")
    opts.on( '-d', '--description Description', "Description of the security group" ) do |description|
      params[:securityGroup][:description] = description
    end
    build_common_options(opts, options, [:json, :dry_run])
  end
  optparse.parse!(args)
  if args.count < 1
    puts optparse
    exit 1
  end
  params[:securityGroup][:name] = args[0]
  connect(options)
  begin
    if options[:dry_run]
      print_dry_run @security_groups_interface.dry.create(params)
      return
    end
    json_response = @security_groups_interface.create(params)
    if options[:json]
      print JSON.pretty_generate(json_response)
      print "\n"
      return
    end
    list([])
  rescue RestClient::Exception => e
    print_rest_exception(e, options)
    exit 1
  end
end

#connect(opts) ⇒ Object 



15
16
17
18
19
 
# File 'lib/morpheus/cli/security_groups.rb', line 15

def connect(opts)
  @api_client = establish_remote_appliance_connection(opts)
  @security_groups_interface = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url).security_groups
  @active_security_group = ::Morpheus::Cli::SecurityGroups.load_security_group_file
end

#get(args) ⇒ Object 



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
106
107
108
 
# File 'lib/morpheus/cli/security_groups.rb', line 70

def get(args)
  options = {}
  optparse = Morpheus::Cli::OptionParser.new do |opts|
    opts.banner = subcommand_usage("[id]")
    build_common_options(opts, options, [:json, :dry_run])
  end
  optparse.parse!(args)
  if args.count < 1
    puts optparse
    exit 1
  end
  connect(options)
  begin
    if options[:dry_run]
      print_dry_run @security_groups_interface.dry.get({id: args[0]})
      return
    end
    json_response = @security_groups_interface.get({id: args[0]})
    if options[:json]
      print JSON.pretty_generate(json_response)
      print "\n"
      return
    end
    security_group = json_response['securityGroup']
    print_h1 "Morpheus Security Group"
    print cyan
    description_cols = {
      "ID" => 'id',
      "Name" => 'name',
      "Description" => 'description',
      #"Account" => lambda {|it| it['account'] ? it['account']['name'] : '' },
    }
    print_description_list(description_cols, security_group)
    print reset,"\n"
  rescue RestClient::Exception => e
    print_rest_exception(e, options)
    exit 1
  end
end

#handle(args) ⇒ Object 



21
22
23
 
# File 'lib/morpheus/cli/security_groups.rb', line 21

def handle(args)
  handle_subcommand(args)
end

#list(args) ⇒ Object 



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
 
# File 'lib/morpheus/cli/security_groups.rb', line 26

def list(args)
  options = {}
  optparse = Morpheus::Cli::OptionParser.new do |opts|
    opts.banner = subcommand_usage()
    build_common_options(opts, options, [:json, :dry_run])
  end
  optparse.parse!(args)
  connect(options)
  begin
    params = {}
    if options[:dry_run]
      print_dry_run @security_groups_interface.dry.list(params)
      return
    end
    json_response = @security_groups_interface.list(params)
    if options[:json]
      print JSON.pretty_generate(json_response)
      print "\n"
      return
    end
    security_groups = json_response['securityGroups']
    print_h1 "Morpheus Security Groups"
    if security_groups.empty?
      print yellow,"No Security Groups currently configured.",reset,"\n"
    else
      active_id = @active_security_group[@appliance_name.to_sym]
      security_groups.each do |security_group|
        if @active_security_group[@appliance_name.to_sym] == security_group['id']
          print cyan, "=> #{security_group['id']}: #{security_group['name']} (#{security_group['description']})\n"
        else
          print cyan, "   #{security_group['id']}: #{security_group['name']} (#{security_group['description']})\n"
        end
      end
      if active_id
        print cyan, "\n# => - current", reset, "\n"
      end
    end
    print reset,"\n"
        rescue RestClient::Exception => e
    print_rest_exception(e, options)
    exit 1
  end
end

#remove(args) ⇒ Object 



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
 
# File 'lib/morpheus/cli/security_groups.rb', line 145

def remove(args)
  options = {}
  optparse = Morpheus::Cli::OptionParser.new do |opts|
    opts.banner = subcommand_usage("[id]")
    build_common_options(opts, options, [:json, :dry_run])
  end
  optparse.parse!(args)
  if args.count < 1
    puts optparse
    return
  end
  connect(options)
  begin
    json_response = @security_groups_interface.get({id: args[0]})
    security_group = json_response['securityGroup']
    if security_group.nil?
      puts "Security Group not found by id #{args[0]}"
      return
    end
    if options[:dry_run]
      print_dry_run @security_groups_interface.dry.delete(security_group['id'])
      return
    end
    json_response = @security_groups_interface.delete(security_group['id'])
    if options[:json]
      print JSON.pretty_generate(json_response)
      print "\n"
      return
    end
    list([])
  rescue RestClient::Exception => e
    print_rest_exception(e, options)
    exit 1
  end
end

#unuse(args) ⇒ Object 



227
228
229
 
# File 'lib/morpheus/cli/security_groups.rb', line 227

def unuse(args)
  use(args + ['--none'])
end

#use(args) ⇒ Object 



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
222
223
224
225
 
# File 'lib/morpheus/cli/security_groups.rb', line 181

def use(args)
  options = {}
  optparse = Morpheus::Cli::OptionParser.new do |opts|
    opts.banner = subcommand_usage("[id] [--none]")
    opts.on('--none','--none', "Do not use an active group.") do |json|
      options[:unuse] = true
    end
    build_common_options(opts, options, [])
  end
  optparse.parse!(args)
  if args.length < 1 && !options[:unuse]
    puts optparse
    return
  end
  connect(options)
  begin

    if options[:unuse]
      if @active_security_group[@appliance_name.to_sym] 
        @active_security_group.delete(@appliance_name.to_sym)
      end
      ::Morpheus::Cli::SecurityGroups.save_security_group(@active_security_group)
      unless options[:quiet]
        print cyan
        puts "Switched to no active security group."
        print reset
      end
      print reset
      return # exit 0
    end

    json_response = @security_groups_interface.get({id: args[0]})
    security_group = json_response['securityGroup']
    if !security_group.nil?
      @active_security_group[@appliance_name.to_sym] = security_group['id']
      ::Morpheus::Cli::SecurityGroups.save_security_group(@active_security_group)
      puts cyan, "Using Security Group #{args[0]}", reset
    else
      puts red, "Security Group not found", reset
    end
  rescue RestClient::Exception => e
    print_rest_exception(e, options)
    exit 1
  end
end