bobble 0.0.4 → 0.0.5

data/lib/bobble/checker.rb

@@ -0,0 +1,112 @@
+require 'bobble/notifier/twilio'
+require 'bobble/notifier/google_voice'
+require 'bobble/notifier/gmail'
+
+module Bobble
+  class Checker
+    class << self
+
+      def check(url, options={})
+        {
+          # The status that is considered a success.
+          # If nil, any status other than 50x is success.
+          :success_status => nil,
+          # Specified headers are required for the response
+          # to be considered a success.
+          # Regex values can be specified.
+          :success_header => nil,
+          # Response body must match the specified body.
+          # Can be a regex.
+          :success_body => nil
+        }.update(options)
+
+        begin
+          response = http_request(url)
+          assert_success(response, options)
+          puts "Successful!: #{url}"
+        rescue Exception => e
+          message = "FAILED: #{url} - #{e.message}"
+          puts message
+
+          self.send_notification(message, url)
+        end
+      end
+
+      def http_request(url)
+        uri = URI.parse(url)
+        http = Net::HTTP.new(uri.host, uri.port)
+        request = Net::HTTP::Get.new(uri.request_uri)
+        return http.request(request)
+      end
+
+      # raises an exception if not successful
+      def assert_success(response, options)
+        if response.code.to_i >= 500
+          raise Exception.new("Response status code error: #{response.code}")
+        end
+
+        if options[:success_status]
+          unless response.code.to_i == options[:success_status].to_i
+            raise Exception.new("Response status code error. Is: #{response.code} Expected: #{options[:success_status]}")
+          end
+        end
+
+        if options[:success_header]
+          options[:success_header].each do |k,v|
+            header_value = response.header[k.to_s]
+            if v.is_a?(Regexp)
+              unless header_value =~ v
+                raise Exception.new("Response header '#{k}' should match regexp #{v} ('#{header_value}' does not match)")
+              end
+            else
+              unless header_value == v
+                raise Exception.new("Response header '#{k}' should have value '#{v}' (instead is #{header_value})")
+              end
+            end
+          end
+        end
+
+        if options[:success_body]
+          match = options[:success_body]
+          if match.is_a?(Regexp)
+            unless response.body =~ match
+              raise Exception.new("Response body should match regexp #{match}, but does not.")
+            end
+          else
+            unless response.body.strip == match.strip
+              raise Exception.new("Response body should be '#{match}', instead is '#{response.body}'")
+            end
+          end
+        end
+
+      end
+
+      def send_notification(message, url)
+        if Bobble.option(:gmail)
+          begin
+            Notifier::Gmail.send(message, url)
+          rescue Exception => e
+            puts "Gmail Notifier failed: #{e.message}"
+          end
+        end
+
+        if Bobble.option(:twilio)
+          begin
+            Notifier::Twilio.send(message)
+          rescue Exception => e
+            puts "Twilio Notifier failed: #{e.message}"
+          end
+        end
+
+        if Bobble.option(:google_voice)
+          begin
+            Notifier::GoogleVoice.send(message)
+          rescue Exception => e
+            puts "Google Voice Notifier failed: #{e.message}"
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/bobble/notifier/gmail.rb

@@ -0,0 +1,29 @@
+require 'pony'
+
+module Bobble
+  module Notifier
+    class Gmail
+      class << self
+
+        def send(message, url)
+          Pony.mail({
+                      :to => ENV["BOBBLE_GMAIL_TO_EMAIL"],
+                      :via => :smtp,
+                      :via_options => {
+                        :address              => 'smtp.gmail.com',
+                        :port                 => '587',
+                        :enable_starttls_auto => true,
+                        :user_name            => ENV["BOBBLE_GMAIL_USERNAME"],
+                        :password             => ENV["BOBBLE_GMAIL_PASSWORD"],
+                        :authentication       => :plain, # :plain, :login, :cram_md5, no auth by default
+                        :domain               => "localhost.localdomain" # the HELO domain provided by the client to the server
+                      },
+                      :body => message,
+                      :subject => "Bobble: #{url} is down"
+                    })
+        end
+
+      end
+    end
+  end
+end

