panda-motd 0.0.8 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e96416cb34d16f565eb89916c6156cd2e293a04f0ba8ffeede5df5e92d73acc3
-  data.tar.gz: abc7ad56a5b1a0d90dcbf4cd4a7f948bed78232357f2ac7fc352f7388d136d0f
+  metadata.gz: 7036e8cca753cf96292eb0c4dd7bb29524852e486265b5de0403089af3f7897d
+  data.tar.gz: c9237796bc85552104a451801e202a0bd57e139f4c2bafc2cea9d2275ff92291
 SHA512:
-  metadata.gz: 452c47ad204b5dfb044abf370c0f0643c24bcfea8565ca5329ad487717cd116fd1d4c88378985be2d14a4759269cbe363d7b86bfdba31425e673d9f5be4aecef
-  data.tar.gz: a2209d1666149ac3222c005b79af5d3edb129a79dec9bc12c8c07c906ad26aa384dfc0222378ce3aa4c8da097cb425c33e4de76b67d92a0c2677ead2ba493366
+  metadata.gz: 8336450885b5e4f51d1afe19c75a18f3c3eca1928cc1ce9967287933960f0e695c2f6d2908ad3820e0cfcd705ade94388c9aa9657a9a894a0a1ee8d59090df03
+  data.tar.gz: f5b94c6be1aa9859735f0aa3eddeba09283c76664930127724ab628f29a82e917fbb1f63494ffa626514e82fd4e33891bbe607c7fa6222d506ec259d384a0e85

data/lib/panda_motd.rb

@@ -6,7 +6,11 @@ class PandaMOTD
     if ARGV[0].nil?
       puts 'You must provide a config file path as an argument to panda-motd.'
     else
-      return MOTD.new(ARGV[0])
+      MOTD.new(ARGV[0])
     end
   end
+
+  def self.root
+    File.expand_path('..', __dir__)
+  end
 end

data/lib/panda_motd/component.rb

@@ -0,0 +1,26 @@
+class Component
+  attr_reader :name, :errors, :results, :config
+
+  def initialize(motd, name)
+    @name = name
+    @motd = motd
+    @config = motd.config.component_config(@name)
+    @errors = []
+  end
+
+  def process
+    raise NotImplementedError
+  end
+
+  def to_s
+    raise NotImplementedError
+  end
+
+  def lines_before
+    @motd.config.component_config(@name)['lines_before'] || 1
+  end
+
+  def lines_after
+    @motd.config.component_config(@name)['lines_after'] || 1
+  end
+end

data/lib/panda_motd/components/ascii_text_art.rb

@@ -1,29 +1,21 @@
 require 'artii'
 require 'colorize'
 
-class ASCIITextArt
-  attr_reader :name, :errors, :results
-
+class ASCIITextArt < Component
   def initialize(motd)
-    @name = 'ascii_text_art'
-    @motd = motd
-    @config = motd.config.component_config(@name)
-    @errors = []
+    super(motd, 'ascii_text_art')
   end
 
   def process
     @text = `#{@config['command']}`
-
-    begin
-      @art = Artii::Base.new font: @config['font']
-      @results = @art.asciify(@text)
-      @results = @results.colorize(@config['color'].to_sym) if @config['color']
-    rescue Errno::EISDIR # Artii doesn't handle invalid font names very well
-      @errors << ComponentError.new(self, 'Invalid font name')
-    end
+    @art = Artii::Base.new font: @config['font']
+    @results = @art.asciify(@text)
+    @results = @results.colorize(@config['color'].to_sym) if @config['color']
+  rescue Errno::EISDIR # Artii doesn't handle invalid font names very well
+    @errors << ComponentError.new(self, 'Invalid font name')
   end
 
   def to_s
-    return @results
+    @results
   end
 end

data/lib/panda_motd/components/fail_2_ban.rb

