DivKit get dict values

gistfile1.txt

import com.yandex.div.evaluable.EvaluableType
import com.yandex.div.evaluable.EvaluationContext
import com.yandex.div.evaluable.ExpressionContext
import com.yandex.div.evaluable.Function
import com.yandex.div.evaluable.FunctionArgument
import org.json.JSONArray
import org.json.JSONObject

class GetDictValuesFunction : Function() {

    override val name = "getOptDictValues"

    override val declaredArgs = listOf(
        FunctionArgument(type = EvaluableType.DICT), // variable name
    )

    override val resultType = EvaluableType.ARRAY
    override val isPure = false

    override fun evaluate(
        evaluationContext: EvaluationContext,
        expressionContext: ExpressionContext,
        args: List<Any>
    ): Any {
        val fallback = JSONArray()
        val dict = args[0] as? JSONObject ?: return fallback
        val result = JSONArray()
        dict.keys().forEach { key ->
            result.put(dict.get(key))
        }
        return result
    }
}