@imessaging/telegram-mtproto 0.8.19 → 0.8.21

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@imessaging/telegram-mtproto",
-  "version": "0.8.19",
+  "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.15",
+    "@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,
@@ -31,15 +32,25 @@ import type {
   ResolvedPeer,
   SelfAccount,
   SelfAwareTransport,
+  OutboundDocument,
   SendResult,
+  SentPart,
+  StoredDraft,
   TelegramPeerStore,
   TelegramRecipient,
   TransportStatus,
   TypingTransport,
 } from "@imessaging/core";
-import { formatText, formattedOf, isTelegramRecipient, MemoryPeerStore } from "@imessaging/core";
+import {
+  formatText,
+  formattedOf,
+  isTelegramRecipient,
+  MemoryPeerStore,
+  PartialSendError,
+} from "@imessaging/core";
 import { toTelegramEntities } from "./entities";
 import { peerToChat, unmarkPeerId } from "./incoming";
+import { stat } from "node:fs/promises";
 import bigInt from "big-integer";
 import { Api, TelegramClient } from "telegram";
 import { CustomFile } from "telegram/client/uploads";
@@ -155,6 +166,7 @@ export class TelegramMtprotoTransport
   implements
     CommandMenuTransport,
     ContactTransport,
+    DraftReaderTransport,
     DraftTransport,
     EditableMessageTransport,
     EditedMessageTransport,
@@ -436,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. */
@@ -848,7 +905,25 @@ export class TelegramMtprotoTransport
 
   async send(message: OutboundMessage): Promise<SendResult> {
     this.assertConnected();
-    return this.trackOwn(this.sendWithRetry(message, false));
+    return this.sendUploaded(message);
+  }
+
+  /** Файлы грузятся один раз, до отправки: повтор после смены пира их не перезагружает. */
+  private async sendUploaded(message: OutboundMessage): Promise<SendResult> {
+    const documents = message.documents ?? [];
+    const asCaption = message.textAsCaption === true && documents.length > 0;
+    // Длина известна только с `format`; иначе разметку разбирает gramjs, и судит Telegram.
+    const formatted = formattedOf(message);
+    if (asCaption && formatted && formatted.text.length > CAPTION_LIMIT) {
+      throw new Error(`Подпись длиннее ${CAPTION_LIMIT} знаков`);
+    }
+    // Пир до загрузки: несуществующий получатель не стоит загрузки гигабайта.
+    if (documents.length > 0) await this.getInputPeer(message.recipient);
+    // Сначала все файлы в Telegram, потом сообщения: сбой загрузки не оставит в чате полсообщения.
+    const uploaded: Api.TypeInputFile[] = [];
+    for (const document of documents) uploaded.push(await this.uploadDocument(document));
+    // Своё ждут только на время отправки: долгая загрузка не задерживает чужие исходящие события.
+    return this.trackOwn(this.sendWithRetry(message, uploaded, false));
   }
 
   /**
@@ -990,7 +1065,12 @@ export class TelegramMtprotoTransport
       .sort((a, b) => Number(a.messageId) - Number(b.messageId));
   }
 
-  private async sendWithRetry(message: OutboundMessage, retried: boolean): Promise<SendResult> {
+  private async sendWithRetry(
+    message: OutboundMessage,
+    uploaded: Api.TypeInputFile[],
+    retried: boolean,
+  ): Promise<SendResult> {
+    const parts: SentPart[] = [];
     try {
       const peer = await this.getInputPeer(message.recipient);
       // С `format` уходят готовые выделения: gramjs тогда не разбирает текст своим markdown вовсе.
@@ -998,37 +1078,68 @@ export class TelegramMtprotoTransport
       const text = ft ? ft.text : message.text;
       const parseMode = ft ? undefined : message.parseMode?.toLowerCase();
       const formattingEntities = ft ? toTelegramEntities(ft) : undefined;
-      const sent = await this.client.sendMessage(peer, {
-        message: text,
-        parseMode,
-        formattingEntities,
-        replyTo: message.replyToMessageId ?? message.threadId,
-        buttons: sendMarkup(message),
-      });
-      this.rememberOwn(this.ownSent, peer, sent.id);
-      for (const document of message.documents ?? []) {
-        const data = Buffer.from(document.data);
-        const file = await this.client.sendFile(peer, {
-          file: new CustomFile(document.filename, data.length, "", data),
-          caption: document.caption,
-          replyTo: message.threadId,
+      const documents = message.documents ?? [];
+      const asCaption = message.textAsCaption === true && documents.length > 0;
+
+      let first: Api.Message | undefined;
+      const replyTo = message.replyToMessageId ?? message.threadId;
+      if (!asCaption && (text.trim() !== "" || documents.length === 0)) {
+        first = await this.client.sendMessage(peer, {
+          message: text,
+          parseMode,
+          formattingEntities,
+          replyTo,
+          buttons: sendMarkup(message),
         });
-        this.rememberOwn(this.ownSent, peer, file.id);
+        this.rememberOwn(this.ownSent, peer, first.id);
+        parts.push({ messageId: String(first.id), kind: "text", media: false });
+      }
+      // Текстового сообщения нет — ответ и кнопки несёт первый файл.
+      const fileLeads = first === undefined;
+      for (const [index, document] of documents.entries()) {
+        const captioned = asCaption && index === 0 && text.trim() !== "";
+        const leads = fileLeads && index === 0;
+        const sent = await this.client.sendFile(peer, {
+          file: documentMedia(document, uploaded[index]!),
+          caption: captioned ? text : document.caption,
+          parseMode: captioned ? parseMode : undefined,
+          formattingEntities: captioned ? formattingEntities : undefined,
+          replyTo: leads ? replyTo : message.threadId,
+          buttons: leads ? sendMarkup(message) : undefined,
+        });
+        this.rememberOwn(this.ownSent, peer, sent.id);
+        first ??= sent;
+        parts.push({ messageId: String(sent.id), kind: "document", media: hasFileMedia(sent) });
       }
       this.lastActivityAt = new Date();
       // Получатель есть, только если ответ принёс users: короткий ответ лички (UpdateShortSentMessage) их не несёт.
-      const chat = sent.chat;
+      const chat = first!.chat;
       return {
         transportId: this.id,
-        messageId: String(sent.id),
+        messageId: String(first!.id),
         recipientId: this.getPeerId(peer),
+        parts,
         ...(chat instanceof Api.User ? { mutualContact: Boolean(chat.mutualContact) } : {}),
       };
     } catch (error: unknown) {
+      // Часть уже в чате — повтор её задвоит; вызывающему отдаём, что именно ушло.
+      if (parts.length > 0) throw new PartialSendError(parts, error);
       if (retried || !this.isInvalidPeerError(error)) throw error;
       await this.peerStore.delete(this.options.accountId, message.recipient);
-      return this.sendWithRetry(message, true);
+      return this.sendWithRetry(message, uploaded, true);
+    }
+  }
+
+  /** Загрузка без отправки. С `path` файл читается с диска частями, без предела 20 МБ на буфер. */
+  private async uploadDocument(document: OutboundDocument): Promise<Api.TypeInputFile> {
+    if (document.data === undefined && !document.path) {
+      throw new Error(`Вложение ${document.filename}: нет ни data, ни path`);
     }
+    const file =
+      document.data !== undefined
+        ? new CustomFile(document.filename, document.data.length, "", Buffer.from(document.data))
+        : new CustomFile(document.filename, (await stat(document.path)).size, document.path);
+    return this.client.uploadFile({ file, workers: UPLOAD_WORKERS });
   }
 
   private async getInputPeer(recipient: MessageRecipient): Promise<Api.TypeInputPeer> {
@@ -1199,3 +1310,68 @@ export class TelegramMtprotoTransport
     return telegramError.code === 400 && INVALID_PEER_ERRORS.has(telegramError.errorMessage ?? "");
   }
 }
