Make KotlinScript annotation accepting all params needed to define a script

plugins/scripting/scripting-cli/src/org/jetbrains/kotlin/scripting/compiler/plugin/LazyScriptDefinitionFromDiscoveredClass.kt

@@ -66,6 +66,8 @@ class LazyScriptDefinitionFromDiscoveredClass internal constructor(
 
     override val scriptFileExtensionWithDot: String by lazy(LazyThreadSafetyMode.PUBLICATION) {
         val ext = annotationsFromAsm.find { it.name == KotlinScriptFileExtension::class.simpleName!! }?.args?.first()
+            // TODO: check whether it actually works
+            ?: annotationsFromAsm.find { it.name == KotlinScript::class.simpleName }?.args?.get(1)
             ?: scriptDefinition.let {
                 it.getOrNull(ScriptDefinitionProperties.fileExtension) ?: "kts"
             }