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 public class JavaForUnwrapper
extends JavaUnwrapper
{
23 public JavaForUnwrapper() {
24 super(CodeInsightBundle
.message("unwrap.for"));
27 public boolean isApplicableTo(PsiElement e
) {
28 return e
instanceof PsiForStatement
|| e
instanceof PsiForeachStatement
;
32 protected void doUnwrap(PsiElement element
, Context context
) throws IncorrectOperationException
{
33 if (element
instanceof PsiForStatement
) {
34 unwrapInitializer(element
, context
);
36 unwrapBody(element
, context
);
38 context
.delete(element
);
41 private void unwrapInitializer(PsiElement element
, Context context
) throws IncorrectOperationException
{
42 PsiStatement init
= ((PsiForStatement
)element
).getInitialization();
43 context
.extractFromBlockOrSingleStatement(init
, element
);
46 private void unwrapBody(PsiElement element
, Context context
) throws IncorrectOperationException
{
47 PsiStatement body
= ((PsiLoopStatement
)element
).getBody();
48 context
.extractFromBlockOrSingleStatement(body
, element
);