@@ -0,0 +1,44 @@
+class Fail2Ban < Component
+  def initialize(motd)
+    super(motd, 'fail_2_ban')
+  end
+
+  def process
+    @results = {
+      jails: {}
+    }
+
+    @config['jails'].each do |jail|
+      status = jail_status(jail)
+      @results[:jails][jail] = {
+        total: status[:total],
+        current: status[:current]
+      }
+    end
+  end
+
+  def to_s
+    result = "Fail2Ban:\n"
+    @results[:jails].each do |name, stats|
+      result += "  #{name}:\n"
+      result += "    Total bans:   #{stats[:total]}\n"
+      result += "    Current bans: #{stats[:current]}\n"
+    end
+
+    result.gsub(/\s$/, '')
+  end
+
+  private
+
+  def jail_status(jail)
+    cmd_result = `fail2ban-client status #{jail}`
+    if cmd_result =~ /Sorry but the jail '#{jail}' does not exist/
+      @errors << ComponentError.new(self, "Invalid jail name '#{jail}'.")
+    else
+      total = cmd_result.match(/Total banned:\s+([0-9]+)/)[1].to_i
+      current = cmd_result.match(/Currently banned:\s+([0-9]+)/)[1].to_i
+    end
+
+    { total: total, current: current }
+  end
+end

data/lib/panda_motd/components/filesystems.rb

@@ -1,15 +1,9 @@
 require 'ruby-units'
 require 'colorize'
 
-class Filesystems
-  attr_reader :name, :errors
-  attr_reader :results
-
+class Filesystems < Component
   def initialize(motd)
-    @name = 'filesystems'
-    @motd = motd
-    @config = motd.config.component_config(@name)
-    @errors = []
+    super(motd, 'filesystems')
   end
 
   def process
@@ -17,69 +11,56 @@ class Filesystems
   end
 
   def to_s
-    name_col_size = @results.select { |r| r.is_a? Hash }.map { |r| r[:pretty_name].length }.max
-    name_col_size = 0 if name_col_size.nil?
+    name_col_size = @results.select { |r| r.is_a? Hash }
+                            .map { |r| r[:pretty_name].length }
+                            .max || 0
 
-    name_col_size_with_padding = (name_col_size + 6) > 13 ? (name_col_size + 6) : 13
+    size_w_padding = (name_col_size + 6) > 13 ? (name_col_size + 6) : 13
 
-    result = 'Filesystems'.ljust(name_col_size_with_padding, ' ')
+    result = 'Filesystems'.ljust(size_w_padding, ' ')
     result += "Size  Used  Free  Use%\n"
 
     @results.each do |filesystem|
-      # handle filesystem not found
-      if filesystem.is_a? String
-        result += "  #{filesystem}\n"
-        next
-      end
-
-      result += "  #{filesystem[:pretty_name]}".ljust(name_col_size_with_padding, ' ')
-
-      [:size, :used, :avail].each do |metric|
-        units = if filesystem[metric] > 10**12
-                  'terabytes'
-                elsif filesystem[metric] > 10**9
-                  'gigabytes'
-                elsif filesystem[metric] > 10**6
-                  'megabytes'
-                elsif filesystem[metric] > 10**3
-                  'kilobytes'
-                else
-                  'bytes'
-                end
-
-        value = Unit.new("#{filesystem[metric]} bytes").convert_to(units)
-
-        # we have 4 characters of space to include the number, a potential decimal point, and
-        # the unit character at the end. if the whole number component is 3+ digits long then
-        # we omit the decimal point and just display the whole number component. if the whole
-        # number component is 2 digits long, we can't afford to use a decimal point, so we still
-        # only display the whole number component. if the whole number component is 1 digit long,
-        # we use the single whole number component digit, a decimal point, a single fractional
-        # digit, and the unit character.
-        whole_number_length = value.scalar.floor.to_s.length
-        round_amount = whole_number_length > 1 ? 0 : 1
-        formatted_value = value.scalar.round(round_amount).to_s + units[0].upcase
-        result += formatted_value.rjust(4, ' ') + '  '
-      end
-
-      percentage_used = ((filesystem[:used].to_f / filesystem[:size].to_f) * 100).round
-      result += (percentage_used.to_s.rjust(3, ' ') + '%').send(percentage_color(percentage_used))
-
-      result += "\n"
-
-      total_ticks = name_col_size_with_padding + 18
-      used_ticks = (total_ticks * (percentage_used.to_f / 100)).round
-
-      result += "  [#{('=' * used_ticks).send(percentage_color(percentage_used))}#{('=' * (total_ticks - used_ticks)).light_black}]"
-      result += "\n" unless filesystem == @results.last
+      result += format_filesystem(filesystem, size_w_padding)
     end
 
