124 lines
2.2 KiB
Ruby
124 lines
2.2 KiB
Ruby
require 'set'
|
|
|
|
module Rails
|
|
class Application
|
|
module PathParent
|
|
def method_missing(id, *args)
|
|
name = id.to_s
|
|
|
|
if name =~ /^(.*)=$/
|
|
@children[$1] = Path.new(args.first, @root)
|
|
elsif path = @children[name]
|
|
path
|
|
else
|
|
super
|
|
end
|
|
end
|
|
end
|
|
|
|
class Root
|
|
include PathParent
|
|
|
|
attr_reader :path
|
|
def initialize(path)
|
|
raise unless path.is_a?(String)
|
|
|
|
@children = {}
|
|
|
|
# TODO: Move logic from set_root_path initializer
|
|
@path = File.expand_path(path)
|
|
@root = self
|
|
@load_once, @eager_load, @all_paths = [], [], []
|
|
end
|
|
|
|
def load_once
|
|
@load_once.uniq!
|
|
@load_once
|
|
end
|
|
|
|
def eager_load
|
|
@eager_load.uniq!
|
|
@eager_load
|
|
end
|
|
|
|
def all_paths
|
|
@all_paths.uniq!
|
|
@all_paths
|
|
end
|
|
|
|
def load_paths
|
|
all_paths.map { |path| path.paths }.flatten
|
|
end
|
|
|
|
def add_to_load_path
|
|
load_paths.reverse_each do |path|
|
|
$LOAD_PATH.unshift(path) if File.directory?(path)
|
|
end
|
|
end
|
|
end
|
|
|
|
class Path
|
|
include PathParent
|
|
|
|
attr_reader :path
|
|
attr_accessor :glob
|
|
|
|
def initialize(path, root)
|
|
@children = {}
|
|
@root = root
|
|
@paths = [path].flatten
|
|
@glob = "**/*.rb"
|
|
end
|
|
|
|
def push(path)
|
|
@paths.push path
|
|
end
|
|
|
|
alias << push
|
|
|
|
def unshift(path)
|
|
@paths.unshift path
|
|
end
|
|
|
|
def concat(paths)
|
|
@paths.concat paths
|
|
end
|
|
|
|
def load_once!
|
|
@load_once = true
|
|
@root.load_once.push *self.paths
|
|
end
|
|
|
|
def load_once?
|
|
@load_once
|
|
end
|
|
|
|
def eager_load!
|
|
@eager_load = true
|
|
@root.all_paths << self
|
|
@root.eager_load.push *self.paths
|
|
end
|
|
|
|
def eager_load?
|
|
@eager_load
|
|
end
|
|
|
|
def load_path!
|
|
@load_path = true
|
|
@root.all_paths << self
|
|
end
|
|
|
|
def load_path?
|
|
@load_path
|
|
end
|
|
|
|
def paths
|
|
@paths.map do |path|
|
|
path.index('/') == 0 ? path : File.join(@root.path, path)
|
|
end
|
|
end
|
|
|
|
alias to_a paths
|
|
end
|
|
end
|
|
end |