notifu 1.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e84a8d91a613dee08b4ebfa63f7c05082c4230d8
+  data.tar.gz: c0bf2890e4e519adc62b6f76b553fb706451146c
+SHA512:
+  metadata.gz: 9ea47dec858e01b1dcea6586a29a491a0fc5fd0cb98731f0308bd010fe51eae593b4f3f092d454b2ef0d02336b495eea67197c58034bc73ada6751fe2f77bffd
+  data.tar.gz: b212ae242588fea5fc7b9ff199fa0776337b34ab0c5ca3afe24498eb967ae972b7076c4476122debbdf7f653b27111687a01329d8e4953a993d463752450a552

data/bin/notifu

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+
+path = File.symlink?(__FILE__) ? File.readlink(__FILE__) : __FILE__
+$basepath = '/' + File.dirname(path).split('/')[1..-2].join('/') + '/lib/notifu/'
+$actorpath = $basepath + "actors/"
+# $sidekiq_bin = $basepath.sub(/app\//, '') + ".rvm/wrappers/notifu/sidekiq"
+$sidekiq_bin = "sidekiq"
+
+require 'notifu'
+
+Notifu::CONFIG = Notifu::Config.new.get
+
+Ohm.redis = Redic.new Notifu::CONFIG[:redis_data]
+
+Notifu::CLI::Root.start(ARGV)

data/lib/notifu.rb

@@ -0,0 +1,12 @@
+require 'yaml'
+require 'json'
+require 'thor'
+require 'ohm'
+require "notifu/mixins"
+require "notifu/util"
+require "notifu/config"
+require "notifu/logger"
+require "notifu/model"
+require "notifu/cli/service"
+require "notifu/cli/object"
+require "notifu/cli"

data/lib/notifu/actors/gammu_sms_bridge.rb

@@ -0,0 +1,41 @@
+module Notifu
+  module Actors
+    class GammuSmsBridge < Notifu::Actor
+
+      self.name = "gammu_sms_bridge"
+      self.desc = "Old-school NetCat-like SMS bridge to Gammu"
+      self.retry = 3
+
+      def act
+        data = OpenStruct.new({
+          notifu_id: self.issue.notifu_id,
+          host: self.issue.host,
+          message: self.issue.message,
+          service: self.issue.service,
+          status: self.issue.code.to_state,
+          first_event: Time.at(self.issue.time_created.to_i),
+          duration: (Time.now.to_i - self.issue.time_created.to_i).duration,
+          occurrences_count: self.issue.occurrences_count,
+          occurrences_trigger: self.issue.occurrences_trigger
+        })
+        message = ERB.new(self.template).result(data.instance_eval {binding})
+
+        self.contacts.each do |contact|
+          cell = contact.cell
+          template = ERB.new File.new("sms.erb").read, nil, "%"
+          message = "template.result(self.issue)"
+
+          # send message to sms-bridge
+          socket = TCPSocket.new Notifu::CONFIG[:actors][:gammu_sms_bridge][:host], Notifu::CONFIG[:actors][:gammu_sms_bridge][port]
+          socket.send contact.cell.to_s + "--" + message
+          socket.close
+          socket = nil
+        end
+      end
+
+      def template
+          "<%= data[:status] %> [<%= data[:host] %>/<%= data[:service] %>]: (<%= data[:message] %>) <%= data[:duration] %> [<%= data[:notifu_id] %>]"
+      end
+    end
+  end
+end

data/lib/notifu/actors/slack_chan.rb

@@ -0,0 +1,29 @@
+module Notifu
+  module Actors
+    class SlackChan < Notifu::Actor
+
+      require 'excon'
+      require 'erb'
+
+      self.name = "slack_msg"
+      self.desc = "Sends message to a Slack contact"
+      self.retry = 3
+
+
+      def act
+        cfg = Notifu::CONFIG[:actors][:slack]
+        contacts = self.contacts.map { |contact| contact.cell }
+        req_string = Notifu::CONFIG[:actors][:twilio_call][:api] +
+                      "?token="       + Notifu::CONFIG[:actors][:twilio_call][:token] +
+                      "&status="      + self.issue.code.to_state +
+                      "&hostname="    + self.issue.host +
+                      "&service="     + self.issue.service +
+                      "&description=" + ERB::Util.url_encode(self.issue.message.to_s) +
+                      "&call_group="  + ERB::Util.url_encode(contacts.to_json) +
+                      "&init=1"
+        Excon.get req_string if self.issue.code.to_i == 2
+      end
+
+    end
+  end
+end

