greenhat 0.3.3 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ac7b0e70e5a13dc6f8f88a4f12fec812e38e7d838537abe83b7d4dc2e8b43663
-  data.tar.gz: 84246bc31bd38af74e9f4f8d95f2f7ffa1ca33978dc0d0862ec0b50eaff20196
+  metadata.gz: 4205de8b256423c08291a212511d4d6ecd613d21f232d0d6b4641881779f1303
+  data.tar.gz: 695ce3fb852d8a3167b853b3b8c07448f031305e51072f34e7afb3aed75ca50e
 SHA512:
-  metadata.gz: ba104ea8d7b4e52babf8bdf27ea4f3dbdb9e92820589881bab639a01e8558c355389a0f10b3c17dbe5c7ac881f10364ad06ec0c973934f06d735f948bb57fbd3
-  data.tar.gz: a2013229826758ca61335f5cf8d58c3a4a4206e656dce084f90781c264e4930c5c3fd09b7475a43dd31819bc2562e85e5799de775cc5db9dfaa3f73c0228ac3a
+  metadata.gz: 6d4653882946a783ddc0e11094205e42b33338afe53334cc9774956047a1b00eb54bca964925efbed47ca4c410e5007d6e5ac79f10376274b50beb1ffaedc8f0
+  data.tar.gz: c2002b0458b334799e4b04a446a9ca0193091f264e7077b9eac84c310423e6c0547d79094da45fa5fa9ce475d110756952602d3274b34f97b4162e7ee4226871

data/bin/greenhat

@@ -2,11 +2,8 @@
 
 require 'greenhat'
 
-# DEV ONLY
-require 'pry'
-
 Dir.mktmpdir('greenhat-sauce') do |tmp|
   $TMP = tmp
   $STDOUT_CLONE = $stdout.clone
-  GreenHat::Cli.start ARGV
+  GreenHat::Entrypoint.start ARGV
 end

data/lib/greenhat/accessors/disk.rb

@@ -30,7 +30,6 @@ module GreenHat
     end
 
     # Unified Output Handler
-    # rubocop:disable Metrics/MethodLength
     def self.format_output(file, name = false, limit = nil, filter = %w[tmpfs loop])
       output = []
 
@@ -62,7 +61,7 @@ module GreenHat
           disk.use.rjust(5).ljust(5).pastel(:green),
           '  ['.pastel(:blue),
           ('=' * (disk.use.to_i / 2)).pastel(:green),
-          ' ' * (50 - disk.use.to_i / 2),
+          ' ' * (50 - (disk.use.to_i / 2)),
           ']'.pastel(:blue)
         ].join
 
@@ -78,6 +77,46 @@ module GreenHat
 
       output
     end
-    # rubocop:enable Metrics/MethodLength
+
+    # Unified Output Handler
+    def self.markdown_format(file, name = false, limit = nil, filter = %w[tmpfs loop])
+      output = []
+
+      output << file.friendly_name if name
+
+      # Reject TMPFS
+      disks = file.data.sort_by { |x| x.use.to_i }.reverse
+
+      # Filter
+      disks.reject! { |x| filter.any? { |y| x.filesystem.include? y } }
+
+      disks = disks[0..limit - 1] if limit
+
+      pad_mount, pad_size, pad_used, pad_avail = GreenHat::Disk.padding(disks)
+
+      # Headers
+      output << [
+        'Mount'.ljust(pad_mount),
+        'Size'.ljust(pad_size),
+        'Used'.ljust(pad_used),
+        'Avail'.ljust(pad_avail),
+        '% Use'.ljust(pad_avail)
+      ].join
+
+      # Table Summary
+      disks.map do |disk|
+        # Whole Thing
+        output << [
+          disk.mounted_on.ljust(pad_mount),
+          disk[:size].to_s.ljust(pad_size),
+          disk.used.to_s.ljust(pad_used),
+          disk.avail.to_s.ljust(pad_avail),
+          disk.use.rjust(5).ljust(5)
+        ].join
+      end
+
+      output
+    end
+    # =
   end
 end

data/lib/greenhat/accessors/gitlab.rb

@@ -35,6 +35,7 @@ module GreenHat
     end
 
     # Show GitLab Services in a grid / include versions
+    # rubocop:disable Metrics/PerceivedComplexity
     def self.services(archive, indent = 0)
       manifest = archive.things.find { |x| x.type == 'gitlab/version-manifest.json' }
       gitlab_status = archive.things.find { |x| x.name == 'gitlab_status' }
@@ -43,6 +44,7 @@ module GreenHat
 
       list = gitlab_status.data.keys.sort.map do |service|
         color = gitlab_status.data.dig(service, 0, :status) == 'run' ? :green : :red
