Update handling of reactions (#277)

* fix: reactions argument in callback

* replace object with Set in reactions

* update reactions in UI

packages/status-js/src/client.ts

@@ -104,7 +104,7 @@ class Client {
   }
 
   public handleWakuMessage = (wakuMessage: WakuMessage): void => {
-    handleWakuMessage(wakuMessage, this, this.community, this.account)
+    handleWakuMessage(wakuMessage, this, this.community)
   }
 }
 

packages/status-js/src/client/chat.ts

@@ -277,7 +277,7 @@ export class Chat {
   public handleEmojiReaction = (
     messageId: string,
     reaction: EmojiReaction,
-    isMe: boolean
+    publicKey: string
   ) => {
     let messageIndex = this.messages.length
     while (--messageIndex >= 0) {
@@ -295,7 +295,7 @@ export class Chat {
     this.messages[messageIndex].reactions = getReactions(
       reaction,
       this.messages[messageIndex].reactions,
-      isMe
+      publicKey
     )
 
     this.emitMessages(this.messages)
@@ -379,16 +379,30 @@ export class Chat {
   public sendReaction = async (
     chatId: string,
     messageId: string,
-    reaction: EmojiReaction.Type
+    reaction: keyof ChatMessage['reactions']
   ) => {
+    if (!this.client.account) {
+      throw new Error('Account not initialized')
+    }
+
+    const message = this.getMessage(messageId)
+
+    if (!message) {
+      throw new Error('Message not found')
+    }
+
+    const retracted = message.reactions[reaction].has(
+      this.client.account.publicKey
+    )
+
     const payload = EmojiReaction.encode({
       clock: BigInt(Date.now()),
       chatId: chatId,
-      messageType: 'COMMUNITY_CHAT',
+      messageType: 'COMMUNITY_CHAT' as MessageType,
-      grant: new Uint8Array([]),
       messageId,
-      retracted: false,
       type: reaction,
+      retracted,
+      grant: new Uint8Array([]),
     })
 
     await this.client.sendWakuMessage(

packages/status-js/src/client/community/get-reactions.ts

@@ -3,40 +3,25 @@ import type { EmojiReaction } from '../../../protos/emoji-reaction'
 type Reaction = Exclude<`${EmojiReaction.Type}`, 'UNKNOWN_EMOJI_REACTION_TYPE'>
 
 export type Reactions = {
-  [key in Reaction]: {
+  [key in Reaction]: Set<string>
-    count: number
-    me: boolean
-  }
 }
 
 export function getReactions(
   reaction: EmojiReaction,
   reactions: Reactions,
-  isMe: boolean
+  publicKey: string
-) {
+): Reactions {
   const { type, retracted } = reaction
 
   if (type === 'UNKNOWN_EMOJI_REACTION_TYPE') {
     return reactions
   }
 
-  const _reaction = {
+  if (retracted) {
-    count: reactions[type].count,
+    reactions[type].delete(publicKey)
-    me: reactions[type].me,
-  }
-
-  if (retracted && _reaction.count !== 0) {
-    _reaction.count -= 1
   } else {
-    _reaction.count += 1
+    reactions[type].add(publicKey)
   }
 
-  if (isMe) {
+  return reactions
-    _reaction.me = retracted ? false : true
-  }
-
-  return {
-    ...reactions,
-    [type]: _reaction,
-  }
 }

packages/status-js/src/client/community/handle-waku-message.ts

@@ -15,7 +15,6 @@ import { recoverPublicKey } from '../../utils/recover-public-key'
 import { getChatUuid } from './get-chat-uuid'
 import { mapChatMessage } from './map-chat-message'
 
-import type { Account } from '../../account'
 import type { Client } from '../../client'
 import type { Community } from './community'
 import type { WakuMessage } from 'js-waku'
@@ -24,8 +23,7 @@ export function handleWakuMessage(
   wakuMessage: WakuMessage,
   // state
   client: Client,
-  community: Community,
+  community: Community
-  account?: Account
 ): void {
   // decode (layers)
   if (!wakuMessage.payload) {
@@ -151,11 +149,14 @@ export function handleWakuMessage(
 
       const messageId = decodedPayload.messageId
       const chatUuid = getChatUuid(decodedPayload.chatId)
-      const isMe = account?.publicKey === `0x${bytesToHex(publicKey)}`
 
-      community.chats
+      const chat = community.chats.get(chatUuid)
-        .get(chatUuid)
+
-        ?.handleEmojiReaction(messageId, decodedPayload, isMe)
+      chat?.handleEmojiReaction(
+        messageId,
+        decodedPayload,
+        `0x${bytesToHex(publicKey)}`
+      )
 
       success = true
 

packages/status-js/src/client/community/map-chat-message.ts

@@ -6,40 +6,24 @@ export function mapChatMessage(
   props: {
     messageId: string
     chatUuid: string
+    publicKey: string
   }
 ): ChatMessage {
-  const { messageId, chatUuid } = props
+  const { messageId, chatUuid, publicKey } = props
 
   const message = {
     ...decodedMessage,
     messageId,
     chatUuid,
     pinned: false,
+    signer: publicKey,
     reactions: {
-      THUMBS_UP: {
+      THUMBS_UP: new Set<string>(),
-        count: 0,
+      THUMBS_DOWN: new Set<string>(),
-        me: false,
+      LOVE: new Set<string>(),
-      },
+      LAUGH: new Set<string>(),
-      THUMBS_DOWN: {
+      SAD: new Set<string>(),
-        count: 0,
+      ANGRY: new Set<string>(),
-        me: false,
-      },
-      LOVE: {
-        count: 0,
-        me: false,
-      },
-      LAUGH: {
-        count: 0,
-        me: false,
-      },
-      SAD: {
-        count: 0,
-        me: false,
-      },
-      ANGRY: {
-        count: 0,
-        me: false,
-      },
     },
   }
 

packages/status-react/src/components/reaction-popover/index.tsx

@@ -1,5 +1,6 @@
 import React from 'react'
 
+import { useAccount } from '~/src/protocol'
 import { styled } from '~/src/styles/config'
 import { Flex, Image, Popover, PopoverTrigger } from '~/src/system'
 
@@ -43,18 +44,22 @@ export const emojis: Record<Reaction, { url: string; symbol: string }> = {
 export const ReactionPopover = (props: Props) => {
   const { reactions, children, onClick, ...popoverProps } = props
 
+  const { account } = useAccount()
+
   return (
     <PopoverTrigger {...popoverProps}>
       {children}
       <Popover side="top" align="center" sideOffset={6}>
         <Flex gap={1} css={{ padding: 8 }}>
           {Object.entries(emojis).map(([type, emoji]) => {
-            const reaction = reactions[type]
+            const value = reactions[type]
+            const me = account ? value.has('0x' + account.publicKey) : false
+
             return (
               <Button
                 key={type}
-                onClick={() => onClick(reaction as Reaction)}
+                onClick={() => onClick(type)}
-                active={reaction.me}
+                active={me}
                 aria-label={`React with ${emoji.symbol}`}
               >
                 <Image width={30} src={emoji.url} alt={emoji.symbol} />

packages/status-react/src/routes/chat/components/chat-message/index.tsx

@@ -84,7 +84,9 @@ export const ChatMessage = (props: Props) => {
   }
 
   const handleReaction = (reaction: Reaction) => {
-    client.community.chats.get(chatId).sendReaction(chatId, messageId, reaction)
+    client.community
+      .getChatById(chatId)
+      .sendReaction(chatId, messageId, reaction)
   }
 
   const handleReplyClick = () => {

packages/status-react/src/routes/chat/components/chat-message/reactions.tsx

@@ -2,7 +2,7 @@ import React from 'react'
 
 import { emojis, ReactionPopover } from '~/src/components/reaction-popover'
 import { ReactionIcon } from '~/src/icons/reaction-icon'
-import { Reaction } from '~/src/protocol/use-messages'
+import { useAccount } from '~/src/protocol'
 import { styled } from '~/src/styles/config'
 import { Flex, Image, Text } from '~/src/system'
 
@@ -16,9 +16,7 @@ interface Props {
 export const MessageReactions = (props: Props) => {
   const { reactions, onClick } = props
 
-  const hasReaction = Object.values(reactions).some(
+  const hasReaction = Object.values(reactions).some(value => value.size > 0)
-    reaction => reaction.count !== 0
-  )
 
   if (hasReaction === false) {
     return null
@@ -26,12 +24,12 @@ export const MessageReactions = (props: Props) => {
 
   return (
     <Flex align="center" css={{ paddingTop: 6 }} gap={1}>
-      {Object.entries(reactions).map(([reaction, value]) => (
+      {Object.entries(emojis).map(([type, emoji]) => (
         <Reaction
-          key={reaction}
+          key={type}
-          emoji={emojis[reaction as Reaction]}
+          emoji={emoji}
-          reaction={value}
+          value={reactions[type]}
-          onClick={() => onClick(reaction as Reaction)}
+          onClick={() => onClick(type)}
         />
       ))}
 
@@ -59,25 +57,30 @@ interface ReactionProps {
     url: string
     symbol: string
   }
-  reaction: Props['reactions']['smile']
+  value: Props['reactions']['HEART']
   onClick: VoidFunction
 }
 
 const Reaction = (props: ReactionProps) => {
-  const { emoji, reaction, onClick } = props
+  const { emoji, value, onClick } = props
 
-  if (reaction.count === 0) {
+  const { account } = useAccount()
+
+  const count = value.size
+  const me = account ? value.has('0x' + account.publicKey) : false
+
+  if (value.size === 0) {
     return null
   }
 
   return (
     <Button
       onClick={onClick}
-      active={reaction.me}
+      active={me}
-      aria-label={`${emoji.symbol}, ${reaction.count} reaction, press to react`}
+      aria-label={`${emoji.symbol}, ${count} reaction, press to react`}
     >
       <Image width={14} src={emoji.url} alt={emoji.symbol} />
-      <Text size="12">{reaction.count}</Text>
+      <Text size="12">{count}</Text>
     </Button>
   )
 }