icinga2 0.7.0.1 → 0.8.1.2

data/lib/icinga2/network.rb

@@ -17,90 +17,73 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.get( params = {} )
+    def self.api_data( params )
+
+      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new('missing params') if( params.size.zero? )
 
-      host    = params.dig(:host)
       url     = params.dig(:url)
       headers = params.dig(:headers)
       options = params.dig(:options)
       payload = params.dig(:payload) || {}
-      result  = {}
-      max_retries   = 30
-      times_retried = 0
-
-      return get_with_payload( params ) if  payload.count >= 1
 
-      headers.delete( 'X-HTTP-Method-Override' )
+      raise ArgumentError.new('Missing url') if( url.nil? )
+      raise ArgumentError.new('Missing headers') if( headers.nil? )
+      raise ArgumentError.new('Missing options') if( options.nil? )
+      raise ArgumentError.new('only Hash for payload are allowed') unless( payload.is_a?(Hash) )
 
-      result  = {}
+      rest_client = RestClient::Resource.new( URI.encode( url ), options )
 
-      rest_client = RestClient::Resource.new(
-        URI.encode( url ),
-        options
-      )
+      max_retries = 10
+      retried     = 0
 
       begin
-
-        data     = rest_client.get( headers )
-        results  = JSON.parse( data.body )
-        results  = results.dig('results')
-
-        result[:status] = 200
-        result[:data] ||={}
-
-        results.each do |r|
-          attrs = r.dig('attrs')
-          if( !attrs.nil? )
-            result[:data][attrs['name']] = {
-              name: attrs['name'],
-              display_name: attrs['display_name'],
-              type: attrs['type']
-            }
-          else
-            result = r
-          end
-
+        if payload
+          headers['X-HTTP-Method-Override'] = 'GET'
+          payload = JSON.generate(payload)
+          res = rest_client.post(payload, headers)
+        else
+          res = rest_client.get(headers)
         end
+
       rescue RestClient::Unauthorized => e
 
-        result = {
+        return {
           status: 401,
-          name: host,
           message: 'unauthorized'
         }
 
-      rescue RestClient::ExceptionWithResponse => e
+      rescue RestClient::NotFound => e
 
-        error  = e.response ? e.response : nil
-        error  = JSON.parse( error )
+        message = format( 'not found (request %s)', url )
+#         $stderr.puts( message )
 
-        result = {
-          status: error['error'].to_i,
-          name: host,
-          message: error['status']
+        return {
+          status: 404,
+          message: message
         }
-      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
-
-        if( times_retried < max_retries )
 
-          times_retried += 1
-          $stderr.puts(format( 'Cannot execute request %s', url ))
-          $stderr.puts(format( '   cause: %s', e ))
-          $stderr.puts(format( '   retry %d / %d', times_retried, max_retries ))
+      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
 
-          sleep( 4 )
+        if( retried < max_retries )
+          retried += 1
+          $stderr.puts(format("Cannot execute request against '%s': '%s' (retry %d / %d)", url, e, retried, max_retries))
+          sleep(2)
           retry
         else
-          $stderr.puts(format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url ))
+
+          message = format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
+#           $stderr.puts( message )
 
           return {
             status: 500,
-            message: format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
+            message: message
           }
         end
       end
 
-      result
+      body = res.body
+      JSON.parse(body)
     end
 
     # static function for GET Requests without filters
@@ -115,168 +98,26 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.get_raw( params = {} )
-
-      host    = params.dig(:host)
-      url     = params.dig(:url)
-      headers = params.dig(:headers)
-      options = params.dig(:options)
-      payload = params.dig(:payload) || {}
-      result  = {}
-      max_retries   = 30
-      times_retried = 0
-
-      return get_with_payload( params ) if  payload.count >= 1
-
-      headers.delete( 'X-HTTP-Method-Override' )
-
-      result  = {}
-
-      rest_client = RestClient::Resource.new(
-        URI.encode( url ),
-        options
-      )
-
-      begin
-
-        data     = rest_client.get( headers )
-        results  = JSON.parse( data.body )
-        results  = results.dig('results')
-
-        result[:status] = 200
-        result[:data] ||={}
-
-        results.each do |r|
-
-          r.reject! { |x| x == 'perfdata' }
-          result[:data][r.dig('name')] = r.dig('status')
-        end
-
-      rescue RestClient::ExceptionWithResponse => e
+    def self.application_data( params )
 
-        error  = e.response ? e.response : nil
-        error  = JSON.parse( error )
+      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new('missing params') if( params.size.zero? )
 
-        result = {
-          status: error['error'].to_i,
-          name: host,
-          message: error['status']
-        }
-      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
-
-        if( times_retried < max_retries )
-
-          times_retried += 1
-          $stderr.puts(format( 'Cannot execute request %s', url ))
-          $stderr.puts(format( '   cause: %s', e ))
-          $stderr.puts(format( '   retry %d / %d', times_retried, max_retries ))
-
-          sleep( 4 )
-          retry
-        else
-          $stderr.puts(format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url ))
-
-          return {
-            status: 500,
-            message: format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
-          }
-        end
-      end
-
-      result
-    end
-
-
-    # static function for GET Requests with payload
-    # internal we use here an POST Request and overwrite a http header
-    #
-    # @param [Hash] params
-    # @option params [String] :host
-    # @option params [String] :url
-    # @option params [String] :headers
-    # @option params [String] :options
-    # @option params [Hash] :payload
-    #
-    #
-    # @return [Hash]
-    #
-    def self.get_with_payload( params = {} )
-
-      host    = params.dig(:host)
       url     = params.dig(:url)
       headers = params.dig(:headers)
       options = params.dig(:options)
