fix: downloadMediaMessage return type matched parameter (#757)

* fix: downloadMediaMessage return type matched parameter

* chore: keyword-spacing eslint rule warn instead of error

* chore: linting

aaaaaaaaaaaaaaaaaaaa

.eslintrc.json

@@ -27,6 +27,9 @@
                 "selector": "TSEnumDeclaration",
                 "message": "Don't declare enums, use literals instead"
             }
+        ],
+        "keyword-spacing": [
+            "warn"
         ]
     }
 }

src/Utils/messages.ts

@@ -3,6 +3,7 @@ import axios from 'axios'
 import { randomBytes } from 'crypto'
 import { promises as fs } from 'fs'
 import { Logger } from 'pino'
+import { type Transform } from 'stream'
 import { proto } from '../../WAProto'
 import { MEDIA_KEYS, URL_REGEX, WA_DEFAULT_EPHEMERAL } from '../Defaults'
 import {
@@ -868,16 +869,14 @@ const REUPLOAD_REQUIRED_STATUS = [410, 404]
 /**
  * Downloads the given message. Throws an error if it's not a media message
  */
-export const downloadMediaMessage = async(
+export const downloadMediaMessage = async<Type extends 'buffer' | 'stream'>(
 	message: WAMessage,
-	type: 'buffer' | 'stream',
+	type: Type,
 	options: MediaDownloadOptions,
 	ctx?: DownloadMediaMessageContext
 ) => {
-	try {
 	const result = await downloadMsg()
-		return result
-	} catch(error) {
+		.catch(async(error) => {
 			if(ctx) {
 				if(axios.isAxiosError(error)) {
 					// check if the message requires a reupload
@@ -892,7 +891,9 @@ export const downloadMediaMessage = async(
 			}
 
 			throw error
-	}
+		})
+
+	return result as Type extends 'buffer' ? Buffer : Transform
 
 	async function downloadMsg() {
 		const mContent = extractMessageContent(message.message)