Decouple CelTypeResolver from full protobuf implementation

PiperOrigin-RevId: 725790812

Author: l46kok

extensions/src/main/java/dev/cel/extensions/BUILD.bazel

@@ -147,7 +147,7 @@ java_library(
         "//common/types",
         "//compiler:compiler_builder",
         "//runtime",
-        "//runtime:runtime_helper",
+        "//runtime:proto_message_runtime_equality",
         "@maven//:com_google_errorprone_error_prone_annotations",
         "@maven//:com_google_guava_guava",
     ],

extensions/src/main/java/dev/cel/extensions/CelSetsExtensions.java

@@ -29,7 +29,7 @@
 import dev.cel.runtime.CelRuntime;
 import dev.cel.runtime.CelRuntimeBuilder;
 import dev.cel.runtime.CelRuntimeLibrary;
-import dev.cel.runtime.RuntimeEquality;
+import dev.cel.runtime.ProtoMessageRuntimeEquality;
 import java.util.Collection;
 import java.util.Iterator;
 import java.util.Set;
@@ -66,9 +66,6 @@ public final class CelSetsExtensions implements CelCompilerLibrary, CelRuntimeLi
           + " are unique, so size does not factor into the computation. If either list is empty,"
           + " the result will be false.";
 
-  private static final RuntimeEquality RUNTIME_EQUALITY =
-      new RuntimeEquality(DynamicProto.create(DefaultMessageFactory.INSTANCE));
-
   /** Denotes the set extension function. */
   public enum Function {
     CONTAINS(
@@ -111,15 +108,17 @@ String getFunction() {
   }
 
   private final ImmutableSet<Function> functions;
-  private final CelOptions celOptions;
+  private final ProtoMessageRuntimeEquality runtimeEquality;
 
   CelSetsExtensions(CelOptions celOptions) {
     this(celOptions, ImmutableSet.copyOf(Function.values()));
   }
 
   CelSetsExtensions(CelOptions celOptions, Set<Function> functions) {
     this.functions = ImmutableSet.copyOf(functions);
-    this.celOptions = celOptions;
+    this.runtimeEquality =
+        ProtoMessageRuntimeEquality.create(
+            DynamicProto.create(DefaultMessageFactory.INSTANCE), celOptions);
   }
 
   @Override
@@ -208,7 +207,7 @@ private <T> boolean contains(Object o, Collection<T> list) {
   }
 
   private boolean objectsEquals(Object o1, Object o2) {
-    return RUNTIME_EQUALITY.objectEquals(o1, o2, celOptions);
+    return runtimeEquality.objectEquals(o1, o2);
   }
 
   private <T> boolean setIntersects(Collection<T> listA, Collection<T> listB) {

publish/BUILD.bazel

@@ -9,7 +9,7 @@ RUNTIME_TARGETS = [
     "//runtime/src/main/java/dev/cel/runtime",
     "//runtime/src/main/java/dev/cel/runtime:base",
     "//runtime/src/main/java/dev/cel/runtime:interpreter",
-    "//runtime/src/main/java/dev/cel/runtime:runtime_helper",
+    "//runtime/src/main/java/dev/cel/runtime:runtime_helpers",
     "//runtime/src/main/java/dev/cel/runtime:unknown_attributes",
 ]
 

runtime/BUILD.bazel

@@ -25,9 +25,33 @@ java_library(
 )
 
 java_library(
-    name = "runtime_helper",
+    name = "runtime_helpers",
     visibility = ["//visibility:public"],
-    exports = ["//runtime/src/main/java/dev/cel/runtime:runtime_helper"],
+    exports = ["//runtime/src/main/java/dev/cel/runtime:runtime_helpers"],
+)
+
+java_library(
+    name = "proto_message_runtime_helpers",
+    visibility = ["//visibility:public"],
+    exports = ["//runtime/src/main/java/dev/cel/runtime:proto_message_runtime_helpers"],
+)
+
+java_library(
+    name = "runtime_equality",
+    visibility = ["//visibility:public"],
+    exports = ["//runtime/src/main/java/dev/cel/runtime:runtime_equality"],
+)
+
+java_library(
+    name = "proto_message_runtime_equality",
+    visibility = ["//visibility:public"],
+    exports = ["//runtime/src/main/java/dev/cel/runtime:proto_message_runtime_equality"],
+)
+
+java_library(
+    name = "type_resolver",
+    visibility = ["//visibility:public"],
+    exports = ["//runtime/src/main/java/dev/cel/runtime:type_resolver"],
 )
 
 java_library(

runtime/src/main/java/dev/cel/runtime/BUILD.bazel

@@ -37,9 +37,25 @@ INTERPRETER_SOURCES = [
 ]
 
 java_library(
-    name = "cel_type_resolver",
-    srcs = ["CelTypeResolver.java"],
+    name = "type_resolver",
+    srcs = ["TypeResolver.java"],
+    tags = [
+    ],
+    deps = [
+        "//common/types",
+        "//common/types:type_providers",
+        "@maven//:com_google_errorprone_error_prone_annotations",
+        "@maven//:com_google_guava_guava",
+        "@maven//:com_google_protobuf_protobuf_java",
+    ],
+)
+
+java_library(
+    name = "descriptor_type_resolver",
+    srcs = ["DescriptorTypeResolver.java"],
+    visibility = ["//visibility:private"],
     deps = [
+        ":type_resolver",
         "//common/types",
         "//common/types:type_providers",
         "@maven//:com_google_errorprone_error_prone_annotations",
@@ -56,8 +72,7 @@ java_library(
     deps = [
         ":evaluation_exception",
         ":metadata",
-        "//:auto_value",
-        "//common",
+        "//common:cel_ast",
         "//common/annotations",
         "@maven//:com_google_code_findbugs_annotations",
         "@maven//:com_google_errorprone_error_prone_annotations",
@@ -75,12 +90,12 @@ java_library(
     exports = [":base"],
     deps = [
         ":base",
-        ":cel_type_resolver",
         ":evaluation_exception",
         ":evaluation_exception_builder",
         ":evaluation_listener",
         ":metadata",
-        ":runtime_helper",
+        ":runtime_helpers",
+        ":type_resolver",
         ":unknown_attributes",
         "//:auto_value",
         "//common",
@@ -105,31 +120,80 @@ java_library(
 )
 
 java_library(
-    name = "runtime_helper",
+    name = "runtime_equality",
     srcs = [
         "RuntimeEquality.java",
-        "RuntimeHelpers.java",
     ],
     tags = [
     ],
-    # NOTE: do not grow this dependencies arbitrarily
     deps = [
+        ":runtime_helpers",
         "//common:error_codes",
         "//common:options",
         "//common:runtime_exception",
-        "//common/annotations",
         "//common/internal:comparison_functions",
-        "//common/internal:converter",
+        "@maven//:com_google_errorprone_error_prone_annotations",
+        "@maven//:com_google_guava_guava",
+        "@maven//:com_google_protobuf_protobuf_java",
+    ],
+)
+
+java_library(
+    name = "proto_message_runtime_equality",
+    srcs = [
+        "ProtoMessageRuntimeEquality.java",
+    ],
+    tags = [
+    ],
+    deps = [
+        ":proto_message_runtime_helpers",
+        ":runtime_equality",
+        "//common:options",
+        "//common/annotations",
         "//common/internal:dynamic_proto",
         "//common/internal:proto_equality",
         "@maven//:com_google_errorprone_error_prone_annotations",
         "@maven//:com_google_guava_guava",
         "@maven//:com_google_protobuf_protobuf_java",
+    ],
+)
+
+java_library(
+    name = "runtime_helpers",
+    srcs = [
+        "RuntimeHelpers.java",
+    ],
+    tags = [
+    ],
+    deps = [
+        "//common:error_codes",
+        "//common:options",
+        "//common:runtime_exception",
+        "//common/internal:converter",
+        "@maven//:com_google_errorprone_error_prone_annotations",
+        "@maven//:com_google_guava_guava",
+        "@maven//:com_google_protobuf_protobuf_java",
         "@maven//:com_google_re2j_re2j",
         "@maven//:org_threeten_threeten_extra",
     ],
 )
 
+java_library(
+    name = "proto_message_runtime_helpers",
+    srcs = [
+        "ProtoMessageRuntimeHelpers.java",
+    ],
+    tags = [
+    ],
+    deps = [
+        ":runtime_helpers",
+        "//common:options",
+        "//common/annotations",
+        "//common/internal:dynamic_proto",
+        "@maven//:com_google_protobuf_protobuf_java",
+    ],
+)
+
 # keep sorted
 RUNTIME_SOURCES = [
     "CelFunctionOverload.java",
@@ -194,9 +258,13 @@ java_library(
     tags = [
     ],
     deps = [
+        ":descriptor_type_resolver",
         ":evaluation_exception",
         ":evaluation_listener",
-        ":runtime_helper",
+        ":interpreter",
+        ":proto_message_runtime_equality",
+        ":runtime_equality",
+        ":runtime_helpers",
         ":runtime_type_provider_legacy",
         ":unknown_attributes",
         "//:auto_value",
@@ -214,7 +282,6 @@ java_library(
         "//common/types:cel_types",
         "//common/values:cel_value_provider",
         "//common/values:proto_message_value_provider",
-        "//runtime:interpreter",
         "@maven//:com_google_code_findbugs_annotations",
         "@maven//:com_google_errorprone_error_prone_annotations",
         "@maven//:com_google_guava_guava",
@@ -271,6 +338,7 @@ java_library(
     name = "runtime_type_provider_legacy",
     srcs = ["RuntimeTypeProviderLegacyImpl.java"],
     deps = [
+        ":interpreter",
         ":unknown_attributes",
         "//common:error_codes",
         "//common:options",
@@ -282,7 +350,6 @@ java_library(
         "//common/values:cel_value",
         "//common/values:cel_value_provider",
         "//common/values:proto_message_value",
-        "//runtime:interpreter",
         "@maven//:com_google_errorprone_error_prone_annotations",
         "@maven//:com_google_guava_guava",
     ],

runtime/src/main/java/dev/cel/runtime/CelRuntimeLegacyImpl.java

@@ -241,10 +241,12 @@ public CelRuntimeLegacyImpl build() {
               runtimeTypeFactory, DefaultMessageFactory.create(celDescriptorPool));
 
       DynamicProto dynamicProto = DynamicProto.create(runtimeTypeFactory);
+      RuntimeEquality runtimeEquality = ProtoMessageRuntimeEquality.create(dynamicProto, options);
 
       ImmutableMap.Builder<String, CelFunctionBinding> functionBindingsBuilder =
           ImmutableMap.builder();
-      for (CelFunctionBinding standardFunctionBinding : newStandardFunctionBindings(dynamicProto)) {
+      for (CelFunctionBinding standardFunctionBinding :
+          newStandardFunctionBindings(runtimeEquality)) {
         functionBindingsBuilder.put(
             standardFunctionBinding.getOverloadId(), standardFunctionBinding);
       }
@@ -277,13 +279,17 @@ public CelRuntimeLegacyImpl build() {
       }
 
       return new CelRuntimeLegacyImpl(
-          new DefaultInterpreter(runtimeTypeProvider, dispatcher.immutableCopy(), options),
+          new DefaultInterpreter(
+              DescriptorTypeResolver.create(),
+              runtimeTypeProvider,
+              dispatcher.immutableCopy(),
+              options),
           options,
           this);
     }
 
     private ImmutableSet<CelFunctionBinding> newStandardFunctionBindings(
-        DynamicProto dynamicProto) {
+        RuntimeEquality runtimeEquality) {
       CelStandardFunctions celStandardFunctions;
       if (standardEnvironmentEnabled) {
         celStandardFunctions =
@@ -336,7 +342,7 @@ private ImmutableSet<CelFunctionBinding> newStandardFunctionBindings(
         return ImmutableSet.of();
       }
 
-      return celStandardFunctions.newFunctionBindings(dynamicProto, options);
+      return celStandardFunctions.newFunctionBindings(runtimeEquality, options);
     }
 
     private static CelDescriptorPool newDescriptorPool(