Better error handling

src/WAClient/Base.ts

@@ -59,7 +59,7 @@ export default class WhatsAppWebBase extends WAConnection {
      * @param jid the ID of the person/group who you are updating
      * @param type your presence
      */
-    async updatePresence(jid: string, type: Presence) {
+    async updatePresence(jid: string | null, type: Presence) {
         const json = [
             'action',
             { epoch: this.msgCount.toString(), type: 'set' },

src/WAClient/Messages.ts

@@ -17,7 +17,7 @@ import {
     WAUrlInfo,
 } from './Constants'
 import { generateMessageID, sha256, hmacSign, aesEncrypWithIV, randomBytes } from '../WAConnection/Utils'
-import { WAMessageContent, WAMetric, WAFlag, WANode, WAMessage, WAMessageProto } from '../WAConnection/Constants'
+import { WAMessageContent, WAMetric, WAFlag, WANode, WAMessage, WAMessageProto, BaileysError } from '../WAConnection/Constants'
 import { validateJIDForSending, generateThumbnail, getMediaKeys, decodeMediaMessageBuffer, extensionForMediaMessage } from './Utils'
 import { proto } from '../../WAMessage/WAMessage'
 
@@ -173,11 +173,11 @@ export default class WhatsAppWebMessages extends WhatsAppWebGroups {
      */
     async updateMediaMessage (message: WAMessage) {
         const content = message.message?.audioMessage || message.message?.videoMessage || message.message?.imageMessage || message.message?.stickerMessage || message.message?.documentMessage 
-        if (!content) throw new Error (`given message ${message.key.id} is not a media message`)
+        if (!content) throw new BaileysError (`given message ${message.key.id} is not a media message`, message)
         
         const query = ['query',{type: 'media', index: message.key.id, owner: message.key.fromMe ? 'true' : 'false', jid: message.key.remoteJid, epoch: this.msgCount.toString()},null]
         const response = await this.query (query, [WAMetric.queryMedia, WAFlag.ignore])
-        if (parseInt(response[1].code) !== 200) throw new Error ('unexpected status ' + response[1].code)
+        if (parseInt(response[1].code) !== 200) throw new BaileysError ('unexpected status ' + response[1].code, response)
         
         Object.keys (response[1]).forEach (key => content[key] = response[1][key]) // update message
     }
@@ -213,7 +213,7 @@ export default class WhatsAppWebMessages extends WhatsAppWebGroups {
                 } else if ('text' in message) {
                     m.extendedTextMessage = message as WATextMessage
                 } else {
-                    throw new Error ('message needs to be a string or object with property \'text\'')
+                    throw new BaileysError ('message needs to be a string or object with property \'text\'', message)
                 }
                 break
             case MessageType.location:

src/WAClient/Tests.ts

@@ -12,7 +12,6 @@ const testJid = process.env.TEST_JID || '1234@s.whatsapp.net' // set TEST_JID=xy
 
 async function sendAndRetreiveMessage(client: WAClient, content, type: MessageType, options: MessageOptions = {}) {
     const response = await client.sendMessage(testJid, content, type, options)
-    assert.strictEqual(response.status, 200)
     const messages = await client.loadConversation(testJid, 1, null, true)
     assert.strictEqual(messages[0].key.id, response.messageID)
     return messages[0]

src/WAClient/Utils.ts

@@ -2,7 +2,7 @@ import { MessageType, HKDFInfoKeys, MessageOptions, WAMessageType } from './Cons
 import Jimp from 'jimp'
 import * as fs from 'fs'
 import fetch from 'node-fetch'
-import { WAMessage, WAMessageContent } from '../WAConnection/Constants'
+import { WAMessage, WAMessageContent, BaileysError } from '../WAConnection/Constants'
 import { hmacSign, aesDecryptWithIV, hkdf } from '../WAConnection/Utils'
 import { proto } from '../../WAMessage/WAMessage'
 import { randomBytes } from 'crypto'
@@ -55,11 +55,8 @@ const extractVideoThumb = async (
     new Promise((resolve, reject) => {
         const cmd = `ffmpeg -ss ${time} -i ${path} -y -s ${size.width}x${size.height} -vframes 1 -f image2 ${destPath}`
         exec(cmd, (err) => {
-            if (err) {
-                reject(err)
-            } else {
-                resolve()
-            }
+            if (err) reject(err)
+            else resolve()
         })
     }) as Promise<void>
 
@@ -112,10 +109,10 @@ export async function decodeMediaMessageBuffer(message: WAMessageContent) {
     */
     const type = Object.keys(message)[0] as MessageType
     if (!type) {
-        throw new Error('unknown message type')
+        throw new BaileysError('unknown message type', message)
     }
     if (type === MessageType.text || type === MessageType.extendedText) {
-        throw new Error('cannot decode text message')
+        throw new BaileysError('cannot decode text message', message)
     }
     if (type === MessageType.location || type === MessageType.liveLocation) {
         return new Buffer(message[type].jpegThumbnail)
@@ -123,7 +120,7 @@ export async function decodeMediaMessageBuffer(message: WAMessageContent) {
     let messageContent: proto.IVideoMessage | proto.IImageMessage | proto.IAudioMessage | proto.IDocumentMessage
     if (message.productMessage) {
         const product = message.productMessage.product?.productImage
-        if (!product) throw new Error ('product has no image')
+        if (!product) throw new BaileysError ('product has no image', message)
         messageContent = product
     } else {
         messageContent = message[type]
@@ -134,7 +131,7 @@ export async function decodeMediaMessageBuffer(message: WAMessageContent) {
     const buffer = await fetched.buffer()
 
     if (buffer.length <= 10) {
-        throw new Error ('Empty buffer returned. File has possibly been deleted from WA servers. Run `client.updateMediaMessage()` to refresh the url')
+        throw new BaileysError ('Empty buffer returned. File has possibly been deleted from WA servers. Run `client.updateMediaMessage()` to refresh the url', {status: 404})
     }
 
     const decryptedMedia = (type: MessageType) => {
@@ -163,7 +160,7 @@ export async function decodeMediaMessageBuffer(message: WAMessageContent) {
             if (i === 0) { console.log (`decryption of ${type} media with original HKDF key failed`) }
         }
     }
-    throw new Error('Decryption failed, HMAC sign does not match')
+    throw new BaileysError('Decryption failed, HMAC sign does not match', {status: 400})
 }
 export function extensionForMediaMessage(message: WAMessageContent) {
     const getExtension = (mimetype: string) => mimetype.split(';')[0].split('/')[1]