From 0eda1f3969c4e7e801892ea512202c7969850021 Mon Sep 17 00:00:00 2001
From: Alexey Sedunov <alexey.sedunov@jetbrains.com>
Date: Mon, 14 Mar 2016 17:54:10 +0300
Subject: [PATCH] Spring Support: SpEL injections  #KT-11406 Fixed

---
 Changelog.md                                  |  1 +
 .../src/META-INF/kotlin-spring.xml            |  2 +
 .../idea/spring/el/KotlinSpringELInjector.kt  | 82 +++++++++++++++++++
 3 files changed, 85 insertions(+)
 create mode 100644 idea/idea-ultimate/src/org/jetbrains/kotlin/idea/spring/el/KotlinSpringELInjector.kt

diff --git a/Changelog.md b/Changelog.md
index 4876619712d..b0cafb02769 100644
--- a/Changelog.md
+++ b/Changelog.md
@@ -99,6 +99,7 @@ New features:
 - [KT-11098](https://youtrack.jetbrains.com/issue/KT-11098) Inspection on final classes/functions annotated with Spring @Configuration/@Component/@Bean
 - [KT-11405](https://youtrack.jetbrains.com/issue/KT-11405) Navigation and Find Usages for Spring beans referenced in annotation arguments and BeanFactory method calls
 - [KT-3741](https://youtrack.jetbrains.com/issue/KT-3741) Show Spring-specific line markers on Kotlin classes
+- [KT-11406](https://youtrack.jetbrains.com/issue/KT-11406) Support Spring EL injections inside of Kotlin string literals
 
 General issues fixed:
 
diff --git a/idea/idea-ultimate/src/META-INF/kotlin-spring.xml b/idea/idea-ultimate/src/META-INF/kotlin-spring.xml
index 347c6c334ca..12b5f6831cf 100644
--- a/idea/idea-ultimate/src/META-INF/kotlin-spring.xml
+++ b/idea/idea-ultimate/src/META-INF/kotlin-spring.xml
@@ -2,6 +2,8 @@
   <extensions defaultExtensionNs="com.intellij">
       <psi.referenceContributor language="kotlin" implementation="org.jetbrains.kotlin.idea.spring.references.KotlinSpringReferenceContributor"/>
 
+      <multiHostInjector implementation="org.jetbrains.kotlin.idea.spring.el.KotlinSpringELInjector"/>
+
       <codeInsight.lineMarkerProvider language="kotlin" implementationClass="org.jetbrains.kotlin.idea.spring.lineMarking.KotlinSpringClassAnnotator"/>
 
       <localInspection implementationClass="org.jetbrains.kotlin.idea.spring.inspections.KotlinFinalClassOrFunSpringInspection"
diff --git a/idea/idea-ultimate/src/org/jetbrains/kotlin/idea/spring/el/KotlinSpringELInjector.kt b/idea/idea-ultimate/src/org/jetbrains/kotlin/idea/spring/el/KotlinSpringELInjector.kt
new file mode 100644
index 00000000000..c1f9c2ef4b5
--- /dev/null
+++ b/idea/idea-ultimate/src/org/jetbrains/kotlin/idea/spring/el/KotlinSpringELInjector.kt
@@ -0,0 +1,82 @@
+/*
+ * Copyright 2010-2016 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.idea.spring.el
+
+import com.intellij.ide.plugins.PluginManager
+import com.intellij.lang.injection.MultiHostInjector
+import com.intellij.lang.injection.MultiHostRegistrar
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiLanguageInjectionHost
+import com.intellij.psi.impl.source.jsp.el.ELContextProvider
+import com.intellij.spring.constants.SpringAnnotationsConstants
+import com.intellij.spring.el.SpringELLanguage
+import com.intellij.spring.el.SpringElTemplateParser
+import com.intellij.spring.el.contextProviders.SpringElContextProvider
+import org.jetbrains.kotlin.descriptors.ConstructorDescriptor
+import org.jetbrains.kotlin.idea.caches.resolve.analyze
+import org.jetbrains.kotlin.idea.imports.importableFqName
+import org.jetbrains.kotlin.psi.KtAnnotationEntry
+import org.jetbrains.kotlin.psi.KtStringTemplateExpression
+import org.jetbrains.kotlin.psi.KtValueArgument
+import org.jetbrains.kotlin.psi.psiUtil.getStrictParentOfType
+import org.jetbrains.kotlin.psi.psiUtil.isPlain
+import org.jetbrains.kotlin.resolve.calls.callUtil.getResolvedCall
+import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
+import org.jetbrains.kotlin.utils.ifEmpty
+
+class KotlinSpringELInjector : MultiHostInjector {
+    companion object {
+        private val RELEVANT_ANNOTATIONS = setOf(SpringAnnotationsConstants.VALUE, SpringAnnotationsConstants.SCHEDULED)
+
+        // SpEL support requires both Spring and J2EE plugins
+        // This code works around the inability to specify multiple dependencies for a given plugin config file
+        private val contextRecorder by lazy {
+            if ("com.intellij.javaee" in PluginManager.getDisabledPlugins()) return@lazy null
+            { host: PsiElement -> host.putUserData(ELContextProvider.ourContextProviderKey, SpringElContextProvider(host)) }
+        }
+    }
+
+    override fun getLanguagesToInject(registrar: MultiHostRegistrar, host: PsiElement) {
+        if (host is KtStringTemplateExpression) {
+            if (!host.isPlain()) return
+
+            val argument = host.parent as? KtValueArgument ?: return
+            val argumentName = argument.getArgumentName()
+            if (argumentName != null && argumentName.asName.asString() != "value") return
+
+            val entry = argument.getStrictParentOfType<KtAnnotationEntry>() ?: return
+            val context = entry.analyze(BodyResolveMode.PARTIAL)
+            val resolvedCall = entry.getResolvedCall(context) ?: return
+            val annotation = (resolvedCall.resultingDescriptor as? ConstructorDescriptor)?.containingDeclaration ?: return
+            if (annotation.importableFqName?.asString() !in RELEVANT_ANNOTATIONS) return
+
+            val text = host.getText()
+            val ranges = SpringElTemplateParser.parse(text).ifEmpty { return }
+            for (range in ranges) {
+                registrar
+                        .startInjecting(SpringELLanguage.INSTANCE)
+                        .addPlace(null, null, host as PsiLanguageInjectionHost, range)
+                        .doneInjecting()
+            }
+
+            contextRecorder!!(host)
+        }
+    }
+
+    override fun elementsToInjectIn() = contextRecorder?.let { listOf(KtStringTemplateExpression::class.java) } ?: emptyList()
+}
+