icinga2 0.9.0.1 → 0.9.2.1

data/lib/icinga2/network.rb

@@ -17,73 +17,47 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.api_data( params )
+    def api_data( params )
 
-      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new(format('wrong type. \'params\' must be an Hash, given \'%s\'', params.class.to_s)) 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) || {}
+      payload = params.dig(:payload)
 
       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) )
 
       rest_client = RestClient::Resource.new( URI.encode( url ), options )
 
-      max_retries = 10
-      retried     = 0
+      if( payload )
+        raise ArgumentError.new('only Hash for payload are allowed') unless( payload.is_a?(Hash) )
+        headers['X-HTTP-Method-Override'] = 'GET'
+        method = 'POST'
+      else
+        headers['X-HTTP-Method-Override'] = 'GET'
+        method = 'GET'
+      end
 
       begin
-        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
+        data = request( rest_client, method, headers, payload )
 
-        return {
-          status: 401,
-          message: 'unauthorized'
-        }
+        data = JSON.parse( data ) if( data.is_a?(String) )
+        data = data.deep_string_keys
+        data = data.dig('results') if( data.is_a?(Hash) )
 
-      rescue RestClient::NotFound => e
+        return data
 
-        message = format( 'not found (request %s)', url )
-#         $stderr.puts( message )
+      rescue => e
 
-        return {
-          status: 404,
-          message: message
-        }
+        logger.error(e)
+        logger.error(e.backtrace.join("\n"))
 
-      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
-
-        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
-
-          message = format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
-#           $stderr.puts( message )
-
-          return {
-            status: 500,
-            message: message
-          }
-        end
+        return nil
       end
-
-      body = res.body
-      JSON.parse(body)
     end
 
     # static function for GET Requests without filters
@@ -98,9 +72,9 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.application_data( params )
+    def icinga_application_data( params )
 
-      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new(format('wrong type. \'params\' must be an Hash, given \'%s\'', params.class.to_s)) unless( params.is_a?(Hash) )
       raise ArgumentError.new('missing params') if( params.size.zero? )
 
       url     = params.dig(:url)
@@ -111,13 +85,23 @@ module Icinga2
       raise ArgumentError.new('Missing headers') if( headers.nil? )
       raise ArgumentError.new('Missing options') if( options.nil? )
 
-      data    = Network.api_data( url: url, headers: headers, options: options )
+      begin
+
+        data = api_data( url: url, headers: headers, options: options )
+        data = data.first if( data.is_a?(Array) )
+
+        data
 
-      return nil unless( data.dig(:status).nil? )
+        return data.dig('status') unless( data.nil? )
 
-      results = data.dig('results')
+      rescue => e
+
+        logger.error(e)
+        logger.error(e.backtrace.join("\n"))
+
+        return nil
+      end
 
-      return results.first.dig('status') unless( results.nil? )
     end
 
     # static function for POST Requests
@@ -131,9 +115,9 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.post( params )
+    def post( params )
 
-      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new(format('wrong type. \'params\' must be an Hash, given \'%s\'', params.class.to_s)) unless( params.is_a?(Hash) )
       raise ArgumentError.new('missing params') if( params.size.zero? )
 
       url     = params.dig(:url)
@@ -146,71 +130,25 @@ module Icinga2
       raise ArgumentError.new('Missing options') if( options.nil? )
       raise ArgumentError.new('only Hash for payload are allowed') unless( payload.is_a?(Hash) )
 
-      max_retries = 30
-      retried     = 0
-      result      = {}
-
+      rest_client = RestClient::Resource.new( URI.encode( url ), options )
       headers['X-HTTP-Method-Override'] = 'POST'
 
-      rest_client = RestClient::Resource.new(
-        URI.encode( url ),
-        options
-      )
-
       begin
+        data = request( rest_client, 'POST', headers, payload )
 
-        data = rest_client.post(
-          JSON.generate( payload ),
-          headers
-        )
-
-        data    = JSON.parse( data )
-
-        results = data.dig('results').first
-
-        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 )
+        data = JSON.parse( data ) if( data.is_a?(String) )
+        data = data.deep_string_keys
+        data = data.dig('results').first if( data.is_a?(Hash) )
 
-        results = error.dig( 'results' )
+        return { 'code' => data.dig('code').to_i, 'name' => data.dig('name'), 'status' => data.dig('status') } unless( data.nil? )
 
-        if( !results.nil? )
-          return {
-            status: results.dig('code').to_i,
-            name: results.dig('name'),
-            message: results.dig('status'),
-            error: results.dig('errors')
-          }
-        else
-          return {
-            status: error.dig( 'error' ).to_i,
-            message: error.dig( 'status' )
-          }
-        end
+      rescue => e
 
-      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
+        logger.error(e)
+        logger.error(e.backtrace.join("\n"))
 
