Questionable changes

Author: arnautov-anton

src/components/Channel/hooks/useEditMessageHandler.ts

@@ -1,26 +1,40 @@
-import type { MessageRequest, StreamChat, UpdateMessageOptions } from 'stream-chat';
+import type {
+  LocalMessage,
+  MessageRequest,
+  MessageResponse,
+  StreamChat,
+  UpdateMessageAPIResponse,
+  UpdateMessageOptions,
+} from 'stream-chat';
 
 import { useChatContext } from '../../../context/ChatContext';
 
 type UpdateHandler = (
   cid: string,
-  updatedMessage: MessageRequest,
+  updatedMessage: LocalMessage | MessageResponse,
   options?: UpdateMessageOptions,
 ) => ReturnType<StreamChat['updateMessage']>;
 
-export const useEditMessageHandler = (doUpdateMessageRequest?: UpdateHandler) => {
+export const useEditMessageHandler = (
+  doUpdateMessageRequest?: UpdateHandler,
+): ((
+  updatedMessage: LocalMessage | MessageResponse,
+  options?: UpdateMessageOptions,
+) => Promise<UpdateMessageAPIResponse>) => {
   const { channel, client } = useChatContext('useEditMessageHandler');
 
-  return (updatedMessage: MessageRequest, options?: UpdateMessageOptions) => {
+  return (
+    updatedMessage: LocalMessage | MessageResponse,
+    options?: UpdateMessageOptions,
+  ) => {
     if (doUpdateMessageRequest && channel) {
       return Promise.resolve(
         doUpdateMessageRequest(channel.cid, updatedMessage, options),
       );
     }
     return client.updateMessage({
-      // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-      id: updatedMessage.id!,
-      message: updatedMessage,
+      id: updatedMessage.id,
+      message: updatedMessage as unknown as MessageRequest,
       ...options,
     });
   };

src/components/MessageComposer/QuotedMessagePreview.tsx

@@ -26,6 +26,7 @@ import {
   isVoiceRecordingAttachment,
   type LocalMessage,
   type MessageComposerState,
+  type MessageResponse,
   type PollResponseData,
   type SharedLocationResponseData,
   type TranslationLanguages,
@@ -55,7 +56,7 @@ const messageComposerStateStoreSelector = (state: MessageComposerState) => ({
 });
 
 export type QuotedMessagePreviewProps = {
-  getQuotedMessageAuthor?: (message: LocalMessage) => string;
+  getQuotedMessageAuthor?: (message: LocalMessage | MessageResponse) => string;
   renderText?: MessageContextValue['renderText'];
 };
 
@@ -96,7 +97,7 @@ type GroupedAttachments = Record<AttachmentType, Attachment[]> & {
   total: number;
 };
 
-const getGroupedAttachments = (quotedMessage: LocalMessage | null) => {
+const getGroupedAttachments = (quotedMessage: LocalMessage | MessageResponse | null) => {
   const groupedAttachments = {
     documents: [],
     giphies: [],
@@ -169,7 +170,7 @@ type PreviewType =
   | 'mixed';
 
 const getAttachmentIconWithType = (
-  quotedMessage: LocalMessage | null,
+  quotedMessage: LocalMessage | MessageResponse | null,
   giphyVersionName: GiphyVersions,
 ): {
   groupedAttachments: GroupedAttachments;
@@ -314,7 +315,7 @@ export const QuotedMessagePreview = ({
 };
 
 type QuotedMessagePreviewUIProps = QuotedMessagePreviewProps & {
-  quotedMessage: LocalMessage;
+  quotedMessage: LocalMessage | MessageResponse;
   authorLabel?: ReactNode;
   className?: string;
   onClick?: MouseEventHandler<HTMLDivElement>;