Upload 18 files

controllers/promptGroupController.js

@@ -1,5 +1,6 @@
 const asyncHandler = require('express-async-handler');
 const PromptGroup = require('../models/PromptGroup');
+const logger = require('../utils/logger');
 
 // @desc    获取所有提示词组
 // @route   GET /api/prompt-groups
@@ -176,19 +177,60 @@ const deletePrompt = asyncHandler(async (req, res) => {
   }
 });
 
-// @desc    添加DSL文件到提示词组
+// @desc    添加DSL文件到提示词组 - 修改为支持文本内容
 // @route   POST /api/prompt-groups/:id/dsl-files
 // @access  Private
 const addDslFileToGroup = asyncHandler(async (req, res) => {
-  const { name, fileData, mimeType } = req.body;
+  try {
+    const { name, content, fileData, mimeType } = req.body;
+    logger.info(`正在添加DSL文件: ${name}`);
+    
+    // 验证请求数据
+    if (!name) {
+      res.status(400);
+      throw new Error('文件名是必需的');
+    }
 
-  const promptGroup = await PromptGroup.findById(req.params.id);
+    // 确保至少提供了文本内容或文件数据中的一个
+    if (!content && !fileData) {
+      res.status(400);
+      throw new Error('请提供YAML内容或上传文件');
+    }
+    
+    const promptGroup = await PromptGroup.findById(req.params.id);
+
+    if (!promptGroup) {
+      res.status(404);
+      throw new Error('提示词组未找到');
+    }
+    
+    // 确保文件名以 .yml 结尾
+    let fileName = name;
+    if (!fileName.toLowerCase().endsWith('.yml')) {
+      fileName = fileName.replace(/\.[^/.]+$/, '') + '.yml';
+    }
+    
+    // 处理内容: 优先使用提供的内容，如果没有则尝试解码文件数据
+    let yamlContent = content;
+    
+    // 如果提供了fileData但没有提供content，尝试将fileData转换为文本
+    if (!yamlContent && fileData) {
+      try {
+        // 尝试将Base64解码为文本
+        const buffer = Buffer.from(fileData, 'base64');
+        yamlContent = buffer.toString('utf-8');
+        logger.info(`成功从文件数据提取YAML内容`);
+      } catch (error) {
+        logger.error(`从文件数据提取内容失败: ${error.message}`);
+        res.status(400);
+        throw new Error('无法从上传的文件中提取内容');
+      }
+    }
 
-  if (promptGroup) {
     const newDslFile = {
-      name,
-      fileData,
-      mimeType: mimeType || 'application/octet-stream',
+      name: fileName,
+      content: yamlContent,
+      mimeType: mimeType || 'application/x-yaml',
       uploadedAt: new Date(),
     };
 
@@ -196,10 +238,74 @@ const addDslFileToGroup = asyncHandler(async (req, res) => {
     promptGroup.updatedAt = new Date();
 
     const updatedPromptGroup = await promptGroup.save();
+    logger.info(`DSL文件添加成功: ${fileName}`);
+    
     res.status(201).json(updatedPromptGroup.dslFiles[updatedPromptGroup.dslFiles.length - 1]);
-  } else {
-    res.status(404);
-    throw new Error('提示词组未找到');
+  } catch (error) {
+    logger.error(`添加DSL文件时出错: ${error.message}`);
+    if (!res.statusCode || res.statusCode === 200) {
+      res.status(500);
+    }
+    throw error;
+  }
+});
+
+// @desc    更新DSL文件
+// @route   PUT /api/prompt-groups/:id/dsl-files/:fileId
+// @access  Private
+const updateDslFile = asyncHandler(async (req, res) => {
+  try {
+    const { name, content } = req.body;
+    
+    if (!name && !content) {
+      res.status(400);
+      throw new Error('请提供要更新的文件名或内容');
+    }
+    
+    const promptGroup = await PromptGroup.findById(req.params.id);
+
+    if (!promptGroup) {
+      res.status(404);
+      throw new Error('提示词组未找到');
+    }
+    
+    const fileIndex = promptGroup.dslFiles.findIndex(
+      file => file._id.toString() === req.params.fileId
+    );
+    
+    if (fileIndex === -1) {
+      res.status(404);
+      throw new Error('DSL文件未找到');
+    }
+    
+    // 更新文件名 - 确保以.yml结尾
+    if (name) {
+      let updatedName = name;
+      if (!updatedName.toLowerCase().endsWith('.yml')) {
+        updatedName = updatedName.replace(/\.[^/.]+$/, '') + '.yml';
+      }
+      promptGroup.dslFiles[fileIndex].name = updatedName;
+    }
+    
+    // 更新内容
+    if (content) {
+      promptGroup.dslFiles[fileIndex].content = content;
+    }
+    
+    // 更新时间戳
+    promptGroup.dslFiles[fileIndex].uploadedAt = new Date();
+    promptGroup.updatedAt = new Date();
+    
+    const updatedPromptGroup = await promptGroup.save();
+    logger.info(`DSL文件更新成功: ${promptGroup.dslFiles[fileIndex].name}`);
+    
+    res.status(200).json(updatedPromptGroup.dslFiles[fileIndex]);
+  } catch (error) {
+    logger.error(`更新DSL文件时出错: ${error.message}`);
+    if (!res.statusCode || res.statusCode === 200) {
+      res.status(500);
+    }
+    throw error;
   }
 });
 
@@ -242,5 +348,6 @@ module.exports = {
   updatePrompt,
   deletePrompt,
   addDslFileToGroup,
+  updateDslFile,
   deleteDslFile,
 };

models/PromptGroup.js

@@ -24,7 +24,7 @@ const promptSchema = new mongoose.Schema(
   }
 );
 
-// DSL 文件 Schema
+// DSL 文件 Schema - 修改以存储文本内容
 const dslFileSchema = new mongoose.Schema(
   {
     name: {
@@ -32,13 +32,13 @@ const dslFileSchema = new mongoose.Schema(
       required: [true, '请输入文件名称'],
       trim: true,
     },
-    fileData: {
+    content: {  // 添加: 存储 YAML 文本内容
       type: String,
-      required: [true, '请提供文件数据'],
+      required: [true, '请提供 YAML 内容'],
     },
     mimeType: {
       type: String,
-      default: 'application/octet-stream',
+      default: 'application/x-yaml',
     },
     uploadedAt: {
       type: Date,

routes/promptGroupRoutes.js

@@ -10,6 +10,7 @@ const {
   updatePrompt,
   deletePrompt,
   addDslFileToGroup,
+  updateDslFile,
   deleteDslFile,
 } = require('../controllers/promptGroupController');
 const { protect } = require('../middleware/auth');
@@ -37,6 +38,7 @@ router.route('/:id/dsl-files')
   .post(protect, addDslFileToGroup);
 
 router.route('/:id/dsl-files/:fileId')
+  .put(protect, updateDslFile)
   .delete(protect, deleteDslFile);
 
 module.exports = router;