morpheus-cli 2.10.0 → 2.10.1

data/lib/morpheus/cli/security_group_rules.rb

@@ -9,210 +9,220 @@ require 'morpheus/cli/cli_command'
 class Morpheus::Cli::SecurityGroupRules
   include Morpheus::Cli::CliCommand
 
-	def initialize() 
-		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
-		@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).request_credentials()
-		@security_group_rules_interface = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url).security_group_rules
-		@active_security_group = ::Morpheus::Cli::SecurityGroups.load_security_group_file
-	end
-
-
-	def handle(args) 
-		if @access_token.empty?
-			print_red_alert "Invalid Credentials. Unable to acquire access token. Please verify your credentials and try again."
-			return 1
-		end
-		if args.empty?
-			puts "\nUsage: morpheus security-group-rules [list, add_custom_rule, remove]\n\n"
-			return 
-		end
-
-		case args[0]
-			when 'list'
-				list(args[1..-1])
-			when 'add_custom_rule'
-				add_custom_rule(args[1..-1])
-			when 'add_instance_rule'
-				add_instance_rule(args[1..-1])
-			when 'remove'
-				remove(args[1..-1])
-			else
-				puts "\nUsage: morpheus security-group-rules [list,add_custom_rule,remove]\n\n"
-		end
-	end
-
-	def add_custom_rule(args)
-		usage = <<-EOT
-Usage: morpheus security-group-rules add_custom_rule SOURCE_CIDR PORT_RANGE PROTOCOL [options]
-  SOURCE_CIDR: CIDR to white-list
-  PORT_RANGE: Port value (i.e. 123) or port range (i.e. 1-65535)
-  PROTOCOL: tcp, udp, icmp\n\n'
+  register_subcommands :list, :'add-custom-rule', :'add-instance-rule', :remove
+  set_default_subcommand :list
+  
+  def connect(opts)
+    @api_client = establish_remote_appliance_connection(opts)
+    @security_group_rules_interface = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url).security_group_rules
+    @active_security_group = ::Morpheus::Cli::SecurityGroups.load_security_group_file
+  end
+
+  def handle(args)
+    handle_subcommand(args)
+  end
+
+  def add_custom_rule(args)
+    usage = <<-EOT
+Usage: morpheus #{command_name} add-custom-rule SOURCE_CIDR PORT_RANGE PROTOCOL [options]
+\tSOURCE_CIDR: CIDR to white-list
+\tPORT_RANGE: Port value (i.e. 123) or port range (i.e. 1-65535)
+\tPROTOCOL: tcp, udp, icmp
 EOT
-		options = {}
-		security_group_id = nil 
-		optparse = OptionParser.new do|opts|
-			opts.banner = usage
-			opts.on( '-s', '--secgroup SECGROUP', "Security Group ID (Use will use security as set with 'security-groups use id'" ) do |id|
-				security_group_id = id
-			end
-			build_common_options(opts, options, [])
-		end
-		optparse.parse(args)
-
-		if args.count < 3
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-
-		if security_group_id.nil?
-			security_group_id = @active_security_group[@appliance_name.to_sym]
-		end
-
-		if security_group_id.nil?
-			puts "Security Group ID must be specified with options or set using 'security-groups use id'"
-			exit
-		end
-
-		params = {
-			:rule => {
-				:source => args[0],
-				:portRange => args[1],
-				:protocol => args[2],
-				:customRule => true
-			}
-		}
-
-		begin
-			@security_group_rules_interface.create(security_group_id, params)
-			list([])
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	def add_instance_rule(args)
-		usage = <<-EOT
-Usage: morpheus security-group-rules add_instance_rule SOURCE_CIDR INSTANCE_TYPE_ID [options]
-  SOURCE_CIDR: CIDR to white-list
-  INSTANCE_TYPE_ID: ID of the Instance Type to access
+    options = {}
+    security_group_id = nil
+    optparse = OptionParser.new do|opts|
+      opts.banner = usage
+      opts.on( '-s', '--secgroup SECGROUP', "Security Group ID (Use will use security as set with 'security-groups use id'" ) do |id|
+        security_group_id = id
+      end
+      build_common_options(opts, options, [:json, :dry_run])
+    end
+    optparse.parse!(args)
+
+    if args.count < 3
+      puts optparse.banner
+      exit 1
+    end
+
+    if security_group_id.nil? && @active_security_group
+      security_group_id = @active_security_group[@appliance_name.to_sym]
+    end
+
+    if security_group_id.nil?
+      puts "Security Group ID must be specified with options or set using 'security-groups use id'"
+      exit 1
+    end
+
+    params = {
+      :rule => {
+        :source => args[0],
+        :portRange => args[1],
+        :protocol => args[2],
+        :customRule => true
+      }
+    }
+    connect(options)
+    begin
+      if options[:dry_run]
+        print_dry_run @security_group_rules_interface.dry.create(security_group_id, params)
+        return
+      end
+      json_response = @security_group_rules_interface.create(security_group_id, params)
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+        print "\n"
+        return
+      end
+      list([security_group_id])
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def add_instance_rule(args)
+    usage = <<-EOT
+Usage: morpheus #{command_name} add_instance_rule SOURCE_CIDR INSTANCE_TYPE_ID [options]
+\tSOURCE_CIDR: CIDR to white-list
+\tINSTANCE_TYPE_ID: ID of the Instance Type to access
 EOT
