greenhat 0.1.4

data/lib/greenhat/thing/spinner.rb

@@ -0,0 +1,52 @@
+module GreenHat
+  # Info Formatter
+  module Spinner
+    def spin_start(title)
+      # Ignore when Quiet is Set
+      return true if Cli.quiet
+
+      @spin_time = Time.now
+      @spinner = TTY::Spinner.new(
+        "#{time.colorize(:light_black)} - [:spinner] :title", hide_cursor: true, success_mark: '✔'.colorize(:green)
+      )
+      @spinner.update(title: title)
+
+      # Don't Auto spin when debug output is happening
+      @spinner.auto_spin unless ENV['DEBUG']
+    end
+
+    def spin_done
+      # Ignore when Quiet is Set
+      return true if Cli.quiet
+
+      title = @spinner.tokens[:title]
+      spin_end = humanize(@spin_time)
+
+      title_update = if spin_end.blank?
+                       title
+                     else
+                       "#{title} (#{spin_end.colorize(:blue)})"
+                     end
+
+      @spinner.update(title: title_update)
+      @spinner.success
+    end
+
+    def time
+      Time.now.strftime('%I:%M:%S').colorize(:light_black)
+    end
+
+    # Replace TimeDifference with https://stackoverflow.com/a/4136485/1678507
+    def humanize(time)
+      miliseconds = (Time.now - time) * 1000
+
+      [[1000, :ms], [60, :s], [60, :m], [24, :h]].map do |count, name|
+        next unless miliseconds.positive?
+
+        miliseconds, n = miliseconds.divmod(count)
+
+        "#{n.to_i}#{name}" unless n.to_i.zero?
+      end.compact.reverse.join(' ')
+    end
+  end
+end

data/lib/greenhat/thing/super_log.rb

@@ -0,0 +1,102 @@
+# TODO: Deprecate, was used specifically for Elk
+# rubocop:disable  /AbcSize, Metrics/CyclomaticComplexity, Metrics/MethodLength, Metrics/PerceivedComplexity, Metrics/
+module GreenHat
+  # Log Identifier
+  module SuperLog
+    def self.log?(kind)
+      [
+        'gitaly/current',
+        'gitaly/gitaly_ruby_json.log',
+        'gitlab-rails/api_json.log',
+        'gitlab-rails/application_json.log',
+        'gitlab-rails/audit_json.log',
+        'gitlab-rails/graphql_json.log',
+        'gitlab-rails/integrations_json.log',
+        'gitlab-rails/production_json.log',
+        'gitlab-rails/sidekiq_client.log',
+        'gitlab-shell/gitlab-shell.log',
+        'gitlab-workhorse/current',
+        'puma/puma_stdout.log',
+        'sidekiq/current',
+        'gitlab-rails/importer.log',
+        'gitlabsos.log',
+        'nginx/gitlab_access.log',
+        'patroni/current'
+      ].any? { |x| x.include? kind.to_s }
+
+      true
+    end
+
+    # ==========================================================================
+    # File Pattern Matching
+    # ==========================================================================
+    def self.api_json_format?(path)
+      ['rails_api_json_log'].any? { |x| path.include? x }
+    end
+
+    def self.unicorn_stderr_format?(path)
+      %w[
+        unicorn_stderr
+      ].any? { |x| path.include? x }
+    end
+
+    def self.reconfigure_format?(path)
+      %w[
+        reconfigure
+      ].any? { |x| path.include? x }
+    end
+
+    def self.shellwords?(path)
+      %w[
+        gitlab_pages_current
+        alertmanager_current
+        registry_current
+        prometheus_current
+
+      ].any? { |x| path.include? x }
+    end
+
+    def self.time_space?(path)
+      %w[
+        postgresql_current
+        redis_current
+        unicorn_current
+        gitlab_monitor_current
+        sidekiq_exporter_log
+      ].any? { |x| path.include? x }
+    end
+
+    def self.json_format?(path)
+      %w[
+        production_json_log
+        gitaly_current
+        geo_log
+        sidekiq_current
+        sidekiq_log
+        gitlab_shell_gitlab_shell_log
+        gitlab_rails_audit_json_log
+        gitlab_rails_application_json_log
+      ].any? { |x| path.include? x }
+    end
+
+    def self.dmesg_format?(path)
+      %w[
+        dmesg
+      ].any? { |x| path.include? x }
+    end
+    # ==========================================================================
+
+    # Identify Formatter
+    def self.type?(path)
+      return :api_json_format if api_json_format?(path)
+      return :unicorn_stderr_format if unicorn_stderr_format?(path)
+      return :reconfigure_format if reconfigure_format?(path)
+      return :shellwords if shellwords?(path)
+      return :time_space if time_space?(path)
+      return :json_format if json_format?(path)
+      return :dmesg_format if dmesg_format?(path)
+
+      nil
+    end
+  end
+end

