daemon-kit 0.1.3

data/lib/daemon_kit/patches/force_kill_wait.rb

@@ -0,0 +1,120 @@
+# Shamelessly taken from http://blog.rapleaf.com/dev/?p=19
+
+require 'rubygems'
+require 'daemons'
+require 'timeout'
+
+module Daemons
+
+  class ApplicationGroup
+
+    # We want to redefine find_applications to not rely on
+    # pidfiles (e.g. find application if pidfile is gone)
+    # We recreate the pid files if they're not there.
+    def find_applications(dir)
+      # Find pid_files, like original implementation
+      pid_files = PidFile.find_files(dir, app_name)
+      @monitor = Monitor.find(dir, app_name + '_monitor')
+      pid_files.reject! {|f| f =~ /_monitor.pid$/}
+      
+      # Find the missing pids based on the UNIX pids
+      pidfile_pids = pid_files.map {|pf| PidFile.existing(pf).pid}
+      missing_pids = unix_pids - pidfile_pids
+
+      # Create pidfiles that are gone
+      if missing_pids.size > 0
+        puts "[daemons_ext]: #{missing_pids.size} missing pidfiles: " + 
+             "#{missing_pids.inspect}... creating pid file(s)."
+        missing_pids.each do |pid|
+          pidfile = PidFile.new(dir, app_name, multiple)
+          pidfile.pid = pid # Doesn't seem to matter if it's a string or Fixnum
+        end
+      end
+
+      # Now get all the pid file again
+      pid_files = PidFile.find_files(dir, app_name)
+
+      return pid_files.map {|f|
+        app = Application.new(self, {}, PidFile.existing(f))
+        setup_app(app)
+        app
+      }
+    end
+
+    # Specify :force_kill_wait => (seconds to wait) and this method will
+    # block until the process is dead.  It first sends a TERM signal, then
+    # a KILL signal (-9) if the process hasn't died after the wait time.
+    def stop_all(force = false)
+      @monitor.stop if @monitor
+      
+      wait = options[:force_kill_wait].to_i
+      if wait > 0
+        puts "[daemons_ext]: Killing #{app_name} with force after #{wait} secs."
+
+        # Send term first, don't delete PID files.
+        @applications.each {|a| a.send_sig('TERM')}
+
+        begin
+          started_at = Time.now
+          Timeout::timeout(wait) do
+            num_pids = unix_pids.size
+            while num_pids > 0
+              time_left = wait - (Time.now - started_at)
+              puts "[daemons_ext]: Waiting #{time_left.round} secs on " +
+                   "#{num_pids} #{app_name}(s)..."
+              sleep 1
+              num_pids = unix_pids.size
+            end 
+          end
+        rescue Timeout::Error
+          @applications.each {|a| a.send_sig('KILL')}
+        ensure
+          # Delete Pidfiles
+          @applications.each {|a| a.zap!}
+        end
+
+        puts "[daemons_ext]: All #{app_name}(s) dead."
+      else
+        @applications.each {|a| 
+          if force
+            begin; a.stop; rescue ::Exception; end
+          else
+            a.stop
+          end
+        }
+      end
+    end
+    
+    private
+
+    # Find UNIX pids based on app_name.  CAUTION: This has only been tested on
+    # Mac OS X and CentOS.
+    def unix_pids
+      pids = []
+      x = `ps auxw | grep -v grep | awk '{print $2, $11}' | grep #{app_name}`
+      if x && x.chomp!
+        processes = x.split(/\n/).compact
+        processes = processes.delete_if do |p|
+          pid, name = p.split(/\s/)
+          # We want to make sure that the first part of the process name matches
+          # so that app_name matches app_name_22
+          app_name != name[0..(app_name.length - 1)]
+        end
+        pids = processes.map {|p| p.split(/\s/)[0].to_i}
+      end
+
+      pids
+    end
+
+  end
+
+  class Application
+
+    # Send signal to the process, rescue if process deson't exist
+    def send_sig(sig)
+      Process.kill(sig, @pid.pid) rescue Errno::ESRCH
+    end
+
+  end
+
+end

data/lib/daemon_kit/tasks.rb

@@ -0,0 +1,2 @@
+# Load all the take tasks in the gem
+Dir[File.join(File.dirname(__FILE__), '**/*.rake')].each { |rake| load rake }

data/lib/daemon_kit/tasks/framework.rake

