morpheus-cli 0.1.0 → 0.1.1

data/lib/morpheus/cli/security_groups.rb

@@ -0,0 +1,192 @@
+# 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::SecurityGroups
+  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_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,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-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)
+		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|
+					print cyan, "=  #{security_group['id']}: #{security_group['name']} (#{security_group['description']})\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 get(args)
+		if args.count < 1
+			puts "\nUsage: morpheus security-groups get ID\n\n"
+			return
+		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 => e
+			puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			return nil
+		end
+	end
+
+	def add(args)
+		if args.count < 1
+			puts "\nUsage: morpheus security-groups add NAME [options]\n\n"
+			return
+		end
+
+		options = {: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|
+				options[:securityGroup][:description] = description
+			end
+
+			opts.on( '-h', '--help', "Prints this help" ) do
+				puts opts
+				exit
+			end
+		end
+		optparse.parse(args)
+
+		begin
+			@security_groups_interface.create(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 remove(args)
+		if args.count < 1
+			puts "\nUsage: morpheus security-groups remove ID\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
+			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 use(args) 
+		if args.length < 1
+			puts "Usage: morpheus security-groups use ID"
+			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 => e
+			puts "Error Communicating with the Appliance. Please try again later. #{e}"
+			return nil
+		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
+
+end

data/lib/morpheus/cli/servers.rb

@@ -3,10 +3,13 @@ require 'io/console'
 require 'rest_client'
 require 'term/ansicolor'
 require 'optparse'
+require 'morpheus/cli/cli_command'
 
 
 class Morpheus::Cli::Servers
 	include Term::ANSIColor
+  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()
@@ -259,4 +262,4 @@ private
 		end
 		return group_results['groups'][0]
 	end
-end
+end

data/lib/morpheus/cli/version.rb

@@ -1,5 +1,5 @@
 module Morpheus
   module Cli
-    VERSION = "0.1.0"
+    VERSION = "0.1.1"
   end
 end

data/lib/morpheus/cli/zones.rb

@@ -3,9 +3,10 @@ require 'io/console'
 require 'rest_client'
 require 'term/ansicolor'
 require 'optparse'
-
+require 'morpheus/cli/cli_command'
 
 class Morpheus::Cli::Zones
+  include Morpheus::Cli::CliCommand
 	include Term::ANSIColor
 	def initialize() 
 		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
@@ -21,7 +22,7 @@ class Morpheus::Cli::Zones
 			return 1
 		end
 		if args.empty?
-			puts "\nUsage: morpheus zones [list,add,remove] [name]\n\n"
+			puts "\nUsage: morpheus zones [list,add,remove,firewall_disable,firewall_enable,security_groups,apply_security_groups] [name]\n\n"
 			return
 		end
 
@@ -32,8 +33,16 @@ class Morpheus::Cli::Zones
 				add(args[1..-1])
 			when 'remove'
 				remove(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 zones [list,add,remove] [name]\n\n"
+				puts "\nUsage: morpheus zones [list,add,remove,firewall_disable,firewall_enable,security_groups,apply_security_groups] [name]\n\n"
 		end
 	end
 
@@ -163,6 +172,145 @@ class Morpheus::Cli::Zones
 		end
 	end
 
+	def firewall_disable(args)
+		if args.count < 1
+			puts "\nUsage: morpheus zones firewall_disable [name]\n\n"
+			return
+		end
+		begin
+			zone_results = @zones_interface.get({name: args[0]})
+			if zone_results['zones'].empty?
+				puts "Zone not found by name #{args[0]}"
+				return
+			end
+			@zones_interface.firewall_disable(zone_results['zones'][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 zones firewall_enable [name]\n\n"
+			return
+		end
+		begin
+			zone_results = @zones_interface.get({name: args[0]})
+			if zone_results['zones'].empty?
+				puts "Zone not found by name #{args[0]}"
+				return
+			end
+			@zones_interface.firewall_enable(zone_results['zones'][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 zones security_groups [name]\n\n"
+			return
+		end
+		begin
+			zone_results = @zones_interface.get({name: args[0]})
+			if zone_results['zones'].empty?
+				puts "Zone not found by name #{args[0]}"
+				return
+			end
+
+			zone_id = zone_results['zones'][0]['id']
+			json_response = @zones_interface.security_groups(zone_id)
+
+			securityGroups = json_response['securityGroups']
+			print "\n" ,cyan, bold, "Morpheus Security Groups for Zone:#{zone_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 zones 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
+			zone_results = @zones_interface.get({name: args[0]})
+			if zone_results['zones'].empty?
+				puts "Zone not found by name #{args[0]}"
+				return
+			end
+
+			@zones_interface.apply_security_groups(zone_results['zones'][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 zone_type_for_id(id)
@@ -202,4 +350,4 @@ class Morpheus::Cli::Zones
 		end
 		return group_results['groups'][0]
 	end
-end
+end

data/lib/morpheus/rest_client.rb

@@ -0,0 +1,50 @@
+require 'rest_client'
+
+module Morpheus
+  # A wrapper around rest_client so we can more easily deal with passing options (like turning on/off SSL verification)
+  class RestClient
+
+    class << self
+
+      def execute(options)
+        opts = options.merge({})
+
+        unless ssl_verification_enabled?
+          opts[:verify_ssl] = OpenSSL::SSL::VERIFY_NONE
+        end
+
+        ::RestClient::Request.execute opts
+      end
+      
+      def post(url, payload)
+        execute url: url, payload: payload, method: :post
+      end
+
+      def ssl_verification_enabled?
+        begin
+          @@ssl_verification_enabled.nil? ? true : @@ssl_verification_enabled
+        rescue
+          @@ssl_verification_enabled = true
+        end
+      end
+
+      def enable_ssl_verification=(verify)
+        @@ssl_verification_enabled = verify
+      end
+
+      private
+
+      def build_request_args(url, method, payload)
+        args = {url: url,
+                method: method,
+                payload: payload}
+
+        unless ssl_verification_enabled?
+          args[:verify_ssl] = OpenSSL::SSL::VERIFY_NONE
+        end
+        
+        args
+      end
+    end
+  end
+end