Fetch the like status along with other update details

pbteja1998 · pbteja1998 · commit 2691fead4add · 2021-02-28T10:44:49.000+05:30
diff --git a/src/components/HomePageFeed.tsx b/src/components/HomePageFeed.tsx
@@ -94,13 +94,7 @@ export function HomePageFeedUpdate({
 
   const { count: likesCount, hasLiked, toggleLike } = useLikes({
     initialCount: update.likes.data,
-    query: {
-      key: ['api/fauna/has-liked-update', update.id],
-      endpoint: '/api/fauna/has-liked-update',
-      body: {
-        updateId: update.id,
-      },
-    },
+    initialHasLiked: update.hasLiked,
     mutation: {
       endpoint: '/api/fauna/toggle-update-like',
       body: {
@@ -268,9 +262,6 @@ export function HomePageFeedUpdate({
                         updateId={update.id}
                         key={comment.id}
                         comment={comment}
-                        isLastComment={
-                          index === update.comments.data.length - 1
-                        }
                       >
                         {comment.description}
                       </UpdateComment>
diff --git a/src/components/goals/UpdateComment.tsx b/src/components/goals/UpdateComment.tsx
@@ -25,12 +25,10 @@ export default function UpdateComment({
   updateId,
   comment,
   children,
-  isLastComment = false,
 }: {
   updateId: string
   children: string
   comment: UpdateCommentType
-  isLastComment?: boolean
 }) {
   const queryClient = useQueryClient()
   const [isInEditMode, setIsInEditMode] = useState(false)
@@ -40,13 +38,7 @@ export default function UpdateComment({
   const [isLikeModalOpen, setIsLikeModalOpen] = useState(false)
   const { count: likesCount, hasLiked, toggleLike } = useLikes({
     initialCount: comment.likes.data,
-    query: {
-      key: ['api/fauna/has-liked-comment', comment.id],
-      endpoint: '/api/fauna/has-liked-comment',
-      body: {
-        commentId: comment.id,
-      },
-    },
+    initialHasLiked: comment.hasLiked,
     mutation: {
       endpoint: '/api/fauna/toggle-comment-like',
       body: {
diff --git a/src/components/useLikes.ts b/src/components/useLikes.ts
@@ -1,5 +1,5 @@
-import { useEffect, useReducer } from 'react'
-import { useMutation, useQuery } from 'react-query'
+import { useReducer } from 'react'
+import { useMutation } from 'react-query'
 
 type LikeData = {
   count: number
@@ -24,30 +24,20 @@ function reducer(state: LikeData, action: { type: string; payload?: any }) {
 
 export function useLikes({
   initialCount,
-  query,
+  initialHasLiked,
   mutation,
 }: {
   initialCount: number
-  query: {
-    key: string | string[]
-    endpoint: string
-    body: any
-  }
+  initialHasLiked: boolean
   mutation: {
     endpoint: string
     body: any
   }
 }) {
-  const initialState: LikeData = { count: initialCount, hasLiked: false }
-  const { isLoading, isError, data } = useQuery(query.key, () => {
-    return fetch(query.endpoint, {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify(query.body),
-    }).then((res) => res.json())
-  })
+  const initialState: LikeData = {
+    count: initialCount,
+    hasLiked: initialHasLiked,
+  }
 
   const { mutate } = useMutation(() => {
     return fetch(mutation.endpoint, {
@@ -66,15 +56,6 @@ export function useLikes({
 
   const [{ count, hasLiked }, dispatch] = useReducer(reducer, initialState)
 
-  useEffect(() => {
-    if (!isLoading && !isError) {
-      dispatch({
-        type: 'set',
-        payload: { hasLiked: data.liked, count: initialCount },
-      })
-    }
-  }, [data?.liked, initialCount, isError, isLoading])
-
   const toggleLike = () => {
     dispatch({ type: 'toggle' })
     mutate()
diff --git a/src/pages/api/fauna/all-updates.ts b/src/pages/api/fauna/all-updates.ts
@@ -1,6 +1,8 @@
 import { NextApiHandler, NextApiRequest, NextApiResponse } from 'next'
 
 import faunadb from 'faunadb'
+import { getSession } from 'next-auth/client'
+import { User } from 'src/pages/members'
 const q = faunadb.query
 const isProduction = process.env.NODE_ENV === 'production'
 const client = new faunadb.Client({
@@ -14,6 +16,7 @@ const FaunaCreateHandler: NextApiHandler = async (
   req: NextApiRequest,
   res: NextApiResponse
 ) => {
+  const session = await getSession({ req })
   try {
     const response: any = await client.query(
       q.Map(q.Paginate(q.Match(q.Index('all_recent_updates'))), (result) => {
@@ -26,8 +29,22 @@ const FaunaCreateHandler: NextApiHandler = async (
         const createdAt = q.ToMillis(
           q.Select(['data', 'timestamps', 'createdAt'], goalUpdateDoc)
         )
+        const updateId = q.Select(['ref', 'id'], goalUpdateDoc)
+        let hasLiked = false
+        if (session) {
+          const userId = (session.user as User).id
+          const ref = q.Match(q.Index('unique_update_user_like'), [
+            q.Ref(q.Collection('goal_updates'), updateId),
+            q.Ref(q.Collection('users'), userId),
+          ])
+          hasLiked = q.If(
+            q.Exists(ref),
+            q.Select(['data', 'liked'], q.Get(ref)),
+            false
+          ) as boolean
+        }
         return {
-          id: q.Select(['ref', 'id'], goalUpdateDoc),
+          id: updateId,
           goal: {
             id: q.Select(['ref', 'id'], goalDoc),
             title: q.Select(['data', 'title'], goalDoc),
@@ -41,12 +58,27 @@ const FaunaCreateHandler: NextApiHandler = async (
               const postedByDoc = q.Get(
                 q.Select(['data', 'postedBy'], commentDoc)
               )
+              let hasLiked = false
+              const commentId = q.Select(['ref', 'id'], commentDoc)
+              if (session) {
+                const userId = (session.user as User).id
+                const ref = q.Match(q.Index('unique_comment_user_like'), [
+                  q.Ref(q.Collection('update_comments'), commentId),
+                  q.Ref(q.Collection('users'), userId),
+                ])
+                hasLiked = q.If(
+                  q.Exists(ref),
+                  q.Select(['data', 'liked'], q.Get(ref)),
+                  false
+                ) as boolean
+              }
               return {
-                id: q.Select(['ref', 'id'], commentDoc),
+                id: commentId,
                 description: q.Select(['data', 'description'], commentDoc),
                 createdAt: q.ToMillis(
                   q.Select(['data', 'timestamps', 'createdAt'], commentDoc)
                 ),
+                hasLiked,
                 likes: q.Count(
                   q.Filter(
                     q.Paginate(
@@ -73,6 +105,7 @@ const FaunaCreateHandler: NextApiHandler = async (
               }
             }
           ),
+          hasLiked,
           likes: q.Count(
             q.Filter(
               q.Paginate(
diff --git a/src/pages/index.tsx b/src/pages/index.tsx
@@ -8,6 +8,7 @@ export type UpdateCommentType = {
   description: string
   createdAt: number
   postedBy: User
+  hasLiked: boolean
   likes: {
     data: number
   }
@@ -19,6 +20,7 @@ export type HomePageFeedUpdateType = {
   description: string
   createdAt: number
   postedBy: User
+  hasLiked: boolean
   likes: {
     data: number
   }
