hospital 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4fe2ccd9fe0117efeaf99d430e269bee6e11e27463161d7038f4550ad79dcabb
-  data.tar.gz: 045b72881224032291189022ede2c9eec3aacc8edb52a688d05570e0ad54d871
+  metadata.gz: 5c2d250697eb93225f8674b4374369688af80a6d32efcbfc70406ae2130d0bba
+  data.tar.gz: bb30e0f93629c922d9723c943061e9dddad28fcab7406ab24496cd5f893a5b6d
 SHA512:
-  metadata.gz: 7ead12b8b24f8aaae387ac81c400950f02d682260a136cecb5f97007e0607a89d55900ccd5e112f82d1137484295990042dfe0f98e91a570755b354ac08e2181
-  data.tar.gz: a269f71f8736c77172da1798f6f78623dccbcb887167193b7b0c998611786df554f5a035e66f2acf2dd6b292b6647caf7d0c822d2fed6301d68c85788c4322c6
+  metadata.gz: 6ca63659e83bd3a662612b41b2a645776ca06132241fd5033c169f8fb4be1547445fb34e2160731adced86f7ba6a8995f92a6d1fa573751002ce6d2c85869c70
+  data.tar.gz: 91f8b05eddcbfc0fbbc1cb44b7d9a0c340312207267bd4589237480f9fd95a5473af493d0c274d4231c6aaa914fb2820b7e3985039b1ad089eb523764d5ce89c

data/lib/hospital/diagnosis.rb

@@ -2,7 +2,7 @@ class Hospital::Diagnosis
   attr_reader :infos, :warnings, :errors, :name, :results
 
   def initialize name
-    @name               = name.to_s
+    @name = name.to_s
     reset
   end
 
@@ -27,6 +27,12 @@ class Hospital::Diagnosis
     "[#{vars.map{|v| "'#{v}'"}.join(', ')}]"
   end
 
+  def hide_value value
+    "#{value}"
+      .gsub(/(?<=.{1}).(?=.{2})/, '*')
+      .gsub(/\*{10,}/, '*****...*****')
+  end
+
   class Result
     attr_reader :message, :prefix
 
@@ -74,15 +80,6 @@ class Hospital::Diagnosis
   end
 
   def put_results
-    put_header "Checking #{name}:"
     results.each &:put
   end
-
-  private
-
-  def put_header message
-    puts ''
-    puts "### #{message}"
-  end
-
 end

data/lib/hospital/formatter.rb

@@ -0,0 +1,19 @@
+module Formatter
+  refine String do
+    def h1
+      "\n\e[4m\e[1m#{self}\e[0m"
+    end
+
+    def h2
+      "\n\e[4m#{self}\e[0m"
+    end
+
+    def red
+      "\e[31m#{self}\e[0m"
+    end
+
+    def yellow
+      "\e[33m#{self}\e[0m"
+    end
+  end
+end

data/lib/hospital/tasks/checkup.rake

@@ -3,13 +3,24 @@
 require_relative '../../hospital'
 
 desc 'Check system setup sanity.'
-task :doctor => :environment do
+task :doctor, [:verbose] => :environment do |t, args|
   # at_exit { Rake::Task['doctor:summary'].invoke if $!.nil? }
 
+  ActiveRecord::Base.connection_pool.disconnect!
+  ActiveSupport.on_load(:active_record) do
+    config = ActiveRecord::Base.configurations[Rails.env]
+    config['pool'] = 100
+    ActiveRecord::Base.establish_connection(config)
+  end
+  
+  verbose = args[:verbose] == "true"
+
   # silence warnings about double constant definitions
   Kernel::silence_warnings do
+    p "eager load all classes" if verbose
     Rails.application.eager_load!
   end
-
-  Hospital.checkup_all
+  
+  p "start checkup" if verbose
+  Hospital.do_checkup_all verbose: verbose
 end

data/lib/hospital/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Hospital
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

data/lib/hospital.rb

@@ -2,49 +2,113 @@
 
 require_relative "hospital/version"
 require_relative "hospital/diagnosis"
+require_relative "hospital/formatter"
+
+using Formatter
 
 module Hospital
   require_relative 'railtie' if defined?(Rails)
 
   class Error < StandardError; end
 
-  @@checkups    = {}
-  @@diagnosises = {}
+  class Checkup 
+    attr_reader :code, :condition, :diagnosis, :group, :skipped, :klass
 
