Added Manifest Fetcher

Author: LokeshDogga13

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/artifacts/ManifestFetcher.kt

@@ -0,0 +1,113 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.artifacts
+
+import com.intellij.openapi.util.text.StringUtil
+import com.intellij.util.io.DigestUtil
+import com.intellij.util.io.HttpRequests
+import software.amazon.awssdk.utils.UserHomeDirectoryUtils
+import software.aws.toolkits.core.utils.exists
+import software.aws.toolkits.core.utils.getLogger
+import software.aws.toolkits.core.utils.readText
+import software.aws.toolkits.jetbrains.core.getTextFromUrl
+import software.aws.toolkits.jetbrains.core.saveFileFromUrl
+import software.aws.toolkits.jetbrains.services.amazonq.project.manifest.ManifestManager
+import java.nio.file.Path
+import java.nio.file.Paths
+
+
+class ManifestFetcher {
+
+    private val lspManifestUrl = "https://aws-toolkit-language-servers.amazonaws.com/codewhisperer/0/manifest.json"
+    private val manifestManager = ManifestManager()
+    private val lspManifestFilePath: Path = Paths.get(UserHomeDirectoryUtils.userHomeDirectory(), ".aws", "amazonq", "cache", "lspManifest.js")
+
+    companion object {
+        private val logger = getLogger<ManifestFetcher>()
+    }
+
+    /**
+     * Method which will be used to fetch latest manifest.
+     * */
+    fun fetch() : ManifestManager.Manifest? {
+        val localManifest = fetchManifestFromLocal()
+        if (localManifest != null) {
+            return localManifest
+        }
+        val remoteManifest = fetchManifestFromRemote() ?: return null
+        return remoteManifest
+    }
+
+    private fun fetchManifestFromRemote() : ManifestManager.Manifest? {
+        val manifest : ManifestManager.Manifest?
+        try {
+            val manifestString = getTextFromUrl(lspManifestUrl)
+            manifest = manifestManager.readManifestFile(manifestString) ?: return null
+        }
+        catch (e: Exception) {
+            logger.error("error fetching lsp manifest from remote URL ${e.message}", e)
+            return null
+        }
+        if (manifest.isManifestDeprecated == true) {
+            logger.info("Manifest is deprecated")
+            return null
+        }
+        updateManifestCache()
+        logger.info("Using manifest found from remote URL")
+        return manifest
+    }
+
+    private fun updateManifestCache() {
+        try {
+            saveFileFromUrl(lspManifestUrl, lspManifestFilePath)
+        }
+        catch (e: Exception) {
+            logger.error("error occurred while saving lsp manifest to local cache ${e.message}", e)
+        }
+    }
+
+    private fun fetchManifestFromLocal() : ManifestManager.Manifest? {
+        var manifest : ManifestManager.Manifest? = null
+        val localETag = getManifestETagFromLocal()
+        val remoteETag = getManifestETagFromUrl()
+        // If local and remote have same ETag, we can re-use the manifest file from local to fetch artifacts.
+        if (localETag != null && remoteETag != null && localETag == remoteETag) {
+            try {
+                val manifestContent = lspManifestFilePath.readText()
+                manifest = manifestManager.readManifestFile(manifestContent) ?: return null
+            }
+            catch (e: Exception) {
+                logger.error("error reading lsp manifest file from local ${e.message}", e)
+                return null
+            }
+        }
+        logger.info("Re-using lsp manifest from local.")
+        return manifest
+    }
+
+    private fun getManifestETagFromLocal() : String? {
+        if(lspManifestFilePath.exists()) {
+            val messageDigest = DigestUtil.md5()
+            DigestUtil.updateContentHash(messageDigest, lspManifestFilePath)
+            return StringUtil.toHexString(messageDigest.digest())
+        }
+        return null
+    }
+
+    private fun getManifestETagFromUrl() : String? {
+        try {
+            val actualETag = HttpRequests.head(lspManifestUrl)
+
+                .userAgent("AWS Toolkit for JetBrains")
+                .connect { request ->
+                    request.connection.headerFields["ETag"]?.firstOrNull().orEmpty()
+                }
+            return actualETag.trim('"')
+        }
+        catch (e: Exception) {
+            logger.error("error fetching ETag of lsp manifest from url.", e)
+        }
+        return null
+    }
+}

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/project/manifest/ManifestManager.kt

@@ -41,12 +41,45 @@ class ManifestManager {
         @JsonProperty("contents")
         val contents: List<TargetContent>? = emptyList(),
     )
+
+    data class RunTime(
+        @JsonIgnoreProperties(ignoreUnknown = true)
+        @JsonProperty("name")
+        val name: String? = null,
+        @JsonProperty("version")
+        val version: String? = null
+    )
+
+    data class Capabilities(
+        @JsonIgnoreProperties(ignoreUnknown = true)
+        @JsonProperty("name")
+        val name: String? = null,
+        @JsonProperty("version")
+        val version: String? = null
+    )
+
+    data class Protocol(
+        @JsonIgnoreProperties(ignoreUnknown = true)
+        @JsonProperty("name")
+        val name: String?= null,
+        @JsonProperty("version")
+        val version: String? = null
+    )
+
     data class Version(
         @JsonIgnoreProperties(ignoreUnknown = true)
         @JsonProperty("serverVersion")
         val serverVersion: String? = null,
         @JsonProperty("isDelisted")
         val isDelisted: Boolean? = null,
+        @JsonProperty("runtime")
+        val runtime: RunTime? = null,
+        @JsonProperty("capabilities")
+        val capabilities: List<Capabilities>? = emptyList(),
+        @JsonProperty("protocol")
+        val protocol: List<Protocol>? = emptyList(),
+        @JsonProperty("thirdPartyLicenses")
+        val thirdPartyLicenses: String? = null,
         @JsonProperty("targets")
         val targets: List<VersionTarget>? = emptyList(),
     )
@@ -67,7 +100,7 @@ class ManifestManager {
 
     fun getManifest(): Manifest? = fetchFromRemoteAndSave()
 
-    private fun readManifestFile(content: String): Manifest? {
+    fun readManifestFile(content: String): Manifest? {
         try {
             return mapper.readValue<Manifest>(content)
         } catch (e: Exception) {