+
+/**
+ * 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;
+
+/** Предел подписи к файлу у обычной (не Premium) учётки. */
+const CAPTION_LIMIT = 1024;
+
+/**
+ * Что отдать в `sendFile`. С `video` или `mimeType` не фото/звука — готовый InputMediaUploadedDocument:
+ * sendFile сам `mimeType` не принимает, а видео без атрибутов шлёт 1×1 длиной 0. Фото и звук — gramjs.
+ */
+function documentMedia(document: OutboundDocument, file: Api.TypeInputFile) {
+  const video = document.video;
+  const own = document.mimeType && !/^(image|audio)\//.test(document.mimeType);
+  if (!video && !own) return file;
+  return new Api.InputMediaUploadedDocument({
+    file,
+    mimeType: document.mimeType ?? (video ? "video/mp4" : "application/octet-stream"),
+    attributes: [
+      new Api.DocumentAttributeFilename({ fileName: document.filename }),
+      ...(video
+        ? [
+            new Api.DocumentAttributeVideo({
+              w: video.width,
+              h: video.height,
+              duration: video.duration,
+              supportsStreaming: video.supportsStreaming,
+            }),
+          ]
+        : []),
+    ],
+  });
+}
+
+/** Файл дошёл: у сообщения есть документ или фото. */
+function hasFileMedia(message: Api.Message): boolean {
+  return (
+    (message.media instanceof Api.MessageMediaDocument &&
+      message.media.document instanceof Api.Document) ||
+    (message.media instanceof Api.MessageMediaPhoto && message.media.photo instanceof Api.Photo)
+  );
+}