From dc4c33a07be138370bbba2ecacb9caa4486f49dc Mon Sep 17 00:00:00 2001
From: Michael Telatynski <7t3chguy@gmail.com>
Date: Fri, 19 Apr 2024 12:40:07 +0100
Subject: [PATCH] Remove legacy consumers of the OverflowMenuContext in favour
 of HOCs

Signed-off-by: Michael Telatynski <7t3chguy@gmail.com>
---
 .../views/rooms/MessageComposerButtons.tsx           | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/components/views/rooms/MessageComposerButtons.tsx b/src/components/views/rooms/MessageComposerButtons.tsx
index 7f23efbce2..88e8f2def3 100644
--- a/src/components/views/rooms/MessageComposerButtons.tsx
+++ b/src/components/views/rooms/MessageComposerButtons.tsx
@@ -289,20 +289,22 @@ function voiceRecordingButton(props: IProps, narrow: boolean): ReactElement | nu
 }
 
 function pollButton(room: Room, relation?: IEventRelation): ReactElement {
-    return <PollButton key="polls" room={room} relation={relation} />;
+    return (
+        <OverflowMenuContext.Consumer key="polls">
+            {(context) => <PollButton room={room} relation={relation} context={context} />}
+        </OverflowMenuContext.Consumer>
+    );
 }
 
 interface IPollButtonProps {
     room: Room;
     relation?: IEventRelation;
+    context: React.ContextType<typeof OverflowMenuContext>;
 }
 
 class PollButton extends React.PureComponent<IPollButtonProps> {
-    public static contextType = OverflowMenuContext;
-    public context!: React.ContextType<typeof OverflowMenuContext>;
-
     private onCreateClick = (): void => {
-        this.context?.(); // close overflow menu
+        this.props.context?.(); // close overflow menu
         const canSend = this.props.room.currentState.maySendEvent(
             M_POLL_START.name,
             MatrixClientPeg.safeGet().getSafeUserId(),