Implement Native extension

PiperOrigin-RevId: 904195105

Author: l46kok

bundle/BUILD.bazel

@@ -13,14 +13,6 @@ java_library(
     ],
 )
 
-java_library(
-    name = "cel_experimental_factory",
-    visibility = ["//:internal"],
-    exports = [
-        "//bundle/src/main/java/dev/cel/bundle:cel_experimental_factory",
-    ],
-)
-
 java_library(
     name = "environment",
     exports = ["//bundle/src/main/java/dev/cel/bundle:environment"],

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

@@ -35,7 +35,6 @@ java_library(
         "@cel_spec//proto/cel/expr:checked_java_proto",
         "@maven//:com_google_code_findbugs_annotations",
         "@maven//:com_google_errorprone_error_prone_annotations",
-        "@maven//:com_google_guava_guava",
         "@maven//:com_google_protobuf_protobuf_java",
     ],
 )
@@ -54,22 +53,6 @@ java_library(
         "//compiler:compiler_builder",
         "//parser",
         "//runtime",
-    ],
-)
-
-java_library(
-    name = "cel_experimental_factory",
-    srcs = ["CelExperimentalFactory.java"],
-    tags = [
-    ],
-    deps = [
-        ":cel",
-        ":cel_impl",
-        "//checker",
-        "//common:options",
-        "//common/annotations",
-        "//compiler",
-        "//parser",
         "//runtime:runtime_planner_impl",
     ],
 )
