File tree Expand file tree Collapse file tree 2 files changed +16
-6
lines changed
src/main/java/ee/carlrobert/codegpt/ui/checkbox Expand file tree Collapse file tree 2 files changed +16
-6
lines changed Original file line number Diff line number Diff line change @@ -24,11 +24,16 @@ public PsiElementCheckboxTree(@NotNull PsiElement rootElement) {
2424 }
2525
2626 public List <CheckedFile > getCheckedFiles () {
27- return Arrays .stream (getCheckedNodes (
28- PsiElement .class ,
29- node -> Optional .ofNullable (node .getContainingFile ())
30- .map (PsiFile ::getVirtualFile )
31- .isPresent ()))
27+ var checkedNodes = getCheckedNodes (
28+ PsiElement .class ,
29+ node -> Optional .ofNullable (node .getContainingFile ())
30+ .map (PsiFile ::getVirtualFile )
31+ .isPresent ());
32+ if (checkedNodes .length > 1000 ) {
33+ throw new RuntimeException ("Too many files selected" );
34+ }
35+
36+ return Arrays .stream (checkedNodes )
3237 .map (item -> new CheckedFile (new File (item .getContainingFile ().getVirtualFile ().getPath ())))
3338 .collect (toList ());
3439 }
Original file line number Diff line number Diff line change @@ -20,7 +20,12 @@ public VirtualFileCheckboxTree(@NotNull VirtualFile[] rootFiles) {
2020 }
2121
2222 public List <CheckedFile > getCheckedFiles () {
23- return Arrays .stream (getCheckedNodes (VirtualFile .class , Objects ::nonNull ))
23+ var checkedNodes = getCheckedNodes (VirtualFile .class , Objects ::nonNull );
24+ if (checkedNodes .length > 1000 ) {
25+ throw new RuntimeException ("Too many files selected" );
26+ }
27+
28+ return Arrays .stream (checkedNodes )
2429 .map (item -> new CheckedFile (new File (item .getPath ())))
2530 .collect (toList ());
2631 }
You can’t perform that action at this time.
0 commit comments