undertaker 0.1.0

data/.document

@@ -0,0 +1,6 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE
+examples/*

data/.gitignore

@@ -0,0 +1,22 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC
+config.yml

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2010 Waine Kerr
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,42 @@
+undertaker
+===========
+
+periodic websites status checker storing in redis
+
+## Install
+
+  (sudo) gem install undertaker
+
+## Configuration
+
+It's easy- look at the [example]("http://github.com/wainekerr/undertaker/blob/master/examples/config.yml") :)
+
+## Run it
+
+Currently, the only way to do it is manually (which is useless) or in a crontab. A 5-minutes crontab is a good deal:
+
+  `*/5 * * * * undertaker /path/to/config.yml`
+
+## What it does
+
+It makes an HTTP request on the configured websites, then store the result as a JSON Array in redis e.g :
+
+  `undertaker:hostname => [http_code, http_message]`
+
+If the requests returns an exception, the `http_code` will be -999 and `http_message` the exception.
+
+After, it's up to you do to what you want to do with that data, not the job of undertaker :-)
+
+## Command-line options
+
+You can add theses options after the `/path/to/config.yml` when launching `undertaker` :
+
+* _verbose_ enable verbosity
+* _report_ display an ASCII formatted results of the last test
+* _jsonreport_ returns the results of the last test on JSON format
+* _reset_ reset the content
+* _unlock_ unlock undertaker (to prevent multiple instances running at the same time)
+
+## Copyright
+
+Copyright (c) 2010 Waine Kerr. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,55 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "undertaker"
+    gem.summary = %Q{Periodic HTTP status checker, stored in Redis}
+    gem.description = %Q{Small utility who checks HTTP status of websites, and store the results in Redis.}
+    gem.email = "wk@heldscalla.org"
+    gem.homepage = "http://wainekerr.github.com/undertaker/"
+    gem.authors = ["Waine Kerr"]
+    gem.add_dependency "redis-namespace", ">= 0.10.0"
+    gem.add_dependency "json", ">= 1.4.6"
+    gem.add_dependency "terminal-table", ">= 1.4.2"
+    gem.add_development_dependency "thoughtbot-shoulda", ">= 0"
+    gem.add_development_dependency "yard", ">= 0"
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new
+rescue LoadError
+  task :yardoc do
+    abort "YARD is not available. In order to run yardoc, you must: sudo gem install yard"
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/undertaker

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+#
+# Runner for Undertaker
+
+(puts "Usage: undertaker config.yml [report|json_report|verbose|unlock|reset]" ; exit 1) if ARGV[0].nil?
+
+require File.dirname(__FILE__) + '/../lib/undertaker'
+
+begin
+  verbose = ARGV[1]=='verbose' ? true : false
+  @undertaker = Undertaker.new(ARGV[0], verbose)
+  if ARGV[1]=="unlock"
+    @undertaker.unlock
+    exit 0
+  elsif ARGV[1]=="report"
+    @undertaker.report
+  elsif ARGV[1]=="json_report"
+    @undertaker.json_report
+  elsif ARGV[1]=="reset"
+    @undertaker.reset
+  else
+    puts "[INFO    ] Undertaker starting- #{@undertaker.config['websites'].size} websites to test!" if verbose
+    @undertaker.check!
+    @undertaker.report if verbose
+  end
+rescue
+  puts "[OH NOES ] #{$!}"
+  exit 1
+end  

data/lib/undertaker.rb

@@ -0,0 +1,88 @@
+%w{redis/namespace net/http uri yaml json terminal-table/import}.each {|r| require r }
+
+class Undertaker
+  attr_reader :config
+  
+  def initialize(config_file, verbose=false)
+    raise "Configuration file not found" unless File.exists?(config_file)
+    @verbose = verbose
+    @config = YAML.load_file(config_file)
+    @redis = Redis::Namespace.new("#{@config['redis']['namespace']}", :redis => Redis.new(@config['redis']))
+  end
+  
+  def lock
+    @redis.set "locked", "true"
+  end
+  
+  def unlock
+    @redis.set "locked", nil
+  end
+  
+  def check!
+    raise "Looks like another instance is running. Exiting." if @redis.get("locked")=="true"
+    lock
+    @threads = []
+    @config['websites'].each do |website|
+      @threads << Thread.new { check(website) }
+    end
+    @threads.each { |th| th.join }
+    unlock
+  end
+  
+  def check(website)
+    begin
+      url = URI.parse(website)
+      puts "[TEST    ] Starting request on #{website}" if @verbose
+      req = Net::HTTP::Get.new(url.path)
+      res = Net::HTTP.start(url.host, url.port) {|http|
+        http.request(req)
+      }
+      @redis.set url.host, [res.code.to_i, res.message, Time.now.strftime('%s').to_i].to_json
+      @redis.expire url.host, @config['redis']['expires']
+      puts "[TESTED  ] #{url} => #{res.code}, #{res.message}" if @verbose
+    rescue
+      @redis.set url.host, [0, $!, Time.now.strftime('%s').to_i].to_json
+      @redis.expire url.host, @config['redis']['expires']
+      puts "[FAILED  ] #{url} => #{$!}" if @verbose
+    end
+  end
+  
+  def report
+    @website_table = table do |t|
+      t.headings = 'Website', 'Code', 'Message'
+      @config['websites'].each do |website|
+        url = URI.parse(website)
+        result = @redis.get url.host
+        if result
+          result = JSON.parse(result)
+          t << [website, result[0], result[1]]
+        else
+          t << [website, "N/A", "No Info"]
+        end
+      end
+    end
+    puts @website_table
+  end
+  
+  def json_report
+    @array = []
+    @config['websites'].each do |website|
+      url = URI.parse(website)
+      result = @redis.get url.host
+      if result
+        result = JSON.parse(result)
+        @array << [website, result[0], result[1]]
+      else
+        @array << [website, "N/A", "No Info"]
+      end
+    end
+    puts @array.to_json
+  end
+  
+  def reset
+    @config['websites'].each do |website|
+      url = URI.parse(website)
+      result = @redis.del url.host
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'undertaker'
+
+class Test::Unit::TestCase
+end

data/test/test_undertaker.rb

@@ -0,0 +1,7 @@
+require 'helper'
+
+class TestUndertaker < Test::Unit::TestCase
+  should "probably rename this file and start testing for real" do
+    flunk "hey buddy, you should probably rename this file and start testing for real"
+  end
+end

metadata

@@ -0,0 +1,138 @@
+--- !ruby/object:Gem::Specification 
+name: undertaker
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Waine Kerr
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-08-28 00:00:00 +02:00
+default_executable: undertaker
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: redis-namespace
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 10
+        - 0
+        version: 0.10.0
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: json
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 4
+        - 6
+        version: 1.4.6
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: terminal-table
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 4
+        - 2
+        version: 1.4.2
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: thoughtbot-shoulda
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: yard
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id005
+description: Small utility who checks HTTP status of websites, and store the results in Redis.
+email: wk@heldscalla.org
+executables: 
+- undertaker
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.md
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.md
+- Rakefile
+- VERSION
+- bin/undertaker
+- lib/undertaker.rb
+- test/helper.rb
+- test/test_undertaker.rb
+has_rdoc: true
+homepage: http://wainekerr.github.com/undertaker/
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Periodic HTTP status checker, stored in Redis
+test_files: 
+- test/helper.rb
+- test/test_undertaker.rb