@@ -117,7 +100,6 @@ java_library(
     tags = [
     ],
     deps = [
-        ":cel_factory",
         ":environment_exception",
         ":required_fields_checker",
         "//:auto_value",
@@ -190,7 +172,6 @@ java_library(
         "//common:options",
         "//common/internal:env_visitor",
         "//common/types:cel_proto_types",
-        "//common/types:cel_types",
         "//common/types:type_providers",
         "//compiler:compiler_builder",
         "//extensions",

bundle/src/main/java/dev/cel/bundle/CelExperimentalFactory.java

@@ -20,6 +20,7 @@
 import dev.cel.compiler.CelCompilerImpl;
 import dev.cel.parser.CelParserImpl;
 import dev.cel.runtime.CelRuntime;
+import dev.cel.runtime.CelRuntimeImpl;
 import dev.cel.runtime.CelRuntimeLegacyImpl;
 
 /** Helper class to configure the entire CEL stack in a common interface. */
@@ -44,6 +45,30 @@ public static CelBuilder standardCelBuilder() {
         .setStandardEnvironmentEnabled(true);
   }
 
+  /**
+   * Creates a builder for configuring CEL for the parsing, optional type-checking, and evaluation
+   * of expressions using the Program Planner.
+   *
+   * <p>The {@code ProgramPlanner} architecture provides key benefits over the {@link
+   * #standardCelBuilder()}:
+   *
+   * <ul>
+   *   <li><b>Performance:</b> Programs can be cached for improving evaluation speed.
+   *   <li><b>Parsed-only expression evaluation:</b> Unlike the traditional stack which required
+   *       supplying type-checked expressions, this architecture handles both parsed-only and
+   *       type-checked expressions.
+   * </ul>
+   */
+  public static CelBuilder plannerCelBuilder() {
+    return CelImpl.newBuilder(
+            CelCompilerImpl.newBuilder(
+                CelParserImpl.newBuilder(),
+                CelCheckerLegacyImpl.newBuilder().setStandardEnvironmentEnabled(true)),
+            CelRuntimeImpl.newBuilder())
+        // CEL-Internal-2
+        .setOptions(CelOptions.current().enableHeterogeneousNumericComparisons(true).build());
+  }
+
   /** Combines a prebuilt {@link CelCompiler} and {@link CelRuntime} into {@link Cel}. */
   public static Cel combine(CelCompiler celCompiler, CelRuntime celRuntime) {
     return CelImpl.combine(celCompiler, celRuntime);

bundle/src/main/java/dev/cel/bundle/CelFactory.java

@@ -56,3 +56,8 @@ java_library(
     name = "comprehensions",
     exports = ["//extensions/src/main/java/dev/cel/extensions:comprehensions"],
 )
+
+java_library(
+    name = "native",
+    exports = ["//extensions/src/main/java/dev/cel/extensions:native"],
+)

extensions/BUILD.bazel

@@ -34,6 +34,7 @@ java_library(
         ":encoders",
         ":lists",
         ":math",
+        ":native",
         ":optional_library",
         ":protos",
         ":regex",
@@ -42,6 +43,7 @@ java_library(
         ":strings",
         "//common:options",
         "//extensions:extension_library",
+        "//runtime",
         "@maven//:com_google_errorprone_error_prone_annotations",
         "@maven//:com_google_guava_guava",
     ],
@@ -318,3 +320,20 @@ java_library(
         "@maven//:com_google_guava_guava",
     ],
 )
+
+java_library(
+    name = "native",
+    srcs = ["CelNativeExtensions.java"],
+    tags = [
+    ],
+    deps = [
+        "//common/exceptions:attribute_not_found",
+        "//common/types",
+        "//common/types:type_providers",
+        "//common/values",
+        "//common/values:cel_byte_string",
+        "//common/values:cel_value_provider",
+        "//runtime",
+        "@maven//:com_google_guava_guava",
+    ],
+)

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

@@ -22,6 +22,7 @@
 import com.google.errorprone.annotations.InlineMe;
 import dev.cel.common.CelOptions;
 import dev.cel.extensions.CelMathExtensions.Function;
+import dev.cel.runtime.CelRuntimeLibrary;
 import java.util.Set;
 
 /**
@@ -350,6 +351,67 @@ public static CelComprehensionsExtensions comprehensions() {
     return COMPREHENSIONS_EXTENSIONS;
   }
 
+  /**
+   * Creates a {@link CelRuntimeLibrary} that registers the given POJO classes.
+   *
+   * <p>All POJO classes are exposed to CEL using their fully qualified canonical name. For example,
+   * if you have a class `com.example.Account`:
+   *
+   * <pre>{@code
+   * package com.example;
+   * public class Account {
+   *   public int id;
+   * }
+   * }</pre>
+   *
+   * The type `com.example.Account` would be exported to CEL using its full name. If you set the
+   * container to `com.example` on the compiler, you can use it simply as `Account`: `Account{id:
+   * 1234}` would create a new `Account` instance with the `id` field populated.
+   *
+   * <p>Properties are discovered by scanning public fields and public getter methods. For field
+   * selection (reading) and object creation (writing), resolution happens in the following order of
+   * precedence:
+   *
+   * <ol>
+   *   <li>Standard JavaBeans getter (e.g., `getFoo()`) or setter (e.g., `setFoo(...)`)
+   *   <li>Boolean getter (e.g., `isFoo()`) for boolean properties
+   *   <li>Prefix-less getter (e.g., `foo()`) matching a declared field name
+   *   <li>Public field directly (e.g., `public String foo`)
+   * </ol>
+   *
+   * <p>The type-mapping between Java and CEL is as follows:
+   *
+   * <table>
+   *   <tr><th>Java type</th><th>CEL type</th></tr>
+   *   <tr><td>boolean, Boolean</td><td>bool</td></tr>
+   *   <tr><td>byte[]</td><td>bytes</td></tr>
+   *   <tr><td>float, Float, double, Double</td><td>double</td></tr>
+   *   <tr><td>int, Integer, long, Long</td><td>int</td></tr>
+   *   <tr><td>com.google.common.primitives.UnsignedLong</td><td>uint</td></tr>
+   *   <tr><td>String</td><td>string</td></tr>
+   *   <tr><td>java.time.Duration</td><td>duration</td></tr>
+   *   <tr><td>java.time.Instant</td><td>timestamp</td></tr>
+   *   <tr><td>java.util.List</td><td>list</td></tr>
+   *   <tr><td>java.util.Map</td><td>map</td></tr>
+   *   <tr><td>java.util.Optional</td><td>optional_type</td></tr>
+   * </table>
+   *
+   * <p>Note: This is only supported for the planner runtime (e.g., {@code
+   * CelRuntimeFactory.plannerRuntimeBuilder()}).
+   *
+   * <p>Note: Native Java arrays (except `byte[]`) are not supported. Use {@link java.util.List}
+   * instead.
+   *
+   * <p>Note: If there is a name collision with a Protocol Buffer type, the protobuf type will take
+   * precedence.
+   *
+   * @throws IllegalArgumentException if a class contains any fields or getters with types that
+   *     cannot be mapped to CEL, or if an anonymous or local class is passed.
+   */
+  public static CelRuntimeLibrary nativeTypes(Class<?>... classes) {
+    return CelNativeExtensions.nativeTypes(classes);
+  }
+
   /**
    * Retrieves all function names used by every extension libraries.
    *