lib/seedbank/runner.rb in seedbank-0.3.0 vs lib/seedbank/runner.rb in seedbank-0.4.0.pre

@@ -1,16 +1,38 @@
 module Seedbank
-  class Runner < Module
-  
-    def initialize(task)
-      @task = task
-      super()
+  class Runner
+
+    def initialize
+      @_memoized = {}
     end
 
+    def let(name, &block)
+      name = String(name)
+
+      raise ArgumentError.new("#{name} is already defined") if respond_to?(name, true)
+
+      __eigenclass.instance_exec(name) do |name|
+        define_method(name) do
+          @_memoized.fetch(name) { |key| @_memoized[key] = instance_eval(&block) }
+        end
+      end
+
+    end
+
+    def let!(name, &block)
+      let(name, &block)
+      send name
+    end
+
+    def evaluate(seed_task, seed_file)
+      @_seed_task = seed_task
+      instance_eval(File.read(seed_file), seed_file)
+    end
+
     # Run this seed after the specified dependencies have run
     # @param dependencies [Array] seeds to run before the block is executed
-    # 
+    #
     # If a block is specified the contents of the block are executed after all the
     # dependencies have been executed.
     #
     # If no block is specified just the dependencies are run. This makes it possible
     # to create shared dependencies. For example
@@ -20,12 +42,26 @@
     #
     # Would look for a db/seeds/shared/users.seeds.rb seed and execute it.
     def after(*dependencies, &block)
       dependencies.flatten!
       dependencies.map! { |dep| "db:seed:#{dep}"}
-      dependent_task_name =  @task.name + ':body'
-      dependent_task = Rake::Task.define_task(dependent_task_name => dependencies, &block)
+      dependent_task_name =  @_seed_task.name + ':body'
+
+      if Rake::Task.task_defined?(dependent_task_name)
+        dependent_task = Rake::Task[dependent_task_name]
+      else
+        dependent_task = Rake::Task.define_task(dependent_task_name => dependencies, &block)
+      end
+
       dependent_task.invoke
     end
-  
+
+    private
+
+    def __eigenclass
+      class << self
+        self
+      end
+    end
+
   end
 end