Skip to content

Commit

Permalink
feat: Added WPP.chat.editMessage
Browse files Browse the repository at this point in the history
  • Loading branch information
icleitoncosta committed Nov 19, 2022
1 parent 746d349 commit c63dcd4
Show file tree
Hide file tree
Showing 3 changed files with 128 additions and 0 deletions.
88 changes: 88 additions & 0 deletions src/chat/functions/editMessage.ts
@@ -0,0 +1,88 @@
/*!
* Copyright 2021 WPPConnect Team
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

import { WPPError } from '../../util';
import { ChatModel, MsgKey, MsgModel, Wid } from '../../whatsapp';
import {
canEditMessage,
fetchLinkPreview,
findFirstWebLink,
sendMessageEdit,
} from '../../whatsapp/functions';
import { LinkPreviewOptions, RawMessage } from '..';
import { getMessageById, prepareRawMessage } from '.';

/**
* Send a text message
*
* @example
* WPP.chat.editMessage('[number]@c.us', 'New text for message', {
* linkPreview: true,
* mentionedJidList: ['5521985232287@c.us']
* });
* ```
* @category Message
*/
export async function editMessage(
msgId: string | MsgKey,
newText: any,
options?: LinkPreviewOptions & { mentionedJidList?: Wid[] }
): Promise<MsgModel> {
const msg = await getMessageById(msgId);

const canEdit = canEditMessage(msg);
if (!canEdit) {
throw new WPPError(`edit_message_error`, `Cannot edit this message`);
}
let jidList: Wid[] | undefined = [];
let linkPreview: any = false;

if (options?.mentionedJidList) {
const rawMessage: RawMessage = {
body: newText,
type: 'chat',
};

const raw = await prepareRawMessage(msg.chat as ChatModel, rawMessage, {
mentionedList: options.mentionedJidList,
});
jidList = raw.mentionedJidList;
}

if (options?.linkPreview != false) {
const override =
typeof options?.linkPreview === 'object' ? options.linkPreview : {};

if (newText) {
try {
const link = findFirstWebLink(newText);
if (link) {
const preview = await fetchLinkPreview(link);
if (preview?.data) {
linkPreview = { ...preview.data, ...override };
}
}
} catch (error) {}
}
}

await sendMessageEdit(msg as MsgModel, newText, {
linkPreview: linkPreview,
mentionedJidList: jidList as any,
});

return await getMessageById(msgId);
}
1 change: 1 addition & 0 deletions src/chat/functions/index.ts
Expand Up @@ -22,6 +22,7 @@ export { clear } from './clear';
export { delete } from './delete';
export { deleteMessage, DeleteMessageReturn } from './deleteMessage';
export { downloadMedia } from './downloadMedia';
export { editMessage } from './editMessage';
export { find } from './find';
export { forwardMessage, ForwardMessagesOptions } from './forwardMessage';
export { generateMessageID } from './generateMessageID';
Expand Down
39 changes: 39 additions & 0 deletions src/whatsapp/functions/sendMessageEdit.ts
@@ -0,0 +1,39 @@
/*!
* Copyright 2022 WPPConnect Team
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

import { LinkPreviewOptions } from '../../chat';
import { exportModule } from '../exportModule';
import { Wid } from '../misc/Wid';
import { MsgModel } from '../models';

/** @whatsapp 262489
*/
export declare function sendMessageEdit(
msg: MsgModel,
text: string,
options: {
linkPreview: LinkPreviewOptions | null;
mentionedJidList: Wid[] | [];
}
): Promise<any>;

exportModule(
exports,
{
sendMessageEdit: 'sendMessageEdit',
},
(m) => m.sendMessageEdit
);

0 comments on commit c63dcd4

Please sign in to comment.