kpm 0.5.3 → 0.6.0

data/lib/kpm/system_helpers/cpu_information.rb

@@ -0,0 +1,72 @@
+module KPM
+  module SystemProxy
+    module CpuInformation
+      class << self
+        def fetch
+          cpu_info = nil
+          if OS.windows?
+            cpu_info = fetch_windows
+          elsif OS.linux?
+            cpu_info = fetch_linux
+          elsif OS.mac?
+            cpu_info = fetch_mac
+          end
+
+          cpu_info
+        end
+
+        def get_labels
+          labels = [{:label => :cpu_detail},
+                    {:label => :value}]
+          labels
+        end
+
+        private
+          def fetch_linux
+            cpu_data = `cat /proc/cpuinfo 2>&1`.gsub("\t",'')
+            cpu = get_hash(cpu_data)
+            cpu
+          end
+
+          def fetch_mac
+            cpu_data = `system_profiler SPHardwareDataType | grep -e "Processor" -e "Cores" -e "Cache" 2>&1`
+            cpu = get_hash(cpu_data)
+            cpu
+          end
+
+          def fetch_windows
+            cpu_name = `wmic cpu get Name`.split("\n\n")
+            cpu_caption = `wmic cpu get Caption`.split("\n\n")
+            cpu_max_clock_speed = `wmic cpu get MaxClockSpeed`.split("\n\n")
+            cpu_device_id = `wmic cpu get DeviceId`.split("\n\n")
+            cpu_status = `wmic cpu get Status`.split("\n\n")
+
+            cpu = Hash.new
+            cpu[cpu_name[0].to_s.strip] = {:cpu_detail => cpu_name[0].to_s.strip, :value => cpu_name[1].to_s.strip}
+            cpu[cpu_caption[0].to_s.strip] = {:cpu_detail => cpu_caption[0].to_s.strip, :value => cpu_caption[1].to_s.strip}
+            cpu[cpu_max_clock_speed[0].to_s.strip] = {:cpu_detail => cpu_max_clock_speed[0].to_s.strip, :value => cpu_max_clock_speed[1].to_s.strip}
+            cpu[cpu_device_id[0].to_s.strip] = {:cpu_detail => cpu_device_id[0].to_s.strip, :value => cpu_device_id[1].to_s.strip}
+            cpu[cpu_status[0].to_s.strip] = {:cpu_detail => cpu_status[0].to_s.strip, :value => cpu_status[1].to_s.strip}
+
+            cpu
+          end
+
+          def get_hash(data)
+            cpu = Hash.new
+
+            unless data.nil?
+              data.split("\n").each do |info|
+                infos = info.split(':')
+
+                unless infos[0].to_s.strip.eql?('flags')
+                  cpu[infos[0].to_s.strip] = {:cpu_detail => infos[0].to_s.strip, :value => infos[1].to_s.strip}
+                end
+              end
+            end
+
+            cpu
+          end
+        end
+    end
+  end
+end

data/lib/kpm/system_helpers/disk_space_information.rb

