flapjack 0.4.12 → 0.5.1

data/lib/flapjack/filters/ok.rb

@@ -0,0 +1,13 @@
+module Flapjack
+  module Filters
+    class Ok
+      def initialize(opts={})
+        @log = opts[:log]
+      end
+
+      def block?(result)
+        !result.warning? || !result.critical?
+      end
+    end
+  end
+end

data/lib/flapjack/inifile.rb

@@ -0,0 +1,44 @@
+#!/usr/bin/env ruby 
+
+# Nothing Flapjack-specific here - feel free to reuse.
+
+module Flapjack
+  class Inifile
+  
+    def initialize(string)
+      @data = {}
+  
+      string.split("\n").each do |line|
+        # sections
+        if line =~ /^\s*\[(.+)\]\s*(;.+)*$/
+          @current_section = $1
+          @data[@current_section] ||= {}
+        end
+        # parameters
+        if line =~ /^\s*(.+)\s*=\s*([^;]+)\s*(;.+)*$/ # after = captures up to ; then groups everything after ;
+          key = $1.strip
+          value = $2.strip
+          @data[@current_section][key] = value
+        end
+      end
+    end
+  
+    def [](key)
+      @data[key]
+    end
+  
+    def keys
+      @data.keys
+    end
+  
+    def self.read(filename)
+      self.new(File.read(filename))
+    end
+
+    def all
+      @data
+    end
+  
+  end
+end
+

data/lib/flapjack/{notifier.rb → notifier_engine.rb}

@@ -3,14 +3,13 @@
 require 'ostruct'
 
 module Flapjack
-  class Notifier
+  class NotifierEngine
    
-    attr_reader :recipients, :log, :notifiers
+    attr_reader :log, :notifiers
   
     def initialize(opts={})
-      @log = opts[:logger]
+      @log = opts[:log]
       raise "you have to specify a logger" unless @log
-      @recipients = (opts[:recipients] || [])
       
       @notifiers = []
       if opts[:notifiers]
@@ -19,15 +18,20 @@ module Flapjack
           @log.info("using the #{n.class.to_s.split("::").last} notifier")
         end
       else
-        @log.warning("there are no notifiers")
+        @log.warning("There are no notifiers! flapjack-notifier won't be useful.")
       end
     end
    
-    # FIXME: use opts={} convention
-    def notify!(result, event)
+    def notify!(options={})
+      result     = options[:result]
+      event      = options[:event]
+      recipients = options[:recipients]
+
+      raise ArgumentError, "A result + event were not passed!" unless result && event
+
       @notifiers.each do |n|
-        @recipients.each do |recipient|
-          @log.info("Notifying #{recipient.name} via #{n.class} about check #{result.id}")
+        recipients.each do |recipient|
+          @log.info("Notifying #{recipient.name} via #{n.class} about check #{result.check_id}")
           n.notify(:result => result, :who => recipient, :event => event)
         end
       end

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

@@ -3,40 +3,39 @@
 require 'rubygems'
 require 'net/smtp'
 require 'tmail'
-require 'log4r'
 
 module Flapjack
   module Notifiers
 
     class Mailer
 
+      attr_accessor :log, :from_address
+
       def initialize(opts={})
-        if opts[:from_address]
-          @from_address = opts[:from_address]
-        else
-          raise ArgumentError, "from address must be provided"
-        end
-        @website_uri  = opts[:website_uri] ? opts[:website_uri].gsub(/\/$/, '') : "http://#{`hostname`}"
-        @log = opts[:logger]
-        @log ||= ::Log4r::Logger.new("notifier")
+        @log = opts[:log]
+        @from_address = opts[:from_address]
+        @website_uri = opts[:website_uri]
+        
+        raise ArgumentError, "from address must be provided" unless @from_address
       end
 
       def notify(opts={})
         raise ArgumentError, "a recipient was not specified" unless opts[:who]
         raise ArgumentError, "a result was not specified" unless opts[:result]
-    
+
+        # potential FIXME: refactor TMail out entirely?
         mail = TMail::Mail.new
         mail.to = opts[:who].email
         mail.from = @from_address
-        mail.subject = "Check: #{opts[:result].id}, Status: #{opts[:result].status}"
+        mail.subject = "Check: #{opts[:result].check_id}, Status: #{opts[:result].status}"
         mail.body = <<-DESC
-          Check #{opts[:result].id} returned the status "#{opts[:result].status}".
+          Check #{opts[:result].check_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}
+            #{@website_uri}/issue/#{opts[:result].check_id}
         DESC
 
         begin 

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

@@ -31,8 +31,8 @@ module Flapjack
         raise ArgumentError, "a result was not specified" unless opts[:result]
     
         text = <<-DESC
-          Check #{opts[:result].id} returned the status "#{opts[:result].status}".
-            http://localhost:4000/checks/#{opts[:result].id}
+          Check #{opts[:result].check_id} returned the status "#{opts[:result].status}".
+            http://localhost:4000/checks/#{opts[:result].check_id}
         DESC
    
         message = Jabber::Message.new(opts[:who].jid, text)

data/lib/flapjack/patches.rb

@@ -22,5 +22,30 @@ module Log4r
     def error(args)
       err(args)
     end
+
+    def warning(args)
+      warn(args)
+    end
+  end
+end
+
+# extracted from Extlib. 
+# FIXME: what's the licensing here?
+class String
+  def camel_case                                                                                                                                    
+    return self if self !~ /_/ && self =~ /[A-Z]+.*/                                                                                                
+    split('_').map{|e| e.capitalize}.join                                                                                                           
+  end   
+end
+
+# http://gist.github.com/151324
+class Hash
+  def symbolize_keys
+    inject({}) do |acc, (k,v)|
+      key = String === k ? k.to_sym : k
+      value = Hash === v ? v.symbolize_keys : v
+      acc[key] = value
+      acc
+    end
   end
 end

