blob: b8aed59cc70ece5437d9e0bd0e59fd867ab8f9de (
plain)
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
85
86
87
88
89
90
91
92
93
94
95
|
/*
* Copyright 2000-2009 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.editorActions;
import com.intellij.psi.tree.TokenSet;
import com.intellij.psi.tree.IElementType;
import com.intellij.openapi.editor.highlighter.HighlighterIterator;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.Document;
public class SimpleTokenSetQuoteHandler implements QuoteHandler {
protected final TokenSet myLiteralTokenSet;
public SimpleTokenSetQuoteHandler(IElementType... _literalTokens) {
this(TokenSet.create(_literalTokens));
}
public SimpleTokenSetQuoteHandler(TokenSet tokenSet) {
myLiteralTokenSet = tokenSet;
}
@Override
public boolean isClosingQuote(HighlighterIterator iterator, int offset) {
final IElementType tokenType = iterator.getTokenType();
if (myLiteralTokenSet.contains(tokenType)){
int start = iterator.getStart();
int end = iterator.getEnd();
return end - start >= 1 && offset == end - 1;
}
return false;
}
@Override
public boolean isOpeningQuote(HighlighterIterator iterator, int offset) {
if (myLiteralTokenSet.contains(iterator.getTokenType())){
int start = iterator.getStart();
return offset == start;
}
return false;
}
@Override
public boolean hasNonClosedLiteral(Editor editor, HighlighterIterator iterator, int offset) {
int start = iterator.getStart();
try {
Document doc = editor.getDocument();
CharSequence chars = doc.getCharsSequence();
int lineEnd = doc.getLineEndOffset(doc.getLineNumber(offset));
while (!iterator.atEnd() && iterator.getStart() < lineEnd) {
IElementType tokenType = iterator.getTokenType();
if (myLiteralTokenSet.contains(tokenType)) {
if (isNonClosedLiteral(iterator, chars)) return true;
}
iterator.advance();
}
}
finally {
while(iterator.atEnd() || iterator.getStart() != start) iterator.retreat();
}
return false;
}
protected boolean isNonClosedLiteral(HighlighterIterator iterator, CharSequence chars) {
if (iterator.getStart() >= iterator.getEnd() - 1 ||
chars.charAt(iterator.getEnd() - 1) != '\"' && chars.charAt(iterator.getEnd() - 1) != '\'') {
return true;
}
return false;
}
@Override
public boolean isInsideLiteral(HighlighterIterator iterator) {
return myLiteralTokenSet.contains(iterator.getTokenType());
}
}
|
