ducksboard_reporter 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: faaa03c65806c52554a5e2afe37b04914c1dccb6
-  data.tar.gz: 7820258a49917f566545b8298760c2fc4461d448
+  metadata.gz: 0f166259ade7200cbecbe4b6c47afebab9ec0168
+  data.tar.gz: b4c7a1700503fb55d363461d70069f35f39fafc4
 SHA512:
-  metadata.gz: 90b1b70ee0b2b8fc0b838b20a95b682ae024c2dd6a2b9f48cc43a944672a9dec9baee1fa842e8678fc1d07a125ac6d52461abb1c58a652f362ccc8fb10984075
-  data.tar.gz: b1779d9639ce483333c7ee84429074c5e6e36033b5ed7c5fa7ee2c9455a7c5f80c7c53c7f53846ae5e5d85c5ef696cfc3138f57018e6f3422b1f72af7d0680b4
+  metadata.gz: e9c0f3bacc413c914382430327ed53dc1e23eacb800aa69c0c363c6fc667b70261d77e1a8393775b6bd8d9ac210581155ac285a9776fc0a52669aab5432154ac
+  data.tar.gz: 7a83efa936333b143b766f3df60043a858ba3623a66a0517779362ab74c349e562a5ab6cf8b03fe9ae91ac317c1cb1b5e69b1d3f3c1582085947886a1a37c429

data/bin/ducksboard-reporter

@@ -4,10 +4,30 @@ $:.unshift File.expand_path("../../lib", __FILE__)
 require "ducksboard_reporter"
 require "trollop"
 require "yaml"
+require "rund"
 
 opts = Trollop::options do
   opt :config_file, "Path to config file", short: "f", type: :string, required: true
+  opt :pid_file, "Path to pid file", short: "p", type: :string
+  opt :log_file, "Path to log file", short: "l", type: :string
+  opt :run_dir,  "Path to run dir", type: :string
+  opt :user,  "The daemon wil run as specified user", type: :string
+  opt :group,  "The daemon wil run as specified group", type: :string
+  opt :daemonize, "Detach from terminal", short: "d", default: false
 end
 
+# Rund.debug = true
+
 config = YAML.load_file(opts[:config_file])
+
+if opts[:daemonize]
+  Rund.pid_file = opts[:pid_file]
+  Rund.log_file = opts[:log_file]
+  Rund.run_dir  = opts[:run_dir]
+  Rund.user     = opts[:user]
+  Rund.group    = opts[:group]
+  Rund.daemonize!
+end
+
 DucksboardReporter::App.new(config).start
+

data/lib/ducksboard_reporter/version.rb

@@ -1,3 +1,3 @@
 module DucksboardReporter
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/rund.rb

