cloud66-bluepill 0.0.62

data/lib/bluepill/dsl.rb

@@ -0,0 +1,12 @@
+# -*- encoding: utf-8 -*-
+module Bluepill
+  def self.application(app_name, options = {}, &block)
+    app_proxy = AppProxy.new(app_name, options)
+    if block.arity == 0
+      app_proxy.instance_eval &block
+    else
+      app_proxy.instance_exec(app_proxy, &block)
+    end
+    app_proxy.app.load
+  end
+end

data/lib/bluepill/dsl/app_proxy.rb

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+module Bluepill
+  class AppProxy
+    APP_ATTRIBUTES = [:working_dir, :uid, :gid, :environment, :auto_start ]
+
+    attr_accessor *APP_ATTRIBUTES
+    attr_reader :app
+
+    def initialize(app_name, options)
+      @app = Application.new(app_name.to_s, options)
+    end
+
+    def process(process_name, &process_block)
+      attributes = {}
+      APP_ATTRIBUTES.each { |a| attributes[a] = self.send(a) }
+
+      process_factory = ProcessFactory.new(attributes, process_block)
+
+      process = process_factory.create_process(process_name, @app.pids_dir)
+      group = process_factory.attributes.delete(:group)
+
+      @app.add_process(process, group)
+    end
+  end
+end

data/lib/bluepill/dsl/process_factory.rb

@@ -0,0 +1,122 @@
+# -*- encoding: utf-8 -*-
+module Bluepill
+  class ProcessFactory
+    attr_reader :attributes
+
+    @@process_keys = Hash.new
+    @@pid_files = Hash.new
+
+    def initialize(attributes, process_block)
+      @attributes = attributes
+      @process_block = process_block
+    end
+
+    def create_process(name, pids_dir)
+      self.assign_default_pid_file(name, pids_dir)
+
+      process = ProcessProxy.new(name, @attributes, @process_block)
+      child_process_block = @attributes.delete(:child_process_block)
+      @attributes[:child_process_factory] = ProcessFactory.new(@attributes, child_process_block) if @attributes[:monitor_children]
+
+      self.validate_process! process
+      process.to_process
+    end
+
+    def create_child_process(name, pid, logger)
+      attributes = {}
+      [:start_grace_time, :stop_grace_time, :restart_grace_time].each {|a| attributes[a] = @attributes[a]}
+      attributes[:actual_pid] = pid
+      attributes[:logger] = logger
+
+      child = ProcessProxy.new(name, attributes, @process_block)
+      self.validate_child_process! child
+      process = child.to_process
+
+      process.determine_initial_state
+      process
+    end
+
+    protected
+
+    def assign_default_pid_file(process_name, pids_dir)
+      unless @attributes.key?(:pid_file)
+        group_name = @attributes[:group]
+        default_pid_name = [group_name, process_name].compact.join('_').gsub(/[^A-Za-z0-9_\-]/, "_")
+        @attributes[:pid_file] = File.join(pids_dir, default_pid_name + ".pid")
+      end
+    end
+
+    def validate_process!(process)
+      # validate uniqueness of group:process
+      process_key = [process.attributes[:group], process.name].join(":")
+      if @@process_keys.key?(process_key)
+        $stderr.print "Config Error: You have two entries for the process name '#{process.name}'"
+        $stderr.print " in the group '#{process.attributes[:group]}'" if process.attributes.key?(:group)
+        $stderr.puts
+        exit(6)
+      else
+        @@process_keys[process_key] = 0
+      end
+
+      # validate required attributes
+      [:start_command].each do |required_attr|
+        if !process.attributes.key?(required_attr)
+          $stderr.puts "Config Error: You must specify a #{required_attr} for '#{process.name}'"
+          exit(6)
+        end
+      end
+
+      # validate uniqueness of pid files
+      pid_key = process.attributes[:pid_file].strip
+      if @@pid_files.key?(pid_key)
+        $stderr.puts "Config Error: You have two entries with the pid file: #{pid_key}"
+        exit(6)
+      else
+        @@pid_files[pid_key] = 0
+      end
+
+      #validate stop_signals array
+      stop_grace_time = process.attributes[:stop_grace_time]
+      stop_signals = process.attributes[:stop_signals]
+
+      unless stop_signals.nil?
+        #Start with the more helpful error messages before the 'odd number' message.
+        delay_sum = 0
+        stop_signals.each_with_index do |s_or_d, i|
+          if i % 2 == 0
+            signal = s_or_d
+            unless signal.is_a? Symbol
+              $stderr.puts "Config Error: Invalid stop_signals!  Expected a symbol (signal) at position #{i} instead of '#{signal}'."
+              exit(6)
+            end
+          else
+            delay = s_or_d
+            unless delay.is_a? Fixnum
+              $stderr.puts "Config Error: Invalid stop_signals!  Expected a number (delay) at position #{i} instead of '#{delay}'."
+              exit(6)
+            end
+            delay_sum += delay
+          end
+        end
+
+        unless stop_signals.size % 2 == 1
+          $stderr.puts "Config Error: Invalid stop_signals!  Expected an odd number of elements."
+          exit(6)
+        end
+
+        if stop_grace_time.nil? || stop_grace_time <= delay_sum
+          $stderr.puts "Config Error: Stop_grace_time should be greater than the sum of stop_signals delays!"
+          exit(6)
+        end
+      end
+    end
+
+    def validate_child_process!(child)
+      unless child.attributes.has_key?(:stop_command)
+        $stderr.puts "Config Error: Invalid child process monitor for #{child.name}"
+        $stderr.puts "You must specify a stop command to monitor child processes."
+        exit(6)
+      end
+    end
+  end
+end

