/** * @flow * WriteBatch representation wrapper */ import { buildNativeMap } from './utils/serialize'; import { isObject, isString } from '../../utils'; import { getNativeModule } from '../../utils/native'; import type DocumentReference from './DocumentReference'; import type Firestore from './'; import type { FirestoreWriteOptions } from '../../types'; type DocumentWrite = { data?: Object, options?: Object, path: string, type: 'DELETE' | 'SET' | 'UPDATE', } /** * @class WriteBatch */ export default class WriteBatch { _firestore: Firestore; _writes: DocumentWrite[]; constructor(firestore: Firestore) { this._firestore = firestore; this._writes = []; } commit(): Promise { return getNativeModule(this._firestore).documentBatch(this._writes); } delete(docRef: DocumentReference): WriteBatch { // TODO: Validation // validate.isDocumentReference('docRef', docRef); // validate.isOptionalPrecondition('deleteOptions', deleteOptions); this._writes.push({ path: docRef.path, type: 'DELETE', }); return this; } set(docRef: DocumentReference, data: Object, writeOptions?: FirestoreWriteOptions) { // TODO: Validation // validate.isDocumentReference('docRef', docRef); // validate.isDocument('data', data); // validate.isOptionalPrecondition('writeOptions', writeOptions); const nativeData = buildNativeMap(data); this._writes.push({ data: nativeData, options: writeOptions, path: docRef.path, type: 'SET', }); return this; } update(docRef: DocumentReference, ...args: any[]): WriteBatch { // TODO: Validation // validate.isDocumentReference('docRef', docRef); let data = {}; if (args.length === 1) { if (!isObject(args[0])) { throw new Error('DocumentReference.update failed: If using two arguments, the second must be an object.'); } data = args[0]; } else if (args.length % 2 === 1) { throw new Error('DocumentReference.update failed: Must have a document reference, followed by either a single object argument, or equal numbers of key/value pairs.'); } else { for (let i = 0; i < args.length; i += 2) { const key = args[i]; const value = args[i + 1]; if (!isString(key)) { throw new Error(`DocumentReference.update failed: Argument at index ${i + 1} must be a string`); } data[key] = value; } } const nativeData = buildNativeMap(data); this._writes.push({ data: nativeData, path: docRef.path, type: 'UPDATE', }); return this; } }