Clean up a bunch of duplicated work in the comments list stuff

web/components/contract/contract-tabs.tsx

@@ -54,6 +54,13 @@ export function ContractTabs(props: {
     />
   )
 
+  const generalBets = outcomeType === 'FREE_RESPONSE' ? [] : visibleBets
+  const generalComments = comments.filter(
+    (comment) =>
+      comment.answerOutcome === undefined &&
+      (outcomeType === 'FREE_RESPONSE' ? comment.betId === undefined : true)
+  )
+
   const commentActivity =
     outcomeType === 'FREE_RESPONSE' ? (
       <>
@@ -69,8 +76,8 @@ export function ContractTabs(props: {
           <div className={'mb-4 w-full border-b border-gray-200'} />
           <ContractCommentsActivity
             contract={contract}
-            bets={visibleBets}
-            comments={comments}
+            bets={generalBets}
+            comments={generalComments}
             tips={tips}
             user={user}
           />

web/components/feed/contract-activity.tsx

@@ -10,7 +10,7 @@ import { FeedCommentThread, CommentInput } from './feed-comments'
 import { User } from 'common/user'
 import { CommentTipMap } from 'web/hooks/use-tip-txns'
 import { LiquidityProvision } from 'common/liquidity-provision'
-import { sortBy, uniq } from 'lodash'
+import { groupBy, sortBy, uniq } from 'lodash'
 import { Col } from 'web/components/layout/col'
 
 export function ContractBetsActivity(props: {
@@ -62,47 +62,35 @@ export function ContractCommentsActivity(props: {
   user: User | null | undefined
 }) {
   const { bets, contract, comments, user, tips } = props
-
-  const nonFreeResponseComments = comments.filter(
-    (comment) =>
-      comment.answerOutcome === undefined &&
-      (contract.outcomeType === 'FREE_RESPONSE'
-        ? comment.betId === undefined
-        : true)
-  )
-  const nonFreeResponseBets =
-    contract.outcomeType === 'FREE_RESPONSE' ? [] : bets
-
-  const betsByCurrentUser = nonFreeResponseBets.filter(
-    (bet) => bet.userId === user?.id
-  )
-  const commentsByCurrentUser = nonFreeResponseComments.filter(
-    (comment) => comment.userId === user?.id
-  )
-
-  const parentComments = comments.filter((comment) => !comment.replyToCommentId)
+  const betsByUserId = groupBy(bets, (bet) => bet.userId)
+  const commentsByUserId = groupBy(comments, (c) => c.userId)
+  const commentsByParentId = groupBy(comments, (c) => c.replyToCommentId ?? '_')
+  const topLevelComments = commentsByParentId['_'] ?? []
 
   return (
     <div>
       <CommentInput
         contract={contract}
-        betsByCurrentUser={betsByCurrentUser}
-        commentsByCurrentUser={commentsByCurrentUser}
+        betsByCurrentUser={(user && betsByUserId[user.id]) ?? []}
+        commentsByCurrentUser={(user && commentsByUserId[user.id]) ?? []}
       />
-      {parentComments.map((parent, idx) => (
+      {topLevelComments.map((parent, idx) => (
         <div key={parent.id} className={'relative pb-4'}>
-          {idx !== parentComments.length - 1 ? (
+          {idx !== topLevelComments.length - 1 ? (
             <span
               className="absolute top-5 left-5 -ml-px h-[calc(100%-2rem)] w-0.5 bg-gray-200"
               aria-hidden="true"
             />
           ) : null}
           <FeedCommentThread
+            user={user}
             contract={contract}
             parentComment={parent}
-            comments={comments}
+            threadComments={commentsByParentId[parent.id] ?? []}
             tips={tips}
             bets={bets}
+            betsByUserId={betsByUserId}
+            commentsByUserId={commentsByUserId}
           />
         </div>
       ))}
@@ -130,6 +118,10 @@ export function FreeResponseContractCommentsActivity(props: {
     })
     .filter((answer) => answer != null)
 
+  const betsByUserId = groupBy(bets, (bet) => bet.userId)
+  const commentsByUserId = groupBy(comments, (c) => c.userId)
+  const commentsByOutcome = groupBy(comments, (c) => c.answerOutcome ?? '_')
+
   return (
     <div>
       {answers.map((answer) => (
@@ -142,9 +134,11 @@ export function FreeResponseContractCommentsActivity(props: {
             contract={contract}
             user={user}
             answer={answer}
-            comments={comments}
+            answerComments={commentsByOutcome[answer.number.toString()]}
             tips={tips}
             bets={bets}
+            betsByUserId={betsByUserId}
+            commentsByUserId={commentsByUserId}
           />
         </div>
       ))}

web/components/feed/feed-answer-comment-group.tsx

@@ -1,5 +1,6 @@
 import { Answer } from 'common/answer'
 import { Bet } from 'common/bet'
+import { FreeResponseContract } from 'common/contract'
 import { ContractComment } from 'common/comment'
 import React, { useEffect, useState } from 'react'
 import { Col } from 'web/components/layout/col'
@@ -15,20 +16,31 @@ import {
 } from 'web/components/feed/feed-comments'
 import { CopyLinkDateTimeComponent } from 'web/components/feed/copy-link-date-time'
 import { useRouter } from 'next/router'
-import { groupBy } from 'lodash'
+import { Dictionary } from 'lodash'
 import { User } from 'common/user'
 import { useEvent } from 'web/hooks/use-event'
 import { CommentTipMap } from 'web/hooks/use-tip-txns'
 
 export function FeedAnswerCommentGroup(props: {
-  contract: any
+  contract: FreeResponseContract
   user: User | undefined | null
   answer: Answer
-  comments: ContractComment[]
+  answerComments: ContractComment[]
   tips: CommentTipMap
   bets: Bet[]
+  betsByUserId: Dictionary<Bet[]>
+  commentsByUserId: Dictionary<ContractComment[]>
 }) {
-  const { answer, contract, comments, tips, bets, user } = props
+  const {
+    answer,
+    contract,
+    answerComments,
+    tips,
+    bets,
+    betsByUserId,
+    commentsByUserId,
+    user,
+  } = props
   const { username, avatarUrl, name, text } = answer
 
   const [replyToUser, setReplyToUser] =
@@ -38,11 +50,6 @@ export function FeedAnswerCommentGroup(props: {
   const router = useRouter()
 
   const answerElementId = `answer-${answer.id}`
-  const betsByUserId = groupBy(bets, (bet) => bet.userId)
-  const commentsByUserId = groupBy(comments, (comment) => comment.userId)
-  const commentsList = comments.filter(
-    (comment) => comment.answerOutcome === answer.number.toString()
-  )
   const betsByCurrentUser = (user && betsByUserId[user.id]) ?? []
   const commentsByCurrentUser = (user && commentsByUserId[user.id]) ?? []
   const isFreeResponseContractPage = !!commentsByCurrentUser
@@ -155,7 +162,7 @@ export function FeedAnswerCommentGroup(props: {
       </Row>
       <CommentRepliesList
         contract={contract}
-        commentsList={commentsList}
+        comments={answerComments}
         betsByUserId={betsByUserId}
         smallAvatar={true}
         bets={bets}

web/components/feed/feed-comments.tsx

@@ -3,7 +3,7 @@ import { ContractComment } from 'common/comment'
 import { User } from 'common/user'
 import { Contract } from 'common/contract'
 import React, { useEffect, useState } from 'react'
-import { minBy, maxBy, groupBy, partition, sumBy, Dictionary } from 'lodash'
+import { minBy, maxBy, partition, sumBy, Dictionary } from 'lodash'
 import { useUser } from 'web/hooks/use-user'
 import { formatMoney } from 'common/util/format'
 import { useRouter } from 'next/router'
@@ -31,28 +31,30 @@ import { Content, TextEditor, useTextEditor } from '../editor'
 import { Editor } from '@tiptap/react'
 
 export function FeedCommentThread(props: {
+  user: User | null | undefined
   contract: Contract
-  comments: ContractComment[]
+  threadComments: ContractComment[]
   tips: CommentTipMap
   parentComment: ContractComment
   bets: Bet[]
+  betsByUserId: Dictionary<Bet[]>
+  commentsByUserId: Dictionary<ContractComment[]>
 }) {
-  const { contract, comments, bets, tips, parentComment } = props
+  const {
+    user,
+    contract,
+    threadComments,
+    commentsByUserId,
+    bets,
+    betsByUserId,
+    tips,
+    parentComment,
+  } = props
   const [showReply, setShowReply] = useState(false)
-  const [replyToUser, setReplyToUser] = useState<{
-    id: string
-    username: string
-  }>()
-  const betsByUserId = groupBy(bets, (bet) => bet.userId)
-  const user = useUser()
-  const commentsList = comments.filter(
-    (comment) =>
-      parentComment.id && comment.replyToCommentId === parentComment.id
-  )
-  commentsList.unshift(parentComment)
+  const [replyTo, setReplyTo] = useState<{ id: string; username: string }>()
 
   function scrollAndOpenReplyInput(comment: ContractComment) {
-    setReplyToUser({ id: comment.userId, username: comment.userUsername })
+    setReplyTo({ id: comment.userId, username: comment.userUsername })
     setShowReply(true)
   }
 
@@ -64,7 +66,7 @@ export function FeedCommentThread(props: {
       />
       <CommentRepliesList
         contract={contract}
-        commentsList={commentsList}
+        comments={[parentComment].concat(threadComments)}
         betsByUserId={betsByUserId}
         tips={tips}
         bets={bets}
@@ -79,12 +81,10 @@ export function FeedCommentThread(props: {
           <CommentInput
             contract={contract}
             betsByCurrentUser={(user && betsByUserId[user.id]) ?? []}
-            commentsByCurrentUser={comments.filter(
-              (c) => c.userId === user?.id
-            )}
+            commentsByCurrentUser={(user && commentsByUserId[user.id]) ?? []}
             parentCommentId={parentComment.id}
-            replyToUser={replyToUser}
-            parentAnswerOutcome={comments[0].answerOutcome}
+            replyToUser={replyTo}
+            parentAnswerOutcome={parentComment.answerOutcome}
             onSubmitComment={() => setShowReply(false)}
           />
         </Col>
@@ -95,7 +95,7 @@ export function FeedCommentThread(props: {
 
 export function CommentRepliesList(props: {
   contract: Contract
-  commentsList: ContractComment[]
+  comments: ContractComment[]
   betsByUserId: Dictionary<Bet[]>
   tips: CommentTipMap
   scrollAndOpenReplyInput: (comment: ContractComment) => void
@@ -105,7 +105,7 @@ export function CommentRepliesList(props: {
 }) {
   const {
     contract,
-    commentsList,
+    comments,
     betsByUserId,
     tips,
     smallAvatar,
@@ -115,7 +115,7 @@ export function CommentRepliesList(props: {
   } = props
   return (
     <>
-      {commentsList.map((comment, commentIdx) => (
+      {comments.map((comment, commentIdx) => (
         <div
           key={comment.id}
           id={comment.id}