-		
-		options = {}
-		security_group_id = nil 
-		optparse = OptionParser.new do|opts|
-			opts.banner = usage
-			opts.on( '-s', '--secgroup secgroup', "Security Group ID (Use will use security as set with 'security-groups use id'" ) do |id|
-				security_group_id = id
-			end
-			build_common_options(opts, options, [])
-		end
-		optparse.parse(args)
-		if args.count < 2
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-		if security_group_id.nil?
-			security_group_id = @active_security_group[@appliance_name.to_sym]
-		end
-
-		if security_group_id.nil?
-			puts "Security Group ID must be specified with options or set using 'security-groups use id'"
-			exit
-		end
-
-		params = {
-			:rule => {
-				:source => args[0],
-				:instanceTypeId => args[1]
-			}
-		}
-
-		begin
-			@security_group_rules_interface.create(security_group_id, params)
-			list([security_group_id])
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	def list(args)
-		options = {}
-		security_group_id = nil 
-		optparse = OptionParser.new do|opts|
-			opts.banner = "\nUsage: morpheus security-group-rules list [ID]"
-			build_common_options(opts, options, [:json])
-		end
-		security_group_id = args[0].to_i
-		optparse.parse(args)
-
-		if security_group_id.nil?
-			security_group_id = @active_security_group[@appliance_name.to_sym]
-		end
-
-		if security_group_id.nil?
-			puts "Security Group ID must be specified with options or set using 'security-groups use id'"
-			exit
-		end
-
-		begin
-			params = {}
-			json_response = @security_group_rules_interface.get(security_group_id, params)
-			if options[:json]
-				print JSON.pretty_generate(json_response)
-				print "\n"
-				return
-			end
-			rules = json_response['rules']
-			print "\n" ,cyan, bold, "Morpheus Security Group Rules for Security Group ID:#{security_group_id}\n","==================", reset, "\n\n"
-			if rules.empty?
-				puts yellow,"No Security Group Rules currently configured.",reset
-			else
-				rules.each do |rule|
-					print cyan, "=  #{rule['id']} - (CIDR:#{rule['source']}, Port Range:#{rule['portRange']}, Protocol:#{rule['protocol']}, Custom Rule:#{rule['customRule']}, Instance Type:#{rule['instanceTypeId']})\n"
-				end
-			end
-			print reset,"\n\n"
-			
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	def remove(args)
-		
-		options = {}
-		security_group_id = nil 
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus security-group-rules remove ID [options]"
-			opts.on( '-s', '--secgroup secgroup', "Security Group ID (Use will use security as set with 'security-groups use id'" ) do |id|
-				security_group_id = id
-			end
-			build_common_options(opts, options, [])
-		end
-		optparse.parse(args)
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-		if security_group_id.nil?
-			security_group_id = @active_security_group[@appliance_name.to_sym]
-		end
-
-		if security_group_id.nil?
-			puts "Security Group ID must be specified with options or set using 'security-groups use id'"
-			exit
-		end
-
-		begin
-			@security_group_rules_interface.delete(security_group_id, args[0])
-			list([security_group_id])
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
+
+    options = {}
+    security_group_id = nil
+    optparse = OptionParser.new do|opts|
+      opts.banner = usage
+      opts.on( '-s', '--secgroup secgroup', "Security Group ID (Use will use security as set with 'security-groups use id'" ) do |id|
+        security_group_id = id
+      end
+      build_common_options(opts, options, [:json, :dry_run])
+    end
+    optparse.parse!(args)
+    if args.count < 2
+      puts optparse.banner
+      exit 1
+    end
+    if security_group_id.nil? && @active_security_group
+      security_group_id = @active_security_group[@appliance_name.to_sym]
+    end
+
+    if security_group_id.nil?
+      puts "Security Group ID must be specified with options or set using 'security-groups use id'"
+      exit
+    end
+
+    params = {
+      :rule => {
+        :source => args[0],
+        :instanceTypeId => args[1]
+      }
+    }
+    connect(options)
+    begin
+      if options[:dry_run]
+        print_dry_run @security_group_rules_interface.dry.create(security_group_id, params)
+        return
+      end
+      json_response = @security_group_rules_interface.create(security_group_id, params)
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+        print "\n"
+        return
+      end
+      list([security_group_id])
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def list(args)
+    options = {}
+    security_group_id = nil
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[id]")
+      build_common_options(opts, options, [:json, :dry_run])
+    end
+    optparse.parse!(args)
+    security_group_id = args[0]
+    if security_group_id.nil? && @active_security_group
+      security_group_id = @active_security_group[@appliance_name.to_sym]
+    end
+
+    if security_group_id.nil?
+      puts "Security Group ID must be specified with options or set using 'security-groups use id'"
+      exit 1
+    end
+    connect(options)
+    begin
+      params = {}
+      if options[:dry_run]
+        print_dry_run @security_group_rules_interface.dry.get(security_group_id, params)
+        return
+      end
+      json_response = @security_group_rules_interface.get(security_group_id, params)
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+        print "\n"
+        return
+      end
+      rules = json_response['rules']
+      print "\n" ,cyan, bold, "Morpheus Security Group Rules for Security Group ID:#{security_group_id}\n","==================", reset, "\n\n"
+      if rules.empty?
+        puts yellow,"No Security Group Rules currently configured.",reset
+      else
+        rules = rules.sort {|x,y| x["id"] <=> y["id"] }
+        rules.each do |rule|
+          print cyan, "=  #{rule['id']} - (CIDR:#{rule['source']}, Port Range:#{rule['portRange']}, Protocol:#{rule['protocol']}, Custom Rule:#{rule['customRule']}, Instance Type:#{rule['instanceTypeId']})\n"
+        end
+      end
+      print reset,"\n"
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def remove(args)
+    options = {}
+    security_group_id = nil
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[id] [options]")
+      opts.on( '-s', '--secgroup secgroup', "Security Group ID (Use will use security as set with 'security-groups use id'" ) do |id|
+        security_group_id = id
+      end
+      build_common_options(opts, options, [:json, :dry_run])
+    end
+    optparse.parse!(args)
+    if args.count < 1
+      puts optparse
+      exit 1
+    end
+    if security_group_id.nil? && @active_security_group
+      security_group_id = @active_security_group[@appliance_name.to_sym]
+    end
+
+    if security_group_id.nil?
+      puts "Security Group ID must be specified with options or set using 'security-groups use id'"
+      exit
+    end
+    connect(options)
+    begin
+      if options[:dry_run]
+        print_dry_run @security_group_rules_interface.dry.delete(security_group_id, args[0])
+        return
+      end
+      json_response = @security_group_rules_interface.delete(security_group_id, args[0])
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+        print "\n"
+        return
+      end
+      list([security_group_id])
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
 
 end

