require 'puppet/indirector/terminus'

# The base class for YAML indirection termini.
class Puppet::Indirector::Yaml < Puppet::Indirector::Terminus
    def initialize
        # Make sure our base directory exists.
        Puppet.settings.use(:yaml)
    end

    # Read a given name's file in and convert it from YAML.
    def find(name)
        raise ArgumentError.new("You must specify the name of the object to retrieve") unless name
        file = path(name)
        return nil unless FileTest.exist?(file)

        begin
            return YAML.load(File.read(file))
        rescue => detail
            raise Puppet::Error, "Could not read YAML data for %s(%s): %s" % [indirection.name, name, detail]
        end
    end

    # Convert our object to YAML and store it to the disk.
    def save(object)
        raise ArgumentError.new("You can only save objects that respond to :name") unless object.respond_to?(:name)

        file = path(object.name)

        basedir = File.dirname(file)

        # This is quite likely a bad idea, since we're not managing ownership or modes.
        unless FileTest.exist?(basedir)
            Dir.mkdir(basedir)
        end

        File.open(file, "w", 0660) { |f| f.print YAML.dump(object) }
    end

    private

    # Return the path to a given node's file.
    def path(name)
        File.join(Puppet[:yamldir], self.name.to_s, name.to_s + ".yaml")
    end
end