-    return result
+    result
   end
 
   private
 
-  def percentage_color(percentage)
+  def format_filesystem(filesystem, size)
+    return "  #{filesystem}\n" if filesystem.is_a? String # handle fs not found
+
+    # filesystem name
+    result = ''
+    result += "  #{filesystem[:pretty_name]}".ljust(size, ' ')
+
+    # statistics (size, used, free, use%)
+    [:size, :used, :avail].each do |metric|
+      result += format_metric(filesystem, metric)
+    end
+    percent_used = calc_percent_used(filesystem)
+    result += format_percent_used(percent_used)
+    result += "\n"
+
+    # visual bar representation of use%
+    result += generate_usage_bar(filesystem, size, percent_used)
+
+    result
+  end
+
+  def generate_usage_bar(filesystem, size, percent_used)
+    result = ''
+    total_ticks = size + 18
+    used_ticks = (total_ticks * (percent_used.to_f / 100)).round
+    result += "  [#{('=' * used_ticks).send(pct_color(percent_used))}"\
+              "#{('=' * (total_ticks - used_ticks)).light_black}]"
+    result += "\n" unless filesystem == @results.last
+    result
+  end
+
+  def pct_color(percentage)
     case percentage
     when 0..75 then :green
     when 76..95 then :yellow
@@ -88,9 +69,52 @@ class Filesystems
     end
   end
 
+  def calc_percent_used(filesystem)
+    ((filesystem[:used].to_f / filesystem[:size].to_f) * 100).round
+  end
+
+  def format_percent_used(percent_used)
+    (percent_used.to_s.rjust(3, ' ') + '%').send(pct_color(percent_used))
+  end
+
+  def calc_metric(value)
+    Unit.new("#{value} bytes").convert_to(calc_units(value))
+  end
+
+  def format_metric(filesystem, metric)
+    # we have 4 characters of space to include the number, a potential
+    # decimal point, and the unit character at the end. if the whole number
+    # component is 3+ digits long then we omit the decimal point and just
+    # display the whole number component. if the whole number component is
+    # 2 digits long, we can't afford to use a decimal point, so we still
+    # only display the whole number component. if the whole number
+    # component is 1 digit long, we use the single whole number component
+    # digit, a decimal point, a single fractional digit, and the unit
+    # character.
+
+    value = calc_metric(filesystem[metric])
+    whole_number_length = value.scalar.floor.to_s.length
+    round_amount = whole_number_length > 1 ? 0 : 1
+    formatted = value.scalar.round(round_amount).to_s + value.units[0].upcase
+    formatted.rjust(4, ' ') + '  '
+  end
+
+  def calc_units(value)
+    if value > 10**12
+      'terabytes'
+    elsif value > 10**9
+      'gigabytes'
+    elsif value > 10**6
+      'megabytes'
+    elsif value > 10**3
+      'kilobytes'
+    else
+      'bytes'
+    end
+  end
+
   def parse_filesystem_usage(filesystems)
-    entries = `BLOCKSIZE=1024 df --output=source,size,used,avail`.lines
-                                                                 .drop(1)
+    entries = `BLOCKSIZE=1024 df --output=source,size,used,avail`.lines.drop(1)
 
     filesystems.map do |filesystem, pretty_name|
       matching_entry = entries.map(&:split).find { |e| e.first == filesystem }

data/lib/panda_motd/components/last_login.rb

@@ -1,13 +1,8 @@
 require 'date'
 
-class LastLogin
-  attr_reader :name, :errors, :results
-
+class LastLogin < Component
   def initialize(motd)
-    @name = 'last_login'
-    @motd = motd
-    @config = motd.config.component_config(@name)
-    @errors = []
+    super(motd, 'last_login')
   end
 
   def process
