# Author::    Tyler Rick
# Copyright:: Copyright (c) 2007 QualitySmith, Inc.
# License::   Ruby License
# Submit to Facets?:: Yes

require 'rubygems'
require 'facets/more/filelist'

require 'facets/core/kernel/require_local'
require_local '../file/exact_match_regexp'

module Kernel

  # Recursively <tt>require</tt>s all Ruby files in <tt>base_dir</tt> or any of its subdirectories.
  # <tt>:exclude</tt>:       An array of regular expressions or glob patterns that will be passed to FileList#exclude. If you specify this option, a file will not be included if it matches *any* of these patterns.
  # <tt>:exclude_files</tt>: An array of filenames to exclude. These will be matched exactly, so if you tell it to exclude 'bar.rb', 'foobar.rb' will _not_ be excluded.
  #
  # Note:
  #
  # Examples:
  #   require_all 'lib/', :exclude => [/ignore/, /bogus/]   # will require 'lib/a.rb', 'lib/long/path/b.rb', but not 'lib/ignore/c.rb'
  #   require_all File.dirname(__FILE__), :exclude_files => ['blow_up_stuff.rb']
  def require_all(base_dir, options = {})
    base_dir += '/' unless base_dir[-1].chr == '/'
    exclusions = nil
    files = FileList[base_dir + "**/*.rb"]
    files = files.exclude(*exclusions)                                       if (exclusions = options.delete(:exclude))
    files = files.exclude(*exclusions.map {|a| File.exact_match_regexp(a) }) if (exclusions = options.delete(:exclude_files))

    files.each do |filename|
      # puts "requiring #{filename}" if filename =~ /test/
      require filename
    end
  end

end

#  _____         _
# |_   _|__  ___| |_
#   | |/ _ \/ __| __|
#   | |  __/\__ \ |_
#   |_|\___||___/\__|
#
=begin test
require 'tmpdir'
require 'fileutils'

class TheTest < Test::Unit::TestCase
  def setup
    @main_subdir = 'require_all_test'
    @base_dir = "#{Dir.tmpdir}/#{@main_subdir}"
    FileUtils.mkdir              @base_dir
    FileUtils.mkdir_p @deep_dir = @base_dir + "/really/really/deep/subdir"
    $loaded = Array.new(2)
  end
  def teardown
    FileUtils.rm_rf @base_dir
  end


  def test_repeat_requires
    $loaded[0] = 0
    File.open("#{@base_dir}/moo.rb", "w") {|f| f.puts "$loaded[0] += 1"}
    require_all File.dirname(@base_dir)
    assert_equal [1, nil], $loaded

    require "#{@base_dir}/moo.rb"
    assert_equal [1, nil], $loaded

    # Good! It refuses to load it again, even if we drop the ".rb" part!
    require "#{@base_dir}/moo"
    assert_equal [1, nil], $loaded

    # But, we can still trick it!
    $LOAD_PATH << @base_dir
    require "moo"
    assert_equal [2, nil], $loaded

    load "moo.rb"
    assert_equal [3, nil], $loaded
  end

  def test_deep_subdir
    File.open("#{@base_dir}/flip.rb", "w") {|f| f.puts "$loaded[0] = true"}
    File.open("#{@deep_dir}/flop.rb", "w") {|f| f.puts "$loaded[1] = true"}

    require_all File.dirname(@base_dir)
    assert_equal [true, true], $loaded
  end

  def test_exclude_pattern
    File.open("#{@base_dir}/require_me.rb", "w")       {|f| f.puts "$loaded[0] = true"}
    File.open("#{@base_dir}/please_ignore_me.rb", "w") {|f| f.puts "$loaded[1] = true"}

    require_all File.dirname(@base_dir), :exclude => [/ignore/]
    assert_equal [true, nil], $loaded
  end

  def test_exclude_filename
    File.open("#{@base_dir}/yes.rb", "w") {|f| f.puts "$loaded[0] = true"}
    File.open("#{@base_dir}/all.rb", "w") {|f| f.puts "$loaded[1] = true"}

    # :todo: Interesting how none of these patterns work. I would have expected them to. Is there a bug in FileList? Find out...
    #   /usr/lib/ruby/gems/1.8/gems/facets-1.8.51/lib/facets/more/filelist.rb
    #require_all File.dirname(@base_dir), :exclude => ['all.rb']
    #require_all File.dirname(@base_dir), :exclude => ['**/all']
    #require_all File.dirname(@base_dir), :exclude => [/^all\.rb$/]
    # This works, but it's too much to expect users to type out!:
    #require_all File.dirname(@base_dir), :exclude => [/(\/|^)all\.rb$/]
    
    # So...... I added an :exclude_files option so that people wouldn't have to!
    require_all File.dirname(@base_dir), :exclude_files => ['all.rb']

    assert_equal [true, nil], $loaded
  end

end
=end