Browse Source

🐞 fix: 修复总结文件接口中路径错误的问题

Pchen. 2 months ago
parent
commit
02457d4678
1 changed files with 19 additions and 2 deletions
  1. 19 2
      apis/AI/GetFileSummary.js

+ 19 - 2
apis/AI/GetFileSummary.js

@@ -2,6 +2,7 @@ const API = require("../../lib/API")
 const db = require("../../plugin/DataBase/db")
 const AccessControl = require("../../lib/AccessControl")
 const { BaseStdResponse } = require("../../BaseStdResponse")
+const path = require('path')
 
 class GetFileSummary extends API {
     constructor() {
@@ -24,8 +25,24 @@ class GetFileSummary extends API {
                 ...BaseStdResponse.ACCESS_DENIED
             })
 
-        let sql = 'SELECT create_time, start_time, end_time, result, repo_hash FROM file_summary_tasks WHERE create_user = ? AND repo_id = ? AND repo_hash = ? AND filepath = ?'
-        let rows = await db.query(sql, [uuid, id, hash, filePath])
+        let sql = 'SELECT state, path FROM repos WHERE create_user = ? AND id = ?'
+        let rows = await db.query(sql, [uuid, id])
+        if (!rows || rows.length === 0)
+            return res.json({
+                ...BaseStdResponse.ERR,
+                msg: '未找到仓库'
+            })
+
+        if (rows[0].state !== 1 || !rows[0].path)
+            return res.json({
+                ...BaseStdResponse.ERR,
+                msg: '仓库未成功克隆!'
+            })
+
+        filePath = path.join(rows[0].path, filePath)
+
+        sql = 'SELECT create_time, start_time, end_time, result, repo_hash FROM file_summary_tasks WHERE create_user = ? AND repo_id = ? AND repo_hash = ? AND filepath = ?'
+        rows = await db.query(sql, [uuid, id, hash, filePath])
 
         if (!rows || rows.length === 0)
             return res.json({