@@ -16,55 +11,64 @@ class LastLogin
   end
 
   def to_s
-    <<~LAST
+    <<~HEREDOC
       Last Login:
-      #{@results.map do |user, logins|
-        logins_part =
-          if logins.empty?
-            '    no logins found for user.'
-          else
-            longest_location_size = logins.map { |l| l[:location].length }.max
-            logins.map do |login|
-              location_part = login[:location].ljust(longest_location_size, ' ')
-              start_part = login[:time_start].strftime('%m/%d/%Y %I:%M%p')
-              end_part = if login[:time_end].is_a? String # still logged in text
-                           login[:time_end].green
-                         else
-                           "#{((login[:time_end] - login[:time_start]) * 24 * 60).to_i} minutes"
-                         end
-              "    from #{location_part} at #{start_part} (#{end_part})"
-            end.join("\n")
-          end
-        <<~USER
-            #{user}:
-          #{logins_part}
-          USER
-      end.join("\n")}
-    LAST
+      #{@results.map { |u, l| parse_result(u, l) }.join("\n")}
+    HEREDOC
   end
 
   private
 
+  def parse_result(user, logins)
+    logins_part = if logins.empty?
+                    '    no logins found for user.'
+                  else
+                    longest_size = logins.map { |l| l[:location].length }.max
+                    logins.map { |l| parse_login(l, longest_size) }.join("\n")
+                  end
+    <<~HEREDOC
+        #{user}:
+      #{logins_part}
+    HEREDOC
+  end
+
+  def parse_login(login, longest_size)
+    location = login[:location].ljust(longest_size, ' ')
+    start = login[:time_start].strftime('%m/%d/%Y %I:%M%p')
+    finish = if login[:time_end].is_a? String # not a date
+               if login[:time_end] == 'still logged in'
+                 login[:time_end].green
+               else
+                 login[:time_end].yellow
+               end
+             else
+               "#{((login[:time_end] - login[:time_start]) / 60).to_i} minutes"
+             end
+    "    from #{location} at #{start} (#{finish})"
+  end
+
   def parse_last_logins(users)
     users.map do |(username, num_logins)|
-      user_logins =
-        `last --time-format=iso #{username}`
-        .lines
-        .select { |entry| entry.start_with?(username) }
-        .take(num_logins)
-        .map do |entry|
-          data = entry.chomp.split(/(?:\s{2,})|(?:\s-\s)/)
-          time_end = data[4] == 'still logged in' ? data[4] : DateTime.parse(data[4])
+      cmd_result = `last --time-format=iso #{username}`
+      logins = cmd_result.lines
+                         .select { |entry| entry.start_with?(username) }
+                         .take(num_logins)
 
-          {
-            username: username,
-            location: data[2],
-            time_start: DateTime.parse(data[3]),
-            time_end: time_end
-          }
-        end
-
-      [username.to_sym, user_logins]
+      [username.to_sym, logins.map! { |l| hashify_login(l, username) }]
     end.to_h
   end
+
+  def hashify_login(login, username)
+    re = login.chomp.split(/(?:\s{2,})|(?<=\d)(?:\s-\s)/)
+    date = re[4].scan(/\d{4}-\d{2}-[\dT:]+-\d{4}/)
+
+    time_end = date.any? ? Time.parse(re[4]) : re[4]
+
+    {
+      username: username,
+      location: re[2],
+      time_start: Time.parse(re[3]),
+      time_end: time_end
+    }
+  end
 end

data/lib/panda_motd/components/service_status.rb

@@ -1,14 +1,8 @@
 require 'colorize'
 
-class ServiceStatus
-  attr_reader :name, :errors
-  attr_reader :results
-
+class ServiceStatus < Component
   def initialize(motd)
-    @name = 'service_status'
-    @motd = motd
-    @config = motd.config.component_config(@name)
-    @errors = []
+    super(motd, 'service_status')
   end
 
   def process
@@ -18,8 +12,9 @@ class ServiceStatus
 
   def to_s
     return "Services:\n  No matching services found." unless @results.any?
+
     longest_name_size = @results.keys.map { |k| k.to_s.length }.max
