icinga2 0.5.2

data/lib/icinga2/notifications.rb

@@ -0,0 +1,158 @@
+
+module Icinga2
+
+  module Notifications
+
+
+    def enableHostNotification( host )
+
+      return hostNotification( { :name => host, :enable_notifications => true } )
+    end
+
+
+    def disableHostNotification( host )
+
+      return self.hostNotification( { :name => host, :enable_notifications => false } )
+    end
+
+
+    def enableServiceNotification( params = {} )
+
+      host    = params.get(:host)
+      service = params.get(:service)
+
+      if( host == nil )
+
+        return {
+          :status  => 404,
+          :message => 'missing host name'
+        }
+      end
+
+      return self.serviceNotification( { :name => host, :service => service, :enable_notifications => true } )
+    end
+
+
+    def disableServiceNotification( host )
+
+      return self.serviceNotification( { :name => host, :enable_notifications => false } )
+    end
+
+
+    def enableHostgroupNotification( group )
+
+      return self.hostgroupNotification( { :host_group => group, :enable_notifications => true } )
+    end
+
+
+    def disableHostgroupNotification( group )
+
+      return self.hostgroupNotification( { :host_group => group, :enable_notifications => false } )
+    end
+
+
+    def listNotifications( params = {} )
+
+      name = params.dig(:name)
+
+      result = Network.get( {
+        :host     => name,
+        :url      => sprintf( '%s/v1/objects/notifications/%s', @icingaApiUrlBase, name ),
+        :headers  => @headers,
+        :options  => @options
+      } )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+    # PRIVATE SECTION
+    #
+    def hostNotification( params = {} )
+
+      name          = params.dig(:name)
+      notifications = params.dig(:enable_notifications) || false
+
+      payload = {
+        "attrs"     => {
+          "enable_notifications" => notifications
+        }
+      }
+
+      result = Network.post( {
+        :host    => name,
+        :url     => sprintf( '%s/v1/objects/hosts/%s', @icingaApiUrlBase, name ),
+        :headers => @headers,
+        :options => @options,
+        :payload => payload
+      } )
+
+      logger.debug( result.class.to_s )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+    def hostgroupNotification( params = {} )
+
+      group         = params.dig(:host_group)
+      notifications = params.dig(:enable_notifications) || false
+
+      payload = {
+        "filter"    => sprintf( '"%s" in host.groups', group ),
+        "attrs"     => {
+          "enable_notifications" => notifications
+        }
+      }
+
+      result = Network.post( {
+        :host    => name,
+        :url     => sprintf( '%s/v1/objects/services', @icingaApiUrlBase ),
+        :headers => @headers,
+        :options => @options,
+        :payload => payload
+      } )
+
+      logger.debug( result.class.to_s )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+    def serviceNotification( params = {} )
+
+      name          = params.dig(:name)
+      service       = params.dig(:service)
+      notifications = params.dig(:enable_notifications) || false
+
+      payload = {
+        "filter"    => sprintf( 'host.name=="%s"', name ),
+        "attrs"     => {
+          "enable_notifications" => notifications
+        }
+      }
+
+      logger.debug( payload )
+      logger.debug( sprintf( '%s/v1/objects/services', @icingaApiUrlBase ) )
+
+      result = Network.post( {
+        :host    => name,
+        :url     => sprintf( '%s/v1/objects/services', @icingaApiUrlBase ),
+        :headers => @headers,
+        :options => @options,
+        :payload => payload
+      } )
+
+      logger.debug( result.class.to_s )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+  end
+
+end

data/lib/icinga2/servicegroups.rb