data/lib/bobble/notifier/google_voice.rb

@@ -0,0 +1,85 @@
+require 'bobble/util'
+
+#
+# Google Voice Notifier
+# Adapted from: http://brettterpstra.com/sms-from-the-command-line-with-google-voice/
+#
+module Bobble
+  module Notifier
+    class GoogleVoice
+      class << self
+
+        def postit(uri_str, data, header = nil, limit = 3)
+          raise ArgumentError, 'HTTP redirect too deep' if limit == 0
+          url = URI.parse(uri_str)
+          http = Net::HTTP.new(url.host,443)
+          http.use_ssl = true
+          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+          response,content = http.post(url.path,data,header)
+          case response
+          when Net::HTTPSuccess     then content
+          when Net::HTTPRedirection then postit(response['location'],data,header, limit - 1)
+          else
+            puts response.inspect
+            response.error!
+          end
+        end
+
+        def getit(uri_str, header, limit = 3)
+          raise ArgumentError, 'HTTP redirect too deep' if limit == 0
+          url = URI.parse(uri_str)
+          http = Net::HTTP.new(url.host,url.port)
+          http.use_ssl = true
+          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+          response,content = http.get(url.path,header)
+          case response
+          when Net::HTTPSuccess     then content
+          when Net::HTTPRedirection then getit(response['location'],header, limit - 1)
+          else
+            response.error!
+          end
+        end
+
+        def send(message)
+          message = Bobble::Util.shorten_to_text_message(message)
+
+          username = ENV["BOBBLE_GVOICE_USERNAME"]
+          password = ENV["BOBBLE_GVOICE_PASSWORD"]
+          # TODO: support multiple numbers
+          number = ENV["BOBBLE_GVOICE_TO_PHONENUMBER"]
+          numbers = [number]
+
+          data = "accountType=GOOGLE&Email=#{username}&Passwd=#{password}&service=grandcentral&source=brettterpstra-CLISMS-2.0"
+          res = postit('https://www.google.com/accounts/ClientLogin',data)
+
+          if res
+            authcode = res.match(/Auth=(.+)/)[1]
+            header = {'Authorization' => "GoogleLogin auth=#{authcode.strip}",'Content-Length' => '0'}
+            newres = getit('https://www.google.com/voice',header)
+
+            if newres
+              rnrse = newres.match(/'_rnr_se': '([^']+)'/)[1]
+              numbers.each do |num|
+                data = "_rnr_se=#{rnrse}&phoneNumber=#{num.strip}&text=#{message}&id="
+                finalres = postit('https://www.google.com/voice/sms/send/',data,header)
+
+                if finalres["ok"]
+                  puts "Message sent to #{num}"
+                else
+                  puts "Error sending to #{num}"
+                end
+              end
+
+            else
+              newres.error!
+            end
+
+          else
+            res.error!
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/bobble/notifier/twilio.rb

@@ -0,0 +1,35 @@
+require 'twilio-ruby'
+require 'bobble/util'
+
+module Bobble
+  module Notifier
+    class Twilio
+      class << self
+
+        @@client = nil
+
+        def create_client
+          return if @@client
+
+          account_sid = ENV['BOBBLE_TWILIO_SID']
+          auth_token = ENV['BOBBLE_TWILIO_TOKEN']
+
+          @@client = Twilio::REST::Client.new account_sid, auth_token
+        end
+        
+        def send(message)
+          create_client
+          message = Bobble::Util.shorten_to_text_message(message)
+
+          params = {
+            :from => ENV['BOBBLE_TWILIO_FROM_PHONENUMBER'],
+            :to => ENV['BOBBLE_TWILIO_TO_PHONENUMBER'],
+            :body => message
+          }
+          @@client.account.sms.messages.create(params)
+        end
+
+      end
+    end
+  end
+end

data/lib/bobble/util.rb

@@ -1,13 +1,15 @@
-class Bobble::Util
-  class << self
+module Bobble
+  class Util
+    class << self
 
