greenhat 0.3.4 → 0.5.0

data/lib/greenhat/host.rb

@@ -14,17 +14,23 @@ class Host < Teron
   # end
 
   def archive_name
-    File.basename archive
+    archive.friendly_name
   end
 
   def find_file(file_name)
     files.find { |x| x.name == file_name }
   end
 
-  def icon
-    release_file = find_file files.map(&:name).grep(/_release/).first
+  def things
+    archive.things
+  end
 
-    release = release_file.raw.join
+  def find_thing(param)
+    things.find { |x| x.name.include? param }
+  end
+
+  def icon
+    release = find_thing('etc/os-release').data['ID_LIKE']
 
     if release.include? 'ubuntu'
       'fa-ubuntu'
@@ -57,41 +63,36 @@ class Host < Teron
   end
   # ---------------------------
 
-  def manifest
-    file = find_file 'gitlab_version_manifest_json'
+  def gitlab_version
+    file = find_thing 'gitlab/version-manifest.json'
     return nil unless file
 
-    Oj.load file.raw.join
+    file.data.build_version
   end
 
   def uptime
-    file = find_file 'uptime'
+    file = find_thing 'uptime'
     return nil unless file
 
     file.raw
   end
 
   def uname
-    file = find_file 'uname'
+    file = find_thing 'uname'
     return nil unless file
 
     file.raw.join
   end
 
   def cpuinfo
-    file = find_file 'cpuinfo'
+    file = find_thing 'lscpu'
     return nil unless file
 
-    file.raw.join("\n").split("\n\n").map do |cpu|
-      all = cpu.split("\n").map do |row|
-        row.delete("\t").split(': ')
-      end
-      { details: all[1..], order: all[0].last }
-    end
+    file.data
   end
 
   def cpu_speed
-    file = find_file 'lscpu'
+    file = find_thing 'lscpu'
     return nil unless file
 
     details = file.raw.find { |x| x.include? 'MHz' }
@@ -99,7 +100,7 @@ class Host < Teron
   end
 
   def total_memory
-    file = find_file 'free_m'
+    file = find_thing 'free_m'
     return nil unless file
 
     value = file.raw.dig(1, 1).to_i
@@ -107,28 +108,28 @@ class Host < Teron
   end
 
   def free_m
-    file = find_file 'free_m'
+    file = find_thing 'free_m'
     return nil unless file
 
     file.raw
   end
 
   def df_h
-    file = find_file 'df_h'
+    file = find_thing 'df_h'
     return nil unless file
 
     file.raw
   end
 
   def netstat
-    file = find_file 'netstat'
+    file = find_thing 'netstat'
     return nil unless file
 
     file.raw
   end
 
   def ulimit
-    file = find_file 'ulimit'
+    file = find_thing 'ulimit'
     return nil unless file
 
     results = file.raw.map do |entry|
@@ -142,7 +143,7 @@ class Host < Teron
   end
 
   def processes
-    file = find_file 'ps'
+    file = find_thing 'ps'
     return nil unless file
 
     headers = file.raw.first.split(' ', 11)
@@ -155,7 +156,7 @@ class Host < Teron
   end
 
   def systemctl_unit_files
-    file = find_file 'systemctl_unit_files'
+    file = find_thing 'systemctl_unit_files'
     return nil unless file
 
     all = file.raw[1..-2].map do |x|
@@ -167,16 +168,3 @@ class Host < Teron
     all.sort_by(&:unit)
   end
 end
-
-# # Slim Wrapper
-# class List
-#   attr_accessor :hosts
-
-#   def initialize
-#     self.hosts = []
-#   end
-
-#   def slim(file, host = nil)
-#     Slim::Template.new("#{__dir__}/views/#{file}.slim").render(host)
-#   end
-# end

data/lib/greenhat/settings.rb

@@ -3,22 +3,50 @@ module GreenHat
   # Helper for all things user environment / settings / history
   module Settings
     def self.settings
+      # round: [2],
+      # page: [:true] Automatic,
       @settings ||= {
-        history: [],
         assume_json: true,
         fuzzy_file_match: true,
-
-        # round: [2],
-        # page: [:true] Automatic,
-        truncate: TTY::Screen.width * 4,
+        assume_raw: false,
+        truncate: TTY::Screen.width * 4, # Dynamic Setting
         color: true
       }
     end
 
+    def self.configure_choices
+      [
+
+        { name: 'Assume JSON - auto injest json', value: :assume_json },
+        { name: 'Assume Raw - do not attempt unknown parsing', value: :assume_raw },
+        { name: 'Fuzzy File Match - all partial matching', value: :fuzzy_file_match },
+        { name: 'Color Output', value: :color }
+      ]
+    end
+
+    def self.configure_settings
+      puts 'Current Settings'.pastel(:green)
+      ap @settings.except(:truncate)
+      puts
+
+      prompt = TTY::Prompt.new
+      choice = prompt.select('Change Which Setting?', configure_choices, filter: true)
+
+      setting = configure_choices.find { |x| x.value == choice }
+      value = prompt.yes?("#{setting[:name]} (Y/n)")
+      @settings[choice] = value
+
+      File.write(settings_file, Oj.dump(@settings.except(:truncate)))
+    end
+
     def self.assume_json?
       settings.assume_json
     end
 
