morpheus-cli 0.1.0 → 0.1.1

data/lib/morpheus/cli/cli_registry.rb

@@ -0,0 +1,75 @@
+
+module Morpheus
+  module Cli
+    class CliRegistry
+
+      def initialize
+        @commands = {}
+      end
+      
+      class << self
+        
+        def instance
+          @instance ||= CliRegistry.new
+        end
+
+        def exec(command_name, args)
+          instance.get(command_name).new.handle(args)
+        end
+
+        def add(klass, command_name=nil)
+          klass_command_name = cli_ize(klass.name.split('::')[-1])
+
+          if has_command?(klass_command_name) && !command_name.nil?
+            instance.remove(klass_command_name)
+            instance.add(command_name, klass)
+          else
+            instance.add(klass_command_name, klass)
+          end
+        end
+
+        def has_command?(command_name)
+          if command_name.nil? || command_name == ''
+            false
+          else
+            !instance.get(command_name).nil?
+          end
+        end
+
+        def all
+          instance.all
+        end
+
+        private
+
+        def cli_ize(klass_name)
+          # borrowed from ActiveSupport
+          return klass_name unless klass_name =~ /[A-Z-]|::/
+          word = klass_name.to_s.gsub(/::/, '/')
+          word.gsub!(/(?:(?<=([A-Za-z\d]))|\b)(?=\b|[^a-z])/) { "#{$1 && '_'}" }
+          word.gsub!(/([A-Z\d]+)([A-Z][a-z])/,'\1_\2')
+          word.gsub!(/([a-z\d])([A-Z])/,'\1_\2')
+          word.tr!("-", "_")
+          word.downcase!
+          word.chop.tr('_', '-')
+        end
+      end
+
+      def all
+        @commands
+      end
+
+      def get(cmd_name)
+        @commands[cmd_name.to_sym]
+      end
+
+      def add(cmd_name, klass)
+        @commands[cmd_name.to_sym] = klass
+      end
+
+      def remove(cmd_name)
+        @commands.delete(cmd_name.to_sym)
+      end
+    end
+  end
+end

data/lib/morpheus/cli/credentials.rb

@@ -1,6 +1,7 @@
 require 'yaml'
 require 'io/console'
-require 'rest_client'
+require 'optparse'
+
 module Morpheus
 	module Cli
 		class Credentials
@@ -21,7 +22,7 @@ module Morpheus
 
 					oauth_url = File.join(@appliance_url, "/oauth/token")
 					begin
-						authorize_response = RestClient.post oauth_url, {grant_type: 'password', scope:'write', client_id: 'morph-cli', username: username, password: password}
+						authorize_response = Morpheus::RestClient.post oauth_url, {grant_type: 'password', scope:'write', client_id: 'morph-cli', username: username, password: password}
 
 						json_response = JSON.parse(authorize_response.to_s)
 						access_token = json_response['access_token']
@@ -96,4 +97,4 @@ module Morpheus
 			end
 		end
 	end
-end
+end

data/lib/morpheus/cli/deploys.rb

@@ -5,9 +5,14 @@ require 'term/ansicolor'
 require 'optparse'
 require 'filesize'
 require 'table_print'
+require 'morpheus/cli/cli_command'
 
 class Morpheus::Cli::Deploys
+  include Morpheus::Cli::CliCommand
 	include Term::ANSIColor
+
+  cli_command_name :deploy
+  
 	def initialize() 
 		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
 		@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).request_credentials()
@@ -133,7 +138,7 @@ class Morpheus::Cli::Deploys
 	#
 	def load_deploy_file
 		if !File.exist? "morpheus.yml"
-			puts "No morheus.yml file detected in the current directory. Nothing to do."
+			puts "No morpheus.yml file detected in the current directory. Nothing to do."
 			return nil
 		end
 
@@ -150,4 +155,4 @@ class Morpheus::Cli::Deploys
 		end
 		return deploy_args
 	end
