Add CelType as a property to CelValue

PiperOrigin-RevId: 584665130

Author: l46kok

common/src/main/java/dev/cel/common/values/BUILD.bazel

@@ -37,6 +37,7 @@ java_library(
     ],
     deps = [
         "//common/annotations",
+        "//common/types:type_providers",
         "@maven//:com_google_errorprone_error_prone_annotations",
     ],
 )
@@ -51,6 +52,7 @@ java_library(
         ":cel_value",
         "//:auto_value",
         "//common/annotations",
+        "//common/types",
         "//common/types:type_providers",
         "@maven//:com_google_errorprone_error_prone_annotations",
         "@maven//:com_google_guava_guava",

common/src/main/java/dev/cel/common/values/BoolValue.java

@@ -16,6 +16,8 @@
 
 import com.google.auto.value.AutoValue;
 import com.google.errorprone.annotations.Immutable;
+import dev.cel.common.types.CelType;
+import dev.cel.common.types.SimpleType;
 
 /** BoolValue is a simple CelValue wrapper around Java booleans. */
 @AutoValue
@@ -30,6 +32,11 @@ public boolean isZeroValue() {
     return !value();
   }
 
+  @Override
+  public CelType celType() {
+    return SimpleType.BOOL;
+  }
+
   public static BoolValue create(Boolean value) {
     return new AutoValue_BoolValue(value);
   }

common/src/main/java/dev/cel/common/values/BytesValue.java

@@ -16,6 +16,8 @@
 
 import com.google.auto.value.AutoValue;
 import com.google.errorprone.annotations.Immutable;
+import dev.cel.common.types.CelType;
+import dev.cel.common.types.SimpleType;
 
 /** BytesValue is a simple CelValue wrapper around CelByteString (immutable byte string). */
 @AutoValue
@@ -30,6 +32,11 @@ public boolean isZeroValue() {
     return value().isEmpty();
   }
 
+  @Override
+  public CelType celType() {
+    return SimpleType.BYTES;
+  }
+
   public static BytesValue create(CelByteString value) {
     return new AutoValue_BytesValue(value);
   }

common/src/main/java/dev/cel/common/values/CelValue.java

@@ -16,6 +16,7 @@
 
 import com.google.errorprone.annotations.Immutable;
 import dev.cel.common.annotations.Internal;
+import dev.cel.common.types.CelType;
 
 /**
  * A representation of a CEL value for the runtime. Clients should never directly extend from
@@ -35,8 +36,10 @@ public abstract class CelValue {
   /** Returns true if the {@link #value()} is a zero value for its type. */
   public abstract boolean isZeroValue();
 
+  /** The CelType that represents this value. */
+  public abstract CelType celType();
+
   // TOOD(b/309695452): Add CelEquals method
-  // TODO: Add a getter for CelType
 
   public CelValue() {}
 }

common/src/main/java/dev/cel/common/values/DoubleValue.java

@@ -15,6 +15,8 @@
 package dev.cel.common.values;
 
 import com.google.errorprone.annotations.Immutable;
+import dev.cel.common.types.CelType;
+import dev.cel.common.types.SimpleType;
 
 /** DoubleValue is a simple CelValue wrapper around Java doubles. */
 @Immutable
@@ -35,6 +37,11 @@ public boolean isZeroValue() {
     return value() == 0;
   }
 
+  @Override
+  public CelType celType() {
+    return SimpleType.DOUBLE;
+  }
+
   public static DoubleValue create(double value) {
     return new DoubleValue(value);
   }

common/src/main/java/dev/cel/common/values/DurationValue.java

@@ -16,6 +16,8 @@
 
 import com.google.auto.value.AutoValue;
 import com.google.errorprone.annotations.Immutable;
+import dev.cel.common.types.CelType;
+import dev.cel.common.types.SimpleType;
 import java.time.Duration;
 
 /** DurationValue is a simple CelValue wrapper around {@link java.time.Duration} */
@@ -31,6 +33,11 @@ public boolean isZeroValue() {
     return value().isZero();
   }
 
+  @Override
+  public CelType celType() {
+    return SimpleType.DURATION;
+  }
+
   public static DurationValue create(Duration value) {
     return new AutoValue_DurationValue(value);
   }

common/src/main/java/dev/cel/common/values/EnumValue.java

@@ -16,6 +16,8 @@
 
 import com.google.auto.value.AutoValue;
 import com.google.errorprone.annotations.Immutable;
+import dev.cel.common.types.CelType;
+import dev.cel.common.types.SimpleType;
 
 /**
  * EnumValue is a simple CelValue wrapper around Java enums.
@@ -35,6 +37,12 @@ public boolean isZeroValue() {
     return false;
   }
 
+  @Override
+  public CelType celType() {
+    // (b/178627883) Strongly typed enum is not supported yet
+    return SimpleType.INT;
+  }
+
   public static <E extends Enum<E>> EnumValue<E> create(Enum<E> value) {
     return new AutoValue_EnumValue<>(value);
   }

common/src/main/java/dev/cel/common/values/ErrorValue.java

@@ -16,6 +16,8 @@
 
 import com.google.auto.value.AutoValue;
 import dev.cel.common.annotations.Internal;
+import dev.cel.common.types.CelType;
+import dev.cel.common.types.SimpleType;
 
 /**
  * CelErrorValue represent the intermediate error that occurs during evaluation in the form of Java
@@ -39,6 +41,11 @@ public boolean isZeroValue() {
     return false;
   }
 
+  @Override
+  public CelType celType() {
+    return SimpleType.ERROR;
+  }
+
   public static ErrorValue create(Exception value) {
     return new AutoValue_ErrorValue(value);
   }

common/src/main/java/dev/cel/common/values/IntValue.java

@@ -15,6 +15,8 @@
 package dev.cel.common.values;
 
 import com.google.errorprone.annotations.Immutable;
+import dev.cel.common.types.CelType;
+import dev.cel.common.types.SimpleType;
 
 /** IntValue is a simple CelValue wrapper around Java longs. */
 @Immutable
@@ -35,6 +37,11 @@ public boolean isZeroValue() {
     return value() == 0;
   }
 
+  @Override
+  public CelType celType() {
+    return SimpleType.INT;
+  }
+
   public static IntValue create(long value) {
     return new IntValue(value);
   }

common/src/main/java/dev/cel/common/values/ListValue.java

@@ -17,6 +17,9 @@
 import com.google.errorprone.annotations.CanIgnoreReturnValue;
 import com.google.errorprone.annotations.DoNotCall;
 import com.google.errorprone.annotations.Immutable;
+import dev.cel.common.types.CelType;
+import dev.cel.common.types.ListType;
+import dev.cel.common.types.SimpleType;
 import java.util.Collection;
 import java.util.Comparator;
 import java.util.List;
@@ -31,6 +34,7 @@
  */
 @Immutable
 public abstract class ListValue<E extends CelValue> extends CelValue implements List<E> {
+  private static final ListType LIST_TYPE = ListType.create(SimpleType.DYN);
 
   @Override
   @SuppressWarnings("Immutable") // ListValue APIs prohibit mutation.
@@ -41,6 +45,11 @@ public boolean isZeroValue() {
     return isEmpty();
   }
 
+  @Override
+  public CelType celType() {
+    return LIST_TYPE;
+  }
+
   /**
    * Guaranteed to throw an exception and leave the list unmodified.
    *