zabbirc 0.0.1

data/lib/zabbirc/services/events.rb

@@ -0,0 +1,31 @@
+module Zabbirc
+  module Services
+    class Events < Base
+      def iterate
+        synchronize do
+          recent_events = Zabbix::Event.recent
+          recent_events = filter_out_repeated_events(recent_events)
+
+          recent_events.each do |event|
+            @service.ops.interested_in(event).notify event
+          end
+        end
+      end
+
+      private
+
+      def filter_out_repeated_events events
+        triggers = events.group_by{|e| e.related_object.id }
+        triggers.collect do |_id, events|
+          events.sort_by{|e| e.created_at }.last
+        end
+      end
+
+      def send_notifications op, events
+        events.each do |event|
+          op.notify event
+        end
+      end
+    end
+  end
+end

data/lib/zabbirc/services/ops.rb

@@ -0,0 +1,38 @@
+module Zabbirc
+  module Services
+    class Ops < Base
+      def iterate
+        synchronize do
+          @cinch_bot.channels.each do |channel|
+            sync_ops channel
+          end
+        end
+      end
+
+      private
+
+      def channel_nicks channel
+        channel.users.keys.collect(&:nick)
+      end
+
+      def channel_find_user channel, nick
+        channel.users.keys.find { |irc_user| irc_user.nick == nick }
+      end
+
+      def sync_ops channel
+        nicks = channel_nicks channel
+        zabbix_users = Zabbix::User.get filter: { alias: nicks }
+        zabbix_users.each do |zabbix_user|
+          irc_user = channel_find_user channel, zabbix_user.alias
+          op = @service.ops.add(Op.new(zabbix_user: zabbix_user, irc_user: irc_user))
+          op.add_channel channel
+        end
+
+        @service.ops.each do |op|
+          op.remove_channel channel unless nicks.include? op.nick
+        end
+        true
+      end
+    end
+  end
+end

data/lib/zabbirc/setting.rb

@@ -0,0 +1,31 @@
+module Zabbirc
+  class Setting
+    DEFAULTS = {
+        notify: true,
+        primary_channel: nil,
+        events_priority: :information
+    }
+
+    def initialize
+      @options = ActiveSupport::HashWithIndifferentAccess.new DEFAULTS.deep_dup
+    end
+
+    def set name, value
+      @options[name] = value
+    end
+
+    def get name
+      @options[name]
+    end
+
+    def fetch name, value
+      @options[name] ||= value
+    end
+
+    def to_s
+      @options.collect do |k, v|
+        "#{k}: #{v}"
+      end.join(", ")
+    end
+  end
+end

data/lib/zabbirc/stop_error.rb

@@ -0,0 +1,3 @@
+# This class is used for stopping service threads
+class StopError < StandardError
+end

data/lib/zabbirc/zabbix/connection.rb

@@ -0,0 +1,26 @@
+require 'zabbix/client'
+
+module Zabbirc
+  module Zabbix
+    class Connection
+      attr_reader :client
+
+      def self.get_connection
+        Thread.current[:zabbix_connection] ||= self.new
+      end
+
+      def self.test_connection
+        self.new
+        true
+      rescue => e
+        Zabbirc.logger.fatal "Could not connect to zabbix: #{e}"
+        false
+      end
+
+      def initialize
+        @client = ::Zabbix::Client.new(Zabbirc.config.zabbix_api_url, debug: false)
+        @client.user.login(user: Zabbirc.config.zabbix_login, password: Zabbirc.config.zabbix_password)
+      end
+    end
+  end
+end

data/lib/zabbirc/zabbix/event.rb

