dogeventer 1.0.0

data/lib/dogeventer/event_emitter.rb

@@ -0,0 +1,131 @@
+module DogEventer
+  class EventEmitter
+    attr_reader :events
+
+    def wait(seconds)
+      @start_time += seconds
+    end
+  end
+
+  class NagiosCheck < EventEmitter
+    attr_reader :host
+
+    def initialize(check_name, host, start_time)
+      @check_name = check_name
+      @host = host
+      @start_time = start_time
+      @events = []
+    end
+
+    def msg_title(alert_state)
+      "#{@check_name} is #{alert_state.to_s} on #{@host}"
+    end
+
+    def event_object()
+      @check_name
+    end
+
+    def event_type()
+      "monitoring.alert"
+    end
+
+    def alert_type(state)
+      {
+        :critical => :error,
+        :ok       => :success,
+        :warning  => :warning,
+        :warn     => :warn
+      }[state]
+    end
+
+    def date_happened(date=nil)
+      (date || @start_time).to_i
+    end
+
+    def source_type_name()
+      "Nagios"
+    end
+
+    def to_event(alert_state, date=nil)
+      @events << {
+        :event_object  => event_object,
+        :event_type    => event_type,
+        :alert_type    => alert_type(alert_state),
+        :date_happened => date_happened(date),
+        :msg_title     => msg_title(alert_state),
+        :source_type_name => source_type_name,
+        :host          => host
+      }
+    end
+
+    def critical(date=nil)
+      to_event :critical
+    end
+
+    def warning(date=nil)
+      to_event :warning
+    end
+
+    def ok(date=nil)
+      to_event :ok
+    end
+  end
+
+  class PingdomCheck < EventEmitter
+    attr_reader :url
+
+    def initialize(check_name, url, start_time)
+      @check_name = check_name
+      @url = url
+      @start_time = start_time
+      @events = []
+    end
+
+    def msg_title(alert_state)
+      "#{@check_name} is #{alert_state.to_s}"
+    end
+
+    def event_object()
+      @check_name
+    end
+
+    def event_type()
+      "monitoring.alert"
+    end
+
+    def alert_type(state)
+      {
+        :down => :error,
+        :up   => :success
+      }[state]
+    end
+
+    def date_happened(date=nil)
+      (date || @start_time).to_i
+    end
+
+    def source_type_name()
+      "Pingdom"
+    end
+
+    def to_event(alert_state, date=nil)
+      @events << {
+        :event_object  => event_object,
+        :event_type    => event_type,
+        :alert_type    => alert_type(alert_state),
+        :date_happened => date_happened(date),
+        :msg_title     => msg_title(alert_state),
+        :source_type_name => source_type_name,
+        :host          => "Pingdom:#{@check_name}"
+      }
+    end
+
+    def down(date=nil)
+      to_event :down
+    end
+    def up(date=nil)
+      to_event :up
+    end
+
+  end
+end

data/lib/dogeventer/scope.rb

@@ -0,0 +1,36 @@
+module DogEventer
+  class Scope
+    attr_reader :events
+    def initialize(start_time, events=nil)
+      @start_time = start_time
+      @events = events || []
+    end
+  end
+
+  class Host < Scope
+    def initialize(start_time, host_name)
+      super start_time
+      @host_name = host_name
+    end
+  
+    def nagios_check(check_name, &block)
+      n = NagiosCheck.new(check_name, @host_name, @start_time)
+      n.instance_eval &block
+      @events += n.events
+    end
+  end
+
+  class Url < Scope
+    def initialize(start_time, url)
+      super start_time
+      @url = url
+    end
+  
+    def pingdom_check(check_name, &block)
+      p = PingdomCheck.new(check_name, @url, @start_time)
+      p.instance_eval &block
+      @events += p.events
+    end
+  end
+end
+

data/lib/dogeventer.rb

@@ -0,0 +1,91 @@
+require 'dogapi'
+require 'dogeventer/scope'
+require 'dogeventer/event_emitter'
+
+module DogEventer
+  class DogEventer
+    def initialize(api_key, application_key=nil)
+      @dog = Dogapi::Client.new(api_key, application_key)
+      @events = []
+    end
+
+    def generate(start_time, &block)
+      @start_time = start_time
+      instance_eval &block
+    
+      @events.each do |event|
+        puts event.inspect
+        @dog.emit_event(Dogapi::Event.new('', event), :host => event[:host])
+      end
+    end
+
+    def host(host_name, &block)
+      h = Host.new(@start_time, host_name)
+      @events += (h.instance_eval &block)
+    end
+  
+    def url(url, &block)
+      u = Url.new(@start_time, url)
+      @events += (u.instance_eval &block)
+    end
+  
+  end
+end
+
+def dogeventer(start_time=nil, api_key=nil, application_key=nil, &block)
+  require 'optparse'
+  
+  all_config = {
+    :default => {:start_time => Time.now.to_i - 60*60},
+    :cli => {},
+    :file => {}
+  }
+  config_file = "~/.dogrc"
+  
+  optparse = OptionParser.new do |opts|
+    opts.banner = "Usage: ruby #{$0} [options]"
+    opts.on("-t", "--time TIME", Float, "Start time in seconds since unix epoch. Defaults to the current time minus an hour.") { |v| 
+      all_config[:cli][:start_time] = start_time
+    }
+    opts.on("-c", "--config CONFIG_FILE", "Path to config file. Defaults to #{config_file}") { |v|
+      config_file = v
+    }
+    opts.on("--api_key API_KEY", "Api key. Overrides config values.") { |v| 
+      all_config[:cli][:api_key] = v
+    }
+    opts.on("--app_key [APP_KEY]", "App key. Overrides config values.") { |v| 
+      all_config[:cli][:app_key] = v
+    }
+    
+  end
+  optparse.parse!
+
+  if not config_file
+    config_file = default_config_file
+  end
+  
+  config_file = File.expand_path config_file
+  if File.exists? config_file
+    File.open(config_file) do |f| 
+      f.readlines.each do |line|
+        keyval = line.split "="
+        if keyval.length == 2
+          key = keyval[0].strip
+          val = keyval[1].strip
+          if key == 'apikey'
+            all_config[:file][:api_key] = val
+          elsif key == 'appkey'
+            all_config[:file][:app_key] = val
+          end
+        end
+      end
+    end
+  end
+  
+  config = all_config.reduce({}) do |merged, to_merge|
+    merged.merge to_merge[1]
+  end
+  
+  DogEventer::DogEventer.new(config[:api_key], config[:app_key]).generate config[:start_time], &block
+end
+

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification 
+name: dogeventer
+version: !ruby/object:Gem::Version 
+  hash: 23
+  prerelease: 
+  segments: 
+  - 1
+  - 0
+  - 0
+  version: 1.0.0
+platform: ruby
+authors: 
+- Datadog, Inc.
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-12-02 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: dogapi
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 19
+        segments: 
+        - 1
+        - 2
+        - 6
+        version: 1.2.6
+  type: :runtime
+  version_requirements: *id001
+description: A DSL for generating Datadog events
+email: packages@datadoghq.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/dogeventer/event_emitter.rb
+- lib/dogeventer/scope.rb
+- lib/dogeventer.rb
+homepage: http://datadoghq.com/
+licenses: 
+- BSD
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: A DSL for generating Datadog events
+test_files: []
+