Added default headers in PKGS APIs calls for IDE version and PKGS plu…

…gin version. (#61)
JetBrains · Feb 7, 2024 · fc8710d · fc8710d
1 parent 0377dc4
commit fc8710d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/package-search-api-models b/package-search-api-models
diff --git a/...tlin/com/jetbrains/packagesearch/plugin/services/PackageSearchApplicationCachesService.kt b/...tlin/com/jetbrains/packagesearch/plugin/services/PackageSearchApplicationCachesService.kt
@@ -6,12 +6,15 @@ import com.intellij.ide.actions.cache.AsyncRecoveryResult
 import com.intellij.ide.actions.cache.RecoveryAction
 import com.intellij.ide.actions.cache.RecoveryScope
 import com.intellij.openapi.Disposable
+import com.intellij.openapi.application.ApplicationInfo
 import com.intellij.openapi.application.appSystemDir
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.components.Service.Level
+import com.intellij.openapi.components.service
 import com.jetbrains.packagesearch.plugin.PackageSearchBundle
 import com.jetbrains.packagesearch.plugin.core.PackageSearch
 import com.jetbrains.packagesearch.plugin.core.nitrite.buildDefaultNitrate
+import com.jetbrains.packagesearch.plugin.core.utils.IntelliJApplication
 import com.jetbrains.packagesearch.plugin.core.utils.PKGSInternalAPI
 import com.jetbrains.packagesearch.plugin.utils.ApiPackageCacheEntry
 import com.jetbrains.packagesearch.plugin.utils.ApiRepositoryCacheEntry
@@ -20,8 +23,10 @@ import com.jetbrains.packagesearch.plugin.utils.KtorDebugLogger
 import com.jetbrains.packagesearch.plugin.utils.PackageSearchApiPackageCache
 import com.jetbrains.packagesearch.plugin.utils.PackageSearchProjectService
 import io.ktor.client.engine.java.Java
+import io.ktor.client.plugins.DefaultRequest
 import io.ktor.client.plugins.logging.LogLevel
 import io.ktor.client.plugins.logging.Logging
+import io.ktor.http.headers
 import java.util.concurrent.CompletableFuture
 import kotlin.io.path.absolutePathString
 import kotlin.io.path.div
@@ -76,6 +81,12 @@ class PackageSearchApplicationCachesService(private val coroutineScope: Coroutin
                 logger = KtorDebugLogger()
                 filter { it.attributes.getOrNull(PackageSearchApiClient.Attributes.Cache) == true }
             }
+            install(DefaultRequest) {
+                headers {
+                    append("JB-Plugin-Version", PackageSearch.pluginVersion)
+                    append("JB-IDE-Version", IntelliJApplication.service<ApplicationInfo>().strictVersion)
+                }
+            }
         }
     )