data/lib/flapjack/persistence/couch.rb

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

data/lib/flapjack/persistence/couch/connection.rb

@@ -0,0 +1,66 @@
+#!/usr/bin/env ruby
+
+module Flapjack
+  module Persistence
+    class Couch
+      class Connection
+        class << self
+          attr_accessor :host, :port
+          def setup(options={})
+            @host = options[:host]
+            @port = options[:port]
+            @database = options[:database]
+          end
+  
+          def get(id)
+            uri = "/#{@database}/#{id}"
+            req = ::Net::HTTP::Get.new(uri)
+            request(req)
+          end
+  
+          def post(options={})
+            document = options[:document]
+            uri = "/#{@database}/"
+  
+            req = ::Net::HTTP::Post.new(uri)
+            req["content-type"] = "application/json"
+            req.body = document.to_json
+  
+            request(req) 
+          end
+  
+          def put(options={})
+            document = options[:document]
+            uri = "/#{@database}/#{(options[:document]["id"] || options[:document]["_id"])}"
+  
+            req = ::Net::HTTP::Put.new(uri)
+            req["content-type"] = "application/json"
+            req.body = document.to_json
+  
+            request(req) 
+          end
+
+          def delete(options={})
+            document = options[:document]
+
+            uri = "/#{@database}/#{(options[:document]["id"] || options[:document]["_id"])}"
+  
+            req = ::Net::HTTP::Put.new(uri)
+            req["content-type"] = "application/json"
+            req.body = document.to_json
+
+            request(req)
+          end
+  
+          def request(request)
+            response = Net::HTTP.start(@host, @port) {|http| http.request(request)}
+           
+            @parser = Yajl::Parser.new
+            hash = @parser.parse(response.body)
+          end
+        end
+      end
+    end # class Couch
+  end # module Persistence
+end # module Flapjack
+

data/lib/flapjack/persistence/couch/couch.rb

@@ -0,0 +1,63 @@
+#!/usr/bin/env ruby
+
+require 'net/http'
+require 'yajl/json_gem'
+
+module Flapjack
+  module Persistence
+    class Couch
+
+      attr_accessor :config
+
+      def initialize(options={})
+        @options = options
+        @config = OpenStruct.new(options)
+        @log = @config.log
+        
+        Flapjack::Persistence::Couch::Connection.setup(@options)
+      end
+
+      def any_parents_failed?(id)
+        id == "1" ? false : true
+      end
+
+      def save_check(attrs)
+        true
+      end
+
+      def get_check(id)
+        id == "4" ? nil : true
+      end
+
+      def delete_check(id)
+        true
+      end
+
+      def all_checks
+        [true, true, true]
+      end
+
+      def save_check_relationship(attrs)
+        true
+      end
+
+      def create_event(result)
+        true
+      end
+
+      def all_events_for(id)
+        [true, true, true]
+      end
+
+      def all_events
+        [true, true, true]
+      end
+
+      def all_check_relationships
+        [true, true, true]
+      end
+
+    end # class Couch
+  end # module Persistence
+end # module Flapjack
+

data/lib/flapjack/persistence/data_mapper.rb

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

data/lib/flapjack/persistence/data_mapper/data_mapper.rb

@@ -0,0 +1,67 @@
+#!/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
+
+module Flapjack
+  module Persistence
+    class DataMapper
+      def initialize(options={})
+        @options = options
+        @config = OpenStruct.new(options)
+        @log = @config.log
+        connect
+      end
+
+      def any_parents_failed?(result)
+        check = Check.get(result.check_id)
+        if check
+          check.worst_parent_status != 0
+        else
+          @log.warning("Check with id #{result.check_id} doesn't exist!")
+          false # this will notify
+        end
+      end
+
+      def save(result)
+        check = Check.get(result.check_id)
+        if check
+          check.status = result.status
+          check.save
+        else
+          @log.warning("Check with id #{result.check_id} doesn't exist!")
+          true
+        end
+      end
+
+      def create_event(result)
+        event = Event.new(:check_id => result.check_id)
+        event.save
+      end
+
+      private 
+      def connect
+        raise ArgumentError, "Database URI wasn't specified" unless @config.uri
+        ::DataMapper.setup(:default, @config.uri)
+        validate_structure
+      end
+
+      def validate_structure
+        begin
+          ::DataMapper.repository(:default).adapter.execute("SELECT 'id' FROM 'checks';")
+        rescue Sqlite3Error => e
+          @log.warning("The specified database doesn't appear to have any structure!")
+          @log.warning("Exiting.")
+          raise
+        end
+      end
+    end
+  end
+end
+

data/lib/flapjack/{models → persistence/data_mapper/models}/check.rb

@@ -74,17 +74,13 @@ class Check
   end
 
 
-  GOOD = 0
-  BAD  = 1
-  UGLY = 2
-  
   def pretty_print_status
     case self.status
-    when GOOD
+    when 0
       "good"
-    when BAD
+    when 1
       "bad"
-    when UGLY
+    when 2
       "ugly"
     end
   end

data/lib/flapjack/persistence/data_mapper/models/event.rb

@@ -0,0 +1,17 @@
+class Event
+  include DataMapper::Resource
+
+  timestamps :at
+
+  belongs_to :check
+
+  property :id, Serial, :key => true
+  property :check_id, Integer, :nullable => false
+  #property :check_output, Text, :nullable => false
+
+  # dm-timestamps
+  property :created_at, DateTime
+  property :updated_at, DateTime
+  property :deleted_at, ParanoidDateTime
+
+end