vrr/ProxyAI
2
0
Fork 0
mirror of https://github.com/carlrobertoh/ProxyAI.git synced 2026-05-20 17:52:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: include unsaved content in prompt

Browse source
This commit is contained in:
carlrobertoh@gmail.com 2025-02-10 18:42:04 +00:00
parent 1a3fbdd94a
commit 7cd9644932
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/main/java/ee/carlrobert/codegpt/ReferencedFile.java
View file

@ -1,5 +1,6 @@
package ee.carlrobert.codegpt;
import com.intellij.openapi.fileEditor.FileDocumentManager;
import com.intellij.openapi.vfs.VfsUtilCore;
import com.intellij.openapi.vfs.VirtualFile;
import java.io.File;
@ -24,10 +25,19 @@ public record ReferencedFile(String fileName, String filePath, String fileConten
return new ReferencedFile(
virtualFile.getName(),
virtualFile.getPath(),
readContent(virtualFile)
getVirtualFileContent(virtualFile)
);
}
private static String getVirtualFileContent(VirtualFile virtualFile) {
var documentManager = FileDocumentManager.getInstance();
var document = documentManager.getDocument(virtualFile);
if (document != null && documentManager.isDocumentUnsaved(document)) {
return document.getText();
}
return readContent(virtualFile);
}
private static String readContent(File file) {
try {
return new String(Files.readAllBytes(Paths.get(file.getPath())));