@firebaseextensions/firestore-bigquery-change-tracker 1.1.14 → 1.1.15

package/lib/bigquery/partitioning.js

@@ -4,6 +4,8 @@ exports.Partitioning = void 0;
 const firebase = require("firebase-admin");
 const logs = require("../logs");
 const schema_1 = require("./schema");
+const bigquery_1 = require("@google-cloud/bigquery");
+const types_1 = require("../types");
 class Partitioning {
     constructor(config, table, schema) {
         this.config = config;
@@ -32,7 +34,11 @@ class Partitioning {
         return metadata.schema.fields;
     }
     isValidPartitionTypeDate(value) {
-        return value instanceof firebase.firestore.Timestamp;
+        /* Check if valid timestamp value from sdk */
+        if (value instanceof firebase.firestore.Timestamp)
+            return true;
+        /* Check if valid date/time value from console */
+        return Object.prototype.toString.call(value) === "[object Date]";
     }
     hasHourAndDatePartitionConfig() {
         if (this.config.timePartitioning === "HOUR" &&
@@ -110,6 +116,19 @@ class Partitioning {
             return false;
         return this.hasValidCustomPartitionConfig();
     }
+    convertDateValue(fieldValue) {
+        const { timePartitioningFieldType } = this.config;
+        /* Return as Datetime value */
+        if (timePartitioningFieldType === types_1.PartitionFieldType.DATETIME) {
+            return bigquery_1.BigQuery.datetime(fieldValue.toISOString()).value;
+        }
+        /* Return as Date value */
+        if (timePartitioningFieldType === types_1.PartitionFieldType.DATE) {
+            return bigquery_1.BigQuery.date(fieldValue.toISOString().substring(0, 10)).value;
+        }
+        /* Return as Timestamp  */
+        return bigquery_1.BigQuery.timestamp(fieldValue).value;
+    }
     /*
       Extracts a valid Partition field from the Document Change Event.
       Matches result based on a pre-defined Firestore field matching the event data object.
@@ -130,8 +149,14 @@ class Partitioning {
         if (this.isValidPartitionTypeString(fieldValue)) {
             return { [fieldName]: fieldValue };
         }
-        if (this.isValidPartitionTypeDate(fieldValue))
-            return { [fieldName]: fieldValue.toDate() };
+        if (this.isValidPartitionTypeDate(fieldValue)) {
+            /* Return converted console value */
+            if (fieldValue.toDate) {
+                return { [fieldName]: this.convertDateValue(fieldValue.toDate()) };
+            }
+            /* Return standard date value */
+            return { [fieldName]: fieldValue };
+        }
         logs.firestoreTimePartitionFieldError(event.documentName, fieldName, firestoreFieldName, fieldValue);
         return {};
     }

package/lib/types.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PartitionFieldType = void 0;
+var PartitionFieldType;
+(function (PartitionFieldType) {
+    PartitionFieldType["DATE"] = "DATE";
+    PartitionFieldType["DATETIME"] = "DATETIME";
+    PartitionFieldType["TIMESTAMP"] = "TIMESTAMP";
+})(PartitionFieldType = exports.PartitionFieldType || (exports.PartitionFieldType = {}));

package/package.json

@@ -5,7 +5,7 @@
     "url": "github.com/firebase/extensions.git",
     "directory": "firestore-bigquery-export/firestore-bigquery-change-tracker"
   },
-  "version": "1.1.14",
+  "version": "1.1.15",
   "description": "Core change-tracker library for Cloud Firestore Collection BigQuery Exports",
   "main": "./lib/index.js",
   "scripts": {