-    <<~HEREDOC
+    result = <<~HEREDOC
       Services:
       #{@results.map do |(name, status)|
         name_part = name.to_s.ljust(longest_name_size, ' ') + ':'
@@ -27,28 +22,32 @@ class ServiceStatus
         "  #{name_part} #{status_part}"
       end.join("\n")}
     HEREDOC
+
+    result.gsub(/\s$/, '')
   end
 
   private
 
   def parse_service(service)
     cmd_result = `systemctl is-active #{service[0]}`.strip
-    @errors << ComponentError.new(self, 'Unable to parse systemctl output') unless valid_responses.include? cmd_result
-    return cmd_result
+    unless valid_responses.include? cmd_result
+      @errors << ComponentError.new(self, 'Unable to parse systemctl output')
+    end
+    cmd_result
   end
 
   def parse_services(services)
-    services.map { |service| [service[1].to_sym, parse_service(service).to_sym] }.to_h
+    services.map { |s| [s[1].to_sym, parse_service(s).to_sym] }.to_h
   end
 
   def service_colors
-    return {
+    {
       active: :green,
       inactive: :red
     }
   end
 
   def valid_responses
-    return ['active', 'inactive']
+    ['active', 'inactive']
   end
 end

data/lib/panda_motd/components/ssl_certificates.rb

@@ -1,13 +1,8 @@
 require 'date'
 
-class SSLCertificates
-  attr_reader :name, :errors, :results
-
+class SSLCertificates < Component
   def initialize(motd)
-    @name = 'ssl_certificates'
-    @motd = motd
-    @config = motd.config.component_config(@name)
-    @errors = []
+    super(motd, 'ssl_certificates')
   end
 
   def process
@@ -16,48 +11,75 @@ class SSLCertificates
   end
 
   def to_s
-    longest_name_size = @results.map { |r| r[0].length }.max
     <<~HEREDOC
       SSL Certificates:
-      #{@results.map do |cert|
+      #{sorted_results.map do |cert|
           return "  #{cert}" if cert.is_a? String # print the not found message
 
-          name_portion = cert[0].ljust(longest_name_size + 6, ' ')
-          status = cert_status(cert[1])
-          status = cert_status_strings[status].to_s.colorize(cert_status_colors[status])
-          date_portion = cert[1].strftime('%e %b %Y %H:%M:%S%p')
-          "  #{name_portion} #{status} #{date_portion}"
+          parse_cert(cert)
         end.join("\n")}
     HEREDOC
   end
 
   private
 
+  def parse_cert(cert)
+    name_portion = cert[0].ljust(longest_cert_name_length + 6, ' ')
+    status_sym = cert_status(cert[1])
+    status = cert_status_strings[status_sym].to_s
+    colorized_status = status.colorize(cert_status_colors[status_sym])
+    date_portion = cert[1].strftime('%e %b %Y %H:%M:%S%p')
+    "  #{name_portion} #{colorized_status} #{date_portion}"
+  end
+
+  def longest_cert_name_length
+    @results.map { |r| r[0].length }.max
+  end
+
+  def sorted_results
+    if @config['sort_method'] == 'alphabetical'
+      @results.sort_by { |c| c[0] }
+    elsif @config['sort_method'] == 'expiration'
+      @results.sort_by { |c| c[1] }
+    else # default to alphabetical
+      @results.sort_by { |c| c[0] }
+    end
+  end
+
   def cert_dates(certs)
-    return certs.map do |name, path|
+    certs.map do |name, path|
       if File.exist?(path)
-        cmd_result = `openssl x509 -in #{path} -dates`
-        parsed = cmd_result.match(/notAfter=([\w\s:]+)\n/)
-        if parsed.nil?
-          @errors << ComponentError.new(self, 'Unable to find certificate expiration date')
-        else
-          begin
-            expiry_date = DateTime.parse(parsed[1])
-            [name, expiry_date]
-          rescue ArgumentError
-            @errors << ComponentError.new(self, 'Found expiration date, but unable to parse as date')
-          end
-        end
+        parse_result(name, path)
       else
         "Certificate #{name} not found at path: #{path}"
       end
