|
25 | 25 | import com.intellij.codeInsight.lookup.Lookup; |
26 | 26 | import com.intellij.codeInsight.lookup.LookupElement; |
27 | 27 | import com.intellij.codeInsight.lookup.LookupElementPresentation; |
| 28 | +import com.intellij.codeInsight.template.*; |
28 | 29 | import com.intellij.openapi.editor.Document; |
29 | 30 | import com.intellij.openapi.editor.Editor; |
30 | 31 | import com.intellij.openapi.project.Project; |
31 | 32 | import com.intellij.openapi.util.TextRange; |
32 | 33 | import com.intellij.psi.PsiDocumentManager; |
| 34 | +import com.intellij.psi.PsiElement; |
| 35 | +import com.intellij.psi.util.PsiTreeUtil; |
33 | 36 | import de.halirutan.mathematica.codeinsight.completion.SymbolInformationProvider.SymbolInformation; |
| 37 | +import de.halirutan.mathematica.parsing.psi.api.FunctionCall; |
| 38 | +import de.halirutan.mathematica.settings.MathematicaSettings; |
| 39 | +import de.halirutan.mathematica.settings.MathematicaSettings.SmartEnterResult; |
34 | 40 | import org.jetbrains.annotations.NotNull; |
| 41 | +import org.jetbrains.annotations.Nullable; |
35 | 42 |
|
36 | 43 | /** |
37 | 44 | * @author patrick (30.11.16). |
38 | 45 | */ |
| 46 | +@SuppressWarnings("AnonymousClassVariableHidesContainingMethodVariable") |
39 | 47 | public class BuiltinSymbolLookupElement extends LookupElement { |
40 | 48 |
|
41 | 49 | private final SymbolInformation myInfo; |
42 | 50 | private static final char OPEN_BRACKET = '['; |
43 | 51 | private static final char CLOSING_BRACKET = ']'; |
44 | 52 |
|
45 | | - public BuiltinSymbolLookupElement(SymbolInformation info) { |
| 53 | + BuiltinSymbolLookupElement(SymbolInformation info) { |
46 | 54 | myInfo = info; |
47 | 55 | } |
48 | 56 |
|
@@ -70,33 +78,78 @@ public boolean isCaseSensitive() { |
70 | 78 |
|
71 | 79 | @Override |
72 | 80 | public void handleInsert(InsertionContext context) { |
73 | | - Editor editor = context.getEditor(); |
74 | | - Document document = editor.getDocument(); |
75 | | - context.commitDocument(); |
| 81 | + final SmartEnterResult smartEnterSetting = MathematicaSettings.getInstance().getSmartEnterResult(); |
| 82 | + Editor editor = context.getEditor(); |
| 83 | + Document document = editor.getDocument(); |
| 84 | + context.commitDocument(); |
76 | 85 |
|
77 | | - char completionChar = context.getCompletionChar(); |
78 | | - context.setAddCompletionChar(false); |
| 86 | + char completionChar = context.getCompletionChar(); |
| 87 | + context.setAddCompletionChar(false); |
79 | 88 |
|
80 | | - if (completionChar == Lookup.COMPLETE_STATEMENT_SELECT_CHAR) { |
81 | | - if (myInfo.function) { |
| 89 | + if (completionChar == Lookup.COMPLETE_STATEMENT_SELECT_CHAR) { |
| 90 | + if (myInfo.function) { |
| 91 | + if (smartEnterSetting.equals(SmartEnterResult.INSERT_BRACES)) { |
| 92 | + document.insertString(context.getTailOffset(), "[]"); |
| 93 | + final int currentPosition = context.getTailOffset(); |
| 94 | + editor.getCaretModel().moveToOffset(currentPosition - 1); |
| 95 | + } else if (smartEnterSetting.equals(SmartEnterResult.INSERT_CODE) || smartEnterSetting.equals(SmartEnterResult.INSERT_TEMPLATE)) { |
82 | 96 | document.insertString(context.getTailOffset(), Character.toString(OPEN_BRACKET)); |
83 | 97 | final int currentPosition = context.getTailOffset(); |
84 | 98 | document.insertString(currentPosition, myInfo.getCallPattern()); |
85 | 99 | document.insertString(context.getTailOffset(), Character.toString(CLOSING_BRACKET)); |
86 | 100 | final int endOffset = getFirstArgumentRange(myInfo).getEndOffset() + currentPosition; |
87 | 101 | editor.getSelectionModel().setSelection(currentPosition, endOffset); |
88 | 102 | editor.getCaretModel().moveToOffset(endOffset); |
89 | | - } else { |
90 | | - document.insertString(context.getTailOffset(), " "); |
91 | | - editor.getCaretModel().moveToOffset(context.getTailOffset()); |
92 | | - } |
93 | 103 |
|
94 | | - } |
| 104 | + if (smartEnterSetting.equals(SmartEnterResult.INSERT_TEMPLATE)) { |
| 105 | + |
| 106 | + PsiDocumentManager.getInstance(context.getProject()).commitDocument(context.getDocument()); |
| 107 | + final PsiElement headOfFunction = PsiTreeUtil.findElementOfClassAtRange(context.getFile(), context.getStartOffset(), context.getTailOffset(), PsiElement.class); |
| 108 | + final TemplateBuilderFactory factory = TemplateBuilderFactoryImpl.getInstance(); |
| 109 | + final TemplateBuilderImpl builder = (TemplateBuilderImpl) factory.createTemplateBuilder(headOfFunction); |
| 110 | + if (headOfFunction instanceof FunctionCall) { |
| 111 | + final PsiElement[] children = headOfFunction.getChildren(); |
| 112 | + for (int i = 1; i < children.length; i++) { |
| 113 | + final PsiElement child = children[i]; |
| 114 | + |
| 115 | + builder.replaceElement(child, new Expression() { |
| 116 | + @Nullable |
| 117 | + @Override |
| 118 | + public Result calculateResult(ExpressionContext context) { |
| 119 | + return new TextResult(child.getText()); |
| 120 | + } |
| 121 | + |
| 122 | + @Nullable |
| 123 | + @Override |
| 124 | + public Result calculateQuickResult(ExpressionContext context) { |
| 125 | + return new TextResult(child.getText()); |
| 126 | + } |
| 127 | + |
| 128 | + @Nullable |
| 129 | + @Override |
| 130 | + public LookupElement[] calculateLookupItems(ExpressionContext context) { |
| 131 | + return LookupElement.EMPTY_ARRAY; |
| 132 | + } |
| 133 | + }); |
| 134 | + } |
| 135 | + Template template = builder.buildInlineTemplate(); |
| 136 | + TemplateManager.getInstance(context.getProject()).startTemplate(context.getEditor(), template); |
95 | 137 |
|
96 | | - if (completionChar == ' ') { |
97 | | - context.setAddCompletionChar(true); |
| 138 | + } |
| 139 | + |
| 140 | + } |
| 141 | + } |
| 142 | + } else { |
| 143 | + document.insertString(context.getTailOffset(), " "); |
| 144 | + editor.getCaretModel().moveToOffset(context.getTailOffset()); |
98 | 145 | } |
99 | 146 |
|
| 147 | + } |
| 148 | + |
| 149 | + if (completionChar == ' ') { |
| 150 | + context.setAddCompletionChar(true); |
| 151 | + } |
| 152 | + |
100 | 153 | final Project project = context.getProject(); |
101 | 154 | PsiDocumentManager.getInstance(project).commitDocument(document); |
102 | 155 | } |
|
0 commit comments