[lombok] IDEA-255688 Review: Simplify annotation add QuickFix

GitOrigin-RevId: 9e9cd1a47074bee11c2eea589c6660902cf3f37f
This commit is contained in:
Michail Plushnikov
2023-12-14 15:53:43 +01:00
committed by intellij-monorepo-bot
parent 60f3b284cf
commit 953dbacb92

View File

@@ -1,7 +1,6 @@
package de.plushnikov.intellij.plugin.handler;
import com.intellij.codeInsight.ExceptionUtil;
import com.intellij.codeInsight.intention.AddAnnotationFix;
import com.intellij.codeInsight.intention.AddAnnotationPsiFix;
import com.intellij.java.analysis.JavaAnalysisBundle;
import com.intellij.modcommand.ActionContext;
@@ -9,10 +8,10 @@ import com.intellij.modcommand.ModPsiUpdater;
import com.intellij.modcommand.Presentation;
import com.intellij.modcommand.PsiUpdateModCommandAction;
import com.intellij.psi.*;
import com.intellij.psi.codeStyle.JavaCodeStyleManager;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.util.JavaPsiConstructorUtil;
import de.plushnikov.intellij.plugin.LombokClassNames;
import de.plushnikov.intellij.plugin.quickfix.PsiQuickFixFactory;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
@@ -21,7 +20,7 @@ import org.jetbrains.annotations.Nullable;
*/
public class AddSneakyThrowsAnnotationCommandAction extends PsiUpdateModCommandAction<PsiElement> {
public AddSneakyThrowsAnnotationCommandAction(PsiElement context) {
public AddSneakyThrowsAnnotationCommandAction(@NotNull PsiElement context) {
super(context);
}
@@ -52,8 +51,8 @@ public class AddSneakyThrowsAnnotationCommandAction extends PsiUpdateModCommandA
protected void invoke(@NotNull ActionContext context, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) {
PsiMethod psiMethod = PsiTreeUtil.getParentOfType(element, PsiMethod.class);
if (null != psiMethod) {
AddAnnotationFix fix = PsiQuickFixFactory.createAddAnnotationFix(LombokClassNames.SNEAKY_THROWS, psiMethod);
fix.applyFix();
final PsiAnnotation addedAnnotation = psiMethod.getModifierList().addAnnotation(LombokClassNames.SNEAKY_THROWS);
JavaCodeStyleManager.getInstance(element.getProject()).shortenClassReferences(addedAnnotation);
}
}
}