require 'representable/bindings/yaml_bindings'

module Representable
  module YAML
    include Hash

    def self.included(base)
      base.class_eval do
        include Representable
        #self.representation_wrap = true # let representable compute it.
        register_feature Representable::YAML
      end
    end


    def from_yaml(doc, options={})
      hash = Psych.load(doc)
      from_hash(hash, options, PropertyBinding)
    end

    # Returns a Nokogiri::XML object representing this object.
    def to_ast(options={})
      #root_tag = options[:wrap] || representation_wrap

      Psych::Nodes::Mapping.new.tap do |map|
        create_representation_with(map, options, PropertyBinding)
      end
    end

    def to_yaml(*args)
      stream = Psych::Nodes::Stream.new
      stream.children << doc = Psych::Nodes::Document.new

      doc.children << to_ast(*args)
      stream.to_yaml
    end
  end
end