From 3a98b49556e48b912a949e8782846d38c8dc8eb2 Mon Sep 17 00:00:00 2001
From: Ilya Kirillov <ilya.kirillov@jetbrains.com>
Date: Wed, 29 May 2019 14:21:57 +0300
Subject: [PATCH] New J2K: fix type in ast conversion name

---
 .../kotlin/nj2k/ConversionsRunner.kt          |  2 +-
 .../conversions/DefaultArgumentsConversion.kt |  4 +-
 .../JavaStandardMethodsConversion.kt          | 75 +++++++++++++++++++
 3 files changed, 77 insertions(+), 4 deletions(-)
 create mode 100644 nj2k/src/org/jetbrains/kotlin/nj2k/conversions/JavaStandardMethodsConversion.kt

diff --git a/nj2k/src/org/jetbrains/kotlin/nj2k/ConversionsRunner.kt b/nj2k/src/org/jetbrains/kotlin/nj2k/ConversionsRunner.kt
index 92cab272061..3e9858406ca 100644
--- a/nj2k/src/org/jetbrains/kotlin/nj2k/ConversionsRunner.kt
+++ b/nj2k/src/org/jetbrains/kotlin/nj2k/ConversionsRunner.kt
@@ -58,7 +58,7 @@ object ConversionsRunner {
             +PrimaryConstructorDetectConversion(context)
             +InsertDefaultPrimaryConstructorConversion(context)
             +FieldToPropertyConversion()
-            +JavaStandartMethodsConversion(context)
+            +JavaStandardMethodsConversion(context)
             +JavaMethodToKotlinFunctionConversion(context)
             +MainFunctionConversion(context)
             +AssertStatementConversion(context)
diff --git a/nj2k/src/org/jetbrains/kotlin/nj2k/conversions/DefaultArgumentsConversion.kt b/nj2k/src/org/jetbrains/kotlin/nj2k/conversions/DefaultArgumentsConversion.kt
index 2354276df48..510ffe7e626 100644
--- a/nj2k/src/org/jetbrains/kotlin/nj2k/conversions/DefaultArgumentsConversion.kt
+++ b/nj2k/src/org/jetbrains/kotlin/nj2k/conversions/DefaultArgumentsConversion.kt
@@ -5,9 +5,7 @@
 
 package org.jetbrains.kotlin.nj2k.conversions
 
-import org.jetbrains.kotlin.nj2k.NewJ2kConverterContext
-import org.jetbrains.kotlin.nj2k.copyTreeAndDetach
-import org.jetbrains.kotlin.nj2k.jvmAnnotation
+import org.jetbrains.kotlin.nj2k.*
 import org.jetbrains.kotlin.nj2k.tree.*
 import org.jetbrains.kotlin.nj2k.tree.impl.JKFieldAccessExpressionImpl
 import org.jetbrains.kotlin.nj2k.tree.impl.JKUniverseMethodSymbol
diff --git a/nj2k/src/org/jetbrains/kotlin/nj2k/conversions/JavaStandardMethodsConversion.kt b/nj2k/src/org/jetbrains/kotlin/nj2k/conversions/JavaStandardMethodsConversion.kt
new file mode 100644
index 00000000000..2065c04ad11
--- /dev/null
+++ b/nj2k/src/org/jetbrains/kotlin/nj2k/conversions/JavaStandardMethodsConversion.kt
@@ -0,0 +1,75 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.nj2k.conversions
+
+import com.intellij.psi.PsiMethod
+import org.jetbrains.kotlin.idea.refactoring.fqName.getKotlinFqName
+import org.jetbrains.kotlin.nj2k.NewJ2kConverterContext
+import org.jetbrains.kotlin.j2k.ast.Nullability
+import org.jetbrains.kotlin.nj2k.tree.*
+import org.jetbrains.kotlin.nj2k.tree.impl.*
+
+class JavaStandardMethodsConversion(private val context: NewJ2kConverterContext) : RecursiveApplicableConversionBase() {
+    override fun applyToElement(element: JKTreeElement): JKTreeElement {
+        if (element !is JKClass) return recurse(element)
+        for (declaration in element.classBody.declarations) {
+            if (declaration !is JKJavaMethodImpl) continue
+            if (fixToStringMethod(declaration)) continue
+            if (fixFinalizeMethod(declaration, element)) continue
+            if (fixCloneMethod(declaration)) {
+                val hasNoCloneableInSuperClasses =
+                    declaration.psi<PsiMethod>()
+                        ?.findSuperMethods()
+                        ?.all { superMethod ->
+                            superMethod.containingClass?.getKotlinFqName()?.asString() == "java.lang.Object"
+                        } == true
+                if (hasNoCloneableInSuperClasses) {
+                    element.inheritance.implements +=
+                        JKTypeElementImpl(
+                            JKClassTypeImpl(
+                                JKUnresolvedClassSymbol("Cloneable"),
+                                emptyList(), Nullability.NotNull
+                            )
+                        )
+                }
+                continue
+            }
+        }
+        return recurse(element)
+    }
+
+    private fun fixToStringMethod(method: JKJavaMethodImpl): Boolean {
+        if (method.name.value != "toString") return false
+        if (method.parameters.isNotEmpty()) return false
+        val type = (method.returnType.type as? JKClassType)
+            ?.takeIf { it.classReference.name == "String" }
+            ?.updateNullability(Nullability.NotNull) ?: return false
+        method.returnType = JKTypeElementImpl(type)
+        return true
+    }
+
+    private fun fixCloneMethod(method: JKJavaMethodImpl): Boolean {
+        if (method.name.value != "clone") return false
+        if (method.parameters.isNotEmpty()) return false
+        val type = (method.returnType.type as? JKClassType)
+            ?.takeIf { it.classReference.name == "Object" }
+            ?.updateNullability(Nullability.NotNull) ?: return false
+        method.returnType = JKTypeElementImpl(type)
+        return true
+    }
+
+    private fun fixFinalizeMethod(method: JKJavaMethodImpl, containingClass: JKClass): Boolean {
+        if (method.name.value != "finalize") return false
+        if (method.parameters.isNotEmpty()) return false
+        if (method.returnType.type != JKJavaVoidType) return false
+        if (method.modality == Modality.OVERRIDE) {
+            method.modality =
+                if (containingClass.modality == Modality.OPEN) Modality.OPEN
+                else Modality.FINAL
+        }
+        return true
+    }
+}
\ No newline at end of file