ChangeFunctionReturnTypeFix for TYPE_MISMATCH error

idea/src/org/jetbrains/jet/plugin/quickfix/ChangeFunctionReturnTypeFix.java

@@ -185,4 +185,24 @@ public class ChangeFunctionReturnTypeFix extends JetIntentionAction<JetNamedFunc
             }
         };
     }
+
+    @NotNull
+    public static JetIntentionActionFactory createFactoryForTypeMismatch() {
+        return new JetIntentionActionFactory() {
+            @Nullable
+            @Override
+            public IntentionAction createAction(Diagnostic diagnostic) {
+                JetExpression expression = (JetExpression) diagnostic.getPsiElement();
+                JetNamedFunction function = QuickFixUtil.getParentElementOfType(diagnostic, JetNamedFunction.class);
+
+                if (function != null && (function.getInitializer() == expression || expression.getParent() instanceof JetReturnExpression)) {
+                    BindingContext context = AnalyzerFacadeWithCache.analyzeFileWithCache((JetFile) diagnostic.getPsiFile()).getBindingContext();
+                    JetType type = context.get(BindingContext.EXPRESSION_TYPE, expression);
+                    assert type != null : "Expression type mismatch, but expression has no type";
+                    return new ChangeFunctionReturnTypeFix(function, type);
+                }
+                return null;
+            }
+        };
+    }
 }

idea/src/org/jetbrains/jet/plugin/quickfix/QuickFixes.java

@@ -222,6 +222,7 @@ public class QuickFixes {
         factories.put(COMPONENT_FUNCTION_RETURN_TYPE_MISMATCH, ChangeFunctionReturnTypeFix.createFactoryForComponentFunctionReturnTypeMismatch());
         factories.put(HAS_NEXT_FUNCTION_TYPE_MISMATCH, ChangeFunctionReturnTypeFix.createFactoryForHasNextFunctionTypeMismatch());
         factories.put(COMPARE_TO_TYPE_MISMATCH, ChangeFunctionReturnTypeFix.createFactoryForCompareToTypeMismatch());
+        factories.put(TYPE_MISMATCH, ChangeFunctionReturnTypeFix.createFactoryForTypeMismatch());
 
         factories.put(EXPECTED_PARAMETER_TYPE_MISMATCH, ChangeTypeFix.createFactoryForExpectedParameterTypeMismatch());
         factories.put(EXPECTED_RETURN_TYPE_MISMATCH, ChangeTypeFix.createFactoryForExpectedReturnTypeMismatch());

idea/testData/quickfix/typeMismatch/afterTypeMismatchInInitializer.kt

@@ -0,0 +1,2 @@
+// "Change 'foo' function return type to 'String'" "true"
+fun foo(): String = <caret>""

idea/testData/quickfix/typeMismatch/afterTypeMismatchInReturnStatement.kt

@@ -0,0 +1,4 @@
+// "Change 'foo' function return type to 'String'" "true"
+fun foo(): String {
+    return ""<caret>
+}

idea/testData/quickfix/typeMismatch/beforeTypeMismatchInInitializer.kt

@@ -0,0 +1,2 @@
+// "Change 'foo' function return type to 'String'" "true"
+fun foo(): Int = <caret>""

idea/testData/quickfix/typeMismatch/beforeTypeMismatchInReturnStatement.kt

@@ -0,0 +1,4 @@
+// "Change 'foo' function return type to 'String'" "true"
+fun foo() {
+    return ""<caret>
+}

idea/tests/org/jetbrains/jet/plugin/quickfix/QuickFixTestGenerated.java

@@ -1063,6 +1063,16 @@ public class QuickFixTestGenerated extends AbstractQuickFixTest {
             doTest("idea/testData/quickfix/typeMismatch/beforeReturnTypeMismatch.kt");
         }
         
+        @TestMetadata("beforeTypeMismatchInInitializer.kt")
+        public void testTypeMismatchInInitializer() throws Exception {
+            doTest("idea/testData/quickfix/typeMismatch/beforeTypeMismatchInInitializer.kt");
+        }
+        
+        @TestMetadata("beforeTypeMismatchInReturnStatement.kt")
+        public void testTypeMismatchInReturnStatement() throws Exception {
+            doTest("idea/testData/quickfix/typeMismatch/beforeTypeMismatchInReturnStatement.kt");
+        }
+        
     }
     
     @TestMetadata("idea/testData/quickfix/typeProjection")