+    end.compact # remove nil entries, will have nil if error ocurred
+  end
+
+  def parse_result(name, path)
+    cmd_result = `openssl x509 -in #{path} -dates`
+    # match indices: 1 - month, 2 - day, 3 - time, 4 - year, 5 - zone
+    exp = /notAfter=([A-Za-z]+) (\d+) ([\d:]+) (\d{4}) ([A-Za-z]+)\n/
+    parsed = cmd_result.match(exp)
+
+    if parsed.nil?
+      @errors << ComponentError.new(self, 'Unable to find certificate '\
+                                          'expiration date')
+      nil
+    else
+      expiry_date = Time.parse([1, 2, 4, 3, 5].map { |n| parsed[n] }.join(' '))
+      [name, expiry_date]
     end
+  rescue ArgumentError
+    @errors << ComponentError.new(self, 'Found expiration date, but unable '\
+                                        'to parse as date')
+    [name, Time.now]
   end
 
   def cert_status(expiry_date)
-    if (DateTime.now...DateTime.now + 30).cover? expiry_date # ... range excludes end
+    if (Time.now...Time.now + 30).cover? expiry_date # ... range excludes last
       :expiring
-    elsif DateTime.now >= expiry_date
+    elsif Time.now >= expiry_date
       :expired
     else
       :valid
@@ -65,7 +87,7 @@ class SSLCertificates
   end
 
   def cert_status_colors
