From 779dc5c1e2e49001ebcfcce5b981b9201c02f897 Mon Sep 17 00:00:00 2001
From: Ilya Kirillov <ilya.kirillov@jetbrains.com>
Date: Wed, 13 Feb 2019 13:26:57 +0300
Subject: [PATCH] New J2K: Add RETURN_TYPE_MISMATCH_ON_OVERRIDE to
 postprocessing

---
 .../src/org/jetbrains/kotlin/nj2k/NewJ2kPostProcessings.kt | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/nj2k/nj2k-services/src/org/jetbrains/kotlin/nj2k/NewJ2kPostProcessings.kt b/nj2k/nj2k-services/src/org/jetbrains/kotlin/nj2k/NewJ2kPostProcessings.kt
index e83ede2a8d3..21c1385f44e 100644
--- a/nj2k/nj2k-services/src/org/jetbrains/kotlin/nj2k/NewJ2kPostProcessings.kt
+++ b/nj2k/nj2k-services/src/org/jetbrains/kotlin/nj2k/NewJ2kPostProcessings.kt
@@ -169,6 +169,13 @@ object NewJ2KPostProcessingRegistrar {
                         ?: return@registerDiagnosticBasedProcessing
                 action.invoke(element.project, null, element.containingKtFile)
             },
+            registerDiagnosticBasedProcessing(Errors.RETURN_TYPE_MISMATCH_ON_OVERRIDE) { element: KtNamedDeclaration, diagnostic ->
+                val action =
+                    ChangeCallableReturnTypeFix.ReturnTypeMismatchOnOverrideFactory
+                        .createActionsForAllProblems(listOf(diagnostic)).singleOrNull()
+                        ?: return@registerDiagnosticBasedProcessing
+                action.invoke(element.project, null, element.containingKtFile)
+            },
 
             RemoveRedundantTypeQualifierProcessing(),
             RemoveRedundantExpressionQualifierProcessing(),