greenhat 0.1.4

data/lib/greenhat/logbot.rb

@@ -0,0 +1,86 @@
+# Log Helper
+module LogBot
+  # For Amazing Print
+  def self.settings
+    { ruby19_syntax: true, multiline: false }
+  end
+
+  def self.output(name, message, level, duration = nil)
+    time = Time.now.to_s.colorize(:light_black)
+    output = time
+    output += "  #{color_level(level)}"
+    output += " - #{name.to_s.colorize(:light_blue)} "
+    output += "(#{duration.round(2)}) ".colorize(:magenta) if duration
+
+    output += ' '
+    if message
+      output += case message
+                when String then message
+                else
+                  message.ai(settings)
+                end
+    end
+
+    puts output unless ENV['TESTING']
+  end
+
+  def self.color_level(level)
+    case level
+    when :debug
+      level.to_s.upcase.colorize(:light_cyan)
+    when :info
+      level.to_s.upcase.colorize(:cyan)
+    when :warn
+      level.to_s.upcase.colorize(:yellow)
+    when :error
+      level.to_s.upcase.colorize(:light_red)
+    when :fatal
+      level.to_s.upcase.colorize(:red)
+    end
+  end
+
+  def self.info(name, message = nil, level = :info)
+    output(name, message, level)
+  end
+
+  def self.debug(name, message = nil, level = :debug)
+    output(name, message, level)
+  end
+
+  def self.warn(name, message = nil, level = :warn)
+    output(name, message, level)
+  end
+
+  def self.error(name, message = nil, level = :error)
+    output(name, message, level)
+  end
+
+  def self.fatal(name, message = nil, level = :fatal)
+    output(name, message, level)
+  end
+
+  def self.do(name, message = nil, level = nil)
+    began_at = clock_time
+
+    level ||= :info
+    result = yield
+
+    duration = (clock_time - began_at).round(3)
+
+    output(name, message, level, duration)
+
+    result
+  end
+
+  def self.measure
+    began_at = clock_time
+    result = yield
+    duration = (clock_time - began_at).round(3)
+    [result, duration]
+  end
+
+  # From Rack Time
+  def self.clock_time
+    Process.clock_gettime(Process::CLOCK_MONOTONIC)
+  end
+end

data/lib/greenhat/pry_helpers.rb

@@ -0,0 +1,51 @@
+def show(data)
+  pastel = Pastel.new
+  TTY::Pager.page do |pager|
+    data.each do |e|
+      table = TTY::Table.new(header: e.keys, rows: [e], orientation: :vertical)
+      # do some work with the line
+
+      output = table.render(:unicode, multiline: false) do |renderer|
+        renderer.border.style = :cyan
+        # renderer.filter = ->(val, row_index, col_index) do
+        #   val.ai
+        # end
+
+        renderer.filter = lambda { |val, _row_index, col_index|
+          puts val.class unless val.instance_of? String
+          if col_index == 1
+            if val.numeric?
+              pastel.red.blue(val)
+            else
+              val
+            end
+          else
+            val
+          end
+          # .ai(raw: true).gsub('"','')
+        }
+      end
+
+      pager.write("\n#{output}") # write line to the pager
+    end
+  end
+end
+
+def filter(data, params = {})
+  result = data.clone
+  params.each do |k, v|
+    result.select! do |row|
+      if row.key? k.to_sym
+        row[k.to_sym].include? v
+      else
+        false
+      end
+    end
+  end
+
+  result
+end
+
+def version
+  Thing.find_by(name: 'gitlab_version_manifest_json').data.build_version
+end

data/lib/greenhat/settings.rb

@@ -0,0 +1,51 @@
+# Top level namespace
+module GreenHat
+  # Helper for all things user environment / settings / history
+  module Settings
+    def self.settings
+      @settings ||= {
+        history: []
+      }
+    end
+
+    def self.start
+      Dir.mkdir dir unless Dir.exist? dir
+
+      # CMD History Loading / Tracking
+      File.write(cmd_file, "\n") unless File.exist? cmd_file
+    end
+
+    def self.dir
+      "#{ENV['HOME']}/.greenhat"
+    end
+
+    # ----------------------------------
+    # Command History
+    def self.cmd_history
+      "#{dir}/cmd_history"
+    end
+
+    def self.cmd_file
+      "#{dir}/cmd_history"
+    end
+
+    def self.cmd_history_clean
+      File.read(cmd_file).split("\n")
+    end
+
+    def self.cmd_write
+      File.write(cmd_file, cmd_history_clean.join("\n"))
+    end
+
+    def self.cmd_add(line)
+      File.write(cmd_file, line, File.size(cmd_file), mode: 'a')
+    end
+
+    # ----------------------------------
+
+    # File Load History
+    def self.history_file
+      "#{dir}/file_history"
+    end
+  end
+end

