lib/rio/ext.rb in rio-0.3.4 vs lib/rio/ext.rb in rio-0.3.6

@@ -34,27 +34,33 @@
 # <b>Rio is pre-alpha software. 
 # The documented interface and behavior is subject to change without notice.</b>
 
 
 require 'rio/ext/csv'
+require 'rio/ext/yaml'
 module RIO
   module Ext #:nodoc: all
-    OUTPUT_SYMS = CSV::Output.instance_methods.build_hash { |sym| [sym.to_s,1] }
+    OUTPUT_SYMS = (CSV::Output.instance_methods + YAML::Output.instance_methods).build_hash { |sym| [sym.to_s,1] }
+
     module Cx
       include CSV::Cx
+      include YAML::Cx
     end
   end
   module Ext
     module Input
       def add_extensions(obj)
+        #p "add_extensions(#{obj.inspect})"
         obj.extend(CSV::Input) if obj.csv?
+        obj.extend(YAML::Input) if obj.yaml?
         obj
       end
       module_function :add_extensions
     end
     module Output
       def add_extensions(obj)
         obj.extend(CSV::Output) if obj.csv?
+        obj.extend(YAML::Output) if obj.yaml?
         obj
       end
       module_function :add_extensions
     end
   end