-    def shorten_to_text_message(message)
-      text_message = message
-      if message.length > 140
-        text_message = message[0..136] + "..."
+      def shorten_to_text_message(message)
+        text_message = message
+        if message.length > 140
+          text_message = message[0..136] + "..."
+        end
+        return message
       end
-      return message
-    end
 
+    end
   end
 end

data/lib/bobble.rb

@@ -1,4 +1,6 @@
-class Bobble
+require 'bobble/checker'
+
+module Bobble
   class << self
 
     # default options are guessed based on what environment variables are present
@@ -8,56 +10,15 @@ class Bobble
       :gmail => !!ENV["BOBBLE_GMAIL_USERNAME"]
     }
 
-    def options(o)
-      @@options.update(o)
+    def option(key)
+      @@options[key]
     end
 
-    def check(url)
-      begin
-        response = Net::HTTP.get(URI.parse(url))
-        raise Exception.new("empty response") if response == ""
-        puts "Successful!: #{url}"
-      rescue Exception => e
-        message = "FAILED: #{url} - #{e.message}"
-        puts message
-
-        send_notification(message, url)
-      end
-    end
-
-    def send_notification(message, url)
-
-      if @@options[:gmail]
-        begin
-          GmailNotifier.send(message, url)
-        rescue Exception => e
-          puts "Gmail Notifier failed: #{e.message}"
-        end
-      end
-
-      if @@options[:twilio]
-        begin
-          TwilioNotifier.send(message)
-        rescue Exception => e
-          puts "Twilio Notifier failed: #{e.message}"
-        end
-      end
-
-      if @@options[:google_voice]
-        begin
-          GoogleVoiceNotifier.send(message)
-        rescue Exception => e
-          puts "Google Voice Notifier failed: #{e.message}"
-        end
-      end
-
+    def check(url, options={})
+      Bobble::Checker.check(url, options)
     end
 
   end
 end
 
-require 'bobble/twilio_notifier'
-require 'bobble/google_voice_notifier'
-require 'bobble/gmail_notifier'
-
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bobble
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-24 00:00:00.000000000Z
+date: 2012-07-06 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: twilio-ruby
-  requirement: &70329909850180 !ruby/object:Gem::Requirement
+  requirement: &70256347466300 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: 3.6.0
   type: :runtime
   prerelease: false
-  version_requirements: *70329909850180
+  version_requirements: *70256347466300
 - !ruby/object:Gem::Dependency
   name: pony
-  requirement: &70329909849480 !ruby/object:Gem::Requirement
+  requirement: &70256347465820 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,19 +32,20 @@ dependencies:
         version: '1.4'
   type: :runtime
   prerelease: false
-  version_requirements: *70329909849480
-description: For freely pinging your favorite web services & freely or cheaply getting
-  email/SMS notifications when they're down.
+  version_requirements: *70256347465820
+description: For pinging your web services (any URL) & freely or cheaply getting email/SMS
+  notifications when they're down.
 email: ahfarmer@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
 - lib/bobble.rb
-- lib/bobble/gmail_notifier.rb
-- lib/bobble/google_voice_notifier.rb
-- lib/bobble/twilio_notifier.rb
+- lib/bobble/checker.rb
 - lib/bobble/util.rb
+- lib/bobble/notifier/gmail.rb
+- lib/bobble/notifier/google_voice.rb
+- lib/bobble/notifier/twilio.rb
 homepage: https://github.com/ahfarmer/bobble
 licenses: []
 post_install_message: 

data/lib/bobble/gmail_notifier.rb

@@ -1,25 +0,0 @@
-require 'pony'
-
-class Bobble::GmailNotifier
-  class << self
-
-    def send(message, url)
-      Pony.mail({
-                  :to => ENV["BOBBLE_GMAIL_TO_EMAIL"],
-                  :via => :smtp,
-                  :via_options => {
-                    :address              => 'smtp.gmail.com',
-                    :port                 => '587',
-                    :enable_starttls_auto => true,
-                    :user_name            => ENV["BOBBLE_GMAIL_USERNAME"],
-                    :password             => ENV["BOBBLE_GMAIL_PASSWORD"],
-                    :authentication       => :plain, # :plain, :login, :cram_md5, no auth by default
-                    :domain               => "localhost.localdomain" # the HELO domain provided by the client to the server
-                  },
-                  :body => message,
-                  :subject => "Bobble: #{url} is down"
-                })
-    end
-
-  end
-end