@@ -0,0 +1,75 @@
+namespace :daemon_kit do
+  namespace :freeze do
+    desc "Lock this application to the current gem (by unpacking it into vendor/daemon_kit)"
+    task :gems do
+      deps = %w()
+      require 'rubygems'
+      require 'rubygems/gem_runner'
+
+      kit = (version = ENV['VERSION']) ?
+        Gem.cache.find_name('daemon-kit', "= #{version}").first :
+        Gem.cache.find_name('daemon-kit').sort_by { |g| g.version }.last
+
+      version ||= kit.version
+
+      unless kit
+        puts "No daemon_kit gem #{version} is installed.  Do 'gem list daemon_kit' to see what you have available."
+        exit
+      end
+
+      puts "Freezing the gem for DaemonKit #{kit.version}"
+      rm_rf   "vendor/daemon_kit"
+      mkdir_p "vendor/daemon_kit"
+
+      begin
+        chdir("vendor/daemon_kit") do
+          kit.dependencies.select { |g| deps.include? g.name }.each do |g|
+            Gem::GemRunner.new.run(["unpack", g.name, "--version", g.version_requirements.to_s])
+            mv(Dir.glob("#{g.name}*").first, g.name)
+          end
+
+          Gem::GemRunner.new.run(["unpack", "daemon-kit", "--version", "=#{version}"])
+          FileUtils.mv(Dir.glob("daemon-kit*").first, "daemon-kit")
+        end
+      rescue Exception
+        rm_rf "vendor/daemon_kit"
+        raise
+      end
+    end
+
+    desc 'Lock to latest edge daemon_kit'
+    task :edge do
+      require 'open-uri'
+      #version = ENV["RELEASE"] || "edge"
+      commits = "http://github.com/api/v1/yaml/kennethkalmer/daemon-kit/commits/master"
+      url     = "http://github.com/kennethkalmer/daemon-kit/zipball/master"
+
+      rm_rf   "vendor/daemon_kit"
+      mkdir_p "vendor/daemon_kit"
+      
+      chdir 'vendor/daemon_kit' do
+        latest_revision = YAML.load(open(commits))["commits"].first["id"]
+        
+        puts "Downloading DaemonKit from #{url}"
+        File.open('daemon-kit.zip', 'wb') do |dst|
+          open url do |src|
+            while chunk = src.read(4096)
+              dst << chunk
+            end
+          end
+        end
+
+        puts 'Unpacking DaemonKit'
+        rm_rf 'daemon-kit'
+        `unzip daemon-kit.zip`
+        FileUtils.mv(Dir.glob("kennethkalmer-daemon-kit*").first, "daemon-kit")
+        %w(daemon-kit.zip).each do |goner|
+          rm_f goner
+        end
+
+        touch "REVISION_#{latest_revision}"
+      end
+    end
+    
+  end
+end

data/rubygems_generators/install_rspec/USAGE

@@ -0,0 +1,5 @@
+Description:
+  
+  
+Usage:
+  

data/rubygems_generators/install_rspec/install_rspec_generator.rb

@@ -0,0 +1,57 @@
+
+class InstallRspecGenerator < RubiGen::Base
+  
+  default_options :author => nil
+  
+  attr_reader :gem_name, :module_name
+  
+  def initialize(runtime_args, runtime_options = {})
+    super
+    @destination_root = File.expand_path(destination_root)
+    @gem_name = base_name
+    @module_name  = @gem_name.camelcase
+    extract_options
+  end
+
+  def manifest
+    record do |m|
+      # Ensure appropriate folder(s) exists
+      m.directory 'spec'
+      m.directory 'tasks'
+
+      m.template           'spec.rb', "spec/#{gem_name}_spec.rb"
+      
+      m.template_copy_each %w( spec.opts spec_helper.rb ), 'spec'
+      m.file_copy_each     %w( rspec.rake ), 'tasks'
+    end
+  end
+
+  protected
+    def banner
+      <<-EOS
+Install rspec BDD testing support. 
+
+Includes a rake task (tasks/rspec.rake) to be loaded by the root Rakefile,
+which provides a "spec" task.
+
+USAGE: #{$0} [options]
+EOS
+    end
+
+    def add_options!(opts)
+      # opts.separator ''
+      # opts.separator 'Options:'
+      # For each option below, place the default
+      # at the top of the file next to "default_options"
+      # opts.on("-a", "--author=\"Your Name\"", String,
+      #         "Some comment about this option",
+      #         "Default: none") { |x| options[:author] = x }
+    end
+    
+    def extract_options
+      # for each option, extract it into a local variable (and create an "attr_reader :author" at the top)
+      # Templates can access these value via the attr_reader-generated methods, but not the
+      # raw instance variable value.
+      # @author = options[:author]
+    end
+end

data/rubygems_generators/install_rspec/templates/spec.rb

@@ -0,0 +1,11 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+# Time to add your specs!
+# http://rspec.info/
+describe "Place your specs here" do
+  
+  it "find this spec in spec directory" do
+    violated "Be sure to write your specs"
+  end
+  
+end

data/rubygems_generators/install_rspec/templates/spec/spec.opts

@@ -0,0 +1 @@
+--colour

data/rubygems_generators/install_rspec/templates/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+begin
+  require 'spec'
+rescue LoadError
+  require 'rubygems'
+  gem 'rspec'
+  require 'spec'
+end
+
+require File.dirname(__FILE__) + '/../config/boot'
+require '<%= gem_name %>'

