morpheus-cli 2.10.0 → 2.10.1

data/lib/morpheus/cli/license.rb

@@ -5,112 +5,94 @@ require 'table_print'
 require 'morpheus/cli/cli_command'
 
 class Morpheus::Cli::License
-	include Morpheus::Cli::CliCommand
+  include Morpheus::Cli::CliCommand
 
-	def initialize() 
-		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance	
-	end
+  register_subcommands :get, :apply
+  alias_subcommand :details, :get
 
-	def connect(opts)
-		if opts[:remote]
-			@appliance_url = opts[:remote]
-			@appliance_name = opts[:remote]
-			@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).request_credentials(opts)
-		else
-			@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).request_credentials(opts)
-		end
-		@api_client = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url)		
-		@license_interface = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url).license
-		
-		if @access_token.empty?
-			print_red_alert "Invalid Credentials. Unable to acquire access token. Please verify your credentials and try again."
-			exit 1
-		end
-	end
+  def initialize() 
+    # @appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
+  end
 
+  def connect(opts)
+    @api_client = establish_remote_appliance_connection(opts)
+    @api_client = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url)		
+    @license_interface = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url).license
+  end
 
-	def handle(args) 
-		if args.empty?
-			puts "\nUsage: morpheus license [details, apply]\n\n"
-			return 
-		end
+  def handle(args)
+    handle_subcommand(args)
+  end
 
-		case args[0]
-			when 'apply'
-				apply(args[1..-1])	
-			when 'details'
-				details(args[1..-1])
-			else
-				puts "\nUsage: morpheus license [details, apply]\n\n"
-				exit 127
-		end
-	end
+  def get(args)
+    options = {}
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage()
+      build_common_options(opts, options, [:json, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    connect(options)
+    begin
+      if options[:dry_run]
+        print_dry_run @license_interface.dry.get()
+        return
+      end
+      license = @license_interface.get()
+      if options[:json]
+        puts JSON.pretty_generate(license)
+      else
+        if license['license'].nil?
+          puts "No License Currently Applied to the appliance."
+          exit 1
+        else
+          print "\n", cyan, "License\n=======\n"
+          max_memory = Filesize.from("#{license['license']['maxMemory']} B").pretty
+          max_storage = Filesize.from("#{license['license']['maxStorage']} B").pretty
+          used_memory = Filesize.from("#{license['usedMemory']} B").pretty
+          puts "Account: #{license['license']['accountName']}"
+          puts "Start Date: #{license['license']['startDate']}"
+          puts "End Date: #{license['license']['endDate']}"
+          puts "Memory: #{used_memory} / #{max_memory}"
+          puts "Max Storage: #{max_storage}"
+        end
+        print reset,"\n"
+      end
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      return false
+    end
+  end
 
-
-	def details(args)
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus license details"
-			build_common_options(opts, options, [:json, :remote])
-		end
-		optparse.parse(args)
-		connect(options)
-		begin
-			license = @license_interface.get()
-			
-			if options[:json]
-					puts JSON.pretty_generate(license)
-			else
-				if license['license'].nil?
-					puts "No License Currently Applied to the appliance."
-					exit 1
-				else
-					print "\n", cyan, "License\n=======\n"
-					max_memory = Filesize.from("#{license['license']['maxMemory']} B").pretty
-					max_storage = Filesize.from("#{license['license']['maxStorage']} B").pretty
-					used_memory = Filesize.from("#{license['usedMemory']} B").pretty
-					puts "Account: #{license['license']['accountName']}"
-					puts "Start Date: #{license['license']['startDate']}"
-					puts "End Date: #{license['license']['endDate']}"
-					puts "Memory: #{used_memory} / #{max_memory}"
-					puts "Max Storage: #{max_storage}"
-				end
-				print reset,"\n\n"
-			end
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-
-	def apply(args)
-		key = args[0]
-		options = {}
-		account_name = nil
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus license apply [key]"
-			build_common_options(opts, options, [:json, :remote])
-		end
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-		optparse.parse(args)
-
-		connect(options)
-		
-		begin
-			license_results = @license_interface.apply(key)
-
-			if options[:json]
-					puts JSON.pretty_generate(license_results)
-			else
-				puts "License applied successfully!"
-			end
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
+  def apply(args)
+    options = {}
+    account_name = nil
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[key]")
+      build_common_options(opts, options, [:json, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    connect(options)
+    begin
+      if args[0]
+        key = args[0]
+      else
+        v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldName' => 'licenseKey', 'fieldLabel' => 'License Key', 'type' => 'text', 'required' => true}], options[:options])
+        instance_name = v_prompt['licenseKey'] || ''
+      end
+      if options[:dry_run]
+        print_dry_run @license_interface.dry.apply(key)
+        return
+      end
+      license_results = @license_interface.apply(key)
+      if options[:json]
+        puts JSON.pretty_generate(license_results)
+      else
+        puts "License applied successfully!"
+      end
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      return false
+    end
+  end
 
 end

data/lib/morpheus/cli/load_balancers.rb

@@ -6,331 +6,345 @@ require 'table_print'
 require 'morpheus/cli/cli_command'
 
 class Morpheus::Cli::LoadBalancers
-	include Morpheus::Cli::CliCommand
+  include Morpheus::Cli::CliCommand
 
-	def initialize() 
-		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance	
-	end
+  register_subcommands :list, :get, :add, :update, :remove, {:types => :lb_types}
+  alias_subcommand :details, :get
 
-	def connect(opts)
-		if opts[:remote]
-			@appliance_url = opts[:remote]
-			@appliance_name = opts[:remote]
-			@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).request_credentials(opts)
-		else
-			@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).request_credentials(opts)
-		end
-		@api_client = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url)		
-		@load_balancers_interface = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url).load_balancers
-		
-		if @access_token.empty?
-			print_red_alert "Invalid Credentials. Unable to acquire access token. Please verify your credentials and try again."
-			exit 1
-		end
-	end
+  def initialize() 
+    # @appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
+  end
 