data/lib/bluepill/dsl/process_proxy.rb

@@ -0,0 +1,44 @@
+# -*- encoding: utf-8 -*-
+module Bluepill
+  class ProcessProxy
+    attr_reader :attributes, :watches, :name
+    def initialize(process_name, attributes, process_block)
+      @name = process_name
+      @attributes = attributes
+      @watches = {}
+
+      if process_block.arity == 0
+        instance_eval &process_block
+      else
+        instance_exec(self, &process_block)
+      end
+    end
+
+    def method_missing(name, *args)
+      if args.size == 1 && name.to_s =~ /^(.*)=$/
+        @attributes[$1.to_sym] = args.first
+      elsif args.size == 1
+        @attributes[name.to_sym] = args.first
+      elsif args.size == 0 && name.to_s =~ /^(.*)!$/
+        @attributes[$1.to_sym] = true
+      elsif args.empty? && @attributes.key?(name.to_sym)
+        @attributes[name.to_sym]
+      else
+        super
+      end
+    end
+
+    def checks(name, options = {})
+      @watches[name] = options
+    end
+
+    def monitor_children(&child_process_block)
+      @attributes[:monitor_children] = true
+      @attributes[:child_process_block] = child_process_block
+    end
+
+    def to_process
+      Process.new(@name, @watches, @attributes)
+    end
+  end
+end

data/lib/bluepill/group.rb

@@ -0,0 +1,72 @@
+# -*- encoding: utf-8 -*-
+module Bluepill
+  class Group
+    attr_accessor :name, :processes, :logger
+    attr_accessor :process_logger
+
+    def initialize(name, options = {})
+      self.name = name
+      self.processes = []
+      self.logger = options[:logger]
+    end
+
+    def add_process(process)
+      process.logger = self.logger.prefix_with(process.name)
+      self.processes << process
+    end
+
+    def tick
+      self.processes.each do |process|
+        process.tick
+      end
+    end
+
+    def determine_initial_state
+      self.processes.each do |process|
+        process.determine_initial_state
+      end
+    end
+
+    # proxied events
+    [:start, :unmonitor, :stop, :restart].each do |event|
+      class_eval <<-END
+        def #{event}(process_name = nil)
+          threads = []
+          affected = []
+          self.processes.each do |process|
+            next if process_name && process_name != process.name
+            affected << [self.name, process.name].join(":")
+            threads << Thread.new { process.handle_user_command("#{event}") }
+          end
+          threads.each { |t| t.join }
+          affected
+        end
+      END
+    end
+
+    def status(process_name = nil)
+      lines = []
+      if process_name.nil?
+        prefix = self.name ? "  " : ""
+        lines << "#{self.name}:" if self.name
+
+        self.processes.each do |process|
+          lines << "%s%s(pid:%s): %s" % [prefix, process.name, process.actual_pid, process.state]
+          if process.monitor_children?
+            process.children.each do |child|
+              lines << "  %s%s: %s" % [prefix, child.name, child.state]
+            end
+          end
+        end
+      else
+        self.processes.each do |process|
+          next if process_name != process.name
+          lines << "%s%s(pid:%s): %s" % [prefix, process.name, process.actual_pid, process.state]
+          lines << process.statistics.to_s
+        end
+      end
+      lines << ""
+    end
+
+  end
+end

