@cnbcool/cnb-cli 1.1.0 → 1.2.0

package/skills/cnb-api/scripts/core/utils/upload.js

@@ -6,7 +6,6 @@ Object.defineProperty(exports, "__esModule", {
 exports.handleUpload = handleUpload;
 var _fs = _interopRequireDefault(require("fs"));
 var _path = _interopRequireDefault(require("path"));
-var _mimeTypes = require("mime-types");
 function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
 /**
  * 完整上传流程
@@ -16,19 +15,105 @@ function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e
  * 4. 返回最终结果
  */
 
+/** 常见文件扩展名 → MIME 类型映射（仅用于上传场景，不引入外部依赖） */
+const MIME_MAP = {
+  '.txt': 'text/plain',
+  '.html': 'text/html',
+  '.htm': 'text/html',
+  '.css': 'text/css',
+  '.csv': 'text/csv',
+  '.js': 'application/javascript',
+  '.mjs': 'application/javascript',
+  '.json': 'application/json',
+  '.xml': 'application/xml',
+  '.pdf': 'application/pdf',
+  '.zip': 'application/zip',
+  '.gz': 'application/gzip',
+  '.gzip': 'application/gzip',
+  '.tar': 'application/x-tar',
+  '.7z': 'application/x-7z-compressed',
+  '.rar': 'application/vnd.rar',
+  '.doc': 'application/msword',
+  '.docx': 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+  '.xls': 'application/vnd.ms-excel',
+  '.xlsx': 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+  '.ppt': 'application/vnd.ms-powerpoint',
+  '.pptx': 'application/vnd.openxmlformats-officedocument.presentationml.presentation',
+  '.png': 'image/png',
+  '.jpg': 'image/jpeg',
+  '.jpeg': 'image/jpeg',
+  '.gif': 'image/gif',
+  '.svg': 'image/svg+xml',
+  '.webp': 'image/webp',
+  '.ico': 'image/x-icon',
+  '.bmp': 'image/bmp',
+  '.mp3': 'audio/mpeg',
+  '.wav': 'audio/wav',
+  '.ogg': 'audio/ogg',
+  '.mp4': 'video/mp4',
+  '.webm': 'video/webm',
+  '.avi': 'video/x-msvideo',
+  '.mov': 'video/quicktime',
+  '.woff': 'font/woff',
+  '.woff2': 'font/woff2',
+  '.ttf': 'font/ttf',
+  '.otf': 'font/otf',
+  '.md': 'text/markdown',
+  '.yaml': 'text/yaml',
+  '.yml': 'text/yaml',
+  '.sh': 'application/x-sh',
+  '.py': 'text/x-python',
+  '.rb': 'text/x-ruby',
+  '.go': 'text/x-go',
+  '.rs': 'text/x-rust',
+  '.ts': 'text/typescript',
+  '.tsx': 'text/typescript',
+  '.jsx': 'text/jsx',
+  '.java': 'text/x-java-source',
+  '.c': 'text/x-c',
+  '.h': 'text/x-c',
+  '.cpp': 'text/x-c++src',
+  '.hpp': 'text/x-c++hdr',
+  '.log': 'text/plain',
+  '.patch': 'text/x-diff',
+  '.diff': 'text/x-diff'
+};
+function mimeLookup(filePath) {
+  const ext = _path.default.extname(filePath).toLowerCase();
+  return MIME_MAP[ext] || 'application/octet-stream';
+}
+
 // 上传文件大小上限：100MB
 const MAX_FILE_SIZE = 100 * 1024 * 1024;
 
+/** 上传 API 的 path 参数 */
+
+/** Issue 上传请求体 */
+
+/** Pull 上传请求体 */
+
+/** 上传 API 函数签名 */
+
+/** 上传 API 响应 */
+
+/** 上传失败时的 data */
+
+/** Issue 上传成功时的 data */
+
+/** Pull 上传成功时的 data */
+
+/** handleUpload 的返回值 */
+
 /**
  * 处理完整上传流程
  * @param shortcut 解析后的快捷命令
  * @param filePath 本地文件路径
  * @param toolFunction 原始上传 API 函数（获取 upload_url）
- * @param toolsParam API 调用的 path 参数（repo 或 {repo, number}）
+ * @param pathParams API 调用的 path 参数（repo 或 {repo, number}）
  */
-async function handleUpload(shortcut, filePath, toolFunction, toolsParam) {
+async function handleUpload(shortcut, filePath, toolFunction, pathParams) {
   // 校验 file 参数
-  if (!filePath || typeof filePath !== 'string') {
+  if (!filePath) {
     return {
       status: 400,
       data: {
@@ -73,7 +158,7 @@ async function handleUpload(shortcut, filePath, toolFunction, toolsParam) {
       }
     };
   }
-  const contentType = (0, _mimeTypes.lookup)(filePath) || 'application/octet-stream';
+  const contentType = mimeLookup(filePath);
 
   // 2. 调用上传 API 获取 upload_url
   const isIssueUpload = shortcut.module === 'issues';
@@ -85,10 +170,10 @@ async function handleUpload(shortcut, filePath, toolFunction, toolsParam) {
     name: fileName,
     size: fileSize
   };
-  const uploadResponse = await toolFunction(toolsParam, requestBody);
+  const uploadResponse = await toolFunction(pathParams, requestBody);
 
-  // 提取 upload_url（兼容标准响应和裸响应）
-  const responseData = uploadResponse?.data ?? uploadResponse;
+  // 提取 upload_url
+  const responseData = uploadResponse?.data;
   const uploadUrl = responseData?.upload_url;
   if (!uploadUrl) {
     return uploadResponse; // 获取 URL 失败，直接返回原始错误
@@ -100,14 +185,21 @@ async function handleUpload(shortcut, filePath, toolFunction, toolsParam) {
     'Content-Type': contentType,
     'Content-Length': String(fileSize)
   };
-  const putResponse = await fetch(uploadUrl, {
-    method: 'PUT',
-    headers: putHeaders,
-    body: fileStream,
-    // @ts-ignore duplex required for streaming body in Node.js fetch
-    duplex: 'half'
-  });
+  let putResponse;
+  try {
+    putResponse = await fetch(uploadUrl, {
+      method: 'PUT',
+      headers: putHeaders,
+      body: fileStream,
+      // @ts-ignore duplex required for streaming body in Node.js fetch
+      duplex: 'half'
+    });
+  } catch (e) {
+    fileStream.destroy();
+    throw e;
+  }
   if (!putResponse.ok) {
+    fileStream.destroy();
     const errText = await putResponse.text().catch(() => '');
     return {
       status: putResponse.status,

package/skills/cnb-api/scripts/modules/knowledge-base/search-npc.js

@@ -16,7 +16,7 @@ function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e
  * ## AUTHOR: bapelin                                                     ##
  * ## SOURCE: https://cnb.cool/cnb/skills/cnb-api-generate   ##
  * -------------------------------------------------------------------------
- * @Source /npc/search
+ * @Source /search/npc
  */
 
 /**
@@ -41,7 +41,7 @@ function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e
 * @tags KnowledgeBase
 * @name searchNpc
 * @summary 全局语义搜索 NPC 角色
-* @request get:/npc/search
+* @request get:/search/npc
 
 ----------------------------------
 * @param {SearchNpcParams} arg0 - searchNpc request params
@@ -58,13 +58,13 @@ async function _default({
     ...axiosConfig,
     _next_req: req,
     options: options,
-    url: "/npc/search",
-    _apiTag: "/npc/search",
+    url: "/search/npc",
+    _apiTag: "/search/npc",
     method: "get",
     params: query,
     _originParams: {
       method: "get",
-      _apiTag: "/npc/search",
+      _apiTag: "/search/npc",
       query: query
     }
   });