Contents

require "fileutils"

module Lita
  # Converts Lita to a daemon process.
  class Daemon
    # @param pid_path [String] The path to the PID file.
    # @param log_path [String] The path to the log file.
    # @param kill_existing [Boolean] Whether or not to kill existing processes.
    def initialize(pid_path, log_path, kill_existing)
      @pid_path = pid_path
      @log_path = log_path
      @kill_existing = kill_existing
    end

    # Converts Lita to a daemon process.
    # @return [void]
    def daemonize
      handle_existing_process
      Process.daemon(true)
      File.open(@pid_path, "w") { |f| f.write(Process.pid) }
      set_up_logs
      at_exit { FileUtils.rm(@pid_path) if File.exist?(@pid_path) }
    end

    private

    # Abort if Lita is already running.
    def ensure_not_running
      abort I18n.t("lita.daemon.pid_exists", path: @pid_path) if File.exist?(@pid_path)
    end

    # Call the appropriate method depending on kill mode.
    def handle_existing_process
      if @kill_existing && File.exist?(@pid_path)
        kill_existing_process
      else
        ensure_not_running
      end
    end

    # Try to kill an existing process.
    def kill_existing_process
      pid = File.read(@pid_path).to_s.strip.to_i
      Process.kill("TERM", pid)
    rescue Errno::ESRCH, RangeError, Errno::EPERM
      abort I18n.t("lita.daemon.kill_failure", pid: pid)
    end

    # Redirect the standard streams to a log file.
    def set_up_logs
      log_file = File.new(@log_path, "a")
      stdout.reopen(log_file)
      stderr.reopen(log_file)
      stderr.sync = stdout.sync = true
    end

    def stdout
      $stdout
    end

    def stderr
      $stderr
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
lita-3.3.1	lib/lita/daemon.rb
lita-3.3.0	lib/lita/daemon.rb
lita-3.2.0	lib/lita/daemon.rb
lita-3.1.0	lib/lita/daemon.rb
lita-3.0.4	lib/lita/daemon.rb