@@ -0,0 +1,94 @@
+module Zabbirc
+  module Zabbix
+    class Event < Resource::Base
+      has_many :hosts
+
+      def self.recent options={}
+        params = {
+            acknowledged: false,
+            time_from: Zabbirc.config.notify_about_events_from_last.ago.utc.to_i,
+            priority_from: 0,
+            selectRelatedObject: :extend,
+            selectHosts: :extend
+        }.merge(options)
+
+        priority_from = Priority.new(params.delete(:priority_from))
+        events = get params
+        events = events.find_all{|e| e.priority >= priority_from }
+        events.sort{|x,y| x.priority <=> y.priority }
+      end
+
+      attr_reader :attrs
+
+      delegate :priority, :priority_code, to: :related_object
+
+      def related_object
+        raise AttributeError, "`source` attribute required" if @attrs[:source].blank?
+        raise AttributeError, "`object` attribute required" if @attrs[:object].blank?
+        @related_object ||= determine_related_object
+      end
+
+      def acknowledge message
+        res = api.event.acknowledge(eventids: id, message: message)
+        res["eventids"].collect(&:to_i).include? id.to_i
+      end
+
+      def acknowledged?
+        acknowledged.to_i == 1
+      end
+
+      def created_at
+        Time.at(clock.to_i)
+      end
+
+      def value
+        case @attrs[:source].to_i
+        when 0
+          case @attrs[:value].to_i
+          when 0
+            :ok
+          when 1
+            :problem
+          end
+        else
+          @attrs[:value]
+        end
+      end
+
+      alias_method :state, :value
+
+      def message
+        desc = related_object.description
+        if desc.include?("{HOST.NAME}")
+          desc.sub("{HOST.NAME}", hosts.collect(&:host).join(', '))
+        else
+          "#{desc} on #{hosts.collect(&:host).join(', ')}"
+        end
+      end
+
+      def label
+        format_label "|%id| %time [%priority-code] %msg - %state"
+      end
+
+      def format_label fmt
+        fmt.gsub("%priority-code", "#{priority.code}").
+            gsub("%priority-num", "#{priority.number}").
+            gsub("%time", "#{created_at.to_formatted_s(:short)}").
+            gsub("%msg", "#{message}").
+            gsub("%id", "#{id}").
+            gsub("%state", "#{state}")
+      end
+
+      private
+
+      def determine_related_object
+        case @attrs[:object].to_i
+        when 0
+          @attrs[:relatedObject] ? Trigger.new(@attrs[:relatedObject]) : Trigger.find(@attrs[:objectid])
+        else
+          raise StandardError, "related object #{@attrs[:object].to_i} not implemented yet"
+        end
+      end
+    end
+  end
+end

data/lib/zabbirc/zabbix/host.rb

@@ -0,0 +1,6 @@
+module Zabbirc
+  module Zabbix
+    class Host < Resource::Base
+    end
+  end
+end

data/lib/zabbirc/zabbix/resource/associations.rb

@@ -0,0 +1,30 @@
+module Zabbirc
+  module Zabbix
+    module Resource
+      module Associations
+        def has_many name
+          define_method name do
+            @associations ||= ActiveSupport::HashWithIndifferentAccess.new
+            @associations[name] ||= begin
+              assoc_class = Zabbix.const_get(name.to_s.singularize.camelize)
+              hash_data = @attrs[name]
+              if hash_data.blank?
+                this = self.class.find id, :"select#{name.to_s.camelize}" => :extend
+                raise StandardError, "zabbix response does not contain #{name}" if this[name].blank?
+                hash_data = this[name]
+              end
+
+              hash_data.collect do |obj|
+                assoc_class.new obj
+              end
+            end
+          end
+        end
+
+        def has_one name
+
+        end
+      end
+    end
+  end
+end

data/lib/zabbirc/zabbix/resource/base.rb

