Add setDoc, updateDoc, deleteDoc and addDoc

Author: schmidt-sebastian

packages/firestore/exp/index.node.ts

@@ -44,7 +44,13 @@ export {
 
 export { runTransaction, Transaction } from '../lite/src/api/transaction';
 
-export { getDoc } from './src/api/reference';
+export {
+  getDoc,
+  addDoc,
+  setDoc,
+  deleteDoc,
+  updateDoc
+} from './src/api/reference';
 
 export {
   FieldValue,

packages/firestore/exp/src/api/reference.ts

@@ -20,16 +20,28 @@
 import * as firestore from '../../index';
 
 import { Firestore } from './database';
-import { DocumentKeyReference } from '../../../src/api/user_data_reader';
+import {
+  DocumentKeyReference,
+  ParsedUpdateData
+} from '../../../src/api/user_data_reader';
 import { debugAssert } from '../../../src/util/assert';
 import { cast } from '../../../lite/src/api/util';
 import { DocumentSnapshot } from './snapshot';
 import {
+  applyFirestoreDataConverter,
   getDocViaSnapshotListener,
   SnapshotMetadata
 } from '../../../src/api/database';
 import { ViewSnapshot } from '../../../src/core/view_snapshot';
-import { DocumentReference } from '../../../lite/src/api/reference';
+import {
+  CollectionReference,
+  doc,
+  DocumentReference,
+  newUserDataReader
+} from '../../../lite/src/api/reference';
+import { invokeCommitRpc } from '../../../src/remote/datastore';
+import { DeleteMutation, Precondition } from '../../../src/model/mutation';
+import { FieldPath } from '../../../lite/src/api/field_path';
 
 export function getDoc<T>(
   reference: firestore.DocumentReference<T>
@@ -42,6 +54,131 @@ export function getDoc<T>(
   });
 }
 
+export function setDoc<T>(
+  reference: firestore.DocumentReference<T>,
+  data: T
+): Promise<void>;
+export function setDoc<T>(
+  reference: firestore.DocumentReference<T>,
+  data: Partial<T>,
+  options: firestore.SetOptions
+): Promise<void>;
+export function setDoc<T>(
+  reference: firestore.DocumentReference<T>,
+  data: T,
+  options?: firestore.SetOptions
+): Promise<void> {
+  const ref = cast<DocumentReference<T>>(reference, DocumentReference);
+  const firestore = cast(ref.firestore, Firestore);
+
+  const [convertedValue] = applyFirestoreDataConverter(
+    ref._converter,
+    data,
+    'setDoc'
+  );
+  const dataReader = newUserDataReader(firestore);
+  const parsed = dataReader.parseSetData(
+    'setDoc',
+    ref._key,
+    convertedValue,
+    options
+  );
+
+  return firestore
+    ._getFirestoreClient()
+    .then(firestoreClient =>
+      firestoreClient.write(parsed.toMutations(ref._key, Precondition.none()))
+    );
+}
+
+export function updateDoc(
+  reference: firestore.DocumentReference<unknown>,
+  data: firestore.UpdateData
+): Promise<void>;
+export function updateDoc(
+  reference: firestore.DocumentReference<unknown>,
+  field: string | firestore.FieldPath,
+  value: unknown,
+  ...moreFieldsAndValues: unknown[]
+): Promise<void>;
+export function updateDoc(
+  reference: firestore.DocumentReference<unknown>,
+  fieldOrUpdateData: string | firestore.FieldPath | firestore.UpdateData,
+  value?: unknown,
+  ...moreFieldsAndValues: unknown[]
+): Promise<void> {
+  const ref = cast<DocumentReference<unknown>>(reference, DocumentReference);
+  const firestore = cast(ref.firestore, Firestore);
+  const dataReader = newUserDataReader(firestore);
+
+  let parsed: ParsedUpdateData;
+  if (
+    typeof fieldOrUpdateData === 'string' ||
+    fieldOrUpdateData instanceof FieldPath
+  ) {
+    parsed = dataReader.parseUpdateVarargs(
+      'updateDoc',
+      ref._key,
+      fieldOrUpdateData,
+      value,
+      moreFieldsAndValues
+    );
+  } else {
+    parsed = dataReader.parseUpdateData(
+      'updateDoc',
+      ref._key,
+      fieldOrUpdateData
+    );
+  }
+
+  return firestore
+    ._getFirestoreClient()
+    .then(firestoreClient =>
+      firestoreClient.write(
+        parsed.toMutations(ref._key, Precondition.exists(true))
+      )
+    );
+}
+
+export function deleteDoc(
+  reference: firestore.DocumentReference
+): Promise<void> {
+  const ref = cast<DocumentReference<unknown>>(reference, DocumentReference);
+  const firestore = cast(ref.firestore, Firestore);
+  return firestore
+    ._getFirestoreClient()
+    .then(firestoreClient =>
+      firestoreClient.write([new DeleteMutation(ref._key, Precondition.none())])
+    );
+}
+
+export function addDoc<T>(
+  reference: firestore.CollectionReference<T>,
+  data: T
+): Promise<firestore.DocumentReference<T>> {
+  const collRef = cast<CollectionReference<T>>(reference, CollectionReference);
+  const firestore = cast(collRef, Firestore);
+  const docRef = doc(collRef);
+
+  const [convertedValue] = applyFirestoreDataConverter(
+    collRef._converter,
+    data,
+    'addDoc'
+  );
+
+  const dataReader = newUserDataReader(collRef.firestore);
+  const parsed = dataReader.parseSetData('addDoc', docRef._key, convertedValue);
+
+  return firestore
+    ._getFirestoreClient()
+    .then(firestoreClient =>
+      firestoreClient.write(
+        parsed.toMutations(docRef._key, Precondition.exists(false))
+      )
+    )
+    .then(() => docRef);
+}
+
 /**
  * Converts a ViewSnapshot that contains the single document specified by `ref`
  * to a DocumentSnapshot.

packages/firestore/exp/test/helpers.ts

@@ -26,6 +26,8 @@ import {
   DEFAULT_SETTINGS
 } from '../../test/integration/util/settings';
 import { collection } from '../../lite/src/api/reference';
+import { setDoc } from '../src/api/reference';
+import { AutoId } from '../../src/util/misc';
 
 let appCount = 0;
 
@@ -43,6 +45,14 @@ export async function withTestDbSettings(
   return fn(firestore);
 }
 
+export function withTestCollection(
+  fn: (collRef: firestore.CollectionReference) => void | Promise<void>
+): Promise<void> {
+  return withTestDb(db => {
+    return fn(collection(db, AutoId.newId()));
+  });
+}
+
 export function withTestDb(
   fn: (db: firestore.FirebaseFirestore) => void | Promise<void>
 ): Promise<void> {
@@ -54,3 +64,14 @@ export function withTestDoc(
 ): Promise<void> {
   return withTestDb(db => fn(doc(collection(db, 'test-collection'))));
 }
+
+export function withTestDocAndInitialData(
+  data: firestore.DocumentData,
+  fn: (doc: firestore.DocumentReference) => void | Promise<void>
+): Promise<void> {
+  return withTestDb(async db => {
+    const ref = doc(collection(db, 'test-collection'));
+    await setDoc(ref, data);
+    return fn(ref);
+  });
+}