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
|
package com.intellij.structuralsearch.impl.matcher.handlers;
import com.intellij.dupLocator.iterators.NodeIterator;
import com.intellij.dupLocator.iterators.SiblingNodeIterator;
import com.intellij.psi.PsiComment;
import com.intellij.psi.PsiElement;
import com.intellij.structuralsearch.StructuralSearchUtil;
import com.intellij.structuralsearch.impl.matcher.MatchContext;
import com.intellij.structuralsearch.impl.matcher.iterators.SsrFilteringNodeIterator;
import org.jetbrains.annotations.NotNull;
import java.util.ArrayList;
import java.util.List;
public final class TopLevelMatchingHandler extends MatchingHandler implements DelegatingHandler {
private final MatchingHandler delegate;
public TopLevelMatchingHandler(@NotNull MatchingHandler _delegate) {
delegate = _delegate;
setFilter(_delegate.getFilter());
}
public boolean match(final PsiElement patternNode, final PsiElement matchedNode, final MatchContext matchContext) {
final boolean matched = delegate.match(patternNode, matchedNode, matchContext);
if (matched) {
List<PsiElement> matchedNodes = matchContext.getMatchedNodes();
if (matchedNodes == null) {
matchedNodes = new ArrayList<PsiElement>();
matchContext.setMatchedNodes(matchedNodes);
}
PsiElement elementToAdd = matchedNode;
if (patternNode instanceof PsiComment && StructuralSearchUtil.isDocCommentOwner(matchedNode)) {
// psicomment and psidoccomment are placed inside the psimember next to them so
// simple topdown matching should do additional "dances" to cover this case.
elementToAdd = matchedNode.getFirstChild();
assert elementToAdd instanceof PsiComment;
}
matchedNodes.add(elementToAdd);
}
if ((!matched || matchContext.getOptions().isRecursiveSearch()) &&
matchContext.getPattern().getStrategy().continueMatching(matchedNode) &&
matchContext.shouldRecursivelyMatch()
) {
matchContext.getMatcher().matchContext(
new SsrFilteringNodeIterator(
new SiblingNodeIterator(matchedNode.getFirstChild())
)
);
}
return matched;
}
@Override
public boolean canMatch(PsiElement patternNode, PsiElement matchedNode) {
return delegate.canMatch(patternNode, matchedNode);
}
@Override
public boolean matchSequentially(final NodeIterator nodes, final NodeIterator nodes2, final MatchContext context) {
return delegate.matchSequentially(nodes, nodes2, context);
}
public boolean match(final PsiElement patternNode,
final PsiElement matchedNode, final int start, final int end, final MatchContext context) {
return match(patternNode, matchedNode, context);
}
public boolean isMatchSequentiallySucceeded(final NodeIterator nodes2) {
return true;
}
@Override
public boolean shouldAdvanceTheMatchFor(final PsiElement patternElement, final PsiElement matchedElement) {
return delegate.shouldAdvanceTheMatchFor(patternElement, matchedElement);
}
public MatchingHandler getDelegate() {
return delegate;
}
}
|