data/lib/notifu/actors/slack_msg.rb

@@ -0,0 +1,29 @@
+module Notifu
+  module Actors
+    class SlackMsg < Notifu::Actor
+
+      require 'excon'
+      require 'erb'
+
+      self.name = "slack_msg"
+      self.desc = "Sends message to a Slack contact"
+      self.retry = 3
+
+
+      def act
+        cfg = Notifu::CONFIG[:actors][:slack]
+        contacts = self.contacts.map { |contact| contact.cell }
+        req_string = Notifu::CONFIG[:actors][:twilio_call][:api] +
+                      "?token="       + Notifu::CONFIG[:actors][:twilio_call][:token] +
+                      "&status="      + self.issue.code.to_state +
+                      "&hostname="    + self.issue.host +
+                      "&service="     + self.issue.service +
+                      "&description=" + ERB::Util.url_encode(self.issue.message.to_s) +
+                      "&call_group="  + ERB::Util.url_encode(contacts.to_json) +
+                      "&init=1"
+        Excon.get req_string if self.issue.code.to_i == 2
+      end
+
+    end
+  end
+end

data/lib/notifu/actors/smtp.rb

@@ -0,0 +1,73 @@
+module Notifu
+  module Actors
+    class Smtp < Notifu::Actor
+
+      require 'ostruct'
+      require 'mail'
+
+      self.name = "smtp"
+      self.desc = "SMTP notifier"
+      self.retry = 2
+
+      def act
+        data = OpenStruct.new({
+          notifu_id: self.issue.notifu_id,
+          host: self.issue.host,
+          message: self.issue.message,
+          service: self.issue.service,
+          status: self.issue.code.to_state,
+          first_event: Time.at(self.issue.time_created.to_i),
+          duration: (Time.now.to_i - self.issue.time_created.to_i).duration,
+          occurrences_count: self.issue.occurrences_count,
+          occurrences_trigger: self.issue.occurrences_trigger
+        })
+        contacts = self.contacts.map { |contact| "#{contact.full_name} <#{contact.mail}>"}
+        text_message = ERB.new(self.text_template).result(data.instance_eval {binding})
+        html_message = ERB.new(self.html_template).result(data.instance_eval {binding})
+        mail = Mail.new do
+          from Notifu::CONFIG[:actors][:smtp][:from]
+          subject "#{data[:status]}/#{data[:host]}/#{data[:service]}"
+          to contacts
+          text_part do
+            body text_message
+          end
+          html_part do
+            content_type 'text/html; charset=UTF-8'
+            body html_message
+          end
+        end
+        mail.delivery_method :sendmail
+        mail.deliver
+      end
+
+      def text_template
+        %{
+<%= data[:message] %>
+
+Notifu ID: <%= data[:notifu_id] %>
+Host: <%= data[:host] %>
+Service: <%= data[:service] %>
+Status: <%= data[:status] %>
+First event: <%= Time.at(data[:first_event]).to_s %>
+Duration: <%= data[:duration] %>
+Occurences: <%= data[:occurrences_count] %>/<%= data[:occurrences_trigger] %> (occured/trigger)
+}
+      end
+
+      def html_template
+        %{
+<h3><%= data[:message] %></h3><br/>
+
+<strong>Notifu ID: </strong><%= data[:notifu_id] %><br/>
+<strong>Host: </strong><%= data[:host] %><br/>
+<strong>Service: </strong><%= data[:service] %><br/>
+<strong>Status: </strong><%= data[:status] %><br/>
+<strong>First event: </strong><%= Time.at(data[:first_event]).to_s %><br/>
+<strong>Duration: </strong><%= data[:duration] %><br/>
+<strong>Occurences: </strong><%= data[:occurrences_count] %>/<%= data[:occurrences_trigger] %> (occured/trigger)<br/>
+}
+      end
+
+    end
+  end
+end

data/lib/notifu/actors/stdout.rb

