Firestore

Author: samtstern

firestore/main/index.js

@@ -1,31 +1,32 @@
 const debug = require('debug')('firestore-snippets-node');
 
 // [START firestore_deps]
-const admin = require('firebase-admin');
+import { initializeApp, applicationDefault, cert } from 'firebase-admin/app';
+import { getFirestore, Timestamp, FieldValue } from 'firebase-admin/firestore';
 // [END firestore_deps]
 
 // We supress these logs when not in NODE_ENV=debug for cleaner Mocha output
 const console = {log: debug};
 
-async function initializeApp() {
+async function demoInitializeApp() {
   process.env.GCLOUD_PROJECT = 'firestorebeta1test2';
   // [START initialize_app]
 
-  admin.initializeApp({
-    credential: admin.credential.applicationDefault()
+  initializeApp({
+    credential: applicationDefault()
   });
 
-  const db = admin.firestore();
+  const db = getFirestore();
   // [END initialize_app]
   return db;
 }
 
 async function initializeAppFunctions() {
   process.env.GCLOUD_PROJECT = 'firestorebeta1test2';
   // [START initialize_app_functions]
-  admin.initializeApp();
+  initializeApp();
 
-  const db = admin.firestore();
+  const db = getFirestore();
 
   // [END initialize_app_functions]
   return db;
@@ -36,11 +37,11 @@ async function initializeAppSA() {
 
   const serviceAccount = require('./path/to/serviceAccountKey.json');
 
-  admin.initializeApp({
-    credential: admin.credential.cert(serviceAccount)
+  initializeApp({
+    credential: cert(serviceAccount)
   });
 
-  const db = admin.firestore();
+  const db = getFirestore();
 
   // [END initialize_app_service_account]
   return db;
@@ -171,7 +172,7 @@ async function dataTypes(db) {
     stringExample: 'Hello, World!',
     booleanExample: true,
     numberExample: 3.14159265,
-    dateExample: admin.firestore.Timestamp.fromDate(new Date('December 10, 1815')),
+    dateExample: Timestamp.fromDate(new Date('December 10, 1815')),
     arrayExample: [5, true, 'hello'],
     nullExample: null,
     objectExample: {
@@ -249,11 +250,11 @@ async function updateDocumentArray(db) {
 
   // Atomically add a new region to the "regions" array field.
   const unionRes = await washingtonRef.update({
-    regions: admin.firestore.FieldValue.arrayUnion('greater_virginia')
+    regions: FieldValue.arrayUnion('greater_virginia')
   });
   // Atomically remove a region from the "regions" array field.
   const removeRes = await washingtonRef.update({
-    regions: admin.firestore.FieldValue.arrayRemove('east_coast')
+    regions: FieldValue.arrayRemove('east_coast')
   });
   // [END firestore_data_set_array_operations]
   // [END update_document_array]
@@ -270,7 +271,7 @@ async function updateDocumentIncrement(db) {
 
   // Atomically increment the population of the city by 50.
   const res = await washingtonRef.update({
-    population: admin.firestore.FieldValue.increment(50)
+    population: FieldValue.increment(50)
   });
   // [END firestore_data_set_numeric_increment]
   // [END update_document_increment]
@@ -314,9 +315,6 @@ async function updateServerTimestamp(db) {
 
   // [START update_with_server_timestamp]
   // [START firestore_data_set_server_timestamp]
-  // Get the `FieldValue` object
-  const FieldValue = admin.firestore.FieldValue;
-
   // Create a document reference
   const docRef = db.collection('objects').doc('some-id');
 
@@ -334,9 +332,6 @@ async function updateDeleteField(db) {
   const admin = require('firebase-admin');
   // [START update_delete_field]
   // [START firestore_data_delete_field]
-  // Get the `FieldValue` object
-  const FieldValue = admin.firestore.FieldValue;
-
   // Create a document reference
   const cityRef = db.collection('cities').doc('BJ');
 
@@ -1058,8 +1053,8 @@ async function deleteQueryBatch(db, query, resolve) {
 
 describe('Firestore Smoketests', () => {
 
-  admin.initializeApp();
-  const db = admin.firestore();
+  initializeApp();
+  const db = getFirestore();
 
   it('should get an empty document', () => {
     return getDocumentEmpty(db);

firestore/solution-aggregation/functions/index.js

@@ -1,7 +1,7 @@
 const functions = require('firebase-functions');
-const admin = require('firebase-admin');
+import { getFirestore } from 'firebase-admin/firestore';
 
-const db = admin.firestore();
+const db = getFirestore();
 // [START_EXCLUDE]
 const settings = {timestampsInSnapshots: true};
 db.settings(settings);

firestore/solution-deletes/functions/index.js

@@ -1,8 +1,10 @@
-const admin = require('firebase-admin');
+import { initializeApp } from 'firebase-admin/app';
+import { getAuth } from 'firebase-admin/auth';
+
 const firebase_tools = require('firebase-tools');
 const functions = require('firebase-functions');
 
-admin.initializeApp();
+initializeApp();
 
 
 /**
@@ -17,8 +19,7 @@ admin.initializeApp();
 exports.mintAdminToken = functions.https.onCall(async (data, context) => {
   const uid = data.uid;
 
-  const token = await admin
-    .auth()
+  const token = await getAuth()
     .createCustomToken(uid, { admin: true });
 
   return { token };

firestore/solution-sharded-timestamp/nonShardedTimestamps.js

@@ -1,9 +1,11 @@
+import { initializeApp } from 'firebase-admin/app';
+import { getFirestore, Timestamp } from 'firebase-admin/firestore';
 const util = require('util');
-const admin = require('firebase-admin');
-admin.initializeApp();
+
+initializeApp();
 
 // Create a new client
-const fs = admin.firestore();
+const fs = getFirestore();
 
 // [START fs_sharded_timestamps_pre_insert]
 async function insertData() {
@@ -16,7 +18,7 @@ async function insertData() {
       },
       exchange: 'EXCHG1',
       instrumentType: 'commonstock',
-      timestamp: admin.firestore.Timestamp.fromMillis(
+      timestamp: Timestamp.fromMillis(
           Date.parse('2019-01-01T13:45:23.010Z'))
     },
     {
@@ -27,7 +29,7 @@ async function insertData() {
       },
       exchange: 'EXCHG2',
       instrumentType: 'commonstock',
-      timestamp: admin.firestore.Timestamp.fromMillis(
+      timestamp: Timestamp.fromMillis(
           Date.parse('2019-01-01T13:45:23.101Z'))
     },
     {
@@ -38,7 +40,7 @@ async function insertData() {
       },
       exchange: 'EXCHG1',
       instrumentType: 'etf',
-      timestamp: admin.firestore.Timestamp.fromMillis(
+      timestamp: Timestamp.fromMillis(
           Date.parse('2019-01-01T13:45:23.001Z'))
     }
   ];

firestore/solution-sharded-timestamp/shardedTimestamps.js

@@ -1,9 +1,11 @@
 const util = require('util');
-const admin = require('firebase-admin');
-admin.initializeApp();
+import { initializeApp } from 'firebase-admin/app';
+import { getFirestore, Timestamp } from 'firebase-admin/firestore';
+
+initializeApp();
 
 // Create a new client
-const fs = admin.firestore();
+const fs = getFirestore();
 const MAX_IN_VALUES = 10;
 
 // [START fs_sharded_timestamps_define_shards]
@@ -44,7 +46,7 @@ async function insertData() {
       },
       exchange: 'EXCHG1',
       instrumentType: 'commonstock',
-      timestamp: admin.firestore.Timestamp.fromMillis(
+      timestamp: Timestamp.fromMillis(
           Date.parse('2019-01-01T13:45:23.010Z'))
     },
     {
@@ -56,7 +58,7 @@ async function insertData() {
       },
       exchange: 'EXCHG2',
       instrumentType: 'commonstock',
-      timestamp: admin.firestore.Timestamp.fromMillis(
+      timestamp: Timestamp.fromMillis(
           Date.parse('2019-01-01T13:45:23.101Z'))
     },
     {
@@ -68,7 +70,7 @@ async function insertData() {
       },
       exchange: 'EXCHG1',
       instrumentType: 'etf',
-      timestamp: admin.firestore.Timestamp.fromMillis(
+      timestamp: Timestamp.fromMillis(
           Date.parse('2019-01-01T13:45:23.001Z'))
     }
   ];