object subclass working

2 files changed, 21 insertions, 0 deletions

diff --git a/lib/psych/visitors/to_ruby.rb b/lib/psych/visitors/to_ruby.rb
index f6a4d69..39c9194 100644
--- a/lib/psych/visitors/to_ruby.rb
+++ b/lib/psych/visitors/to_ruby.rb
@@ -97,6 +97,17 @@ module Psych
 
       def visit_Psych_Nodes_Mapping o
         case o.tag
+        when /!ruby\/object(:.*)?$/
+          name = $1.sub(/^:/, '')
+          h = Hash[*o.children.map { |c| accept c }]
+          s = name.split('::').inject(Object) { |k,sub|
+            k.const_get sub
+          }.allocate
+          h.each do |k,v|
+            s.instance_variable_set(:"@#{k}", v)
+          end
+          s
+
         when /!ruby\/struct(:.*)?$/
           klassname = $1
           h = Hash[*o.children.map { |c| accept c }].to_a
diff --git a/lib/psych/visitors/yast_builder.rb b/lib/psych/visitors/yast_builder.rb
index 296cb36..11e3b34 100644
--- a/lib/psych/visitors/yast_builder.rb
+++ b/lib/psych/visitors/yast_builder.rb
@@ -22,6 +22,16 @@ module Psych
         raise TypeError, "Can't dump #{target.class}"
       end
 
+      def visit_Object o
+        tag = ['!ruby/object', o.class.name].join(':')
+        @stack.push append Nodes::Mapping.new(nil, tag, false)
+        o.instance_variables.each do |iv|
+          accept iv.to_s.sub(/^@/, '')
+          accept o.instance_variable_get(iv)
+        end
+        @stack.pop
+      end
+
       def visit_Struct o
         tag = ['!ruby/struct', o.class.name].compact.join(':')