activesupport/lib/active_support/actionable_error.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

# frozen_string_literal: true

require "active_support/concern"

module ActiveSupport
  # Actionable errors let's you define actions to resolve an error.
  #
  # To make an error actionable, include the <tt>ActiveSupport::ActionableError</tt>
  # module and invoke the +action+ class macro to define the action.
  #
  # An action needs a name and a procedure to execute. The name can be shown by
  # the action dispatching mechanism.
  module ActionableError
    extend Concern

    NonActionable = Class.new(StandardError)

    NoActions = Hash.new do |_, label| # :nodoc:
      raise NonActionable, "Cannot find action \"#{label}\" for #{self}"
    end

    included do
      class_attribute :_actions, default: NoActions.dup
    end

    def self.actions(error) # :nodoc:
      case error
      when String
        actions(error.constantize)
      when ActionableError, -> it { Class === it && it < ActionableError }
        error._actions
      when Exception
        NoActions
      else
        raise NonActionable, "#{error} is non-actionable"
      end
    end

    def self.dispatch(error, label) # :nodoc:
      actions(error)[label].call
    end

    module ClassMethods
      # Defines an action that can resolve the error.
      #
      #   class PendingMigrationError < MigrationError
      #     include ActiveSupport::ActionableError
      #
      #     action "Run pending migrations" do
      #       ActiveRecord::Tasks::DatabaseTasks.migrate
      #     end
      #   end
      def action(label, &block)
        _actions[label] = block
      end
    end
  end
end