-  def self.included(base)
-    raise Hospital::Error.new("Cannot include Hospital, please extend instead.")
-  end
+    def initialize klass, code, group: :general, title: nil, condition: -> { true }
+      @klass      = klass
+      @code       = code
+      @group      = group
+      @condition  = condition
+      @diagnosis  = Hospital::Diagnosis.new([klass.to_s, title].compact.join(' - '))
+    end
 
-  def self.extended(base)
-    @@checkups[base] = -> (diagnosis) do
-      diagnosis.add_warning("#{base}: No checks defined! Please call checkup with a lambda.")
+    def reset_diagnosis
+      diagnosis.reset
     end
-    @@diagnosises[base] = Hospital::Diagnosis.new(base)
-  end
 
-  def checkup &code
-    @@checkups[self] = code
-  end
+    def check verbose: false
+      diagnosis.reset
 
-  def self.checkup klass
-    @@diagnosises[klass].reset
-    @@checkups[klass].call(@@diagnosises[klass])
-    @@diagnosises[klass]
+      if condition.nil? || condition.call
+        @skipped = false
+        code.call(diagnosis)
+        diagnosis
+      else
+        @skipped = true
+        nil
+      end
+    rescue StandardError => e
+      diagnosis.add_error "Unrescued exception in #{klass}.checkup:\n#{e.inspect}.\nThis is a bug inside the checkup method that should be fixed!"
+    end
   end
 
-  def self.checkup_all
-    errcount = 0
-    @@checkups.keys.each do |klass|
-      checkup(klass)
-      diagnosis = @@diagnosises[klass]
-      diagnosis.put_results
-      errcount += diagnosis.errors.count
+  @@checkups = {}
+
+  class << self
+
+    def included(klass)
+      raise Hospital::Error.new("Cannot include Hospital, please extend instead.")
+    end
+
+    def extended(klass)
+      # only relevant if the class does not yet define a real checkup method
+      @@checkups[klass] = Checkup.new klass, -> (diagnosis) do
+        diagnosis.add_warning("#{klass}: No checks defined! Please call checkup with a lambda.")
+      end, group: :general
+    end
+
+    def do_checkup_all verbose: false
+      errcount  = 0
+      warcount  = 0
+
+      threads = @@checkups.map do |klass, checkup|
+        Thread.new do
+          Thread.current.report_on_exception = false
+          checkup.check(verbose: verbose)
+        end
+      end
+
+      threads.each &:join
+
+      @@checkups.group_by{|klass, checkup| checkup.group}.map do |group, checkups|
+        puts group_header(group)
+        first = false
+
+        checkups.each do |klass, checkup|
+          if diagnosis = checkup.diagnosis
+            errcount += diagnosis.errors.count
+            warcount += diagnosis.warnings.count
+
+            if !checkup.skipped
+              puts "Checking #{diagnosis.name}:".h2
+              diagnosis.put_results
+            elsif verbose
+              puts "Skipped #{diagnosis.name}.".h2
+            end
+          end
+        end
+      end
+
+      puts <<~END
+
+        #{"Summary:".h1}
+        #{"Errors:   #{errcount}".red}
+        #{"Warnings: #{warcount}".yellow}
+      END
+    end
+
+    # used to call the checkup for a specific class directly (in specs)
+    def do_checkup(klass)
+      @@checkups[klass].check
     end
 
-    puts <<~END
+    def group_header group
+      "### #{group.to_s.capitalize.gsub(/_/, ' ')} checks".h1
+    end
+  end
 
-      Summary:
-      Errors: #{errcount}
-    END
+  def checkup if: -> { true }, group: :general, title: nil, &code
+    @@checkups[self] = Checkup.new self, code, group: group, title: title, condition: binding.local_variable_get('if')
   end
+
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hospital
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Alexander
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-12-21 00:00:00.000000000 Z
+date: 2023-12-26 00:00:00.000000000 Z
 dependencies: []
 description: Imagine a team of little doctors creating diagnoses and creating a useful
   report.
@@ -22,6 +22,7 @@ files:
 - README.md
 - lib/hospital.rb
 - lib/hospital/diagnosis.rb
+- lib/hospital/formatter.rb
 - lib/hospital/tasks/checkup.rake
 - lib/hospital/version.rb
 - lib/railtie.rb