From 143555bbd8ddca1812d77c5a870ec33c6f65a625 Mon Sep 17 00:00:00 2001
From: martincupela <martin.cupela@gmail.com>
Date: Thu, 5 Sep 2024 12:38:38 +0200
Subject: [PATCH] feat: use own anchor root for DialogAnchor

---
 src/components/Dialog/DialogAnchor.tsx        |   8 +-
 src/components/Dialog/DialogPortal.tsx        |  27 ++-
 .../MessageActions/MessageActions.tsx         |   5 +-
 .../MessageActions/MessageActionsBox.tsx      | 163 ++++++++++--------
 4 files changed, 109 insertions(+), 94 deletions(-)

diff --git a/src/components/Dialog/DialogAnchor.tsx b/src/components/Dialog/DialogAnchor.tsx
index 31e62dcec..20c7434f0 100644
--- a/src/components/Dialog/DialogAnchor.tsx
+++ b/src/components/Dialog/DialogAnchor.tsx
@@ -1,3 +1,4 @@
+import clsx from 'clsx';
 import { Placement } from '@popperjs/core';
 import React, { ComponentProps, PropsWithChildren, useEffect, useRef } from 'react';
 import { usePopper } from 'react-popper';
@@ -48,7 +49,7 @@ export function useDialogAnchor<T extends HTMLElement>({
 
 type DialogAnchorProps = PropsWithChildren<Partial<DialogAnchorOptions>> & {
   id: string;
-} & ComponentProps<'div' | 'span'>;
+} & ComponentProps<'div'>;
 
 export const DialogAnchor = ({
   children,
@@ -56,6 +57,7 @@ export const DialogAnchor = ({
   id,
   placement = 'auto',
   referenceElement = null,
+  ...restDivProps
 }: DialogAnchorProps) => {
   const open = useDialogIsOpen(id);
   const { attributes, popperElementRef, styles } = useDialogAnchor<HTMLDivElement>({
@@ -67,8 +69,10 @@ export const DialogAnchor = ({
   return (
     <DialogPortalEntry dialogId={id}>
       <div
+        {...restDivProps}
         {...attributes.popper}
-        className={className}
+        className={clsx('str-chat__dialog-contents', className)}
+        data-testid='str-chat__dialog-contents'
         ref={popperElementRef}
         style={styles.popper}
       >
diff --git a/src/components/Dialog/DialogPortal.tsx b/src/components/Dialog/DialogPortal.tsx
index 9110b42e5..e31a3582f 100644
--- a/src/components/Dialog/DialogPortal.tsx
+++ b/src/components/Dialog/DialogPortal.tsx
@@ -18,22 +18,17 @@ export const DialogPortalDestination = () => {
   );
 
   return (
-    <>
-      <div
-        className='str-chat__dialog-overlay'
-        onClick={() => dialogsManager.closeAll()}
-        style={{
-          height: '100%',
-          inset: '0',
-          overflow: 'hidden',
-          position: 'absolute',
-          width: '100%',
-          zIndex: shouldRender ? '2' : '-1',
-        }}
-      >
-        <div data-str-chat__portal-id={dialogsManager.id} />
-      </div>
-    </>
+    <div
+      className='str-chat__dialog-overlay'
+      data-str-chat__portal-id={dialogsManager.id}
+      data-testid='str-chat__dialog-overlay'
+      onClick={() => dialogsManager.closeAll()}
+      style={
+        {
+          '--str-chat__dialog-overlay-height': shouldRender ? '100%' : '0',
+        } as React.CSSProperties
+      }
+    ></div>
   );
 };
 
diff --git a/src/components/MessageActions/MessageActions.tsx b/src/components/MessageActions/MessageActions.tsx
index 9860f9b7d..41dd364e7 100644
--- a/src/components/MessageActions/MessageActions.tsx
+++ b/src/components/MessageActions/MessageActions.tsx
@@ -146,9 +146,6 @@ export const MessageActions = <
       toggleOpen={dialog?.toggleSingle}
     >
       <DialogAnchor
-        className={clsx('str-chat__message-actions-box', {
-          'str-chat__message-actions-box--open': dialogIsOpen,
-        })}
         id={dialogId}
         placement={isMine ? 'top-end' : 'top-start'}
         referenceElement={actionsBoxButtonRef.current}
@@ -162,6 +159,8 @@ export const MessageActions = <
           handleMute={handleMute}
           handlePin={handlePin}
           isUserMuted={isMuted}
+          mine={isMine}
+          open={dialogIsOpen}
         />
       </DialogAnchor>
       <button
diff --git a/src/components/MessageActions/MessageActionsBox.tsx b/src/components/MessageActions/MessageActionsBox.tsx
index b7e618317..fdbee888d 100644
--- a/src/components/MessageActions/MessageActionsBox.tsx
+++ b/src/components/MessageActions/MessageActionsBox.tsx
@@ -1,4 +1,5 @@
-import React from 'react';
+import clsx from 'clsx';
+import React, { ComponentProps } from 'react';
 
 import { MESSAGE_ACTIONS } from '../Message/utils';
 
@@ -27,7 +28,9 @@ export type MessageActionsBoxProps<
   StreamChatGenerics extends DefaultStreamChatGenerics = DefaultStreamChatGenerics
 > = Pick<MessageContextValue<StreamChatGenerics>, PropsDrilledToMessageActionsBox> & {
   isUserMuted: () => boolean;
-};
+  mine: boolean;
+  open: boolean;
+} & ComponentProps<'div'>;
 
 const UnMemoizedMessageActionsBox = <
   StreamChatGenerics extends DefaultStreamChatGenerics = DefaultStreamChatGenerics
@@ -35,6 +38,7 @@ const UnMemoizedMessageActionsBox = <
   props: MessageActionsBoxProps<StreamChatGenerics>,
 ) => {
   const {
+    className,
     getMessageActions,
     handleDelete,
     handleEdit,
@@ -43,6 +47,8 @@ const UnMemoizedMessageActionsBox = <
     handleMute,
     handlePin,
     isUserMuted,
+    open,
+    ...restDivProps
   } = props;
 
   const {
@@ -70,81 +76,92 @@ const UnMemoizedMessageActionsBox = <
     }
   };
 
+  const rootClassName = clsx('str-chat__message-actions-box', className, {
+    'str-chat__message-actions-box--open': open,
+  });
+
   const buttonClassName =
     'str-chat__message-actions-list-item str-chat__message-actions-list-item-button';
 
   return (
-    <div
-      aria-label={t('aria/Message Options')}
-      className='str-chat__message-actions-list'
-      role='listbox'
-    >
-      <CustomMessageActionsList customMessageActions={customMessageActions} message={message} />
-      {messageActions.indexOf(MESSAGE_ACTIONS.quote) > -1 && (
-        <button
-          aria-selected='false'
-          className={buttonClassName}
-          onClick={handleQuote}
-          role='option'
-        >
-          {t<string>('Reply')}
-        </button>
-      )}
-      {messageActions.indexOf(MESSAGE_ACTIONS.pin) > -1 && !message.parent_id && (
-        <button aria-selected='false' className={buttonClassName} onClick={handlePin} role='option'>
-          {!message.pinned ? t<string>('Pin') : t<string>('Unpin')}
-        </button>
-      )}
-      {messageActions.indexOf(MESSAGE_ACTIONS.markUnread) > -1 && !threadList && !!message.id && (
-        <button
-          aria-selected='false'
-          className={buttonClassName}
-          onClick={handleMarkUnread}
-          role='option'
-        >
-          {t<string>('Mark as unread')}
-        </button>
-      )}
-      {messageActions.indexOf(MESSAGE_ACTIONS.flag) > -1 && (
-        <button
-          aria-selected='false'
-          className={buttonClassName}
-          onClick={handleFlag}
-          role='option'
-        >
-          {t<string>('Flag')}
-        </button>
-      )}
-      {messageActions.indexOf(MESSAGE_ACTIONS.mute) > -1 && (
-        <button
-          aria-selected='false'
-          className={buttonClassName}
-          onClick={handleMute}
-          role='option'
-        >
-          {isUserMuted() ? t<string>('Unmute') : t<string>('Mute')}
-        </button>
-      )}
-      {messageActions.indexOf(MESSAGE_ACTIONS.edit) > -1 && (
-        <button
-          aria-selected='false'
-          className={buttonClassName}
-          onClick={handleEdit}
-          role='option'
-        >
-          {t<string>('Edit Message')}
-        </button>
-      )}
-      {messageActions.indexOf(MESSAGE_ACTIONS.delete) > -1 && (
-        <button
-          aria-selected='false'
-          className={buttonClassName}
-          onClick={handleDelete}
-          role='option'
-        >
-          {t<string>('Delete')}
-        </button>
-      )}
+    <div {...restDivProps} className={rootClassName} data-testid='message-actions-box'>
+      <div
+        aria-label={t('aria/Message Options')}
+        className='str-chat__message-actions-list'
+        role='listbox'
+      >
+        <CustomMessageActionsList customMessageActions={customMessageActions} message={message} />
+        {messageActions.indexOf(MESSAGE_ACTIONS.quote) > -1 && (
+          <button
+            aria-selected='false'
+            className={buttonClassName}
+            onClick={handleQuote}
+            role='option'
+          >
+            {t<string>('Reply')}
+          </button>
+        )}
+        {messageActions.indexOf(MESSAGE_ACTIONS.pin) > -1 && !message.parent_id && (
+          <button
+            aria-selected='false'
+            className={buttonClassName}
+            onClick={handlePin}
+            role='option'
+          >
+            {!message.pinned ? t<string>('Pin') : t<string>('Unpin')}
+          </button>
+        )}
+        {messageActions.indexOf(MESSAGE_ACTIONS.markUnread) > -1 && !threadList && !!message.id && (
+          <button
+            aria-selected='false'
+            className={buttonClassName}
+            onClick={handleMarkUnread}
+            role='option'
+          >
+            {t<string>('Mark as unread')}
+          </button>
+        )}
+        {messageActions.indexOf(MESSAGE_ACTIONS.flag) > -1 && (
+          <button
+            aria-selected='false'
+            className={buttonClassName}
+            onClick={handleFlag}
+            role='option'
+          >
+            {t<string>('Flag')}
+          </button>
+        )}
+        {messageActions.indexOf(MESSAGE_ACTIONS.mute) > -1 && (
+          <button
+            aria-selected='false'
+            className={buttonClassName}
+            onClick={handleMute}
+            role='option'
+          >
+            {isUserMuted() ? t<string>('Unmute') : t<string>('Mute')}
+          </button>
+        )}
+        {messageActions.indexOf(MESSAGE_ACTIONS.edit) > -1 && (
+          <button
+            aria-selected='false'
+            className={buttonClassName}
+            onClick={handleEdit}
+            role='option'
+          >
+            {t<string>('Edit Message')}
+          </button>
+        )}
+        {messageActions.indexOf(MESSAGE_ACTIONS.delete) > -1 && (
+          <button
+            aria-selected='false'
+            className={buttonClassName}
+            onClick={handleDelete}
+            role='option'
+          >
+            {t<string>('Delete')}
+          </button>
+        )}
+      </div>
     </div>
   );
 };