morpheus-cli 2.10.0 → 2.10.1

data/lib/morpheus/cli/config_file.rb

@@ -0,0 +1,126 @@
+require 'fileutils'
+require 'time'
+require 'morpheus/cli/cli_registry'
+require 'morpheus/logging'
+require 'term/ansicolor'
+
+class Morpheus::Cli::ConfigFile
+  include Term::ANSIColor
+  class << self
+    def init(filename=nil)
+      @instance ||= Morpheus::Cli::ConfigFile.new(filename)
+    end
+
+    def instance
+      #@instance ||= init(Morpheus::Cli.config_filename)
+      @instance or raise "#{self}.init() must be called!"
+    end
+
+  end
+
+  attr_reader :filename
+  attr_reader :config
+
+  def initialize(fn)
+    @config = {}
+    # only create the file if we're using the default, otherwise error
+    if fn
+      @filename = File.expand_path(fn)
+    else
+      @filename = File.expand_path(Morpheus::Cli.config_filename)
+      if !Dir.exists?(File.dirname(@filename))
+        FileUtils.mkdir_p(File.dirname(@filename))
+      end
+      if !File.exists?(@filename)
+        print "#{Term::ANSIColor.dark}Initializing default config file#{Term::ANSIColor.reset}\n" if Morpheus::Logging.debug?
+        FileUtils.touch(@filename)
+        save_file()
+      end
+    end
+    load_file()
+  end
+
+  def load_file
+    #puts "loading config #{@filename}"
+    @config = {}
+    if !@filename
+      return false
+    end
+    if !File.exist?(@filename)
+      raise "Morpheus cli config file not found: #{@filename}"
+    end
+    file_contents = File.read(@filename)
+    file_contents.split
+    config_text = File.open(@filename).read
+    config_lines = config_text.split(/\n/)
+    #config_lines = config_lines.reject {|line| line =~ /^\#/} # strip comments
+    config_lines.each_with_index do |line, line_index|
+      line_num = line_index + 1
+      line = line.strip
+      #puts "parsing config line #{line_num} : #{line}"
+      next if line.empty?
+      next if line =~ /^\#/ # skip comments
+
+      if line =~ /^alias\s+/
+        alias_name, command_string = Morpheus::Cli::CliRegistry.parse_alias_definition(line)
+        if alias_name.empty? || command_string.empty?
+          print "#{dark} #=> bad config line #{line_num} invalid alias definition: #{line}\n" if Morpheus::Logging.debug?
+        else
+          # @config[:aliases] ||= []
+          # @config[:aliases] << {name: alias_name, command: command_string}
+          begin
+            Morpheus::Cli::CliRegistry.instance.add_alias(alias_name, command_string)
+          rescue => err
+            print "#{dark} #=> #{err.message}#{reset}\n" if Morpheus::Logging.debug?
+          end
+        end
+      elsif line =~ /^colorize/
+        Term::ANSIColor::coloring = true
+      elsif line =~ /^uncolorize/
+        Term::ANSIColor::coloring = false
+      # what else shall we make configurable?
+      else
+        print "#{dark} #=> config line #{line_num} unrecognized: #{line}#{reset}\n" if Morpheus::Logging.debug?
+      end
+    end
+
+    # if @config[:aliases]
+    #   @config[:aliases].each do |it|
+    #     Morpheus::Cli::CliRegistry.instance.add_alias(it[:name], it[:command])
+    #   end
+    # end
+
+    #puts "done loading config from #{}"
+
+    return @config
+  end
+
+  def save_file
+    if !@filename
+      print "#{Term::ANSIColor.dark}Skipping config file save because filename has not been set#{Term::ANSIColor.reset}\n" if Morpheus::Logging.debug?
+      return false
+    end
+    print "#{dark} #=> Saving config file #{@filename}#{reset}\n" if Morpheus::Logging.debug?
+    out = ""
+    out << "# .morpheusrc file #{@filename}"
+    out << "\n"
+    out << "# Auto-generated by morpheus #{Morpheus::Cli::VERSION} on #{Time.now.iso8601}"
+    out << "\n\n"
+    out << "# aliases"
+    out << "\n"
+    Morpheus::Cli::CliRegistry.instance.all_aliases.each do |k, v|
+      out << "alias #{k}='#{v}'"
+      out << "\n"
+    end
+    out << "\n"
+    File.open(@filename, 'w') {|f| f.write(out) }
+    return true
+  end
+
+  # this will load any local changes, and then resave everything in the config (shell)
+  def reload_file
+    load_file
+    save_file
+  end
+
+end

