commands 0.0.3 → 0.0.4

data/commands.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name    = 'commands'
-  s.version = '0.0.3'
+  s.version = '0.0.4'
   s.author  = 'David Heinemeier Hansson'
   s.email   = 'david@37signals.com'
   s.summary = 'Run Rake/Rails commands through the console'

data/lib/commands.rb

@@ -13,33 +13,36 @@ class Commands
   include Rake::DSL
   
   def initialize
-    load Rails.root.join('Rakefile')
-    Rails.application.load_generators
+    load_rake_tasks
+    load_rails_generators
   end
   
-  def rake(task, *args)
-    silence_active_record_logger { Rake::Task[task].invoke(*args) }
-
-    # Rake by default only allows tasks to be run once per session
-    Rake.application.tasks.each(&:reenable)
-
+  def rake(task = nil, *args)
+    task.nil? ? print_rake_tasks : invoke_rake_task(task, *args)
     "Completed"
-  rescue SystemExit
-    # Swallow exit/abort calls as we'll never want to exit the IRB session
+  rescue SystemExit, RuntimeError
     "Failed"
   end
 
   # FIXME: Turn this into calls directly to the test classes, so we don't have to load environment again.
   # Also need to toggle the environment to test and back to dev after running.
   def test(what = nil)
-    case what
-    when NilClass, :all
-      rake "test:units"
-    when String
-      ENV['TEST'] = "test/#{what}_test.rb"
-      rake "test:single"
-      ENV['TEST'] = nil
+    forking do
+      case what
+      when NilClass
+        print_test_usage
+      when "all"
+        rake "test"
+      when /^[^\/]+$/ # models
+        rake "test:#{what}"
+      when /[\/]+/ # models/person
+        ENV['TEST'] = "test/#{what}_test.rb"
+        rake "test:single"
+        ENV['TEST'] = nil
+      end
     end
+
+    "Completed"
   end
 
   def generate(argv = nil)
@@ -56,6 +59,52 @@ class Commands
   
   
   private
+    def load_rake_tasks
+      Rake::TaskManager.record_task_metadata = true # needed to capture comments from define_task
+      load Rails.root.join('Rakefile')
+    end
+    
+    def load_rails_generators
+      Rails.application.load_generators
+    end
+
+
+    def print_rake_tasks
+      Rake.application.options.show_tasks = :tasks
+      Rake.application.options.show_task_pattern = Regexp.new('')
+      Rake.application.display_tasks_and_comments
+    end
+
+    def invoke_rake_task(task, *args)
+      silence_active_record_logger { Rake::Task[task].invoke(*args) }
+      Rake.application.tasks.each(&:reenable) # Rake by default only allows tasks to be run once per session
+    end
+
+
+    def print_test_usage
+      puts <<-EOT
+Usage:
+  test "WHAT"
+
+Description:
+    Runs either a full set of test suites or single suite.
+    
+    If you supply WHAT with either models, controllers, helpers, integration, or performance,
+    those whole sets will be run.
+    
+    If you supply WHAT with models/person, just test/models/person_test.rb will be run.
+EOT
+    end
+
+    def forking
+      pid = Kernel.fork do
+        yield
+        Kernel.exit
+      end
+      Process.wait pid
+    end
+
+
     def generator(name, argv = nil)
       if argv.nil?
         # FIXME: I don't know why we can't just catch SystemExit here, then we wouldn't need this if block

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: commands
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -34,6 +34,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ./commands-0.0.2.gem
+- ./commands-0.0.3.gem
 - ./commands.gemspec
 - ./Gemfile
 - ./Gemfile.lock