@@ -0,0 +1,84 @@
+module KPM
+  module SystemProxy
+    module DiskSpaceInformation
+      class << self
+
+        def fetch
+          disk_space_info = nil
+          if OS.windows?
+            disk_space_info = fetch_windows
+          elsif OS.linux?
+            disk_space_info = fetch_linux_mac(5)
+          elsif OS.mac?
+            disk_space_info = fetch_linux_mac(8)
+          end
+
+          disk_space_info
+        end
+
+        def get_labels
+          labels = []
+          @@data_keys.each { |key| labels.push({:label => key.gsub(' ','_').to_sym})}
+          labels
+        end
+
+        private
+          def fetch_linux_mac(cols_count)
+            disk_space_info = `df 2>&1`
+            disk_space = get_hash(disk_space_info,cols_count,true)
+            disk_space
+          end
+
+          def fetch_windows
+            disk_space_info = `wmic logicaldisk get size,freespace,caption 2>&1`
+            disk_space = get_hash(disk_space_info,3,false)
+            disk_space
+          end
+
+          def get_hash(data, cols_count, merge_last_two_columns)
+            disk_space = Hash.new
+
+            unless data.nil?
+
+              data_table = data.split("\n")
+
+              @@data_keys = data_table[0].split(' ')
+
+              if merge_last_two_columns
+                @@data_keys[@@data_keys.length - 2] = @@data_keys[@@data_keys.length - 2] + ' ' + @@data_keys[@@data_keys.length - 1]
+                @@data_keys.delete_at(@@data_keys.length - 1)
+              end
+
+              row_num = 0
+              data_table.each do |row|
+                cols = row.split(' ')
+                row_num += 1
+                unless cols[0].to_s.eql?(@@data_keys[0])
+                  key = 'DiskInfo_'+row_num.to_s
+                  disk_space[key] = Hash.new
+                  cols.each_index do |idx|
+                    if idx > cols_count
+                      break
+                    end
+
+                    value = cols[idx].to_s.strip
+                    if idx == cols_count && cols.length - 1 > idx
+                      for i in cols_count+1..cols.length
+                        value += ' ' + cols[i].to_s.strip
+                      end
+                    end
+
+                    disk_space[key][@@data_keys[idx].gsub(' ','_').to_sym] = value
+                  end
+                end
+
+              end
+            end
+
+            disk_space
+          end
+
+      end
+    end
+  end
+end

data/lib/kpm/system_helpers/entropy_available.rb

@@ -0,0 +1,52 @@
+module KPM
+  module SystemProxy
+    module EntropyAvailable
+      class << self
+
+        def fetch
+          entropy_available = nil
+          if OS.windows?
+            entropy_available = fetch_windows
+          elsif OS.linux?
+            entropy_available = fetch_linux
+          elsif OS.mac?
+            entropy_available = fetch_mac
+          end
+
+          entropy_available
+        end
+
+        def get_labels
+          labels = [{:label => :entropy},
+                    {:label => :value}]
+          labels
+        end
+
+        private
+          def fetch_linux
+            entropy_available_data = `cat /proc/sys/kernel/random/entropy_avail 2>&1`.gsub("\n",'')
+            entropy_available = get_hash(entropy_available_data)
+            entropy_available
+          end
+
+          def fetch_mac
+            entropy_available = get_hash('-')
+            entropy_available
+          end
+
+          def fetch_windows
+            entropy_available = get_hash('-')
+            entropy_available
+          end
+
+          def get_hash(data)
+            entropy_available = Hash.new
+            entropy_available['entropy_available'] = {:entropy => 'available', :value => data}
+
+            entropy_available
+          end
+
+      end
+    end
+  end
+end

data/lib/kpm/system_helpers/memory_information.rb

@@ -0,0 +1,71 @@
+module KPM
+  module SystemProxy
+    module MemoryInformation
+      class << self
+
+        def fetch
+          memory_info = nil
+          if OS.windows?
+            memory_info = fetch_windows
+          elsif OS.linux?
+            memory_info = fetch_linux
+          elsif OS.mac?
+            memory_info = fetch_mac
+          end
+
+          memory_info
+        end
+
+        def get_labels
+          labels = [{:label => :memory_detail},
+                    {:label => :value}]
+          labels
+        end
+
+        private
+          def fetch_linux
+            mem_data = `cat /proc/meminfo 2>&1`.gsub("\t",'')
+            mem = get_hash(mem_data)
+            mem
+          end
+
+          def fetch_mac
+            mem_data = `vm_stat 2>&1`.gsub(".",'')
+            mem = get_hash(mem_data)
+
+            mem.each_key do |key|
+              mem[key][:value] = ((mem[key][:value].to_i * 4096) / 1024 / 1024).to_s + 'MB'
+              mem[key][:memory_detail] = mem[key][:memory_detail].gsub('Pages','Memory')
+            end
+
+            mem_total_data = `system_profiler SPHardwareDataType | grep "  Memory:" 2>&1`
+            mem_total = get_hash(mem_total_data)
+
+            mem = mem_total.merge(mem)
+
+            mem
+          end
+
+          def fetch_windows
+            mem_data = `systeminfo | findstr /C:"Total Physical Memory" /C:"Available Physical Memory"`
+            mem = get_hash(mem_data)
+            mem
+          end
+
+          def get_hash(data)
+            mem = Hash.new
+
+            unless data.nil?
+              data.split("\n").each do |info|
+                infos = info.split(':')
+                mem[infos[0].to_s.strip] = {:memory_detail => infos[0].to_s.strip, :value => infos[1].to_s.strip}
+              end
+            end
+
+            mem
+          end
+
+      end
+    end
+  end
+end

