From 4e04d9b8874fdbc8a811a3897f6d637ba288b598 Mon Sep 17 00:00:00 2001
From: Nikolay Krasko <Nikolay.Krasko@jetbrains.com>
Date: Tue, 24 Mar 2015 01:16:35 +0300
Subject: [PATCH] Add more toString() methods and additional wrapping for
 Kotlin elements

---
 .../kotlin/asJava/KotlinLightField.java       |  2 +-
 .../asJava/KotlinLightMethodForDeclaration.kt |  2 ++
 .../asJava/KotlinLightNoOriginElements.kt     | 34 +++++++++++++++++++
 .../asJava/KotlinWrappingLightClass.java      |  6 ++--
 .../annotations/testAnnotationsOnFunction.kt  |  6 +++-
 .../testAnnotationsWithParameters.kt          |  3 +-
 idea/testData/search/junit/testJunit4.kt      |  7 +++-
 idea/testData/search/junit/testJunit4Alias.kt |  4 ++-
 8 files changed, 55 insertions(+), 9 deletions(-)
 create mode 100644 compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightNoOriginElements.kt

diff --git a/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightField.java b/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightField.java
index 7956051f9e8..7ad9444b3cd 100644
--- a/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightField.java
+++ b/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightField.java
@@ -148,7 +148,7 @@ public abstract class KotlinLightField<T extends JetDeclaration, D extends PsiFi
 
     @Override
     public String toString() {
-        return "PsiField:" + getName();
+        return "KotlinLightField:" + getName();
     }
 
     @NotNull
diff --git a/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightMethodForDeclaration.kt b/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightMethodForDeclaration.kt
index 663aef4792b..06872c9c1b3 100644
--- a/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightMethodForDeclaration.kt
+++ b/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightMethodForDeclaration.kt
@@ -125,4 +125,6 @@ open public class KotlinLightMethodForDeclaration(
             getContainingClass() == other.getContainingClass()
 
     override fun hashCode(): Int = (getName().hashCode() * 31 + origin.hashCode()) * 31 + getContainingClass()!!.hashCode()
+
+    override fun toString(): String = "KotlinLightMethodForDeclaration:" + getName()
 }
