flying 0.0.2 → 0.0.3

data/.travis.yml

@@ -0,0 +1 @@
+script: bundle exec rspec spec

data/README.markdown

@@ -12,10 +12,10 @@ Create a Ruby file anywhere with the following content:
 # flying.rb
 require "flying"
 
-Flying.start do
-  site "http://www.google.com"
-  site "http://www.nytimes.com"
-end
+site "http://www.google.com"
+site "http://www.nytimes.com"
+
+start # starts monitoring
 ```
 
 Now, just run in your terminal:
@@ -31,32 +31,32 @@ We want to be able to use the following DSL:
 ```ruby
 require "flying"
 
-Flying.start do
-  # Defines the interval between verifications
-  timer 1
+# Defines the interval between verifications
+timer 1
+
+# Send an email if any error occured
+notify_me :email => "email@gmail.com"
 
-  # Send an email if any error occured
-  notify_me :email => "email@gmail.com"
+# When an error happens, monitoring stops. :on_error_continue
+# makes Flying ignore the error and continue monitoring.
+site "http://www.google.com", :as => :google, :on_error => :continue
 
-  # When an error happens, monitoring stops. :on_error_continue
-  # makes Flying ignore the error and continue monitoring.
-  site "http://www.google.com", :as => :google, :on_error_continue
+# If :google is down (above, :as => :google), Flying will notify the related services,
+# e.g. 'www.mysite.com is probably not working properly because :google is offline'
+site "http://www.mysite.com", :depends_on => :google
 
-  # If :google is down (above, :as => :google), Flying will notify the related services,
-  # e.g. 'www.mysite.com is probably not working properly because :google is offline'
-  site "http://www.mysite.com", :depends_on => :google
+# Besides checking site availability, makes sure the response.body is a valid XML,
+# ideal for making sure 3rd-party webservices are working.
+site "http://www.fedex.com/webservice", :format => :xml
 
-  # Besides checking site availability, makes sure the response.body is a valid XML,
-  # ideal for making sure 3rd-party webservices are working.
-  site "http://www.fedex.com/webservice", :format => :xml
+# Logs to the server via SSH, goes to a Rails app directory and do the following:
+#
+# - Reads production.log file and checks if any error happened (e.g. Errno... )
+# - Looks for config/database.yml and tries connecting to all possible DBs, except
+#   'development' and 'test'
+rails_app "ssh://kurko@google.com", :path => "/var/rails/google"
 
