panda-motd 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 83d9afcd650c9756e2d67f1b943636e1f36fd82f2bf980efdbdc5ecaa15de0d4
-  data.tar.gz: b63abe376f04963ef5c4a6d5609ddcaef7a6f5070e27e762ffa8dc1b61a91fb3
+  metadata.gz: 602cdbb28a1c8eb7b902dd7c1dee4f0caffb0663fcda3d4b505ca9f53b2aebf7
+  data.tar.gz: 638ca548913a3de96d7534bc22f84aa1fef57e4ee45ad18c4b295de7ce8add42
 SHA512:
-  metadata.gz: e53ec2f4ebd333e722d6c5954ffecd243af8edb3c84d791e8117e02ce23509866b1eedc09557f282553794f469082f29d79b714903ce0dd0b28979ebc246dd11
-  data.tar.gz: 272253a011490cdef506dff3fe273a1819795a9fcac8f25a16ad4b3f7a445e2373718d8a2d07ed525d99e13a46b98fb4de3c2ffb373330f74a1fd37e7d73a757
+  metadata.gz: ca2a3254437207351125a08175382980cc43905a22b9f84127993e900d9a2e20213b89d79f973fc9505b8ee38423e24b8542e78135f6cb4423f729782d60400f
+  data.tar.gz: fb2e551bc3bcf911fc9d01aad4af7906aaad435ec19e89026232e34e53ef536d6eae592a6a68e9700b69ac3924d7714e4d22a299e6f75746acbdf951880eaa1e

data/bin/panda-motd

@@ -2,5 +2,5 @@
 
 require 'panda_motd'
 
-motd = MOTD.new
+motd = PandaMOTD.new_motd
 puts motd

data/lib/panda_motd.rb

@@ -1,4 +1,8 @@
-Dir['lib/panda_motd/**/*.rb'].each { |r| require r.gsub(%r{lib\/|\.rb}, '') }
+require 'require_all'
+require_rel 'panda_motd'
 
 class PandaMOTD
+  def self.new_motd
+    return MOTD.new
+  end
 end

data/lib/panda_motd/components/ascii_text_art.rb

@@ -1,12 +1,18 @@
 require 'artii'
+require 'colorize'
 
 class AsciiTextArt
-  def initialize(text, font)
-    @text = text
-    @art = Artii::Base.new font: font
+  def initialize(motd)
+    @motd = motd
+    @config = motd.config.component_config('ascii_text_art')
+  end
+
+  def process
+    @text = `hostname`
+    @art = Artii::Base.new font: @config['font']
   end
 
   def to_s
-    return "\n#{@art.asciify(@text)}\n"
+    return @art.asciify(@text).red
   end
 end

data/lib/panda_motd/components/service_status.rb

@@ -0,0 +1,101 @@
+require 'colorize'
+
+class ServiceStatus
+  attr_reader :results
+
+  def initialize(motd)
+    @motd = motd
+    @config = motd.config.component_config('service_status')
+  end
+
+  def process
+    @services = @config['services']
+    @results = {}
+
+    parse_services
+  end
+
+  def to_s
+    if @results.any?
+      result = "Services:\n"
+      longest_name_size = @results.keys.map { |k| k.to_s.length }.max + 1 # add 1 for the ':' at the end
+      @results.each do |name, status|
+        name_part = if name.to_s.length > longest_name_size
+                      name.to_s.slice(1..longest_name_size)
+                    else
+                      (name.to_s + ':').ljust(longest_name_size, ' ')
+                    end
+        result += "  #{name_part} #{status}\n"
+      end
+
+      return result
+    else
+      return "Services:\n  No matching services found."
+    end
+  end
+
+  private
+
+  def parse_services
+    case Gem::Platform.local.os
+    when 'darwin'
+      @results = parse_services_macos(@services)
+    when 'linux'
+      @results = parse_services_linux(@services)
+    end
+
+    return self
+  end
+
+  def parse_services_macos(services)
+    results = {}
+
+    # valid statuses are started, stopped, error, and unknown
+    `brew services list`.split("\n").each do |line|
+      parsed_name = line.split[0]
+      parsed_status = line.split[1]
+
+      matching_service = services.find { |service, _name| service == parsed_name }
+
+      if matching_service
+        results[parsed_name.to_sym] = parsed_status.send(macos_service_colors[parsed_status.to_sym])
+      end
+    end
+
+    return results
+  end
+
+  def parse_services_linux(services)
+    results = {}
+
+    `systemctl | grep '\.service'`.split("\n").each do |line|
+      parsed_name = line.split[0].gsub('.service', '')
+      parsed_status = line.split[3]
+
+      matching_service = services.find { |service, _name| service == parsed_name }
+
+      if matching_service
+        results[parsed_name.to_sym] = parsed_status.send(linux_service_colors[parsed_status.to_sym])
+      end
+    end
+
+    return results
+  end
+
+  def macos_service_colors
+    return {
+      started: :green,
+      stopped: :white,
+      error: :red,
+      unknown: :yellow
+    }
+  end
+
+  def linux_service_colors
+    return {
+      running: :green,
+      exited: :white,
+      failed: :red
+    }
+  end
+end

