flapjack 0.4.10

data/lib/flapjack/cli/worker_manager.rb

@@ -0,0 +1,47 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'ostruct'
+require 'optparse' 
+
+class WorkerManagerOptions
+  def self.parse(args)
+    options = OpenStruct.new
+    opts = OptionParser.new do |opts|
+      opts.banner = "Usage: flapjack-worker-manager <command> [options]"
+      opts.separator " "
+      opts.separator "  where <command> is one of:"
+      opts.separator "     start            start a worker"
+      opts.separator "     stop             stop all workers"
+      opts.separator "     restart          restart workers"
+      opts.separator " "
+      opts.separator "  and [options] are:"
+
+      opts.on('-w', '--workers N', 'number of workers to spin up') do |workers|
+        options.workers = workers.to_i
+      end
+      opts.on('-c', '--checks-directory DIR', 'sandboxed check directory') do |dir|
+        options.check_directory = dir
+      end
+    end
+
+    begin
+      opts.parse!(args)
+    rescue => e
+      puts e.message.capitalize + "\n\n"
+      puts opts
+      exit 1
+    end
+
+    options.workers ||= 5
+
+    unless %w(start stop restart).include?(args[0])
+      puts opts
+      exit 1
+    end
+
+    options
+  end
+end
+
+

data/lib/flapjack/database.rb

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby 
+
+require 'dm-core'
+require 'dm-validations'
+require 'dm-types'
+require 'dm-timestamps'
+
+Dir.glob(File.join(File.dirname(__FILE__), 'models', '*.rb')).each do |model|
+  require model
+end

data/lib/flapjack/models/check.rb

@@ -0,0 +1,92 @@
+class Check
+  include DataMapper::Resource
+
+  timestamps :at
+
+  has n, :related_checks, :child_key => [:child_id, :parent_id]
+  has n, :events
+
+  #has n, :parent_checks, :through => :related_checks, 
+  #       :child_key => :child_id,  :class_name => "Check"
+  #has n, :child_checks,  :through => :related_checks, 
+  #       :child_key => :parent_id, :class_name => "Check"
+
+  belongs_to :node
+  belongs_to :check_template
+
+  property :id, Serial, :key => true
+  property :command, Text, :nullable => false
+  property :params, Yaml
+  property :name, String, :nullable => false
+  property :enabled, Boolean, :default => false
+  property :status, Integer, :default => 0
+
+  # dm-timestamps
+  property :created_at, DateTime
+  property :updated_at, DateTime
+  property :deleted_at, ParanoidDateTime
+
+  # copy command onto check
+  before :valid? do 
+    if self.check_template && self.command.blank?
+      self.command = self.check_template.command
+      self.name = self.check_template.name
+      self.params = (self.check_template.params || {})
+    end
+  end
+
+  def parameters_and_values
+    names = parameter_names_from_command
+    hash = {}
+    names.each { |name| hash[name] = params ? params[name] : "" }
+    hash["$FQDN"] = self.node_fqdn # pkey of node check belongs to
+    hash
+  end
+
+  def parameter_names_from_command
+    self.command.split.grep(/\$\w/)
+  end
+
+  def executed_command
+    c = self.command
+    parameters_and_values.each_pair do |param, value|
+      value = value.to_s
+      c.gsub!(param, value)
+    end
+    return c
+  end
+
+  # FIXME: this should work through related checks association, but doesn't
+  def parent_checks
+    RelatedCheck.all(:child_id => self.id).map {|rc| rc.parent_check}
+  end
+
+  def child_checks
+    RelatedCheck.all(:parent_id => self.id).map {|rc| rc.child_check}
+  end
+
+  def worst_parent_status
+    if parent_checks.size > 0
+      self.parent_checks.map { |parent| parent.status }.sort.pop
+    else
+      0
+    end
+  end
+
+
+  GOOD = 0
+  BAD  = 1
+  UGLY = 2
+  
+  def pretty_print_status
+    case self.status
+    when GOOD
+      "good"
+    when BAD
+      "bad"
+    when UGLY
+      "ugly"
+    end
+  end
+
+end

data/lib/flapjack/models/check_template.rb

@@ -0,0 +1,18 @@
+class CheckTemplate
+  include DataMapper::Resource
+
+  has n, :checks
+
+  property :id,      Serial
+  property :command, Text, :nullable => false
+  property :name,    String, :nullable => false
+  property :params,  Yaml
+
+  validates_is_unique :command, :name
+
+  def name_for_select
+    cmd = (self.command.size > 30 ? "#{self.command[0..50]}..." : self.command)
+    "#{self.name} (#{cmd})"
+  end
+
+end