-end
+end

data/lib/morpheus/cli/groups.rb

@@ -3,10 +3,12 @@ require 'io/console'
 require 'rest_client'
 require 'term/ansicolor'
 require 'optparse'
-
+require 'morpheus/cli/cli_command'
 
 class Morpheus::Cli::Groups
+  include Morpheus::Cli::CliCommand
 	include Term::ANSIColor
+  
 	def initialize() 
 		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
 		@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).request_credentials()
@@ -168,4 +170,4 @@ class Morpheus::Cli::Groups
 	def self.save_groups(group_map)
 		File.open(groups_file_path, 'w') {|f| f.write group_map.to_yaml } #Store
 	end
-end
+end

data/lib/morpheus/cli/instance_types.rb

@@ -4,8 +4,10 @@ require 'rest_client'
 require 'term/ansicolor'
 require 'optparse'
 require 'filesize'
+require 'morpheus/cli/cli_command'
 
 class Morpheus::Cli::InstanceTypes
+  include Morpheus::Cli::CliCommand
 	include Term::ANSIColor
 	def initialize() 
 		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
@@ -98,4 +100,4 @@ class Morpheus::Cli::InstanceTypes
 			return nil
 		end
 	end
-end
+end

data/lib/morpheus/cli/instances.rb

@@ -5,8 +5,10 @@ require 'term/ansicolor'
 require 'optparse'
 require 'filesize'
 require 'table_print'
+require 'morpheus/cli/cli_command'
 
 class Morpheus::Cli::Instances
+  include Morpheus::Cli::CliCommand
 	include Term::ANSIColor
 	def initialize() 
 		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
@@ -24,7 +26,7 @@ class Morpheus::Cli::Instances
 			return 1
 		end
 		if args.empty?
-			puts "\nUsage: morpheus instances [list,add,remove,stop,start,restart,resize,upgrade,clone,envs,setenv,delenv] [name]\n\n"
+			puts "\nUsage: morpheus instances [list,add,remove,stop,start,restart,resize,upgrade,clone,envs,setenv,delenv,firewall_disable,firewall_enable,security_groups,apply_security_groups] [name]\n\n"
 			return 
 		end
 
@@ -51,6 +53,14 @@ class Morpheus::Cli::Instances
 				setenv(args[1..-1])	
 			when 'delenv'
 				delenv(args[1..-1])	
+			when 'firewall_disable'
+				firewall_disable(args[1..-1])	
+			when 'firewall_enable'
+				firewall_enable(args[1..-1])	
+			when 'security_groups'	
+				security_groups(args[1..-1])	
+			when 'apply_security_groups'	
+				apply_security_groups(args[1..-1])	
 			else
 				puts "\nUsage: morpheus instances [list,add,remove,stop,start,restart,resize,upgrade,clone,envs,setenv,delenv] [name]\n\n"
 		end
@@ -439,6 +449,145 @@ class Morpheus::Cli::Instances
 		end
 	end
 
