aboutsummaryrefslogtreecommitdiffstats
path: root/railties/lib/rails/paths.rb
diff options
context:
space:
mode:
Diffstat (limited to 'railties/lib/rails/paths.rb')
-rw-r--r--railties/lib/rails/paths.rb124
1 files changed, 124 insertions, 0 deletions
diff --git a/railties/lib/rails/paths.rb b/railties/lib/rails/paths.rb
new file mode 100644
index 0000000000..d2f6d83659
--- /dev/null
+++ b/railties/lib/rails/paths.rb
@@ -0,0 +1,124 @@
+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 \ No newline at end of file