data/lib/bobble/google_voice_notifier.rb

@@ -1,81 +0,0 @@
-require 'bobble/util'
-
-#
-# Google Voice Notifier
-# Adapted from: http://brettterpstra.com/sms-from-the-command-line-with-google-voice/
-#
-class Bobble::GoogleVoiceNotifier
-  class << self
-
-    def postit(uri_str, data, header = nil, limit = 3)
-        raise ArgumentError, 'HTTP redirect too deep' if limit == 0
-        url = URI.parse(uri_str)
-        http = Net::HTTP.new(url.host,443)
-        http.use_ssl = true
-        http.verify_mode = OpenSSL::SSL::VERIFY_NONE
-        response,content = http.post(url.path,data,header)
-        case response
-          when Net::HTTPSuccess     then content
-          when Net::HTTPRedirection then postit(response['location'],data,header, limit - 1)
-          else
-            puts response.inspect
-            response.error!
-        end
-    end
-
-    def getit(uri_str, header, limit = 3)
-        raise ArgumentError, 'HTTP redirect too deep' if limit == 0
-        url = URI.parse(uri_str)
-        http = Net::HTTP.new(url.host,url.port)
-        http.use_ssl = true
-        http.verify_mode = OpenSSL::SSL::VERIFY_NONE
-        response,content = http.get(url.path,header)
-        case response
-          when Net::HTTPSuccess     then content
-          when Net::HTTPRedirection then getit(response['location'],header, limit - 1)
-          else
-            response.error!
-        end
-    end
-
-    def send(message)
-      message = Bobble::Util.shorten_to_text_message(message)
-
-      username = ENV["BOBBLE_GVOICE_USERNAME"]
-      password = ENV["BOBBLE_GVOICE_PASSWORD"]
-      # TODO: support multiple numbers
-      number = ENV["BOBBLE_GVOICE_TO_PHONENUMBER"]
-      numbers = [number]
-
-      data = "accountType=GOOGLE&Email=#{username}&Passwd=#{password}&service=grandcentral&source=brettterpstra-CLISMS-2.0"
-      res = postit('https://www.google.com/accounts/ClientLogin',data)
-
-      if res
-        authcode = res.match(/Auth=(.+)/)[1]
-        header = {'Authorization' => "GoogleLogin auth=#{authcode.strip}",'Content-Length' => '0'}
-        newres = getit('https://www.google.com/voice',header)
-
-        if newres
-          rnrse = newres.match(/'_rnr_se': '([^']+)'/)[1]
-          numbers.each do |num|
-            data = "_rnr_se=#{rnrse}&phoneNumber=#{num.strip}&text=#{message}&id="
-            finalres = postit('https://www.google.com/voice/sms/send/',data,header)
-
-            if finalres["ok"]
-              puts "Message sent to #{num}"
-            else
-              puts "Error sending to #{num}"
-            end
-          end
-
-        else
-          newres.error!
-        end
-
-      else
-        res.error!
-      end
-    end
-
-  end
-end

data/lib/bobble/twilio_notifier.rb

@@ -1,31 +0,0 @@
-require 'twilio-ruby'
-require 'bobble/util'
-
-class Bobble::TwilioNotifier
-  class << self
-
-    @@client = nil
-
-    def create_client
-      return if @@client
-
-      account_sid = ENV['BOBBLE_TWILIO_SID']
-      auth_token = ENV['BOBBLE_TWILIO_TOKEN']
-
-      @@client = Twilio::REST::Client.new account_sid, auth_token
-    end
-    
-    def send(message)
-      create_client
-      message = Bobble::Util.shorten_to_text_message(message)
-
-      params = {
-        :from => ENV['BOBBLE_TWILIO_FROM_PHONENUMBER'],
-        :to => ENV['BOBBLE_TWILIO_TO_PHONENUMBER'],
-        :body => message
-      }
-      @@client.account.sms.messages.create(params)
-    end
-
-  end
-end