MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-10-06 06:21:16 +07:00
Code Issues Packages Projects Releases Wiki Activity

CPP-2906 Code Fold of control flow statements (if/else, do/while, for, switch)

Browse Source 
move java-code to shared package
This commit is contained in:
Alexey Utkin
2018-01-10 18:33:05 +03:00
parent 39f15eda94
commit 356657307b
1 changed files with 4 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
java/java-impl/src/com/intellij/codeInsight/folding/impl/CollapseBlockHandler.java
-99
View File
@@ -1,99 +0,0 @@
/*
* Copyright 2000-2015 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.codeInsight.folding.impl;
import com.intellij.codeInsight.CodeInsightActionHandler;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.FoldRegion;
import com.intellij.openapi.editor.ex.FoldingModelEx;
import com.intellij.openapi.project.Project;
import com.intellij.psi.*;
import com.intellij.psi.util.PsiTreeUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
/**
* @author ven
*/
public class CollapseBlockHandler implements CodeInsightActionHandler {
private static final String ourPlaceHolderText = "{...}";
private static final Logger LOG = Logger.getInstance("#com.intellij.codeInsight.folding.impl.CollapseBlockHandler");
@Override
public void invoke(@NotNull final Project project, @NotNull final Editor editor, @NotNull final PsiFile file) {
int[] targetCaretOffset = {-1};
editor.getFoldingModel().runBatchFoldingOperation(() -> {
final EditorFoldingInfo info = EditorFoldingInfo.get(editor);
FoldingModelEx model = (FoldingModelEx) editor.getFoldingModel();
PsiElement element = file.findElementAt(editor.getCaretModel().getOffset() - 1);
if (!(element instanceof PsiJavaToken) || ((PsiJavaToken) element).getTokenType() != JavaTokenType.RBRACE) {
element = file.findElementAt(editor.getCaretModel().getOffset());
}
if (element == null) return;
PsiCodeBlock block = PsiTreeUtil.getParentOfType(element, PsiCodeBlock.class);
FoldRegion previous = null;
FoldRegion myPrevious = null;
while (block != null) {
int start = block.getTextRange().getStartOffset();
int end = block.getTextRange().getEndOffset();
FoldRegion existing = FoldingUtil.findFoldRegion(editor, start, end);
if (existing != null) {
if (existing.isExpanded()) {
existing.setExpanded(false);
targetCaretOffset[0] = existing.getEndOffset();
return;
}
previous = existing;
if (info.getPsiElement(existing) == null) myPrevious = existing;
block = PsiTreeUtil.getParentOfType(block, PsiCodeBlock.class);
continue;
}
if (!model.intersectsRegion(start, end)) {
FoldRegion region = model.addFoldRegion(start, end, ourPlaceHolderText);
LOG.assertTrue(region != null);
region.setExpanded(false);
if (myPrevious != null && info.getPsiElement(region) == null) {
info.removeRegion(myPrevious);
model.removeFoldRegion(myPrevious);
}
targetCaretOffset[0] = block.getTextRange().getEndOffset() < editor.getCaretModel().getOffset() ? start : end;
return;
} else break;
}
if (previous != null) {
previous.setExpanded(false);
if (myPrevious != null) {
info.removeRegion(myPrevious);
model.removeFoldRegion(myPrevious);
}
targetCaretOffset[0] = previous.getEndOffset();
}
});
if (targetCaretOffset[0] >= 0) editor.getCaretModel().moveToOffset(targetCaretOffset[0]);
}
@Nullable
@Override
public PsiElement getElementToMakeWritable(@NotNull PsiFile currentFile) {
return null;
}
@Override
public boolean startInWriteAction() {
return true;
}
}