koishi-plugin-share-links-analysis 0.5.0 → 0.5.1

package/lib/core.d.ts

@@ -1,5 +1,6 @@
 import { Context, Session } from 'koishi';
 import { Link, PluginConfig, ParsedInfo } from './types';
+export declare const parsers_str: string[];
 /**
  * 从文本中解析出所有支持的链接
  * @param content 消息内容

package/lib/core.js

@@ -34,6 +34,7 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.parsers_str = void 0;
 exports.resolveLinks = resolveLinks;
 exports.processLink = processLink;
 exports.init = init;
@@ -42,6 +43,7 @@ const Xiaohongshu = __importStar(require("./parsers/xiaohongshu"));
 const Twitter = __importStar(require("./parsers/twitter"));
 // 定义所有支持的解析器
 const parsers = [Bilibili, Xiaohongshu, Twitter];
+exports.parsers_str = ['bilibili', 'xiaohongshu', 'twitter'];
 /**
  * 从文本中解析出所有支持的链接
  * @param content 消息内容

package/lib/index.js

@@ -24,7 +24,7 @@ exports.Config = koishi_1.Schema.intersect([
             koishi_1.Schema.const('1').description('低清晰度优先'),
             koishi_1.Schema.const('2').description('高清晰度优先'),
         ]).role('radio').default('1').description("发送的视频清晰度优先策略"),
-        Max_size: koishi_1.Schema.number().default(5).description("允许发送的最大文件大小（Mb）"),
+        Max_size: koishi_1.Schema.number().default(20).description("允许发送的最大文件大小（Mb）"),
         Max_size_tip: koishi_1.Schema.string().default('文件体积过大，策略已阻止发送').description("对大文件的文字提示内容"),
         MinimumTimeInterval: koishi_1.Schema.number().default(600).description("若干秒内不再处理相同链接，防止刷屏").min(1),
         waitTip_Switch: koishi_1.Schema.union([
@@ -53,8 +53,11 @@ exports.Config = koishi_1.Schema.intersect([
         useNumeral: koishi_1.Schema.boolean().default(true).description("使用格式化数字 (如 10000 -> 1万)"),
         showError: koishi_1.Schema.boolean().default(false).description("当链接不正确时提醒发送者"),
         allow_sensitive: koishi_1.Schema.boolean().default(false).description("允许NSFW内容"),
-        proxy: koishi_1.Schema.string().description("代理设置"),
     }).description("高级解析设置"),
+    koishi_1.Schema.object({
+        proxy: koishi_1.Schema.string().description("代理设置"),
+        proxy_settings: koishi_1.Schema.object(Object.fromEntries(core_1.parsers_str.map(parser => [parser, koishi_1.Schema.boolean().default(false)]))),
+    }).description("代理设置"),
     koishi_1.Schema.object({
         onebotReadDir: koishi_1.Schema.string().description('OneBot 实现 (如 NapCat) 所在的容器或环境提供的路径前缀。').default("/app/.config/QQ/NapCat/temp"),
         localDownloadDir: koishi_1.Schema.string().description('与上述路径对应的、Koishi 所在的容器或主机可以访问的路径前缀。').default("/koishi/data/temp"),

package/lib/types.d.ts

@@ -27,6 +27,7 @@ export interface PluginConfig {
     showError: boolean;
     allow_sensitive: boolean;
     proxy: string;
+    proxy_settings: object;
     onebotReadDir: string;
     localDownloadDir: string;
     userAgent: string;

package/lib/utils.d.ts

@@ -8,6 +8,6 @@ import { Logger, Session } from "koishi";
  */
 export declare function numeral(num: number, config: PluginConfig): string;
 export declare function escapeHtml(str: string): string;
-export declare function getFileSize(url: string, proxy: string | undefined, userAgent: string | undefined): Promise<number | null>;
+export declare function getFileSize(url: string, proxy: string | undefined, userAgent: string | undefined, logger: Logger): Promise<number | null>;
 export declare function sendResult_plain(session: Session, config: PluginConfig, result: ParsedInfo, logger: Logger): Promise<void>;
 export declare function sendResult_forward(session: Session, config: PluginConfig, result: ParsedInfo, logger: Logger, mixed_sending?: boolean): Promise<void>;

package/lib/utils.js

@@ -170,7 +170,22 @@ async function downloadAndMapUrl(url, proxy, userAgent, localDownloadDir, onebot
         });
     });
 }