-      payload = params.dig(:payload) || {}
-      result  = {}
-      max_retries   = 30
-      times_retried = 0
-
-      headers['X-HTTP-Method-Override'] = 'GET'
-
-      rest_client = RestClient::Resource.new(
-        URI.encode( url ),
-        options
-      )
-
-      begin
-
-        response = rest_client.post(
-          JSON.generate( payload ),
-          headers
-        )
-
-        response_code = response.code
-        response_body = response.body
-        node         = {}
-
-        data    = JSON.parse( response_body )
-        results = data.dig('results')
-
-        unless( results.nil? )
-
-          results.each do |r|
-            node[r.dig('name')] = r
-          end
-
-          result = {
-            status: response_code,
-            nodes: node
-          }
-        end
-
-      rescue RestClient::ExceptionWithResponse => e
-
-        error  = e.response ? e.response : nil
-        error  = JSON.parse( error )
 
-        result = {
-          status: error['error'].to_i,
-          name: host,
-          message: error['status']
-        }
-      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH  => e
-
-        if( times_retried < max_retries )
-
-          times_retried += 1
-          $stderr.puts(format( 'Cannot execute request %s', url ))
-          $stderr.puts(format( '   cause: %s', e ))
-          $stderr.puts(format( '   retry %d / %d', times_retried, max_retries ))
+      raise ArgumentError.new('Missing url') if( url.nil? )
+      raise ArgumentError.new('Missing headers') if( headers.nil? )
+      raise ArgumentError.new('Missing options') if( options.nil? )
 
-          sleep( 4 )
-          retry
-        else
-          $stderr.puts( 'Exiting request ...' )
+      data    = Network.api_data( url: url, headers: headers, options: options )
 
-          return {
-            status: 500,
-            message: format( 'Errno::ECONNREFUSED for request: %s', url )
-          }
-        end
-      end
+      return nil unless( data.dig(:status).nil? )
 
-      result
+      results = data.dig('results')
 
+      return results.first.dig('status') unless( results.nil? )
     end
 
     # static function for POST Requests
@@ -290,18 +131,26 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.post( params = {} )
+    def self.post( params )
+
+      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new('missing params') if( params.size.zero? )
 
       url     = params.dig(:url)
       headers = params.dig(:headers)
       options = params.dig(:options)
       payload = params.dig(:payload)
-      max_retries   = 30
-      times_retried = 0
 
-      headers['X-HTTP-Method-Override'] = 'POST'
+      raise ArgumentError.new('Missing url') if( url.nil? )
+      raise ArgumentError.new('Missing headers') if( headers.nil? )
+      raise ArgumentError.new('Missing options') if( options.nil? )
+      raise ArgumentError.new('only Hash for payload are allowed') unless( payload.is_a?(Hash) )
 
