feat(client): improve logging

Logging is now:
1. Streaming
4. Configurable in-memory
5. Generally more robust
6. Usable with any underlying http client

Author: stainless-app[bot]

README.md

@@ -615,8 +615,6 @@ The SDK throws custom unchecked exception types:
 
 ## Logging
 
-The SDK uses the standard [OkHttp logging interceptor](https://github.com/square/okhttp/tree/master/okhttp-logging-interceptor).
-
 Enable logging by setting the `IMAGE_KIT_LOG` environment variable to `info`:
 
 ```sh
@@ -629,6 +627,19 @@ Or to `debug` for more verbose logging:
 export IMAGE_KIT_LOG=debug
 ```
 
+Or configure the client manually using the `logLevel` method:
+
+```java
+import io.imagekit.client.ImageKitClient;
+import io.imagekit.client.okhttp.ImageKitOkHttpClient;
+import io.imagekit.core.LogLevel;
+
+ImageKitClient client = ImageKitOkHttpClient.builder()
+    .fromEnv()
+    .logLevel(LogLevel.INFO)
+    .build();
+```
+
 ## ProGuard and R8
 
 Although the SDK uses reflection, it is still usable with [ProGuard](https://github.com/Guardsquare/proguard) and [R8](https://developer.android.com/topic/performance/app-optimization/enable-app-optimization) because `image-kit-java-core` is published with a [configuration file](image-kit-java-core/src/main/resources/META-INF/proguard/image-kit-java-core.pro) containing [keep rules](https://www.guardsquare.com/manual/configuration/usage).

image-kit-java-client-okhttp/build.gradle.kts

@@ -7,7 +7,6 @@ dependencies {
     api(project(":image-kit-java-core"))
 
     implementation("com.squareup.okhttp3:okhttp:4.12.0")
-    implementation("com.squareup.okhttp3:logging-interceptor:4.12.0")
 
     testImplementation(kotlin("test"))
     testImplementation("org.assertj:assertj-core:3.27.7")

image-kit-java-client-okhttp/src/main/kotlin/io/imagekit/client/okhttp/ImageKitOkHttpClient.kt

@@ -6,6 +6,7 @@ import com.fasterxml.jackson.databind.json.JsonMapper
 import io.imagekit.client.ImageKitClient
 import io.imagekit.client.ImageKitClientImpl
 import io.imagekit.core.ClientOptions
+import io.imagekit.core.LogLevel
 import io.imagekit.core.Sleeper
 import io.imagekit.core.Timeout
 import io.imagekit.core.http.Headers
@@ -277,6 +278,15 @@ class ImageKitOkHttpClient private constructor() {
          */
         fun maxRetries(maxRetries: Int) = apply { clientOptions.maxRetries(maxRetries) }
 
+        /**
+         * The level at which to log request and response information.
+         *
+         * [fromEnv] will set the level from environment variables. See [LogLevel.fromEnv].
+         *
+         * Defaults to [LogLevel.fromEnv].
+         */
+        fun logLevel(logLevel: LogLevel) = apply { clientOptions.logLevel(logLevel) }
+
         /**
          * Your ImageKit private API key (starts with `private_`). You can find this in the
          * [ImageKit dashboard](https://imagekit.io/dashboard/developer/api-keys).

image-kit-java-client-okhttp/src/main/kotlin/io/imagekit/client/okhttp/ImageKitOkHttpClientAsync.kt

@@ -6,6 +6,7 @@ import com.fasterxml.jackson.databind.json.JsonMapper
 import io.imagekit.client.ImageKitClientAsync
 import io.imagekit.client.ImageKitClientAsyncImpl
 import io.imagekit.core.ClientOptions
+import io.imagekit.core.LogLevel
 import io.imagekit.core.Sleeper
 import io.imagekit.core.Timeout
 import io.imagekit.core.http.Headers
@@ -277,6 +278,15 @@ class ImageKitOkHttpClientAsync private constructor() {
          */
         fun maxRetries(maxRetries: Int) = apply { clientOptions.maxRetries(maxRetries) }
 
+        /**
+         * The level at which to log request and response information.
+         *
+         * [fromEnv] will set the level from environment variables. See [LogLevel.fromEnv].
+         *
+         * Defaults to [LogLevel.fromEnv].
+         */
+        fun logLevel(logLevel: LogLevel) = apply { clientOptions.logLevel(logLevel) }
+
         /**
          * Your ImageKit private API key (starts with `private_`). You can find this in the
          * [ImageKit dashboard](https://imagekit.io/dashboard/developer/api-keys).

image-kit-java-client-okhttp/src/main/kotlin/io/imagekit/client/okhttp/OkHttpClient.kt

@@ -35,7 +35,6 @@ import okhttp3.Request
 import okhttp3.RequestBody
 import okhttp3.RequestBody.Companion.toRequestBody
 import okhttp3.Response
-import okhttp3.logging.HttpLoggingInterceptor
 import okio.BufferedSink
 import okio.buffer
 import okio.sink
@@ -93,18 +92,6 @@ internal constructor(@JvmSynthetic internal val okHttpClient: okhttp3.OkHttpClie
     private fun newCall(request: HttpRequest, requestOptions: RequestOptions): Call {
         val clientBuilder = okHttpClient.newBuilder()
 
-        val logLevel =
-            when (System.getenv("IMAGE_KIT_LOG")?.lowercase()) {
-                "info" -> HttpLoggingInterceptor.Level.BASIC
-                "debug" -> HttpLoggingInterceptor.Level.BODY
-                else -> null
-            }
-        if (logLevel != null) {
-            clientBuilder.addNetworkInterceptor(
-                HttpLoggingInterceptor().setLevel(logLevel).apply { redactHeader("Authorization") }
-            )
-        }
-
         requestOptions.timeout?.let {
             clientBuilder
                 .connectTimeout(it.connect())

image-kit-java-core/src/main/kotlin/io/imagekit/core/ClientOptions.kt

@@ -5,6 +5,7 @@ package io.imagekit.core
 import com.fasterxml.jackson.databind.json.JsonMapper
 import io.imagekit.core.http.Headers
 import io.imagekit.core.http.HttpClient
+import io.imagekit.core.http.LoggingHttpClient
 import io.imagekit.core.http.PhantomReachableClosingHttpClient
 import io.imagekit.core.http.QueryParams
 import io.imagekit.core.http.RetryingHttpClient
@@ -97,6 +98,14 @@ private constructor(
      * Defaults to 2.
      */
     @get:JvmName("maxRetries") val maxRetries: Int,
+    /**
+     * The level at which to log request and response information.
+     *
+     * [fromEnv] will set the level from environment variables. See [LogLevel.fromEnv].
+     *
+     * Defaults to [LogLevel.fromEnv].
+     */
+    @get:JvmName("logLevel") val logLevel: LogLevel,
     /**
      * Your ImageKit private API key (starts with `private_`). You can find this in the
      * [ImageKit dashboard](https://imagekit.io/dashboard/developer/api-keys).
@@ -175,6 +184,7 @@ private constructor(
         private var responseValidation: Boolean = false
         private var timeout: Timeout = Timeout.default()
         private var maxRetries: Int = 2
+        private var logLevel: LogLevel = LogLevel.fromEnv()
         private var privateKey: String? = null
         private var password: String? = "do_not_set"
         private var webhookSecret: String? = null
@@ -192,6 +202,7 @@ private constructor(
             responseValidation = clientOptions.responseValidation
             timeout = clientOptions.timeout
             maxRetries = clientOptions.maxRetries
+            logLevel = clientOptions.logLevel
             privateKey = clientOptions.privateKey
             password = clientOptions.password
             webhookSecret = clientOptions.webhookSecret
@@ -304,6 +315,15 @@ private constructor(
          */
         fun maxRetries(maxRetries: Int) = apply { this.maxRetries = maxRetries }
 
+        /**
+         * The level at which to log request and response information.
+         *
+         * [fromEnv] will set the level from environment variables. See [LogLevel.fromEnv].
+         *
+         * Defaults to [LogLevel.fromEnv].
+         */
+        fun logLevel(logLevel: LogLevel) = apply { this.logLevel = logLevel }
+
         /**
          * Your ImageKit private API key (starts with `private_`). You can find this in the
          * [ImageKit dashboard](https://imagekit.io/dashboard/developer/api-keys).
@@ -430,6 +450,7 @@ private constructor(
          * System properties take precedence over environment variables.
          */
         fun fromEnv() = apply {
+            logLevel(LogLevel.fromEnv())
             (System.getProperty("imagekit.baseUrl") ?: System.getenv("IMAGE_KIT_BASE_URL"))?.let {
                 baseUrl(it)
             }
@@ -497,7 +518,13 @@ private constructor(
             return ClientOptions(
                 httpClient,
                 RetryingHttpClient.builder()
-                    .httpClient(httpClient)
+                    .httpClient(
+                        LoggingHttpClient.builder()
+                            .httpClient(httpClient)
+                            .clock(clock)
+                            .level(logLevel)
+                            .build()
+                    )
                     .sleeper(sleeper)
                     .clock(clock)
                     .maxRetries(maxRetries)
@@ -512,6 +539,7 @@ private constructor(
                 responseValidation,
                 timeout,
                 maxRetries,
+                logLevel,
                 privateKey,
                 password,
                 webhookSecret,

image-kit-java-core/src/main/kotlin/io/imagekit/core/LogLevel.kt

@@ -0,0 +1,33 @@
+// File generated from our OpenAPI spec by Stainless.
+
+package io.imagekit.core
+
+/** The level at which to log request and response information. */
+enum class LogLevel {
+    /** No logging. */
+    OFF,
+    /** Minimal request and response summary logs. No headers or bodies are logged. */
+    INFO,
+    /** [INFO] logs plus details about request failures. */
+    ERROR,
+    /**
+     * Full request and response logs. Sensitive headers are redacted, but sensitive data in request
+     * and response bodies may still be visible.
+     */
+    DEBUG;
+
+    /** Returns whether this level is at or higher than the given [level]. */
+    fun shouldLog(level: LogLevel): Boolean = ordinal >= level.ordinal
+
+    companion object {
+
+        /** Returns a [LogLevel] based on the `IMAGE_KIT_LOG` environment variable. */
+        fun fromEnv() =
+            when (System.getenv("IMAGE_KIT_LOG")?.lowercase()) {
+                "info" -> INFO
+                "error" -> ERROR
+                "debug" -> DEBUG
+                else -> OFF
+            }
+    }
+}

image-kit-java-core/src/main/kotlin/io/imagekit/core/Utils.kt

@@ -5,6 +5,7 @@ package io.imagekit.core
 import io.imagekit.errors.ImageKitInvalidDataException
 import java.util.Collections
 import java.util.SortedMap
+import java.util.SortedSet
 import java.util.concurrent.CompletableFuture
 import java.util.concurrent.locks.Lock
 
@@ -16,6 +17,11 @@ internal fun <T : Any> T?.getOrThrow(name: String): T =
 internal fun <T> List<T>.toImmutable(): List<T> =
     if (isEmpty()) Collections.emptyList() else Collections.unmodifiableList(toList())
 
+@JvmSynthetic
+internal fun <V : Comparable<V>> SortedSet<V>.toImmutable(): SortedSet<V> =
+    if (isEmpty()) Collections.emptySortedSet()
+    else Collections.unmodifiableSortedSet(toSortedSet(comparator() ?: Comparator.naturalOrder()))
+
 @JvmSynthetic
 internal fun <K, V> Map<K, V>.toImmutable(): Map<K, V> =
     if (isEmpty()) immutableEmptyMap() else Collections.unmodifiableMap(toMap())