@@ -0,0 +1,192 @@
+require "fileutils"
+require "etc"
+
+module Rund
+  extend self
+
+  def daemonize!
+    # Keeep $stderr IO to display error and debug messages until
+    # daemonizing is finished
+    keep_local_stderr
+
+    File.umask(0000)
+    Process.daemon(true, true)
+    change_group if @group
+    change_user  if @user
+    write_pid    if @pid_file
+    change_dir
+    redirect_input
+    redirect_output
+
+    close_local_stderr
+  end
+
+  def log_file=(path)
+    @log_file = clean_path(path)
+  end
+
+  def pid_file=(path)
+    @pid_file = clean_path(path)
+  end
+
+  def run_dir=(path)
+    @run_dir = clean_path(path)
+  end
+
+  def input_file=(path)
+    @input_file = clean_path(path)
+  end
+
+  def user=(user)
+    @user = user
+  end
+
+  def group=(group)
+    @group = group
+  end
+
+  def debug=(flag)
+    @debug = !!flag
+  end
+
+  private
+
+  def change_user
+    error("Not running as root. Cannot change user of process.") if Process.uid != 0
+    begin
+      user = (@user.is_a? Integer) ? Etc.getpwuid(@user) : Etc.getpwnam(@user)
+    rescue ArgumentError
+      error("User #{@user} does not exist")
+    end
+
+    debug { "Changing user to #{user.name} UID=#{user.uid}" }
+
+    Process::UID.change_privilege(user.uid)
+  end
+
+  def change_group
+    error("Not running as root. Cannot change group of process.") if Process.uid != 0
+    begin
+      group = (@group.is_a? Integer) ? Etc.getgrgid(@group) : Etc.getgrnam(@group)
+    rescue ArgumentError
+      error("Group #{@group} does not exist")
+    end
+
+    debug { "Changing group to #{group.name} GID=#{group.gid}" }
+
+    begin
+      Process::GID.change_privilege(group.gid)
+    rescue Errno::EPERM
+      error("Cannot change to group #{group.name}. Permission denied.")
+    end
+  end
+
+  def keep_local_stderr
+    @stderr = $stderr.dup
+    @stderr.sync
+  end
+
+  def close_local_stderr
+    @stderr.close
+  end
+
+  def clean_path(path)
+    path = path.to_s.strip
+    path.empty? ? nil : File.absolute_path(path)
+  end
+
+  def change_dir
+    @run_dir ||= "/"
+    debug { "Changing to #{@run_dir} as run dir" }
+    Dir.chdir(@run_dir)
+  end
+
+  def redirect_output
+    @log_file ||= "/dev/null"
+    debug { "Logging to #{@log_file}" }
+
+    log_dir = File.dirname(@log_file)
+    error("Log file dir does not exist #{log_dir}") unless File.exists?(log_dir)
+
+    begin
+      unless File.exists?(@log_file)
+        FileUtils.touch(@log_file)
+        File.chmod(0644, @log_file)
+      end
+    rescue Errno::EACCES
+      error("Cannot write log file #{@log_file}. Permission denied.")
+    end
+
+    $stderr.reopen(@log_file, "a")
+    $stdout.reopen($stderr)
+    $stdout.sync = $stderr.sync = true
+  end
+
+  def redirect_input
+    @input_file ||= "/dev/null"
+    $stdin.reopen(@input_file, "r")
+  end
+
+  def write_pid
+    debug { "Writing pid to #{@pid_file}" }
+
+    begin
+      File.open(@pid_file, File::CREAT | File::EXCL | File::WRONLY) {|f| f.write("#{Process.pid}") }
+    rescue Errno::EACCES
+      error("Cannot write pid file #{@pid_file}. Permission denied.")
+    rescue Errno::EEXIST
+      check_pid
+      retry
+    end
+
+    at_exit do
+      begin
+        File.delete(@pid_file) if @pid_file && File.exists?(@pid_file)
+      rescue Errno::EACCES
+        debug { "Cannot delete pid file #{@pid_file}. Permission denied." }
+      end
+    end
+  end
+
+  def pid_status
+    return :exited unless File.exists?(@pid_file)
+    return :dead if pid == 0
+
+    Process.kill(0, pid) # check process status
+    :running
+  rescue Errno::ESRCH
+    :dead
+  rescue Errno::EPERM
+    :not_owned
+  end
+
+  def check_pid
+    case pid_status
+    when :running, :not_owned
+      error("A process is already running with pid #{pid}")
+    when :dead
+      debug { "Deleting stale pid file" }
+      File.delete(@pid_file)
+    end
+  end
+
+  def pid
+    @pid ||= File.read(@pid_file).strip.to_i
+  end
+
+  def error(msg)
+    @stderr << "ERROR: #{msg}\n"
+    exit 1
+  end
+
+  def debug(msg = nil)
+    return unless @debug
+    msg = yield if block_given?
+    @stderr << "DEBUG: #{msg}\n"
+  end
+
+  def error(msg)
+    @stderr << "ERROR: #{msg}\n"
+    exit 1
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ducksboard_reporter
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - unnu
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-30 00:00:00.000000000 Z
+date: 2014-10-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -174,6 +174,7 @@ files:
 - lib/ducksboard_reporter/reporters/random.rb
 - lib/ducksboard_reporter/version.rb
 - lib/ducksboard_reporter/widget.rb
+- lib/rund.rb
 - spec/ducksboard_reporter_spec.rb
 - spec/reporter_spec.rb
 - spec/spec_helper.rb