+	def firewall_disable(args)
+		if args.count < 1
+			puts "\nUsage: morpheus instances firewall_disable [name]\n\n"
+			return
+		end
+		begin
+			instance_results = @instances_interface.get({name: args[0]})
+			if instance_results['instances'].empty?
+				puts "Instance not found by name #{args[0]}"
+				return
+			end
+			@instances_interface.firewall_disable(instance_results['instances'][0]['id'])
+			security_groups([args[0]])
+		rescue RestClient::Exception => e
+			if e.response.code == 400
+				error = JSON.parse(e.response.to_s)
+				::Morpheus::Cli::ErrorHandler.new.print_errors(error)
+			else
+				puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			end
+			return nil
+		end
+	end
+
+	def firewall_enable(args)
+		if args.count < 1
+			puts "\nUsage: morpheus instances firewall_enable [name]\n\n"
+			return
+		end
+		begin
+			instance_results = @instances_interface.get({name: args[0]})
+			if instance_results['instances'].empty?
+				puts "Instance not found by name #{args[0]}"
+				return
+			end
+			@instances_interface.firewall_enable(instance_results['instances'][0]['id'])
+			security_groups([args[0]])
+		rescue RestClient::Exception => e
+			if e.response.code == 400
+				error = JSON.parse(e.response.to_s)
+				::Morpheus::Cli::ErrorHandler.new.print_errors(error)
+			else
+				puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			end
+			return nil
+		end
+	end
+
+	def security_groups(args)
+		if args.count < 1
+			puts "\nUsage: morpheus instances security_groups [name]\n\n"
+			return
+		end
+		begin
+			instance_results = @instances_interface.get({name: args[0]})
+			if instance_results['instances'].empty?
+				puts "Instance not found by name #{args[0]}"
+				return
+			end
+
+			instance_id = instance_results['instances'][0]['id']
+			json_response = @instances_interface.security_groups(instance_id)
+
+			securityGroups = json_response['securityGroups']
+			print "\n" ,cyan, bold, "Morpheus Security Groups for Instance:#{instance_id}\n","==================", reset, "\n\n"
+			print cyan, "Firewall Enabled=#{json_response['firewallEnabled']}\n\n"
+			if securityGroups.empty?
+				puts yellow,"No security groups currently applied.",reset
+			else
+				securityGroups.each do |securityGroup|
+					print cyan, "=  #{securityGroup['id']} (#{securityGroup['name']}) - (#{securityGroup['description']})\n"
+				end
+			end
+			print reset,"\n\n"
+
+		rescue RestClient::Exception => e
+			if e.response.code == 400
+				error = JSON.parse(e.response.to_s)
+				::Morpheus::Cli::ErrorHandler.new.print_errors(error)
+			else
+				puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			end
+			return nil
+		end
+	end
+
+	def apply_security_groups(args)
+		usage = <<-EOF
+Usage: morpheus instances apply_security_groups [name] [options]
+EOF
+		if args.count < 1
+			puts usage
+			return
+		end
+
+		options = {}
+		clear_or_secgroups_specified = false
+		optparse = OptionParser.new do|opts|
+			opts.banner = usage
+			opts.on( '-c', '--clear', "Clear all security groups" ) do
+				options[:securityGroupIds] = []
+				clear_or_secgroups_specified = true
+			end
+			opts.on( '-s', '--secgroups SECGROUPS', "Apply the specified comma separated security group ids" ) do |secgroups|
+				options[:securityGroupIds] = secgroups.split(",")
+				clear_or_secgroups_specified = true
+			end
+			opts.on( '-h', '--help', "Prints this help" ) do
+				puts opts
+				exit
+			end
+		end
+		optparse.parse(args)
+
+		if !clear_or_secgroups_specified 
+			puts usage
+			exit
+		end
+
+		begin
+			instance_results = @instances_interface.get({name: args[0]})
+			if instance_results['instances'].empty?
+				puts "Instance not found by name #{args[0]}"
+				return
+			end
+
+			@instances_interface.apply_security_groups(instance_results['instances'][0]['id'], options)
+			security_groups([args[0]])
+		rescue RestClient::Exception => e
+			if e.response.code == 400
+				error = JSON.parse(e.response.to_s)
+				::Morpheus::Cli::ErrorHandler.new.print_errors(error)
+			else
+				puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			end
+			return nil
+		end
+	end
+
 private 
 	def find_group_by_name(name)
 		group_results = @groups_interface.get(name)
@@ -457,4 +606,4 @@ private
 		end
 		return instance_type_results['instanceTypes'][0]
 	end
-end
+end

data/lib/morpheus/cli/remote.rb

@@ -14,6 +14,7 @@ class Morpheus::Cli::Remote
 	def handle(args) 
 		if args.empty?
 			puts "\nUsage: morpheus remote [list,add,remove,use] [name] [host]\n\n"
