Feature/cid 4110 fix full scan jwt expiration (#134)

* cid-4110: Fix fetch cache and expiration

* cid-4110: Modify reconnect check

* cid-4110: Modify installation token expiration

Author: henriq-amaral-leanix

src/main/kotlin/net/leanix/githubagent/handler/BrokerStompSessionHandler.kt

@@ -52,7 +52,7 @@ class BrokerStompSessionHandler(
 
     override fun handleTransportError(session: StompSession, exception: Throwable) {
         logger.error("Connection error: ${exception.message}")
-        if (isConnected) {
+        if (isConnected && session.sessionId == webSocketService.stompSession?.sessionId) {
             isConnected = false
             logger.error("Session closed. This could be due to a network error or the server closing the connection.")
             logger.info("Reconnecting...")

src/main/kotlin/net/leanix/githubagent/services/CachingService.kt

@@ -6,6 +6,7 @@ import com.github.benmanes.caffeine.cache.Expiry
 import jakarta.annotation.PostConstruct
 import net.leanix.githubagent.config.GitHubEnterpriseProperties
 import org.springframework.stereotype.Service
+import java.util.concurrent.TimeUnit
 
 @Service
 class CachingService(
@@ -22,7 +23,7 @@ class CachingService(
                 value: CacheValue,
                 currentTime: Long
             ): Long {
-                return value.expiry?.times(1_000_000_000) ?: Long.MAX_VALUE
+                return TimeUnit.SECONDS.toNanos(value.expiry ?: Long.MAX_VALUE)
             }
 
             override fun expireAfterUpdate(
@@ -31,7 +32,7 @@ class CachingService(
                 currentTime: Long,
                 currentDuration: Long
             ): Long {
-                return value.expiry?.times(1_000_000_000) ?: Long.MAX_VALUE
+                return TimeUnit.SECONDS.toNanos(value.expiry ?: Long.MAX_VALUE)
             }
 
             override fun expireAfterRead(

src/main/kotlin/net/leanix/githubagent/services/FullScanService.kt

@@ -1,22 +1,26 @@
 package net.leanix.githubagent.services
 
 import net.leanix.githubagent.exceptions.UnableToSendMessageException
+import org.springframework.scheduling.annotation.Async
 import org.springframework.stereotype.Service
 
 @Service
 class FullScanService(
     private val gitHubScanningService: GitHubScanningService,
-    private val syncLogService: SyncLogService
+    private val syncLogService: SyncLogService,
+    private val gitHubAuthenticationService: GitHubAuthenticationService
 ) {
 
     companion object {
         var requireScan: Boolean = false
     }
 
+    @Async
     fun verifyAndStartScan() {
         if (requireScan) {
             runCatching {
                 requireScan = false
+                gitHubAuthenticationService.generateAndCacheJwtToken()
                 syncLogService.sendFullScanStart(null)
                 gitHubScanningService.scanGitHubResources()
             }.onSuccess {

src/main/kotlin/net/leanix/githubagent/services/GitHubAuthenticationService.kt

@@ -32,7 +32,8 @@ class GitHubAuthenticationService(
 ) {
 
     companion object {
-        private const val JWT_EXPIRATION_DURATION = 600000L
+        private const val JWT_EXPIRATION_DURATION_IN_SECONDS = 600L
+        private const val INSTALLATION_JWT_EXPIRATION_DURATION_IN_SECONDS = 3600L
         private const val pemPrefix = "-----BEGIN RSA PRIVATE KEY-----"
         private const val pemSuffix = "-----END RSA PRIVATE KEY-----"
         private val logger = LoggerFactory.getLogger(GitHubAuthenticationService::class.java)
@@ -76,7 +77,7 @@ class GitHubAuthenticationService(
         return runCatching {
             Jwts.builder()
                 .setIssuedAt(Date())
-                .setExpiration(Date(System.currentTimeMillis() + JWT_EXPIRATION_DURATION))
+                .setExpiration(Date(System.currentTimeMillis() + (JWT_EXPIRATION_DURATION_IN_SECONDS * 1000)))
                 .setIssuer(cachingService.get("githubAppId").toString())
                 .signWith(privateKey, SignatureAlgorithm.RS256)
                 .compact()
@@ -89,7 +90,7 @@ class GitHubAuthenticationService(
     private fun verifyAndCacheJwtToken(jwt: String) {
         runCatching {
             gitHubEnterpriseService.verifyJwt(jwt)
-            cachingService.set("jwtToken", jwt, JWT_EXPIRATION_DURATION)
+            cachingService.set("jwtToken", jwt, JWT_EXPIRATION_DURATION_IN_SECONDS)
             logger.info("JWT token generated and cached successfully")
         }.onFailure {
             logger.error("Failed to verify and cache JWT token", it)
@@ -103,7 +104,11 @@ class GitHubAuthenticationService(
     ) {
         installations.forEach { installation ->
             val installationToken = gitHubClient.createInstallationToken(installation.id, "Bearer $jwtToken").token
-            cachingService.set("installationToken:${installation.id}", installationToken, 3600L)
+            cachingService.set(
+                "installationToken:${installation.id}",
+                installationToken,
+                INSTALLATION_JWT_EXPIRATION_DURATION_IN_SECONDS
+            )
         }
     }