require "yml_reader/version"
require 'yaml'
require 'erb'

module YmlReader

  #
  # Set the directory to use when reading yml files
  #
  def yml_directory=(directory)
    @yml_directory = directory
  end

  #
  # Returns the directory to be used when reading yml files
  #
  def yml_directory
    return @yml_directory if @yml_directory
    return default_directory if self.respond_to? :default_directory
    nil
  end

  #
  # Loads the requested file.  It will look for the file in the
  # directory specified by a call to the yml_directory= method.
  # The parameter can also be a comma delimited list of files to
  # load and merge.
  #
  def load(filename)
    files= filename.include?(',') ? filename.split(',') : [filename]
    @yml = files.inject({}) do |total_merge,file|
      total_merge.merge!(::YAML.load(ERB.new(File.read("#{yml_directory}/#{file}")).result(binding)))
    end
  end
  
  def include_yml(filename)
    ERB.new(IO.read("#{yml_directory}/#{filename}")).result
  end

end