|
8 | 8 | import org.ice1000.devkt.openapi.AnnotationHolder;
|
9 | 9 | import org.ice1000.devkt.openapi.ColorScheme;
|
10 | 10 | import org.ice1000.devkt.openapi.ExtendedDevKtLanguage;
|
| 11 | +import org.ice1000.devkt.openapi.PsiUtils; |
11 | 12 | import org.jetbrains.annotations.NotNull;
|
12 | 13 | import org.jetbrains.annotations.Nullable;
|
13 | 14 | import org.jetbrains.kotlin.com.intellij.psi.PsiElement;
|
| 15 | +import org.jetbrains.kotlin.com.intellij.psi.TokenType; |
14 | 16 | import org.jetbrains.kotlin.com.intellij.psi.tree.IElementType;
|
15 | 17 |
|
16 | 18 | public class Solidity<T> extends ExtendedDevKtLanguage<T> {
|
@@ -45,13 +47,27 @@ public T attributesOf(IElementType type, ColorScheme<? extends T> colorScheme) {
|
45 | 47 | @Override
|
46 | 48 | public void annotate(PsiElement element, AnnotationHolder<? super T> document, ColorScheme<? extends T> colorScheme) {
|
47 | 49 | super.annotate(element, document, colorScheme);
|
48 |
| - if (element instanceof SolNumberLiteral) document.highlight(element, colorScheme.getNumbers()); |
49 |
| - else { |
| 50 | + if (element instanceof SolNumberLiteral) { |
| 51 | + document.highlight(element, colorScheme.getNumbers()); |
| 52 | + return; |
| 53 | + } |
| 54 | + function: |
| 55 | + { |
50 | 56 | PsiElement prevSibling = element.getPrevSibling();
|
51 |
| - if (prevSibling == null) return; |
52 |
| - if (prevSibling.getNode().getElementType() == SolidityTokenTypes.FUNCTION) { |
| 57 | + if (prevSibling == null) break function; |
| 58 | + if (PsiUtils.getNodeType(prevSibling) == TokenType.WHITE_SPACE) prevSibling = prevSibling.getPrevSibling(); |
| 59 | + if (prevSibling == null) break function; |
| 60 | + if (PsiUtils.getNodeType(prevSibling) == SolidityTokenTypes.FUNCTION) |
53 | 61 | document.highlight(element, colorScheme.getFunction());
|
54 |
| - } |
| 62 | + } |
| 63 | + clazz: |
| 64 | + { |
| 65 | + PsiElement prevSibling = element.getPrevSibling(); |
| 66 | + if (prevSibling == null) break clazz; |
| 67 | + if (PsiUtils.getNodeType(prevSibling) == TokenType.WHITE_SPACE) prevSibling = prevSibling.getPrevSibling(); |
| 68 | + if (prevSibling == null) break clazz; |
| 69 | + if (PsiUtils.getNodeType(prevSibling) == SolidityTokenTypes.CONTRACT) |
| 70 | + document.highlight(element, colorScheme.getNamespace()); |
55 | 71 | }
|
56 | 72 | }
|
57 | 73 |
|
|
0 commit comments