Update the "sets" extension to be compatible with CelEnvironmentExporter

PiperOrigin-RevId: 789087644

Author: dmitriplotnikov

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

@@ -118,7 +118,8 @@ public Builder addStandardExtensions(CelOptions options) {
           CelExtensions.getExtensionLibrary("lists", options),
           CelExtensions.getExtensionLibrary("math", options),
           CelExtensions.getExtensionLibrary("protos", options),
-          CelExtensions.getExtensionLibrary("regex", options));
+          CelExtensions.getExtensionLibrary("regex", options),
+          CelExtensions.getExtensionLibrary("sets", options));
       // TODO: add support for remaining standard extensions
       return this;
     }

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

@@ -203,6 +203,7 @@ java_library(
         "//common/internal:dynamic_proto",
         "//common/types",
         "//compiler:compiler_builder",
+        "//extensions:extension_library",
         "//runtime",
         "//runtime:proto_message_runtime_equality",
         "@maven//:com_google_errorprone_error_prone_annotations",

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

@@ -315,6 +315,8 @@ public static CelExtensionLibrary<? extends CelExtensionLibrary.FeatureSet> getE
         return CelProtoExtensions.library();
       case "regex":
         return CelRegexExtensions.library();
+      case "sets":
+        return CelSetsExtensions.library(options);
       // TODO: add support for remaining standard extensions
       default:
         throw new IllegalArgumentException("Unknown standard extension '" + name + "'");

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

@@ -41,7 +41,8 @@
  * rewrite the AST into a map to achieve a O(1) lookup.
  */
 @Immutable
-final class CelSetsExtensions implements CelCompilerLibrary, CelRuntimeLibrary {
+final class CelSetsExtensions
+    implements CelCompilerLibrary, CelRuntimeLibrary, CelExtensionLibrary.FeatureSet {
 
   private static final String SET_CONTAINS_OVERLOAD_DOC =
       "Returns whether the first list argument contains all elements in the second list"
@@ -90,6 +91,28 @@ final class CelSetsExtensions implements CelCompilerLibrary, CelRuntimeLibrary {
                   ListType.create(TypeParamType.create("T")),
                   ListType.create(TypeParamType.create("T")))));
 
+  private static final class Library implements CelExtensionLibrary<CelSetsExtensions> {
+    private final CelSetsExtensions version0;
+
+    Library(CelOptions celOptions) {
+      version0 = new CelSetsExtensions(celOptions);
+    }
+
+    @Override
+    public String name() {
+      return "sets";
+    }
+
+    @Override
+    public ImmutableSet<CelSetsExtensions> versions() {
+      return ImmutableSet.of(version0);
+    }
+  }
+
+  static CelExtensionLibrary<CelSetsExtensions> library(CelOptions options) {
+    return new Library(options);
+  }
+
   private final ImmutableSet<SetsFunction> functions;
   private final SetsExtensionsRuntimeImpl setsExtensionsRuntime;
 
@@ -105,6 +128,16 @@ final class CelSetsExtensions implements CelCompilerLibrary, CelRuntimeLibrary {
     this.setsExtensionsRuntime = new SetsExtensionsRuntimeImpl(runtimeEquality, functions);
   }
 
+  @Override
+  public int version() {
+    return 0;
+  }
+
+  @Override
+  public ImmutableSet<CelFunctionDecl> functions() {
+    return ImmutableSet.copyOf(FUNCTION_DECL_MAP.values());
+  }
+
   @Override
   public void setCheckerOptions(CelCheckerBuilder checkerBuilder) {
     functions.forEach(

extensions/src/test/java/dev/cel/extensions/CelSetsExtensionsTest.java

@@ -72,6 +72,17 @@ public final class CelSetsExtensionsTest {
                   Math::toIntExact))
           .build();
 
+  @Test
+  public void library() {
+    CelExtensionLibrary<?> library =
+        CelExtensions.getExtensionLibrary("sets", CelOptions.DEFAULT);
+    assertThat(library.name()).isEqualTo("sets");
+    assertThat(library.latest().version()).isEqualTo(0);
+    assertThat(library.version(0).functions().stream().map(CelFunctionDecl::name))
+        .containsExactly("sets.contains", "sets.equivalent", "sets.intersects");
+    assertThat(library.version(0).macros()).isEmpty();
+  }
+
   @Test
   public void contains_integerListWithSameValue_succeeds() throws Exception {
     ImmutableList<Integer> list = ImmutableList.of(1, 2, 3, 4);