Internal Changes

PiperOrigin-RevId: 941243631

Author: l46kok

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

@@ -52,7 +52,8 @@ public final class CelComprehensionsExtensions
   private static final TypeParamType TYPE_PARAM_V = TypeParamType.create("V");
   private static final MapType MAP_KV_TYPE = MapType.create(TYPE_PARAM_K, TYPE_PARAM_V);
 
-  enum Function {
+  /** Enumeration of functions for Comprehensions extension. */
+  public enum Function {
     MAP_INSERT(
         CelFunctionDecl.newFunctionDeclaration(
             MAP_INSERT_FUNCTION,
@@ -72,6 +73,10 @@ enum Function {
 
     private final CelFunctionDecl functionDecl;
 
+    public CelFunctionDecl functionDecl() {
+      return functionDecl;
+    }
+
     String getFunction() {
       return functionDecl.name();
     }
@@ -81,20 +86,25 @@ String getFunction() {
     }
   }
 
-  private static final CelExtensionLibrary<CelComprehensionsExtensions> LIBRARY =
-      new CelExtensionLibrary<CelComprehensionsExtensions>() {
-        private final CelComprehensionsExtensions version0 = new CelComprehensionsExtensions();
+  private static final class Library implements CelExtensionLibrary<CelComprehensionsExtensions> {
+    private final CelComprehensionsExtensions version0;
+
+    Library() {
+      version0 = new CelComprehensionsExtensions();
+    }
+
+    @Override
+    public String name() {
+      return "comprehensions";
+    }
 
-        @Override
-        public String name() {
-          return "comprehensions";
-        }
+    @Override
+    public ImmutableSet<CelComprehensionsExtensions> versions() {
+      return ImmutableSet.of(version0);
+    }
+  }
 
-        @Override
-        public ImmutableSet<CelComprehensionsExtensions> versions() {
-          return ImmutableSet.of(version0);
-        }
-      };
+  private static final Library LIBRARY = new Library();
 
   static CelExtensionLibrary<CelComprehensionsExtensions> library() {
     return LIBRARY;
@@ -103,7 +113,7 @@ static CelExtensionLibrary<CelComprehensionsExtensions> library() {
   private final ImmutableSet<Function> functions;
 
   CelComprehensionsExtensions() {
-    this.functions = ImmutableSet.copyOf(Function.values());
+    this.functions = ImmutableSet.of(Function.MAP_INSERT);
   }
 
   @Override