morpheus-cli 5.3.2.3 → 5.4.1

data/lib/morpheus/cli/mixins/storage_servers_helper.rb

@@ -0,0 +1,156 @@
+require 'morpheus/cli/mixins/print_helper'
+require 'morpheus/cli/option_types'
+require 'morpheus/rest_client'
+# Mixin for Morpheus::Cli command classes
+# Provides common methods for storage server management
+# including storage servers and storage server types
+module Morpheus::Cli::StorageServersHelper
+
+  def self.included(klass)
+    klass.send :include, Morpheus::Cli::PrintHelper
+  end
+
+  def storage_servers_interface
+    # @api_client.storage_servers
+    raise "#{self.class} has not defined @storage_servers_interface" if @storage_servers_interface.nil?
+    @storage_servers_interface
+  end
+
+  def storage_server_types_interface
+    # @api_client.storage_server_types
+    raise "#{self.class} has not defined @storage_server_types_interface" if @storage_server_types_interface.nil?
+    @storage_server_types_interface
+  end
+
+  def storage_server_object_key
+    'storageServer'
+  end
+
+  def storage_server_list_key
+    'storageServers'
+  end
+
+  def storage_server_label
+    'Storage Server'
+  end
+
+  def storage_server_label_plural
+    'Storage Server'
+  end
+
+  def storage_server_type_object_key
+    'storageServerType'
+  end
+
+  def storage_server_type_list_key
+    'storageServerTypes'
+  end
+
+  def storage_server_type_label
+    'Storage Server Type'
+  end
+
+  def storage_server_type_label_plural
+    'Storage Server Types'
+  end
+
+  def find_storage_server_by_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return find_storage_server_by_id(val)
+    else
+      return find_storage_server_by_name(val)
+    end
+  end
+
+  def find_storage_server_by_id(id)
+    begin
+      json_response = storage_servers_interface.get(id.to_i)
+      return json_response[storage_server_object_key]
+    rescue RestClient::Exception => e
+      if e.response && e.response.code == 404
+        print_red_alert "Storage Server not found by id #{id}"
+      else
+        raise e
+      end
+    end
+  end
+
+  def find_storage_server_by_name(name)
+    lbs = storage_servers_interface.list({name: name.to_s})[storage_server_list_key]
+    if lbs.empty?
+      print_red_alert "Storage Server not found by name #{name}"
+      return nil
+    elsif lbs.size > 1
+      print_red_alert "#{lbs.size} storage servers found by name #{name}"
+      #print_lbs_table(lbs, {color: red})
+      print reset,"\n\n"
+      return nil
+    else
+      return lbs[0]
+    end
+  end
+
+  def get_available_storage_server_types(refresh=false)
+    if !@available_storage_server_types || refresh
+      @available_storage_server_types = storage_server_types_interface.list({max:1000})[storage_server_type_list_key]
+    end
+    return @available_storage_server_types
+  end
+
+  def storage_server_type_for_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return storage_server_type_for_id(val)
+    else
+      return storage_server_type_for_name(val)
+    end
+  end
+
+  def storage_server_type_for_id(id)
+    return get_available_storage_server_types().find { |z| z['id'].to_i == id.to_i}
+  end
+
+  def storage_server_type_for_name(name)
+    return get_available_storage_server_types().find { |z| z['name'].downcase == name.downcase || z['code'].downcase == name.downcase}
+  end
+
+  def find_storage_server_type_by_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return find_storage_server_type_by_id(val)
+    else
+      return find_storage_server_type_by_name(val)
+    end
+  end
+
+  def find_storage_server_type_by_id(id)
+    begin
+      json_response = storage_server_types_interface.get(id.to_i)
+      return json_response[storage_server_type_object_key]
+    rescue RestClient::Exception => e
+      if e.response && e.response.code == 404
+        print_red_alert "Storage Server Type not found by id #{id}"
+        return nil
+      else
+        raise e
+      end
+    end
+  end
+
+  def find_storage_server_type_by_name(name)
+    json_response = storage_server_types_interface.list({name: name.to_s})
+    storage_server_types = json_response[storage_server_type_list_key]
+    if storage_server_types.empty?
+      print_red_alert "Storage Server Type not found by name #{name}"
+      return storage_server_types
+    elsif storage_server_types.size > 1
+      print_red_alert "#{storage_server_types.size} storage server types found by name #{name}"
+      rows = storage_server_types.collect do |it|
+        {id: it['id'], name: it['name']}
+      end
+      puts as_pretty_table(rows, [:id, :name], {color:red})
+      return nil
+    else
+      return storage_server_types[0]
+    end
+  end
+
+end

data/lib/morpheus/cli/mixins/storage_volumes_helper.rb