-  # Logs to the server via SSH, goes to a Rails app directory and do the following:
-  #
-  # - Reads production.log file and checks if any error happened (e.g. Errno... )
-  # - Looks for config/database.yml and tries connecting to all possible DBs, except
-  #   'development' and 'test'
-  rails_app "ssh://kurko@google.com", :path => "/var/rails/google"
-end
+start
 ```
 
 ###License

data/lib/flying/aliases/aliases.rb

@@ -1,7 +1,10 @@
 module Kernel
   def site(referer, *options)
-    process = Flying::Bot::Up.new
-    assessment = process.assess(referer, options)
-    puts process.message unless assessment
+    bot = Flying::Bot::Up.new(referer, options)
+    Flying.add_bot(bot)
+  end
+  
+  def start
+    Flying.start
   end
 end

data/lib/flying/bots/up.rb

@@ -3,27 +3,57 @@ require "net/http"
 module Flying
   module Bot
     class Up
-      attr_reader :message
-      
-      def initialize
+      attr_reader :message, :error, :referer
+      #
+      # => Name
+      #
+      # How a service is known. Useful reusing and setting dependencies.
+      #
+      #   e.g. `site "etc.com", :as => :google`
+      #   name is :google.
+      #
+      # => Dependency
+      #
+      # When a service depends on another, it has dependencies. This is
+      # an array with the Name of each service.
+      #
+      attr_accessor :name, :dependency
+
+      def initialize(referer, options = {})
+        @referer = referer
+        @options = options
         @message = "Ok."
+        @error = false
+        @name = options[:as] if options.include?(:as)
+        @dependency = []
+        
+        # Dependency can be one or an array of symbols, whereas @dependency 
+        # must be only a flat array
+        if options.include?(:depends_on)
+          @dependency << options[:depends_on] if options[:depends_on].kind_of? Symbol
+          @dependency = options[:depends_on] if options[:depends_on].kind_of? Array
+        end
       end
       
-      def assess(referer, *options)
+      # Starts assessment of the current service using variables set during
+      # initialization
+      def assess
+        @error = false
         begin
-          response_code = get_http_response_code(referer)
+          response_code = get_http_response_code(@referer)
         rescue
           Flying.an_error_ocurred(true)
-          set_error_message(referer, false, $!)
+          set_error_message(@referer, false, $!)
           return false
         end
         return true if ["200", "302"].include? response_code
         Flying.an_error_ocurred(true)
-        set_error_message(referer, response_code.to_s)
+        set_error_message(@referer, response_code.to_s)
         false
       end
       
       def set_error_message(referer, response_code, error_details = '')
+        @error = true
         case response_code
         when false
           @message = message_unreachable
@@ -34,7 +64,7 @@ module Flying
         else
           @message = message_unknown_error + "(#{error_details})"
         end
-        @message = referer + ": " + @message
+        @message = "\e[31m" + referer + ": " + @message + "\e[0m"
       end
       
       def get_http_response_code referer

data/lib/flying/monitor.rb

@@ -0,0 +1,49 @@
+# This class instance holds the current monitoring.
+module Flying
+  class Monitor
+    
+    def initialize
+      # This is where each bot instance is saved.
+      @bots = []
+      @an_error_ocurred = false
+      @first_attempt = true
+      @total_attempts = 0
+    end
+    
+    def add_bot(bot)
+      @bots << bot
+    end
+    
+    # Loops through each bot, making them perform verifications
+    def perform
+      puts "Running..."
+      looping = true
+      
+      while(looping) do
+        @bots.each { |bot|
+          has_error = bot.error
+          assessment = bot.assess
+          if has_error == false && assessment == false
+            @an_error_ocurred = true
+            puts bot.message
+          elsif has_error && assessment
+            puts "\e[32m" + bot.referer + " is back online." + "\e[0m"
+          end
+        }
+        if @first_attempt
+          if @an_error_ocurred
+            puts "Some services are down. I'll continue monitoring the targets anyway."
+          else
+            puts "Everything's working fine. I'll continue monitoring the targets."
+          end
+          
+          @first_attempt = false
+        end
+        @total_attempts += 1
+        sleep(5)
+      end
+      
+    end
+    
+  end
+end

data/lib/flying/version.rb

@@ -1,3 +1,3 @@
 module Flying
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/lib/bots/up_spec.rb

@@ -0,0 +1,34 @@
+require "spec_helper"
+
+describe Flying::Bot::Up do
+  context "initializing" do
+    it "sets an alias" do
+      bot = Flying::Bot::Up.new("http://www.google.com", :as => :google)
+      bot.name.should == :google
+    end
+
+    it "has a service as dependency" do
+      bot = Flying::Bot::Up.new("http://www.google.com", :depends_on => :google)
+      bot.dependency.should include(:google)
+
+      bot = Flying::Bot::Up.new("http://www.google.com", :depends_on => [:google, :fedex] )
+      bot.dependency.should include(:google)
+      bot.dependency.should include(:fedex)
+    end
+  end
+  
+  context "assessing" do
+    subject { Flying::Bot::Up.new("http://www.google.com") }
+
+    it "checks if site is up" do
+      subject.stub(:get_http_response_code).and_return("302")
+      subject.assess.should be_true
+    end
+
+    it "returns false if 404 and saves a message" do
+      subject.stub(:get_http_response_code).and_return("404")
+      subject.assess.should be_false
+      subject.message.should =~ /google\.com.*target was simply not found \(404\)/
+    end
+  end
+end

data/spec/lib/flying_spec.rb

@@ -0,0 +1,13 @@
+require "spec_helper"
+
+describe Flying do
+  context "regarding errors" do
+    subject { Flying::Bot::Up.new("http://www.google.com") }
+    
+    it "no error happened" do
+      subject.stub(:get_http_response_code).and_return("302")
+      subject.assess.should be_true
+      subject.error.should be_false
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -1 +1,4 @@
-require File.expand_path("../../lib/flying.rb", __FILE__)
+require File.expand_path("../../lib/flying.rb", __FILE__)
+
+RSpec.configure do |c|
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: flying
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-11-26 00:00:00.000000000 Z
+date: 2011-11-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70101949261140 !ruby/object:Gem::Requirement
+  requirement: &70171055504220 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,7 +21,7 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70101949261140
+  version_requirements: *70171055504220
 description: Verifies if each given server is not down.
 email:
 - chavedomundo@gmail.com
@@ -31,6 +31,7 @@ extra_rdoc_files: []
 files:
 - .gitignore
 - .rspec
+- .travis.yml
 - Gemfile
 - README.markdown
 - Rakefile
@@ -38,8 +39,10 @@ files:
 - lib/flying.rb
 - lib/flying/aliases/aliases.rb
 - lib/flying/bots/up.rb
+- lib/flying/monitor.rb
 - lib/flying/version.rb
-- spec/bots/up_spec.rb
+- spec/lib/bots/up_spec.rb
+- spec/lib/flying_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/kurko/flying
 licenses: []
@@ -66,5 +69,6 @@ signing_key:
 specification_version: 3
 summary: Verifies if services are up and running.
 test_files:
-- spec/bots/up_spec.rb
+- spec/lib/bots/up_spec.rb
+- spec/lib/flying_spec.rb
 - spec/spec_helper.rb

data/spec/bots/up_spec.rb

@@ -1,20 +0,0 @@
-require "spec_helper"
-
-describe Flying::Bot::Up do
-  subject { Flying::Bot::Up.new }
-  
-  context "up" do
-    it "checks if site is up" do
-      subject.stub(:get_http_response_code).and_return("302")
-      subject.assess("http://www.google.com").should be_true
-    end
-  end
-  
-  context "down" do
-    it "returns false if 404 and saves a message" do
-      subject.stub(:get_http_response_code).and_return("404")
-      subject.assess("http://www.google.com").should be_false
-      subject.message.should == "http://www.google.com: The target was simply not found (404)."
-    end
-  end
-end