DaemonKit::Application

# File lib/daemon_kit/application.rb, line 11
def exec( file )
  raise DaemonNotFound.new( file ) unless File.exist?( file )

  DaemonKit.configuration.daemon_name ||= File.basename( file )

  command, configs, args = Arguments.parse( ARGV )

  case command
  when :run
    parse_arguments( args )
    run( file )
  when :start
    parse_arguments( args )
    start( file )
  when :stop
    stop
  end
end

# File lib/daemon_kit/application.rb, line 101
def exit!( code = 0 )
end

# File lib/daemon_kit/application.rb, line 31
def run( file )
  self.chroot
  self.clean_fd
  self.redirect_io( true )

  DaemonKit.configuration.log_stdout = true

  require file
end

# File lib/daemon_kit/application.rb, line 92
def running!
  Initializer.continue!

  yield DaemonKit.configuration if block_given?
end

# File lib/daemon_kit/application.rb, line 42
def start( file )
  self.drop_privileges
  self.daemonize
  self.chroot
  self.clean_fd
  self.redirect_io

  require file
end

# File lib/daemon_kit/application.rb, line 52
def stop
  @pid_file = PidFile.new( DaemonKit.configuration.pid_file )

  unless @pid_file.running?
    @pid_file.cleanup
    puts "Nothing to stop"
    exit
  end

  target_pid = @pid_file.pid

  puts "Sending TERM to #{target_pid}"
  Process.kill( 'TERM', target_pid )

  if seconds = DaemonKit.configuration.force_kill_wait
    begin
      Timeout::timeout( seconds ) do
        loop do
          puts "Waiting #{seconds} seconds for #{target_pid} before sending KILL"

          break unless @pid_file.running?

          seconds -= 1
          sleep 1
        end
      end
    rescue Timeout::Error
      Process.kill( 'KILL', target_pid )
    end
  end

  if @pid_file.running?
    puts "Process still running, leaving pidfile behind! Consider using configuration.force_kill_wait."
  else
    @pid_file.cleanup
  end
end

# File lib/daemon_kit/application.rb, line 133
def chroot
  Dir.chdir '/'
  File.umask 0000
end

# File lib/daemon_kit/application.rb, line 140
def clean_fd
  ObjectSpace.each_object(IO) do |io|
    unless [STDIN, STDOUT, STDERR].include?(io)
      begin
        unless io.closed?
          io.close
        end
      rescue ::Exception
      end
    end
  end
end

# File lib/daemon_kit/application.rb, line 113
def daemonize
  @pid_file = PidFile.new( DaemonKit.configuration.pid_file )
  @pid_file.ensure_stopped!

  if RUBY_VERSION < "1.9"
    exit if fork
    Process.setsid
    exit if fork
  else
    Process.daemon( true, true )
  end

  @pid_file.write!

  # TODO: Convert into shutdown hook
  at_exit { @pid_file.cleanup }
end

# File lib/daemon_kit/application.rb, line 167
def drop_privileges
  if DaemonKit.configuration.group
    begin
      group = Etc.getgrnam( DaemonKit.configuration.group )
      Process::Sys.setgid( group.gid.to_i )
    rescue => e
      $stderr.puts "Caught exception while trying to drop group privileges: #{e.message}"
    end
  end
  if DaemonKit.configuration.user
    begin
      user = Etc.getpwnam( DaemonKit.configuration.user )
      Process::Sys.setuid( user.uid.to_i )
    rescue => e
      $stderr.puts "Caught exception while trying to drop user privileges: #{e.message}"
    end
  end
end

# File lib/daemon_kit/application.rb, line 106
def parse_arguments( args )
  Arguments.parser_available = true
  DaemonKit.arguments = Arguments.new
  DaemonKit.arguments.parse( args )
end

# File lib/daemon_kit/application.rb, line 155
def redirect_io( simulate = false )
  begin
    STDIN.reopen '/dev/null'
  rescue ::Exception
  end

  unless simulate
    STDOUT.reopen '/dev/null', 'a'
    STDERR.reopen '/dev/null', 'a'
  end
end