-        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
-
-          message = format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
-          $stderr.puts( message )
-
-          return {
-            status: 500,
-            message: message
-          }
-        end
+        return nil
       end
-
-      result
     end
 
     # static function for PUT Requests
@@ -225,9 +163,9 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.put( params )
+    def put( params )
 
-      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new(format('wrong type. \'params\' must be an Hash, given \'%s\'', params.class.to_s)) unless( params.is_a?(Hash) )
       raise ArgumentError.new('missing params') if( params.size.zero? )
 
       url     = params.dig(:url)
@@ -240,72 +178,31 @@ module Icinga2
       raise ArgumentError.new('Missing options') if( options.nil? )
       raise ArgumentError.new('only Hash for payload are allowed') unless( payload.is_a?(Hash) )
 
-      max_retries = 30
-      retried     = 0
-      result      = {}
-
+      rest_client = RestClient::Resource.new( URI.encode( url ), options )
       headers['X-HTTP-Method-Override'] = 'PUT'
 
-      rest_client = RestClient::Resource.new(
-        URI.encode( url ),
-        options
-      )
-
       begin
-        data = rest_client.put(
-          JSON.generate( payload ),
-          headers
-        )
-        data    = JSON.parse( data )
-        results = data.dig('results').first
-
-        return { status: results.dig('code').to_i, message: results.dig('status') } unless( results.nil? )
 
-      rescue RestClient::ExceptionWithResponse => e
+        data = request( rest_client, 'PUT', headers, payload )
+        data = JSON.parse( data ) if( data.is_a?(String) )
+        data = data.deep_string_keys
 
-        error  = e.response ? e.response : nil
-        error = JSON.parse( error ) if  error.is_a?( String )
-
-        results = error.dig('results')
-
-        return { status: error.dig('error').to_i, message: error.dig('status'), error: results } if( results.nil? )
-
-        if( results.is_a?( Hash ) && results.count != 0 )
-
-          return {
-            status: results.dig('code').to_i,
-            name: results.dig('name'),
-            message: results.dig('status'),
-            error: results
-          }
+        if( data.is_a?(Hash) )
+          results = data.dig('results')
+          results = results.first if( results.is_a?(Array) )
         else
-          return {
-            status: results.first.dig('code').to_i,
-            message: format('%s (possible, object already exists)', results.first.dig('status') ),
-            error: results
-          }
+          results = data
         end
 
-      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
+        return { 'code' => results.dig('code').to_i, 'name' => results.dig('name'), 'status' => results.dig('status') } unless( results.nil? )
 
-        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
+      rescue => e
 
-          message = format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
-          $stderr.puts( message )
+        logger.error(e)
+        logger.error(e.backtrace.join("\n"))
 
-          return {
-            status: 500,
-            message: message
-          }
-        end
+        return nil
       end
-
-      result
     end
 
     # static function for DELETE Requests
@@ -318,9 +215,9 @@ module Icinga2
     #
     # @return [Hash]
     #
-    def self.delete( params )
+    def delete( params )
 
-      raise ArgumentError.new('only Hash are allowed') unless( params.is_a?(Hash) )
+      raise ArgumentError.new(format('wrong type. \'params\' must be an Hash, given \'%s\'', params.class.to_s)) unless( params.is_a?(Hash) )
       raise ArgumentError.new('missing params') if( params.size.zero? )
 
       url     = params.dig(:url)
@@ -331,74 +228,175 @@ module Icinga2
       raise ArgumentError.new('Missing headers') if( headers.nil? )
       raise ArgumentError.new('Missing options') if( options.nil? )
 
+      rest_client = RestClient::Resource.new( URI.encode( url ), options )
+      headers['X-HTTP-Method-Override'] = 'DELETE'
+
+      begin
+        data = request( rest_client, 'DELETE', headers )
+
+        data = JSON.parse( data ) if( data.is_a?(String) )
+        data = data.deep_string_keys
+
+        if( data.is_a?(Hash) )
+          results = data.dig('results')
+          results = results.first if( results.is_a?(Array) )
+        else
+          results = data
+        end
+
+        return { 'code' => results.dig('code').to_i, 'name' => results.dig('name'), 'status' => results.dig('status') } unless( results.nil? )
+
+      rescue => e
+        logger.error(e)
+        logger.error(e.backtrace.join("\n"))
+
+        return nil
+      end
+    end
+
+    private
+    #
+    # internal functionfor the Rest-Client Request
+    #
+    def request( client, method, headers, data = {} )
+
+      # logger.debug( "request( #{client.to_s}, #{method}, #{headers}, #{options}, #{data} )" )
+
+      raise ArgumentError.new('client must be an RestClient::Resource') unless( client.is_a?(RestClient::Resource) )
+      raise ArgumentError.new('method must be an \'GET\', \'POST\', \'PUT\' or \'DELETE\'') unless( %w[GET POST PUT DELETE].include?(method) )
+
+      unless( data.nil? )
+        raise ArgumentError.new(format('data must be an Hash (%s)', data.class.to_s)) unless( data.is_a?(Hash) )
+      end
+
       max_retries = 3
       retried     = 0
 
