1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/railties/lib/rails/initializable.rb

110 lines
No EOL
2.6 KiB
Ruby

module Rails
module Initializable
def self.included(base)
base.extend ClassMethods
end
class Initializer
attr_reader :name, :block
def initialize(name, context, options, &block)
@name, @context, @options, @block = name, context, options, block
end
def before
@options[:before]
end
def after
@options[:after]
end
def global
@options[:global]
end
alias global? global
def run(*args)
@context.instance_exec(*args, &block)
end
def bind(context)
return self if @context
Initializer.new(@name, context, @options, &block)
end
end
class Collection < Array
def initialize(initializers = [])
super()
initializers.each do |initializer|
if initializer.before
index = index_for(initializer.before)
elsif initializer.after
index = index_for(initializer.after)
index += 1 if index
else
index = length
end
insert(index || -1, initializer)
end
end
def +(other)
Collection.new(to_a + other.to_a)
end
def index_for(name)
initializer = find { |i| i.name == name }
initializer && index(initializer)
end
end
def run_initializers(*args)
return if instance_variable_defined?(:@ran)
initializers.each do |initializer|
initializer.run(*args)
end
@ran = true
end
def initializers
@initializers ||= begin
initializers = self.class.initializers_for(:instance)
Collection.new(initializers.map { |i| i.bind(self) })
end
end
module ClassMethods
def initializers
@initializers ||= []
end
def initializers_for(scope = :global)
initializers = Collection.new
ancestors.reverse_each do |klass|
next unless klass.respond_to?(:initializers)
initializers = initializers + klass.initializers.select { |i|
(scope == :global) == !!i.global?
}
end
initializers
end
def initializer(name, opts = {}, &blk)
raise ArgumentError, "A block must be passed when defining an initializer" unless blk
@initializers ||= []
@initializers << Initializer.new(name, nil, opts, &blk)
end
def run_initializers(*args)
return if @ran
initializers_for(:global).each do |initializer|
instance_exec(*args, &initializer.block)
end
@ran = true
end
end
end
end