3 * Copyright 2000-2009 JetBrains s.r.o.
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
17 package com
.intellij
.codeInsight
.generation
.surroundWith
;
19 import com
.intellij
.codeInsight
.CodeInsightBundle
;
20 import com
.intellij
.openapi
.editor
.Editor
;
21 import com
.intellij
.openapi
.project
.Project
;
22 import com
.intellij
.openapi
.util
.TextRange
;
23 import com
.intellij
.psi
.*;
24 import com
.intellij
.psi
.codeStyle
.CodeStyleManager
;
25 import com
.intellij
.util
.IncorrectOperationException
;
26 import org
.jetbrains
.annotations
.NonNls
;
28 class JavaWithWhileSurrounder
extends JavaStatementsSurrounder
{
29 public String
getTemplateDescription() {
30 return CodeInsightBundle
.message("surround.with.while.template");
33 public TextRange
surroundStatements(Project project
, Editor editor
, PsiElement container
, PsiElement
[] statements
) throws IncorrectOperationException
{
34 PsiManager manager
= PsiManager
.getInstance(project
);
35 PsiElementFactory factory
= JavaPsiFacade
.getInstance(manager
.getProject()).getElementFactory();
36 CodeStyleManager codeStyleManager
= CodeStyleManager
.getInstance(project
);
38 statements
= SurroundWithUtil
.moveDeclarationsOut(container
, statements
, true);
39 if (statements
.length
== 0){
43 @NonNls String text
= "while(true){\n}";
44 PsiWhileStatement whileStatement
= (PsiWhileStatement
)factory
.createStatementFromText(text
, null);
45 whileStatement
= (PsiWhileStatement
)codeStyleManager
.reformat(whileStatement
);
47 whileStatement
= (PsiWhileStatement
)container
.addAfter(whileStatement
, statements
[statements
.length
- 1]);
49 PsiCodeBlock bodyBlock
= ((PsiBlockStatement
)whileStatement
.getBody()).getCodeBlock();
50 bodyBlock
.addRange(statements
[0], statements
[statements
.length
- 1]);
51 container
.deleteChildRange(statements
[0], statements
[statements
.length
- 1]);
53 return whileStatement
.getCondition().getTextRange();