smoke 0.0.1

data/History.txt

@@ -0,0 +1,3 @@
+== 0.0.1 2008-03-29
+
+* Initial release

data/License.txt

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) 2008 Craig R Webster <craig@barkingiguana.com>
+
+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/Manifest.txt

@@ -0,0 +1,46 @@
+History.txt
+License.txt
+Manifest.txt
+README.txt
+Rakefile
+bin/smoke
+bin/smoked
+config/hoe.rb
+config/requirements.rb
+etc/smoke.d/processors/smoke/remote_port_status_check_processor.rb
+etc/smoke.d/processors/smoke/system_load_processor.rb
+etc/smoke.d/signals/smoke/random_status_demo.rb
+etc/smoke.d/signals/smoke/remote_port_status_check.rb
+etc/smoke.d/signals/smoke/system_load.rb
+lib/smoke.rb
+lib/smoke/client.rb
+lib/smoke/client/connection.rb
+lib/smoke/client/signal_runner.rb
+lib/smoke/client/version.rb
+lib/smoke/server.rb
+lib/smoke/server/signal_handler.rb
+lib/smoke/server/signal_router.rb
+lib/smoke/server/version.rb
+lib/smoke/signal.rb
+lib/smoke/signal_processor.rb
+lib/smoke/version.rb
+load_average.png
+log/debug.log
+log/development.log
+log/production.log
+log/server.log
+log/test.log
+script/destroy
+script/generate
+script/txt2html
+setup.rb
+tasks/deployment.rake
+tasks/environment.rake
+tasks/website.rake
+test/test_helper.rb
+test/test_server.rb
+website/index.html
+website/index.txt
+website/javascripts/rounded_corners_lite.inc.js
+website/stylesheets/screen.css
+website/template.rhtml

data/README.txt

@@ -0,0 +1,3 @@
+== Smoke Server
+
+Run bin/smoked to listen for connections from Smoke clients.

data/Rakefile

@@ -0,0 +1,4 @@
+require 'config/requirements'
+require 'config/hoe' # setup Hoe + all gem configuration
+
+Dir['tasks/**/*.rake'].each { |rake| load rake }

data/bin/smoke

@@ -0,0 +1,43 @@
+#!/usr/bin/env ruby
+
+GEM_ROOT = File.expand_path(File.dirname(File.dirname(__FILE__)))
+SIGNAL_DIRECTORIES = [ "#{GEM_ROOT}/etc/smoke.d/signals", "/etc/smoke.d/signals" ]
+
+$: << File.join(GEM_ROOT, 'lib')
+SIGNAL_DIRECTORIES.each_with_index do |directory, index|
+  if directory =~ /[^\/]$/
+    directory += '/'
+    SIGNAL_DIRECTORIES[index] = directory
+  end
+  $: << directory
+end
+
+require 'rubygems'
+require 'active_support'
+require 'smoke/signal'
+require 'smoke/client'
+
+signals = returning([]) do |signal_list|
+  SIGNAL_DIRECTORIES.each do |directory|
+    Dir[directory + '**/*.rb'].collect do |signal_file|
+      require signal_file
+      signal = signal_file.sub(directory, '').sub('.rb', '').classify
+      begin
+        puts "Loading #{signal}."
+        signal_class = signal.constantize
+        if signal_class.ancestors.include? Smoke::Signal
+          signal_list << signal_class
+        else
+          puts "Expected #{signal} (#{signal_file}) to extend Smoke::Signal."
+          exit 2
+        end
+      rescue NameError
+        puts "Expected #{signal_file} to define #{signal}."
+        exit 1
+      end
+    end
+  end
+end.flatten.compact.uniq
+
+client = Smoke::Client.new signals
+client.join

data/bin/smoked