data/lib/bluepill/logger.rb

@@ -0,0 +1,63 @@
+# -*- encoding: utf-8 -*-
+module Bluepill
+  class Logger
+    LOG_METHODS = [:emerg, :alert, :crit, :err, :warning, :notice, :info, :debug]
+
+    def initialize(options = {})
+      @options  = options
+      @logger   = options[:logger] || self.create_logger
+      @prefix   = options[:prefix]
+      @stdout   = options[:stdout]
+      @prefixes = {}
+    end
+
+    LOG_METHODS.each do |method|
+      eval <<-END
+        def #{method}(msg, prefix = [])
+          if @logger.is_a?(self.class)
+            @logger.#{method}(msg, [@prefix] + prefix)
+          else
+            s_prefix = prefix.size > 0 ? "[\#{prefix.compact.join(':')}] " : ""
+            if @stdout
+              $stdout.puts("[#{method}]: \#{s_prefix}\#{msg}")
+              $stdout.flush
+            end
+            @logger.#{method}("\#{s_prefix}\#{msg}")
+          end
+        end
+      END
+    end
+
+    def prefix_with(prefix)
+      @prefixes[prefix] ||= self.class.new(:logger => self, :prefix => prefix)
+    end
+
+    def reopen
+      if @logger.is_a?(self.class)
+        @logger.reopen
+      else
+        @logger = create_logger
+      end
+    end
+
+    protected
+    def create_logger
+      if @options[:log_file]
+        LoggerAdapter.new(@options[:log_file])
+      else
+        Syslog.close if Syslog.opened? # need to explictly close it before reopening it
+        Syslog.open(@options[:identity] || 'bluepilld', Syslog::LOG_PID, Syslog::LOG_LOCAL6)
+      end
+    end
+
+    class LoggerAdapter < ::Logger
+      LOGGER_EQUIVALENTS =
+        {:debug => :debug, :err => :error, :warning => :warn, :info => :info, :emerg => :fatal, :alert => :warn, :crit => :fatal, :notice => :info}
+
+      LOG_METHODS.each do |method|
+        next if method == LOGGER_EQUIVALENTS[method]
+        alias_method method, LOGGER_EQUIVALENTS[method]
+      end
+    end
+  end
+end

data/lib/bluepill/process.rb

