05a147171e48c88d22864251d3f5c0376760596a
[ccsdk/cds.git] /
1 /*
2  * Copyright © 2017-2018 AT&T Intellectual Property.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.onap.ccsdk.cds.controllerblueprints.core.scripts
18
19 import org.onap.ccsdk.cds.controllerblueprints.core.BluePrintProcessorException
20 import org.slf4j.LoggerFactory
21 import java.util.*
22 import kotlin.reflect.full.createInstance
23 import kotlin.script.experimental.api.*
24 import kotlin.script.experimental.host.BasicScriptingHost
25 import kotlin.script.experimental.jvm.defaultJvmScriptingHostConfiguration
26 import kotlin.script.experimental.jvmhost.JvmScriptCompiler
27 import kotlin.script.experimental.jvmhost.impl.withDefaults
28
29 val blueprintScriptCompiler = JvmScriptCompiler(defaultJvmScriptingHostConfiguration,
30         BluePrintsCompilerProxy(defaultJvmScriptingHostConfiguration.withDefaults()))
31
32 open class BlueprintScriptingHost(evaluator: ScriptEvaluator) : BasicScriptingHost(blueprintScriptCompiler, evaluator) {
33
34     override fun eval(script: SourceCode, scriptCompilationConfiguration: ScriptCompilationConfiguration,
35                       configuration: ScriptEvaluationConfiguration?): ResultWithDiagnostics<EvaluationResult> =
36
37             runInCoroutineContext {
38
39                 blueprintScriptCompiler(script, scriptCompilationConfiguration)
40                         .onSuccess {
41                             evaluator(it, configuration)
42                         }.onFailure { failedResult ->
43                             val messages = failedResult.reports.joinToString("\n")
44                             throw BluePrintProcessorException(messages)
45                         }
46             }
47 }
48
49 open class BluePrintScriptEvaluator(private val scriptClassName: String) : ScriptEvaluator {
50
51     private val log = LoggerFactory.getLogger(BluePrintScriptEvaluator::class.java)!!
52
53     override suspend operator fun invoke(compiledScript: CompiledScript<*>,
54                                          scriptEvaluationConfiguration: ScriptEvaluationConfiguration?
55     ): ResultWithDiagnostics<EvaluationResult> =
56             try {
57                 log.debug("Getting script class name($scriptClassName) from the compiled sources ")
58
59                 val bluePrintCompiledScript = compiledScript as BluePrintCompiledScript
60                 bluePrintCompiledScript.scriptClassFQName = scriptClassName
61
62                 val classResult = compiledScript.getClass(scriptEvaluationConfiguration)
63                 when (classResult) {
64                     is ResultWithDiagnostics.Failure -> classResult
65                     is ResultWithDiagnostics.Success -> {
66
67                         val scriptClass = classResult.value
68                         val args = ArrayList<Any?>()
69                         scriptEvaluationConfiguration?.get(ScriptEvaluationConfiguration.providedProperties)?.forEach {
70                             args.add(it.value)
71                         }
72                         scriptEvaluationConfiguration?.get(ScriptEvaluationConfiguration.implicitReceivers)?.let {
73                             args.addAll(it)
74                         }
75                         scriptEvaluationConfiguration?.get(ScriptEvaluationConfiguration.constructorArgs)?.let {
76                             args.addAll(it)
77                         }
78
79                         val instance = if (args.isNotEmpty()) {
80                             scriptClass.java.constructors.single().newInstance(*args.toArray())
81                                     ?: throw BluePrintProcessorException("failed to create instance from the script")
82                         } else {
83                             scriptClass.createInstance()
84                         }
85
86                         log.debug("Created script instance of type ${instance.javaClass}")
87
88                         ResultWithDiagnostics.Success(EvaluationResult(ResultValue.Value(scriptClass.qualifiedName!!,
89                                 instance, "", instance),
90                                 scriptEvaluationConfiguration))
91                     }
92                 }
93             } catch (e: Throwable) {
94                 ResultWithDiagnostics.Failure(e.asDiagnostics("Error evaluating script"))
95             }
96 }