+        status = gitlab_status.data.dig(service, 0, :status) == 'run' ? '↑' : '↓'
 
         # Collect Service version from manifest
         version = manifest.data.software[service.to_sym]&.display_version
@@ -50,7 +52,7 @@ module GreenHat
         # If able to identify version use / fallback
         if version
           [
-            service.pastel(color),
+            "#{service}#{status}".pastel(color),
             "(#{version})".pastel(:bright_black)
           ].join(' ')
         else
@@ -71,5 +73,32 @@ module GreenHat
 
       table.render(:unicode, padding: [0, 1, 0, 1], indent: indent)
     end
+    # rubocop:enable Metrics/PerceivedComplexity
+
+    def self.services_markdown(archive)
+      # manifest = archive.things.find { |x| x.type == 'gitlab/version-manifest.json' }
+      gitlab_status = archive.things.find { |x| x.name == 'gitlab_status' }
+
+      list = gitlab_status.data.keys.sort.map do |service|
+        status = gitlab_status.data.dig(service, 0, :status) == 'run' ? '↑' : '↓'
+
+        "#{service}#{status}"
+      end
+
+      # Keep Alphabetical Sort
+      groups = list.each_slice((list.size / 3.to_f).round).to_a
+
+      table = TTY::Table.new do |t|
+        loop do
+          break if groups.all?(&:empty?)
+
+          t << groups.map(&:shift)
+        end
+      end
+
+      "```\n#{table.render(:basic)}\n```"
+    end
+
+    # ==
   end
 end

data/lib/greenhat/accessors/memory.rb

@@ -6,7 +6,7 @@ module GreenHat
     end
 
     def self.percentage(used, total)
-      return 0 if used.to_i.zero?
+      return 0 if used.to_i.zero? || total.to_i.zero?
 
       # Show at least one bar if below 1%
       [1, ((used.to_i / total.to_f).round(1) * 100).round].max

data/lib/greenhat/archive.rb

@@ -7,7 +7,7 @@ module GreenHat
       Dir.mkdir path
 
       # Initially Copy file into tmpdir
-      FileUtils.cp(archive_path, "#{path}/")
+      FileUtils.cp_r(archive_path.to_s, "#{path}/")
 
       # Extract Everything
       loop do
@@ -28,9 +28,6 @@ module GreenHat
 
       archive = Archive.new(name: archive_path, path: path)
       archive.save
-      # file = list[2]
-      # thing = archive.things_create(file: file)
-      # thing.setup
 
       list.each do |file|
         # Ensure Valid Content
@@ -54,8 +51,13 @@ module GreenHat
     # Handle Different Types of Archives
     def self.unpack(archive_path, path)
       case File.extname archive_path
+      when '.tgz'
+        base_path = archive_path.gsub(File.basename(archive_path), '')
+        `bsdtar -xzf "#{archive_path}" -C #{base_path}`
+        FileUtils.rm(archive_path)
       when '.tar'
-        `bsdtar -xf "#{archive_path}" -C #{path}`
+        base_path = archive_path.gsub(File.basename(archive_path), '')
+        `bsdtar -xf "#{archive_path}" -C #{base_path}`
         FileUtils.rm(archive_path)
       when '.gz'
         `gzip -d "#{archive_path}"`
@@ -75,7 +77,7 @@ module GreenHat
     end
 
     def self.archive_types
-      %w[s gz tar]
+      %w[s gz tar tgz]
     end
     # -- Archive Loader ----------------------------------------------
   end
@@ -102,7 +104,16 @@ class Archive < Teron
     "#<Archive name: '#{name}'>"
   end
 
-  def report
-    GreenHat::Report.new(self)
+  def report(flags)
+    GreenHat::Report.new(self, flags)
+  end
+
+  def report_markdown
+    GreenHat::ReportMarkdown.new(self)
+  end
+
+  # Helper for finding thing
+  def thing?(thing_name = 'gitlab-workhorse/current')
+    things.find { |x| x.name == thing_name }
   end
 end

data/lib/greenhat/cli.rb

@@ -1,8 +1,31 @@
 # Top level namespace
 module GreenHat
-  # CLI Methods
+  # CLI Methods / Shell Specific COmmands
   # rubocop:disable Metrics/ModuleLength
   module Cli
+    # CLI Start
+    def self.start
+      value ||= '' # Empty Start
+
+      loop do
+        line = reader.read_line(readline_notch, value: value)
+        value = '' # Remove Afterwards
+
+        if reader.breaker
+          value = line
+          puts ''
+          next
+        end
+
+        break if line =~ /^exit/i
+
+        Settings.cmd_add(line) unless line.blank?
+
+        process
+        run
+      end
+    end
+
     def self.cursor
       @cursor = TTY::Cursor
     end
@@ -348,117 +371,6 @@ module GreenHat
       @quiet = !@quiet
     end
 
