CID-3774: Move logic of processing installation event to backend

Author: mohamedlajmileanix

src/main/kotlin/net/leanix/githubagent/dto/InstallationEventPayload.kt

@@ -0,0 +1,6 @@
+package net.leanix.githubagent.dto
+
+data class InstallationRequestDTO(
+    val id: Long,
+    val account: Account,
+)

src/main/kotlin/net/leanix/githubagent/dto/InstallationRequestDTO.kt

@@ -15,6 +15,7 @@ import java.util.concurrent.CountDownLatch
 class BrokerStompSessionHandler(
     private val artifactDownloadHandler: ArtifactDownloadHandler,
     private val repositoryGetHandler: RepositoryGetHandler,
+    private val installationGetHandler: InstallationGetHandler,
 ) : StompSessionHandlerAdapter() {
     @Lazy
     @Autowired
@@ -32,6 +33,7 @@ class BrokerStompSessionHandler(
         latch.countDown()
         session.subscribe("/user/queue/message/artifact", artifactDownloadHandler)
         session.subscribe("/user/queue/message/repository", repositoryGetHandler)
+        session.subscribe("/user/queue/message/installation", installationGetHandler)
     }
 
     override fun handleException(

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

@@ -0,0 +1,34 @@
+package net.leanix.githubagent.handler
+
+import net.leanix.githubagent.dto.InstallationRequestDTO
+import net.leanix.githubagent.services.WebhookEventService
+import org.slf4j.LoggerFactory
+import org.springframework.beans.factory.annotation.Autowired
+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 InstallationGetHandler(
+    @Lazy @Autowired
+    private val webhookEventService: WebhookEventService
+) : StompFrameHandler {
+
+    private val logger = LoggerFactory.getLogger(InstallationGetHandler::class.java)
+
+    override fun getPayloadType(headers: StompHeaders): Type {
+        return InstallationRequestDTO::class.java
+    }
+
+    override fun handleFrame(headers: StompHeaders, payload: Any?) {
+        payload?.let {
+            val dto = payload as InstallationRequestDTO
+            logger.info("Received installation get message from server for organisation: ${dto.account.login}")
+            runCatching {
+                webhookEventService.handleInstallationCreated(dto)
+            }
+        }
+    }
+}

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

@@ -3,7 +3,7 @@ package net.leanix.githubagent.services
 import com.fasterxml.jackson.module.kotlin.jacksonObjectMapper
 import com.fasterxml.jackson.module.kotlin.readValue
 import net.leanix.githubagent.client.GitHubClient
-import net.leanix.githubagent.dto.InstallationEventPayload
+import net.leanix.githubagent.dto.InstallationRequestDTO
 import net.leanix.githubagent.dto.ManifestFileAction
 import net.leanix.githubagent.dto.ManifestFileUpdateDto
 import net.leanix.githubagent.dto.PushEventCommit
@@ -37,7 +37,6 @@ class WebhookEventService(
     fun consumeWebhookEvent(eventType: String, payload: String) {
         when (eventType.uppercase()) {
             "PUSH" -> handlePushEvent(payload)
-            "INSTALLATION" -> handleInstallationEvent(payload)
             else -> {
                 logger.info("Sending event of type: $eventType")
                 webSocketService.sendMessage("/events/other/$eventType", payload)
@@ -67,23 +66,16 @@ class WebhookEventService(
         }
     }
 
-    private fun handleInstallationEvent(payload: String) {
-        val installationEventPayload: InstallationEventPayload = objectMapper.readValue(payload)
-        if (installationEventPayload.action == "created") {
-            handleInstallationCreated(installationEventPayload)
-        }
-    }
-
-    private fun handleInstallationCreated(installationEventPayload: InstallationEventPayload) {
+    fun handleInstallationCreated(installationRequestDTO: InstallationRequestDTO) {
         while (cachingService.get("runId") != null) {
             logger.info("A full scan is already in progress, waiting for it to finish.")
             Thread.sleep(waitingTime)
         }
-        syncLogService.sendFullScanStart(installationEventPayload.installation.account.login)
+        syncLogService.sendFullScanStart(installationRequestDTO.account.login)
         kotlin.runCatching {
             val jwtToken = cachingService.get("jwtToken") ?: throw JwtTokenNotFound()
             val installation = gitHubClient.getInstallation(
-                installationEventPayload.installation.id.toLong(),
+                installationRequestDTO.id,
                 "Bearer $jwtToken"
             )
             gitHubEnterpriseService.validateEnabledPermissionsAndEvents(

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

@@ -8,18 +8,17 @@ import io.mockk.verify
 import net.leanix.githubagent.client.GitHubClient
 import net.leanix.githubagent.dto.Account
 import net.leanix.githubagent.dto.Installation
+import net.leanix.githubagent.dto.InstallationRequestDTO
 import net.leanix.githubagent.dto.ItemResponse
 import net.leanix.githubagent.dto.ManifestFileAction
 import net.leanix.githubagent.dto.ManifestFileUpdateDto
-import net.leanix.githubagent.dto.Organization
 import net.leanix.githubagent.dto.RepositoryItemResponse
 import net.leanix.githubagent.shared.MANIFEST_FILE_NAME
 import org.junit.jupiter.api.BeforeEach
 import org.junit.jupiter.api.Test
 import org.springframework.beans.factory.annotation.Autowired
 import org.springframework.boot.test.context.SpringBootTest
 import org.springframework.test.context.ActiveProfiles
-import java.util.UUID
 
 const val UNSUPPORTED_MANIFEST_EXTENSION = "leanix.yml"
 
@@ -419,19 +418,12 @@ class WebhookEventServiceTest {
         every { cachingService.set("runId", any(), any()) } just runs
         every { cachingService.remove("runId") } just runs
 
-        val eventType = "INSTALLATION"
-        val payload = """{
-          "action": "created",
-          "installation": {
-            "id": 30,
-            "account": {
-              "login": "test-org",
-              "id": 20
-            }
-          }
-        }"""
+        val installationRequestDTO = InstallationRequestDTO(
+            30,
+            Account("test-org")
+        )
 
-        webhookEventService.consumeWebhookEvent(eventType, payload)
+        webhookEventService.handleInstallationCreated(installationRequestDTO)
 
         verify(exactly = 6) { cachingService.get("runId") }
     }
@@ -456,37 +448,6 @@ class WebhookEventServiceTest {
         }
     }
 
-    @Test
-    fun `should send the org to the backend when an new installation is created`() {
-        val runId = UUID.randomUUID()
-        every { cachingService.get("runId") } returnsMany listOf("value", null, runId)
-        every { cachingService.set("runId", any(), any()) } just runs
-        every { cachingService.remove("runId") } just runs
-        every { gitHubAPIService.getPaginatedOrganizations(any()) } returns
-            listOf(Organization("testOrganization", 1))
-
-        val eventType = "INSTALLATION"
-        val payload = """{
-          "action": "created",
-          "installation": {
-            "id": 30,
-            "account": {
-              "login": "test-org",
-              "id": 20
-            }
-          }
-        }"""
-
-        webhookEventService.consumeWebhookEvent(eventType, payload)
-
-        verify {
-            webSocketService.sendMessage(
-                "$runId/organizations",
-                any()
-            )
-        }
-    }
-
     fun createItemResponse(repoName: String, organization: String): ItemResponse {
         return ItemResponse(
             MANIFEST_FILE_NAME,