data/lib/morpheus/cli/credentials.rb

@@ -1,134 +1,158 @@
+require 'fileutils'
 require 'yaml'
+require 'json'
 require 'io/console'
 require 'rest_client'
-#require 'optparse'
+require 'morpheus/logging'
 require 'morpheus/cli/mixins/print_helper'
-require 'json'
 
 module Morpheus
-	module Cli
-		class Credentials
-			include Morpheus::Cli::PrintHelper
+  module Cli
+    class Credentials
+      include Morpheus::Cli::PrintHelper
+
+      @@saved_credentials_map = nil
 
-			def initialize(appliance_name, appliance_url)
-				@appliance_url = appliance_url
-				@appliance_name = appliance_name
-			end
-			
-			def request_credentials(opts = {})
-				username = nil
-				password = nil
-				creds = nil
-				skip_save = false
-				# We should return an access Key for Morpheus CLI Here
-				if !opts[:remote_username].nil?
-					username = opts[:remote_username]
-					password = opts[:remote_password]
-					skip_save = opts[:remote_url] ? true : false
-				else
-					creds = load_saved_credentials
-				end
-				if !creds
-					print "\nEnter Morpheus Credentials for #{@appliance_name} - #{@appliance_url}\n\n",reset
-					if username.nil? || username.empty?
-						# print "Username: "
-						print "Username: #{required_blue_prompt} "
-						username = $stdin.gets.chomp!
-					end
-					if password.nil? || password.empty?
-						# print "Password: "
-						print "Password: #{required_blue_prompt} "
-						password = STDIN.noecho(&:gets).chomp!
-						print "\n"
-					end
+      def initialize(appliance_name, appliance_url)
+        @appliance_url = appliance_url
+        @appliance_name = appliance_name
+      end
+      
+      def request_credentials(opts = {})
+        #puts "request_credentials(#{opts})"
+        username = nil
+        password = nil
+        creds = nil
+        skip_save = false
+        # We should return an access Key for Morpheus CLI Here
+        if !opts[:remote_username].nil?
+          username = opts[:remote_username]
+          password = opts[:remote_password]
+          skip_save = opts[:remote_url] ? true : false
+        else
+          creds = load_saved_credentials
+        end
+        if creds
+          return creds
+        end
+        unless opts[:quiet] || opts[:no_prompt]
+          # if username.empty? || password.empty?
+            print "Enter Morpheus Credentials for #{display_appliance(@appliance_name, @appliance_url)}\n",reset
+          # end
+          if username.empty?
+            print "Username: #{required_blue_prompt} "
+            username = $stdin.gets.chomp!
+          else
+            print "Username: #{required_blue_prompt} #{username}\n"
+          end
+          if password.empty?
+            print "Password: #{required_blue_prompt} "
+            password = STDIN.noecho(&:gets).chomp!
+            print "\n"
+          else
+            print "Password: #{required_blue_prompt} \n"
+          end
+        end
+        if username.empty? || password.empty?
+          print_red_alert "Username and password are required to login."
+          return nil
+        end
+        begin
+          auth_interface = Morpheus::AuthInterface.new(@appliance_url)
+          json_response = auth_interface.login(username, password)
+          if opts[:json]
+            print JSON.pretty_generate(json_response)
+            print reset, "\n"
+          end
+          access_token = json_response['access_token']
+          if !access_token.empty?
+            unless skip_save
+              save_credentials(@appliance_name, access_token)
+            end
+            return access_token
+          else
+            print_red_alert "Credentials not verified."
+            return nil
+          end
+        rescue ::RestClient::Exception => e
+          #raise e
+          if (e.response && e.response.code == 400)
+            print_red_alert "Credentials not verified."
+            if opts[:json]
+              json_response = JSON.parse(e.response.to_s)
+              print JSON.pretty_generate(json_response)
+              print reset, "\n"
+            end
+          else
+            print_rest_exception(e, opts)
+          end
+        end
 
