class FileServing::Fileset

# File lib/puppet/file_serving/fileset.rb, line 15
def self.merge(*filesets)
  result = {}

  filesets.each do |fileset|
    fileset.files.each do |file|
      result[file] ||= fileset.path
    end
  end

  result
end

# File lib/puppet/file_serving/fileset.rb, line 57
def initialize(path, options = {})
  if Puppet.features.microsoft_windows?
    # REMIND: UNC path
    path = path.chomp(File::SEPARATOR) unless path =~ /^[A-Za-z]:\/$/
  else
    path = path.chomp(File::SEPARATOR) unless path == File::SEPARATOR
  end
  raise ArgumentError.new("Fileset paths must be fully qualified: #{path}") unless Puppet::Util.absolute_path?(path)

  @path = path

  # Set our defaults.
  @ignore = []
  @links = :manage
  @recurse = false
  @recurselimit = :infinite

  if options.is_a?(Puppet::Indirector::Request)
    initialize_from_request(options)
  else
    initialize_from_hash(options)
  end

  raise ArgumentError.new("Fileset paths must exist") unless stat = stat(path)
  raise ArgumentError.new("Fileset recurse parameter must not be a number anymore, please use recurselimit") if @recurse.is_a?(Integer)
end

# File lib/puppet/file_serving/fileset.rb, line 31
def files
  files = perform_recursion

  # Now strip off the leading path, so each file becomes relative, and remove
  # any slashes that might end up at the beginning of the path.
  result = files.collect { |file| file.sub(%r{^#{Regexp.escape(@path)}/*}, '') }

  # And add the path itself.
  result.unshift(".")

  result
end

# File lib/puppet/file_serving/fileset.rb, line 52
def ignore=(values)
  values = [values] unless values.is_a?(Array)
  @ignore = values
end

# File lib/puppet/file_serving/fileset.rb, line 45
def ignore?(path)
  return false if @ignore == [nil]

  # 'detect' normally returns the found result, whereas we just want true/false.
  ! @ignore.detect { |pattern| File.fnmatch?(pattern, path) }.nil?
end

# File lib/puppet/file_serving/fileset.rb, line 98
def initialize_from_hash(options)
  options.each do |option, value|
    method = option.to_s + "="
    begin
      send(method, value)
    rescue NoMethodError
      raise ArgumentError, "Invalid option '#{option}'"
    end
  end
end

# File lib/puppet/file_serving/fileset.rb, line 109
def initialize_from_request(request)
  [:links, :ignore, :recurse, :recurselimit, :checksum_type].each do |param|
    if request.options.include?(param) # use 'include?' so the values can be false
      value = request.options[param]
    elsif request.options.include?(param.to_s)
      value = request.options[param.to_s]
    end
    next if value.nil?
    value = true if value == "true"
    value = false if value == "false"
    value = Integer(value) if value.is_a?(String) and value =~ /^\d+$/
    send(param.to_s + "=", value)
  end
end

# File lib/puppet/file_serving/fileset.rb, line 84
def links=(links)
  links = links.to_sym
  raise(ArgumentError, "Invalid :links value '#{links}'") unless [:manage, :follow].include?(links)
  @links = links
  @stat_method = links == :manage ? :lstat : :stat
end

# File lib/puppet/file_serving/fileset.rb, line 93
def recurse?(depth)
  # recurse if told to, and infinite recursion or current depth not at the limit
  self.recurse and (self.recurselimit == :infinite or depth <= self.recurselimit)
end

# File lib/puppet/file_serving/fileset.rb, line 162
def stat(path)
  @stat_method ||= self.links == :manage ? :lstat : :stat

  begin
    return File.send(@stat_method, path)
  rescue
    # If this happens, it is almost surely because we're
    # trying to manage a link to a file that does not exist.
    return nil
  end
end