@@ -0,0 +1,101 @@
+
+module Icinga2
+
+  module Servicegroups
+
+
+    def addServicegroup( params = {} )
+
+      name        = params.dig(:name)
+      displayName = params.dig(:display_name)
+
+      if( name == nil )
+
+        return {
+          :status  => 404,
+          :message => 'missing servicegroup name'
+        }
+      end
+
+      payload = {
+        "attrs" => {
+          "display_name"         => displayName
+        }
+      }
+
+      result = Network.put( {
+        :host    => name,
+        :url     => sprintf( '%s/v1/objects/servicegroups/%s', @icingaApiUrlBase, name ),
+        :headers => @headers,
+        :options => @options,
+        :payload => payload
+      } )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+    def deleteServicegroup( params = {} )
+
+      name = params.dig(:name)
+
+      if( name == nil )
+
+        return {
+          :status  => 404,
+          :message => 'missing servicegroup name'
+        }
+      end
+
+      result = Network.delete( {
+        :host    => name,
+        :url     => sprintf( '%s/v1/objects/servicegroups/%s?cascade=1', @icingaApiUrlBase, name ),
+        :headers => @headers,
+        :options => @options
+      } )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+    def listServicegroups( params = {} )
+
+      name = params.dig(:name)
+
+      result = Network.get( {
+        :host     => name,
+        :url      => sprintf( '%s/v1/objects/servicegroups/%s', @icingaApiUrlBase, name ),
+        :headers  => @headers,
+        :options  => @options
+      } )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+    def existsServicegroup?( name )
+
+
+      result = self.listServicegroups( { :name => name } )
+
+      if( result.is_a?( String ) )
+        result = JSON.parse( result )
+      end
+
+      status = result.dig('status')
+
+      if( status != nil && status == 200 )
+        return true
+      end
+
+      return false
+
+    end
+
+
+  end
+
+end

data/lib/icinga2/services.rb