-      headers['X-HTTP-Method-Override'] = 'DELETE'
+      begin
 
-      result  = {}
+        case method.upcase
+        when 'GET'
+          response = client.get( headers )
+        when 'POST'
+          response = client.post( data.to_json, headers )
+        when 'PATCH'
+          response = client.patch( data, headers )
+        when 'PUT'
+          # response = @api_instance[endpoint].put( data, @headers )
+          client.put( data.to_json, headers ) do |response, req, _result|
+
+            @req           = req
+            @response_raw  = response
+            @response_body = response.body
+            @response_code = response.code.to_i
+
+            # logger.debug('----------------------------')
+            # logger.debug(@response_raw)
+            # logger.debug(@response_body)
+            # logger.debug(@response_code)
+            # logger.debug('----------------------------')
+
+            case response.code
+            when 200
+              return @response_body
+            when 400
+              raise RestClient::BadRequest
+            when 404
+              raise RestClient::NotFound
+            when 500
+              raise RestClient::InternalServerError
+            else
+              response.return
+            end
+          end
+
+        when 'DELETE'
+          response = client.delete( @headers )
+        else
+          @logger.error( "Error: #{__method__} is not a valid request method." )
+          return false
+        end
 
-      rest_client = RestClient::Resource.new(
-        URI.encode( url ),
-        options
-      )
+        response_body    = response.body
+        response_headers = response.headers
+        response_body    = JSON.parse( response_body )
 
-      begin
-        data     = rest_client.get( headers )
+        return response_body
 
-        if( data )
+      rescue RestClient::BadRequest
 
-          data    = JSON.parse( data )
-          results = data.dig('results').first
+        response_body = JSON.parse(response_body) if response_body.is_a?(String)
 
-          return { status: results.dig('code').to_i, name: results.dig('name'), message: results.dig('status') } unless( results.nil? )
-        end
+        return {
+         'results' => [{
+           'code' => 400,
+           'status' => response_body.nil? ? 'Bad Request' : response_body
+          }]
+        }
 
-      rescue RestClient::ExceptionWithResponse => e
+      rescue RestClient::Unauthorized
 
-        error  = e.response ? e.response : nil
+        return {
+            'code' => 401,
+            'status' => format('Not authorized to connect \'%s\' - wrong username or password?', @icinga_api_url_base)
+        }
 
-        error = JSON.parse( error ) if  error.is_a?( String )
+      rescue RestClient::NotFound
 
-        results = error.dig('results')
+        return {
+          'results' => [{
+            'code' => 404,
+            'status' => 'Object not Found'
+          }]
+        }
 
-        if( results.nil? )
-          return {
-            status: error.dig( 'error' ).to_i,
-            message: error.dig( 'status' )
-          }
-        else
+      rescue RestClient::InternalServerError
 
-          results = results.first if( results.is_a?(Array) )
+        response_body = JSON.parse(@response_body) if @response_body.is_a?(String)
+
+        results = response_body.dig('results')
+        results = results.first if( results.is_a?(Array) )
+        status  = results.dig('status')
+        errors  = results.dig('errors')
+        errors  = errors.first if( errors.is_a?(Array) )
+        errors  = errors.sub(/ \'.*\'/,'')
+
+        return {
+          'results' => [{
+            'code' => 500,
+            'status' => format('%s (%s)', status, errors).delete('.')
+          }]
+        }
 
-          return {
-            status: results.dig('code').to_i,
-            name: results.dig('name'),
-            message: results.dig('status'),
-            error: results.dig('errors')
-          }
-        end
       rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
 
-        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
+        # TODO
+        # ist hier ein raise sinnvoll?
+        raise format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, @icinga_api_url_base ) if( retried >= max_retries )
 
-          message = format( "Maximum retries (%d) against '%s' reached. Giving up ...", max_retries, url )
-          $stderr.puts( message )
+        retried += 1
+        $stderr.puts(format("Cannot execute request against '%s': '%s' (retry %d / %d)", @icinga_api_url_base, e, retried, max_retries))
+        sleep(3)
+        retry
 
-          return {
-            status: 500,
-            message: message
-          }
-        end
+      rescue RestClient::ExceptionWithResponse => e
+
+        @logger.error( "Error: #{__method__} #{method_type.upcase} on #{endpoint} error: '#{e}'" )
+        @logger.error( data )
+        @logger.error( @headers )
+        @logger.error( JSON.pretty_generate( response_headers ) )
+
+
+        return {
+          'results' => [{
+            'code' => 500,
+            'status' => e
+          }]
+        }
       end
 
-      result
     end
 
+
   end
 end