data/rubygems_generators/install_rspec/templates/tasks/rspec.rake

@@ -0,0 +1,21 @@
+begin
+  require 'spec'
+rescue LoadError
+  require 'rubygems'
+  require 'spec'
+end
+begin
+  require 'spec/rake/spectask'
+rescue LoadError
+  puts <<-EOS
+To use rspec for testing you must install rspec gem:
+    gem install rspec
+EOS
+  exit(0)
+end
+
+desc "Run the specs under spec/models"
+Spec::Rake::SpecTask.new do |t|
+  t.spec_opts = ['--options', "spec/spec.opts"]
+  t.spec_files = FileList['spec/**/*_spec.rb']
+end

data/script/console

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+# File: script/console
+irb = RUBY_PLATFORM =~ /(:?mswin|mingw)/ ? 'irb.bat' : 'irb'
+
+libs =  " -r irb/completion"
+# Perhaps use a console_lib to store any extra methods I may want available in the cosole
+# libs << " -r #{File.dirname(__FILE__) + '/../lib/console_lib/console_logger.rb'}"
+libs <<  " -r #{File.dirname(__FILE__) + '/../lib/daemon-kit.rb'}"
+puts "Loading daemon-kit gem"
+exec "#{irb} #{libs} --simple-prompt"

data/script/destroy

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/destroy'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Destroy.new.run(ARGV)

data/script/generate

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/generate'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Generate.new.run(ARGV)

data/script/txt2html

@@ -0,0 +1,71 @@
+#!/usr/bin/env ruby
+
+load File.dirname(__FILE__) + "/../Rakefile"
+require 'rubyforge'
+require 'redcloth'
+require 'syntax/convertors/html'
+require 'erb'
+
+download = "http://rubyforge.org/projects/#{$hoe.rubyforge_name}"
+version  = $hoe.version
+
+def rubyforge_project_id
+  RubyForge.new.configure.autoconfig["group_ids"][$hoe.rubyforge_name]
+end
+
+class Fixnum
+  def ordinal
+    # teens
+    return 'th' if (10..19).include?(self % 100)
+    # others
+    case self % 10
+    when 1: return 'st'
+    when 2: return 'nd'
+    when 3: return 'rd'
+    else    return 'th'
+    end
+  end
+end
+
+class Time
+  def pretty
+    return "#{mday}#{mday.ordinal} #{strftime('%B')} #{year}"
+  end
+end
+
+def convert_syntax(syntax, source)
+  return Syntax::Convertors::HTML.for_syntax(syntax).convert(source).gsub(%r!^<pre>|</pre>$!,'')
+end
+
+if ARGV.length >= 1
+  src, template = ARGV
+  template ||= File.join(File.dirname(__FILE__), '/../website/template.html.erb')
+else
+  puts("Usage: #{File.split($0).last} source.txt [template.html.erb] > output.html")
+  exit!
+end
+
+template = ERB.new(File.open(template).read)
+
+title = nil
+body = nil
+File.open(src) do |fsrc|
+  title_text = fsrc.readline
+  body_text_template = fsrc.read
+  body_text = ERB.new(body_text_template).result(binding)
+  syntax_items = []
+  body_text.gsub!(%r!<(pre|code)[^>]*?syntax=['"]([^'"]+)[^>]*>(.*?)</\1>!m){
+    ident = syntax_items.length
+    element, syntax, source = $1, $2, $3
+    syntax_items << "<#{element} class='syntax'>#{convert_syntax(syntax, source)}</#{element}>"
+    "syntax-temp-#{ident}"
+  }
+  title = RedCloth.new(title_text).to_html.gsub(%r!<.*?>!,'').strip
+  body = RedCloth.new(body_text).to_html
+  body.gsub!(%r!(?:<pre><code>)?syntax-temp-(\d+)(?:</code></pre>)?!){ syntax_items[$1.to_i] }
+end
+stat = File.stat(src)
+created = stat.ctime
+modified = stat.mtime
+
+$stdout << template.result(binding)

data/spec/daemon_kit_spec.rb

@@ -0,0 +1,7 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+describe DaemonKit do
+  
+  it "should work"
+  
+end

data/spec/initializer_spec.rb

@@ -0,0 +1,31 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe DaemonKit::Configuration do
+  before(:each) do
+    @configuration = DaemonKit::Configuration.new
+  end
+  
+  it "should know our environment" do
+    @configuration.environment.should_not be_nil
+  end
+
+  it "should have a default log path" do
+    @configuration.log_path.should_not be_nil
+  end
+
+  it "should have a default log level" do
+    @configuration.log_level.should_not be_nil
+  end
+    
+end
+
+
+describe DaemonKit::Initializer do
+
+  it "should setup a logger" do
+    pending
+    DaemonKit::Initializer.run(:initialize_logger)
+    DaemonKit.logger.should_not be_nil
+  end
+  
+end