diff --git a/plugins/lombok/src/main/java/de/plushnikov/intellij/plugin/handler/AddSneakyThrowsAnnotationCommandAction.java b/plugins/lombok/src/main/java/de/plushnikov/intellij/plugin/handler/AddSneakyThrowsAnnotationCommandAction.java index a5eab078e9aa..c89cdcba7df3 100644 --- a/plugins/lombok/src/main/java/de/plushnikov/intellij/plugin/handler/AddSneakyThrowsAnnotationCommandAction.java +++ b/plugins/lombok/src/main/java/de/plushnikov/intellij/plugin/handler/AddSneakyThrowsAnnotationCommandAction.java @@ -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 { - 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); } } }