From 3240cd9abb1f1c1c43262d51e81fe9d04c2215af Mon Sep 17 00:00:00 2001
From: Herman Wikner <wiknerherman@gmail.com>
Date: Fri, 13 Oct 2023 11:50:30 +0200
Subject: [PATCH] refactor: update comment workshop stories

---
 .../src/__workshop__/CommentsListStory.tsx    | 206 +++++++++++-------
 .../__workshop__/CommentsProviderStory.tsx    |  10 +-
 2 files changed, 133 insertions(+), 83 deletions(-)

diff --git a/packages/sanity/src/desk/comments/src/__workshop__/CommentsListStory.tsx b/packages/sanity/src/desk/comments/src/__workshop__/CommentsListStory.tsx
index df571071e3d6..292f8b1407a7 100644
--- a/packages/sanity/src/desk/comments/src/__workshop__/CommentsListStory.tsx
+++ b/packages/sanity/src/desk/comments/src/__workshop__/CommentsListStory.tsx
@@ -1,15 +1,35 @@
-import React, {useCallback, useState} from 'react'
+import React, {useCallback, useMemo, useState} from 'react'
 import {useBoolean, useSelect} from '@sanity/ui-workshop'
+import {Schema} from '@sanity/schema'
+import {uuid} from '@sanity/uuid'
+import {Container, Flex} from '@sanity/ui'
 import {CommentsList} from '../components'
-import {
-  CommentDocument,
-  CommentCreatePayload,
-  CommentEditPayload,
-  CommentStatus,
-  CommentThreadItem,
-} from '../types'
+import {CommentDocument, CommentCreatePayload, CommentEditPayload, CommentStatus} from '../types'
+import {buildCommentThreadItems} from '../utils/buildCommentThreadItems'
+import {useMentionOptions} from '../hooks'
 import {useCurrentUser} from 'sanity'
 
