2 * Copyright 2000-2009 JetBrains s.r.o.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 package com
.intellij
.codeInsight
.unwrap
;
18 import com
.intellij
.codeInsight
.CodeInsightBundle
;
19 import com
.intellij
.psi
.*;
20 import com
.intellij
.util
.IncorrectOperationException
;
22 import java
.util
.List
;
24 public class JavaAnonymousUnwrapper
extends JavaUnwrapper
{
25 public JavaAnonymousUnwrapper() {
26 super(CodeInsightBundle
.message("unwrap.anonymous"));
29 public boolean isApplicableTo(PsiElement e
) {
30 return e
instanceof PsiAnonymousClass
31 && ((PsiAnonymousClass
)e
).getMethods().length
<= 1;
35 public PsiElement
collectAffectedElements(PsiElement e
, List
<PsiElement
> toExtract
) {
36 super.collectAffectedElements(e
, toExtract
);
37 return findElementToExtractFrom(e
);
41 protected void doUnwrap(PsiElement element
, Context context
) throws IncorrectOperationException
{
42 PsiElement from
= findElementToExtractFrom(element
);
44 for (PsiMethod m
: ((PsiAnonymousClass
)element
).getMethods()) {
45 context
.extractFromCodeBlock(m
.getBody(), from
);
48 PsiElement next
= from
.getNextSibling();
49 if (next
instanceof PsiJavaToken
&& ((PsiJavaToken
)next
).getTokenType() == JavaTokenType
.SEMICOLON
) {
50 context
.deleteExactly(from
.getNextSibling());
52 context
.deleteExactly(from
);
55 private PsiElement
findElementToExtractFrom(PsiElement el
) {
56 if (el
.getParent() instanceof PsiNewExpression
) el
= el
.getParent();
57 el
= findTopmostParentOfType(el
, PsiMethodCallExpression
.class);
58 el
= findTopmostParentOfType(el
, PsiAssignmentExpression
.class);
59 el
= findTopmostParentOfType(el
, PsiDeclarationStatement
.class);
61 while (el
.getParent() instanceof PsiExpressionStatement
) {