koishi-plugin-meinv 1.0.3 → 1.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-meinv",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "main": "src/index.js",
   "files": [
     "src"

package/src/index.js

@@ -1,5 +1,10 @@
 const { Context } = require('koishi')
 const axios = require('axios')
+const fs = require('fs').promises
+const path = require('path')
+
+const TEMP_DIR = path.resolve(__dirname, '../temp')
+fs.mkdir(TEMP_DIR, { recursive: true }).catch(() => {})
 
 exports.name = 'meinv'
 exports.using = ['logger']
@@ -8,16 +13,19 @@ exports.apply = (ctx) => {
   ctx.command('meinv', '随机获取小姐姐视频')
     .alias('小姐姐')
     .action(async ({ session }) => {
+      // 第一步：先回复用户提示
+      await session.send('请稍等 视频就在来的路上了......\nby.pldduck')
+
       try {
+        // 第二步：请求 API
         const res = await axios.get('https://www.tmini.net/api/meinv?mp4=json', {
-          timeout: 10000,
+          timeout: 15000,
           headers: {
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
             'Referer': 'https://www.tmini.net/'
           }
         })
 
-        // 解析视频链接（优先 data.link，其次 url）
         let videoUrl = ''
         if (res.data?.data?.link) {
           videoUrl = res.data.data.link
@@ -26,23 +34,51 @@ exports.apply = (ctx) => {
         }
 
         if (!videoUrl) {
-          return '❌ 未获取到视频链接，请稍后再试。'
+          return '❌ 未获取到有效视频链接。'
         }
 
-        // 尝试发送文件（QQ/Telegram 可能失败）
-        try {
-          return await session.send({
-            type: 'file',
-            url: videoUrl,
-            name: 'meinv.mp4'
-          })
-        } catch (e) {
-          ctx.logger('meinv').warn('文件发送失败:', e.message)
-          return `📹 视频链接：${videoUrl}`
-        }
+        // 第三步：生成临时文件路径
+        const filename = `meinv_${Date.now()}.mp4`
+        const filePath = path.join(TEMP_DIR, filename)
+
+        // 第四步：尝试下载并保存文件
+        ctx.logger('meinv').info(`开始下载视频: ${videoUrl}`)
+        const response = await axios({
+          method: 'GET',
+          url: videoUrl,
+          responseType: 'arraybuffer',
+          timeout: 30000
+        })
+
+        // 使用 writeFile 代替 createWriteStream
+        await fs.writeFile(filePath, response.data)
+        ctx.logger('meinv').info(`视频下载完成: ${filePath}`)
+
+        // 第五步：发送本地文件
+        await session.send({
+          type: 'file',
+          path: filePath
+        })
+
+        // 第六步：成功后删除临时文件
+        await fs.unlink(filePath).catch(e => {
+          ctx.logger('meinv').warn(`临时文件删除失败: ${e.message}`)
+        })
+
+        return '✅ 视频已送达！'
+
       } catch (e) {
-        ctx.logger('meinv').error('请求失败:', e.message)
-        return '⚠️ 获取视频时出错，请稍后再试。'
+        ctx.logger('meinv').error('插件执行出错:', e)
+
+        // 清理可能残留的文件
+        const tempFiles = await fs.readdir(TEMP_DIR).catch(() => [])
+        for (const file of tempFiles) {
+          if (file.startsWith('meinv_') && Date.now() - parseInt(file.split('_')[1]) > 300000) {
+            await fs.unlink(path.join(TEMP_DIR, file)).catch(() => {})
+          }
+        }
+
+        return '⚠️ 抱歉，视频加载失败，请稍后再试。'
       }
     })
 }