+      return
 		end
 
 		case args[0]
@@ -160,4 +161,4 @@ class Morpheus::Cli::Remote
 	def self.save_appliances(appliance_map)
 		File.open(appliances_file_path, 'w') {|f| f.write appliance_map.to_yaml } #Store
 	end
-end
+end

data/lib/morpheus/cli/security_group_rules.rb

@@ -0,0 +1,243 @@
+# require 'yaml'
+require 'io/console'
+require 'rest_client'
+require 'term/ansicolor'
+require 'optparse'
+require 'filesize'
+require 'table_print'
+require 'morpheus/cli/cli_command'
+
+class Morpheus::Cli::SecurityGroupRules
+  include Morpheus::Cli::CliCommand
+	include Term::ANSIColor
+	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,bold, "\nInvalid Credentials. Unable to acquire access token. Please verify your credentials and try again.\n\n",reset
+			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'
+EOT
+
+		if args.count < 3
+			puts usage
+			return
+		end
+		
+		security_group_id = nil 
+		optparse = OptionParser.new do|opts|
+			opts.banner = "\nUsage: morpheus security-group-rules add_custom_rule SOURCE_CIDR PORT_RANGE PROTOCOL [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
+			opts.on( '-h', '--help', "Prints this help" ) do
+				puts opts
+				exit
+			end
+		end
+		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
+
+		options = {
+			:rule => {
+				:source => args[0],
+				:portRange => args[1],
+				:protocol => args[2],
+				:customRule => true
+			}
+		}
+
+		begin
+			@security_group_rules_interface.create(security_group_id, options)
+		rescue => e
+			if e.response.code == 400
+				error = JSON.parse(e.response.to_s)
+				::Morpheus::Cli::ErrorHandler.new.print_errors(error)
+			else
+				puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			end
+			return nil
+		end
+		list([])
+	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
+EOT
+		if args.count < 2
+			puts usage
+			return
+		end
+		
+		security_group_id = nil 
+		optparse = OptionParser.new do|opts|
+			opts.banner = "\nmorpheus security-group-rules add_instance_rule SOURCE_CIDR INSTANCE_TYPE_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
+			opts.on( '-h', '--help', "Prints this help" ) do
+				puts opts
+				exit
+			end
+		end
+		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
+
+		options = {
+			:rule => {
+				:source => args[0],
+				:instanceTypeId => args[1]
+			}
+		}
+
+		begin
+			@security_group_rules_interface.create(security_group_id, options)
+		rescue => e
+			if e.response.code == 400
+				error = JSON.parse(e.response.to_s)
+				::Morpheus::Cli::ErrorHandler.new.print_errors(error)
+			else
+				puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			end
+			return nil
+		end
+		list([])
+	end
+
+	def list(args)
+		options = {}
+		security_group_id = nil 
+		optparse = OptionParser.new do|opts|
+			opts.banner = "\nUsage: morpheus security-group-rules list [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
+			opts.on( '-h', '--help', "Prints this help" ) do
+				puts opts
+				exit
+			end
+		end
+		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, options)
+			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 => e
+			puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			return nil
+		end
+	end
+
+	def remove(args)
+		if args.count < 1
+			puts "\nUsage: morpheus security-group-rules remove ID [options]\n\n"
+			return
+		end
+
+		security_group_id = nil 
+		optparse = OptionParser.new do|opts|
+			opts.banner = "\nUsage: 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
+			opts.on( '-h', '--help', "Prints this help" ) do
+				puts opts
+				exit
+			end
+		end
+		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
+			@security_group_rules_interface.delete(security_group_id, args[0])
+			list([])
+		rescue RestClient::Exception => e
+			if e.response.code == 400
+				error = JSON.parse(e.response.to_s)
+				::Morpheus::Cli::ErrorHandler.new.print_errors(error)
+			else
+				puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			end
+			return nil
+		end
+	end
+
+end