-      result  = {}
+      max_retries = 30
+      retried     = 0
+      result      = {}
+
+      headers['X-HTTP-Method-Override'] = 'POST'
 
       rest_client = RestClient::Resource.new(
         URI.encode( url ),
@@ -318,15 +167,7 @@ module Icinga2
         data    = JSON.parse( data )
         results = data.dig('results').first
 
-        unless( results.nil? )
-
-          result = {
-            status: results.dig('code').to_i,
-            name: results.dig('name'),
-            message: results.dig('status')
-          }
-
-        end
+        return { status: results.dig('code').to_i, name: results.dig('name'), message: results.dig('status') } unless( results.nil? )
 
       rescue RestClient::ExceptionWithResponse => e
 
@@ -337,16 +178,16 @@ module Icinga2
 
         if( !results.nil? )
 
-          result = result.first
+#           result = result.first
 
-          result = {
+          return {
             status: results.dig('code').to_i,
             name: results.dig('name'),
             message: results.dig('status'),
             error: results.dig('errors')
           }
         else
-          result = {
+          return {
             status: error.dig( 'error' ).to_i,
             message: error.dig( 'status' )
           }
@@ -354,21 +195,19 @@ module Icinga2
 
       rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
 
-        if( times_retried < max_retries )
-
-          times_retried += 1
-          $stderr.puts(format( 'Cannot execute request %s', url ))
-          $stderr.puts(format( '   cause: %s', e ))
-          $stderr.puts(format( '   retry %d / %d', times_retried, max_retries ))
-
-          sleep( 4 )
+        if( retried < max_retries )
+          retried += 1
+          $stderr.puts(format("Cannot execute request against '%s': '%s' (retry %d / %d)", url, e, retried, max_retries))
+          sleep(2)
           retry
         else
-          $stderr.puts( 'Exiting request ...' )
+
+          message = format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
+          $stderr.puts( message )
 
           return {
             status: 500,
-            message: format( 'Errno::ECONNREFUSED for request: %s', url )
+            message: message
           }
         end
       end
@@ -388,19 +227,26 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.put( params = {} )
+    def self.put( params )
+
+      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new('missing params') if( params.size.zero? )
 
-      host    = params.dig(:host)
       url     = params.dig(:url)
       headers = params.dig(:headers)
       options = params.dig(:options)
       payload = params.dig(:payload)
-      max_retries   = 30
-      times_retried = 0
 
-      headers['X-HTTP-Method-Override'] = 'PUT'
+      raise ArgumentError.new('Missing url') if( url.nil? )
+      raise ArgumentError.new('Missing headers') if( headers.nil? )
+      raise ArgumentError.new('Missing options') if( options.nil? )
+      raise ArgumentError.new('only Hash for payload are allowed') unless( payload.is_a?(Hash) )
 
-      result  = {}
+      max_retries = 30
+      retried     = 0
+      result      = {}
+
+      headers['X-HTTP-Method-Override'] = 'PUT'
 
       rest_client = RestClient::Resource.new(
         URI.encode( url ),
@@ -408,81 +254,55 @@ module Icinga2
       )
 
       begin
-
         data = rest_client.put(
           JSON.generate( payload ),
           headers
         )
-
         data    = JSON.parse( data )
         results = data.dig('results').first
 
-        unless( results.nil? )
-
-          result = {
-            status: results.dig('code').to_i,
-            name: results.dig('name'),
-            message: results.dig('status')
-          }
-
-        end
+        return { status: results.dig('code').to_i, name: results.dig('name'), message: results.dig('status') } unless( results.nil? )
 
       rescue RestClient::ExceptionWithResponse => e
 
         error  = e.response ? e.response : nil
-
         error = JSON.parse( error ) if  error.is_a?( String )
 
-        results = error.dig( 'results' )
-
-        if( !results.nil? )
-
-          if( result.is_a?( Hash ) && result.count != 0 )
-
-            result = result.first
-
-            result = {
-              status: results.dig('code').to_i,
-              name: results.dig('name'),
-              message: results.dig('status'),
-              error: results.dig('errors')
-            }
-
-          else
+        results = error.dig('results')
 
-            result = {
-              status: 204,
-              name: host,
-              message: 'unknown result'
-            }
-          end
+        return { status: error.dig('error').to_i, message: error.dig('status'), error: results } if( results.nil? )
 
+        if( results.is_a?( Hash ) && results.count != 0 )
+#          result = result.first
+          return {
+            status: results.dig('code').to_i,
+            name: results.dig('name'),
+            message: results.dig('status'),
+            error: results
+          }
         else
-
-          result = {
-            status: error.dig( 'error' ).to_i,
-            message: error.dig( 'status' )
+          return {
+            status: results.first.dig('code').to_i,
+            message: format('%s (possible, object already exists)', results.first.dig('status') ),
+            error: results
           }
-
         end
 
       rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
 
-        if( times_retried < max_retries )
-
-          times_retried += 1
-          $stderr.puts(format( 'Cannot execute request %s', url ))
-          $stderr.puts(format( '   cause: %s', e ))
-          $stderr.puts(format( '   retry %d / %d', times_retried, max_retries ))
-
-          sleep( 4 )
+        if( retried < max_retries )
+          retried += 1
+          $stderr.puts(format("Cannot execute request against '%s': '%s' (retry %d / %d)", url, e, retried, max_retries))
+          sleep(2)
           retry
         else
-          $stderr.puts( 'Exiting request ...' )
+
+          message = format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
+          $stderr.puts( message )
 
           return {
             status: 500,
-            message: format( 'Errno::ECONNREFUSED for request: %s', url )
+            message: message
           }
         end
       end
@@ -500,13 +320,21 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.delete( params = {} )
+    def self.delete( params )
+
+      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new('missing params') if( params.size.zero? )
 
       url     = params.dig(:url)
       headers = params.dig(:headers)
       options = params.dig(:options)
-      max_retries   = 3
-      times_retried = 0
+
+      raise ArgumentError.new('Missing url') if( url.nil? )
+      raise ArgumentError.new('Missing headers') if( headers.nil? )
+      raise ArgumentError.new('Missing options') if( options.nil? )
+
+      max_retries = 3
+      retried     = 0
 
       headers['X-HTTP-Method-Override'] = 'DELETE'
 
@@ -522,18 +350,10 @@ module Icinga2
 
         if( data )
 
-          data    = JSON.parse( data ) #.body )
+          data    = JSON.parse( data )
           results = data.dig('results').first
 
-          unless( results.nil? )
-
-            result = {
-              status: results.dig('code').to_i,
-              name: results.dig('name'),
-              message: results.dig('status')
-            }
-
-          end
+          return { status: results.dig('code').to_i, name: results.dig('name'), message: results.dig('status') } unless( results.nil? )
         end
 
       rescue RestClient::ExceptionWithResponse => e
@@ -544,40 +364,33 @@ module Icinga2
 
         results = error.dig('results')
 
-        result = if( !results.nil? )
-
-          {
+        if( results.nil? )
+          return {
+            status: error.dig( 'error' ).to_i,
+            message: error.dig( 'status' )
+          }
+        else
+          return {
             status: results.dig('code').to_i,
             name: results.dig('name'),
             message: results.dig('status')
           }
-
-        else
-
-          {
-            status: error.dig( 'error' ).to_i,
-#            :name        => results.dig('name'),
-            message: error.dig( 'status' )
-          }
-
-                 end
+        end
       rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
 
-        if( times_retried < max_retries )
-
-          times_retried += 1
-          $stderr.puts(format( 'Cannot execute request %s', url ))
-          $stderr.puts(format( '   cause: %s', e ))
-          $stderr.puts(format( '   retry %d / %d', times_retried, max_retries ))
-
-          sleep( 4 )
+        if( retried < max_retries )
+          retried += 1
+          $stderr.puts(format("Cannot execute request against '%s': '%s' (retry %d / %d)", url, e, retried, max_retries))
+          sleep(2)
           retry
         else
-          $stderr.puts( 'Exiting request ...' )
+
+          message = format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
+          $stderr.puts( message )
 
           return {
             status: 500,
-            message: format( 'Errno::ECONNREFUSED for request: %s', url )
+            message: message
           }
         end
       end
@@ -587,5 +400,3 @@ module Icinga2
 
   end
 end
-
-