Add tracing interceptor

Author: FranAguilera

Cargo.Bazel.lock

@@ -108,7 +108,7 @@ final class LoggerCustomer: NSObject, URLSessionDelegate {
                     requestFieldProvider: CustomNetworkFieldProvider(),
                     responseFieldProvider: CustomNetworkResponseFieldProvider()
                 ), ],
-                disableSwizzling: true
+                disableSwizzling: false
             )
 
         Logger.addField(withKey: "field_container_field_key", value: "field_container_value")

Cargo.lock

@@ -53,6 +53,8 @@ class OkHttpEventListenerMethodVisitor(
     ) {
     private val captureOkHttpEventListenerFactory =
         "io/bitdrift/capture/network/okhttp/CaptureOkHttpEventListenerFactory"
+    private val captureOkHttpTracingInterceptor =
+        "io/bitdrift/capture/network/okhttp/CaptureOkHttpTracingInterceptor"
 
     override fun onMethodEnter() {
         super.onMethodEnter()
@@ -64,6 +66,8 @@ class OkHttpEventListenerMethodVisitor(
     }
 
     private fun addOverwritingEventListener() {
+        addTracingInterceptor()
+
         // Add the following call at the beginning of the constructor with the Builder parameter:
         // builder.eventListenerFactory(new CaptureOkHttpEventListenerFactory());
 
@@ -95,9 +99,12 @@ class OkHttpEventListenerMethodVisitor(
             "(Lokhttp3/EventListener\$Factory;)Lokhttp3/OkHttpClient\$Builder;",
             false,
         )
+        visitInsn(Opcodes.POP)
     }
 
     private fun addProxyingEventListener() {
+        addTracingInterceptor()
+
         // Add the following call at the beginning of the constructor with the Builder parameter:
         // builder.eventListenerFactory(new CaptureOkHttpEventListenerFactory(builder.eventListenerFactory));
 
@@ -141,5 +148,35 @@ class OkHttpEventListenerMethodVisitor(
             "(Lokhttp3/EventListener\$Factory;)Lokhttp3/OkHttpClient\$Builder;",
             false,
         )
+        visitInsn(Opcodes.POP)
+    }
+
+    private fun addTracingInterceptor() {
+        // Add the following call at the beginning of the constructor with the Builder parameter:
+        // builder.addInterceptor(new CaptureOkHttpTracingInterceptor());
+
+        // OkHttpClient.Builder is the parameter, retrieved here
+        visitVarInsn(Opcodes.ALOAD, 1)
+
+        // Create CaptureOkHttpTracingInterceptor instance
+        visitTypeInsn(Opcodes.NEW, captureOkHttpTracingInterceptor)
+        visitInsn(Opcodes.DUP)
+        visitMethodInsn(
+            Opcodes.INVOKESPECIAL,
+            captureOkHttpTracingInterceptor,
+            "<init>",
+            "()V",
+            false,
+        )
+
+        // Call addInterceptor(Interceptor)
+        visitMethodInsn(
+            Opcodes.INVOKEVIRTUAL,
+            "okhttp3/OkHttpClient\$Builder",
+            "addInterceptor",
+            "(Lokhttp3/Interceptor;)Lokhttp3/OkHttpClient\$Builder;",
+            false,
+        )
+        visitInsn(Opcodes.POP)
     }
 }

examples/swift/hello_world/LoggerCustomer.swift

@@ -225,6 +225,14 @@ object Capture {
         val deviceId: String?
             get() = logger()?.deviceId
 
+        /**
+         * Whether workflow-controlled tracing is currently active for this session.
+         * Returns `null` prior to SDK start.
+         */
+        @JvmStatic
+        val isTracingActive: Boolean?
+            get() = logger()?.isTracingActive
+
         /**
          * Defines the initialization of a new session within the currently running logger
          * If no logger is started, this is a no-op.

platform/jvm/capture-plugin/src/main/kotlin/io/bitdrift/capture/instrumentation/okhttp/visitor/OkHttpEventListenerMethodVisitor.kt

@@ -111,6 +111,11 @@ internal object CaptureJniLibrary : IBridge, IStreamingReportProcessor {
      */
     external fun getDeviceId(loggerId: Long): String?
 
+    /**
+     * Returns true when workflow-controlled tracing is active for the current session.
+     */
+    external fun isTracingActive(loggerId: Long): Boolean
+
     /**
      * Adds a field that should be attached to all logs emitted by the logger going forward.
      * If a field with a given key has already been registered with the logger, its value is

platform/jvm/capture/src/main/kotlin/io/bitdrift/capture/Capture.kt

@@ -37,6 +37,11 @@ interface ILogger {
      */
     val deviceId: String
 
+    /**
+     * Whether workflow-controlled tracing is currently active for this session.
+     */
+    val isTracingActive: Boolean
+
     /**
      * Defines the initialization of a new session within the current logger.
      */

platform/jvm/capture/src/main/kotlin/io/bitdrift/capture/CaptureJniLibrary.kt

@@ -0,0 +1,33 @@
+// capture-sdk - bitdrift's client SDK
+// Copyright Bitdrift, Inc. All rights reserved.
+//
+// Use of this source code is governed by a source available license that can be found in the
+// LICENSE file or at:
+// https://polyformproject.org/wp-content/uploads/2020/06/PolyForm-Shield-1.0.0.txt
+
+package io.bitdrift.capture
+
+import io.bitdrift.capture.common.RuntimeConfig
+import io.bitdrift.capture.common.RuntimeFeature
+import io.bitdrift.capture.common.RuntimeStringConfig
+
+internal interface IRuntimeProvider {
+    fun isRuntimeFeatureEnabled(feature: RuntimeFeature): Boolean
+
+    fun getRuntimeConfigValue(config: RuntimeConfig): Int
+
+    fun getRuntimeStringConfigValue(config: RuntimeStringConfig): String
+}
+
+internal object CaptureRuntimeProvider : IRuntimeProvider {
+    override fun isRuntimeFeatureEnabled(feature: RuntimeFeature): Boolean =
+        getLoggerProvider()?.isRuntimeFeatureEnabled(feature) ?: feature.defaultValue
+
+    override fun getRuntimeConfigValue(config: RuntimeConfig): Int =
+        getLoggerProvider()?.getRuntimeConfigValue(config) ?: config.defaultValue
+
+    override fun getRuntimeStringConfigValue(config: RuntimeStringConfig): String =
+        getLoggerProvider()?.getRuntimeStringConfigValue(config) ?: config.defaultValue
+
+    private fun getLoggerProvider(): IRuntimeProvider? = Capture.logger() as? IRuntimeProvider
+}

platform/jvm/capture/src/main/kotlin/io/bitdrift/capture/ILogger.kt

@@ -10,13 +10,17 @@ package io.bitdrift.capture
 import io.bitdrift.capture.common.Runtime
 import io.bitdrift.capture.common.RuntimeConfig
 import io.bitdrift.capture.common.RuntimeFeature
+import io.bitdrift.capture.common.RuntimeStringConfig
 
 internal class JniRuntime(
     private val logger: LoggerId,
 ) : Runtime {
     override fun isEnabled(feature: RuntimeFeature): Boolean = Jni.isRuntimeEnabled(logger, feature.featureName, feature.defaultValue)
 
     override fun getConfigValue(config: RuntimeConfig): Int = Jni.runtimeValue(logger, config.configName, config.defaultValue)
+
+    override fun getConfigValue(config: RuntimeStringConfig): String =
+        Jni.runtimeStringValue(logger, config.configName, config.defaultValue)
 }
 
 internal object Jni {
@@ -31,4 +35,10 @@ internal object Jni {
         variableName: String,
         defaultValue: Int,
     ): Int
+
+    external fun runtimeStringValue(
+        logger: LoggerId,
+        variableName: String,
+        defaultValue: String,
+    ): String
 }

platform/jvm/capture/src/main/kotlin/io/bitdrift/capture/IRuntimeProvider.kt

@@ -16,7 +16,9 @@ import androidx.lifecycle.ProcessLifecycleOwner
 import io.bitdrift.capture.attributes.ClientAttributes
 import io.bitdrift.capture.attributes.NetworkAttributes
 import io.bitdrift.capture.common.IWindowManager
+import io.bitdrift.capture.common.RuntimeConfig
 import io.bitdrift.capture.common.RuntimeFeature
+import io.bitdrift.capture.common.RuntimeStringConfig
 import io.bitdrift.capture.common.WindowManager
 import io.bitdrift.capture.error.ErrorReporterService
 import io.bitdrift.capture.error.IErrorReporter
@@ -90,7 +92,8 @@ internal class LoggerImpl(
     private val eventListenerDispatcher: CaptureDispatchers.CommonBackground = CaptureDispatchers.CommonBackground,
     windowManager: IWindowManager = WindowManager(errorHandler),
 ) : IInternalLogger,
-    ICompletedReportsProcessor {
+    ICompletedReportsProcessor,
+    IRuntimeProvider {
     @OptIn(ExperimentalBitdriftApi::class)
     internal val webViewConfiguration: WebViewConfiguration? = configuration.webViewConfiguration
 
@@ -311,6 +314,9 @@ internal class LoggerImpl(
     override val deviceId: String
         get() = CaptureJniLibrary.getDeviceId(this.loggerId) ?: "unknown"
 
+    override val isTracingActive: Boolean
+        get() = CaptureJniLibrary.isTracingActive(this.loggerId)
+
     override fun startNewSession() {
         CaptureJniLibrary.startNewSession(this.loggerId)
     }
@@ -538,6 +544,12 @@ internal class LoggerImpl(
         )
     }
 
+    override fun isRuntimeFeatureEnabled(feature: RuntimeFeature): Boolean = runtime.isEnabled(feature)
+
+    override fun getRuntimeConfigValue(config: RuntimeConfig): Int = runtime.getConfigValue(config)
+
+    override fun getRuntimeStringConfigValue(config: RuntimeStringConfig): String = runtime.getConfigValue(config)
+
     override fun logSessionReplayScreenshot(
         fields: Array<Field>,
         duration: Duration,