data/lib/flapjack/models/node.rb

@@ -0,0 +1,16 @@
+class Node
+  include DataMapper::Resource
+
+  has n, :checks
+
+  property :fqdn, String, :key => true
+
+  validates_is_unique :fqdn
+  validates_format :fqdn, :with => /^[0-9|a-z|A-Z|\-|\.]*$/, 
+                          :message => "not a RFC1035-formatted FQDN (see section 2.3.1)"
+
+  def hostname
+    self.fqdn.split('.').first
+  end
+
+end

data/lib/flapjack/models/related_check.rb

@@ -0,0 +1,13 @@
+class RelatedCheck
+  include DataMapper::Resource
+ 
+  belongs_to :parent_check, :class_name => "Check", :child_key => [:parent_id]
+  belongs_to :child_check, :class_name => "Check", :child_key => [:child_id]
+
+  property :id,        Serial, :key => true
+  #property :parent_check_id, Integer, :nullable => false
+  #property :child_check_id,  Integer, :nullable => false
+
+
+
+end

data/lib/flapjack/notifier.rb

@@ -0,0 +1,35 @@
+#!/usr/bin/env ruby
+
+require 'ostruct'
+
+module Flapjack
+  class Notifier
+   
+    attr_reader :recipients, :log, :notifiers
+  
+    def initialize(opts={})
+      @log = opts[:logger]
+      raise "you have to specify a logger" unless @log
+      @recipients = (opts[:recipients] || [])
+      
+      @notifiers = []
+      if opts[:notifiers]
+        opts[:notifiers].each do |n|
+          @notifiers << n
+          @log.info("using the #{n.class.to_s.split("::").last} notifier")
+        end
+      else
+        @log.warning("there are no notifiers")
+      end
+    end
+    
+    def notify!(result)
+      @notifiers.each do |n|
+        @recipients.each do |recipient|
+          @log.info("Notifying #{recipient.name} via #{n.class} about check #{result.id}")
+          n.notify(:result => result, :who => recipient)
+        end
+      end
+    end
+  end
+end

data/lib/flapjack/notifiers/mailer/init.rb

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+
+require File.join(File.dirname(__FILE__), 'mailer')

data/lib/flapjack/notifiers/mailer/mailer.rb

@@ -0,0 +1,47 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'net/smtp'
+require 'mailfactory'
+
+module Flapjack
+  module Notifiers
+
+    class Mailer
+
+      def initialize(opts={})
+        @from_address = opts[:from_address]
+        @website_uri  = opts[:website_uri] ? opts[:website_uri].gsub(/\/$/, '') : "http://#{`hostname`}"
+        @log = opts[:logger]
+        @log ||= Log4r::Logger.new("notifier")
+      end
+
+      def notify(opts={})
+        raise unless (opts[:who] && opts[:result])
+    
+        mail = MailFactory.new
+        mail.to = opts[:who].email
+        mail.from = @from_address
+        mail.subject = "Check: #{opts[:result].id}, Status: #{opts[:result].status}"
+        mail.text = <<-DESC
+          Check #{opts[:result].id} returned the status "#{opts[:result].status}".
+          
+          Here was the output: 
+            #{opts[:result].output}
+    
+          You can respond to this issue at:
+            #{@website_uri}/issue/#{opts[:result].id}
+        DESC
+
+        begin 
+          Net::SMTP.start('localhost') do |smtp|
+            smtp.sendmail(mail.to_s, mail.from, mail.to)
+          end
+        rescue Errno::ECONNREFUSED
+          @log.error("Couldn't establish connection to mail server!")
+        end
+      end
+    
+    end
+  end
+end

data/lib/flapjack/notifiers/xmpp/init.rb

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby 
+
+require File.join(File.dirname(__FILE__), 'xmpp')

data/lib/flapjack/notifiers/xmpp/xmpp.rb

@@ -0,0 +1,42 @@
+#!/usr/bin/env ruby
+
+require 'xmpp4r-simple'
+
+module Flapjack
+  module Notifiers
+    class Xmpp
+    
+      def initialize(opts={})
+    
+        @jid = opts[:jid]
+        @password = opts[:password]
+        @log = opts[:logger]
+        unless @jid && @password 
+          raise ArgumentError, "You have to provide a username and password"
+        end
+
+        begin 
+          @xmpp = Jabber::Simple.new(@jid, @password)
+        rescue SocketError => e
+          @log.error("XMPP: #{e.message}")
+        end
+    
+      end
+    
+      def notify(opts={})
+    
+        raise unless opts[:who] && opts[:result]
+    
+        message = <<-DESC
+          Check #{opts[:result].id} returned the status "#{opts[:result].status}".
+            http://localhost:4000/checks/#{opts[:result].id}
+        DESC
+    
+        @xmpp.deliver(opts[:who].jid, message)
+    
+      end
+    
+    end
+  end
+end
+

