lib/tori/config.rb in tori-0.0.3 vs lib/tori/config.rb in tori-0.0.4

@@ -1,9 +1,17 @@
 module Tori
   class Config
-    attr_accessor :backend, :filename_callback
+    attr_accessor :backend
     def initialize
       @backend = nil
       @filename_callback = nil
+    end
+
+    def filename_callback(&block)
+      if block_given?
+        @filename_callback = block
+      else
+        @filename_callback
+      end
     end
   end
 end