@@ -0,0 +1,16 @@
+module Notifu
+  module Actors
+    class Stdout < Notifu::Actor
+
+      self.name = "stdout"
+      self.desc = "STDOUT notifier, useful for debug only"
+      self.retry = 0
+
+      def act
+        puts self.issue.to_yaml
+        puts self.contacts.to_yaml
+      end
+
+    end
+  end
+end

data/lib/notifu/actors/twilio_call.rb

@@ -0,0 +1,27 @@
+module Notifu
+  module Actors
+    class TwilioCall < Notifu::Actor
+
+      require 'excon'
+      require 'erb'
+
+      self.name = "twilio_call"
+      self.desc = "POST requst to trigger phone-call"
+      self.retry = 2
+
+      def act
+        contacts = self.contacts.map { |contact| contact.cell }
+        req_string = Notifu::CONFIG[:actors][:twilio_call][:api] +
+                      "?token="       + Notifu::CONFIG[:actors][:twilio_call][:token] +
+                      "&status="      + self.issue.code.to_state +
+                      "&hostname="    + self.issue.host +
+                      "&service="     + self.issue.service +
+                      "&description=" + ERB::Util.url_encode(self.issue.message.to_s) +
+                      "&call_group="  + ERB::Util.url_encode(contacts.to_json) +
+                      "&init=1"
+        Excon.get req_string if self.issue.code.to_i == 2
+      end
+
+    end
+  end
+end

data/lib/notifu/cli.rb

@@ -0,0 +1,13 @@
+module Notifu
+  module CLI
+    class Root < Thor
+      package_name "Notifu"
+
+      desc "object SUBCOMMAND", "Runtime configuration object management (SLAs, contacts & groups)"
+      subcommand "object", Object
+
+      desc "service SUBCOMMAND", "Notifu service processes"
+      subcommand "service", Service
+    end
+  end
+end

data/lib/notifu/cli/object.rb

@@ -0,0 +1,37 @@
+module Notifu
+  module CLI
+    class Object < Thor
+      package_name "Notifu object configuration"
+      ##
+      # DB Sync task
+      #
+      desc "sync", "Syncs locally defined config objects with DB"
+      def sync
+        puts "Syncing data with Redis..."
+        Notifu::Config.new.ohm_init
+        puts "...done"
+      end
+
+      ##
+      # Configure SLA objects
+      #
+      desc "sla ACTION", "Manage SLAs"
+      def sla(name)
+      end
+
+      ##
+      # Configure contact objects
+      #
+      desc "contact ACTION", "Manage contacts"
+      def contact(name)
+      end
+
+      ##
+      # Configure group objects
+      #
+      desc "group ACTION", "Manage groups"
+      def group(name)
+      end
+    end
+  end
+end

data/lib/notifu/cli/service.rb

@@ -0,0 +1,53 @@
+module Notifu
+  module CLI
+    class Service < Thor
+      package_name "Notifu service"
+      ##
+      # API Service
+      #
+      desc "api", "Starts Notifu API"
+      def api
+        # Notifu::API.start
+      end
+
+      ##
+      # Processor Service
+      #
+      desc "processor", "Starts Notifu processor"
+      option :concurrency, :type => :numeric, :default => 2, :aliases => :c
+      def processor
+        Process.setproctitle "notifu-processor"
+        puts "Starting #{options[:concurrency].to_s} processor(s)"
+        puts( $sidekiq_bin + " -c " + options[:concurrency].to_s + " -r " + $basepath + "workers/processor.rb -q processor" )
+        system( $sidekiq_bin + " -c " + options[:concurrency].to_s + " -r " + $basepath + "workers/processor.rb -q processor" )
+      end
+
+      ##
+      # Actor Service
+      #
+      desc "actor", "Starts Notifu actor"
+      option :actor, :type => :string, :default => nil, :aliases => :a
+      option :concurrency, :type => :numeric, :default => 1, :aliases => :c
+      def actor
+        if ! options[:actor]
+          puts "No actor name specified (-a <actor_name>)! Available actors:"
+          Dir[$actorpath + "*.rb"].each do |name|
+            puts name.gsub(/.*\/([a-zA-Z0-9_]+)\.rb/, "  \\1")
+          end
+        else
+          if File.exists?($actorpath + options[:actor] + ".rb") then
+            Process.setproctitle "notifu-actor [#{options[:actor]}]"
+            puts "Starting #{options[:concurrency].to_s} '#{options[:actor]}' actor(s)"
+            puts( $sidekiq_bin + " exec sidekiq -c " + options[:concurrency].to_s + " -r " + $basepath + "workers/actor.rb -q actor-" + options[:actor])
+            system( $sidekiq_bin + " exec sidekiq -c " + options[:concurrency].to_s + " -r " + $basepath + "workers/actor.rb -q actor-" + options[:actor])
+            exit 0
+          else
+            STDERR.puts "Actor '#{options[:actor]}' does not exist"
+            exit 1
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/notifu/config.rb