-    def self.cli_help
-      Shell.version
-      puts
-      puts 'Usage'.pastel(:yellow)
-      puts '  greenhat  <sos-archive.tgz> <sos-archive2.tgz> '
-      puts '  greenhat  <sos-archive.tgz> -q --command=df'
-      puts
-
-      puts 'Options'.pastel(:yellow)
-      puts '  --report, -r'.pastel(:green)
-      puts '    Run `report` against archives and exit'
-      puts
-
-      puts '  --quiet, -r'.pastel(:green)
-      puts '    Surpress GreenHat logging output'
-      puts
-
-      puts '  --load, -l'.pastel(:green)
-      puts '    Automatically attempt to read/parse/preload all included files'
-      puts
-
-      puts '  --command, -c'.pastel(:green)
-      puts '    Run and then exit a GreenHat Shell command'
-      puts
-
-      puts '  --version, -v'.pastel(:green)
-      puts '    Print version and exit'
-      puts
-    end
-
-    # Arguments before general processing
-    def self.pre_args(flags, files)
-      # Help
-      if flags?(%i[help h], flags)
-        cli_help
-        exit 0
-      end
-
-      # Version
-      if flags?(%i[version v], flags)
-        Shell.version
-        exit 0
-      end
-
-      # Quiet Flag
-      @quiet = true if flags?(%i[quiet q], flags)
-
-      # rubocop:disable Style/GuardClause
-      # MIA Files
-      if files.empty? || files.count { |x| File.exist? x }.zero?
-        puts "No arguments or files don't exist".pastel(:red)
-        puts 'Usage: greenhat  <sos-archive.tgz> <sos-archive2.tgz>'
-        cli_help
-        Shell.version
-      end
-      # rubocop:enable Style/GuardClause
-    end
-
-    # Arguments to be handled after general processing
-    def self.post_args(flags)
-      # Run report and exit / Don't Clear for reports
-      if flags?(%i[report r], flags)
-        @quiet = true
-        # Don't Use Pagination
-        GreenHat::Shell.report(['--raw'])
-        exit 0
-      else
-        clear_screen
-      end
-
-      # CTL Tails need to be parsed for new 'things'
-      Thing.where(kind: :gitlab_tail)&.map(&:process)
-
-      Thing.all.each(&:process) if flags?(%i[load l], flags)
-    end
-
-    # Helper to Simplify checking flags
-    def self.flags?(list = [], flags = {})
-      list.any? { |x| flags.key? x }
-    end
-
-    # If no arguments Supplied Print and quit - rather than nasty exception
-    def self.start(raw)
-      Settings.start
-      files, flags, args = Args.parse(raw)
-      pre_args(flags, files)
-      load_files files
-      run_command(args) if args.any? { |arg| run_command?(arg) }
-      post_args(flags)
-
-      value ||= '' # Empty Start
-
-      loop do
-        line = reader.read_line(readline_notch, value: value)
-        value = '' # Remove Afterwards
-
-        if reader.breaker
-          value = line
-          puts ''
-          next
-        end
-
-        break if line =~ /^exit/i
-
-        Settings.cmd_add(line) unless line.blank?
-
-        process
-        run
-      end
-    end
-
     def self.run_command?(arg)
       %i[command c].any? do |x|
         arg[:field] == x
@@ -478,20 +390,6 @@ module GreenHat
       exit 0
     end
 
-    def self.load_files(files)
-      # TODO: Web Helpers?
-      # suppress_output { GreenHat::Web.start }
-
-      # Don't double up on archives / Only Existing files
-      puts 'Loading Archives'.pastel(:blue) unless Cli.quiet
-      files.uniq.each do |file|
-        next unless File.exist?(file)
-
-        puts "- #{file}".pastel(:magenta) unless Cli.quiet
-        ArchiveLoader.load file
-      end
-    end
-
     def self.suppress_output
       original_stderr = $stderr.clone
       original_stdout = $stdout.clone

data/lib/greenhat/entrypoint.rb