@@ -0,0 +1,52 @@
+require_relative 'finders'
+require_relative 'associations'
+
+module Zabbirc
+  module Zabbix
+
+    IDNotUniqueError = Class.new(StandardError)
+
+    module Resource
+      class Base
+        extend Finders, Associations
+
+        def self.set_model_name model_name
+          @model_name = model_name
+        end
+
+        def self.model_name
+          @model_name ||= name.split(/::/).last.underscore
+        end
+
+        def self.api
+          Connection.get_connection.client
+        end
+
+        def api
+          Connection.get_connection.client
+        end
+
+        def initialize attrs
+          @attrs = ActiveSupport::HashWithIndifferentAccess.new attrs
+          raise ArgumentError, "attribute `#{self.class.model_name}id` not found, probably not an Event" unless @attrs.key? :"#{self.class.model_name}id"
+        end
+
+        def id
+          @attrs["#{self.class.model_name}id"]
+        end
+
+        def [] attr
+          @attrs[attr]
+        end
+
+        def method_missing method, *args, &block
+          if args.length == 0 and not block_given? and @attrs.key? method
+            @attrs[method]
+          else
+            super
+          end
+        end
+      end
+    end
+  end
+end

data/lib/zabbirc/zabbix/resource/finders.rb

@@ -0,0 +1,37 @@
+module Zabbirc
+  module Zabbix
+    module Resource
+      module Finders
+        def find id, *options
+          options = options.extract_options!
+          options = options.reverse_merge({
+                                              :"#{model_name}ids" => id
+                                          })
+          res = api.send(model_name).get options
+          if res.size == 0
+            nil
+          elsif res.size > 1
+            raise IDNotUniqueError, "#{model_name.camelize} ID `#{id}` is not unique"
+          else
+            self.new res.first
+          end
+        rescue Errno::ETIMEDOUT => e
+          Zabbirc.logger.error "Zabbix::Resource#find: #{e}"
+          nil
+        end
+
+        def get *options
+          options = options.extract_options!
+          res = api.send(model_name).get options
+          res.collect do |obj|
+            self.new obj
+          end
+        rescue Errno::ETIMEDOUT => e
+          Zabbirc.logger.error "Zabbix::Resource#get: #{e}"
+          []
+        end
+
+      end
+    end
+  end
+end

data/lib/zabbirc/zabbix/trigger.rb

@@ -0,0 +1,47 @@
+module Zabbirc
+  module Zabbix
+    class Trigger < Resource::Base
+      has_many :hosts
+
+      def priority
+        Priority.new(super.to_i)
+      end
+
+      def value
+        case @attrs[:value].to_i
+        when 0
+          :ok
+        when 1
+          :problem
+        else
+          @attrs[:value]
+        end
+      end
+
+      def message
+        if description.include?("{HOST.NAME}")
+          description.sub("{HOST.NAME}", hosts.collect(&:host).join(', '))
+        else
+          "#{description} on #{hosts.collect(&:host).join(', ')}"
+        end
+      end
+
+      def label
+        format_label "%time [%priority-code] %msg - %value"
+      end
+
+      def changed_at
+        Time.at(lastchange.to_i)
+      end
+
+      def format_label fmt
+        fmt.gsub("%priority-code", "#{priority.code}").
+            gsub("%priority-num", "#{priority.number}").
+            gsub("%time", "#{changed_at.to_formatted_s(:short)}").
+            gsub("%msg", "#{message}").
+            gsub("%id", "#{id}").
+            gsub("%value", "#{value}")
+      end
+    end
+  end
+end

data/lib/zabbirc/zabbix/user.rb

@@ -0,0 +1,10 @@
+module Zabbirc
+  module Zabbix
+    class User < Resource::Base
+      def self.find_by_alias _alias, options={}
+        options = options.merge filter: { alias: _alias  }
+        get(options).first
+      end
+    end
+  end
+end

data/lib/zabbirc.rb

@@ -0,0 +1,24 @@
+require 'active_support/all'
+require 'singleton'
+require 'dotenv'
+Dotenv.load
+
+def require_dir dir
+  base_dir = Pathname.new(File.expand_path(File.dirname(__FILE__)))
+  Dir.glob(base_dir.join(dir)).each do |f|
+    require f
+  end
+end
+
+module Zabbirc
+  def self.synchronize &block
+    @mutex ||= Mutex.new
+    @mutex.synchronize &block
+  end
+end
+
+require_dir "zabbirc/*.rb"
+require_dir "zabbirc/irc/*.rb"
+require 'zabbirc/zabbix/resource/base'
+require_dir "zabbirc/zabbix/*.rb"
+require_dir "zabbirc/services/*.rb"

