Tests: rewrite some AbstractScriptConfigurationTest to the new script infrastructure

2019-09-23 10:49:48 +03:00
parent eca7bbdb8c
commit 63e5d4fe95
3 changed files with 91 additions and 37 deletions
@@ -1,23 +1,55 @@
 package org.jetbrains.kotlin.idea.script

 import java.io.File
-import kotlin.script.dependencies.*
-import kotlin.script.experimental.dependencies.*
-import kotlin.script.experimental.location.*
-import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.annotations.KotlinScript
+import kotlin.script.experimental.api.*
+import kotlin.script.experimental.host.ScriptingHostConfiguration
+import kotlin.script.experimental.host.ScriptingHostConfigurationKeys
+import kotlin.script.experimental.jvm.JvmDependency
+import kotlin.script.experimental.jvm.dependenciesFromClassContext
+import kotlin.script.experimental.jvm.jdkHome
+import kotlin.script.experimental.jvm.jvm
+import kotlin.script.experimental.util.PropertiesCollection

-class FromTextDependenciesResolver : DependenciesResolver {
-    @Suppress("UNCHECKED_CAST")
-    override fun resolve(scriptContents: ScriptContents, environment: Environment): DependenciesResolver.ResolveResult {
-        return ScriptDependencies(
-            javaHome = environment["javaHome"] as? File,
-            classpath = (environment["classpath"] as? List<File>).orEmpty(),
-            imports = (environment["imports"] as? List<String>).orEmpty(),
-            sources = (environment["sources"] as? List<File>).orEmpty()
-        ).asSuccess()
+@KotlinScript(
+    displayName = "Definition for tests",
+    fileExtension = "kts",
+    compilationConfiguration = TemplateDefinition::class
+)
+open class Template(val args: Array<String>)
+
+val ScriptingHostConfigurationKeys.getEnvironment by PropertiesCollection.key<() -> Map<String, Any?>?>()
+
+@Suppress("UNCHECKED_CAST")
+object TemplateDefinition : ScriptCompilationConfiguration(
+    {
+        jvm {
+            dependenciesFromClassContext(TemplateDefinition::class)
+        }
+        ide {
+            acceptedLocations(ScriptAcceptedLocation.Everywhere)
+        }
+        refineConfiguration {
+            beforeCompiling { context ->
+                val environment = context.compilationConfiguration[ScriptCompilationConfiguration.hostConfiguration]?.let {
+                    it[ScriptingHostConfiguration.getEnvironment]?.invoke()
+                }.orEmpty()
+
+                context.compilationConfiguration.with {
+                    (environment["javaHome"] as? File)?.let {
+                        jvm.jdkHome(it)
+                    }
+                    (environment["imports"] as? List<String>)?.let {
+                        defaultImports(it)
+                    }
+                    (environment["classpath"] as? List<File>)?.let {
+                        dependencies(JvmDependency(it))
+                    }
+                    (environment["sources"] as? List<File>)?.let {
+                        ide.dependenciesSources(JvmDependency(it))
+                    }
+                }.asSuccess()
+            }
+        }
    }
-}
-
-@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
-@ScriptTemplateDefinition(FromTextDependenciesResolver::class, scriptFilePattern = "script.kts")
-open class Template
+)
@@ -1,24 +1,37 @@
 package custom.scriptDefinition

 import java.io.File
-import kotlin.script.dependencies.*
-import kotlin.script.experimental.dependencies.*
-import kotlin.script.templates.ScriptTemplateDefinition
-import kotlin.script.experimental.location.*
+import kotlin.script.experimental.annotations.KotlinScript
+import kotlin.script.experimental.api.*
+import kotlin.script.experimental.host.ScriptingHostConfiguration
+import kotlin.script.experimental.host.ScriptingHostConfigurationKeys
+import kotlin.script.experimental.jvm.JvmDependency
+import kotlin.script.experimental.util.PropertiesCollection

-class TestDependenciesResolver : DependenciesResolver {
-    override fun resolve(
-            scriptContents: ScriptContents,
-            environment: Environment
-    ): DependenciesResolver.ResolveResult {
-        return DependenciesResolver.ResolveResult.Success(
-            ScriptDependencies(
-                classpath = listOf(environment["template-classes"] as File)
-            )
-        )
+@KotlinScript(
+    fileExtension = "kts",
+    compilationConfiguration = TemplateDefinition::class
+)
+open class Template(val args: Array<String>)
+
+val ScriptingHostConfigurationKeys.getEnvironment by PropertiesCollection.key<() -> Map<String, Any?>?>()
+
+object TemplateDefinition : ScriptCompilationConfiguration(
+    {
+        refineConfiguration {
+            beforeCompiling { context ->
+                val environment =
+                    context.compilationConfiguration[ScriptCompilationConfiguration.hostConfiguration]?.let {
+                        it[ScriptingHostConfiguration.getEnvironment]?.invoke()
+                    }.orEmpty()
+
+                context.compilationConfiguration.with {
+                    dependencies(JvmDependency(environment["template-classes"] as File))
+                }.asSuccess()
+            }
+        }
+        ide {
+            acceptedLocations(ScriptAcceptedLocation.Everywhere)
+        }
    }
-}
-
-@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
-@ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
-open class Template
+)