@imessaging/telegram-mtproto 0.8.20 → 0.8.21

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@imessaging/telegram-mtproto",
-  "version": "0.8.20",
+  "version": "0.8.21",
   "description": "Telegram MTProto user-account transport for imessaging",
   "keywords": [
     "messaging",
@@ -31,7 +31,7 @@
     "test": "bun test"
   },
   "dependencies": {
-    "@imessaging/core": "^0.8.16",
+    "@imessaging/core": "^0.8.17",
     "big-integer": "^1.6.52"
   },
   "peerDependencies": {

package/src/telegram-mtproto-transport.ts

@@ -6,6 +6,7 @@ import type {
   PhoneLookupOptions,
   PhoneLookupTransport,
   PhoneMatch,
+  DraftReaderTransport,
   DraftTransport,
   MessageDraft,
   OutgoingObserverTransport,
@@ -34,6 +35,7 @@ import type {
   OutboundDocument,
   SendResult,
   SentPart,
+  StoredDraft,
   TelegramPeerStore,
   TelegramRecipient,
   TransportStatus,
@@ -164,6 +166,7 @@ export class TelegramMtprotoTransport
   implements
     CommandMenuTransport,
     ContactTransport,
+    DraftReaderTransport,
     DraftTransport,
     EditableMessageTransport,
     EditedMessageTransport,
@@ -445,15 +448,60 @@ export class TelegramMtprotoTransport
             : {}),
         }),
       );
+    await this.withPeerRetry(draft.recipient, save);
+    this.lastActivityAt = new Date();
+  }
+
+  /**
+   * Черновик в поле ввода — перед saveDraft, чтобы не затереть набранное человеком. Тема форума хранит свой
+   * черновик в ForumTopic, общий чат — в Dialog. undefined — черновика нет.
+   */
+  async readDraft(
+    recipient: MessageRecipient,
+    options: { threadId?: number } = {},
+  ): Promise<StoredDraft | undefined> {
+    this.assertConnected();
+    if (!isTelegramRecipient(recipient)) {
+      throw new Error(`Telegram MTProto cannot read a draft for ${recipient.type}`);
+    }
+    const { threadId } = options;
+    if (threadId !== undefined && (!Number.isSafeInteger(threadId) || threadId <= 0)) {
+      throw new Error(`Telegram thread id must be a positive integer: ${threadId}`);
+    }
+    const read = async (peer: Api.TypeInputPeer): Promise<Api.TypeDraftMessage | undefined> => {
+      if (threadId !== undefined) {
+        const result = await this.client.invoke(
+          new Api.messages.GetForumTopicsByID({ peer, topics: [threadId] }),
+        );
+        const topic = result.topics.find(
+          (t): t is Api.ForumTopic => t instanceof Api.ForumTopic && t.id === threadId,
+        );
+        // Нет темы — не «пустое поле»: иначе черновик ушёл бы в General поверх набранного там.
+        if (!topic) throw new Error(`Telegram forum topic not found: ${threadId}`);
+        return topic.draft;
+      }
+      // InputPeer, а не InputDialogPeer: клиент при отправке сам оборачивает пир (_getInputDialog).
+      const result = await this.client.invoke(new Api.messages.GetPeerDialogs({ peers: [peer] }));
+      const dialog = result.dialogs.find((d): d is Api.Dialog => d instanceof Api.Dialog);
+      return dialog?.draft;
+    };
+    const draft = await this.withPeerRetry(recipient, read);
+    this.lastActivityAt = new Date();
+    return toStoredDraft(draft, threadId);
+  }
+
+  /** Как у send: устаревший пир из хранилища — забыть и разрешить заново, один раз. */
+  private async withPeerRetry<T>(
+    recipient: MessageRecipient,
+    call: (peer: Api.TypeInputPeer) => Promise<T>,
+  ): Promise<T> {
     try {
-      await save(await this.getInputPeer(draft.recipient));
+      return await call(await this.getInputPeer(recipient));
     } catch (error: unknown) {
-      // Как у send: устаревший пир из хранилища — забыть и разрешить заново, один раз.
       if (!this.isInvalidPeerError(error)) throw error;
-      await this.peerStore.delete(this.options.accountId, draft.recipient);
-      await save(await this.getInputPeer(draft.recipient));
+      await this.peerStore.delete(this.options.accountId, recipient);
+      return call(await this.getInputPeer(recipient));
     }
-    this.lastActivityAt = new Date();
   }
 
   /** Свои новые исходящие — отправленное человеком с другого устройства; отправленное через send помечено sentByTransport. */
@@ -1263,6 +1311,29 @@ export class TelegramMtprotoTransport
   }
 }
 
+/**
+ * DraftMessage → StoredDraft. В теме saveDraft без ответа ставит ответ на корень темы — это не ответ человеку.
+ */
+function toStoredDraft(
+  draft: Api.TypeDraftMessage | undefined,
+  threadId?: number,
+): StoredDraft | undefined {
+  if (!(draft instanceof Api.DraftMessage)) return undefined;
+  const r = draft.replyTo instanceof Api.InputReplyToMessage ? draft.replyTo : undefined;
+  // Ответ в другой чат (replyToPeerId) — чужой id сообщения, здесь он не годится.
+  const replyTo = r && !r.replyToPeerId ? r.replyToMsgId : undefined;
+  const reply = replyTo !== undefined && replyTo !== threadId ? String(replyTo) : undefined;
+  // Очистка в теме оставляет только ответ на корень. Что-то кроме этого (медиа, чужой ответ) — черновик есть.
+  const onlyRoot =
+    !draft.replyTo || (r !== undefined && !r.replyToPeerId && r.replyToMsgId === threadId);
+  if (draft.message === "" && onlyRoot && !draft.media && !draft.richMessage) return undefined;
+  return {
+    text: draft.message,
+    ...(reply !== undefined ? { replyToMessageId: reply } : {}),
+    date: new Date(draft.date * 1000),
+  };
+}
+
 /** Параллельных частей при загрузке: с одной 25 МБ идут по 512 КБ за круг. */
 const UPLOAD_WORKERS = 4;