data/lib/panda_motd/components/uptime.rb

@@ -3,7 +3,12 @@ require 'sysinfo'
 class Uptime
   attr_reader :days, :hours, :minutes
 
-  def initialize
+  def initialize(motd)
+    @motd = motd
+    @config = motd.config.component_config('ascii_text_art')
+  end
+
+  def process
     sysinfo = SysInfo.new
     uptime = sysinfo.uptime
 

data/lib/panda_motd/config.rb

@@ -0,0 +1,32 @@
+require 'yaml'
+require 'fileutils'
+
+class Config
+  def initialize(file_path = nil)
+    @file_path = File.join(Dir.home, '.config', 'panda-motd.yaml') if file_path.nil?
+    create_config(@file_path) unless File.exist?(@file_path)
+    load_config(@file_path)
+  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
+    # convert each snake_case name to an upper-camel-case name which represents a class, and get that class constant
+    return enabled_list.map { |e| Object.const_get(e.split('_').map(&:capitalize).join) }
+  end
+
+  def component_config(component_name)
+    return @config['components'][component_name.to_s]
+  end
+
+  private
+
+  def create_config(file_path)
+    default_config_path = File.join(File.dirname(__dir__), 'panda_motd', 'default_config.yaml')
+    FileUtils.cp(default_config_path, file_path)
+  end
+
+  def load_config(file_path)
+    @config = YAML.safe_load(File.read(file_path))
+  end
+end

data/lib/panda_motd/default_config.yaml

@@ -0,0 +1,23 @@
+components:
+  ascii_text_art:
+    enabled: true
+    font: slant
+    color: red
+
+  service_status:
+    enabled: true
+    services:
+      chunkwm: chunkwm
+      skhd: skhd
+      elasticsearch@5.6: elasticsearch
+      mysql: MySQL
+
+  uptime:
+    enabled: true
+
+  # disk_info:
+  #   enabled: true
+  #   disks:
+  #     /dev/disk3s1: Macintosh HD
+  #     /dev/disk0s4: Windows
+  #     /dev/disk2s2: Games

data/lib/panda_motd/motd.rb

@@ -1,14 +1,20 @@
+require 'sysinfo'
+
 class MOTD
-  attr_reader :components
+  attr_reader :config, :components
 
   def initialize
-    @components = [
-      AsciiTextArt.new('pandaMOTD', 'slant'),
-      Uptime.new
-    ]
+    @config = Config.new
+
+    @components = []
+    @config.components_enabled.each do |component_class|
+      @components << component_class.new(self)
+    end
+
+    @components.each(&:process)
   end
 
   def to_s
-    return @components.map(&:to_s).join("\n")
+    return @components.map(&:to_s).join("\n\n")
   end
 end

data/lib/panda_motd/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: panda-motd
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Taylor Thurlow
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-05-04 00:00:00.000000000 Z
+date: 2018-05-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: artii
@@ -25,7 +25,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '2.1'
 - !ruby/object:Gem::Dependency
-  name: sysinfo
+  name: colorize
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
@@ -39,19 +39,33 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0.8'
 - !ruby/object:Gem::Dependency
-  name: tty-markdown
+  name: require_all
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: sysinfo
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.3'
+        version: '0.8'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.3'
+        version: '0.8'
 - !ruby/object:Gem::Dependency
   name: byebug
   requirement: !ruby/object:Gem::Requirement
@@ -136,6 +150,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.25'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.12'
 description: Enhance your MOTD with useful at-a-glance information.
 email: taylorthurlow8@gmail.com
 executables:
@@ -145,8 +173,12 @@ extra_rdoc_files: []
 files:
 - bin/panda-motd
 - lib/panda_motd.rb
+- lib/panda_motd/component.rb
 - lib/panda_motd/components/ascii_text_art.rb
+- lib/panda_motd/components/service_status.rb
 - lib/panda_motd/components/uptime.rb
+- lib/panda_motd/config.rb
+- lib/panda_motd/default_config.yaml
 - lib/panda_motd/motd.rb
 - lib/panda_motd/version.rb
 homepage: https://github.com/taylorthurlow/panda-motd