From fb31740062566d0e9510fa5dda1c796b074660fa Mon Sep 17 00:00:00 2001
From: Phodal Huang <h@phodal.com>
Date: Thu, 24 Mar 2022 15:06:38 +0800
Subject: [PATCH] fix: fix open image files issues

---
 .../archguard/git/scanner/LanguageService.kt         |  8 ++++----
 .../archguard/git/scanner/LanguageServiceTest.kt     | 12 ++++++++++++
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/scan_git/src/main/kotlin/com/thoughtworks/archguard/git/scanner/LanguageService.kt b/scan_git/src/main/kotlin/com/thoughtworks/archguard/git/scanner/LanguageService.kt
index acae0727..b0a5a5f4 100644
--- a/scan_git/src/main/kotlin/com/thoughtworks/archguard/git/scanner/LanguageService.kt
+++ b/scan_git/src/main/kotlin/com/thoughtworks/archguard/git/scanner/LanguageService.kt
@@ -39,9 +39,9 @@ class LanguageService {
             return langs[0]
         }
 
-        var primaryLanguage = langs[0]
+        var primaryLanguage = ""
         langs.forEach {
-            if (languageMap[it]!!.keywords.isNullOrEmpty()) {
+            if (languageMap[it]?.keywords.isNullOrEmpty()) {
                 primaryLanguage = it
             }
         }
@@ -50,7 +50,7 @@ class LanguageService {
     }
 
     fun detectLanguages(name: String): List<String> {
-        var language: List<String> = listOf()
+        val language: MutableList<String> = mutableListOf()
 
         val dotCount = name.count { it == '.' }
 
@@ -65,7 +65,7 @@ class LanguageService {
             }
 
             // make others file a shebang
-            language = listOf(SHE_BANG)
+            language += SHE_BANG
         }
 
         // Lookup in case the full name matches
diff --git a/scan_git/src/test/kotlin/com/thoughtworks/archguard/git/scanner/LanguageServiceTest.kt b/scan_git/src/test/kotlin/com/thoughtworks/archguard/git/scanner/LanguageServiceTest.kt
index 8c32e56d..4392dd18 100644
--- a/scan_git/src/test/kotlin/com/thoughtworks/archguard/git/scanner/LanguageServiceTest.kt
+++ b/scan_git/src/test/kotlin/com/thoughtworks/archguard/git/scanner/LanguageServiceTest.kt
@@ -35,4 +35,16 @@ internal class LanguageServiceTest {
         val lang = LanguageService()
         assertEquals("d.ts", lang.getExtension("types.d.ts"))
     }
+
+    @Test
+    fun should_return_json_for_file() {
+        val lang = LanguageService()
+        assertEquals("JSON", lang.determineLanguage("api.json"))
+    }
+
+    @Test
+    fun should_return_empty_for_not_a_language_file() {
+        val lang = LanguageService()
+        assertEquals("", lang.determineLanguage("logo.png"))
+    }
 }
\ No newline at end of file