From 96ac29640d1b1a5501e61a1841323128fe11bc79 Mon Sep 17 00:00:00 2001
From: Carl-Robert Linnupuu <carlrobertoh@gmail.com>
Date: Fri, 26 Jul 2024 13:49:37 +0300
Subject: [PATCH] refactor: improve folder suggestions update strategy

---
 .../ui/textarea/SuggestionUpdateStrategy.kt   | 20 ++++---------------
 .../ui/textarea/SuggestionsPopupManager.kt    |  2 ++
 2 files changed, 6 insertions(+), 16 deletions(-)

diff --git a/src/main/kotlin/ee/carlrobert/codegpt/ui/textarea/SuggestionUpdateStrategy.kt b/src/main/kotlin/ee/carlrobert/codegpt/ui/textarea/SuggestionUpdateStrategy.kt
index 3e6f7a8d..fc044354 100644
--- a/src/main/kotlin/ee/carlrobert/codegpt/ui/textarea/SuggestionUpdateStrategy.kt
+++ b/src/main/kotlin/ee/carlrobert/codegpt/ui/textarea/SuggestionUpdateStrategy.kt
@@ -107,7 +107,10 @@ class FolderSuggestionActionStrategy : SuggestionUpdateStrategy {
                 .filter { path ->
                     val file = virtualFileManager.findFileByNioPath(path)
                     val isProjectFile =
-                        file != null && runReadAction { projectFileIndex.isInContent(file) }
+                        file != null && runReadAction {
+                            projectFileIndex.isInSourceContent(file)
+                                    || projectFileIndex.isInTestSourceContent(file)
+                        }
                     path.isDirectory() && !path.startsWith(".") && isProjectFile
                 }
                 .forEach { folder ->
@@ -147,21 +150,6 @@ class PersonaSuggestionActionStrategy : SuggestionUpdateStrategy {
     }
 }
 
-class CreatePersonaActionStrategy : SuggestionUpdateStrategy {
-    override fun populateSuggestions(
-        project: Project,
-        listModel: DefaultListModel<SuggestionItem>,
-    ) {
-    }
-
-    override fun updateSuggestions(
-        project: Project,
-        listModel: DefaultListModel<SuggestionItem>,
-        searchText: String,
-    ) {
-    }
-}
-
 class DefaultSuggestionActionStrategy : SuggestionUpdateStrategy {
     override fun populateSuggestions(
         project: Project,
diff --git a/src/main/kotlin/ee/carlrobert/codegpt/ui/textarea/SuggestionsPopupManager.kt b/src/main/kotlin/ee/carlrobert/codegpt/ui/textarea/SuggestionsPopupManager.kt
index 1f016507..6241bb4e 100644
--- a/src/main/kotlin/ee/carlrobert/codegpt/ui/textarea/SuggestionsPopupManager.kt
+++ b/src/main/kotlin/ee/carlrobert/codegpt/ui/textarea/SuggestionsPopupManager.kt
@@ -114,6 +114,8 @@ class SuggestionsPopupManager(
 
     fun updateSuggestions(searchText: String) {
         currentActionStrategy.updateSuggestions(project, listModel, searchText)
+        list.revalidate()
+        list.repaint()
     }
 
     fun reset(clearPrevious: Boolean = true) {