fix: memoize integration provided fn

GetStream · Dec 5, 2024 · baa322b · baa322b
1 parent 7eb09c9
commit baa322b
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 6 deletions.
diff --git a/package/src/components/Message/Message.tsx b/package/src/components/Message/Message.tsx
@@ -314,6 +314,10 @@ const MessageWithContext = <
     updateMessage,
   } = props;
   const isMessageAIGenerated = messagesContext.isMessageAIGenerated;
+  const isAIGenerated = useMemo(
+    () => isMessageAIGenerated(message),
+    [message, isMessageAIGenerated],
+  );
   const { client } = chatContext;
   const {
     theme: {
@@ -466,7 +470,7 @@ const MessageWithContext = <
       case 'poll':
         return !!message.poll_id;
       case 'ai_text':
-        return isMessageAIGenerated(message);
+        return isAIGenerated;
       case 'text':
       default:
         return !!message.text;

diff --git a/package/src/components/Message/MessageSimple/MessageContent.tsx b/package/src/components/Message/MessageSimple/MessageContent.tsx
@@ -1,4 +1,4 @@
-import React from 'react';
+import React, { useMemo } from 'react';
 import {
   AnimatableNumericValue,
   LayoutChangeEvent,
@@ -188,6 +188,11 @@ const MessageContentWithContext = <
     setMessageContentWidth(width);
   };
 
+  const isAIGenerated = useMemo(
+    () => isMessageAIGenerated(message),
+    [message, isMessageAIGenerated],
+  );
+
   const error = message.type === 'error' || message.status === MessageStatusTypes.FAILED;
 
   const groupStyle = `${alignment}_${groupStyles?.[0]?.toLowerCase?.()}`;
@@ -398,15 +403,15 @@ const MessageContentWithContext = <
                 ) : null;
               }
               case 'ai_text':
-                return isMessageAIGenerated(message) ? (
+                return isAIGenerated ? (
                   <StreamingMessageView
                     key={`ai_message_text_container_${messageContentOrderIndex}`}
                   />
                 ) : null;
               case 'text':
               default:
                 return (otherAttachments.length && otherAttachments[0].actions) ||
-                  isMessageAIGenerated(message) ? null : (
+                  isAIGenerated ? null : (
                   <MessageTextContainer<StreamChatGenerics>
                     key={`message_text_container_${messageContentOrderIndex}`}
                   />

diff --git a/package/src/components/Message/MessageSimple/MessageFooter.tsx b/package/src/components/Message/MessageSimple/MessageFooter.tsx
@@ -1,4 +1,4 @@
-import React from 'react';
+import React, { useMemo } from 'react';
 import { StyleSheet, Text, View } from 'react-native';
 
 import type { Attachment } from 'stream-chat';
@@ -116,6 +116,11 @@ const MessageFooterWithContext = <
   } = useTheme();
   const { t } = useTranslationContext();
 
+  const isAIGenerated = useMemo(
+    () => isMessageAIGenerated(message),
+    [message, isMessageAIGenerated],
+  );
+
   if (isDeleted) {
     return (
       <View style={[styles.container, metaContainer]} testID='message-deleted'>
@@ -131,7 +136,7 @@ const MessageFooterWithContext = <
     return null;
   }
 
-  const isEdited = isEditedMessage(message) && !isMessageAIGenerated(message);
+  const isEdited = isEditedMessage(message) && !isAIGenerated;
 
   return (
     <>