+  def connect(opts)
+    @api_client = establish_remote_appliance_connection(opts)
+    @load_balancers_interface = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url).load_balancers
+  end
 
-	def handle(args) 
-		if args.empty?
-			puts "\nUsage: morpheus load-balancers [list,add, update,remove, details, lb-types]\n\n"
-			return 
-		end
 
-		case args[0]
-			when 'list'
-				list(args[1..-1])
-			when 'add'
-				add(args[1..-1])
-			when 'update'
-				# update(args[1..-1])	
-			when 'details'
-				details(args[1..-1])
-			when 'remove'
-				remove(args[1..-1])
-			when 'lb-types'
-				lb_types(args[1..-1])
-			else
-				puts "\nUsage: morpheus load-balancers [list,add, update,remove, details, lb-types]\n\n"
-				exit 127
-		end
-	end
+  def handle(args)
+    handle_subcommand(args)
+  end
 
-	def list(args)
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus load-balancers list [-s] [-o] [-m]"
-			build_common_options(opts, options, [:list, :json, :remote])
-		end
-		optparse.parse(args)
-		connect(options)
-		begin
-			params = {}
-			[:phrase, :offset, :max, :sort, :direction].each do |k|
-				params[k] = options[k] unless options[k].nil?
-			end
-			json_response = @load_balancers_interface.get(params)
-			if options[:json]
-					print JSON.pretty_generate(json_response)
-			else
-				lbs = json_response['loadBalancers']
-				print "\n" ,cyan, bold, "Morpheus Load Balancers\n","==================", reset, "\n\n"
-				if lbs.empty?
-					puts yellow,"No load balancers currently configured.",reset
-				else
-					print cyan
-					lb_table_data = lbs.collect do |lb|
-						{name: lb['name'], id: lb['id'], type: lb['type']['name']}
-					end
-					tp lb_table_data, :id, :name, :type
-				end
-				print reset,"\n\n"
-			end
-			
-			
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
+  def list(args)
+    options = {}
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage()
+      build_common_options(opts, options, [:list, :json, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    connect(options)
+    begin
+      params = {}
+      [:phrase, :offset, :max, :sort, :direction].each do |k|
+        params[k] = options[k] unless options[k].nil?
+      end
+      if options[:dry_run]
+        print_dry_run @load_balancers_interface.dry.get(params)
+        return
+      end
+      json_response = @load_balancers_interface.get(params)
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+      else
+        lbs = json_response['loadBalancers']
+        print "\n" ,cyan, bold, "Morpheus Load Balancers\n","==================", reset, "\n\n"
+        if lbs.empty?
+          puts yellow,"No load balancers currently configured.",reset
+        else
+          print cyan
+          lb_table_data = lbs.collect do |lb|
+            {name: lb['name'], id: lb['id'], type: lb['type']['name']}
+          end
+          tp lb_table_data, :id, :name, :type
+        end
+        print reset,"\n"
+      end
+                rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
 
-	def details(args)
-				lb_name = args[0]
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus load-balancers details [name]"
-			build_common_options(opts, options, [:json, :remote])
-		end
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-		optparse.parse(args)
-		connect(options)
-		begin
-			lb = find_lb_by_name(lb_name)
+  def get(args)
+    options = {}
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[name]")
+      build_common_options(opts, options, [:json, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    if args.count < 1
+      puts optparse
+      exit 1
+    end
+    lb_name = args[0]
+    connect(options)
+    begin
+      if options[:dry_run]
+        if lb_name.to_s =~ /\A\d{1,}\Z/
+          print_dry_run @load_balancers_interface.dry.get(lb_name.to_i)
+        else
+          print_dry_run @load_balancers_interface.dry.get({name:lb_name})
+        end
+        return
+      end
+      lb = find_lb_by_name_or_id(lb_name)
+      exit 1 if lb.nil?
+      lb_type = load_balancer_type_for_name_or_id(lb['type']['code'])
+      if options[:json]
+        puts JSON.pretty_generate({loadBalancer: lb})
+      else
+        print "\n", cyan, "Lb #{lb['name']} - #{lb['type']['name']}\n\n"
+        # lb_type['optionTypes'].sort { |x,y| x['displayOrder'].to_i <=> y['displayOrder'].to_i }.each do |optionType|
+        # 	puts "  #{optionType['fieldLabel']} : " + (optionType['type'] == 'password' ? "#{task['taskOptions'][optionType['fieldName']] ? '************' : ''}" : "#{task['taskOptions'][optionType['fieldName']] || optionType['defaultValue']}")
+        # end
+        print reset,"\n\n"
+      end
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
 
-			exit 1 if lb.nil?
-			lb_type = find_lb_type_by_name(lb['type']['name'])
-			if options[:json]
-					puts JSON.pretty_generate({task:task})
-			else
-				print "\n", cyan, "Lb #{lb['name']} - #{lb['type']['name']}\n\n"
-				lb_type['optionTypes'].sort { |x,y| x['displayOrder'].to_i <=> y['displayOrder'].to_i }.each do |optionType|
-					puts "  #{optionType['fieldLabel']} : " + (optionType['type'] == 'password' ? "#{task['taskOptions'][optionType['fieldName']] ? '************' : ''}" : "#{task['taskOptions'][optionType['fieldName']] || optionType['defaultValue']}")
-				end
-				print reset,"\n\n"
-			end
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
+  def update(args)
+    lb_name = args[0]
+    options = {}
+    account_name = nil
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[name] [options]")
+      build_common_options(opts, options, [:options, :json, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    if args.count < 1
+      puts optparse
+      exit 1
+    end
+    connect(options)
+    begin
 
-	def update(args)
-		lb_name = args[0]
-		options = {}
-		account_name = nil
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus tasks update [task] [options]"
-			build_common_options(opts, options, [:options, :json, :remote])
-		end
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-		optparse.parse(args)
+      lb = find_lb_by_name_or_id(lb_name)
+      exit 1 if lb.nil?
+      lb_type = load_balancer_type_for_name_or_id(lb['type']['code'])
 
-		connect(options)
-		
-		begin
+      #params = Morpheus::Cli::OptionTypes.prompt(add_load_balancer_option_types, options[:options], @api_client, options[:params]) # options[:params] is mysterious
+      params = options[:options] || {}
 
+      if params.empty?
+        puts optparse
+        option_lines = update_task_option_types(lb_type).collect {|it| "\t-O #{it['fieldContext'] ? (it['fieldContext'] + '.') : ''}#{it['fieldName']}=\"value\"" }.join("\n")
+        puts "\nAvailable Options:\n#{option_lines}\n\n"
+        exit 1
+      end
 
-			task = find_task_by_name_or_code_or_id(lb_name)
-			exit 1 if task.nil?
-			lb_type = find_lb_type_by_name(task['type']['name'])
+      # todo: fix this...
+      #puts "parsed params is : #{params.inspect}"
+      lb_keys = ['name']
+      changes_payload = (params.select {|k,v| task_keys.include?(k) })
+      task_payload = task
+      if changes_payload
+        task_payload.merge!(changes_payload)
+      end
+      puts params
+      if params['taskOptions']
+        task_payload['taskOptions'].merge!(params['taskOptions'])
+      end
+      payload = {task: task_payload}
+      if options[:dry_run]
+        print_dry_run @load_balancers_interface.dry.update(task['id'], payload)
+        return
+      end
+      response = @load_balancers_interface.update(task['id'], payload)
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+        if !response['success']
+          exit 1
+        end
+      else
+        print "\n", cyan, "Task #{response['task']['name']} updated", reset, "\n\n"
+      end
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
 
-			#params = Morpheus::Cli::OptionTypes.prompt(add_user_option_types, options[:options], @api_client, options[:params]) # options[:params] is mysterious
-			params = options[:options] || {}
 
-			if params.empty?
-				puts "\n#{optparse.banner}\n\n"
-				option_lines = update_task_option_types(lb_type).collect {|it| "\t-O #{it['fieldContext'] ? (it['fieldContext'] + '.') : ''}#{it['fieldName']}=\"value\"" }.join("\n")
-				puts "\nAvailable Options:\n#{option_lines}\n\n"
-				exit 1
-			end
+  def lb_types(args)
+    options = {}
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage()
+      build_common_options(opts, options, [:json, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    connect(options)
+    begin
+      if options[:dry_run]
+        print_dry_run @load_balancers_interface.dry.load_balancer_types()
+        return
+      end
+      json_response = @load_balancers_interface.load_balancer_types()
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+      else
+        lb_types = json_response['loadBalancerTypes']
+        print "\n" ,cyan, bold, "Morpheus Load Balancer Types\n","============================", reset, "\n\n"
+        if lb_types.nil? || lb_types.empty?
+          puts yellow,"No lb types currently exist on this appliance. This could be a seed issue.",reset
+        else
+          print cyan
+          lb_table_data = lb_types.collect do |lb_type|
+            {name: lb_type['name'], id: lb_type['id'], code: lb_type['code']}
+          end
+          tp lb_table_data, :id, :name, :code
+        end
 
-			#puts "parsed params is : #{params.inspect}"
-			task_keys = ['name']
-			changes_payload = (params.select {|k,v| task_keys.include?(k) })
-			task_payload = task
-			if changes_payload
-				task_payload.merge!(changes_payload)
-			end
-			puts params
-			if params['taskOptions']
-				task_payload['taskOptions'].merge!(params['taskOptions'])
-			end
+        print reset,"\n"
+      end
+          rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
 
-			request_payload = {task: task_payload}
-			response = @load_balancers_interface.update(task['id'], request_payload)
-			if options[:json]
-				print JSON.pretty_generate(json_response)
-				if !response['success']
-					exit 1
-				end
-			else
-				print "\n", cyan, "Task #{response['task']['name']} updated", reset, "\n\n"
-			end
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
+  # JD: This is broken.. copied from tasks? should optionTypes exist?
+  def add(args)
+    lb_type_name = nil
+    options = {}
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[name] -t LB_TYPE")
+      opts.on( '-t', '--type LB_TYPE', "Load Balancer Type" ) do |val|
+        lb_type_name = val
+      end
+      build_common_options(opts, options, [:options, :json, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    lb_name = args[0]
+    if args.count < 1
+      puts optparse
+      exit 1
+    end
+    if lb_type_name.nil?
+      puts optparse
+      exit 1
+    end
+    connect(options)
+    begin
+      lb_type = load_balancer_type_for_name_or_id(lb_type_name)
+      if lb_type.nil?
+        print_red_alert "LB Type #{lb_type_name} not found!"
+        exit 1
+      end
 
+      payload = {loadBalancer: {name: lb_name, type: {code: lb_type['code'], id: lb_type['id']}}}
 
-	def lb_types(args)
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus load-balancers lb-types"
-			build_common_options(opts, options, [:json, :remote])
-		end
-		optparse.parse(args)
-		connect(options)
-		begin
-			json_response = @load_balancers_interface.load_balancer_types()
-			if options[:json]
-					print JSON.pretty_generate(json_response)
-			else
-				lb_types = json_response['loadBalancerTypes']
-				print "\n" ,cyan, bold, "Morpheus Load Balancer Types\n","============================", reset, "\n\n"
-				if lb_types.nil? || lb_types.empty?
-					puts yellow,"No lb types currently exist on this appliance. This could be a seed issue.",reset
-				else
-					print cyan
-					lb_table_data = lb_types.collect do |lb_type|
-						{name: lb_type['name'], id: lb_type['id'], code: lb_type['code']}
-					end
-					tp lb_table_data, :id, :name, :code
-				end
+      # todo: The options available here are specific by type...
+      #input_options = Morpheus::Cli::OptionTypes.prompt(lb_type['optionTypes'],options[:options],@api_client, options[:params])
+      json_response = @load_balancers_interface.create(payload)
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+      else
+        print "\n", cyan, "LB #{json_response['loadBalancer']['name']} created successfully", reset, "\n\n"
+      end
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
 
-				print reset,"\n\n"
-			end
-			
-			
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
+  def remove(args)
+    lb_name = args[0]
+    options = {}
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[name]")
+      build_common_options(opts, options, [:auto_confirm, :json, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    if args.count < 1
+      puts optparse
+      exit 1
+    end
+    connect(options)
+    begin
+      lb = find_lb_by_name_or_id(lb_name)
+      exit 1 if lb.nil?
+      unless options[:yes] || Morpheus::Cli::OptionTypes.confirm("Are you sure you want to delete the load balancer #{lb['name']}?")
+        exit
+      end
+      if options[:dry_run]
+        print_dry_run @load_balancers_interface.dry.destroy(lb['id'])
+        return
+      end
+      json_response = @load_balancers_interface.destroy(lb['id'])
+      if options[:json]
+        print JSON.pretty_generate(json_response), "\n"
+      else
+        print "\n", cyan, "Load Balancer #{lb['name']} removed", reset, "\n\n"
+      end
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
 
-	def add(args)
-		lb_name = args[0]
-		lb_type_name = nil
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus load-balancers add [lb] -t LB_TYPE"
-			opts.on( '-t', '--type LB_TYPE', "Lb Type" ) do |val|
-				lb_type_name = val
-			end
-			build_common_options(opts, options, [:options, :json, :remote])
-		end
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-		optparse.parse(args)
-		connect(options)
 
-		if lb_type_name.nil?
-			puts "LB Type must be specified...\n#{optparse.banner}"
-			exit 1
-		end
+  private
 
-		lb_type = find_lb_type_by_name(lb_type_name)
-		if lb_type.nil?
-			puts "LB Type not found!"
-			exit 1
-		end
-		input_options = Morpheus::Cli::OptionTypes.prompt(lb_type['optionTypes'],options[:options],@api_client, options[:params])
-		payload = {task: {name: lb_name, taskOptions: input_options['taskOptions'], type: {code: lb_type['code'], id: lb_type['id']}}}
-		begin
-			json_response = @load_balancers_interface.create(payload)
-			if options[:json]
-				print JSON.pretty_generate(json_response)
-			else
-				print "\n", cyan, "LB #{json_response['loadBalancer']['name']} created successfully", reset, "\n\n"
-			end
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
+  def find_lb_by_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return find_lb_by_id(val)
+    else
+      return find_lb_by_name(val)
+    end
+  end
 
-	def remove(args)
-		lb_name = args[0]
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = "Usage: morpheus load-balancers remove [name]"
-			build_common_options(opts, options, [:auto_confirm, :json, :remote])
-		end
-		if args.count < 1
-			puts "\n#{optparse.banner}\n\n"
-			exit 1
-		end
-		optparse.parse(args)
-		connect(options)
-		begin
-			lb = find_lb_by_name(lb_name)
-			exit 1 if lb.nil?
-			unless options[:yes] || Morpheus::Cli::OptionTypes.confirm("Are you sure you want to delete the load balancer #{lb['name']}?")
-				exit
-			end
-			json_response = @load_balancers_interface.destroy(lb['id'])
-			if options[:json]
-					print JSON.pretty_generate(json_response)
-			else
-				print "\n", cyan, "Load Balancer #{lb['name']} removed", reset, "\n\n"
-			end
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
+  def find_lb_by_id(id)
+    begin
+      json_response = @load_balancers_interface.get(id.to_i)
+      return json_response['loadBalancer']
+    rescue RestClient::Exception => e
+      if e.response && e.response.code == 404
+        print_red_alert "Load Balancer not found by id #{id}"
+      else
+        raise e
+      end
+    end
+  end
 
+  def find_lb_by_name(name)
+    lbs = @load_balancers_interface.get({name: name.to_s})['loadBalancers']
+    if lbs.empty?
+      print_red_alert "Load Balancer not found by name #{name}"
+      return nil
+    elsif lbs.size > 1
+      print_red_alert "#{lbs.size} load balancers found by name #{name}"
+      #print_lbs_table(lbs, {color: red})
+      print reset,"\n\n"
+      return nil
+    else
+      return lbs[0]
+    end
+  end
 
-private
-	def find_lb_by_name(val)
-		raise "find_lb_by_name passed a bad name: #{val.inspect}" if val.to_s == ''
-		results = @load_balancers_interface.get(val)
-		result = nil
-		if !results['loadBalancers'].nil? && !results['loadBalancers'].empty?
-			result = results['loadBalancers'][0]
-		elsif val.to_i.to_s == val
-			results = @load_balancers_interface.get(val.to_i)
-			result = results['loadBalancer']
-		end
-		if result.nil?
-			print_red_alert "LB not found by '#{val}'"
-			return nil
-		end
-		return result
-	end
+  def get_available_load_balancer_types(refresh=false)
+    if !@available_load_balancer_types || refresh
+      @available_load_balancer_types = @load_balancers_interface.load_balancer_types['loadBalancerTypes']
+    end
+    return @available_load_balancer_types
+  end
 
-	def find_lb_type_by_name(val)
-		raise "find_lb_type_by_name passed a bad name: #{val.inspect}" if val.to_s == ''
-		results = @load_balancers_interface.load_balancer_types(val)
-		result = nil
-		if !results['loadBalancerTypes'].nil? && !results['loadBalancerTypes'].empty?
-			result = results['loadBalancerTypes'][0]
-		elsif val.to_i.to_s == val
-			results = @load_balancers_interface.load_balancer_types(val.to_i)
-			result = results['loadBalancerType']
-		end
-		if result.nil?
-			print_red_alert "LB Type not found by '#{val}'"
-			return nil
-		end
-		return result
-	end
+  def load_balancer_type_for_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return load_balancer_type_for_id(val)
+    else
+      return load_balancer_type_for_name(val)
+    end
+  end
+
+  def load_balancer_type_for_id(id)
+    return get_available_load_balancer_types().find { |z| z['id'].to_i == id.to_i}
+  end
+
+  def load_balancer_type_for_name(name)
+    return get_available_load_balancer_types().find { |z| z['name'].downcase == name.downcase || z['code'].downcase == name.downcase}
+  end
 
 end