enable lazy conversion

Author: andrewparmet

buildSrc/src/main/kotlin/JavaBasedProjectConventions.kt

@@ -27,6 +27,7 @@ import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 fun Project.javaBasedProjectConventions() {
     repositories {
         mavenCentral()
+        mavenLocal()
     }
 
     dependencies {

gradle/libs.versions.toml

@@ -25,7 +25,7 @@ protobuf-java = "3.21.7"
 protobuf-js = "7.2.6"
 protobufGradlePlugin = "0.9.4"
 protovalidate = "0.6.4"
-protovalidateJava = "0.2.1"
+protovalidateJava = "0.0.0-SNAPSHOT"
 slf4j = "2.0.6"
 
 # build

protokt-protovalidate/build.gradle.kts

@@ -18,6 +18,7 @@ plugins {
 }
 
 enablePublishing()
+trackKotlinApiCompatibility()
 
 dependencies {
     implementation(project(":protokt-reflect"))

protokt-protovalidate/src/main/kotlin/protokt/v1/buf/validate/ProtoktImpl.kt

@@ -0,0 +1,86 @@
+package protokt.v1.buf.validate
+
+import build.buf.protovalidate.internal.evaluator.MessageLike
+import build.buf.protovalidate.internal.evaluator.Value
+import com.google.protobuf.Descriptors.FieldDescriptor
+import protokt.v1.Bytes
+import protokt.v1.Enum
+import protokt.v1.Message
+import protokt.v1.google.protobuf.RuntimeContext
+import protokt.v1.google.protobuf.getField
+import protokt.v1.google.protobuf.hasField
+
+internal class ProtoktMessageLike(
+    val message: Message,
+    val context: RuntimeContext,
+) : MessageLike {
+    override fun hasField(field: FieldDescriptor) =
+        message.hasField(field)
+
+    override fun getField(field: FieldDescriptor) =
+        ProtoktObjectValue(
+            field,
+            message.getField(field)!!,
+            context
+        )
+}
+
+internal class ProtoktMessageValue(
+    private val message: Message,
+    private val context: RuntimeContext,
+) : Value {
+    override fun messageValue() =
+        ProtoktMessageLike(message, context)
+
+    override fun repeatedValue() =
+        emptyList<Value>()
+
+    override fun mapValue() =
+        emptyMap<Value, Value>()
+
+    override fun celValue() =
+        context.convertValue(message)
+
+    override fun <T : Any> jvmValue(clazz: Class<T>) =
+        null
+}
+
+internal class ProtoktObjectValue(
+    private val fieldDescriptor: FieldDescriptor,
+    private val value: Any,
+    private val context: RuntimeContext,
+) : Value {
+    override fun messageValue(): MessageLike =
+        ProtoktMessageLike(value as Message, context)
+
+    override fun repeatedValue() =
+        (value as List<*>).map { ProtoktObjectValue(fieldDescriptor, it!!, context) }
+
+    override fun mapValue(): Map<Value, Value> {
+        val input = value as Map<*, *>
+
+        val keyDesc = fieldDescriptor.messageType.findFieldByNumber(1)
+        val valDesc = fieldDescriptor.messageType.findFieldByNumber(2)
+
+        return input.entries.associate { (key, value) ->
+            Pair(
+                ProtoktObjectValue(keyDesc, key!!, context),
+                ProtoktObjectValue(valDesc, value!!, context),
+            )
+        }
+    }
+
+    override fun celValue() =
+        when (value) {
+            is Enum -> value.value
+            is UInt -> org.projectnessie.cel.common.ULong.valueOf(value.toLong())
+            is ULong -> org.projectnessie.cel.common.ULong.valueOf(value.toLong())
+            is Message, is Bytes -> context.convertValue(value)
+
+            // pray
+            else -> value
+        }
+
+    override fun <T : Any> jvmValue(clazz: Class<T>): T? =
+        context.convertValue(value)?.let(clazz::cast)
+}

protokt-protovalidate/src/main/kotlin/protokt/v1/buf/validate/ProtoktValidator.kt

@@ -38,7 +38,8 @@ import kotlin.reflect.full.findAnnotation
 
 @Beta
 class ProtoktValidator(
-    config: Config = Config.newBuilder().build()
+    config: Config = Config.newBuilder().build(),
+    private val lazyConvert: Boolean = true
 ) {
     private val evaluatorBuilder =
         EvaluatorBuilder(
@@ -87,7 +88,11 @@ class ProtoktValidator(
         evaluatorsByFullTypeName.getValue(
             message::class.findAnnotation<GeneratedMessage>()!!.fullTypeName,
         ).evaluate(
-            MessageValue(message.toDynamicMessage(RuntimeContext(descriptors))),
+            if (lazyConvert) {
+                ProtoktMessageValue(message, RuntimeContext(descriptors))
+            } else {
+                MessageValue(message.toDynamicMessage(RuntimeContext(descriptors)))
+            },
             failFast
         )
 }