@@ -0,0 +1,301 @@
+
+module Icinga2
+
+  module Services
+
+    def addServices( host, services = {} )
+
+      def updateHost( hash, host )
+
+        hash.each do |k, v|
+
+          if( k == "host" && v.is_a?( String ) )
+            v.replace( host )
+
+          elsif( v.is_a?( Hash ) )
+            self.updateHost( v, host )
+
+          elsif( v.is_a?(Array) )
+
+            v.flatten.each { |x| self.updateHost( x, host ) if x.is_a?( Hash ) }
+          end
+        end
+
+        hash
+      end
+
+      services.each do |s,v|
+
+        payload = {
+          "templates" => [ "generic-service" ],
+          "attrs"     => updateHost( v, host )
+        }
+
+        logger.debug( s )
+        logger.debug( v.to_json )
+
+        logger.debug( JSON.pretty_generate( payload ) )
+
+        result = Network.put( {
+          :host    => host,
+          :url     => sprintf( '%s/v1/objects/services/%s!%s', @icingaApiUrlBase, host, s ),
+          :headers => @headers,
+          :options => @options,
+          :payload => payload
+        })
+
+        logger.debug( result )
+
+      end
+
+    end
+
+
+    def unhandledServices( params = {} )
+
+      # taken from https://blog.netways.de/2016/11/18/icinga-2-api-cheat-sheet/
+      # 5) Anzeige aller Services die unhandled sind und weder in Downtime, noch acknowledged sind
+      # /usr/bin/curl -k -s -u 'root:icinga' -H 'X-HTTP-Method-Override: GET' -X POST 'https://127.0.0.1:5665/v1/objects/services' -d '{ "attrs": [ "__name", "state", "downtime_depth", "acknowledgement" ], "filter": "service.state != ServiceOK && service.downtime_depth == 0.0 && service.acknowledgement == 0.0" }''' | jq
+
+    end
+
+
+    def listServices( params = {} )
+
+      name    = params.dig(:host)
+      service = params.dig(:service)
+
+      if( service == nil )
+        url = sprintf( '%s/v1/objects/services/%s', @icingaApiUrlBase, name )
+      else
+        url = sprintf( '%s/v1/objects/services/%s!%s', @icingaApiUrlBase, name, service )
+      end
+
+      result = Network.get( {
+        :host     => name,
+        :url      => url,
+        :headers  => @headers,
+        :options  => @options
+      } )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+    def existsService?( params = {} )
+
+      host    = params.dig(:host)
+      service = params.dig(:service)
+
+      if( host == nil )
+
+        return {
+          :status  => 404,
+          :message => 'missing host name'
+        }
+      end
+
+      result = self.listServices( { :host => host, :service => service } )
+
+      if( result.is_a?( String ) )
+        result = JSON.parse( result )
+      end
+
+      status = result.dig('status')
+
+      if( status != nil && status == 200 )
+        return true
+      end
+
+      return false
+
+    end
+
+
+    def serviceObjects( params = {} )
+
+      attrs   = params.dig(:attrs)
+      filter  = params.dig(:filter)
+      joins   = params.dig(:joins)
+      payload = {}
+
+      if( attrs == nil )
+        attrs = ['name','state','acknowledgement','downtime_depth','last_check']
+      end
+
+      if( joins == nil )
+        joins = ['host.name','host.state','host.acknowledgement','host.downtime_depth','host.last_check']
+      end
+
+      if( attrs != nil )
+        payload['attrs'] = attrs
+      end
+
+      if( filter != nil )
+        payload['filter'] = filter
+      end
+
+      if( joins != nil )
+        payload['joins'] = joins
+      end
+
+      result = Network.get( {
+        :host     => nil,
+        :url      => sprintf( '%s/v1/objects/services', @icingaApiUrlBase ),
+        :headers  => @headers,
+        :options  => @options,
+        :payload  => payload
+      } )
+
+      return JSON.pretty_generate( result )
+
+    end
+
+
+    def serviceProblems()
+
+      data     = self.serviceObjects()
+      problems = 0
+
+      if( data.is_a?(String) )
+        data = JSON.parse(data)
+      end
+
+      nodes = data.dig('nodes')
+
+      nodes.each do |n|
+
+        attrs           = n.last.dig('attrs')
+
+        state           = attrs.dig('state')
+        downtimeDepth   = attrs.dig('downtime_depth')
+        acknowledgement = attrs.dig('acknowledgement')
+
+#         puts state
+
+        if( state != 0 && downtimeDepth == 0 && acknowledgement == 0 )
+          problems += 1 #= problems +1
+        end
+
+      end
+
+      return problems
+
+    end
+
+
+    def problemServices( max_items = 5 )
+
+      count = 0
+      @serviceProblems = {}
+      @serviceProblemsSeverity = {}
+
+      # only fetch the minimal attribute set required for severity calculation
+      servicesData = self.serviceObjects()
+
+      if( servicesData.is_a?(String) )
+
+        servicesData = JSON.parse( servicesData )
+      end
+
+      servicesData = servicesData.dig('nodes')
+
+      servicesData.each do |service,v|
+
+        name  = v.dig('name')
+        state = v.dig('attrs','state')
+#         logger.debug( "Severity for #{name}" )
+        if( state == 0 )
+          next
+        end
+
+        @serviceProblems[name] = self.serviceSeverity(v)
+      end
+
+      @serviceProblems.sort_by {|v| v}.reverse!
+
+      @serviceProblems.keys[1..max_items].each { |k,v| @serviceProblemsSeverity[k] = @serviceProblems[k] }
+
+#       @serviceProblems.each do |k,v|
+#
+#         if( count >= max_items )
+#           break
+#         end
+#
+#         @serviceProblemsSeverity[k] = v
+#
+#         count += 1
+#       end
+
+      return @serviceProblemsSeverity
+    end
+
+
+    # private
+    # stolen from Icinga Web 2
+    # ./modules/monitoring/library/Monitoring/Backend/Ido/Query/ServicestatusQuery.php
+    #
+    def serviceSeverity( service )
+
+      attrs = service.dig('attrs')
+      state           = attrs.dig('state')
+      acknowledgement = attrs.dig('acknowledgement')
+      downtimeDepth   = attrs.dig('downtime_depth')
+
+#       logger.debug( attrs )
+
+      severity = 0
+
+      if( state == 0 )
+
+        if( self.getObjectHasBeenChecked( service ) )
+          severity += 16
+        end
+
+        if( acknowledgement != 0 )
+          severity += 2
+        elsif( downtimeDepth > 0 )
+          severity += 1
+        else
+          severity += 4
+        end
+      else
+        if( self.getObjectHasBeenChecked( service ) )
+          severity += 16
+        elsif( state == 1 )
+          severity += 32
+        elsif( state == 2 )
+          severity += 128
+        elsif( state == 3 )
+          severity += 64
+        else
+          severity += 256
+        end
+
+        # requires joins
+        host_attrs = service.dig('joins','host')
+
+        host_state = host_attrs.dig('state')
+        host_acknowledgement = host_attrs.dig('acknowledgement')
+        host_downtimeDepth   = host_attrs.dig('downtime_depth')
+
+        if( host_state > 0 )
+          severity += 1024
+        elsif( host_acknowledgement )
+          severity += 512
+        elsif( host_downtimeDepth > 0 )
+          severity += 256
+        else
+          severity += 2048
+        end
+      end
+
+      return severity
+    end
+
+
+
+  end
+
+end