Implement quick non-script detection and lazy script discovery..

so ".kt" and ".java" files are not considered as scripts and quickly filtered out, and for the other files the the checks are implemented using sequences, mechanisms provided to supply script definitions lazily, and script discovery is implemented using this mechanisms.
2018-04-18 17:23:39 +02:00
parent 969340167f
commit 15f9ad9531
12 changed files with 322 additions and 128 deletions
@@ -0,0 +1,80 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. 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.scripts
+
+import org.jetbrains.kotlin.cli.common.script.CliScriptDefinitionProvider
+import org.jetbrains.kotlin.script.KotlinScriptDefinition
+import org.jetbrains.kotlin.script.ScriptDefinitionsSource
+import org.jetbrains.kotlin.test.testFramework.KtUsefulTestCase
+import org.junit.Assert
+import org.junit.Test
+import java.util.concurrent.atomic.AtomicInteger
+import kotlin.coroutines.experimental.buildSequence
+import kotlin.script.templates.standard.ScriptTemplateWithArgs
+
+class ScriptProviderTest : KtUsefulTestCase() {
+
+    @Test
+    fun testLazyScriptDefinitionsProvider() {
+
+        val genDefCounter = AtomicInteger()
+        val standardDef = FakeScriptDefinition()
+        val shadedDef = FakeScriptDefinition(".x.kts")
+        val provider = CliScriptDefinitionProvider().apply {
+            setScriptDefinitions(listOf(shadedDef, standardDef))
+            setScriptDefinitionsSources(listOf(TestScriptDefinitionSource(genDefCounter, ".y.kts", ".x.kts")))
+        }
+
+        Assert.assertEquals(0, genDefCounter.get())
+
+        provider.isScript("a.kt").let {
+            Assert.assertFalse(it)
+            Assert.assertEquals(0, genDefCounter.get())
+        }
+
+        provider.isScript("a.y.kts").let {
+            Assert.assertTrue(it)
+            Assert.assertEquals(1, genDefCounter.get())
+        }
+
+        provider.isScript("a.x.kts").let {
+            Assert.assertTrue(it)
+            Assert.assertEquals(2, genDefCounter.get())
+            Assert.assertEquals(0, shadedDef.matchCounter.get())
+        }
+
+        provider.isScript("a.z.kts").let {
+            Assert.assertTrue(it)
+            Assert.assertEquals(2, genDefCounter.get())
+            Assert.assertEquals(1, standardDef.matchCounter.get())
+        }
+
+        provider.isScript("a.ktx").let {
+            Assert.assertFalse(it)
+            Assert.assertEquals(2, genDefCounter.get())
+        }
+    }
+}
+
+private class FakeScriptDefinition(val suffix: String = ".kts") : KotlinScriptDefinition(ScriptTemplateWithArgs::class) {
+    val matchCounter = AtomicInteger()
+    override fun isScript(fileName: String): Boolean = fileName.endsWith(suffix).also {
+        if (it) matchCounter.incrementAndGet()
+    }
+}
+
+private class TestScriptDefinitionSource(val counter: AtomicInteger, val defGens: Iterable<() -> FakeScriptDefinition>) :
+    ScriptDefinitionsSource
+{
+    constructor(counter: AtomicInteger, vararg suffixes: String) : this(counter, suffixes.map { { FakeScriptDefinition(it) } })
+
+    override val definitions: Sequence<KotlinScriptDefinition> = buildSequence {
+        for (gen in defGens) {
+            counter.incrementAndGet()
+            yield(gen())
+        }
+    }
+}