void (@NotNull ActionContext context, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) { PsiElement psiElement = element; CommentTracker ct = new CommentTracker(); JavaCodeStyleManager javaCodeStyleManager = JavaCodeStyleManager.getInstance(context.project()); if (psiElement instanceof PsiMethodReferenceExpression) { String replacement = CommonClassNames.JAVA_UTIL_OBJECTS + "::nonNull"; javaCodeStyleManager.shortenClassReferences(ct.replaceAndRestoreComments(psiElement, replacement)); return; } String nonNullExpression = null; if (psiElement instanceof PsiInstanceOfExpression instanceOf) { nonNullExpression = ct.text(instanceOf.getOperand()); } else if (psiElement instanceof PsiMethodCallExpression call) { PsiExpression arg = ArrayUtil.getFirstElement(call.getArgumentList().getExpressions()); if (arg == null) return; nonNullExpression = ct.text(arg); } if (nonNullExpression == null) return; PsiElement parent = PsiUtil.skipParenthesizedExprUp(psiElement.getParent()); String replacement; if (parent instanceof PsiExpression expression && BoolUtils.isNegation(expression)) { replacement = nonNullExpression + "==null"; psiElement = parent; } else { replacement = nonNullExpression + "!=null"; } javaCodeStyleManager.shortenClassReferences(ct.replaceAndRestoreComments(psiElement, replacement)); }