From 5c4dd75110e4a20948411e2733333e580f186225 Mon Sep 17 00:00:00 2001
From: Alexey Sedunov <alexey.sedunov@jetbrains.com>
Date: Wed, 19 Nov 2014 14:14:38 +0300
Subject: [PATCH] Light Methods: Change PSI element language to Kotlin

---
 .../jet/asJava/FakeLightClassForFileOfPackage.java        | 8 ++++++++
 .../src/org/jetbrains/jet/asJava/KotlinLightField.java    | 8 ++++++++
 .../jet/asJava/KotlinLightMethodForDeclaration.kt         | 3 +++
 .../org/jetbrains/jet/asJava/KotlinLightParameter.java    | 7 +++++++
 .../jetbrains/jet/asJava/KotlinLightTypeParameter.java    | 7 +++++++
 .../jetbrains/jet/asJava/KotlinWrappingLightClass.java    | 7 +++++++
 6 files changed, 40 insertions(+)

diff --git a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/FakeLightClassForFileOfPackage.java b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/FakeLightClassForFileOfPackage.java
index 360673129ab..f085a535353 100644
--- a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/FakeLightClassForFileOfPackage.java
+++ b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/FakeLightClassForFileOfPackage.java
@@ -16,6 +16,7 @@
 
 package org.jetbrains.jet.asJava;
 
+import com.intellij.lang.Language;
 import com.intellij.psi.PsiClass;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiFile;
@@ -27,6 +28,7 @@ import org.jetbrains.jet.lang.psi.JetClassOrObject;
 import org.jetbrains.jet.lang.psi.JetFile;
 import org.jetbrains.jet.lang.resolve.java.jetAsJava.JetJavaMirrorMarker;
 import org.jetbrains.jet.lang.resolve.name.FqName;
+import org.jetbrains.jet.plugin.JetLanguage;
 
 /**
  * This class serves as a workaround for usages of {@link JavaElementFinder#findClasses} which eventually only need names of files
@@ -88,4 +90,10 @@ public class FakeLightClassForFileOfPackage extends AbstractLightClass implement
     public String getText() {
         return null;
     }
+
+    @NotNull
+    @Override
+    public Language getLanguage() {
+        return JetLanguage.INSTANCE;
+    }
 }
diff --git a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightField.java b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightField.java
index f92068cfeb0..da398e97ba2 100644
--- a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightField.java
+++ b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightField.java
@@ -16,6 +16,7 @@
 
 package org.jetbrains.jet.asJava;
 
+import com.intellij.lang.Language;
 import com.intellij.lang.java.JavaLanguage;
 import com.intellij.openapi.util.TextRange;
 import com.intellij.psi.*;
@@ -27,6 +28,7 @@ import org.jetbrains.annotations.NonNls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 import org.jetbrains.jet.lang.psi.JetDeclaration;
+import org.jetbrains.jet.plugin.JetLanguage;
 
 // Copied from com.intellij.psi.impl.light.LightField
 public abstract class KotlinLightField<T extends JetDeclaration, D extends PsiField> extends LightElement
@@ -166,4 +168,10 @@ public abstract class KotlinLightField<T extends JetDeclaration, D extends PsiFi
     public PsiElement getNavigationElement() {
         return getOrigin();
     }
+
+    @NotNull
+    @Override
+    public Language getLanguage() {
+        return JetLanguage.INSTANCE;
+    }
 }
diff --git a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightMethodForDeclaration.kt b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightMethodForDeclaration.kt
index aac358310b3..558958e4596 100644
--- a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightMethodForDeclaration.kt
+++ b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightMethodForDeclaration.kt
@@ -31,6 +31,7 @@ import org.jetbrains.jet.lang.psi.JetProperty
 import org.jetbrains.jet.lang.psi.JetClassOrObject
 import com.intellij.psi.impl.light.LightTypeParameterListBuilder
 import com.intellij.psi.search.SearchScope
+import com.intellij.lang.Language
 
 open public class KotlinLightMethodForDeclaration(
         manager: PsiManager,
@@ -103,6 +104,8 @@ open public class KotlinLightMethodForDeclaration(
 
     override fun getUseScope(): SearchScope = origin.getUseScope()
 
+    override fun getLanguage(): Language = JetLanguage.INSTANCE
+
     override fun equals(other: Any?): Boolean =
             other is KotlinLightMethodForDeclaration &&
             getName() == other.getName() &&
diff --git a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightParameter.java b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightParameter.java
index 38959bee54f..ac3ebbb0660 100644
--- a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightParameter.java
+++ b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightParameter.java
@@ -16,6 +16,7 @@
 
 package org.jetbrains.jet.asJava;
 
+import com.intellij.lang.Language;
 import com.intellij.psi.PsiAnnotationOwner;
 import com.intellij.psi.PsiModifierList;
 import com.intellij.psi.PsiParameter;
@@ -98,4 +99,10 @@ public class KotlinLightParameter extends LightParameter implements KotlinLightE
 
         return setter != null ? setter.getParameter() : null;
     }
+
+    @NotNull
+    @Override
+    public Language getLanguage() {
+        return JetLanguage.INSTANCE;
+    }
 }
diff --git a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightTypeParameter.java b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightTypeParameter.java
index 4ca33c4b685..5ee420eee07 100644
--- a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightTypeParameter.java
+++ b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinLightTypeParameter.java
@@ -16,6 +16,7 @@
 
 package org.jetbrains.jet.asJava;
 
+import com.intellij.lang.Language;
 import com.intellij.psi.*;
 import com.intellij.psi.impl.light.AbstractLightClass;
 import org.jetbrains.annotations.NotNull;
@@ -127,4 +128,10 @@ public class KotlinLightTypeParameter
     public PsiElement getNavigationElement() {
         return getOrigin();
     }
+
+    @NotNull
+    @Override
+    public Language getLanguage() {
+        return JetLanguage.INSTANCE;
+    }
 }
diff --git a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinWrappingLightClass.java b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinWrappingLightClass.java
index 18bfa81bc2d..c49c964640f 100644
--- a/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinWrappingLightClass.java
+++ b/compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/KotlinWrappingLightClass.java
@@ -33,6 +33,7 @@ import kotlin.KotlinPackage;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.jet.lang.psi.*;
 import org.jetbrains.jet.lang.resolve.name.FqName;
+import org.jetbrains.jet.plugin.JetLanguage;
 
 import java.util.List;
 
@@ -171,6 +172,12 @@ public abstract class KotlinWrappingLightClass extends AbstractLightClass implem
         return origin == null ? null : origin.getText();
     }
 
+    @NotNull
+    @Override
+    public Language getLanguage() {
+        return JetLanguage.INSTANCE;
+    }
+
     private boolean isTraitFakeOverride(@NotNull JetDeclaration originMethodDeclaration) {
         if (!(originMethodDeclaration instanceof JetNamedFunction ||
               originMethodDeclaration instanceof JetPropertyAccessor ||