From 7794dd8d64787a63267da2aa763b7f268b0e38dc Mon Sep 17 00:00:00 2001
From: Pavel Prichodko <14926950+prichodko@users.noreply.github.com>
Date: Wed, 15 Jun 2022 16:11:04 +0200
Subject: [PATCH] feat: close reactions popover on click

---
 .../src/components/reaction-popover/index.tsx |  6 ++--
 .../components/chat-message/reactions.tsx     | 28 +++++++++++++------
 2 files changed, 23 insertions(+), 11 deletions(-)

diff --git a/packages/status-react/src/components/reaction-popover/index.tsx b/packages/status-react/src/components/reaction-popover/index.tsx
index 692ad8f5..dd782187 100644
--- a/packages/status-react/src/components/reaction-popover/index.tsx
+++ b/packages/status-react/src/components/reaction-popover/index.tsx
@@ -4,7 +4,7 @@ import { useAccount } from '~/src/protocol'
 import { styled } from '~/src/styles/config'
 import { Flex, Image, Popover, PopoverTrigger } from '~/src/system'
 
-import type { Reaction, Reactions } from '~/src/protocol/use-messages'
+import type { Reaction, Reactions } from '~/src/protocol'
 
 interface Props {
   children: React.ReactElement
@@ -52,13 +52,13 @@ export const ReactionPopover = (props: Props) => {
       <Popover side="top" align="center" sideOffset={6}>
         <Flex gap={1} css={{ padding: 8 }}>
           {Object.entries(emojis).map(([type, emoji]) => {
-            const value = reactions[type]
+            const value = reactions[type as Reaction]
             const me = account ? value.has('0x' + account.publicKey) : false
 
             return (
               <Button
                 key={type}
-                onClick={() => onClick(type)}
+                onClick={() => onClick(type as Reaction)}
                 active={me}
                 aria-label={`React with ${emoji.symbol}`}
               >
diff --git a/packages/status-react/src/routes/chat/components/chat-message/reactions.tsx b/packages/status-react/src/routes/chat/components/chat-message/reactions.tsx
index 0871e90b..62b01e44 100644
--- a/packages/status-react/src/routes/chat/components/chat-message/reactions.tsx
+++ b/packages/status-react/src/routes/chat/components/chat-message/reactions.tsx
@@ -1,4 +1,4 @@
-import React from 'react'
+import React, { useState } from 'react'
 
 import { emojis, ReactionPopover } from '~/src/components/reaction-popover'
 import { ReactionIcon } from '~/src/icons/reaction-icon'
@@ -6,7 +6,7 @@ import { useAccount } from '~/src/protocol'
 import { styled } from '~/src/styles/config'
 import { Flex, Image, Text } from '~/src/system'
 
-import type { Reactions } from '~/src/protocol/use-messages'
+import type { Reaction, Reactions } from '~/src/protocol'
 
 interface Props {
   reactions: Reactions
@@ -16,24 +16,36 @@ interface Props {
 export const MessageReactions = (props: Props) => {
   const { reactions, onClick } = props
 
+  const [open, setOpen] = useState(false)
+
   const hasReaction = Object.values(reactions).some(value => value.size > 0)
 
   if (hasReaction === false) {
     return null
   }
 
+  const handlePopoverClick = (reaction: Reaction) => {
+    onClick(reaction)
+    setOpen(false)
+  }
+
   return (
     <Flex align="center" css={{ paddingTop: 6 }} gap={1}>
       {Object.entries(emojis).map(([type, emoji]) => (
-        <Reaction
+        <ReactionButton
           key={type}
           emoji={emoji}
-          value={reactions[type]}
-          onClick={() => onClick(type)}
+          value={reactions[type as Reaction]}
+          onClick={() => onClick(type as Reaction)}
         />
       ))}
 
-      <ReactionPopover reactions={reactions} onClick={onClick}>
+      <ReactionPopover
+        open={open}
+        onOpenChange={setOpen}
+        reactions={reactions}
+        onClick={handlePopoverClick}
+      >
         <AddReactionButton aria-label="Add Reaction">
           <ReactionIcon width={16} height={16} />
         </AddReactionButton>
@@ -57,11 +69,11 @@ interface ReactionProps {
     url: string
     symbol: string
   }
-  value: Props['reactions']['HEART']
+  value: Reactions['LOVE']
   onClick: VoidFunction
 }
 
-const Reaction = (props: ReactionProps) => {
+const ReactionButton = (props: ReactionProps) => {
   const { emoji, value, onClick } = props
 
   const { account } = useAccount()