@@ -0,0 +1,44 @@
+#!/usr/bin/env ruby
+#
+#  Created on 2008-3-29.
+#  Copyright (c) 2008 Craig R Webster <craig@barkingiguana.com>
+#
+
+$: << File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'rubygems'
+require 'mongrel'
+require 'lib/smoke/server'
+require 'optparse'
+
+OPTIONS = {
+  :port => 8935,
+  :ip_address => "127.0.0.1"
+}
+MANDATORY_OPTIONS = %w(  )
+
+parser = OptionParser.new do |opts|
+  opts.banner = %Q(
+  A framework for processing Smoke signals.
+  http://smoke.rubyforge.org/
+
+  Usage: #{File.basename($0)} [options]
+
+  Options are:)
+  opts.separator ""
+  opts.on("-b", "--bind=IP_ADDRESS", String,
+  "IP address to bind to",
+  "Default: 127.0.0.1 (any IP address)") { |OPTIONS[:ip_address]| }
+  opts.on("-p", "--port=PORT", Integer,
+  "Port to listen on",
+  "Default: 8935") { |OPTIONS[:port]| }
+  opts.on("-h", "--help",
+  "Show this help message.") { puts opts; exit }
+  opts.parse!(ARGV)
+
+  if MANDATORY_OPTIONS && MANDATORY_OPTIONS.find { |option| OPTIONS[option.to_sym].nil? }
+    puts opts; exit
+  end
+end
+
+server = Smoke::Server.new(OPTIONS)
+server.join

data/config/hoe.rb

