mtncloud-cli 1.0.0

data/lib/mtncloud/cli/mixins/load_balancers_helper.rb

@@ -0,0 +1,176 @@
+require 'mtncloud/cli/mixins/print_helper'
+require 'mtncloud/cli/option_types'
+require 'mtncloud/rest_client'
+# Mixin for MtnCloud::Cli command classes
+# Provides common methods for load balancer management
+# including load balancers, load balancer types, virtual servers, etc.
+module MtnCloud::Cli::LoadBalancersHelper
+
+  def self.included(klass)
+    klass.send :include, MtnCloud::Cli::PrintHelper
+  end
+
+  def load_balancers_interface
+    # @api_client.load_balancers
+    raise "#{self.class} has not defined @load_balancers_interface" if @load_balancers_interface.nil?
+    @load_balancers_interface
+  end
+
+  def load_balancer_types_interface
+    # @api_client.load_balancer_types
+    raise "#{self.class} has not defined @load_balancer_types_interface" if @load_balancer_types_interface.nil?
+    @load_balancer_types_interface
+  end
+
+  def load_balancer_object_key
+    'loadBalancer'
+  end
+
+  def load_balancer_list_key
+    'loadBalancers'
+  end
+
+  def load_balancer_label
+    'Load Balancer'
+  end
+
+  def load_balancer_label_plural
+    'Load Balancers'
+  end
+
+  def load_balancer_type_object_key
+    'loadBalancerType'
+  end
+
+  def load_balancer_type_list_key
+    'loadBalancerTypes'
+  end
+
+  def load_balancer_type_label
+    'Load Balancer Type'
+  end
+
+  def load_balancer_type_label_plural
+    'Load Balancer Types'
+  end
+
+  def find_load_balancer_by_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return find_load_balancer_by_id(val)
+    else
+      return find_load_balancer_by_name(val)
+    end
+  end
+
+  def find_load_balancer_by_id(id)
+    begin
+      json_response = load_balancers_interface.get(id.to_i)
+      return json_response[load_balancer_object_key]
+    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_load_balancer_by_name(name)
+    lbs = load_balancers_interface.list({name: name.to_s})[load_balancer_list_key]
+    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
+
+  def get_available_load_balancer_types(refresh=false)
+    if !@available_load_balancer_types || refresh
+      @available_load_balancer_types = load_balancer_types_interface.list({max:1000})[load_balancer_type_list_key]
+    end
+    return @available_load_balancer_types
+  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(val)
+    record = get_available_load_balancer_types().find { |z| z['id'].to_i == val.to_i}
+    label = "Load Balancer Type"
+    if record.nil?
+      print_red_alert "#{label.downcase} not found by id #{val}"
+      return nil
+    end
+    return record
+  end
+
+  def load_balancer_type_for_name(val)
+    records = get_available_load_balancer_types().select { |z| z['name'].downcase == val.downcase || z['code'].downcase == val.downcase}
+    label = "Load Balancer 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 find_load_balancer_type_by_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return find_load_balancer_type_by_id(val)
+    else
+      return find_load_balancer_type_by_name(val)
+    end
+  end
+
+  def find_load_balancer_type_by_id(id)
+    begin
+      json_response = load_balancer_types_interface.get(id.to_i)
+      return json_response[load_balancer_type_object_key]
+    rescue RestClient::Exception => e
+      if e.response && e.response.code == 404
+        print_red_alert "Load Balancer Type not found by id #{id}"
+        return nil
+      else
+        raise e
+      end
+    end
+  end
+
+  def find_load_balancer_type_by_name(name)
+    json_response = load_balancer_types_interface.list({name: name.to_s})
+    load_balancer_types = json_response[load_balancer_type_list_key]
+    if load_balancer_types.empty?
+      print_red_alert "Load Balancer Type not found by name #{name}"
+      return load_balancer_types
+    elsif load_balancer_types.size > 1
+      print_red_alert "#{load_balancer_types.size} load balancer types found by name #{name}"
+      rows = load_balancer_types.collect do |it|
+        {id: it['id'], name: it['name']}
+      end
+      puts as_pretty_table(rows, [:id, :name], {color:red})
+      return nil
+    else
+      return load_balancer_types[0]
+    end
+  end
+
+end

data/lib/mtncloud/cli/mixins/logs_helper.rb

