From ac881e067b65e05e9fc882e433e531d50ee1771e Mon Sep 17 00:00:00 2001
From: Gor Martsen <gor.martsen@gmail.com>
Date: Sat, 9 Mar 2019 01:56:32 -0500
Subject: [PATCH] #6 metric hook support prototype

---
 router-proxy.js     | 100 +++++++++++++++++++++++++++++++++++++++-----
 schema/service.json |   2 +-
 2 files changed, 91 insertions(+), 11 deletions(-)

diff --git a/router-proxy.js b/router-proxy.js
index 772d55d..9dcaa1d 100644
--- a/router-proxy.js
+++ b/router-proxy.js
@@ -387,10 +387,10 @@ function hookCall(targetRequest, phase, callback) {
       debug.log('broadcast sent');
       // If more in queue left - send more
       if (broadcastTargets.length) {
-        _request(getBroadcastRequest, callbackBroadcastRequest)
+        _request(getBroadcastRequest, callbackBroadcastRequest, targetRequest)
       }
     }
-    _request(getBroadcastRequest, callbackBroadcastRequest)
+    _request(getBroadcastRequest, callbackBroadcastRequest, targetRequest)
   }
 
   // send Notify
@@ -452,10 +452,10 @@ function hookCall(targetRequest, phase, callback) {
         // If more groups left - send more
         if (notifyGroups.length) {
           currentNotifyGroup = notifyGroups.shift()
-          _request(getNotifyRequest, callbackNotifyRequest)
+          _request(getNotifyRequest, callbackNotifyRequest, targetRequest)
         }
       }
-      _request(getNotifyRequest, callbackNotifyRequest)
+      _request(getNotifyRequest, callbackNotifyRequest, targetRequest)
     }
   }
 
@@ -571,12 +571,12 @@ function hookCall(targetRequest, phase, callback) {
     // If more groups left - send more
     if (adapterGroups.length) {
       currentAdapterGroup = adapterGroups.shift()
-      return _request(getAdapterRequest, callbackAdapterRequest)
+      return _request(getAdapterRequest, callbackAdapterRequest, targetRequest)
     }
     // return back via callback
     callback()
   }
-  _request(getAdapterRequest, callbackAdapterRequest)
+  _request(getAdapterRequest, callbackAdapterRequest, targetRequest)
 
 }
 
@@ -597,7 +597,7 @@ function findHookTarget(targetRequest, phase, type, group){
       continue
     }
     for (let hook of target.hook) {
-      if (hook.phase !== phase) {
+      if (phase !== null && hook.phase !== phase) {
         continue
       }
       if (hook.type !== type) {
@@ -701,7 +701,7 @@ function getMinLoadedRouter(availableRoutes) {
   return minRouter;
 }
 
-function _request(getRequest, callback) {
+function _request(getRequest, callback, targetRequest, noMetric) {
   let requestOptions = getRequest()
   
   if (requestOptions instanceof Error) {
@@ -716,12 +716,92 @@ function _request(getRequest, callback) {
   if (!(uri.host || (uri.hostname && uri.port)) && !uri.isUnix) {
     return callback(new Error('Invalid URI' + requestOptions.uri))
   }
+
+  let getHeaders = function(router, hookType){
+    let headers = {};
+    // TODO verify date,content-type, transfer-encoding headers
+    let skipHeaders = [
+      'host',
+      'content-type',
+      'date',
+      'connection',
+      'content-length',
+      'transfer-encoding'
+    ]
+    for (var i in targetRequest.requestDetails.headers) {
+      if (skipHeaders.indexOf(i) != -1) {
+        continue
+      }
+      headers[i] = targetRequest.requestDetails.headers[i];
+    }
+    for (var i in router.matchVariables) {
+      headers['mfw-' + i] = router.matchVariables[i];
+    }
+    headers['x-origin-url'] = targetRequest.route
+    headers['x-origin-method'] = targetRequest.method
+    headers['x-hook-type'] = hookType
+    headers['x-endpoint-scope'] = targetRequest.endpoint.scope
+    debug.debug('%s headers %O', targetRequest.route, headers);
+    return headers;
+  }
+  let startTime = Date.now();
   request(requestOptions, function(error, response, body) {
+    let endTime = Date.now();
+    if(!noMetric) {
+      let metricTargets = findHookTarget(targetRequest, null, 'metric')
+      debug.debug('NOTIFY: for %s result: %O', targetRequest.route, metricTargets);
+    
+      if (metricTargets instanceof Array) {
+        let getMetricRequest = function(){
+          if (metricTargets instanceof Error) {
+            return metricTargets
+          }
+          if (!metricTargets.length) {
+            return false
+          }
+          let router = metricTargets.pop()
+          debug.log('Metric route %s result %O', targetRequest.route, router);
+          let headers = getHeaders(router, 'metric')
+          let metricBody = "";
+          if(!router.meta) {
+            metricBody = JSON.stringify({
+              request: targetRequest.requestDetails._buffer,
+              response: body,
+              startTime: startTime,
+              endTime: endTime,
+              headers: requestOptions.headers
+            })
+          }
+          // Sign request for hook
+          headers['x-hook-signature'] = 'sha256='
+            + signature('sha256', metricBody, router.secureKey);
+          return {
+            uri: router.url + targetRequest.path,
+            method: 'NOTIFY',
+            headers: headers,
+            body: targetRequest.requestDetails._buffer
+          }
+        }
+        let callbackMetricRequest = function(err, response, body){
+          // No action on broadcast hook.
+          if (err) {
+            debug.log('metric failed %O', err);
+          }
+          debug.log('metric sent');
+          // If more in queue left - send more
+          if (metricTargets.length) {
+            _request(getMetricRequest, callbackMetricRequest, targetRequest, true)
+          }
+        }
+        _request(getMetricRequest, callbackMetricRequest, targetRequest, true)
+      }
+    }
+    
     if (error) {
       // TODO add limit to re send
       debug.debug('_request Error received: %O', error);
       debug.debug('_request Restart request: %O', requestOptions);
-      return _request(getRequest, callback);
+      return _request(getRequest, callback, targetRequest);
     }
     
     debug.debug('%s body: %s', requestOptions.uri, body);
@@ -880,7 +960,7 @@ function proxyRequest(route, path, method, jsonData, requestDetails, callback) {
 
       })
     }
-    _request(getEndpointRequest, callbackEndpointRequest)
+    _request(getEndpointRequest, callbackEndpointRequest, targetRequest)
     
   })
 }
diff --git a/schema/service.json b/schema/service.json
index 8359c73..435fa0a 100644
--- a/schema/service.json
+++ b/schema/service.json
@@ -20,7 +20,7 @@
           },
           "type": {
             "type": "string",
-            "enum": ["adapter", "notify", "broadcast"],
+            "enum": ["adapter", "notify", "broadcast", "metric"],
             "required": true
           },
           "group": {