Add useCollectionData and useDocumentData

Author: chrisbianca

firestore/helpers/index.ts

@@ -0,0 +1,9 @@
+import { firestore } from 'firebase';
+
+export const snapshotToData = (
+  snapshot: firestore.DocumentSnapshot,
+  idField?: string
+) => ({
+  ...snapshot.data(),
+  ...(idField ? { [idField]: snapshot.id } : null),
+});

firestore/index.ts

@@ -1,2 +1,10 @@
 export { default as useCollection, CollectionHook } from './useCollection';
+export {
+  default as useCollectionData,
+  CollectionDataHook,
+} from './useCollectionData';
 export { default as useDocument, DocumentHook } from './useDocument';
+export {
+  default as useDocumentData,
+  DocumentDataHook,
+} from './useDocumentData';

firestore/useCollectionData.ts

@@ -0,0 +1,23 @@
+import { firestore } from 'firebase';
+import useCollection from './useCollection';
+import { snapshotToData } from './helpers';
+
+export type CollectionDataHook<T> = {
+  error?: object;
+  loading: boolean;
+  value?: T[];
+};
+
+export default <T>(
+  query?: firestore.Query | null,
+  idField?: string
+): CollectionDataHook<T> => {
+  const { error, loading, value } = useCollection(query);
+  return {
+    error,
+    loading,
+    value: (value
+      ? value.docs.map(doc => snapshotToData(doc, idField))
+      : undefined) as T[],
+  };
+};

firestore/useDocumentData.ts

@@ -0,0 +1,21 @@
+import { firestore } from 'firebase';
+import useDocument from './useDocument';
+import { snapshotToData } from './helpers';
+
+export type DocumentDataHook<T> = {
+  error?: object;
+  loading: boolean;
+  value?: T;
+};
+
+export default <T>(
+  docRef?: firestore.DocumentReference | null,
+  idField?: string
+): DocumentDataHook<T> => {
+  const { error, loading, value } = useDocument(docRef);
+  return {
+    error,
+    loading,
+    value: (value ? snapshotToData(value, idField) : undefined) as T,
+  };
+};