-async function getFileSize(url, proxy, userAgent) {
+async function getFileSize(url, proxy, userAgent, logger) {
+    try {
+        // 先尝试HEAD请求（标准方式）
+        const headSize = await tryHeadRequest(url, proxy, userAgent, logger);
+        if (headSize !== null) {
+            return headSize;
+        }
+        // HEAD失败，尝试GET请求获取部分内容
+        return await tryGetRequestForSize(url, proxy, userAgent, logger);
+    }
+    catch (e) {
+        logger.warn(`获取文件大小失败: ${url}`, e);
+        return null;
+    }
+}
+async function tryHeadRequest(url, proxy, userAgent, logger) {
     return new Promise((resolve) => {
         const u = new url_1.URL(url);
         const agent = getProxyAgent(proxy, url);
@@ -180,7 +195,8 @@ async function getFileSize(url, proxy, userAgent) {
             method: 'HEAD',
             timeout: 10000,
             headers: {
-                'User-Agent': userAgent
+                'User-Agent': userAgent,
+                'Referer': 'https://www.bilibili.com/'
             }
         }, (res) => {
             const len = res.headers['content-length'];
@@ -192,11 +208,61 @@ async function getFileSize(url, proxy, userAgent) {
             }
             req.destroy();
         });
-        req.on('error', () => {
+        req.on('error', (err) => {
+            logger.warn(`HEAD请求失败: ${url}`, err);
+            req.destroy();
+            resolve(null);
+        });
+        req.on('timeout', () => {
+            logger.warn(`HEAD请求超时: ${url}`);
+            req.destroy();
+            resolve(null);
+        });
+    });
+}
+async function tryGetRequestForSize(url, proxy, userAgent, logger) {
+    proxy = undefined;
+    return new Promise((resolve) => {
+        const u = new url_1.URL(url);
+        const agent = getProxyAgent(proxy, url);
+        const getter = u.protocol === 'https:' ? require('https').get : require('http').get;
+        const req = getter(url, {
+            agent,
+            timeout: 15000,
+            headers: {
+                'User-Agent': userAgent,
+                'Range': 'bytes=0-1023' // 只请求前1KB
+            }
+        }, (res) => {
+            const contentRange = res.headers['content-range'];
+            if (contentRange) {
+                // 从 Content-Range 头获取总大小，例如: "bytes 0-1023/12345678"
+                const match = contentRange.match(/\/(\d+)$/);
+                if (match) {
+                    resolve(parseInt(match[1], 10));
+                    req.destroy();
+                    return;
+                }
+            }
+            const len = res.headers['content-length'];
+            if (len && /^\d+$/.test(len)) {
+                resolve(parseInt(len, 10));
+            }
+            else {
+                resolve(null);
+            }
+            // 读取少量数据后关闭连接
+            res.on('data', () => {
+                req.destroy();
+            });
+        });
+        req.on('error', (err) => {
+            logger.warn(`GET请求获取大小失败: ${url}`, err);
             req.destroy();
             resolve(null);
         });
         req.on('timeout', () => {
+            logger.warn(`GET请求获取大小超时: ${url}`);
             req.destroy();
             resolve(null);
         });
@@ -211,10 +277,15 @@ async function sendResult_plain(session, config, result, logger) {
     let mediaCoverUrl = result.coverUrl;
     let mediaVideoUrl = result.videoUrl || null;
     let mediaMainbody = result.mainbody;
+    let proxy = undefined;
+    if (config.proxy_settings[result.platform]) {
+        proxy = config.proxy;
+        logger.info("正在使用代理");
+    }
     // --- 下载封面 ---
     if (result.coverUrl) {
         try {
-            mediaCoverUrl = await downloadAndMapUrl(result.coverUrl, config.proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
+            mediaCoverUrl = await downloadAndMapUrl(result.coverUrl, proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
             if (config.logLevel === 'full')
                 logger.info(`封面已下载: ${mediaCoverUrl}`);
         }
@@ -226,7 +297,7 @@ async function sendResult_plain(session, config, result, logger) {
     // --- 视频：先检查大小 ---
     let videoExceedsLimit = false;
     if (result.videoUrl) {
-        const sizeBytes = await getFileSize(result.videoUrl, config.proxy, config.userAgent);
+        const sizeBytes = await getFileSize(result.videoUrl, proxy, config.userAgent, logger);
         const maxBytes = config.Max_size !== undefined ? config.Max_size * 1024 * 1024 : undefined;
         // 日志用 MB（保留 2 位小数）
         const formatMB = (bytes) => (bytes / (1024 * 1024)).toFixed(2);
@@ -246,7 +317,7 @@ async function sendResult_plain(session, config, result, logger) {
             else {
                 // 大小合规，执行下载
                 try {
-                    mediaVideoUrl = await downloadAndMapUrl(result.videoUrl, config.proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
+                    mediaVideoUrl = await downloadAndMapUrl(result.videoUrl, proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
                     if (config.logLevel === 'full') {
                         logger.info(`视频已下载 (${sizeMB} MB): ${mediaVideoUrl}`);
                     }
@@ -265,7 +336,7 @@ async function sendResult_plain(session, config, result, logger) {
         await Promise.all(imgMatches.map(async (match) => {
             const remoteUrl = match[1];
             try {
-                const localUrl = await downloadAndMapUrl(remoteUrl, config.proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
+                const localUrl = await downloadAndMapUrl(remoteUrl, proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
                 urlMap[remoteUrl] = localUrl;
                 if (config.logLevel === 'full')
                     logger.info(`正文图片已下载: ${localUrl}`);
@@ -326,10 +397,15 @@ async function sendResult_forward(session, config, result, logger, mixed_sending
     let mediaCoverUrl = result.coverUrl;
     let mediaVideoUrl = result.videoUrl || null;
     let mediaMainbody = result.mainbody;
+    let proxy = undefined;
+    if (config.proxy_settings[result.platform]) {
+        proxy = config.proxy;
+        logger.info("正在使用代理");
+    }
     // --- 封面 ---
     if (result.coverUrl) {
         try {
-            mediaCoverUrl = await downloadAndMapUrl(result.coverUrl, config.proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
+            mediaCoverUrl = await downloadAndMapUrl(result.coverUrl, proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
         }
         catch (e) {
             logger.warn('封面下载失败', e);
@@ -339,7 +415,7 @@ async function sendResult_forward(session, config, result, logger, mixed_sending
     // --- 视频大小检查 + 下载 ---
     let videoExceedsLimit = false;
     if (result.videoUrl) {
-        const sizeBytes = await getFileSize(result.videoUrl, config.proxy, config.userAgent);
+        const sizeBytes = await getFileSize(result.videoUrl, proxy, config.userAgent, logger);
         const maxBytes = config.Max_size !== undefined ? config.Max_size * 1024 * 1024 : undefined;
         const formatMB = (bytes) => (bytes / (1024 * 1024)).toFixed(2);
         if (sizeBytes === null) {
@@ -357,7 +433,7 @@ async function sendResult_forward(session, config, result, logger, mixed_sending
             }
             else {
                 try {
-                    mediaVideoUrl = await downloadAndMapUrl(result.videoUrl, config.proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
+                    mediaVideoUrl = await downloadAndMapUrl(result.videoUrl, proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
                     if (config.logLevel === 'full') {
                         logger.info(`视频已下载 (${sizeMB} MB): ${mediaVideoUrl}`);
                     }
@@ -375,7 +451,7 @@ async function sendResult_forward(session, config, result, logger, mixed_sending
         const urlMap = {};
         await Promise.all(imgUrls.map(async (url) => {
             try {
-                urlMap[url] = await downloadAndMapUrl(url, config.proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
+                urlMap[url] = await downloadAndMapUrl(url, proxy, config.userAgent, localDownloadDir, onebotReadDir, logger);
             }
             catch (e) {
                 logger.warn(`正文图片下载失败: ${url}`, e);

package/package.json

@@ -2,7 +2,7 @@
   "name": "koishi-plugin-share-links-analysis",
   "description": "自用插件",
   "license": "MIT",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [

package/lib/parsers/tmp.json

@@ -1,46 +0,0 @@
-{
-  allSameType: true,
-  article: null,
-  combinedMediaUrl: null,
-  communityNote: null,
-  conversationID: '1988209837298377142',
-  date: 'Tue Nov 11 11:38:59 +0000 2025',
-  date_epoch: 1762861139,
-  fetched_on: 1763038701,
-  hasMedia: true,
-  hashtags: [],
-  lang: 'zxx',
-  likes: 80,
-  mediaURLs: [
-    'https://pbs.twimg.com/media/G5eKTetbcAE9wtk.jpg'
-  ],
-  media_extended: [
-    {
-      altText: null,
-      id_str: '1988209827773116417',
-      size: {
-        height: 1840,
-        width: 1840
-      },
-      thumbnail_url: 'https://pbs.twimg.com/media/G5eKTetbcAE9wtk.jpg',
-      type: 'image',
-      url: 'https://pbs.twimg.com/media/G5eKTetbcAE9wtk.jpg'
-    }
-  ],
-  pollData: null,
-  possibly_sensitive: false,
-  qrt: null,
-  qrtURL: null,
-  replies: 5,
-  replyingTo: null,
-  replyingToID: null,
-  retweet: null,
-  retweetURL: null,
-  retweets: 4,
-  text: 'https://t.co/b8AyvpjhAE',
-  tweetID: '1988209837298377142',
-  tweetURL: 'https://twitter.com/moyushuang_/status/1988209837298377142',
-  user_name: '魔芋凉粉',
-  user_profile_image_url: 'https://pbs.twimg.com/profile_images/1830265587676962816/KM46JNnI_normal.jpg',
-  user_screen_name: 'moyushuang_'
-}