Merge pull request #129 from leanix/feature/CID-3854/allow-the-user-to-trigger-a-full-scan-from-leanix-workspace

CID-3854: Allow user to trigger a full scan from LeanIX

Author: mohamedlajmileanix

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

@@ -17,6 +17,7 @@ class BrokerStompSessionHandler(
     private val artifactDownloadHandler: ArtifactDownloadHandler,
     private val repositoryGetHandler: RepositoryGetHandler,
     private val installationGetHandler: InstallationGetHandler,
+    private val fullScanHandler: FullScanHandler,
     private val eventPublisher: ApplicationEventPublisher
 ) : StompSessionHandlerAdapter() {
     @Lazy
@@ -33,6 +34,7 @@ class BrokerStompSessionHandler(
         session.subscribe("/user/queue/message/artifact", artifactDownloadHandler)
         session.subscribe("/user/queue/message/repository", repositoryGetHandler)
         session.subscribe("/user/queue/message/installation", installationGetHandler)
+        session.subscribe("/user/queue/message/fullScan", fullScanHandler)
         eventPublisher.publishEvent(ConnectionEstablishedEvent())
     }
 

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

@@ -0,0 +1,42 @@
+package net.leanix.githubagent.handler
+
+import net.leanix.githubagent.services.CachingService
+import net.leanix.githubagent.services.GitHubStartService
+import org.slf4j.LoggerFactory
+import org.springframework.beans.factory.annotation.Autowired
+import org.springframework.beans.factory.annotation.Value
+import org.springframework.context.annotation.Lazy
+import org.springframework.messaging.simp.stomp.StompFrameHandler
+import org.springframework.messaging.simp.stomp.StompHeaders
+import org.springframework.stereotype.Component
+import java.lang.reflect.Type
+
+@Component
+class FullScanHandler(
+    @Lazy @Autowired
+    private val cachingService: CachingService,
+    @Lazy @Autowired
+    private val gitHubStartService: GitHubStartService,
+    @Value("\${webhookEventService.waitingTime}") private val waitingTime: Long,
+) : StompFrameHandler {
+
+    private val logger = LoggerFactory.getLogger(FullScanHandler::class.java)
+
+    override fun getPayloadType(headers: StompHeaders): Type {
+        return MessagePayload::class.java
+    }
+
+    override fun handleFrame(headers: StompHeaders, payload: Any?) {
+        logger.info("Received full scan request")
+        runCatching {
+            while (cachingService.get("runId") != null) {
+                logger.info("A full scan is already in progress, waiting for it to finish.")
+                Thread.sleep(waitingTime)
+            }
+            GitHubStartService.requireScan = true
+            gitHubStartService.verifyAndStartScan()
+        }
+    }
+}
+
+data class MessagePayload(val message: String)

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

@@ -16,12 +16,18 @@ class GitHubStartService(
     private val cachingService: CachingService,
     private val syncLogService: SyncLogService
 ) {
-    var requireScan = true
+    companion object {
+        var requireScan: Boolean = true
+    }
 
     @SuppressWarnings("UnusedParameter")
     @Async
     @EventListener
     fun startAgent(connectionEstablishedEvent: ConnectionEstablishedEvent) {
+        verifyAndStartScan()
+    }
+
+    fun verifyAndStartScan() {
         if (requireScan) {
             runCatching {
                 requireScan = false