koishi-plugin-meinv 1.0.2 → 1.0.4

package/package.json

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

package/src/index.js

@@ -1,5 +1,11 @@
 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,23 +14,80 @@ 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,
-          headers: { 'User-Agent': 'Koishi-Bot' }
+          timeout: 15000,
+          headers: {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0 Safari/537.36',
+            'Referer': 'https://www.tmini.net/'
+          }
         })
-        
-        if (res.data?.status === 'success' && res.data.data?.link) {
-          return session.send({
-            type: 'file',
-            url: res.data.data.link,
-            name: 'meinv.mp4'
-          })
+
+        // 第三步：解析视频链接
+        let videoUrl = ''
+        if (res.data?.data?.link) {
+          videoUrl = res.data.data.link
+        } else if (res.data?.url) {
+          videoUrl = res.data.url
         }
-        return '❌ 获取失败，请稍后再试。'
+
+        if (!videoUrl) {
+          return '❌ 未获取到有效视频链接。'
+        }
+
+        // 第四步：生成临时文件路径
+        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: 'stream',
+          timeout: 30000
+        })
+
+        const writer = fs.createWriteStream(filePath)
+        response.data.pipe(writer)
+
+        // 等待下载完成
+        await new Promise((resolve, reject) => {
+          writer.on('finish', resolve)
+          writer.on('error', reject)
+        })
+
+        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)
-        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 '⚠️ 抱歉，视频加载失败，请稍后再试。'
       }
     })
 }