dev.cel.common.CelProtoJsonAdapter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of runtime Show documentation
Common Expression Language Runtime for Java
There is a newer version: 0.7.1
// Copyright 2024 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package dev.cel.common;

import com.google.common.primitives.UnsignedLong;
import com.google.errorprone.annotations.Immutable;
import com.google.protobuf.ByteString;
import com.google.protobuf.Duration;
import com.google.protobuf.ListValue;
import com.google.protobuf.NullValue;
import com.google.protobuf.Struct;
import com.google.protobuf.Timestamp;
import com.google.protobuf.Value;
import com.google.protobuf.util.Durations;
import com.google.protobuf.util.Timestamps;
import java.util.Base64;
import java.util.Map;

/**
 * {@code CelProtoJsonAdapter} is a utility to handle conversion from Java native objects
 * representing CEL values to Protobuf's structured value which maps to JSON object schema.
 */
@Immutable
public final class CelProtoJsonAdapter {
  private static final long JSON_MAX_INT_VALUE = (1L << 53) - 1;
  private static final long JSON_MIN_INT_VALUE = -JSON_MAX_INT_VALUE;
  private static final UnsignedLong JSON_MAX_UINT_VALUE =
      UnsignedLong.fromLongBits(JSON_MAX_INT_VALUE);

  /**
   * Adapts a map to a JSON Struct.
   *
   * @throws ClassCastException If the key is not a string literal
   * @throws IllegalArgumentException If any of the map's value is not convertible to a canonical
   *     JSON representation defined by protobuf.
   */
  public static  Struct adaptToJsonStructValue(Map map) {
    Struct.Builder struct = Struct.newBuilder();
    for (Map.Entry entry : map.entrySet()) {
      String key = entry.getKey();
      Object keyValue = entry.getValue();

      struct.putFields(key, adaptValueToJsonValue(keyValue));
    }
    return struct.build();
  }

  /**
   * Adapts a native Java object to a JSON value.
   *
   * @throws IllegalArgumentException If the value is not convertible to a canonical JSON *
   *     representation defined by protobuf.
   */
  @SuppressWarnings("unchecked")
  public static Value adaptValueToJsonValue(Object value) {
    Value.Builder json = Value.newBuilder();
    if (value == null || value instanceof NullValue) {
      return json.setNullValue(NullValue.NULL_VALUE).build();
    }
    if (value instanceof Boolean) {
      return json.setBoolValue((Boolean) value).build();
    }
    if (value instanceof Integer || value instanceof Long) {
      long longValue = ((Number) value).longValue();
      if (longValue < JSON_MIN_INT_VALUE || longValue > JSON_MAX_INT_VALUE) {
        return json.setStringValue(Long.toString(longValue)).build();
      }
      return json.setNumberValue((double) longValue).build();
    }
    if (value instanceof UnsignedLong) {
      if (((UnsignedLong) value).compareTo(JSON_MAX_UINT_VALUE) > 0) {
        return json.setStringValue(((UnsignedLong) value).toString()).build();
      }
      return json.setNumberValue((double) ((UnsignedLong) value).longValue()).build();
    }
    if (value instanceof Float || value instanceof Double) {
      return json.setNumberValue(((Number) value).doubleValue()).build();
    }
    if (value instanceof ByteString) {
      return json.setStringValue(
              Base64.getEncoder().encodeToString(((ByteString) value).toByteArray()))
          .build();
    }
    if (value instanceof String) {
      return json.setStringValue((String) value).build();
    }
    if (value instanceof Map) {
      Struct struct = adaptToJsonStructValue((Map) value);
      return json.setStructValue(struct).build();
    }
    if (value instanceof Iterable) {
      ListValue listValue = adaptToJsonListValue((Iterable