Skip to content

Commit 352b459

Browse files
committed
Fix IDE compatibility
1 parent af2ffe7 commit 352b459

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

src/main/kotlin/ai/devchat/plugin/completion/agent/ContextBuilder.kt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ import ai.devchat.common.IDEUtils.foldTextOfLevel
77
import ai.devchat.common.IDEUtils.runInEdtAndGet
88
import ai.devchat.common.Log
99
import ai.devchat.storage.RecentFilesTracker
10-
import com.intellij.openapi.vfs.isFile
1110
import com.intellij.psi.PsiFile
1211
import com.intellij.psi.util.PsiUtilCore.getPsiFile
1312

@@ -157,7 +156,7 @@ class ContextBuilder(val file: PsiFile, val offset: Int) {
157156
val project = file.project
158157
return runInEdtAndGet {
159158
project.getService(RecentFilesTracker::class.java).getRecentFiles().asSequence()
160-
.filter { it.isFile && it.path != filepath }
159+
.filter { it.isValid && !it.isDirectory && it.path != filepath }
161160
.map { CodeSnippet(it.path, getPsiFile(project, it).foldTextOfLevel(2)) }
162161
.filter { it.content.lines().count(String::isBlank) <= 50 }
163162
.takeWhile(::checkAndUpdateTokenCount)

src/main/kotlin/ai/devchat/storage/RecentFilesTracker.kt

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ import com.intellij.openapi.project.Project
1010
import com.intellij.openapi.roots.ProjectFileIndex
1111
import com.intellij.openapi.startup.ProjectActivity
1212
import com.intellij.openapi.vfs.VirtualFile
13-
import com.intellij.openapi.vfs.isFile
1413
import com.intellij.util.messages.MessageBusConnection
1514

1615

@@ -26,7 +25,7 @@ class RecentFilesTracker(private val project: Project) {
2625
val connection: MessageBusConnection = project.messageBus.connect()
2726
connection.subscribe(FileEditorManagerListener.FILE_EDITOR_MANAGER, object : FileEditorManagerListener {
2827
override fun fileOpened(source: FileEditorManager, file: VirtualFile) {
29-
if (file.isFile) {
28+
if (file.isValid && !file.isDirectory) {
3029
addRecentFile(file)
3130
}
3231
}
@@ -37,7 +36,7 @@ class RecentFilesTracker(private val project: Project) {
3736
}
3837

3938
private fun addRecentFile(file: VirtualFile) = runInEdt {
40-
if (file.isFile && projectFileIndex.isInContent(file)) {
39+
if (file.isValid && !file.isDirectory && projectFileIndex.isInContent(file)) {
4140
recentFiles.remove(file)
4241
recentFiles.add(0, file)
4342
if (recentFiles.size > maxSize) {

0 commit comments

Comments
 (0)