theine 0.0.3 → 0.0.4

data/lib/theine/client.rb

@@ -97,7 +97,16 @@ module Theine
 
   private
     def run_command
-      @worker.command_rails(@argv)
+      case @argv[0]
+      when "rake"
+        @argv.shift
+        @worker.command_rake(@argv)
+      when "rspec"
+        @argv.shift
+        @worker.command_rspec(@argv)
+      else
+        @worker.command_rails(@argv)
+      end
     rescue DRb::DRbConnError
       $stderr.puts "\nTheine closed the connection."
     end

data/lib/theine/worker.rb

@@ -46,15 +46,32 @@ module Theine
 
       require 'pry'
       ::Rails.application.config.console = Pry
-
-      Pry.config.input = stdin
-      Pry.config.output = stdout
+      pry_setup
 
       require 'rails/commands'
       sleep 0.1 # allow Pumps to finish
       DRb.stop_service
     end
 
+    def command_rake(argv)
+      pry_setup
+      ::Rails.application.load_tasks
+      argv.each do |task|
+        ::Rake::Task[task].invoke
+      end
+    end
+
+    def command_rspec(argv)
+      pry_setup
+      require 'rspec/core'
+      ::RSpec::Core::Runner.run(argv, $stderr, $stdout)
+    end
+
+    def pry_setup
+      ::Pry.config.input = stdin
+      ::Pry.config.output = stdout
+    end
+
     def stdin=(value)
       @stdin = InputProxy.new(value)
       $stdin = @stdin

metadata

@@ -2,7 +2,7 @@
 name: theine
 version: !ruby/object:Gem::Version
   prerelease:
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Jan Berdajs