data/lib/greenhat/shell.rb

@@ -0,0 +1,92 @@
+module GreenHat
+  # Root Level Shell
+  module Shell
+    def self.pry
+      binding.pry
+    end
+
+    def self.df
+      Disk.df
+    end
+
+    def self.ps
+      Process.ps
+    end
+
+    def self.netstat
+      Network.netstat
+    end
+
+    def self.free
+      Memory.free
+    end
+
+    def self.load_local
+      return false unless TTY::Prompt.new.yes?('Load local Omnibus GitLab Instance files?')
+
+      archive_path = "#{$TMP}/#{Time.now.to_i}_local"
+      Dir.mkdir(archive_path)
+      # archive = Archive.new(name: 'local')
+      archive = Archive.new(name: archive_path, path: archive_path)
+
+      file_list = Dir['/var/log/gitlab/*/current'] + Dir['/var/log/gitlab/*/*.log']
+
+      file_list.each do |file|
+        next if File.size(file).zero?
+
+        puts "- Loading #{file.colorize(:green)}"
+
+        archive.things_create(file: file).setup
+      end
+    end
+
+    def self.debug
+      ENV['DEBUG'] = if ENV['DEBUG']
+                       puts "Debug #{'Off'.colorize(:red)}"
+                       nil
+                     else
+                       puts "Debug #{'On'.colorize(:green)}"
+                       'true'
+                     end
+    end
+
+    def self.uptime
+      Shell::Cat.default ['uptime']
+    end
+
+    def self.hostname
+      Shell::Cat.default ['hostname']
+    end
+
+    def self.uname
+      Shell::Cat.default ['uname']
+    end
+
+    def self.history_clear
+      File.write(GreenHat::Cli.history_file, "\n")
+    end
+
+    def self.history
+      File.read(GreenHat::Cli.history_file).split("\n").each_with_index do |line, i|
+        puts "#{i.to_s.ljust(3).colorize(:magenta)} #{line}"
+      end
+    end
+
+    def self.ls
+      GreenHat::Cli.help
+    end
+
+    def self.ll
+      GreenHat::Cli.help
+    end
+
+    def self.about
+      puts "#{'GreenHat'.colorize(:green)}: #{GreenHat::VERSION.colorize(:blue)}"
+      puts ' - https://gitlab.com/gitlab-com/support/toolbox/greenhat'.colorize(:cyan)
+    end
+  end
+end
+
+# Load All Sub Files
+require_all "#{File.dirname(__FILE__)}/accessors"
+require_all "#{File.dirname(__FILE__)}/shell"

data/lib/greenhat/shell/cat.rb