@@ -0,0 +1,119 @@
+require 'morpheus/cli/mixins/print_helper'
+require 'morpheus/cli/option_types'
+require 'morpheus/rest_client'
+# Mixin for Morpheus::Cli command classes
+# Provides common methods for storage volume management
+# including storage volumes and storage volume types
+module Morpheus::Cli::StorageVolumesHelper
+
+  def self.included(klass)
+    klass.send :include, Morpheus::Cli::PrintHelper
+  end
+
+  def storage_volumes_interface
+    # @api_client.storage_volumes
+    raise "#{self.class} has not defined @storage_volumes_interface" if @storage_volumes_interface.nil?
+    @storage_volumes_interface
+  end
+
+  def storage_volume_types_interface
+    # @api_client.storage_volume_types
+    raise "#{self.class} has not defined @storage_volume_types_interface" if @storage_volume_types_interface.nil?
+    @storage_volume_types_interface
+  end
+
+  def storage_volume_object_key
+    'storageVolume'
+  end
+
+  def storage_volume_list_key
+    'storageVolumes'
+  end
+
+  def storage_volume_label
+    'Storage Volume'
+  end
+
+  def storage_volume_label_plural
+    'Storage Volume'
+  end
+
+  def storage_volume_type_object_key
+    'storageVolumeType'
+  end
+
+  def storage_volume_type_list_key
+    'storageVolumeTypes'
+  end
+
+  def storage_volume_type_label
+    'Storage Volume Type'
+  end
+
+  def storage_volume_type_label_plural
+    'Storage Volume Types'
+  end
+
+  def get_available_storage_volume_types(refresh=false)
+    if !@available_storage_volume_types || refresh
+      @available_storage_volume_types = storage_volume_types_interface.list({max:1000})[storage_volume_type_list_key]
+    end
+    return @available_storage_volume_types
+  end
+
+  def storage_volume_type_for_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return storage_volume_type_for_id(val)
+    else
+      return storage_volume_type_for_name(val)
+    end
+  end
+
+  def storage_volume_type_for_id(val)
+    record = get_available_storage_volume_types().find { |z| z['id'].to_i == val.to_i}
+    label = "Storage Volume Type"
+    if record.nil?
+      print_red_alert "#{label} not found by id #{val}"
+      return nil
+    end
+    return record
+  end
+
+  def storage_volume_type_for_name(val)
+    records = get_available_storage_volume_types().select { |z| z['name'].downcase == val.downcase || z['code'].downcase == val.downcase}
+    label = "Storage Volume Type"
+    if records.empty?
+      print_red_alert "#{label} not found by name '#{val}'"
+      return nil
+    elsif records.size > 1
+      print_red_alert "More than one #{label.downcase} found by name '#{val}'"
+      print_error "\n"
+      puts_error as_pretty_table(records, [:id, :name], {color:red})
+      print_red_alert "Try using ID instead"
+      print_error reset,"\n"
+      return nil
+    else
+      return records[0]
+    end
+  end
+
+  def format_storage_volume_status(record, return_color=cyan)
+    out = ""
+    status_string = record['status']
+    if status_string.nil? || status_string.empty? || status_string == "unknown"
+      out << "#{white}UNKNOWN#{return_color}"
+    elsif status_string == 'provisioned' || status_string == 'unattached'
+      out << "#{cyan}#{status_string.capitalize}#{return_color}"
+    elsif status_string == 'syncing'
+      out << "#{yellow}#{status_string.capitalize}#{return_color}"
+    else
+      out << "#{red}#{status_string ? status_string.capitalize : 'N/A'}#{record['statusMessage'] ? "#{return_color} - #{record['statusMessage']}" : ''}#{return_color}"
+    end
+    out
+  end
+
+  def format_storage_volume_source(storage_volume)
+    storage_volume['source']
+  end
+
+end

data/lib/morpheus/cli/option_parser.rb

@@ -32,27 +32,35 @@ module Morpheus
           #out << "Options:\n"
           # the default way..
           # out << summarize().join("")
-
           # super hacky, should be examining the option, not the fully formatted description
           my_summaries = summarize()
-          summarize().each do |opt_description|
-            is_hidden = (@hidden_options || []).find { |hidden_switch|
-              # opt_description.include?("--#{hidden_switch}")
-              if hidden_switch.start_with?("-")
-                opt_description.to_s.strip.start_with?("#{hidden_switch} ")
-              else
-                opt_description.to_s.strip.start_with?("--#{hidden_switch} ")
-              end
-            }
-            if is_hidden
-              if opts[:show_hidden_options]
-                # out << opt_description + " (hidden)"
-                out << opt_description
+          if opts[:show_hidden_options]
+            my_summaries.each do |full_line|
+              out << full_line
+            end
+          else
+            on_hidden_option = false
+            my_summaries.each do |full_line|
+              opt_description = full_line.to_s.strip
+              if opt_description.start_with?("-")
+                is_hidden = (@hidden_options || []).find { |hidden_switch|
+                  if hidden_switch.start_with?("-")
+                    opt_description.start_with?("#{hidden_switch} ")
+                  else
+                    opt_description.start_with?("--#{hidden_switch} ")
+                  end
+                }
+                if is_hidden
+                  on_hidden_option = true
+                else
+                  on_hidden_option = false
+                  out << full_line
+                end
               else
-                # hidden
+                if on_hidden_option == false
+                  out << full_line
+                end
               end
-            else
-              out << opt_description
             end
           end
         end