1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
// Copyright 2000-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.intellij.json.codeinsight;
import com.intellij.codeInsight.AutoPopupController;
import com.intellij.codeInsight.completion.InsertHandler;
import com.intellij.codeInsight.completion.InsertionContext;
import com.intellij.codeInsight.lookup.LookupElement;
import com.intellij.json.JsonElementTypes;
import com.intellij.json.psi.JsonProperty;
import com.intellij.json.psi.JsonStringLiteral;
import com.intellij.openapi.util.TextRange;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.PsiDocumentManager;
import com.intellij.psi.PsiElement;
import com.intellij.psi.codeStyle.CodeStyleManager;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.util.ObjectUtils;
import org.jetbrains.annotations.NotNull;
public class JsonStringPropertyInsertHandler implements InsertHandler<LookupElement> {
private final String myNewValue;
public JsonStringPropertyInsertHandler(@NotNull String newValue) {
myNewValue = newValue;
}
@Override
public void handleInsert(@NotNull InsertionContext context, @NotNull LookupElement item) {
PsiElement element = context.getFile().findElementAt(context.getStartOffset());
JsonStringLiteral literal = PsiTreeUtil.getParentOfType(element, JsonStringLiteral.class, false);
if (literal == null) return;
JsonProperty property = ObjectUtils.tryCast(literal.getParent(), JsonProperty.class);
if (property == null) return;
final TextRange toDelete;
String textToInsert = "";
TextRange literalRange = literal.getTextRange();
if (literal.getValue().equals(myNewValue)) {
toDelete = new TextRange(literalRange.getEndOffset(), literalRange.getEndOffset());
}
else {
toDelete = literalRange;
textToInsert = StringUtil.wrapWithDoubleQuote(myNewValue);
}
int newCaretOffset = literalRange.getStartOffset() + 1 + myNewValue.length();
boolean showAutoPopup = false;
if (property.getNameElement().equals(literal)) {
if (property.getValue() == null) {
textToInsert += ":\"\"";
newCaretOffset += 3; // "package<caret offset>":"<new caret offset>"
if (needCommaAfter(property)) {
textToInsert += ",";
}
showAutoPopup = true;
}
}
context.getDocument().replaceString(toDelete.getStartOffset(), toDelete.getEndOffset(), textToInsert);
context.getEditor().getCaretModel().moveToOffset(newCaretOffset);
reformat(context, toDelete.getStartOffset(), toDelete.getStartOffset() + textToInsert.length());
if (showAutoPopup) {
AutoPopupController.getInstance(context.getProject()).autoPopupMemberLookup(context.getEditor(), null);
}
}
private static boolean needCommaAfter(@NotNull JsonProperty property) {
PsiElement element = property.getNextSibling();
while (element != null) {
if (element instanceof JsonProperty) {
return true;
}
if (element.getNode().getElementType() == JsonElementTypes.COMMA) {
return false;
}
element = element.getNextSibling();
}
return false;
}
private static void reformat(@NotNull InsertionContext context, int startOffset, int endOffset) {
PsiDocumentManager.getInstance(context.getProject()).commitDocument(context.getDocument());
CodeStyleManager codeStyleManager = CodeStyleManager.getInstance(context.getProject());
codeStyleManager.reformatText(context.getFile(), startOffset, endOffset);
}
}
|