com.anaplan.engineering.azuki.script.generation.ScriptingHelper.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of azuki-script-generation Show documentation

Tools to assist with the recreation of scripts from in-memory Azuki scenarios

The newest version!

package com.anaplan.engineering.azuki.script.generation

import kotlin.reflect.KClassifier
import kotlin.reflect.KFunction
import kotlin.reflect.KType

class ScriptingHelper(private val valueScripters: Map String>) {

    fun scriptifyFunction(fn: KFunction<*>, vararg values: Any?): String {
        if (fn.parameters.filter { !it.isOptional }.size > values.size + 1) {
            throw IllegalStateException("Too few parameters for ${fn.name}: ${values.toList()}")
        }
        val indexedParameters = fn.parameters.drop(1).withIndex()
        val varargIndex = indexedParameters.find { it.value.isVararg }?.index
        val paramStrings = indexedParameters.zip(values).map { (ip, v) ->
            val (i, p) = ip
            val requiredName = if (varargIndex != null && i > varargIndex) {
                p.name
            } else {
                null
            }
            scriptifyParameter(p.type, v, p.isVararg, requiredName)
        }
        return "${fn.name}(${paramStrings.filterNotNull().joinToString(", ")})"
    }

    fun scriptifyParameter(
        expectedType: KType,
        value: Any?,
        isVararg: Boolean,
        requiredName: String?
    ): String? =
        if (isVararg) {
            if (value is Collection<*>) {
                val childType = expectedType.arguments.single().type!!
                if (value.isEmpty()) null else value.map { scriptifyParameter(childType, it, false, null) }
                    .joinToString(", ")
            } else {
                scriptifyParameter(expectedType, value, false, null)
            }
        } else {
            if (requiredName != null) {
                "$requiredName = "
            } else {
                ""
            } + scriptifyValue(expectedType, value)
        }

    private fun ScriptingHelper.scriptifyValue(
        expectedType: KType,
        value: Any?
    ) =
        if (value == null) {
            "null"
        } else {
            val valueScripter = getValueScripter(expectedType)
                ?: getValueScripter(value::class)
                ?: throw IllegalArgumentException("Unable to scriptify value with class ${expectedType.classifier}")
            valueScripter(value)
        }

    private fun getValueScripter(expectedType: KType) =
        if (expectedType.classifier == null) null else getValueScripter(expectedType.classifier!!)

    private fun getValueScripter(expectedClass: KClassifier) = valueScripters.get(expectedClass)
}