data/lib/greenhat/tty/custom_line.rb

@@ -0,0 +1,29 @@
+module TTY
+  class Reader
+    # Shim Helper
+    class Line
+      def move_word_left
+        loop do
+          # Don't go past beginning
+          break if @cursor.zero?
+
+          left 1
+
+          break if self[@cursor].blank?
+        end
+      end
+
+      def move_word_right
+        loop do
+          right 1
+          break if self[@cursor].blank?
+        end
+      end
+
+      # No Tabs
+      def insert(chars)
+        self[@cursor] = chars unless ["\t", "\e[Z"].include?(chars)
+      end
+    end
+  end
+end

data/lib/greenhat/tty/line.rb

@@ -0,0 +1,326 @@
+# https://github.com/piotrmurach/tty-reader/commit/cc38eadf888ac179a77d72c21ebda987cfea3aca
+
+# Disabling Since this is a Shim
+# rubocop:disable all
+
+
+module TTY
+  class Reader
+    class Line
+      # ANSI_MATCHER = /(\[)?\033(\[)?[;?\d]*[\dA-Za-z](\])?/
+
+      # The word break characters list used by shell
+      DEFAULT_WORD_BREAK_CHARACTERS = " \t\n\"\\'`@$><=|&{("
+
+      # Strip ANSI characters from the text
+      #
+      # @param [String] text
+      #
+      # @return [String]
+      #
+      # @api public
+      def self.sanitize(text)
+        text.dup.gsub(ANSI_MATCHER, '')
+      end
+
+      # The editable text
+      # @api public
+      attr_accessor :text
+
+      # The current cursor position witin the text
+      # @api public
+      attr_reader :cursor
+
+      # The line mode
+      # @api public
+      attr_reader :mode
+
+      # The prompt displayed before input
+      # @api public
+      attr_reader :prompt
+
+      # The word separator pattern for splitting the text
+      #
+      # @return [Regexp]
+      #
+      # @api public
+      attr_reader :separator
+
+      # Create a Line instance
+      #
+      # @api private
+      def initialize(text = '', prompt: '', separator: nil)
+        @text   = text.dup
+        @prompt = prompt.dup
+        break_chars = DEFAULT_WORD_BREAK_CHARACTERS.chars
+        @separator = separator || Regexp.union(break_chars)
+        @cursor = [0, @text.length].max
+        @mode   = :edit
+
+        yield self if block_given?
+      end
+
+      # Check if line is in edit mode
+      #
+      # @return [Boolean]
+      #
+      # @public
+      def editing?
+        @mode == :edit
+      end
+
+      # Enable edit mode
+      #
+      # @return [Boolean]
+      #
+      # @public
+      def edit_mode
+        @mode = :edit
+      end
+
+      # Check if line is in replace mode
+      #
+      # @return [Boolean]
+      #
+      # @public
+      def replacing?
+        @mode == :replace
+      end
+
+      # Enable replace mode
+      #
+      # @return [Boolean]
+      #
+      # @public
+      def replace_mode
+        @mode = :replace
+      end
+
+      # Check if cursor reached beginning of the line
+      #
+      # @return [Boolean]
+      #
+      # @api public
+      def start?
+        @cursor.zero?
+      end
+
+      # Check if cursor reached end of the line
+      #
+      # @return [Boolean]
+      #
+      # @api public
+      def end?
+        @cursor == @text.length
+      end
+
+      # Move line position to the left by n chars
+      #
+      # @api public
+      def left(n = 1)
+        @cursor = [0, @cursor - n].max
+      end
+
+      # Move line position to the right by n chars
+      #
+      # @api public
+      def right(n = 1)
+        @cursor = [@text.length, @cursor + n].min
+      end
+
+      # Move cursor to beginning position
+      #
+      # @api public
+      def move_to_start
+        @cursor = 0
+      end
+
+      # Move cursor to end position
+      #
+      # @api public
+      def move_to_end
+        @cursor = @text.length # put cursor outside of text
+      end
+
+      # Insert characters inside a line. When the lines exceeds
+      # maximum length, an extra space is added to accomodate index.
+      #
+      # @param [Integer] i
+      #   the index to insert at
+      #
+      # @param [String] chars
+      #   the characters to insert
+      #
+      # @example
+      #   text = "aaa"
+      #   line[5]= "b"
+      #   => "aaa  b"
+      #
+      # @api public
+      def []=(i, chars)
+        edit_mode
+
+        if i.is_a?(Range)
+          @text[i] = chars
+          @cursor += chars.length
+          return
+        end
+
+        if i <= 0
+          before_text = ''
+          after_text = @text.dup
+        elsif i > @text.length - 1 # insert outside of line input
+          before_text = @text.dup
+          after_text = "\s" * (i - @text.length)
+          @cursor += after_text.length
+        else
+          before_text = @text[0..i - 1].dup
+          after_text  = @text[i..-1].dup
+        end
+
+        @text = if i > @text.length - 1
+                  before_text + after_text + chars
+                else
+                  before_text + chars + after_text
+                end
+
+        @cursor = i + chars.length
+      end
+
+      # Read character
+      #
+      # @api public
+      def [](i)
+        @text[i]
+      end
+
+      # Find a word under the cursor based on the word separator
+      #
+      # @param [Boolean] before
+      #   whether to start searching before or after a break character
+      #
+      # @return [String]
+      #
+      # @api public
+      def word(before: true)
+        @text[range(before: before)]
+      end
+
+      # Find a range of characters under the cursor based on the word separator
+      #
+      # @param [Integer] from
+      #   the start index
+      #
+      # @param [Symbol] before
+      #   whether to start search before or after break character
+      #
+      # @return [Range]
+      #
+      # @api public
+      def range(from: @cursor, before: true)
+        # move back or forward by one character when at a word boundary
+        if word_boundary?
+          from = before ? from - 1 : from + 1
+        end
+
+        # find start position
+        start_pos = @text.rindex(separator, from) || 0
+        start_pos += 1 unless start_pos.zero?
+
+        # find end position
+        end_pos = @text.index(separator, start_pos) || text_size
+        end_pos -= 1 unless @text.empty?
+
+        start_pos..end_pos
+      end
+
+      # Check if cursor is at a word boundary
+      #
+      # @return [Boolean]
+      #
+      # @api private
+      def word_boundary?
+        @text[@cursor] =~ separator
+      end
+
+      # Replace current line with new text
+      #
+      # @param [String] text
+      #
+      # @api public
+      def replace(text)
+        @text = text
+        @cursor = @text.length # put cursor outside of text
+        replace_mode
+      end
+
+      # Insert char(s) at cursor position
+      #
+      # @api public
+      def insert(chars)
+        self[@cursor] = chars
+      end
+
+      # Add char and move cursor
+      #
+      # @api public
+      def <<(char)
+        @text << char
+        @cursor += 1
+      end
+
+      # Remove char from the line at current position
+      #
+      # @api public
+      def delete(n = 1)
+        @text.slice!(@cursor, n)
+      end
+
+      # Remove char from the line in front of the cursor
+      #
+      # @param [Integer] n
+      #   the number of chars to remove
+      #
+      # @api public
+      def remove(n = 1)
+        left(n)
+        @text.slice!(@cursor, n)
+      end
+
+      # Full line with prompt as string
+      #
+      # @api public
+      def to_s
+        "#{@prompt}#{@text}"
+      end
+      alias inspect to_s
+
+      # Prompt size
+      #
+      # @api public
+      def prompt_size
+        p = self.class.sanitize(@prompt).split(/\r?\n/)
+        # return the length of each line + screen width for every line past the first
+        # which accounts for multi-line prompts
+        p.join.length + ((p.length - 1) * TTY::Screen.width)
+      end
+
+      # Text size
+      #
+      # @api public
+      def text_size
+        self.class.sanitize(@text).size
+      end
+
+      # Full line size with prompt
+      #
+      # @api public
+      def size
+        prompt_size + text_size
+      end
+      alias length size
+    end
+  end
+end
+
+# rubocop:enable all