@@ -0,0 +1,125 @@
+module GreenHat
+  # CLI Helper
+  module Shell
+    # Common File Reader File
+    module Cat
+      # rubocop:disable Metrics/MethodLength
+      def self.help
+        puts "\u2500".colorize(:cyan) * 20
+        puts "#{'Cat'.colorize(:yellow)} All the files"
+        puts "\u2500".colorize(:cyan) * 20
+        puts 'Print raw file by just entering the file name'
+        puts
+
+        puts 'Common Options'.colorize(:blue)
+        puts '  --raw'.colorize(:green)
+        puts '    Do not use less/paging'
+        puts
+        puts '  --archive'.colorize(:green)
+        puts '    Limit to specific archvie name (inclusive). Matching SOS tar.gz name'
+        puts '    Ex: --archive=dev-gitlab_20210622154626, --archive=202106,202107'
+        puts '    cat mount --archive=dev-gitlab_202106221546'
+        puts
+
+        puts 'Commands'.colorize(:blue)
+        puts 'ls'.colorize(:green)
+        puts '  List available files'
+        puts '  Options'
+        puts '    -a, --all, show all files including source'
+        puts
+
+        puts '<file names+>'.colorize(:green)
+        puts '  Print any file names'
+        puts '  Ex: `free_m`'
+        puts '  Ex: `ps mount --raw`'
+        puts
+
+        puts "#{'show'.colorize(:green)} <file names>"
+        puts '  Attempt to print formatted output'
+        puts '  Ex: show `free_m`'
+
+        puts '--archive'.colorize(:green)
+        puts '  Limit to specific archvie name (inclusive). Matching SOS tar.gz name'
+        puts '  Ex: --archive=dev-gitlab_20210622154626, --archive=202106,202107'
+        puts
+      end
+      # rubocop:enable Metrics/MethodLength
+
+      # ========================================================================
+      # Default
+      # ========================================================================
+      def self.default(file_list)
+        # Extract Args
+        file_list, _opts, args = ShellHelper.param_parse(file_list)
+
+        # Prepare Log List
+        file_list = ShellHelper.prepare_list(file_list, Thing.all)
+
+        # Convert to Things
+        files = ShellHelper.find_things(file_list, args)
+
+        results = ShellHelper.file_process(files) do |file|
+          [
+            file.friendly_name,
+            file.output(false),
+            "\n"
+          ]
+        end
+
+        ShellHelper.show(results.flatten, args)
+      end
+
+      # ========================================================================
+      # Show Attempted Formatting
+      # ========================================================================
+      def self.show(file_list)
+        # Extract Args
+        file_list, _opts, args = ShellHelper.param_parse(file_list)
+
+        # Prepare Log List
+        file_list = ShellHelper.prepare_list(file_list, Thing.all)
+
+        # Convert to Things
+        files = ShellHelper.find_things(file_list, args)
+
+        results = ShellHelper.file_process(files) do |file|
+          [
+            file.friendly_name,
+            file.output(false),
+            "\n"
+          ]
+        end
+
+        ShellHelper.show(results.flatten, args)
+      end
+
+      # List Files Helpers
+      def self.list(args = [])
+        all = false
+        all = true if args.include?('-a') || args.include?('--all')
+
+        files = Thing.all
+
+        # Sort
+        files.sort_by!(&:name)
+
+        # Short & Uniq
+        files.uniq!(&:name) unless all
+
+        # Print
+        files.each do |log|
+          if all
+            puts "- #{log.friendly_name}"
+          else
+            puts "- #{log.name.colorize(:yellow)}"
+          end
+        end
+      end
+
+      def self.ls(args = [])
+        list(args)
+      end
+      # ------------------------------------------------------------------------
+    end
+  end
+end

data/lib/greenhat/shell/disk.rb

@@ -0,0 +1,68 @@
+module GreenHat
+  # CLI Helper
+  module Shell
+    # Logs
+    module Disk
+      # Alias
+      def self.ls
+        Cli.help
+      end
+
+      # Easy Show All
+      def self.df
+        ShellHelper.file_output GreenHat::Disk.df
+      end
+
+      # rubocop:disable Metrics/MethodLength,Metrics/BlockLength
+      def self.free
+        GreenHat::Disk.df.each do |file|
+          # Reject TMPFS
+          puts file.friendly_name
+          disks = file.data.sort_by { |x| x.use.to_i }.reverse.reject { |x| x.filesystem.include? 'tmpfs' }
+
+          # pad_mount = GreenHat::Disk.max_padding(disks, :mounted_on)
+          # pad_size = GreenHat::Disk.max_padding(disks, :size)
+          # pad_used = GreenHat::Disk.max_padding(disks, :used)
+          # pad_avail = GreenHat::Disk.max_padding(disks, :avail)
+
+          pad_mount, pad_size, pad_used, pad_avail = GreenHat::Disk.padding(disks)
+
+          # Headers
+          puts [
+            'Mount'.ljust(pad_mount).colorize(:blue),
+            'Size'.ljust(pad_size).colorize(:magenta),
+            'Used'.ljust(pad_used).colorize(:cyan),
+            'Avail'.ljust(pad_avail).colorize(:white),
+            '% Use'.ljust(pad_avail).colorize(:green)
+          ].join
+
+          # Table Summary
+          disks.map do |disk|
+            # Pretty Disk Use
+            use = [
+              disk.use.rjust(5).ljust(5).colorize(:green),
+              '  ['.colorize(:blue),
+              ('=' * (disk.use.to_i / 2)).colorize(:green),
+              ' ' * (50 - disk.use.to_i / 2),
+              ']'.colorize(:blue)
+            ].join
+
+            # Whole Thing
+            puts [
+              disk.mounted_on.ljust(pad_mount).colorize(:blue),
+              disk[:size].to_s.ljust(pad_size).colorize(:magenta),
+              disk.used.to_s.ljust(pad_used).colorize(:cyan),
+              disk.avail.to_s.ljust(pad_avail).colorize(:white),
+              use
+            ].join
+          end
+
+          # File End Loop / Break
+          puts
+        end
+      end
+      # rubocop:enable Metrics/MethodLength,Metrics/BlockLength
+      # ------------------------------------------------------------------------
+    end
+  end
+end