forever 0.1.1 → 0.1.2

data/README.md

@@ -1,9 +1,13 @@
-# Forever
+# Ruby Forever
 
 Small daemon framework **for ruby**, with logging, error handler watcher and much more.
 
 The idea of the name and some concepts was taken by [forever for node.js](https://raw.github.com/indexzero/forever) written by Charlie Robbins.
 
+Please don't consider this gem as a port because is not.
+
+The executable name it's called ```foreverb``` to prevent collisions.
+
 ## Why?
 
 There are a lot of alternatives, one of the best is [resque](https://github.com/defunkt/resque), so why another daemons framework?
@@ -100,7 +104,7 @@ $ bin/foo stop
 List daemons:
 
 ``` sh
-$ forever list
+$ foreverb list
 PID     RSS     CPU   CMD
 19838   32512   1.6   Forever: bin/githubwatcher
 ```
@@ -108,7 +112,7 @@ PID     RSS     CPU   CMD
 Stop daemon(s):
 
 ``` sh
-$ forever stop foo
+$ foreverb stop foo
 Do you want really stop Forever: bin/foo  with pid 19538? y
 Killing process Forever: bin/foo  with pid 19538...
 ```

data/bin/forever

@@ -1,34 +1,2 @@
 #!/usr/bin/ruby
-require 'rubygems' unless defined?(Gem)
-require File.expand_path('../../lib/forever/version.rb', __FILE__)
-require 'thor'
-
-class CLI < Thor
-  desc "list", "List Forever daemons"
-  def list
-    say "PID\tRSS\tCPU\tCMD", :green
-    puts daemons.join("\n")
-  end
-
-  desc "stop [DAEMON]", "Stop a specified daemon"
-  def stop(daemon)
-    found = daemons.find_all { |d| d=~/#{daemon}/i }
-    say "Daemon(s) matching '#{daemon}' not found", :red if found.empty?
-    found.each do |daemon|
-      daemon = daemon.split("\t")
-      if yes? "Do you want really stop #{daemon[-1]} with pid #{daemon[0]}?"
-        say "Killing process #{daemon[-1]} with pid #{daemon[0]}...", :green
-        result = `kill #{daemon[0]}`.strip
-        say result, :red if result != ""
-      end
-    end
-  end
-
-  private
-    def daemons
-      `ps axo pid,rss,pcpu,command | grep -vE "^USER|grep" | grep Forever: | awk '{print $1"\t"$2"\t"$3"\t"$4" "$5" "$6}'`.chomp.split("\n")
-    end
-end
-
-ARGV << "-h" if ARGV.empty?
-CLI.start(ARGV)
+warn "Use foreverb, this one is deprecated"

data/bin/foreverb

@@ -0,0 +1,34 @@
+#!/usr/bin/ruby
+require 'rubygems' unless defined?(Gem)
+require File.expand_path('../../lib/forever/version.rb', __FILE__)
+require 'thor'
+
+class CLI < Thor
+  desc "list", "List Forever daemons"
+  def list
+    say "PID\tRSS\tCPU\tCMD", :green
+    puts daemons.join("\n")
+  end
+
+  desc "stop [DAEMON]", "Stop a specified daemon"
+  def stop(daemon)
+    found = daemons.find_all { |d| d=~/#{daemon}/i }
+    say "Daemon(s) matching '#{daemon}' not found", :red if found.empty?
+    found.each do |daemon|
+      daemon = daemon.split("\t")
+      if yes? "Do you want really stop #{daemon[-1]} with pid #{daemon[0]}?"
+        say "Killing process #{daemon[-1]} with pid #{daemon[0]}...", :green
+        result = `kill #{daemon[0]}`.strip
+        say result, :red if result != ""
+      end
+    end
+  end
+
+  private
+    def daemons
+      `ps axo pid,rss,pcpu,command | grep -vE "^USER|grep" | grep Forever: | awk '{print $1"\t"$2"\t"$3"\t"$4" "$5" "$6}'`.chomp.split("\n")
+    end
+end
+
+ARGV << "-h" if ARGV.empty?
+CLI.start(ARGV)

data/lib/forever/base.rb

@@ -5,12 +5,137 @@ module Forever
     def initialize(options={}, &block)
       options.each { |k,v| send(k, v) }
 
+      instance_eval(&block)
+
       Dir.chdir(dir) if exists?(dir)
       Dir.mkdir(File.dirname(log)) if log && !File.exist?(File.dirname(log))
       Dir.mkdir(File.dirname(pid)) if pid && !File.exist?(File.dirname(pid))
 
+      stop!
+
+      return if ARGV[0] == "stop" || on_ready.nil?
+
+      fork do
+        $0 = "Forever: #{$0}"
+        puts "=> Process demonized with pid #{Process.pid}"
+
+        %w(INT TERM KILL).each { |signal| trap(signal)  { stop! } }
+
+        File.open(pid, "w") { |f| f.write(Process.pid.to_s) } if pid
+
+        stream      = log ? File.new(log, "w") : File.open('/dev/null', 'w')
+        stream.sync = true
+
+        STDOUT.reopen(stream)
+        STDERR.reopen(STDOUT)
+
+        begin
+          on_ready.call
+        rescue Exception => e
+          Thread.list.reject { |t| t==Thread.current }.map(&:kill)
+          on_error[e] if on_error
+          stream.print "\n\n%s\n  %s\n\n" % [e.message, e.backtrace.join("\n  ")]
+          sleep 30
+          retry
+        end
+      end
+    end
+
+    ##
+    # Caller file
+    #
+    def file(value=nil)
+      value ? @_file = value : @_file
+    end
+
+    ##
+    # Base working Directory
+    #
+    def dir(value=nil)
+      value ? @_dir = value : @_dir
+    end
+
+    ##
+    # File were we redirect STOUT and STDERR, can be false.
+    #
+    # Default: dir + 'log/[process_name].log'
+    #
+    def log(value=nil)
+      @_log ||= File.join(dir, "log/#{File.basename(file)}.log") if exists?(dir, file)
+      value ? @_log = value : @_log
+    end
+
+    ##
+    # File were we store pid
+    #
+    # Default: dir + 'tmp/[process_name].pid'
+    #
+    def pid(value=nil)
+      @_pid ||= File.join(dir, "tmp/#{File.basename(file)}.pid") if exists?(dir, file)
+      value ? @_pid = value : @_pid
+    end
+
+    ##
+    # Search if there is a running process and stop it
+    #
+    def stop!(kill=true)
+      if exists?(pid)
+        _pid = File.read(pid).to_i
+        puts "=> Found pid #{_pid}..."
+        FileUtils.rm_f(pid)
+        begin
+          puts "=> Killing process #{_pid}..."
+          Process.kill(:KILL, _pid)
+        rescue Errno::ESRCH => e
+          puts "=> #{e.message}"
+        end
+      else
+        puts "=> Pid not found, process seems don't exist!"
+      end
+    end
+
+    ##
+    # Callback raised when an error occour
+    #
+    def on_error(&block)
+      block_given? ? @_on_error = block : @_on_error
+    end
+
+    ##
+    # Callback to fire when the daemon start
+    #
+    def on_ready(&block)
+      block_given? ? @_on_error = block : @_on_error
+    end
+
+    def to_s
+      "#<Forever dir:#{dir}, file:#{file}, log:#{log}, pid:#{pid}>"
+    end
+    alias :inspect :to_s
+
+    def config
+      { :dir => dir, :file => file, :log => log, :pid => pid }.to_yaml
+    end
+    alias :to_yaml :config
+
+    private
+      def exists?(*values)
+        values.all? { |value| value && File.exist?(value) }
+      end
+  end # Base
+end # Foreverrequire 'fileutils'
+
+module Forever
+  class Base
+    def initialize(options={}, &block)
+      options.each { |k,v| send(k, v) }
+
       instance_eval(&block)
 
+      Dir.chdir(dir) if exists?(dir)
+      Dir.mkdir(File.dirname(log)) if log && !File.exist?(File.dirname(log))
+      Dir.mkdir(File.dirname(pid)) if pid && !File.exist?(File.dirname(pid))
+
       stop!
 
       return if ARGV[0] == "stop" || on_ready.nil?

data/lib/forever/version.rb

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

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 1
-  version: 0.1.1
+  - 2
+  version: 0.1.2
 platform: ruby
 authors: 
 - DAddYE
@@ -36,6 +36,7 @@ email:
 - d.dagostino@lipsiasoft.com
 executables: 
 - forever
+- foreverb
 extensions: []
 
 extra_rdoc_files: []
@@ -46,6 +47,7 @@ files:
 - README.md
 - Rakefile
 - bin/forever
+- bin/foreverb
 - forever.gemspec
 - lib/forever.rb
 - lib/forever/base.rb