-					oauth_url = File.join(@appliance_url, "/oauth/token")
-					begin
-						authorize_response = Morpheus::RestClient.execute(method: :post, url: oauth_url, headers:{ params: {grant_type: 'password', scope:'write', client_id: 'morph-cli', username: username}}, payload: {password: password},verify_ssl: false, timeout: 10)
+      end
 
-						json_response = JSON.parse(authorize_response.to_s)
-						access_token = json_response['access_token']
-						if !access_token.empty?
-							save_credentials(access_token) unless skip_save
-							return access_token
-						else
-							print_red_alert "Credentials not verified."
-							return nil
-						end
-					rescue ::RestClient::Exception => e
-						if (e.response && e.response.code == 400)
-							print_red_alert "Credentials not verified."
-							if opts[:json]
-								json_response = JSON.parse(e.response.to_s)
-								print JSON.pretty_generate(json_response)
-          			print reset, "\n\n"
-							end
-						else
-							print_rest_exception(e, opts)
-						end
-						exit 1
-					rescue => e
-						print_red_alert "Error Communicating with the Appliance. #{e}"
-						exit 1
-					end
-				else
-					return creds
-				end
-			end
+      def login(opts = {})
+        clear_saved_credentials(@appliance_name)
+        request_credentials(opts)
+      end
 
-			def login(opts = {})
-				clear_saved_credentials
-				request_credentials(opts)
-			end
+      def logout()
+        clear_saved_credentials(@appliance_name)
+      end
 
-			def logout()
-				clear_saved_credentials
-			end
+      def clear_saved_credentials(appliance_name)
+        @@saved_credentials_map = load_credentials_file || {}
+        @@saved_credentials_map.delete(appliance_name)
+        print "#{dark} #=> updating credentials file #{credentials_file_path}#{reset}\n"  if Morpheus::Logging.debug?
+        File.open(credentials_file_path, 'w') {|f| f.write @@saved_credentials_map.to_yaml } #Store
+      end
 
-			def clear_saved_credentials()
-				credential_map = load_credential_file
-				if credential_map.nil?
-					credential_map = {}
-				end
-				credential_map.delete(@appliance_name)
-				File.open(credentials_file_path, 'w') {|f| f.write credential_map.to_yaml } #Store
-			end
+      def load_saved_credentials(reload=false)
+        if saved_credentials_map && !reload
+          return saved_credentials_map
+        end
+        @@saved_credentials_map = load_credentials_file || {}
+        return @@saved_credentials_map[@appliance_name]
+      end
 
-			def load_saved_credentials()
-				credential_map = load_credential_file
-				if credential_map.nil?
-					return nil
-				else
-					return credential_map[@appliance_name]
-				end
-			end
+      # Provides the current credential information, simply :appliance_name => "access_token"
+      def saved_credentials_map
+        if !defined?(@@saved_credentials_map)
+          @@saved_credentials_map = load_credentials_file
+        end
+        return @@saved_credentials_map ? @@saved_credentials_map[@appliance_name.to_sym] : nil
+      end
 
-			def load_credential_file
-				creds_file = credentials_file_path
-				if File.exist? creds_file
-					return YAML.load_file(creds_file)
-				else
-					return nil
-				end
-			end
+      def load_credentials_file
+        fn = credentials_file_path
+        if File.exist? fn
+          print "#{dark} #=> loading credentials file #{fn}#{reset}\n" if Morpheus::Logging.debug?
+          return YAML.load_file(fn)
+        else
+          return nil
+        end
+      end
 
-			def credentials_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,"credentials")
-			end
+      def credentials_file_path
+        File.join(Morpheus::Cli.home_directory, "credentials")
+      end
 
