Merge pull request #118 from mahboubii/feature/override-do-send-message-request

Feature/override do send message request

Author: jaapbakker88

src/components/Channel.js

@@ -75,6 +75,14 @@ const Channel = withChatContext(
        * Available built-in component (also accepts the same props as): [Attachment](https://getstream.github.io/stream-chat-react-native/#attachment)
        * */
       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 = {

src/components/ChannelInner.js

@@ -124,6 +124,10 @@ export class ChannelInner extends PureComponent {
     isOnline: PropTypes.bool,
     Message: PropTypes.oneOfType([PropTypes.node, PropTypes.elementType]),
     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 = {
@@ -352,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 {
@@ -380,7 +398,16 @@ export class ChannelInner extends PureComponent {
     };
 
     try {
-      const messageResponse = await this.props.channel.sendMessage(messageData);
+      let messageResponse;
+      if (this.props.doSendMessageRequest) {
+        messageResponse = await this.props.doSendMessageRequest(
+          this.props.channel.cid,
+          messageData,
+        );
+      } else {
+        messageResponse = await this.props.channel.sendMessage(messageData);
+      }
+
       // replace it after send is completed
       if (messageResponse.message) {
         messageResponse.message.status = 'received';
@@ -566,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;
@@ -141,6 +142,16 @@ export interface ChannelProps extends ChatContextValue {
   EmptyStateIndicator?: React.ElementType<EmptyStateIndicatorProps>;
   Message?: React.ElementType<MessageUIComponentProps>;
   Attachment?: React.ElementType<AttachmentProps>;
+  /** Function that overrides default sendMessage in chat client */
+  doSendMessageRequest?(
+    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';