lib/sup/draft.rb in sup-0.0.8 vs lib/sup/draft.rb in sup-0.1

@@ -40,10 +40,11 @@
   end
 end
 
 class DraftLoader < Source
   attr_accessor :dir
+  yaml_properties :cur_offset
 
   def initialize cur_offset=0
     dir = Redwood::DRAFT_DIR
     Dir.mkdir dir unless File.exists? dir
     super "draft://#{dir}", cur_offset, true, false
@@ -116,9 +117,7 @@
 
   def get_ids
     Dir.entries(@dir).select { |x| x =~ /^\d+$/ }.map { |x| x.to_i }.sort
   end
 end
-
-Redwood::register_yaml(DraftLoader, %w(cur_offset))
 
 end