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