feat: add icons next to file names in the search result (#136)

* feat: add icons next to file names

* chore: add unit test for icon

* refactor: get instance of FileTypeManager on the class level

Author: jspmarc

src/main/kotlin/com/mituuz/fuzzier/FuzzyAction.kt

@@ -23,6 +23,7 @@
  */
 package com.mituuz.fuzzier
 
+import com.intellij.icons.AllIcons
 import com.intellij.openapi.actionSystem.*
 import com.intellij.openapi.components.service
 import com.intellij.openapi.editor.Caret
@@ -32,6 +33,7 @@ import com.intellij.openapi.editor.actionSystem.EditorActionHandler
 import com.intellij.openapi.editor.actionSystem.EditorActionManager
 import com.intellij.openapi.editor.event.DocumentEvent
 import com.intellij.openapi.editor.event.DocumentListener
+import com.intellij.openapi.fileTypes.FileTypeManager
 import com.intellij.openapi.keymap.KeymapManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.ui.popup.JBPopup
@@ -64,6 +66,7 @@ abstract class FuzzyAction : AnAction() {
     protected lateinit var projectState: FuzzierSettingsService.State
     protected val globalState = service<FuzzierGlobalSettingsService>().state
     protected var defaultDoc: Document? = null
+    private val fileTypeManager = FileTypeManager.getInstance()
 
     @Volatile
     var currentTask: Future<*>? = null
@@ -228,6 +231,10 @@ abstract class FuzzyAction : AnAction() {
                     false -> container.getDisplayString(globalState)
                 }
 
+                val ft = fileTypeManager.getFileTypeByFileName(container.filename)
+                val icon = ft.icon ?: AllIcons.FileTypes.Unknown
+                renderer.icon = icon
+
                 globalState.fileListSpacing.let {
                     renderer.border = BorderFactory.createEmptyBorder(it, 0, it, 0)
                 }

src/test/kotlin/com/mituuz/fuzzier/FuzzyActionTest.kt

@@ -26,6 +26,7 @@ package com.mituuz.fuzzier
 import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.actionSystem.IdeActions
 import com.intellij.openapi.editor.actionSystem.EditorActionManager
+import com.intellij.openapi.fileTypes.FileTypeManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.ui.popup.JBPopup
 import com.intellij.openapi.ui.popup.JBPopupFactory
@@ -168,6 +169,21 @@ class FuzzyActionTest {
         assertEquals("/src/asd", component.text)
     }
 
+    @Test
+    fun `Check renderer icon for different file types`() {
+        val action = getAction()
+        action.setFiletype(FILENAME_ONLY)
+        action.component = SimpleFinderComponent()
+        action.component.isDirSelector = true
+        val renderer = action.getCellRenderer()
+        val dummyList = JList<FuzzyMatchContainer>()
+        val container = FuzzyMatchContainer(FuzzyScore(), "/src/asd", "asd.kt", "")
+        val expectedIcon = FileTypeManager.getInstance().getFileTypeByFileName("asd.kt").icon
+        val component = renderer.getListCellRendererComponent(dummyList, container, 0, false, false) as JLabel
+        assertNotNull(component.icon)
+        assertEquals(expectedIcon, component.icon)
+    }
+
     private fun getAction(): FuzzyAction {
         return object : FuzzyAction() {
             override fun actionPerformed(actionEvent: AnActionEvent) {