@@ -0,0 +1,170 @@
+# Top level namespace
+module GreenHat
+  # Entrypoint Methods
+  # Command Line Args / File Parsing etc
+  module Entrypoint
+    # If no arguments Supplied Print and quit - rather than nasty exception
+    def self.start(raw)
+      @clear = true # Reset screen by default
+      Settings.start
+      files, flags, args = Args.parse(raw)
+      pre_args(flags, files)
+      load_files files
+      Cli.run_command(args) if args.any? { |arg| Cli.run_command?(arg) }
+      post_args(flags)
+      post_setup(flags)
+
+      Cli.start # Kick over to CLI
+    end
+
+    # Arguments before general processing
+    def self.pre_args(flags, files)
+      # Help
+      if flags?(%i[help h], flags)
+        cli_help
+        exit 0
+      end
+
+      # Version
+      if flags?(%i[version v], flags)
+        Shell.version
+        exit 0
+      end
+
+      # Assume Raw Setting
+      Settings.settings[:assume_raw] = true if flags?(%i[no-prompt np], flags)
+      Settings.settings[:assume_raw] = false if flags?(%i[prompt p], flags)
+
+      # Quiet Flag
+      @quiet = true if flags?(%i[quiet q], flags)
+
+      if flags?(%i[load-local ll], flags)
+        # Hook in loading local
+        Shell.load_local(false)
+      elsif files.empty? || files.count { |x| File.exist? x }.zero?
+        # MIA Files
+        puts "No arguments or files don't exist".pastel(:red)
+        puts 'Usage: greenhat  <sos-archive.tgz> <sos-archive2.tgz>'
+        puts 'See `--help`'
+
+        @clear = false # Don't hide messages
+
+        Shell.version
+      end
+    end
+
+    # Arguments to be handled after general processing
+    def self.post_args(flags)
+      # Supress Color
+      GreenHat::Settings.settings[:color] = false if flags?(%i[no-color n], flags)
+
+      # Run report and exit / Don't Clear for reports
+      if flags?(%i[report r], flags)
+        @quiet = true
+        # Don't Use Pagination
+        GreenHat::Shell.report(['--raw'])
+        exit 0
+      elsif flags?(%i[full-report f], flags)
+        @quiet = true
+        # Don't Use Pagination
+        GreenHat::Shell.report(['--raw', '--full'])
+        exit 0
+      elsif flags?(%i[markdown m], flags)
+        @quiet = true
+        # Don't Use Pagination
+        GreenHat::Shell.markdown_report(['--raw'])
+        exit 0
+      elsif @clear
+        Cli.clear_screen
+      end
+    end
+
+    def self.post_setup(flags)
+      # CTL Tails need to be parsed for new 'things'
+      Thing.where(kind: :gitlab_tail)&.map(&:process)
+      Thing.where(kind: :kube_webservice)&.map(&:process)
+
+      Thing.all.each(&:process) if flags?(%i[load l], flags)
+
+      Shell.web if flags?(%i[web w], flags)
+    end
+
+    # Helper to Simplify checking flags
+    def self.flags?(list = [], flags = {})
+      list.any? { |x| flags.key? x }
+    end
+
+    def self.load_files(files)
+      # Don't double up on archives / Only Existing files
+      puts 'Loading Archives'.pastel(:blue) unless Cli.quiet
+      files.uniq.each do |file|
+        next unless File.exist?(file)
+
+        puts "- #{file}".pastel(:magenta) unless Cli.quiet
+        ArchiveLoader.load file
+      end
+    end
+
+    # rubocop:disable Metrics/MethodLength
+    def self.cli_help
+      Shell.version
+      puts
+      puts 'Usage'.pastel(:yellow)
+      puts '  greenhat  <sos-archive.tgz> <sos-archive2.tgz> '
+      puts '  greenhat  <sos-archive.tgz> -q --command=df'
+      puts
+
+      puts 'Options'.pastel(:yellow)
+      puts '  --report, -r'.pastel(:green)
+      puts '    Run `report` against archives and exit'
+      puts
+
+      puts '  --full-report, -f'.pastel(:green)
+      puts '    Run `report` with all details. e.g. include fast-stats top/errors'
+      puts
+
+      puts '  --markdown, -m'.pastel(:green)
+      puts '    Run `markdown_report` against archives and exit'
+      puts
+
+      puts '  --quiet, -r'.pastel(:green)
+      puts '    Surpress GreenHat logging output'
+      puts
+
+      puts '  --load, -l'.pastel(:green)
+      puts '    Automatically attempt to read/parse/preload all included files'
+      puts
+
+      puts '  --load-local, -ll'.pastel(:green)
+      puts '    Load log files from locations on an Omnibus install'
+      puts
+
+      puts '  --prompt, -p'.pastel(:green)
+      puts '    Override settings / Assume Prompt'
+      puts
+
+      puts '  --no-prompt, -np'.pastel(:green)
+      puts '    Do not prompt for file types / Assume Raw or JSON'
+      puts
+
+      puts '  --command, -c'.pastel(:green)
+      puts '    Run and then exit a GreenHat Shell command'
+      puts
+
+      puts '  --web, -w'.pastel(:green)
+      puts '    Start Sinatra Webservice on load (4567)'
+      puts
+
+      puts '  --no-color, -n'.pastel(:green)
+      puts '    Disable color output'
+      puts
+
+      puts '  --version, -v'.pastel(:green)
+      puts '    Print version and exit'
+      puts
+    end
+    # rubocop:enable Metrics/MethodLength
+
+    # ==========================================================
+  end
+end