require 'puppet'
require 'yaml'

module Puppet
    # The transportable objects themselves.  Basically just a hash with some
    # metadata and a few extra methods.  I used to have the object actually
    # be a subclass of Hash, but I could never correctly dump them using
    # YAML.
    class TransObject
        include Enumerable
        attr_accessor :type, :name, :file, :line, :configuration

        attr_writer :tags

        %w{has_key? include? length delete empty? << [] []=}.each { |method|
            define_method(method) do |*args|
                @params.send(method, *args)
            end
        }

        def each
            @params.each { |p,v| yield p, v }
        end

        def initialize(name,type)
            @type = type
            @name = name
            @params = {}
            @tags = []
        end

        def longname
            return [@type,@name].join('--')
        end

        def ref
            unless defined? @ref
                if @type == :component
                    @ref = @name
                else
                    @ref = "%s[%s]" % [type_capitalized, name]
                end
            end
            @ref
        end

        def tags
            return @tags
        end

        # Convert a defined type into a component.
        def to_component
            tmpname = nil

            tmpname = "%s[%s]" % [type_capitalized, self.name]

            trans = TransObject.new(tmpname, :component)

            @params.each { |param,value|
                next unless Puppet::Type::Component.validattr?(param)
                Puppet.debug "Defining %s on %s of type %s" % [param,@name,@type]
                trans[param] = value
            }
            Puppet::Type::Component.create(trans)
        end

        def to_hash
            @params.dup
        end

        def to_s
            return "%s(%s) => %s" % [@type,@name,super]
        end

        def to_manifest
            "#{self.type.to_s} { \'#{self.name}\':\n%s\n}" % @params.collect { |p, v|
                if v.is_a? Array
                    "    #{p} => [\'#{v.join("','")}\']"
                else
                    "    #{p} => \'#{v}\'"
                end
            }.join(",\n")
        end

        def to_yaml_properties
            instance_variables.reject { |v| %w{@ref}.include?(v) }
        end

        def to_ref
            unless defined? @res_ref
                @res_ref = "%s[%s]" % [type_capitalized, self.name]
            end
            @res_ref
        end

        def to_type
            retobj = nil
            if typeklass = Puppet::Type.type(self.type)
                return typeklass.create(self)
            else
                return to_component
            end

            return retobj
        end

        # Return the type fully capitalized correctly.
        def type_capitalized
            type.to_s.split("::").collect { |s| s.capitalize }.join("::")
        end
    end

    # Just a linear container for objects.  Behaves mostly like an array, except
    # that YAML will correctly dump them even with their instance variables.
    class TransBucket
        include Enumerable

        attr_accessor :name, :type, :file, :line, :classes, :keyword, :top, :configuration

        %w{delete shift include? length empty? << []}.each { |method|
            define_method(method) do |*args|
                #Puppet.warning "Calling %s with %s" % [method, args.inspect]
                @children.send(method, *args)
                #Puppet.warning @params.inspect
            end
        }

        # Recursively yield everything.
        def delve(&block)
            @children.each do |obj|
                block.call(obj)
                if obj.is_a? self.class
                    obj.delve(&block)
                else
                    obj
                end
            end
        end

        def each
            @children.each { |c| yield c }
        end

        # Turn our heirarchy into a flat list
        def flatten
            @children.collect do |obj|
                if obj.is_a? Puppet::TransBucket
                    obj.flatten
                else
                    obj
                end
            end.flatten
        end

        def initialize(children = [])
            @children = children
        end

        def push(*args)
            args.each { |arg|
                case arg
                when Puppet::TransBucket, Puppet::TransObject
                    # nada
                else
                    raise Puppet::DevError,
                        "TransBuckets cannot handle objects of type %s" %
                            arg.class
                end
            }
            @children += args
        end

        # Convert to a parseable manifest
        def to_manifest
            unless self.top
                unless defined? @keyword and @keyword
                    raise Puppet::DevError, "No keyword; cannot convert to manifest"
                end
            end

            str = "#{@keyword} #{@name} {\n%s\n}"
            str % @children.collect { |child|
                child.to_manifest
            }.collect { |str|
                if self.top
                    str
                else
                    str.gsub(/^/, "    ") # indent everything once
                end
            }.join("\n\n") # and throw in a blank line
        end

        def to_yaml_properties
            instance_variables
        end

        # Create a resource graph from our structure.
        def to_configuration
            configuration = Puppet::Node::Configuration.new(Facter.value("hostname")) do |config|
                delver = proc do |obj|
                    obj.configuration = config
                    unless container = config.resource(obj.to_ref)
                        container = obj.to_type
                        config.add_resource container
                    end
                    obj.each do |child|
                        child.configuration = config
                        unless resource = config.resource(child.to_ref)
                            next unless resource = child.to_type
                            config.add_resource resource
                        end
                        config.add_edge!(container, resource)
                        if child.is_a?(self.class)
                            delver.call(child)
                        end
                    end
                end
                
                delver.call(self)
            end
            
            return configuration
        end

        def to_ref
            unless defined? @ref
                if self.type and self.name
                    @ref = "%s[%s]" % [self.type, self.name]
                else
                    @ref = nil
                end
            end
            @ref
        end

        def to_type
            unless defined? @type
                Puppet.debug "TransBucket '%s' has no type" % @name
            end

            # Nodes have the same name and type
            if self.name
                tmpname = "%s[%s]" % [@type, self.name]
            else
                tmpname = @type
            end
            trans = TransObject.new(tmpname, :component)
            if defined? @parameters
                @parameters.each { |param,value|
                    Puppet.debug "Defining %s on %s of type %s" %
                        [param,@name,@type]
                    trans[param] = value
                }
            end
            return Puppet::Type::Component.create(trans)
        end

        def param(param,value)
            unless defined? @parameters
                @parameters = {}
            end
            @parameters[param] = value
        end

    end
end