Add UI Trace APIs

Author: AliAbdelfattah

android/src/main/java/com/instabug/reactlibrary/RNInstabugAPMModule.java

@@ -34,7 +34,7 @@ public String getName() {
     }
 
     /**
-     * Enables or disables the feature Chats.
+     * Enables or disables APM.
      * @param isEnabled boolean indicating enabled or disabled.
      */
     @ReactMethod
@@ -52,7 +52,7 @@ public void run() {
     }
 
     /**
-     * Enables or disables the feature Chats.
+     * Enables or disables app launch tracking.
      * @param isEnabled boolean indicating enabled or disabled.
      */
     @ReactMethod
@@ -89,8 +89,10 @@ public void run() {
     }
 
     /**
-     * Starts a trace
+     * Adds a new attribute to trace
      * @param id String id of the trace.
+     * @param key   attribute key
+     * @param value attribute value. Null to remove attribute
      */
     @ReactMethod
     public void setTraceAttribute(final String id, final String key, final String value) {
@@ -107,7 +109,7 @@ public void run() {
     }
 
     /**
-     * Starts a trace
+     * Ends a trace
      * @param id string id of the trace.
      */
     @ReactMethod
@@ -124,5 +126,38 @@ public void run() {
         });
     }
 
+    /**
+     * Starts a UI trace
+     * @param name string name of the UI trace.
+     */
+    @ReactMethod
+    public void startUITrace(final String name) {
+        MainThreadHandler.runOnMainThread(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    APM.startUITrace(name);
+                } catch (Exception e) {
+                    e.printStackTrace();
+                }
+            }
+        });
+    }
 
+    /**
+     * Ends the current running UI trace
+     */
+    @ReactMethod
+    public void endUITrace() {
+        MainThreadHandler.runOnMainThread(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    APM.endUITrace();
+                } catch (Exception e) {
+                    e.printStackTrace();
+                }
+            }
+        });
+    }
 }

index.d.ts

@@ -125,6 +125,20 @@ export namespace NetworkLogger {
   function setRequestFilterExpression(expression: string): void;
   function setProgressHandlerForRequest(handler: () => void): void;
 }
+export class Trace {
+  constructor(id: string, name?: string, attributes?: object);
+  setAttribute(key: string, value: string): void;
+  end(): void;
+}
+export namespace APM {
+  function setEnabled(isEnabled: boolean): void;
+  function setAppLaunchEnabled(isEnabled: boolean): void;
+  function setNetworkEnabledIOS(isEnabled: boolean): void;
+  function setUIHangEnabled(isEnabled: boolean): void;
+  function startTrace(name: string): Trace;
+  function startUITrace(name: string): void;
+  function endUITrace(): void;
+}
 export function startWithToken(
   token: string,
   invocationEvent: invocationEvent[]

ios/RNInstabug/InstabugAPMBridge.h

@@ -13,11 +13,14 @@
 
 - (void)setEnabled:(BOOL)isEnabled;
 - (void)setAppLaunchEnabled:(BOOL)isEnabled;
-- (void)startTrace:(NSString *)name :(NSString *)id;
-- (void)setTraceAttribute:(NSString *)id :(NSString *)key :(NSString *)value;
+- (void)setUIHangEnabled:(BOOL)isEnabled;
+- (void)startTrace:(NSString *)name:(NSString *)id;
+- (void)setTraceAttribute:(NSString *)id:(NSString *)key
+                         :(NSString *)value;
 - (void)endTrace:(NSString *)id;
+- (void)startUITrace:(NSString *)name;
+- (void)endUITrace;
 
 extern NSMutableDictionary *traces;
 
 @end
-

ios/RNInstabug/InstabugAPMBridge.m

@@ -43,6 +43,10 @@ - (id) init
     IBGAPM.appLaunchEnabled = isEnabled;
 }
 
+RCT_EXPORT_METHOD(setUIHangEnabled:(BOOL)isEnabled) {
+    IBGAPM.UIHangEnabled = isEnabled;
+}
+
 RCT_EXPORT_METHOD(startTrace:(NSString *)name :(NSString *)id) {
     IBGTrace *trace = [IBGAPM startTraceWithName:name];
     [traces setObject: trace forKey: id];
@@ -58,6 +62,14 @@ - (id) init
     [trace end];
 }
 
+RCT_EXPORT_METHOD(startUITrace:(NSString *)name) {
+    [IBGAPM startUITraceWithName:name];
+}
+
+RCT_EXPORT_METHOD(endUITrace) {
+    [IBGAPM endUITrace];
+}
+
 
 @synthesize description;
 

modules/APM.js

@@ -37,6 +37,16 @@ export default {
         }
     },
 
+    /**
+     * Enables or disables APM UI Responsivenes tracking feature
+     * @param {boolean} isEnabled 
+     */
+    setUIHangEnabled(isEnabled) {
+        if (Platform.OS === 'ios') {
+            IBGAPM.setUIHangEnabled(isEnabled);
+        }
+    },
+
     /**
      * Starts a custom trace
      * @param {string} name 
@@ -46,4 +56,20 @@ export default {
         IBGAPM.startTrace(name, id);
         return new Trace(id, name);
     },
+
+    /**
+     * Starts a custom trace
+     * @param {string} name 
+     */
+    startUITrace(name) {
+        IBGAPM.startUITrace(name);
+    },
+
+    /**
+     * Starts a custom trace
+     * @param {string} name 
+     */
+    endUITrace() {
+        IBGAPM.endUITrace();
+    },
 };