Skip to content

Commit be7f48d

Browse files
The entire file was added in the context. This was done in the MessageCreationService.createAttachedFilesContext. An assumption was made that the virtual file containing the snippet was of filetype unknown.
Signed-off-by: mydeveloperplanet <[email protected]>
1 parent 2be102c commit be7f48d

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

src/main/java/com/devoxx/genie/action/AddSnippetAction.java

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@
1313
import com.intellij.openapi.project.Project;
1414
import com.intellij.openapi.util.Key;
1515
import com.intellij.openapi.vfs.VirtualFile;
16+
import com.intellij.testFramework.LightVirtualFile;
17+
1618
import lombok.extern.slf4j.Slf4j;
1719
import org.jetbrains.annotations.NotNull;
1820
import org.jetbrains.annotations.Nullable;
@@ -91,18 +93,21 @@ private void createAndAddVirtualFile(@Nullable Project project,
9193
SelectionModel selectionModel = editor.getSelectionModel();
9294
Document document = editor.getDocument();
9395

96+
LightVirtualFile virtualFile = new LightVirtualFile(originalFile.getName(), selectedText);
97+
virtualFile.setFileType(fileType);
98+
9499
int startOffset = selectionModel.getSelectionStart();
95100
int endOffset = selectionModel.getSelectionEnd();
96101
int startLine = document.getLineNumber(startOffset);
97102
int endLine = document.getLineNumber(endOffset);
98-
originalFile.putUserData(ORIGINAL_FILE_KEY, originalFile);
99-
originalFile.putUserData(SELECTED_TEXT_KEY, selectedText);
100-
originalFile.putUserData(SELECTION_START_KEY, startOffset);
101-
originalFile.putUserData(SELECTION_END_KEY, endOffset);
102-
originalFile.putUserData(SELECTION_START_LINE_KEY, startLine);
103-
originalFile.putUserData(SELECTION_END_LINE_KEY, endLine);
104-
105-
FileListManager.getInstance().addFile(project, originalFile);
103+
virtualFile.putUserData(ORIGINAL_FILE_KEY, originalFile);
104+
virtualFile.putUserData(SELECTED_TEXT_KEY, selectedText);
105+
virtualFile.putUserData(SELECTION_START_KEY, startOffset);
106+
virtualFile.putUserData(SELECTION_END_KEY, endOffset);
107+
virtualFile.putUserData(SELECTION_START_LINE_KEY, startLine);
108+
virtualFile.putUserData(SELECTION_END_LINE_KEY, endLine);
109+
110+
FileListManager.getInstance().addFile(project, virtualFile);
106111
}
107112

108113
@Override

0 commit comments

Comments
 (0)