railbow 0.6.0 → 0.7.0

data/lib/railbow/tasks/stats.rake

@@ -1,85 +1,21 @@
 # frozen_string_literal: true
 
-# Defer prepend until CodeStatistics is available.
-# CodeStatistics is autoloaded by Rails when the `stats` task runs,
-# but is not available at rake-file load time.
-#
-# Rails 8.1 namespaced the class to Rails::CodeStatistics, removed
-# the bare `require "code_statistics"` shim, and dropped the built-in
-# `stats` rake task. We handle both old and new layouts.
-Rails.application.config.after_initialize do
-  klass = nil
+require_relative "../stats_formatter"
 
-  begin
-    require "rails/code_statistics"
-    klass = Rails::CodeStatistics
-  rescue LoadError
-    # fall through
-  end
-
-  unless klass
-    begin
-      require "code_statistics"
-      klass = CodeStatistics
-    rescue LoadError
-      # code_statistics not available (e.g., production without railties dev deps)
-    end
-  end
+# Rails 7.2 runs `rails stats` as a rake task that never boots the app, so the
+# railtie initializer does not fire there: install from here as well.
+Railbow::StatsFormatter.install!
 
-  if klass
-    require_relative "../stats_formatter"
-    klass.prepend(Railbow::StatsFormatter)
-  end
-end
-
-# Rails 8.1 removed the built-in `stats` task.  Re-define it so that
-# `rails stats` / `rake stats` keeps working with Railbow formatting.
-unless Rake::Task.task_defined?(:stats)
+# Rails 8.1 dropped the `stats` rake task for the `rails stats` command; this
+# keeps `rake stats` working. Gated on the version, not task_defined?: railtie
+# tasks load before Rails' own, so on older Rails both would run.
+if Rails.gem_version >= Gem::Version.new("8.1")
   desc "Report code statistics (lines / LOC / classes / methods / M/C / LOC/M)"
   task stats: :environment do
-    klass = if defined?(Rails::CodeStatistics)
-      Rails::CodeStatistics
-    elsif defined?(CodeStatistics)
-      CodeStatistics
-    end
-
-    unless klass
-      abort "CodeStatistics is not available. Make sure railties is loaded."
-    end
-
-    stat_directories = [
-      ["Controllers", "app/controllers"],
-      ["Helpers", "app/helpers"],
-      ["Jobs", "app/jobs"],
-      ["Models", "app/models"],
-      ["Mailers", "app/mailers"],
-      ["Channels", "app/channels"],
-      ["JavaScripts", "app/javascript"],
-      ["Libraries", "lib"],
-      ["APIs", "app/apis"],
-      ["Controller tests", "test/controllers"],
-      ["Helper tests", "test/helpers"],
-      ["Job tests", "test/jobs"],
-      ["Model tests", "test/models"],
-      ["Mailer tests", "test/mailers"],
-      ["Channel tests", "test/channels"],
-      ["Integration tests", "test/integration"],
-      ["System tests", "test/system"],
-      ["Controller specs", "spec/controllers"],
-      ["Helper specs", "spec/helpers"],
-      ["Job specs", "spec/jobs"],
-      ["Model specs", "spec/models"],
-      ["Mailer specs", "spec/mailers"],
-      ["Channel specs", "spec/channels"],
-      ["Request specs", "spec/requests"],
-      ["System specs", "spec/system"]
-    ]
-
-    root = Rails.root.to_s
-    pairs = stat_directories
-      .map { |name, dir| [name, File.join(root, dir)] }
+    stat_directories = Rails::CodeStatistics.directories
+      .map { |name, dir| [name, Rails.root.join(dir)] }
       .select { |_, dir| File.directory?(dir) }
 
-    puts klass.new(*pairs)
+    Rails::CodeStatistics.new(*stat_directories).to_s
   end
 end

data/lib/railbow/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Railbow
-  VERSION = "0.6.0"
+  VERSION = "0.7.0"
 end

data/lib/railbow.rb

@@ -7,10 +7,10 @@ require_relative "railbow/config"
 module Railbow
   class Error < StandardError; end
 
-  # Returns true when Railbow formatting should be disabled.
-  # Checks for explicit opt-out, standard conventions, CI, and LLM agents.
-  # RBW_PLAIN always wins; RBW_FORCE beats every auto-detection below it.
+  # True when railbow should leave the stock Rails output alone. RBW_PLAIN
+  # always wins; RBW_FORCE beats every auto-detection below it.
   def self.plain?
+    return true if failure
     return true if Params.plain?
     return false if Params.force?
     return true if ENV.key?("NO_COLOR")
@@ -18,7 +18,31 @@ module Railbow
     return true if ENV.key?("CI")
     !$stdout.tty?
   end
+
+  class << self
+    # The first error railbow's own rendering raised in this process. Once set,
+    # railbow stays plain so a broken formatter cannot fail twice.
+    attr_accessor :failure
+  end
+
+  # A railbow bug must never break the Rails command it decorates: an error
+  # falls back to the stock output and switches railbow off for the process.
+  # Errors Rails raises inside the fallback still propagate.
+  def self.guard(fallback)
+    yield
+  rescue => e
+    record_failure(e)
+    fallback.call
+  end
+
+  def self.record_failure(error)
+    return if failure
+
+    self.failure = error
+    frame = error.backtrace_locations&.find { |l| l.path.to_s.include?("/railbow/") }
+    where = frame ? " at #{frame.path[%r{lib/railbow/.*}] || frame.path}:#{frame.lineno}" : ""
+    warn "railbow: #{error.class}: #{error.message}#{where}; showing the default Rails output instead"
+  end
 end
 
-# Load Railtie if Rails is available
 require_relative "railbow/railtie" if defined?(Rails::Railtie)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: railbow
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Eugene M
@@ -79,6 +79,7 @@ files:
 - lib/railbow.rb
 - lib/railbow/about_formatter.rb
 - lib/railbow/calendar.rb
+- lib/railbow/clock.rb
 - lib/railbow/color_assigner.rb
 - lib/railbow/config.rb
 - lib/railbow/demo/fixtures.rb
@@ -119,7 +120,6 @@ homepage: https://github.com/amberpixels/railbow
 licenses:
 - MIT
 metadata:
-  homepage_uri: https://github.com/amberpixels/railbow
   source_code_uri: https://github.com/amberpixels/railbow
   changelog_uri: https://github.com/amberpixels/railbow/blob/main/CHANGELOG.md
   bug_tracker_uri: https://github.com/amberpixels/railbow/issues