@@ -0,0 +1,131 @@
+require 'mtncloud/cli/mixins/print_helper'
+
+# Mixin for MtnCloud::Cli command classes
+# Provides common methods for fetching log records.
+# The including class must establish @logs_interface, @containers_interface, @servers_interface
+module MtnCloud::Cli::LogsHelper
+
+  def self.included(klass)
+    klass.send :include, MtnCloud::Cli::PrintHelper
+  end
+
+  def logs_interface
+    # @api_client.logs
+    raise "#{self.class} has not defined @logs_interface" if @logs_interface.nil?
+    @logs_interface
+  end
+
+  def instances_interface
+    # @api_client.instances
+    raise "#{self.class} has not defined @instances_interface" if @instances_interface.nil?
+    @instances_interface
+  end
+
+  def containers_interface
+    ## @api_client.containers
+    raise "#{self.class} has not defined @logs_interface" if @containers_interface.nil?
+    @containers_interface
+  end
+
+  def servers_interface
+    # @api_client.servers
+    raise "#{self.class} has not defined @servers_interface" if @servers_interface.nil?
+    @servers_interface
+  end
+
+  def clusters_interface
+    # @api_client.clusters
+    raise "#{self.class} has not defined @clusters_interface" if @clusters_interface.nil?
+    @clusters_interface
+  end
+
+  def format_log_records(log_records, options={}, show_object=true)
+    if options[:table]
+      return format_log_table(log_records, options, show_object)
+    end
+    out = ""
+    table_color = options.key?(:color) ? options[:color] : cyan
+    term_width = current_terminal_width()
+    message_col_width = current_terminal_width() - (show_object ? 56 : 36)
+    if options[:reverse]
+      log_records.reverse!
+    end
+    log_records.each do |log_entry|
+      log_level = format_log_level(log_entry['level'], table_color, 6)
+      out << table_color if table_color
+      # out << "[#{log_entry['ts']}] #{log_level} - #{log_entry['message'].to_s.strip}"
+      out << "#{log_level} "
+      out << "[#{log_entry['ts']}] "
+      if show_object
+        log_src = "#{log_entry['typeCode']} #{log_entry['objectId']}"
+        if options[:details] || options[:all]
+          # log_src = "#{log_entry['typeCode']} #{log_entry['objectId']}"
+        else
+          log_src = truncate_string(log_src, 20)
+        end
+        out << "(#{log_src}) "
+      end
+      log_msg = ""
+      if options[:details] || options[:all]
+        log_msg = log_entry['message'].to_s.strip
+      else
+        log_msg = truncate_string(log_entry['message'].to_s.strip.gsub(/\r?\n/, " "), message_col_width)
+      end
+      out << "#{log_msg}"
+      out << table_color if table_color
+      out << "\n"
+    end
+    return out
+  end
+
+  def format_log_table(logs, options={}, show_object=true)
+    out = ""
+    table_color = options.key?(:color) ? options[:color] : cyan
+    term_width = current_terminal_width()
+    message_col_width = current_terminal_width() - (show_object ? 56 : 36)
+    log_columns = [
+      {"LEVEL" => lambda {|log_entry| format_log_level(log_entry['level'], table_color) } },
+      {"DATE" => lambda {|log_entry| log_entry['ts'] } },
+      {"SOURCE" => lambda {|log_entry| "#{log_entry['typeCode']} #{log_entry['objectId']}" } },
+      {"MESSAGE" => lambda {|log_entry| 
+        if options[:details] || options[:all]
+          log_entry['message'].to_s.strip
+        else
+          truncate_string(log_entry['message'].to_s.strip.gsub(/\r?\n/, " "), message_col_width)
+        end
+      } }
+    ]
+    if show_object != true
+      log_columns = log_columns.reject {|it| it.key?("SOURCE") }
+    end
+    # if options[:include_fields]
+    #   columns = options[:include_fields]
+    # end
+    out << as_pretty_table(logs, log_columns, options.merge(wrap:true))
+    # out << "\n"
+    return out
+  end
+
+  def format_log_level(val, return_color=cyan, label_width=nil)
+    log_level = ''
+    display_value = val.to_s
+    if label_width
+      display_value = display_value.ljust(label_width, ' ')
+    end
+    case val
+    when 'INFO'
+      log_level = "#{blue}#{bold}#{display_value}#{reset}#{return_color}"
+    when 'DEBUG'
+      log_level = "#{white}#{bold}#{display_value}#{reset}#{return_color}"
+    when 'WARN'
+      log_level = "#{yellow}#{bold}#{display_value}#{reset}#{return_color}"
+    when 'ERROR'
+      log_level = "#{red}#{bold}#{display_value}#{reset}#{return_color}"
+    when 'FATAL'
+      log_level = "#{red}#{bold}#{display_value}#{reset}#{return_color}"
+    else
+      log_level = val
+    end
+    return log_level
+  end
+end