diff --git a/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightNoOriginElements.kt b/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightNoOriginElements.kt
new file mode 100644
index 00000000000..9773e0f86e5
--- /dev/null
+++ b/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinLightNoOriginElements.kt
@@ -0,0 +1,34 @@
+/*
+ * Copyright 2010-2015 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin.asJava
+
+import com.intellij.psi.*
+import com.intellij.psi.impl.light.LightField
+import com.intellij.psi.impl.light.LightMethod
+
+
+public class KotlinNoOriginLightMethod(manager: PsiManager, method: PsiMethod, containingClass: PsiClass) :
+        LightMethod(manager, method, containingClass) {
+
+    override fun toString() = "KotlinNoOriginLightMethod:${getName()}" + if (isConstructor()) " ctor" else ""
+}
+
+public class KotlinNoOriginLightField(manager: PsiManager, field: PsiField, containingClass: PsiClass) :
+        LightField(manager, field, containingClass) {
+
+    override fun toString() = "KotlinNoOriginLightField:${getName()}"
+}
\ No newline at end of file
diff --git a/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinWrappingLightClass.java b/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinWrappingLightClass.java
index 39d2928669c..602cdb57443 100644
--- a/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinWrappingLightClass.java
+++ b/compiler/light-classes/src/org/jetbrains/kotlin/asJava/KotlinWrappingLightClass.java
@@ -22,8 +22,6 @@ import com.intellij.navigation.ItemPresentationProviders;
 import com.intellij.psi.*;
 import com.intellij.psi.impl.PsiClassImplUtil;
 import com.intellij.psi.impl.light.AbstractLightClass;
-import com.intellij.psi.impl.light.LightField;
-import com.intellij.psi.impl.light.LightMethod;
 import com.intellij.psi.impl.source.ClassInnerStuffCache;
 import com.intellij.psi.impl.source.PsiExtensibleClass;
 import com.intellij.psi.scope.PsiScopeProcessor;
@@ -138,7 +136,7 @@ public abstract class KotlinWrappingLightClass extends AbstractLightClass implem
                 if (declaration != null) {
                     return new KotlinLightFieldForDeclaration(myManager, declaration, field, KotlinWrappingLightClass.this);
                 }
-                return new LightField(myManager, field, KotlinWrappingLightClass.this);
+                return new KotlinNoOriginLightField(myManager, field, KotlinWrappingLightClass.this);
             }
         });
     }
@@ -157,7 +155,7 @@ public abstract class KotlinWrappingLightClass extends AbstractLightClass implem
                            new KotlinLightMethodForTraitFakeOverride(myManager, method, declaration, KotlinWrappingLightClass.this);
                 }
 
-                return new LightMethod(myManager, method, KotlinWrappingLightClass.this);
+                return new KotlinNoOriginLightMethod(myManager, method, KotlinWrappingLightClass.this);
             }
         });
     }
diff --git a/idea/testData/search/annotations/testAnnotationsOnFunction.kt b/idea/testData/search/annotations/testAnnotationsOnFunction.kt
index 320d701d55a..4c034c683b8 100644
--- a/idea/testData/search/annotations/testAnnotationsOnFunction.kt
+++ b/idea/testData/search/annotations/testAnnotationsOnFunction.kt
@@ -27,4 +27,8 @@ object Test7 {
 }
 
 // ANNOTATION: java.lang.Deprecated
-// SEARCH: PsiMethod:test1, PsiMethod:test2, PsiMethod:test4, PsiMethod:test6, PsiMethod:test7
\ No newline at end of file
+// SEARCH: KotlinLightMethodForDeclaration:test1
+// SEARCH: KotlinLightMethodForDeclaration:test2
+// SEARCH: KotlinLightMethodForDeclaration:test4
+// SEARCH: KotlinLightMethodForDeclaration:test6
+// SEARCH: KotlinLightMethodForDeclaration:test7
\ No newline at end of file
diff --git a/idea/testData/search/annotations/testAnnotationsWithParameters.kt b/idea/testData/search/annotations/testAnnotationsWithParameters.kt
index a947baa4c99..1adb5799508 100644
--- a/idea/testData/search/annotations/testAnnotationsWithParameters.kt
+++ b/idea/testData/search/annotations/testAnnotationsWithParameters.kt
@@ -5,4 +5,5 @@ MyAnnotation("f", "s") class Test1() {}
 annotation class MyAnnotation(val first: String, val second: String)
 
 // ANNOTATION: MyAnnotation
-// SEARCH: PsiMethod:test1, KotlinLightClass:Test1
\ No newline at end of file
+// SEARCH: KotlinLightMethodForDeclaration:test1
+// SEARCH: KotlinLightClass:Test1
\ No newline at end of file
diff --git a/idea/testData/search/junit/testJunit4.kt b/idea/testData/search/junit/testJunit4.kt
index a949d4d5ddc..f242dae05be 100644
--- a/idea/testData/search/junit/testJunit4.kt
+++ b/idea/testData/search/junit/testJunit4.kt
@@ -17,4 +17,9 @@ class MyTestClass {
 }
 
 // ANNOTATION: org.junit.Test
-// SEARCH: PsiMethod:test1, PsiMethod:test2, PsiMethod:test3, PsiMethod:test4, PsiMethod:test5, PsiMethod:test6
\ No newline at end of file
+// SEARCH: KotlinLightMethodForDeclaration:test1
+// SEARCH: KotlinLightMethodForDeclaration:test2
+// SEARCH: KotlinLightMethodForDeclaration:test3
+// SEARCH: KotlinLightMethodForDeclaration:test4
+// SEARCH: KotlinLightMethodForDeclaration:test5
+// SEARCH: KotlinLightMethodForDeclaration:test6
\ No newline at end of file
diff --git a/idea/testData/search/junit/testJunit4Alias.kt b/idea/testData/search/junit/testJunit4Alias.kt
index b3a1ed601ee..0f5beb24aba 100644
--- a/idea/testData/search/junit/testJunit4Alias.kt
+++ b/idea/testData/search/junit/testJunit4Alias.kt
@@ -10,4 +10,6 @@ class MyTestClass {
 }
 
 // ANNOTATION: org.junit.Test
-// SEARCH: PsiMethod:test1, PsiMethod:test2, PsiMethod:test3
\ No newline at end of file
+// SEARCH: KotlinLightMethodForDeclaration:test1
+// SEARCH: KotlinLightMethodForDeclaration:test2
+// SEARCH: KotlinLightMethodForDeclaration:test3
\ No newline at end of file