override doUpdateMessageRequest

Author: mahboubii

src/components/Channel.js

@@ -79,6 +79,10 @@ const Channel = withChatContext(
        * Override send message request (Advanced usage only)
        * */
       doSendMessageRequest: PropTypes.func,
+      /**
+       * Override update message request (Advanced usage only)
+       * */
+      doUpdateMessageRequest: PropTypes.func,
     };
 
     static defaultProps = {

src/components/ChannelInner.js

@@ -126,6 +126,8 @@ export class ChannelInner extends PureComponent {
     Attachment: PropTypes.oneOfType([PropTypes.node, PropTypes.elementType]),
     /** Override send message request (Advanced usage only) */
     doSendMessageRequest: PropTypes.func,
+    /** Override update message request (Advanced usage only) */
+    doUpdateMessageRequest: PropTypes.func,
   };
 
   static defaultProps = {
@@ -354,6 +356,20 @@ export class ChannelInner extends PureComponent {
     return message;
   };
 
+  // eslint-disable-next-line require-await
+  editMessage = async (updatedMessage) => {
+    if (this.props.doUpdateMessageRequest) {
+      return Promise.resolve(
+        this.props.doUpdateMessageRequest(
+          this.props.channel.cid,
+          updatedMessage,
+        ),
+      );
+    }
+
+    return this.props.client.updateMessage(updatedMessage);
+  };
+
   _sendMessage = async (message) => {
     // Scrape the reserved fields if present.
     const {
@@ -577,6 +593,7 @@ export class ChannelInner extends PureComponent {
     updateMessage: this.updateMessage,
     removeMessage: this.removeMessage,
     sendMessage: this.sendMessage,
+    editMessage: this.editMessage,
     retrySendMessage: this.retrySendMessage,
     setEditingState: this.setEditingState,
     clearEditingState: this.clearEditingState,

src/components/MessageInput.js

@@ -374,11 +374,9 @@ const MessageInput = withKeyboardContext(
               // TODO: Remove this line and show an error when submit fails
               this.props.clearEditingState();
 
-              const updateMessagePromise = this.props.client
-                .updateMessage(updatedMessage)
-                .then(() => {
-                  this.props.clearEditingState();
-                });
+              const updateMessagePromise = this.props
+                .editMessage(updatedMessage)
+                .then(this.props.clearEditingState);
               logChatPromiseExecution(updateMessagePromise, 'update message');
             } else {
               try {
@@ -404,7 +402,7 @@ const MessageInput = withKeyboardContext(
 
           updateMessage = async () => {
             try {
-              await this.props.client.updateMessage({
+              await this.props.client.editMessage({
                 ...this.props.editing,
                 text: this.state.text,
               });

types/index.d.ts

@@ -91,6 +91,7 @@ export interface ChannelContextValue {
     updatedMessage: Client.MessageResponse,
     extraState?: object,
   ): void;
+  editMessage?(message: Client.Message): void | Promise<Client.MessageResponse>;
   retrySendMessage?(message: Client.Message): void;
   removeMessage?(updatedMessage: Client.MessageResponse): void;
   setEditingState?(message: Client.Message): void;
@@ -146,6 +147,11 @@ export interface ChannelProps extends ChatContextValue {
     channelId: string,
     message: Client.Message,
   ): void | Promise<Client.MessageResponse>;
+  /** Function that overrides default updateMessage in chat client */
+  doUpdateMessageRequest?(
+    channelId: string,
+    message: Client.Message,
+  ): void | Promise<Client.MessageResponse>;
 }
 
 export type listType = 'channel' | 'message' | 'default';