Add apollo link handler to intercept graphQL request

Author: MahmoudMehisen

index.d.ts

@@ -532,6 +532,12 @@ export namespace NetworkLogger {
    * @param {function} handler
    */
   function setProgressHandlerForRequest(handler: () => void): void;
+  /**
+   * Apollo Link Request Handler to track network log for graphQL using apollo
+   * @param {any} operation 
+   * @param {any} forward 
+   */
+  function apolloLinkRequestHandler(operation: any, forward: any):any;
 }
 export class Trace {
   constructor(id: string, name?: string, attributes?: object);

modules/NetworkLogger.js

@@ -2,11 +2,20 @@ import { NativeModules, Platform } from 'react-native';
 import xhr from '../utils/XhrNetworkInterceptor';
 import IBGEventEmitter from '../utils/IBGEventEmitter.js';
 import InstabugConstants from '../utils/InstabugConstants';
-let { Instabug } = NativeModules;
-
+let { Instabug, IBGAPM } = NativeModules;
 
 var _networkDataObfuscationHandlerSet = false;
 var _requestFilterExpression = false;
+const appendOperationName = (headers, operationName) => {
+  try {
+    let newHeaders = JSON.parse(JSON.stringify(headers));
+    newHeaders[InstabugConstants.GRAPHQL_HEADER] = operationName;
+    return newHeaders;
+  } catch(e) {
+    console.error(e);
+    return headers;
+  }
+};
 
 /**
  * NetworkLogger
@@ -32,6 +41,7 @@ export default {
             try {
               if (Platform.OS === 'android') {
                 Instabug.networkLog(JSON.stringify(network));
+                IBGAPM.networkLog(JSON.stringify(network));
               } else {
                 Instabug.networkLog(network);
               }
@@ -57,13 +67,15 @@ export default {
     }
     _networkDataObfuscationHandlerSet = true;
 
-    IBGEventEmitter.addListener(Instabug,
+    IBGEventEmitter.addListener(
+      Instabug,
       InstabugConstants.NETWORK_DATA_OBFUSCATION_HANDLER_EVENT,
       async data => {
         try {
           const newData = await handler(data);
           if (Platform.OS === 'android') {
             Instabug.networkLog(JSON.stringify(newData));
+            IBGAPM.networkLog(JSON.stringify(newData));
           } else {
             Instabug.networkLog(newData);
           }
@@ -74,7 +86,6 @@ export default {
     );
   },
 
-
   /**
    * Omit requests from being logged based on either their request or response details
    * @param {string} expression
@@ -91,5 +102,15 @@ export default {
     xhr.setOnProgressCallback(handler);
   },
 
+  apolloLinkRequestHandler(operation, forward) {
+    try {
+      operation.setContext(({ headers = {} }) => ({
+        headers: appendOperationName(headers, operation.operationName),
+      }));
+    } catch(e) {
+      console.error(e);
+    }
 
+    return forward(operation);
+  },
 };