Small fixes

src/client/message_parse.ts

@@ -103,7 +103,7 @@ export function _getResponseMessage(
     return;
   }
   const randomToId = new Map<string, number>();
-  const idToMessage = new Map<number, Api.Message>();
+  const idToMessage = new Map<number, CustomMessage>();
   let schedMessage;
   for (const update of updates) {
     if (update instanceof Api.UpdateMessageID) {
@@ -112,36 +112,43 @@ export function _getResponseMessage(
       update instanceof Api.UpdateNewChannelMessage ||
       update instanceof Api.UpdateNewMessage
     ) {
-      new CustomMessage(update.message as unknown as Api.Message)._finishInit(
+      const message = new CustomMessage(
+        update.message as unknown as Api.Message,
+      );
+
+      message._finishInit(
         client,
         entities,
         inputChat,
       );
       if ("randomId" in request || isArrayLike(request)) {
         idToMessage.set(
           update.message.id,
-          update.message as unknown as Api.Message,
+          message,
         );
       } else {
-        return update.message as unknown as Api.Message;
+        return message;
       }
     } else if (
       update instanceof Api.UpdateEditMessage &&
       "peer" in request &&
       entityType_(request.peer) !== EntityType_.CHANNEL
     ) {
-      new CustomMessage(update.message as unknown as Api.Message)._finishInit(
+      const message = new CustomMessage(
+        update.message as unknown as Api.Message,
+      );
+      message._finishInit(
         client,
         entities,
         inputChat,
       );
       if ("randomId" in request) {
         idToMessage.set(
           update.message.id,
-          update.message as unknown as Api.Message,
+          message,
         );
       } else if ("id" in request && request.id === update.message.id) {
-        return update.message;
+        return message;
       }
     } else if (
       update instanceof Api.UpdateEditChannelMessage &&
@@ -150,23 +157,29 @@ export function _getResponseMessage(
         getPeerId((update.message as unknown as Api.Message).peerId!)
     ) {
       if (request.id === update.message.id) {
-        new CustomMessage(update.message as unknown as Api.Message)._finishInit(
+        const message = new CustomMessage(
+          update.message as unknown as Api.Message,
+        );
+        message._finishInit(
           client,
           entities,
           inputChat,
         );
-        return update.message;
+        return message;
       }
     } else if (update instanceof Api.UpdateNewScheduledMessage) {
-      new CustomMessage(update.message as unknown as Api.Message)._finishInit(
+      const message = new CustomMessage(
+        update.message as unknown as Api.Message,
+      );
+      message._finishInit(
         client,
         entities,
         inputChat,
       );
-      schedMessage = update.message as unknown as Api.Message;
+      schedMessage = message;
       idToMessage.set(
         update.message.id,
-        update.message as unknown as Api.Message,
+        message,
       );
     } else if (update instanceof Api.UpdateMessagePoll) {
       if (request.media.poll.id === update.pollId) {

src/client/telegram_client.ts

@@ -592,8 +592,8 @@ export class TelegramClient extends TelegramBaseClient
   ):
     | CustomMessage
     | Api.TypeMessage
-    | Map<number, Api.Message>
-    | (Api.Message | undefined)[]
+    | Map<number, CustomMessage>
+    | (CustomMessage | undefined)[]
     | undefined {
     return parseMethods._getResponseMessage(this, req, result, inputChat);
   }