+const noop = () => {
+  // noop
+}
+
+const schema = Schema.compile({
+  name: 'default',
+  types: [
+    {
+      type: 'document',
+      name: 'article',
+      fields: [
+        {
+          name: 'title',
+          type: 'string',
+          title: 'My string title',
+        },
+      ],
+    },
+  ],
+})
+
 const BASE: CommentDocument = {
   _id: '1',
   _type: 'comment',
@@ -24,7 +44,7 @@ const BASE: CommentDocument = {
   target: {
     documentType: 'article',
     path: {
-      field: JSON.stringify([]),
+      field: 'title',
     },
     document: {
       _dataset: '1',
@@ -52,19 +72,20 @@ const BASE: CommentDocument = {
   ],
 }
 
-const PROPS: CommentThreadItem = {
-  parentComment: BASE,
-  breadcrumbs: [],
-  commentsCount: 1,
-  fieldPath: 'test',
-  replies: [],
-  threadId: '1',
+const MENTION_HOOK_OPTIONS = {
+  documentValue: {
+    _type: 'author',
+    _id: 'grrm',
+    _createdAt: '2021-05-04T14:54:37Z',
+    _rev: '1',
+    _updatedAt: '2021-05-04T14:54:37Z',
+  },
 }
 
 const STATUS_OPTIONS: Record<CommentStatus, CommentStatus> = {open: 'open', resolved: 'resolved'}
 
 export default function CommentsListStory() {
-  const [state, setState] = useState<CommentThreadItem>(PROPS)
+  const [state, setState] = useState<CommentDocument[]>([BASE])
 
   const error = useBoolean('Error', false, 'Props') || null
   const loading = useBoolean('Loading', false, 'Props') || false
@@ -73,95 +94,122 @@ export default function CommentsListStory() {
 
   const currentUser = useCurrentUser()
 
+  const mentionOptions = useMentionOptions(MENTION_HOOK_OPTIONS)
+
   const handleReplySubmit = useCallback(
     (payload: CommentCreatePayload) => {
-      const comment: CommentDocument = {
+      const reply: CommentDocument = {
         ...BASE,
         ...payload,
         _createdAt: new Date().toISOString(),
-        _id: `${state.commentsCount + 1}`,
+        _id: uuid(),
         authorId: currentUser?.id || 'pP5s3g90N',
         parentCommentId: payload.parentCommentId,
       }
 
-      setState((prev) => {
-        return {
-          ...prev,
-          replies: [...prev.replies, comment],
-        }
-      })
+      setState((prev) => [reply, ...prev])
     },
-    [currentUser?.id, state.commentsCount],
+    [currentUser?.id],
   )
 
-  const handleEdit = useCallback(
-    (id: string, payload: CommentEditPayload) => {
-      const isParentEdit = id === state.parentComment._id
-
-      if (isParentEdit) {
-        setState((prev) => {
+  const handleEdit = useCallback((id: string, payload: CommentEditPayload) => {
+    setState((prev) => {
+      return prev.map((item) => {
+        if (item._id === id) {
           return {
-            ...prev,
-            parentComment: {
-              ...prev.parentComment,
-              ...payload,
-              _updatedAt: new Date().toISOString(),
-            },
+            ...item,
+            ...payload,
+            _updatedAt: new Date().toISOString(),
           }
-        })
-      }
-
-      setState((prev) => {
-        return {
-          ...prev,
-          replies: prev.replies.map((item) => {
-            if (item._id === id) {
-              return {
-                ...item,
-                ...payload,
-                _updatedAt: new Date().toISOString(),
-              }
-            }
-
-            return item
-          }),
         }
+
+        return item
       })
-    },
-    [state.parentComment._id],
-  )
+    })
+  }, [])
 
   const handleDelete = useCallback(
     (id: string) => {
+      setState((prev) => prev.filter((item) => item._id !== id))
+    },
+    [setState],
+  )
+
+  const handleNewThreadCreate = useCallback(
+    (payload: CommentCreatePayload) => {
+      const comment: CommentDocument = {
+        ...BASE,
+        ...payload,
+        _createdAt: new Date().toISOString(),
+        _id: uuid(),
+        authorId: currentUser?.id || 'pP5s3g90N',
+      }
+
+      setState((prev) => [comment, ...prev])
+    },
+    [currentUser?.id],
+  )
+
+  const handleStatusChange = useCallback(
+    (id: string, newStatus: CommentStatus) => {
       setState((prev) => {
-        return {
-          ...prev,
-          replies: prev.replies.filter((item) => item._id !== id),
-        }
+        return prev.map((item) => {
+          if (item._id === id) {
+            return {
+              ...item,
+              status: newStatus,
+              _updatedAt: new Date().toISOString(),
+            }
+          }
+
+          if (item.parentCommentId === id) {
+            return {
+              ...item,
+              status: newStatus,
+              _updatedAt: new Date().toISOString(),
+            }
+          }
+
+          return item
+        })
       })
     },
     [setState],
   )
 
+  const threadItems = useMemo(() => {
+    if (!currentUser || emptyState) return []
+
+    const items = buildCommentThreadItems({
+      comments: state.filter((item) => item.status === status),
+      currentUser,
+      documentValue: {},
+      schemaType: schema.get('article'),
+    })
+
+    return items
+  }, [currentUser, emptyState, state, status])
+
   if (!currentUser) return null
 
   return (
-    <CommentsList
-      comments={emptyState ? [] : [state]}
-      currentUser={currentUser}
-      error={error ? new Error('Something went wrong') : null}
-      loading={loading}
-      mentionOptions={{data: [], error: null, loading: false}}
-      onDelete={handleDelete}
-      onEdit={handleEdit}
-      onReply={handleReplySubmit}
-      status={status}
-      onCreateRetry={() => {
-        // ...
-      }}
-      onNewThreadCreate={() => {
-        // ...
-      }}
-    />
+    <Flex height="fill" overflow="hidden" padding={3}>
+      <Container width={1}>
+        <CommentsList
+          comments={threadItems}
+          currentUser={currentUser}
+          error={error ? new Error('Something went wrong') : null}
+          loading={loading}
+          mentionOptions={mentionOptions}
+          onCreateRetry={noop}
+          onDelete={handleDelete}
+          onEdit={handleEdit}
+          onNewThreadCreate={handleNewThreadCreate}
+          onReply={handleReplySubmit}
+          onStatusChange={handleStatusChange}
+          status={status}
+        />
+      </Container>
+    </Flex>
   )
 }
diff --git a/packages/sanity/src/desk/comments/src/__workshop__/CommentsProviderStory.tsx b/packages/sanity/src/desk/comments/src/__workshop__/CommentsProviderStory.tsx
index 403e5179e6d1..29d8047302df 100644
--- a/packages/sanity/src/desk/comments/src/__workshop__/CommentsProviderStory.tsx
+++ b/packages/sanity/src/desk/comments/src/__workshop__/CommentsProviderStory.tsx
@@ -2,7 +2,7 @@
 import React from 'react'
 import {useString} from '@sanity/ui-workshop'
 import {CommentsList} from '../components'
-import {CommentsProvider} from '../context'
+import {CommentsProvider, CommentsSetupProvider} from '../context'
 import {useComments} from '../hooks'
 import {useCurrentUser} from 'sanity'
 
@@ -11,9 +11,11 @@ export default function CommentsProviderStory() {
   const _id = useString('_id', 'grrm') || 'grrm'
 
   return (
-    <CommentsProvider documentType={_type} documentId={_id}>
-      <Inner />
-    </CommentsProvider>
+    <CommentsSetupProvider>
+      <CommentsProvider documentType={_type} documentId={_id}>
+        <Inner />
+      </CommentsProvider>
+    </CommentsSetupProvider>
   )
 }