@@ -0,0 +1,514 @@
+# -*- encoding: utf-8 -*-
+
+# fixes problem with loading on systems with rubyist-aasm installed
+gem "state_machine"
+
+require "state_machine"
+require "daemons"
+require "bluepill/system"
+require "bluepill/process_journal"
+
+module Bluepill
+  class Process
+    CONFIGURABLE_ATTRIBUTES = [
+      :pre_start_command,
+      :start_command,
+      :stop_command,
+      :restart_command,
+
+      :stdout,
+      :stderr,
+      :stdin,
+
+      :daemonize,
+      :pid_file,
+      :working_dir,
+      :environment,
+
+      :start_grace_time,
+      :stop_grace_time,
+      :restart_grace_time,
+
+      :uid,
+      :gid,
+
+      :cache_actual_pid,
+
+      :monitor_children,
+      :child_process_factory,
+
+      :pid_command,
+      :auto_start,
+
+      :supplementary_groups,
+
+      :stop_signals,
+
+      :on_start_timeout,
+    ]
+
+    attr_accessor :name, :watches, :triggers, :logger, :skip_ticks_until, :process_running
+    attr_accessor *CONFIGURABLE_ATTRIBUTES
+    attr_reader :children, :statistics
+
+    state_machine :initial => :unmonitored do
+      # These are the idle states, i.e. only an event (either external or internal) will trigger a transition.
+      # The distinction between down and unmonitored is that down
+      # means we know it is not running and unmonitored is that we don't care if it's running.
+      state :unmonitored, :up, :down
+
+      # These are transitionary states, we expect the process to change state after a certain period of time.
+      state :starting, :stopping, :restarting
+
+      event :tick do
+        transition :starting => :up, :if => :process_running?
+        transition :starting => :down, :unless => :process_running?
+
+        transition :up => :up, :if => :process_running?
+        transition :up => :down, :unless => :process_running?
+
+        # The process failed to die after entering the stopping state. Change the state to reflect
+        # reality.
+        transition :stopping => :up, :if => :process_running?
+        transition :stopping => :down, :unless => :process_running?
+
+        transition :down => :up, :if => :process_running?
+        transition :down => :starting, :unless => :process_running?
+
+        transition :restarting => :up, :if => :process_running?
+        transition :restarting => :down, :unless => :process_running?
+      end
+
+      event :start do
+        transition [:unmonitored, :down] => :starting
+      end
+
+      event :stop do
+        transition :up => :stopping
+      end
+
+      event :unmonitor do
+        transition any => :unmonitored
+      end
+
+      event :restart do
+        transition [:up, :down] => :restarting
+      end
+
+      before_transition any => any, :do => :notify_triggers
+      before_transition :stopping => any, :do => :clean_threads
+
+      after_transition any => :starting, :do => :start_process
+      after_transition any => :stopping, :do => :stop_process
+      after_transition any => :restarting, :do => :restart_process
+
+      after_transition any => any, :do => :record_transition
+    end
+
+    def initialize(process_name, checks, options = {})
+      @name = process_name
+      @event_mutex = Monitor.new
+      @watches = []
+      @triggers = []
+      @children = []
+      @threads = []
+      @statistics = ProcessStatistics.new
+      @actual_pid = options[:actual_pid]
+      self.logger = options[:logger]
+
+      checks.each do |name, opts|
+        if Trigger[name]
+          self.add_trigger(name, opts)
+        else
+          self.add_watch(name, opts)
+        end
+      end
+
+      # These defaults are overriden below if it's configured to be something else.
+      @monitor_children =  false
+      @cache_actual_pid = true
+      @start_grace_time = @stop_grace_time = @restart_grace_time = 3
+      @environment = {}
+      @on_start_timeout = "start"
+
+      CONFIGURABLE_ATTRIBUTES.each do |attribute_name|
+        self.send("#{attribute_name}=", options[attribute_name]) if options.has_key?(attribute_name)
+      end
+
+      # Let state_machine do its initialization stuff
+      super() # no arguments intentional
+    end
+
+    def tick
+      return if self.skipping_ticks?
+      self.skip_ticks_until = nil
+
+      # clear the memoization per tick
+      @process_running = nil
+
+      # Deal with thread cleanup here since the stopping state isn't used
+      clean_threads if self.unmonitored?
+
+      # run state machine transitions
+      super
+
+      if self.up?
+        self.run_watches
+
+        if self.monitor_children?
+          refresh_children!
+          children.each {|child| child.tick}
+        end
+      end
+    end
+
+    def logger=(logger)
+      @logger = logger
+      self.watches.each {|w| w.logger = logger }
+      self.triggers.each {|t| t.logger = logger }
+    end
+
+    # State machine methods
+    def dispatch!(event, reason = nil)
+      @event_mutex.synchronize do
+        @statistics.record_event(event, reason)
+        self.send("#{event}")
+      end
+    end
+
+    def record_transition(transition)
+      unless transition.loopback?
+        @transitioned = true
+
+        # When a process changes state, we should clear the memory of all the watches
+        self.watches.each { |w| w.clear_history! }
+
+        # Also, when a process changes state, we should re-populate its child list
+        if self.monitor_children?
+          self.logger.warning "Clearing child list"
+          self.children.clear
+        end
+        logger.info "Going from #{transition.from_name} => #{transition.to_name}"
+      end
+    end
+
+    def notify_triggers(transition)
+      self.triggers.each {|trigger| trigger.notify(transition)}
+    end
+
+    # Watch related methods
+    def add_watch(name, options = {})
+      self.watches << ConditionWatch.new(name, options.merge(:logger => self.logger))
+    end
+
+    def add_trigger(name, options = {})
+      self.triggers << Trigger[name].new(self, options.merge(:logger => self.logger))
+    end
+
+    def run_watches
+      now = Time.now.to_i
+
+      threads = self.watches.collect do |watch|
+        [watch, Thread.new { Thread.current[:events] = watch.run(self.actual_pid, now) }]
+      end
+
+      @transitioned = false
+
+      threads.inject([]) do |events, (watch, thread)|
+        thread.join
+        if thread[:events].size > 0
+          logger.info "#{watch.name} dispatched: #{thread[:events].join(',')}"
+          thread[:events].each do |event|
+            events << [event, watch.to_s]
+          end
+        end
+        events
+      end.each do |(event, reason)|
+        break if @transitioned
+        self.dispatch!(event, reason)
+      end
+    end
+
+    def determine_initial_state
+      if self.process_running?(true)
+        self.state = 'up'
+      else
+        self.state = (auto_start == false) ? 'unmonitored' : 'down' # we need to check for false value
+      end
+    end
+
+    def handle_user_command(cmd)
+      case cmd
+      when "start"
+        if self.process_running?(true)
+          logger.warning("Refusing to re-run start command on an already running process.")
+        else
+          dispatch!(:start, "user initiated")
+        end
+      when "stop"
+        stop_process
+        dispatch!(:unmonitor, "user initiated")
+      when "restart"
+        restart_process
+      when "unmonitor"
+        # When the user issues an unmonitor cmd, reset any triggers so that
+        # scheduled events gets cleared
+        triggers.each {|t| t.reset! }
+        dispatch!(:unmonitor, "user initiated")
+      end
+    end
+
+    # System Process Methods
+    def process_running?(force = false)
+      @process_running = nil if force # clear existing state if forced
+
+      @process_running ||= signal_process(0)
+      # the process isn't running, so we should clear the PID
+      self.clear_pid unless @process_running
+      @process_running
+    end
+
+    def start_process
+      ProcessJournal.kill_all_from_journal(name) # be sure nothing else is running from previous runs
+      pre_start_process
+      logger.warning "Executing start command: #{start_command}"
+      if self.daemonize?
+        daemon_id = System.daemonize(start_command, self.system_command_options)
+        if daemon_id > 0
+          ProcessJournal.append_pid_to_journal(name, daemon_id)
+          children.each {|child|
+            ProcessJournal.append_pid_to_journal(name, child.actual_id)
+          } if self.monitor_children?
+        end
+        daemon_id
+      else
+        # This is a self-daemonizing process
+        with_timeout(start_grace_time, on_start_timeout) do
+          result = System.execute_blocking(start_command, self.system_command_options)
+
+          unless result[:exit_code].zero?
+            logger.warning "Start command execution returned non-zero exit code:"
+            logger.warning result.inspect
+          end
+        end
+      end
+
+      self.skip_ticks_for(start_grace_time)
+    end
+
+    def pre_start_process
+      return unless pre_start_command
+      logger.warning "Executing pre start command: #{pre_start_command}"
+      result = System.execute_blocking(pre_start_command, self.system_command_options)
+      unless result[:exit_code].zero?
+        logger.warning "Pre start command execution returned non-zero exit code:"
+        logger.warning result.inspect
+      end
+    end
+
+    def stop_process
+      if monitor_children
+        System.get_children(self.actual_pid).each do |child_pid|
+          ProcessJournal.append_pid_to_journal(name, child_pid)
+        end
+      end
+
+      if stop_command
+        cmd = self.prepare_command(stop_command)
+        logger.warning "Executing stop command: #{cmd}"
+
+        with_timeout(stop_grace_time, "stop") do
+          result = System.execute_blocking(cmd, self.system_command_options)
+
+          unless result[:exit_code].zero?
+            logger.warning "Stop command execution returned non-zero exit code:"
+            logger.warning result.inspect
+          end
+        end
+
+      elsif stop_signals
+        # issue stop signals with configurable delay between each
+        logger.warning "Sending stop signals to #{actual_pid}"
+        @threads << Thread.new(self, stop_signals.clone) do |process, stop_signals|
+          signal = stop_signals.shift
+          logger.info "Sending signal #{signal} to #{process.actual_pid}"
+          process.signal_process(signal) # send first signal
+
+          until stop_signals.empty?
+            # we already checked to make sure stop_signals had an odd number of items
+            delay = stop_signals.shift
+            signal = stop_signals.shift
+
+            logger.debug "Sleeping for #{delay} seconds"
+            sleep delay
+            #break unless signal_process(0) #break unless the process can be reached
+            unless process.signal_process(0)
+              logger.debug "Process has terminated."
+              break
+            end
+            logger.info "Sending signal #{signal} to #{process.actual_pid}"
+            process.signal_process(signal)
+          end
+        end
+      else
+        logger.warning "Executing default stop command. Sending TERM signal to #{actual_pid}"
+        signal_process("TERM")
+      end
+      ProcessJournal.kill_all_from_journal(name) # finish cleanup
+      self.unlink_pid # TODO: we only write the pid file if we daemonize, should we only unlink it if we daemonize?
+
+      self.skip_ticks_for(stop_grace_time)
+    end
+
+    def restart_process
+      if restart_command
+        cmd = self.prepare_command(restart_command)
+
+        logger.warning "Executing restart command: #{cmd}"
+
+        with_timeout(restart_grace_time, "restart") do
+          result = System.execute_blocking(cmd, self.system_command_options)
+
+          unless result[:exit_code].zero?
+            logger.warning "Restart command execution returned non-zero exit code:"
+            logger.warning result.inspect
+          end
+        end
+
+        self.skip_ticks_for(restart_grace_time)
+      else
+        logger.warning "No restart_command specified. Must stop and start to restart"
+        self.stop_process
+        # the tick will bring it back.
+      end
+    end
+
+    def clean_threads
+      @threads.each { |t| t.kill }
+      @threads.clear
+    end
+
+    def daemonize?
+      !!self.daemonize
+    end
+
+    def monitor_children?
+      !!self.monitor_children
+    end
+
+    def signal_process(code)
+      code = code.to_s.upcase if code.is_a?(String) || code.is_a?(Symbol)
+      ::Process.kill(code, actual_pid)
+      true
+    rescue Exception => e
+      logger.err "Failed to signal process #{actual_pid} with code #{code}: #{e}"
+      false
+    end
+
+    def cache_actual_pid?
+      !!@cache_actual_pid
+    end
+
+    def actual_pid
+      pid_command ? pid_from_command : pid_from_file
+    end
+
+    def pid_from_file
+      return @actual_pid if cache_actual_pid? && @actual_pid
+      @actual_pid = begin
+        if pid_file
+          if File.exists?(pid_file)
+            str = File.read(pid_file)
+            str.to_i if str.size > 0
+          else
+            logger.warning("pid_file #{pid_file} does not exist or cannot be read")
+            nil
+          end
+        end
+      end
+    end
+
+    def pid_from_command
+      pid = %x{#{pid_command}}.strip
+      (pid =~ /\A\d+\z/) ? pid.to_i : nil
+    end
+
+    def actual_pid=(pid)
+      ProcessJournal.append_pid_to_journal(name, pid) # be sure to always log the pid
+      @actual_pid = pid
+    end
+
+    def clear_pid
+      @actual_pid = nil
+    end
+
+    def unlink_pid
+      System.delete_if_exists(pid_file)
+    end
+
+     # Internal State Methods
+    def skip_ticks_for(seconds)
+      # TODO: should this be addative or longest wins?
+      #       i.e. if two calls for skip_ticks_for come in for 5 and 10, should it skip for 10 or 15?
+      self.skip_ticks_until = (self.skip_ticks_until || Time.now.to_i) + seconds.to_i
+    end
+
+    def skipping_ticks?
+      self.skip_ticks_until && self.skip_ticks_until > Time.now.to_i
+    end
+
+    def refresh_children!
+      # First prune the list of dead children
+      @children.delete_if {|child| !child.process_running?(true) }
+
+      # Add new found children to the list
+      new_children_pids = System.get_children(self.actual_pid) - @children.map {|child| child.actual_pid}
+
+      unless new_children_pids.empty?
+        logger.info "Existing children: #{@children.collect{|c| c.actual_pid}.join(",")}. Got new children: #{new_children_pids.inspect} for #{actual_pid}"
+      end
+
+      # Construct a new process wrapper for each new found children
+      new_children_pids.each do |child_pid|
+        ProcessJournal.append_pid_to_journal(name, child_pid)
+        name = "<child(pid:#{child_pid})>"
+        logger = self.logger.prefix_with(name)
+
+        child = self.child_process_factory.create_child_process(name, child_pid, logger)
+        @children << child
+      end
+    end
+
+    def prepare_command(command)
+      command.to_s.gsub("{{PID}}", actual_pid.to_s)
+    end
+
+    def system_command_options
+      {
+        :uid         => self.uid,
+        :gid         => self.gid,
+        :working_dir => self.working_dir,
+        :environment => self.environment,
+        :pid_file    => self.pid_file,
+        :logger      => self.logger,
+        :stdin       => self.stdin,
+        :stdout      => self.stdout,
+        :stderr      => self.stderr,
+        :supplementary_groups => self.supplementary_groups
+      }
+    end
+
+    def with_timeout(secs, next_state = nil, &blk)
+      # Attempt to execute the passed block. If the block takes
+      # too long, transition to the indicated next state.
+      begin
+        Timeout.timeout(secs.to_f, &blk)
+      rescue Timeout::Error
+        logger.err "Execution is taking longer than expected."
+        logger.err "Did you forget to tell bluepill to daemonize this process?"
+        dispatch!(next_state)
+      end
+    end
+  end
+end
+