lib/eye/patch/overrides.rb in eye-patch-0.0.8 vs lib/eye/patch/overrides.rb in eye-patch-0.0.9

@@ -6,34 +6,64 @@
     File.exists?(filename) ? filename : nil
   end
 end
 
 Eye::System.class_eval do
-  private
+  class << self
+    alias_method :daemonize_without_hook, :daemonize
+    alias_method :exec_without_hook, :exec
 
-  def spawn_options(config = {})
-    o = {pgroup: true, chdir: config[:working_dir] || '/'}
-    o.update(out: [config[:stdout], 'a']) if config[:stdout]
-    o.update(err: [config[:stderr], 'a']) if config[:stderr]
-    o.update(in: config[:stdin]) if config[:stdin]
-    o.update(close_others: !config[:preserve_fds])
+    def daemonize(*args)
+      Eye::Control.invoke_spawn_callback
+      daemonize_without_hook(*args)
+    end
 
-    if Eye::Local.root?
-      o.update(uid: Etc.getpwnam(config[:uid]).uid) if config[:uid]
-      o.update(gid: Etc.getpwnam(config[:gid]).gid) if config[:gid]
+    def exec(*args)
+      Eye::Control.invoke_spawn_callback
+      exec_without_hook(*args)
     end
 
-    o.update(umask: config[:umask]) if config[:umask]
+    private
 
-    o
+    def spawn_options(config = {})
+      o = {pgroup: true, chdir: config[:working_dir] || '/'}
+      o.update(out: [config[:stdout], 'a']) if config[:stdout]
+      o.update(err: [config[:stderr], 'a']) if config[:stderr]
+      o.update(in: config[:stdin]) if config[:stdin]
+      o.update(close_others: !config[:preserve_fds])
+
+      if Eye::Local.root?
+        o.update(uid: Etc.getpwnam(config[:uid]).uid) if config[:uid]
+        o.update(gid: Etc.getpwnam(config[:gid]).gid) if config[:gid]
+      end
+
+      o.update(umask: config[:umask]) if config[:umask]
+
+      o
+    end
   end
 end
 
 Eye::Controller.class_eval do
+
+  def invoke_spawn_callback
+    if respond_to?(:before_spawn)
+      debug "Invoking before_spawn hook"
+      before_spawn
+    end
+  end
+
   private
 
   def parse_config(filename)
-    Eye::Patch.parse(filename)
+    config = Eye::Patch.parse(filename)
+
+    if Eye::Patch.setup_file
+      info "Loading setup from: #{Eye::Patch.setup_file}"
+      require Eye::Patch.setup_file
+    end
+
+    config
   end
 end
 
 Eye::Control