lib/sniff/rake_tasks.rb in sniff-0.4.0 vs lib/sniff/rake_tasks.rb in sniff-0.4.1

@@ -8,20 +8,18 @@
 require 'rocco/tasks'
 
 
 module Sniff
   class RakeTasks
-    def self.define_tasks
-      new.define_tasks
+    def self.define_tasks(&blk)
+      new.define_tasks(&blk)
     end
 
     attr_accessor :earth_domains
 
     def initialize
       yield self if block_given?
-
-      define_task
     end
 
     def earth_domains
       @earth_domains ||= :all
     end
@@ -35,10 +33,9 @@
       full_cmd << "cd #{dir} && " if dir
       full_cmd << "unset GIT_DIR && unset GIT_INDEX_FILE && unset GIT_WORK_TREE && git #{cmd}"
       sh full_cmd, &blk
     end
 
-  private
     def define_tasks
       task :console do
         require 'sniff'
         cwd = Dir.pwd
         Sniff.init cwd, :earth => earth_domains