data/lib/kpm/system_helpers/os_information.rb

@@ -0,0 +1,66 @@
+module KPM
+  module SystemProxy
+    module OsInformation
+      class << self
+
+        def fetch
+          os_information  = nil
+          if OS.windows?
+            os_information  = fetch_windows
+          elsif OS.linux?
+            os_information  = fetch_linux
+          elsif OS.mac?
+            os_information  = fetch_mac
+          end
+
+          os_information
+        end
+
+        def get_labels
+          labels = [{:label => :os_detail},
+                    {:label => :value}]
+          labels
+        end
+
+        private
+          def fetch_linux
+            os_data = `lsb_release -a 2>&1`
+
+            if os_data.nil? || os_data.include?('lsb_release: not found')
+              os_data = `cat /etc/issue 2>&1`
+              os_data = 'Description:'+os_data.gsub('\n \l','')
+            end
+
+            os = get_hash(os_data)
+            os
+          end
+
+          def fetch_mac
+            os_data = `sw_vers`
+            os = get_hash(os_data)
+            os
+          end
+
+          def fetch_windows
+            os_data = `systeminfo | findstr /C:"OS"`
+            os = get_hash(os_data)
+            os
+          end
+
+          def get_hash(data)
+            os = Hash.new
+
+            unless data.nil?
+              data.split("\n").each do |info|
+                infos = info.split(':')
+                os[infos[0].to_s.strip] = {:os_detail => infos[0].to_s.strip, :value => infos[1].to_s.strip}
+              end
+            end
+
+            os
+          end
+
+      end
+    end
+  end
+end

data/lib/kpm/system_helpers/system_proxy.rb

@@ -0,0 +1,28 @@
+require_relative 'cpu_information'
+require_relative 'memory_information'
+require_relative 'disk_space_information'
+require_relative 'entropy_available'
+require_relative 'os_information'
+module KPM
+  module SystemProxy
+
+    module OS
+      def OS.windows?
+        (/cygwin|mswin|mingw|bccwin|wince|emx/ =~ RbConfig::CONFIG["host_os"]) != nil
+      end
+
+      def OS.mac?
+        (/darwin/ =~ RbConfig::CONFIG["host_os"]) != nil
+      end
+
+      def OS.unix?
+        !OS.windows?
+      end
+
+      def OS.linux?
+        OS.unix? and not OS.mac?
+      end
+    end
+
+  end
+end

data/lib/kpm/tasks.rb

@@ -38,7 +38,8 @@ module KPM
         desc 'install config_file', 'Install Kill Bill server and plugins according to the specified YAML configuration file.'
         def install(config_file=nil)
           help = Installer.from_file(config_file).install(options[:force_download], options[:verify_sha1])
-          say help, :green unless help.nil?
+          help = JSON(help)
+          say help['help'], :green unless help['help'].nil?
         end
 
         method_option :destination,
@@ -428,8 +429,13 @@ module KPM
         desc 'system', 'Gather information about the system'
         def system
           system = KPM::System.new
