From bc3e43802b93be68980b55d5d1bddf7cf6bcba24 Mon Sep 17 00:00:00 2001
From: jahooma <jahooma@gmail.com>
Date: Sun, 23 Jan 2022 23:47:20 -0600
Subject: [PATCH] Listen for new folds on /folds

---
 web/hooks/use-fold.ts     | 12 ++++++++-
 web/lib/firebase/folds.ts |  6 ++++-
 web/pages/folds.tsx       | 54 +++++++++++++++++++++++++++++----------
 3 files changed, 57 insertions(+), 15 deletions(-)

diff --git a/web/hooks/use-fold.ts b/web/hooks/use-fold.ts
index 2484c213..f64694df 100644
--- a/web/hooks/use-fold.ts
+++ b/web/hooks/use-fold.ts
@@ -1,6 +1,6 @@
 import { useEffect, useState } from 'react'
 import { Fold } from '../../common/fold'
-import { listenForFold } from '../lib/firebase/folds'
+import { listenForFold, listenForFolds } from '../lib/firebase/folds'
 
 export const useFold = (foldId: string) => {
   const [fold, setFold] = useState<Fold | null | undefined>()
@@ -11,3 +11,13 @@ export const useFold = (foldId: string) => {
 
   return fold
 }
+
+export const useFolds = () => {
+  const [folds, setFolds] = useState<Fold[] | undefined>()
+
+  useEffect(() => {
+    return listenForFolds(setFolds)
+  }, [])
+
+  return folds
+}
diff --git a/web/lib/firebase/folds.ts b/web/lib/firebase/folds.ts
index a339b340..8d2b850a 100644
--- a/web/lib/firebase/folds.ts
+++ b/web/lib/firebase/folds.ts
@@ -2,7 +2,7 @@ import { collection, doc, query, updateDoc, where } from 'firebase/firestore'
 import { Fold } from '../../../common/fold'
 import { Contract, contractCollection } from './contracts'
 import { db } from './init'
-import { getValues, listenForValue } from './utils'
+import { getValues, listenForValue, listenForValues } from './utils'
 
 const foldCollection = collection(db, 'folds')
 
@@ -21,6 +21,10 @@ export async function listAllFolds() {
   return getValues<Fold>(foldCollection)
 }
 
+export function listenForFolds(setFolds: (folds: Fold[]) => void) {
+  return listenForValues(foldCollection, setFolds)
+}
+
 export async function getFoldBySlug(slug: string) {
   const q = query(foldCollection, where('slug', '==', slug))
   const folds = await getValues<Fold>(q)
diff --git a/web/pages/folds.tsx b/web/pages/folds.tsx
index dd013852..993b1b3d 100644
--- a/web/pages/folds.tsx
+++ b/web/pages/folds.tsx
@@ -1,7 +1,7 @@
 import clsx from 'clsx'
 import _ from 'lodash'
 import { useRouter } from 'next/router'
-import { useState } from 'react'
+import { useEffect, useState } from 'react'
 import { Fold } from '../../common/fold'
 import { parseWordsAsTags } from '../../common/util/parse'
 import { ConfirmationButton } from '../components/confirmation-button'
@@ -13,6 +13,7 @@ import { SiteLink } from '../components/site-link'
 import { TagsList } from '../components/tags-list'
 import { Title } from '../components/title'
 import { UserLink } from '../components/user-page'
+import { useFolds } from '../hooks/use-fold'
 import { useUser } from '../hooks/use-user'
 import { createFold } from '../lib/firebase/api-call'
 import { foldPath, listAllFolds } from '../lib/firebase/folds'
@@ -25,22 +26,44 @@ export async function getStaticProps() {
   const curators = await Promise.all(
     folds.map((fold) => getUser(fold.curatorId))
   )
+  const curatorsDict = _.fromPairs(
+    curators.map((curator) => [curator.id, curator])
+  )
 
   return {
     props: {
       folds,
-      curators,
+      curatorsDict,
     },
 
     revalidate: 60, // regenerate after a minute
   }
 }
 
-export default function Folds(props: { folds: Fold[]; curators: User[] }) {
-  const { folds, curators } = props
+export default function Folds(props: {
+  folds: Fold[]
+  curatorsDict: _.Dictionary<User>
+}) {
+  const [curatorsDict, setCuratorsDict] = useState(props.curatorsDict)
 
+  const folds = useFolds() ?? props.folds
   const user = useUser()
 
+  useEffect(() => {
+    // Load User object for curator of new Folds.
+    const newFolds = folds.filter(({ curatorId }) => !curatorsDict[curatorId])
+    if (newFolds.length > 0) {
+      Promise.all(newFolds.map(({ curatorId }) => getUser(curatorId))).then(
+        (newUsers) => {
+          const newUsersDict = _.fromPairs(
+            newUsers.map((user) => [user.id, user])
+          )
+          setCuratorsDict({ ...curatorsDict, ...newUsersDict })
+        }
+      )
+    }
+  })
+
   return (
     <Page>
       <Col className="items-center">
@@ -51,7 +74,7 @@ export default function Folds(props: { folds: Fold[]; curators: User[] }) {
           </Row>
 
           <Col className="gap-4">
-            {folds.map((fold, index) => (
+            {folds.map((fold) => (
               <Row key={fold.id} className="items-center gap-2">
                 <SiteLink href={foldPath(fold)}>{fold.name}</SiteLink>
                 <div />
@@ -61,8 +84,8 @@ export default function Folds(props: { folds: Fold[]; curators: User[] }) {
                   <div className="text-sm text-gray-500 mr-1">Curated by</div>
                   <UserLink
                     className="text-sm text-neutral"
-                    name={curators[index].name}
-                    username={curators[index].username}
+                    name={curatorsDict[fold.curatorId]?.name ?? ''}
+                    username={curatorsDict[fold.curatorId]?.username ?? ''}
                   />
                 </Row>
               </Row>
@@ -94,10 +117,15 @@ function CreateFoldButton() {
       tags: parseWordsAsTags(tags),
     }).then((r) => r.data || {})
 
-    if (result.fold) await router.push(foldPath(result.fold))
-    else console.log(result.status, result.message)
-
-    setIsSubmitting(false)
+    if (result.fold) {
+      await router.push(foldPath(result.fold)).catch((e) => {
+        console.log(e)
+        setIsSubmitting(false)
+      })
+    } else {
+      console.log(result.status, result.message)
+      setIsSubmitting(false)
+    }
   }
 
   return (
@@ -125,7 +153,7 @@ function CreateFoldButton() {
 
       <Spacer h={4} />
 
-      <form>
+      <div>
         <div className="form-control w-full">
           <label className="label">
             <span className="mb-1">Fold name</span>
@@ -167,7 +195,7 @@ function CreateFoldButton() {
             />
           </>
         )}
-      </form>
+      </div>
     </ConfirmationButton>
   )
 }