Add missing requires

jstorimer · Jun 19, 2012 · b1c775f · b1c775f
1 parent 0519b9e
commit b1c775f
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/bin/spin b/bin/spin
@@ -1,17 +1,25 @@
 #!/usr/bin/env ruby
 
+require 'spin/watcher'
+require 'spin/preloader'
 require 'spin/socket'
 
 # requires
 # parse options (preload an extra path(s), add path(s) to watch)
-at_exit { File.delete(Spin::Socket.filepath) }
+
+trap("SIGINT") { exit }
+
 # signal handlers
 #
 # spawn file watcher w/ pipe
 reader = Spin::Watcher.spawn
 
+at_exit { File.delete(Spin::Socket.filepath) }
+push_socket = Spin::Socket.open
+
 # preload rails
 Spin::Preloader.require_files
+
 # IO.select pipe + unix socket
   # add files to be processed to the next test run obj
     # include the recved io if applicable

diff --git a/lib/spin/socket.rb b/lib/spin/socket.rb
@@ -2,6 +2,7 @@
 
 require 'tempfile' # for Dir::tmpdir
 require 'digest/md5'
+require 'socket'
 
 module Spin
   class Socket
@@ -11,6 +12,10 @@ def self.filepath
       slug = Digest::MD5.hexdigest ['spin', Dir.pwd].join('-')
       [Dir::tmpdir, slug].join('/')
     end
+
+    def self.open
+      UNIXServer.open(filepath)
+    end
   end
 end
 
diff --git a/lib/spin/watcher.rb b/lib/spin/watcher.rb
@@ -1,3 +1,5 @@
+require 'listen'
+
 module Spin
   module Watcher
     extend self
@@ -11,7 +13,8 @@ def spawn
       fork {
         rd.close
 
-        Listen.to(AppDirectories + TestDirectories, :filter => /\.rb$/, :latency => 0.1) do |modified, added, removed|
+        dirs = AppDirectories + TestDirectories
+        Listen.to(*dirs, :filter => /\.rb$/, :latency => 0.1) do |modified, added, removed|
           changed_files = [modified + added + removed].uniq
 
           files_to_queue = changed_files.map { |file|