-    return {
+    {
       valid: :green,
       expiring: :yellow,
       expired: :red
@@ -73,7 +95,7 @@ class SSLCertificates
   end
 
   def cert_status_strings
-    return {
+    {
       valid: 'valid until',
       expiring: 'expiring at',
       expired: 'expired at'

data/lib/panda_motd/components/uptime.rb

@@ -1,14 +1,10 @@
 require 'sysinfo'
 
-class Uptime
-  attr_reader :name, :errors
+class Uptime < Component
   attr_reader :days, :hours, :minutes
 
   def initialize(motd)
-    @name = 'uptime'
-    @motd = motd
-    @config = motd.config.component_config(@name)
-    @errors = []
+    super(motd, 'uptime')
   end
 
   def process
@@ -20,7 +16,7 @@ class Uptime
   end
 
   def to_s
-    return "#{@config['prefix'] || 'up'} #{format_uptime}"
+    "#{@config['prefix'] || 'up'} #{format_uptime}"
   end
 
   private

data/lib/panda_motd/config.rb

@@ -2,8 +2,10 @@ require 'yaml'
 require 'fileutils'
 
 class Config
+  attr_reader :file_path
+
   def initialize(file_path = nil)
-    @file_path = file_path.nil? ? File.join(Dir.home, '.config', 'panda-motd.yaml') : file_path
+    @file_path = file_path || File.join(Dir.home, '.config', 'panda-motd.yaml')
     unless File.exist?(@file_path)
       create_config(@file_path)
       puts "panda-motd created a default config file at: #{@file_path}"
@@ -12,31 +14,34 @@ class Config
   end
 
   def components_enabled
-    # first iterate through the config hash and grab all names of enabled components
-    enabled_list = @config['components'].map { |component, setting| component if setting['enabled'] }.compact
+    # iterate config hash and grab names of enabled components
+    enabled = @config['components'].map { |c, s| c if s['enabled'] }.compact
     # get the class constant
-    return enabled_list.map { |e| component_classes[e.to_sym] }
+    enabled.map { |e| Config.component_classes[e.to_sym] }
   end
 
   def component_config(component_name)
-    return @config['components'][component_name.to_s]
+    @config['components'][component_name.to_s]
   end
 
-  private
-
-  def component_classes
-    return {
+  def self.component_classes
+    {
       ascii_text_art: ASCIITextArt,
       service_status: ServiceStatus,
       uptime: Uptime,
       ssl_certificates: SSLCertificates,
       filesystems: Filesystems,
-      last_login: LastLogin
+      last_login: LastLogin,
+      fail_2_ban: Fail2Ban
     }
   end
 
+  private
+
   def create_config(file_path)
-    default_config_path = File.join(File.dirname(__dir__), 'panda_motd', 'default_config.yaml')
+    default_config_path = File.join(
+      File.dirname(__dir__), 'panda_motd', 'default_config.yaml'
+    )
     FileUtils.cp(default_config_path, file_path)
   end
 

data/lib/panda_motd/default_config.yaml

@@ -62,6 +62,10 @@ components:
   #   Displays the validity and expiration dates of SSL certificates.
   #
   # Settings
+  #   sort_method: The method used to sort the list of certificates. If no
+  #     method is specified, the list will be sorted alphabetically. Valid
+  #     values are 'alphabetical' and 'expiration'. The latter will place
+  #     the certificates which are most near expiration at the top.
   #   certs: Pairs following the format "PrettyName: absolute_cert_file_path".
   #     The absolute_cert_file_path is the absolute file path of the SSL
   #     certificate. The pretty name is the name that is used in the MOTD to
@@ -70,8 +74,9 @@ components:
 
   # ssl_certificates:
   #   enabled: true
+  #   sort_method: alphabetical
   #   certs:
-  #     taylorjthurlow.com: /etc/letsencrypt/live/taylorjthurlow.com/cert.pem
+  #     thurlow.io: /etc/letsencrypt/live/thurlow.io/cert.pem
 
   #####
   # Filesystems
@@ -90,6 +95,21 @@ components:
   #     /dev/sda1: Ubuntu
   #     /dev/sdc1: Data
 
+  #####
+  # Fail2Ban
+  #   Displays fail2ban jail statistics.
+  #
+  # Settings
+  #   jails: A list of fail2ban jails to obtain statistics from. The name of the
+  #     jail is the same name used in the `fail2ban-client status jailname`. You
+  #     will get the total banned and currently banned numbers for each jail.
+  #####
+
+  # fail_2_ban:
+  #   enabled: true
+  #   jails:
+  #     - sshd
+  #     - anotherjail
 
   #####
   # Last Login

data/lib/panda_motd/motd.rb

@@ -3,14 +3,14 @@ require 'sysinfo'
 class MOTD
   attr_reader :config, :components
 
-  def initialize(config_path = nil)
+  def initialize(config_path = nil, process = true)
     @config = config_path ? Config.new(config_path) : Config.new
     @components = @config.components_enabled.map { |ce| ce.new(self) }
-    @components.each(&:process)
+    @components.each(&:process) if process
   end
 
   def to_s
-    return @components.map do |c|
+    @components.map do |c|
       if c.errors.any?
         c.errors.map(&:to_s).join("\n")
       else

data/lib/panda_motd/version.rb

@@ -1,4 +1,4 @@
 class PandaMOTD
   #:nodoc:
-  VERSION ||= '0.0.8'.freeze
+  VERSION ||= '0.0.10'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: panda-motd
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.10
 platform: ruby
 authors:
 - Taylor Thurlow
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-10-28 00:00:00.000000000 Z
+date: 2018-11-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: artii
@@ -81,103 +81,131 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0.8'
 - !ruby/object:Gem::Dependency
-  name: byebug
+  name: factory_bot
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: guard
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '2.14'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '2.14'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: guard-rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '4.7'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '4.7'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '3.7'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '3.7'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0.51'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0.51'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rubocop-rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.25'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.25'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0.12'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0.12'
+        version: '0'
 description: Enhance your MOTD with useful at-a-glance information.
 email: taylorthurlow8@gmail.com
 executables:
@@ -187,8 +215,10 @@ extra_rdoc_files: []
 files:
 - bin/panda-motd
 - lib/panda_motd.rb
+- lib/panda_motd/component.rb
 - lib/panda_motd/component_error.rb
 - lib/panda_motd/components/ascii_text_art.rb
+- lib/panda_motd/components/fail_2_ban.rb
 - lib/panda_motd/components/filesystems.rb
 - lib/panda_motd/components/last_login.rb
 - lib/panda_motd/components/service_status.rb
@@ -218,7 +248,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.7.6
+rubygems_version: 2.7.8
 signing_key: 
 specification_version: 4
 summary: Make your MOTD prettier, and more useful.