gosu_extensions 0.1.4 → 0.1.5

data/VERSION

@@ -1 +1 @@
-0.1.4
+0.1.5

data/lib/traits/generator.rb

@@ -10,11 +10,25 @@ module Generator
       self.send :include, InstanceMethods
       
       rate   = options[:every]
-      til    = options[:until] || 100
-      offset = options[:starting_at] || 1
+      til    = options[:until]
+      offset = options[:starting_at]
       
-      InitializerHooks.register self do
-        start_generating klass, rate, til, offset
+      define_method :generated_class do
+        klass
+      end
+      class_eval <<-GENERATION
+def start_generating every_rate = #{rate || 10}, til = #{til || 100}, offset = #{offset || 0}
+  return if til <= 0
+  threaded offset, &generation(generated_class, every_rate, til)
+end
+      GENERATION
+      
+      # If an offset is given, start generating after the offset, else after start_generating is called.
+      #
+      if offset
+        InitializerHooks.register self do
+          start_generating
+        end
       end
       
     end
@@ -23,15 +37,17 @@ module Generator
   
   module InstanceMethods
     
-    def start_generating klass, every_rate, til, offset
-      return if til <= 0
-      threaded offset, &generation(klass, every_rate, til)
-    end
+    # See in #generates.
+    #
+    # def start_generating every_rate, til, offset, klass
+    #   return if til <= 0
+    #   threaded offset, &generation(klass, every_rate, til)
+    # end
     
     def generation klass, every_rate, til
       lambda do
         self.generate klass
-        self.start_generating klass, every_rate, til - every_rate, every_rate
+        self.start_generating every_rate, til - every_rate, every_rate
       end
     end
     

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 4
-  version: 0.1.4
+  - 5
+  version: 0.1.5
 platform: ruby
 authors: 
 - Florian Hanke