-          system.information(options[:bundles_dir], options[:as_json], options[:config_file], options[:kaui_web_path],
+          information = system.information(options[:bundles_dir], options[:as_json], options[:config_file], options[:kaui_web_path],
                              options[:killbill_web_path])
+
+          if options[:as_json]
+            puts information
+          end
+
         end
 
         method_option :export,
@@ -593,6 +599,99 @@ module KPM
         end
 
 
+        method_option :account_export,
+                      :type    => :string,
+                      :default => nil,
+                      :desc    => 'export account for a provided id.'
+        method_option :log_dir,
+                      :type    => :string,
+                      :default => nil,
+                      :desc    => '(Optional) Log directory if the default tomcat location has changed'
+        method_option :config_file,
+                      :type    => :string,
+                      :default => nil,
+                      :desc    => 'Yml that contains killbill api connection and DB connection'
+        method_option :killbill_api_credentials,
+                      :type    => :array,
+                      :default => nil,
+                      :desc    => 'Killbill api credentials <api_key> <api_secrets>'
+        method_option :killbill_credentials,
+                      :type    => :array,
+                      :default => nil,
+                      :desc    => 'Killbill credentials <user> <password>'
+        method_option :killbill_url,
+                      :type    => :string,
+                      :default => nil,
+                      :desc    => 'Killbill URL ex. http://127.0.0.1:8080'
+        method_option :database_name,
+                      :type    => :string,
+                      :default => nil,
+                      :desc    => 'DB name to connect'
+        method_option :database_credentials,
+                      :type    => :array,
+                      :default => nil,
+                      :desc    => 'DB credentials <user> <password>'
+        method_option :database_host,
+                      :type    => :string,
+                      :default => nil,
+                      :desc    => 'Database Host name'
+        method_option :kaui_web_path,
+                      :type    => :string,
+                      :default => nil,
+                      :desc    => 'Path for the KAUI web app'
+        method_option :killbill_web_path,
+                      :type    => :string,
+                      :default => nil,
+                      :desc    => 'Path for the killbill web app'
+        desc 'diagnostic', 'exports and \'zips\' the account data, system, logs and tenant configurations'
+        def diagnostic
+          logger.info 'Please wait processing the request!!!'
+          begin
+            if options[:account_export] && options[:account_export] == 'account_export'
+              raise Interrupt,'--account_export,  please provide a valid account id'
+            end
+
+            if options[:killbill_url] && /https?:\/\/[\S]+/.match(options[:killbill_url]).nil?
+              raise Interrupt,'--killbill_url, required format -> http(s)://something'
+            end
+
+            if options[:killbill_api_credentials] && options[:killbill_api_credentials].size != 2
+              raise Interrupt,'--killbill_api_credentials, required format -> <api_key> <api_secrets>'
+            end
+
+            if options[:killbill_credentials] && options[:killbill_credentials].size != 2
+              raise Interrupt,'--killbill_credentials, required format -> <user> <password>'
+            end
+
+            if options[:database_credentials] && options[:database_credentials].size != 2
+              raise Interrupt,'--database_credentials, required format -> <user> <password>'
+            end
+
+            if options[:database_name] && options[:database_name] == :database_name.to_s
+              raise Interrupt,'--database_credentials, please provide a valid database name'
+            end
+
+            if options[:kaui_web_path] && options[:kaui_web_path] == :kaui_web_path.to_s
+              raise Interrupt,'--kaui_web_path, please provide a valid kaui web path '
+            end
+
+            if options[:killbill_web_path] && options[:killbill_web_path] == :killbill_web_path.to_s
+              raise Interrupt,'--killbill_web_path, please provide a valid killbill web path'
+            end
+
+            diagnostic = KPM::DiagnosticFile.new(options[:config_file],options[:killbill_api_credentials],options[:killbill_credentials],
+                                       options[:killbill_url],options[:database_name],options[:database_credentials],
+                                      options[:database_host], options[:kaui_web_path], options[:killbill_web_path],logger)
+            diagnostic.export_data(options[:account_export],  options[:log_dir])
+
+          rescue Exception => e
+            logger.error "\e[91;1m#{e.message}\e[0m"
+            if not e.is_a?(Interrupt)
+              logger.error e.backtrace.join("\n")
+            end
+          end
+        end
+
         map :pull_ruby_plugin => :install_ruby_plugin,
             :pull_java_plugin => :install_java_plugin