data/lib/flapjack/patches.rb

@@ -0,0 +1,26 @@
+require 'ostruct'
+require 'daemons'
+require 'log4r'
+
+class OpenStruct
+  def to_h
+    @table
+  end
+end
+
+module Daemons
+  class PidFile 
+    # we override this method so creating pid files is fork-safe
+    def filename 
+      File.join(@dir, "#{@progname}#{Process.pid}.pid")
+    end
+  end
+end
+
+module Log4r
+  class Logger
+    def error(args)
+      err(args)
+    end
+  end
+end

data/lib/flapjack/result.rb

@@ -0,0 +1,47 @@
+#!/usr/bin/env ruby
+
+require 'ostruct'
+
+##
+# Representation of a check result, as it's popped off the beanstalk.
+# 
+# Provides several convience methods for querying the status of a result. 
+#
+# Convenience methods are used by the Notifier to determine whether a 
+# notification needs to be sent out.
+module Flapjack
+  class Result < OpenStruct
+  
+    # Whether a check returns an ok status. 
+    def ok?
+      self.retval == 0
+    end
+  
+    # Whether a check has a warning status.
+    def warning?
+      self.retval == 1
+    end
+  
+    # Whether a check has a critical status.
+    def critical?
+      self.retval == 2
+    end
+  
+    # Human readable representation of the check's return value.
+    def status
+      case self.retval
+      when 0 ; "ok"
+      when 1 ; "warning"
+      when 2 ; "critical"
+      end
+    end
+  
+    # The id of a check result.
+    def id
+      # openstruct won't respond, so we have to manually define it
+      @table[:id]
+    end
+  
+  end
+end
+

metadata

@@ -0,0 +1,205 @@
+--- !ruby/object:Gem::Specification 
+name: flapjack
+version: !ruby/object:Gem::Version 
+  version: 0.4.10
+platform: ruby
+authors: 
+- Lindsay Holmwood
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-07-16 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: daemons
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.0.10
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: beanstalk-client
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.0.2
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: log4r
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.0.5
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: xmpp4r-simple
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.8.8
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: mailfactory
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.4.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: dm-core
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.11
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: dm-timestamps
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.11
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: dm-types
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.11
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: dm-validations
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.11
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: data_objects
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.12
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: do_sqlite3
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.12
+    version: 
+description: Flapjack is highly scalable and distributed monitoring system. It understands the Nagios plugin format, and can easily be scaled from 1 server to 1000.
+email: lindsay@holmwood.id.au
+executables: 
+- flapjack-notifier
+- flapjack-notifier-manager
+- flapjack-stats
+- flapjack-worker
+- flapjack-worker-manager
+- install-flapjack-systemwide
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- LICENCE
+- README.md
+- Rakefile
+- TODO.md
+- bin/flapjack-notifier
+- bin/flapjack-notifier-manager
+- bin/flapjack-stats
+- bin/flapjack-worker
+- bin/flapjack-worker-manager
+- bin/install-flapjack-systemwide
+- doc/CONFIGURING.md
+- doc/DEVELOPING.md
+- doc/INSTALL.md
+- etc/default/flapjack-notifier
+- etc/default/flapjack-workers
+- etc/flapjack/flapjack-notifier.yaml.example
+- etc/flapjack/recipients.yaml.example
+- etc/init.d/flapjack-notifier
+- etc/init.d/flapjack-workers
+- flapjack.gemspec
+- lib/flapjack/checks/http_content
+- lib/flapjack/cli/notifier.rb
+- lib/flapjack/cli/notifier_manager.rb
+- lib/flapjack/cli/worker.rb
+- lib/flapjack/cli/worker_manager.rb
+- lib/flapjack/database.rb
+- lib/flapjack/models/check.rb
+- lib/flapjack/models/check_template.rb
+- lib/flapjack/models/node.rb
+- lib/flapjack/models/related_check.rb
+- lib/flapjack/notifier.rb
+- lib/flapjack/notifiers/mailer/init.rb
+- lib/flapjack/notifiers/mailer/mailer.rb
+- lib/flapjack/notifiers/xmpp/init.rb
+- lib/flapjack/notifiers/xmpp/xmpp.rb
+- lib/flapjack/patches.rb
+- lib/flapjack/result.rb
+has_rdoc: true
+homepage: http://flapjack-project.com
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: a scalable and distributed monitoring system
+test_files: []
+