data/spec/bot_spec.rb

@@ -0,0 +1,186 @@
+describe Zabbirc::Irc::PluginMethods do
+  # let(:service) { Zabbirc::ServiceMock.new }
+  let(:mock_message) { double("Cinch::Message", user: mock_user) }
+  let(:mock_user) { double("Cinch::User", nick: mock_nick) }
+  let(:bot) { Zabbirc::MockBot.new }
+  let(:mock_nick) { "op1" }
+  let(:mock_user_settings) { nil }
+
+  before do
+    bot.setup_op mock_nick, mock_user_settings
+  end
+
+  describe "#acknowledge_event" do
+    let(:event) { double "Event", id: 1, label: "Event 1 label" }
+    let(:message) { "ack message" }
+    before do
+      allow(event).to receive(:acknowledge).and_return(true)
+      allow(Zabbirc::Zabbix::Event).to receive(:find).and_return(event)
+    end
+
+    it "should acknowledge event" do
+      expect(mock_message).to receive(:reply).with("#{mock_nick}: Event `#{event.label}` acknowledged with message: #{message}")
+      bot.acknowledge_event mock_message, event.id, message
+    end
+  end
+
+  describe "#host_status" do
+    before do
+      allow(Zabbirc::Zabbix::Host).to receive(:get).and_return(hosts)
+    end
+
+    context "reporting" do
+      before do
+        allow(Zabbirc::Zabbix::Trigger).to receive(:get).and_return(problem_triggers)
+      end
+      let(:host) { double "Host", id: 1, name: "Host-1" }
+      let(:hosts) { [host] }
+      let(:problem_trigger) { double "Trigger", priority: Zabbirc::Priority.new(1), label: "problem_trigger", value: 1 }
+      context "problem trigger" do
+        let(:problem_triggers) { [problem_trigger] }
+        let(:expected_msg) do
+          msg = ["#{mock_nick}: Host: #{host.name} - status: #{problem_triggers.size} problems"]
+          problem_triggers.each do |trigger|
+            msg << "#{mock_nick}: status: #{trigger.label}"
+          end
+          msg.join("\n")
+        end
+        it "should report problem" do
+          expect(mock_message).to receive(:reply).with(expected_msg)
+          bot.host_status mock_message, host.name
+        end
+      end
+
+      context "ok trigger" do
+        let(:problem_triggers) { [] }
+        let(:expected_msg) { "#{mock_nick}: Host: #{host.name} - status: OK" }
+        it "should report problem" do
+          expect(mock_message).to receive(:reply).with(expected_msg)
+          bot.host_status mock_message, host.name
+        end
+      end
+    end # context reporting
+
+    context "host identification" do
+      context "no hosts" do
+        let(:hosts) { [] }
+        let(:host_name) { "undefined_host_name" }
+        it "should not found host" do
+          expect(mock_message).to receive(:reply).with("#{mock_nick}: Host not found `#{host_name}`")
+          bot.host_status mock_message, host_name
+        end
+      end
+
+      context "2 - 10 hosts" do
+        let(:host1) { double "Host1", id: 1, name: "host-1" }
+        let(:host2) { double "Host2", id: 2, name: "host-2" }
+        let(:hosts) { [host1, host2] }
+        let(:expected_msg) { "#{mock_nick}: Found #{hosts.size} hosts: #{hosts.collect(&:name).join(', ')}. Be more specific" }
+        it "should print host names" do
+          expect(mock_message).to receive(:reply).with(expected_msg)
+          bot.host_status mock_message, "host"
+        end
+      end
+
+      context "more than 10 hosts" do
+        let(:hosts) { double "HostsArray", size: 11 }
+        let(:expected_msg) { "#{mock_nick}: Found #{hosts.size} Be more specific" }
+        it "should print host names" do
+          expect(mock_message).to receive(:reply).with(expected_msg)
+          bot.host_status mock_message, "host"
+        end
+      end
+    end # context host identification
+  end # context #host_status
+
+  context "#host_latest" do
+    let(:host) { double "Host", id: 1, name: "Host1" }
+    let(:hosts) { [host] }
+    let(:event1) { double "Event", label: "Event 1 label" }
+    let(:events) { [event1] }
+    let(:expected_msg) do
+      msg = ["#{mock_nick}: Host: #{host.name} - showing last #{events.size} events"]
+      events.each do |event|
+        msg << "#{mock_nick}: !latest: #{event.label}"
+      end
+      msg.join("\n")
+    end
+    before do
+      allow(Zabbirc::Zabbix::Host).to receive(:get).and_return(hosts)
+      allow(Zabbirc::Zabbix::Event).to receive(:get).and_return(events)
+    end
+
+    it "should print latest events" do
+      expect(mock_message).to receive(:reply).with(expected_msg)
+      bot.host_latest mock_message, "Host1", nil, nil
+    end
+  end
+
+  describe "#list_events" do
+    before do
+      allow(Zabbirc::Zabbix::Event).to receive(:recent).and_return(recent_events)
+    end
+    context "no last events" do
+      let(:recent_events) { [] }
+
+      it "should report no last events" do
+        expect(mock_message).to receive(:reply).with("#{mock_nick}: No last events")
+        bot.list_events mock_message
+      end
+    end
+
+    context "no last events" do
+      let(:event1) { double "Event1", label: "Event 1 label" }
+      let(:event2) { double "Event2", label: "Event 2 label" }
+      let(:recent_events) { [event1, event2] }
+      let(:expected_msg) { recent_events.collect{|e| "#{mock_nick}: #{e.label}"}.join("\n") }
+
+      it "should report no last events" do
+        expect(mock_message).to receive(:reply).with(expected_msg)
+        bot.list_events mock_message
+      end
+    end
+  end
+
+  describe "#show_settings" do
+    let(:mock_user_settings) { {primary_channel: "#channel-1", events_priority: "high", notify: false } }
+    let(:expected_msg) { "#{mock_nick}: notify: false, primary_channel: #channel-1, events_priority: high" }
+    it "should show settings" do
+      expect(mock_message).to receive(:reply).with(expected_msg)
+      bot.show_settings mock_message
+    end
+  end
+
+  describe "#set_setting" do
+    shared_examples "set_setting" do |key, value, expected_setting_value|
+      let(:expected_msg) { "#{mock_nick}: setting `#{key}` was set to `#{expected_setting_value}`" }
+      let(:op) { bot.get_op mock_nick }
+      it "should set #{key} setting to #{value}" do
+        expect(mock_message).to receive(:reply).with(expected_msg)
+        bot.set_setting mock_message, key, nil, value
+        expect(op.setting.get(key)).to eq expected_setting_value
+      end
+    end
+
+
+    context "notify" do
+      it_should_behave_like "set_setting", "notify", "false", false
+      it_should_behave_like "set_setting", "notify", "true", true
+    end
+
+    context "events_priority" do
+      it_should_behave_like "set_setting", "events_priority", "high", :high
+      it_should_behave_like "set_setting", "events_priority", "5", :disaster
+    end
+
+    context "primary_channel" do
+      before do
+        op.add_channel double("#channel1", name: "#channel1")
+        op.add_channel double("#channel2double", name: "#channel2")
+      end
+      it_should_behave_like "set_setting", "primary_channel", "#channel1", "#channel1"
+      it_should_behave_like "set_setting", "primary_channel", "#channel2", "#channel2"
+    end
+
+  end
+end