-			def save_credentials(token)
-				credential_map = load_credential_file
-				if credential_map.nil?
-					credential_map = {}
-				end
-				credential_map[@appliance_name] = token
-				File.open(credentials_file_path, 'w') {|f| f.write credential_map.to_yaml } #Store
-			end
-		end
-	end
+      def save_credentials(app_name, token)
+        # credential_map = saved_credentials_map
+        # reloading file is better for now, otherwise you can lose credentials with multiple shells.
+        credential_map = load_credentials_file || {}
+        if credential_map.nil?
+          credential_map = {}
+        end
+        credential_map[app_name] = token
+        begin
+          fn = credentials_file_path
+          print "#{dark} #=> adding credentials to #{fn}#{reset}\n" if Morpheus::Logging.debug?
+          File.open(fn, 'w') {|f| f.write credential_map.to_yaml } #Store
+          FileUtils.chmod(0600, fn)
+        rescue => e
+          puts "failed to save #{fn}. #{e}"  if Morpheus::Logging.debug?
+        end
+      end
+    end
+  end
 end

data/lib/morpheus/cli/dashboard_command.rb

@@ -4,70 +4,62 @@ require 'json'
 
 class Morpheus::Cli::DashboardCommand
   include Morpheus::Cli::CliCommand
-  
-  cli_command_name :dashboard
-  cli_command_hidden # remove once this is done
-
-	def initialize() 
-		@appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
-	end
+  set_command_name :dashboard
+  set_command_hidden # remove once this is done
 
-	def connect(opts)
-		@access_token = Morpheus::Cli::Credentials.new(@appliance_name,@appliance_url).load_saved_credentials()
-		if @access_token.empty?
-			print_red_alert "Invalid Credentials. Unable to acquire access token. Please verify your credentials and try again."
-			exit 1
-		end
-		@api_client = Morpheus::APIClient.new(@access_token,nil,nil, @appliance_url)
-		@dashboard_interface = @api_client.dashboard
-	end
+  def initialize()
+    # @appliance_name, @appliance_url = Morpheus::Cli::Remote.active_appliance
+  end
 
-	def usage
-		"Usage: morpheus dashboard"
-	end
+  def connect(opts)
+    @api_client = establish_remote_appliance_connection(opts)
+    @dashboard_interface = @api_client.dashboard
+  end
 
-	def handle(args)
-		show(args)
-	end
-	
-	def show(args)
-		options = {}
-		optparse = OptionParser.new do|opts|
-			opts.banner = usage
-			build_common_options(opts, options, [:json]) # todo: support :account
-		end
-		optparse.parse(args)
+  def usage
+    "Usage: morpheus #{command_name}"
+  end
 
-		connect(options)
-		begin
-			
-			params = {}
+  def handle(args)
+    show(args)
+  end
+  
+  def show(args)
+    options = {}
+    optparse = OptionParser.new do|opts|
+      opts.banner = usage
+      build_common_options(opts, options, [:json, :dry_run]) # todo: support :account
+    end
+    optparse.parse!(args)
 
-			json_response = @dashboard_interface.get(params)
-			
-			if options[:json]
-				print JSON.pretty_generate(json_response)
-				print "\n"
-			else
-				
+    connect(options)
+    begin
+      params = {}
+      if options[:dry_run]
+        print_dry_run @dashboard_interface.dry.get(params)
+        return
+      end
+      json_response = @dashboard_interface.get(params)
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+        print "\n"
+      else
 
-				# todo: impersonate command and show that info here
+        # todo: impersonate command and show that info here
 
-				print "\n" ,cyan, bold, "Dashboard\n","==================", reset, "\n\n"
-				print cyan
-				
-				print "\n"
-				puts "Coming soon.... see --json"
-				print "\n"
+        print "\n" ,cyan, bold, "Dashboard\n","==================", reset, "\n\n"
+        print cyan
+        print "\n"
+        puts "Coming soon.... see --json"
+        print "\n"
 
-				print reset,"\n"
+        print reset,"\n"
 
-			end
-		rescue RestClient::Exception => e
-			print_rest_exception(e, options)
-			exit 1
-		end
-	end
-	
+      end
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
 
 end