IG: remove newline after modifier list if necessary (IDEA-162989)
[idea/community.git] / plugins / InspectionGadgets / InspectionGadgetsAnalysis / src / com / siyeh / ig / junit / MakePublicStaticFix.java
1 /*
2  * Copyright 2000-2016 JetBrains s.r.o.
3  *
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
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
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.
15  */
16 package com.siyeh.ig.junit;
17
18 import com.intellij.codeInspection.ProblemDescriptor;
19 import com.intellij.openapi.project.Project;
20 import com.intellij.psi.*;
21 import com.intellij.psi.codeStyle.CodeStyleManager;
22 import com.intellij.psi.codeStyle.JavaCodeStyleManager;
23 import com.intellij.psi.codeStyle.JavaCodeStyleSettingsFacade;
24 import com.intellij.psi.util.PsiUtil;
25 import com.intellij.util.IncorrectOperationException;
26 import com.siyeh.ig.InspectionGadgetsFix;
27 import org.jetbrains.annotations.NotNull;
28
29 /**
30 * User: anna
31 * Date: 5/22/13
32 */
33 class MakePublicStaticFix extends InspectionGadgetsFix {
34
35   private final String myName;
36   private final boolean myMakeStatic;
37
38   public MakePublicStaticFix(final String name, final boolean makeStatic) {
39     myName = name;
40     myMakeStatic = makeStatic;
41   }
42
43   @Override
44   protected void doFix(Project project, ProblemDescriptor descriptor) {
45     final PsiElement element = descriptor.getPsiElement();
46     if (element == null) {
47       return;
48     }
49     final PsiElement parent = element.getParent();
50     if (!(parent instanceof PsiMember)) {
51       return;
52     }
53     final PsiMember member = (PsiMember)parent;
54     final PsiModifierList modifierList = member.getModifierList();
55     if (modifierList == null) {
56       return;
57     }
58     modifierList.setModifierProperty(PsiModifier.PUBLIC, true);
59     modifierList.setModifierProperty(PsiModifier.STATIC, myMakeStatic);
60     final PsiElement sibling = modifierList.getNextSibling();
61     if (sibling instanceof PsiWhiteSpace && sibling.getText().contains("\n")) {
62       sibling.replace(PsiParserFacade.SERVICE.getInstance(project).createWhiteSpaceFromText(" "));
63     }
64   }
65
66   @NotNull
67   @Override
68   public String getName() {
69     return myName;
70   }
71
72   @NotNull
73   @Override
74   public String getFamilyName() {
75     return "Make public/static";
76   }
77 }