void (@NotNull ActionContext context, @NotNull PsiParameterListOwner method, @NotNull ModPsiUpdater updater) { if (invokeSingleExpressionLambdaFix(method)) { return; } String value = suggestReturnValue(method); PsiElementFactory factory = JavaPsiFacade.getElementFactory(context.project()); PsiReturnStatement returnStatement = (PsiReturnStatement)factory.createStatementFromText("return " + value + ";", method); PsiCodeBlock body = ObjectUtils.tryCast(method.getBody(), PsiCodeBlock.class); assert body != null; returnStatement = (PsiReturnStatement) body.addBefore(returnStatement, body.getRBrace()); updater.select(Objects.requireNonNull(returnStatement.getReturnValue())); }