feat: don't use rich header if message contains urls

sakarie9 · Mar 18, 2024 · 4b4cdf8 · 4b4cdf8
1 parent fba5d14
commit 4b4cdf8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/main/src/services/ForwardService.ts b/main/src/services/ForwardService.ts
@@ -11,7 +11,7 @@ import {
   segment,
   Sendable,
 } from 'icqq';
-import { fetchFile, getBigFaceUrl, getImageUrlByMd5 } from '../utils/urls';
+import { fetchFile, getBigFaceUrl, getImageUrlByMd5, isContainsUrl } from '../utils/urls';
 import { ButtonLike, FileLike } from 'telegram/define';
 import { getLogger, Logger } from 'log4js';
 import path from 'path';
@@ -392,7 +392,9 @@ export default class ForwardService {
       else if (files.length) {
         messageToSend.file = files;
       }
-      else if (event.message_type === 'group' && (pair.flags | this.instance.flags) & flags.RICH_HEADER && env.WEB_ENDPOINT) {
+      else if (event.message_type === 'group' && (pair.flags | this.instance.flags) & flags.RICH_HEADER && env.WEB_ENDPOINT 
+      // 当消息包含链接时不显示 RICH HEADER
+      && !isContainsUrl(message)) {
         // 没有文件时才能显示链接预览
         richHeaderUsed = true;
         // https://github.com/tdlib/td/blob/437c2d0c6e0ad104022d5ad86ddc8aedc41cb7a8/td/telegram/MessageContent.cpp#L2575

diff --git a/main/src/utils/urls.ts b/main/src/utils/urls.ts
@@ -32,3 +32,7 @@ export async function fetchFile(url: string): Promise<Buffer> {
 export function getAvatar(room: number | Friend | Group) {
   return fetchFile(getAvatarUrl(room));
 }
+
+export function isContainsUrl(msg: string): boolean {
+  return msg.includes("https://") || msg.includes("http://")
+}