data/lib/morpheus/cli/security_groups.rb

@@ -9,197 +9,242 @@ require 'morpheus/cli/cli_command'
 class Morpheus::Cli::SecurityGroups
   include Morpheus::Cli::CliCommand
 
-	def initialize() 
-		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
-		@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).request_credentials()
-		@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
-
-
-	def handle(args) 
-		if @access_token.empty?
-			print_red_alert "Invalid Credentials. Unable to acquire access token. Please verify your credentials and try again."
-			return 1
-		end
-		if args.empty?
-			puts "\nUsage: morpheus security-groups [list,get,add,remove,use] [name]\n\n"
-			return 
-		end
-
-		case args[0]
-			when 'list'
-				list(args[1..-1])
-			when 'get'	
-				get(args[1..-1])
-			when 'add'
-				add(args[1..-1])
-			when 'remove'
-				remove(args[1..-1])
-			when 'use'
-				use(args[1..-1])
-			else
-				puts "\nUsage: morpheus security-groups [list,get,add,remove,use] [name]\n\n"
-		end
-	end
-
-	def list(args)
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "\nUsage: morpheus security-groups list"
-			build_common_options(opts, options, [])
-		end
-		optparse.parse(args)
-		begin
-			json_response = @security_groups_interface.list()
-			security_groups = json_response['securityGroups']
-			print "\n" ,cyan, bold, "Morpheus Security Groups\n","==================", reset, "\n\n"
-			if security_groups.empty?
-				puts yellow,"No Security Groups currently configured.",reset
-			else
-				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
-			end
-			print reset,"\n\n"
-			
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	def get(args)
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus security-groups get ID"
-			build_common_options(opts, options, [])
-		end
-		optparse.parse(args)
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-
-		begin
-			json_response = @security_groups_interface.get({id: args[0]})
-			security_group = json_response['securityGroup']
-			print "\n" ,cyan, bold, "Morpheus Security Group\n","==================", reset, "\n\n"
-			if security_group.nil?
-				puts yellow,"Security Group not found by id #{args[0]}",reset
-			else
-				print cyan, "=  #{security_group['id']}: #{security_group['name']} (#{security_group['description']})\n"
-			end
-			print reset,"\n\n"
-			
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	def add(args)
-		options = {}
-		params = {:securityGroup => {:name => args[0]} }
-		optparse = OptionParser.new do|opts|
-			opts.banner = "\nUsage: morpheus security-groups add NAME [options]"
-			opts.on( '-d', '--description Description', "Description of the security group" ) do |description|
-				params[:securityGroup][:description] = description
-			end
-			build_common_options(opts, options, [])
-		end
-		optparse.parse(args)
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-		begin
-			@security_groups_interface.create(params)
-			list([])
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	def remove(args)
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "\nUsage: morpheus security-groups remove ID"
-			build_common_options(opts, options, [])
-		end
-		optparse.parse(args)
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			return
-		end
-		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
-			@security_groups_interface.delete(security_group['id'])
-			list([])
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	def use(args)
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus security-groups use ID"
-			build_common_options(opts, options, [])
-		end
-		optparse.parse(args)
-		if args.length < 1
-			puts "\n#{optparse.banner}\n\n"
-			return
-		end
-		begin
-			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 "Using Security Group #{args[0]}"
-			else
-				puts "Security Group not found"
-			end
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	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
-
-	def self.security_group_file_path
-		home_dir = Dir.home
-		morpheus_dir = File.join(home_dir,".morpheus")
-		if !Dir.exist?(morpheus_dir)
-			Dir.mkdir(morpheus_dir)
-		end
-		return File.join(morpheus_dir,"securitygroup")
-	end
-
-	def self.save_security_group(security_group_map)
-		File.open(security_group_file_path, 'w') {|f| f.write security_group_map.to_yaml } #Store
-	end
+  register_subcommands :list, :get, :add, :remove, :use, :unuse
+  set_default_subcommand :list
+  
+  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
+
+  def handle(args)
+    handle_subcommand(args)
+  end
+
+
+  def list(args)
+    options = {}
+    optparse = 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 "\n" ,cyan, bold, "Morpheus Security Groups\n","==================", reset, "\n\n"
+      if security_groups.empty?
+        puts yellow,"No Security Groups currently configured.",reset
+      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\n# => - current", reset
+        end
+      end
+      print reset,"\n"
+          rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def get(args)
+    options = {}
+    optparse = 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 "\n" ,cyan, bold, "Morpheus Security Group\n","==================", reset, "\n\n"
+      if security_group.nil?
+        puts yellow,"Security Group not found by id #{args[0]}",reset
+      else
+        print cyan, "=  #{security_group['id']}: #{security_group['name']} (#{security_group['description']})\n"
+      end
+      print reset,"\n\n"
+          rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def add(args)
+    params = {:securityGroup => {}}
+    options = {}
+    optparse = 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
+
+  def remove(args)
+    options = {}
+    optparse = 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
+
+  def use(args)
+    options = {}
+    optparse = 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
+
+  def unuse(args)
+    use(args + ['--none'])
+  end
+
+  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
+
+  def self.security_group_file_path
+    home_dir = Dir.home
+    morpheus_dir = File.join(home_dir,".morpheus")
+    if !Dir.exist?(morpheus_dir)
+      Dir.mkdir(morpheus_dir)
+    end
+    return File.join(morpheus_dir,"securitygroup")
+  end
+
+  def self.save_security_group(security_group_map)
+    File.open(security_group_file_path, 'w') {|f| f.write security_group_map.to_yaml } #Store
+  end
 
 end