@@ -0,0 +1,70 @@
+require 'smoke/version'
+
+AUTHOR = 'Craig R Webster'  # can also be an array of Authors
+EMAIL = "craig@barkingiguana.com"
+DESCRIPTION = "A simple framework for reporting to a central server. What is reported and how it is handled is passed off to plugins called Smoke Signals that are defined by the user."
+GEM_NAME = 'smoke' # what ppl will type to install your gem
+RUBYFORGE_PROJECT = 'smoke' # The unix name for your project
+HOMEPATH = "http://#{RUBYFORGE_PROJECT}.rubyforge.org"
+DOWNLOAD_PATH = "http://rubyforge.org/projects/#{RUBYFORGE_PROJECT}"
+
+@config_file = "~/.rubyforge/user-config.yml"
+@config = nil
+RUBYFORGE_USERNAME = "unknown"
+def rubyforge_username
+  unless @config
+    begin
+      @config = YAML.load(File.read(File.expand_path(@config_file)))
+    rescue
+      puts <<-EOS
+ERROR: No rubyforge config file found: #{@config_file}
+Run 'rubyforge setup' to prepare your env for access to Rubyforge
+ - See http://newgem.rubyforge.org/rubyforge.html for more details
+      EOS
+      exit
+    end
+  end
+  RUBYFORGE_USERNAME.replace @config["username"]
+end
+
+
+REV = nil 
+# UNCOMMENT IF REQUIRED: 
+# REV = `svn info`.each {|line| if line =~ /^Revision:/ then k,v = line.split(': '); break v.chomp; else next; end} rescue nil
+VERS = Smoke::VERSION::STRING + (REV ? ".#{REV}" : "")
+RDOC_OPTS = ['--quiet', '--title', 'smoke documentation',
+    "--opname", "index.html",
+    "--line-numbers", 
+    "--main", "README",
+    "--inline-source"]
+
+class Hoe
+  def extra_deps 
+    @extra_deps.reject! { |x| Array(x).first == 'hoe' } 
+    @extra_deps
+  end 
+end
+
+# Generate all the Rake tasks
+# Run 'rake -T' to see list of generated tasks (from gem root directory)
+hoe = Hoe.new(GEM_NAME, VERS) do |p|
+  p.developer(AUTHOR, EMAIL)
+  p.description = DESCRIPTION
+  p.summary = DESCRIPTION
+  p.url = HOMEPATH
+  p.rubyforge_name = RUBYFORGE_PROJECT if RUBYFORGE_PROJECT
+  p.test_globs = ["test/**/test_*.rb"]
+  p.clean_globs |= ['**/.*.sw?', '*.gem', '.config', '**/.DS_Store']  #An array of file patterns to delete on clean.
+  
+  # == Optional
+  p.changes = p.paragraphs_of("History.txt", 0..1).join("\n\n")
+  #p.extra_deps = []     # An array of rubygem dependencies [name, version], e.g. [ ['active_support', '>= 1.3.1'] ]
+  
+  #p.spec_extras = {}    # A hash of extra values to set in the gemspec.
+  
+end
+
+CHANGES = hoe.paragraphs_of('History.txt', 0..1).join("\\n\\n")
+PATH    = (RUBYFORGE_PROJECT == GEM_NAME) ? RUBYFORGE_PROJECT : "#{RUBYFORGE_PROJECT}/#{GEM_NAME}"
+hoe.remote_rdoc_dir = File.join(PATH.gsub(/^#{RUBYFORGE_PROJECT}\/?/,''), 'rdoc')
+hoe.rsync_args = '-av --delete --ignore-errors'

data/config/requirements.rb

@@ -0,0 +1,17 @@
+require 'fileutils'
+include FileUtils
+
+require 'rubygems'
+%w[rake hoe newgem rubigen].each do |req_gem|
+  begin
+    require req_gem
+  rescue LoadError
+    puts "This Rakefile requires the '#{req_gem}' RubyGem."
+    puts "Installation: gem install #{req_gem} -y"
+    exit
+  end
+end
+
+$:.unshift(File.join(File.dirname(__FILE__), %w[.. lib]))
+
+require 'smoke'

data/etc/smoke.d/processors/smoke/remote_port_status_check_processor.rb

@@ -0,0 +1,8 @@
+module Smoke
+  class RemotePortStatusCheckProcessor < Smoke::SignalProcessor
+    def process(signal)
+      puts signal.inspect
+      puts "-" * 80
+    end
+  end
+end

data/etc/smoke.d/processors/smoke/system_load_processor.rb

@@ -0,0 +1,41 @@
+require 'rubygems'
+require 'scruffy'
+
+module Smoke
+  class SystemLoadProcessor < Smoke::SignalProcessor
+    def initialize(*args)
+      super
+      @load = []
+      0.upto(59) do |n|
+        @load[n] = [0, 0, 0]
+      end
+
+      # Generate a load graph every 60 seconds.
+      Thread.new(@load) do |load|
+        loop do
+          begin
+            graph = Scruffy::Graph.new(:title => "Load Average")
+            graph.value_formatter = Scruffy::Formatters::Number.new(:precision => 2)
+            graph.add(:area, ' 1min', load.collect { |sample| sample[0] }, :opacity => 0.5)
+            graph.add(:area, ' 5min', load.collect { |sample| sample[1] }, :opacity => 0.5)
+            graph.add(:area, '15min', load.collect { |sample| sample[2] }, :opacity => 0.5)
+            graph.render(:min_value => 0, :max_value => 1.00, :width => 700, :as => 'PNG', :to => 'load_average.png')
+            sleep 60
+          rescue => e
+            STDERR.puts e.message + ":\n\n" + e.backtrace.join("\n") + "\n\n"
+          end
+        end
+      end
+    end
+
+    def process(signal)
+      @load.shift
+      load_averages = [
+        signal["report"]["one_minute"],
+        signal["report"]["five_minutes"],
+        signal["report"]["fifteen_minutes"]
+      ]
+      @load.push load_averages
+    end
+  end
+end

data/etc/smoke.d/signals/smoke/random_status_demo.rb

@@ -0,0 +1,7 @@
+module Smoke
+  class RandomStatusDemo < Smoke::Signal
+    def run
+      { :report => { :status => %W(OK FAIL UNKNOWN)[rand(3).round] } }
+    end
+  end
+end

data/etc/smoke.d/signals/smoke/remote_port_status_check.rb

@@ -0,0 +1,34 @@
+module Smoke
+  class RemotePortStatusCheck < Smoke::Signal
+    HOSTS_PORTS_AND_STATUSES = [
+      [ "xeriom.net", 80, :open ],
+      [ "barkingiguana.com", 80, :open ],
+      [ "localhost", 80, :down ]
+    ]
+
+    def initialize(client)
+      @client = client
+    end
+
+    def start
+      HOSTS_PORTS_AND_STATUSES.map do |check|
+        Thread.new(@client, check) do |client, c|
+          host, port, desired_status = *c
+          loop do
+            current_status = begin
+              TCPSocket.new(host, port)
+              :open
+            rescue Errno::ECONNREFUSED
+              :closed
+            end
+            if current_status != desired_status
+              client.report self, { :host => host, :port => port, :status => current_status }
+            end
+            sleep 15
+          end
+        end
+      end
+      loop
+    end
+  end
+end

data/etc/smoke.d/signals/smoke/system_load.rb

@@ -0,0 +1,14 @@
+module Smoke
+  class SystemLoad < Smoke::Signal
+    def run
+      one_minute, five_minutes, fifteen_minutes = %x[uptime].strip.split(/load averages: /).last.split(/ /)
+      {
+        :report => {
+          :one_minute => one_minute.to_f,
+          :five_minutes => five_minutes.to_f,
+          :fifteen_minutes => fifteen_minutes.to_f
+        }
+      }
+    end
+  end
+end

data/lib/smoke.rb

@@ -0,0 +1,5 @@
+$:.unshift File.dirname(__FILE__)
+
+module Smoke
+  
+end

data/lib/smoke/client.rb

@@ -0,0 +1,50 @@
+$:.unshift File.dirname(__FILE__)
+require 'smoke/client/signal_runner'
+
+module Smoke
+  module Client
+    def new(signals = [])
+      @reporter = SignalRunner.new
+
+      polling_signals, interupting_signals = returning([[], []]) do |poll, interupt|
+        signals.map do |signal|
+          if signal.instance_methods.include?("start")
+            interupt << signal
+          else
+            poll << signal
+          end
+        end
+      end
+
+      interupting_signals.map do |signal|
+        Thread.new(self) do |client|
+          signal.new(client).start
+        end
+      end
+
+      loop do
+        polling_signals.map do |signal|
+          Thread.new(@reporter) do |r|
+            begin
+              response = signal.run
+              report(signal, response)
+            rescue => e
+              puts e.message + ":" + e.backtrace.join("\n")
+            end
+          end
+        end
+
+        # FIXME: This should be configurable.
+        sleep 60
+      end
+    end  
+
+    def report(signal, response)
+      source = signal.is_a?(Class) ? signal.to_s : signal.class.to_s
+      response.merge!(:source => source, :time => Time.now)
+      @reporter.signals << response
+    end
+
+    extend self
+  end
+end

data/lib/smoke/client/connection.rb

@@ -0,0 +1,44 @@
+require 'net/http'
+
+module Smoke
+  module Client
+    class Connection
+      DEFAULT_OPTIONS = {
+        :hostname => "localhost",
+        :port     => 8935,
+        :ssl      => false,
+        :path     => '/signals',
+        :timeout  => 3
+      }
+
+      attr_reader :configuration
+
+      def initialize(configuration = {})
+        @configuration = DEFAULT_OPTIONS.merge(configuration)
+        @user_agent = if configuration[:ssl]
+          Net::HTTPS
+        else
+          Net::HTTP
+        end.new(configuration[:hostname], configuration[:port])
+        @user_agent.read_timeout = configuration[:timeout]
+      end
+
+      def send_signal(signal)
+        report_signal = Net::HTTP::Post.new(configuration[:path])
+        if configuration[:username] && configuration[:password]
+          report_signal.basic_auth configuration[:username], configuration[:password]
+        end
+        response = @user_agent.request report_signal, signal.to_xml(:root => :signal)
+        case response
+        when Net::HTTPCreated
+          true
+        else
+          response.error!
+        end
+      rescue => e
+        STDERR.puts "#{e.message}\n\n#{e.backtrace.join("\n")}\n\n"
+        raise
+      end
+    end
+  end
+end