+    def self.assume_raw?
+      settings.assume_raw
+    end
+
     # Allow for future disabling of color output
     def self.color?
       settings.color

data/lib/greenhat/shell/args.rb

@@ -1,12 +1,5 @@
 module GreenHat
   # Module for Argument Parsing
-  # Possible Args
-  # --value=thing
-  # --value / Default => --value=thing
-  # --flag
-  # --slice=value,value2
-  # files
-
   # Variable Breakdown
   # Args:Array supplied or default values for entry looping action
   # Flags:Hash Key/Value filter Actionables
@@ -15,7 +8,14 @@ module GreenHat
     def self.parse(raw, skip_flags = [])
       # Don't affect original object / Better deep clone?
       cmd = raw.clone
-      # cmd = Marshal.load(Marshal.dump(raw))
+
+      # Handle Pipe Params
+      pipe = nil
+      if cmd.include?('|')
+        list = cmd.split('|')
+        cmd = list.shift
+        pipe = list.map { |x| x.join(' ') }.join('|')
+      end
 
       # Extract Files
       files = cmd.grep_v(/^-+([^=]+)/)
@@ -33,6 +33,9 @@ module GreenHat
       # And / Or
       flag_defaults(flags, skip_flags)
 
+      # Set command includes
+      flags[:pipe] = pipe if pipe
+
       [files, flags, args]
     end
 
@@ -78,19 +81,29 @@ module GreenHat
     # Arg Scan (Split -- values into keys)
     def self.arg_scan(arg)
       arg.scan(/^-+([^=]+)=(.*)/).map do |field, value|
+        logic = :include?
         bang = false
+
+        # Exclude Logic
         if field.include? '!'
           field.delete!('!')
           bang = true
         end
 
+        # Numeric Magic Logic
+        logic = :<= if field.include? '<'
+        logic = :>= if field.include? '>'
+        field.delete!('<')
+        field.delete!('>')
+
         # Symbolize
         field = field.to_sym
 
         {
           field: field,
           value: arg_normalize(field, value),
-          bang: bang
+          bang: bang,
+          logic: logic
         }
       end
     end

data/lib/greenhat/shell/faststats.rb

@@ -137,7 +137,7 @@ module GreenHat
       # --limit=X
       # --sort=count,fail,max,median,min,p95,p99,rps,score
       # ========================================================================
-      def self.top(raw)
+      def self.top(raw = [], internal = false)
         unless TTY::Which.exist? 'fast-stats'
           faststats_installation
           return false
@@ -155,6 +155,9 @@ module GreenHat
           ]
         end
 
+        # Quick exit for internal processing
+        return results.flatten if internal
+
         ShellHelper.show(results.flatten, flags)
       end
       # ========================================================================
@@ -162,14 +165,14 @@ module GreenHat
       # ========================================================================
       # ===== [ Fast Stats - Errors ] ====================
       # ========================================================================
-      def self.errors(raw)
+      def self.errors(raw = [], internal = false)
         unless TTY::Which.exist? 'fast-stats'
           faststats_installation
           return false
         end
 
         # Add Color Output
-        raw.push '--color-output'
+        raw.push '--color-output' if Settings.settings.color?
         files, flags, cmd = ShellHelper::Faststats.parse(raw)
 
         LogBot.debug('FastStats CMD', cmd) if ENV['DEBUG']
@@ -182,6 +185,9 @@ module GreenHat
           ]
         end
 
+        # Quick exit for internal processing
+        return results.flatten if internal
+
         ShellHelper.show(results.flatten, flags)
       end
       # ========================================================================
@@ -236,6 +242,20 @@ module GreenHat
           files.any? { |x| thing.name.include? x }
         end
       end
+
+      # =====================================
+      # Internal Helpers
+      # =====================================
+      def self.faststats?
+        TTY::Which.exist?('fast-stats')
+      end
+
+      def self.run(thing, cmd = '')
+        # Blank return if cannot possibly run
+        return {} unless thing || !faststats?
+
+        Oj.load `fast-stats #{cmd} #{thing.file} --format=json`
+      end
     end
   end
 end

data/lib/greenhat/shell/field_helper.rb

@@ -60,7 +60,7 @@ module GreenHat
       auto = fields.split(',').last
 
       # Field Finder
-      matches = fields_find(files, auto, flags)
+      matches = fields_find(files, auto, flags).uniq
 
       if matches.count == 1
         "--#{cmd}=#{(complete + matches).join(',')}"