From c06766cbaf65690092bf1d7081237e4bf2cc864c Mon Sep 17 00:00:00 2001
From: Teddy Sterne <teddy.sterne@fountainlife.com>
Date: Fri, 17 May 2024 14:42:27 -0400
Subject: [PATCH] fix: Modify image message bubble styles

---
 src/screens/DirectMessagesScreen.tsx | 75 +++++++++++++++++++++++++---
 1 file changed, 67 insertions(+), 8 deletions(-)

diff --git a/src/screens/DirectMessagesScreen.tsx b/src/screens/DirectMessagesScreen.tsx
index 748460d5..b8fa82e6 100644
--- a/src/screens/DirectMessagesScreen.tsx
+++ b/src/screens/DirectMessagesScreen.tsx
@@ -1,4 +1,10 @@
-import React, { useCallback, useEffect, useLayoutEffect, useRef } from 'react';
+import React, {
+  useCallback,
+  useEffect,
+  useLayoutEffect,
+  useRef,
+  useState,
+} from 'react';
 import {
   Bubble,
   Composer,
@@ -11,7 +17,13 @@ import {
   MessageTextProps,
   Time,
 } from 'react-native-gifted-chat';
-import { StyleSheet, ActivityIndicator, ScrollView } from 'react-native';
+import {
+  StyleSheet,
+  ActivityIndicator,
+  ScrollView,
+  Image,
+  Dimensions,
+} from 'react-native';
 import {
   useInfinitePrivatePosts,
   useCreatePrivatePostMutation,
@@ -236,10 +248,17 @@ export const DirectMessagesScreen = ({
         return (
           <Bubble
             {...props}
-            wrapperStyle={{
-              left: styles.leftMessageView,
-              right: styles.rightMessageView,
-            }}
+            wrapperStyle={
+              props.currentMessage?.text
+                ? {
+                    left: styles.leftMessageView,
+                    right: styles.rightMessageView,
+                  }
+                : {
+                    right: styles.messageImageBubble,
+                    left: styles.messageImageBubble,
+                  }
+            }
             textStyle={{
               left: styles.leftText,
               right: styles.rightText,
@@ -248,7 +267,8 @@ export const DirectMessagesScreen = ({
             usernameStyle={styles.signatureText}
             renderTime={(timeProps) =>
               timeProps.currentMessage?.createdAt !==
-                (timeProps as any).nextMessage?.createdAt && (
+                (timeProps as any).nextMessage?.createdAt &&
+              props.currentMessage?.text && (
                 <Time
                   {...timeProps}
                   timeTextStyle={{
@@ -418,13 +438,43 @@ const renderCustomImage = memoize(
 const CustomMessageImage = React.memo(
   (props: MessageImageProps<IMessage>) => {
     const { styles } = useStyles(defaultStyles);
+    const [dimensions, setDimensions] = useState({});
+
+    useEffect(() => {
+      const image = props.currentMessage?.image;
+
+      if (image) {
+        Image.getSize(image, (width, height) => {
+          const aspectRatio = width / height;
+          const largerDimension = width > height ? 'width' : 'height';
+          setDimensions({
+            [largerDimension]: 'auto',
+            aspectRatio,
+          });
+        });
+      }
+    }, [props.currentMessage?.image]);
+
     return (
       <View style={{ position: 'relative' }}>
         <ActivityIndicator
           style={styles.messageImageLoadingIcon}
           color={styles.messageImageLoadingIcon?.color}
         />
-        <MessageImage {...props} />
+        <MessageImage
+          {...props}
+          lightboxProps={{
+            underlayColor: styles.messageImageLightbox?.backgroundColor,
+            style: styles.messageImageLightbox,
+          }}
+          imageStyle={[
+            dimensions,
+            {
+              maxWidth: Dimensions.get('screen').width * 0.65,
+            },
+            styles.messageImage,
+          ]}
+        />
       </View>
     );
   },
@@ -599,6 +649,15 @@ const defaultStyles = createStyles('DirectMessagesScreen', (theme) => ({
     ...StyleSheet.absoluteFillObject,
     color: theme.colors.onPrimaryContainer,
   },
+  messageImageBubble: {
+    backgroundColor: 'transparent',
+  },
+  messageImage: {
+    margin: 0,
+  },
+  messageImageLightbox: {
+    backgroundColor: 'transparent',
+  },
 }));
 
 declare module '@styles' {