@@ -0,0 +1,120 @@
+module Notifu
+  class Config
+
+    attr_reader :config
+
+    @@config_path = "/etc/notifu/"
+
+    def initialize
+      begin
+        @config = YAML.load_file(@@config_path + 'notifu.yaml')
+      rescue
+        raise "Failed to load main config file!"
+      end
+    end
+
+    def get
+      @get ||= self.config.deep_symbolize_keys
+    end
+
+    def ohm_init
+      contacts_init
+      slas_init
+      groups_init
+    end
+
+    def contacts_init
+      Dir[@@config_path + 'contacts/*.yaml'].each do |path|
+        cfg = YAML.load_file(path).deep_symbolize_keys
+        begin
+            Notifu::Model::Contact.with(:name, cfg[:name]).update(cfg)
+            puts "Updated contact '#{cfg[:name]}'."
+        rescue
+            Notifu::Model::Contact.create(cfg).save
+            puts "Created contact '#{cfg[:name]}'."
+        end
+      end
+    end
+
+    def slas_init
+      Dir[@@config_path + 'slas/*.yaml'].each do |path|
+        cfg = YAML.load_file(path).deep_symbolize_keys
+        begin
+            Notifu::Model::Sla.with(:name, cfg[:name]).update(cfg)
+            puts "Updated SLA '#{cfg[:name]}'."
+        rescue
+            Notifu::Model::Sla.create(cfg).save
+            puts "Created SLA '#{cfg[:name]}'."
+        end
+      end
+    end
+
+    def groups_init
+      Dir[@@config_path + 'groups/*.yaml'].each do |path|
+        cfg = YAML.load_file(path).deep_symbolize_keys
+        begin
+          group = Notifu::Model::Group.create(name: cfg[:name])
+          puts "Created group '#{cfg[:name]}'."
+        rescue
+          group = Notifu::Model::Group.with(:name, cfg[:name])
+          puts "Found group '#{cfg[:name]}'."
+        end
+
+        contacts = Array.new
+
+        cfg[:primary].each do |contact_id|
+          begin
+            contacts << Notifu::Model::Contact.with(:name, contact_id)
+            puts "Contact '#{contact_id}' accepted as primary."
+          rescue
+            puts "Failed to load primary contact '#{contact_id}'."
+          end
+        end
+
+        group.primary.replace(contacts)
+        group.save
+        puts "Primary contacts for group '#{cfg[:name]}' updated."
+
+        if cfg[:secondary].is_a? Array
+          contacts = Array.new
+
+          cfg[:secondary].each do |contact_id|
+            begin
+              contacts << Notifu::Model::Contact.with(:name, contact_id)
+              puts "Contact '#{contact_id}' accepted as secondary."
+            rescue
+              puts "Failed to load secondary contact '#{contact_id}'."
+              exit 1
+            end
+          end
+
+          group.secondary.replace(contacts)
+          group.save
+          puts "Secondary contacts for group '#{cfg[:name]}' updated."
+
+          if cfg[:tertiary].is_a? Array
+            contacts = Array.new
+
+            cfg[:tertiary].each do |contact_id|
+              begin
+                contacts << Notifu::Model::Contact.with(:name, contact_id)
+                puts "Contact '#{contact_id}' accepted as tertiary."
+              rescue
+                puts "Failed to load tertiary contact '#{contact_id}'."
+                exit 1
+              end
+            end
+
+            group.tertiary.replace(contacts)
+            group.save
+            puts "Tertiary contacts for group '#{cfg[:name]}' updated."
+          else
+            group.tertiary.replace([])
+          end
+        else
+          group.secondary.replace([])
+        end
+      end
+    end
+  end
+end