Add missing features for public preview

Author: wu-hui

firebase-firestore/src/main/java/com/google/firebase/firestore/model/Values.kt

@@ -761,6 +761,19 @@ object Values {
     return Timestamp.newBuilder().setSeconds(seconds).setNanos(truncatedNanoseconds).build()
   }
 
+  @JvmStatic
+  fun getVectorValue(value: Value): DoubleArray? {
+    if (value.valueTypeCase != ValueTypeCase.MAP_VALUE || !isVectorValue(value)) {
+      return null
+    }
+
+    return value.mapValue.fieldsMap[VECTOR_MAP_VECTORS_KEY]
+      ?.arrayValue
+      ?.valuesList
+      ?.map { it.doubleValue }
+      ?.toDoubleArray()
+  }
+
   /**
    * Ensures that the date and time are within what we consider valid ranges.
    *

firebase-firestore/src/main/java/com/google/firebase/firestore/pipeline/EvaluateResult.kt

@@ -19,7 +19,8 @@ import com.google.firebase.firestore.model.Values.encodeValue
 import com.google.firestore.v1.Value
 import com.google.protobuf.Timestamp
 
-internal sealed class EvaluateResult(val value: Value?) {
+internal sealed class EvaluateResult {
+  abstract val value: Value?
   abstract val isError: Boolean
   abstract val isSuccess: Boolean
   abstract val isUnset: Boolean
@@ -49,19 +50,21 @@ internal sealed class EvaluateResult(val value: Value?) {
   }
 }
 
-internal class EvaluateResultValue(value: Value) : EvaluateResult(value) {
+internal data class EvaluateResultValue(override val value: Value) : EvaluateResult() {
   override val isSuccess: Boolean = true
   override val isError: Boolean = false
   override val isUnset: Boolean = false
 }
 
-internal object EvaluateResultError : EvaluateResult(null) {
+internal object EvaluateResultError : EvaluateResult() {
+  override val value: Value? = null
   override val isSuccess: Boolean = false
   override val isError: Boolean = true
   override val isUnset: Boolean = false
 }
 
-internal object EvaluateResultUnset : EvaluateResult(null) {
+internal object EvaluateResultUnset : EvaluateResult() {
+  override val value: Value? = null
   override val isSuccess: Boolean = false
   override val isError: Boolean = false
   override val isUnset: Boolean = true

firebase-firestore/src/main/java/com/google/firebase/firestore/pipeline/FunctionRegistry.kt

@@ -77,6 +77,7 @@ internal object FunctionRegistry {
       "array_contains" to evaluateArrayContains,
       "array_contains_any" to evaluateArrayContainsAny,
       "array_contains_all" to evaluateArrayContainsAll,
+      "array_get" to evaluateArrayGet,
       "array_length" to evaluateArrayLength,
       "timestamp_add" to evaluateTimestampAdd,
       "timestamp_sub" to evaluateTimestampSub,
@@ -104,7 +105,6 @@ internal object FunctionRegistry {
       "split" to evaluateSplit,
       "substring" to evaluateSubstring,
       "ltrim" to evaluateLTrim,
-      "rtrim" to evaluateRTrim,
-      "str_join" to evaluateStrJoin
+      "rtrim" to evaluateRTrim
     )
 }