Merge pull request #22 from leanix/feature/CID-2885/send-GH-app-name-…

…to-backend

CID-2885: Send GitHub App name to backend

src/main/kotlin/net/leanix/githubagent/runners/PostStartupRunner.kt

@@ -1,8 +1,11 @@
 package net.leanix.githubagent.runners
 
+import net.leanix.githubagent.services.CachingService
 import net.leanix.githubagent.services.GitHubAuthenticationService
+import net.leanix.githubagent.services.GitHubEnterpriseService
 import net.leanix.githubagent.services.GitHubScanningService
 import net.leanix.githubagent.services.WebSocketService
+import net.leanix.githubagent.shared.AGENT_METADATA_TOPIC
 import org.springframework.boot.ApplicationArguments
 import org.springframework.boot.ApplicationRunner
 import org.springframework.context.annotation.Profile
@@ -13,12 +16,19 @@ import org.springframework.stereotype.Component
 class PostStartupRunner(
     private val githubAuthenticationService: GitHubAuthenticationService,
     private val webSocketService: WebSocketService,
-    private val gitHubScanningService: GitHubScanningService
+    private val gitHubScanningService: GitHubScanningService,
+    private val gitHubEnterpriseService: GitHubEnterpriseService,
+    private val cachingService: CachingService
 ) : ApplicationRunner {
 
     override fun run(args: ApplicationArguments?) {
         webSocketService.initSession()
         githubAuthenticationService.generateAndCacheJwtToken()
+        val jwt = cachingService.get("jwt") as String
+        webSocketService.sendMessage(
+            AGENT_METADATA_TOPIC,
+            gitHubEnterpriseService.getGitHubApp(jwt).name
+        )
         gitHubScanningService.scanGitHubResources()
     }
 }

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

@@ -18,7 +18,7 @@ class GitHubEnterpriseService(private val githubClient: GitHubClient) {
 
     fun verifyJwt(jwt: String) {
         runCatching {
-            val githubApp = githubClient.getApp("Bearer $jwt")
+            val githubApp = getGitHubApp(jwt)
             validateGithubAppResponse(githubApp)
             logger.info("Authenticated as GitHub App: '${githubApp.name}'")
         }.onFailure {
@@ -48,4 +48,6 @@ class GitHubEnterpriseService(private val githubClient: GitHubClient) {
             throw GitHubAppInsufficientPermissionsException(message)
         }
     }
+
+    fun getGitHubApp(jwt: String) = githubClient.getApp("Bearer $jwt")
 }

src/main/kotlin/net/leanix/githubagent/shared/Constants.kt

@@ -1,6 +1,7 @@
 package net.leanix.githubagent.shared
 
 const val TOPIC_PREFIX = "/app/ghe/"
+const val AGENT_METADATA_TOPIC = "agent"
 
 enum class ManifestFileName(val fileName: String) {
     YAML("leanix.yaml"),

src/test/kotlin/net/leanix/githubagent/runners/PostStartupRunnerTest.kt

@@ -0,0 +1,60 @@
+package net.leanix.githubagent.runners
+
+import io.mockk.every
+import io.mockk.mockk
+import io.mockk.verify
+import net.leanix.githubagent.dto.GitHubAppResponse
+import net.leanix.githubagent.services.CachingService
+import net.leanix.githubagent.services.GitHubAuthenticationService
+import net.leanix.githubagent.services.GitHubEnterpriseService
+import net.leanix.githubagent.services.GitHubScanningService
+import net.leanix.githubagent.services.WebSocketService
+import net.leanix.githubagent.shared.AGENT_METADATA_TOPIC
+import org.junit.jupiter.api.BeforeEach
+import org.junit.jupiter.api.Test
+
+class PostStartupRunnerTest {
+
+    private lateinit var githubAuthenticationService: GitHubAuthenticationService
+    private lateinit var webSocketService: WebSocketService
+    private lateinit var gitHubScanningService: GitHubScanningService
+    private lateinit var gitHubEnterpriseService: GitHubEnterpriseService
+    private lateinit var cachingService: CachingService
+    private lateinit var postStartupRunner: PostStartupRunner
+
+    @BeforeEach
+    fun setUp() {
+        githubAuthenticationService = mockk()
+        webSocketService = mockk()
+        gitHubScanningService = mockk()
+        gitHubEnterpriseService = mockk()
+        cachingService = mockk()
+
+        postStartupRunner = PostStartupRunner(
+            githubAuthenticationService,
+            webSocketService,
+            gitHubScanningService,
+            gitHubEnterpriseService,
+            cachingService
+        )
+
+        every { webSocketService.initSession() } returns Unit
+        every { webSocketService.sendMessage(any(), any()) } returns Unit
+        every { githubAuthenticationService.generateAndCacheJwtToken() } returns Unit
+        every { cachingService.get("jwt") } returns "jwt"
+        every { gitHubScanningService.scanGitHubResources() } returns Unit
+    }
+
+    @Test
+    fun `should send GitHub App name`() {
+        val gitHubAppName = "appName"
+        every { gitHubEnterpriseService.getGitHubApp("jwt") } returns
+            GitHubAppResponse(
+                gitHubAppName, mapOf(), listOf()
+            )
+
+        postStartupRunner.run(mockk())
+
+        verify { webSocketService.sendMessage(AGENT_METADATA_TOPIC, gitHubAppName) }
+    }
+}