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
@@ -180,15 +181,40 @@ const deletePrompt = asyncHandler(async (req, res) => {
 // @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, fileData, mimeType } = req.body;
+    
+    logger.info(`正在添加DSL文件: ${name}`);
+    
+    // 验证请求数据
+    if (!name || !fileData) {
+      res.status(400);
+      throw new Error('文件名和文件数据是必需的');
+    }
+    
+    const promptGroup = await PromptGroup.findById(req.params.id);
 
-  const promptGroup = await PromptGroup.findById(req.params.id);
+    if (!promptGroup) {
+      res.status(404);
+      throw new Error('提示词组未找到');
+    }
+    
+    // 验证文件数据格式
+    if (typeof fileData !== 'string') {
+      res.status(400);
+      throw new Error('文件数据必须是字符串');
+    }
+    
+    // 确保文件名以 .yml 结尾
+    let fileName = name;
+    if (!fileName.toLowerCase().endsWith('.yml')) {
+      fileName = fileName.replace(/\.[^/.]+$/, '') + '.yml';
+    }
 
-  if (promptGroup) {
     const newDslFile = {
-      name,
+      name: fileName,
       fileData,
-      mimeType: mimeType || 'application/octet-stream',
+      mimeType: mimeType || 'application/x-yaml',
       uploadedAt: new Date(),
     };
 
@@ -196,10 +222,67 @@ 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, fileData, mimeType } = req.body;
+    
+    logger.info(`正在更新DSL文件: ${req.params.fileId}`);
+    
+    const promptGroup = await PromptGroup.findById(req.params.id);
+
+    if (!promptGroup) {
+      res.status(404);
+      throw new Error('提示词组未找到');
+    }
+    
+    const fileIndex = promptGroup.dslFiles.findIndex(
+      (f) => f._id.toString() === req.params.fileId
+    );
+    
+    if (fileIndex === -1) {
+      res.status(404);
+      throw new Error('DSL文件未找到');
+    }
+    
+    // 确保文件名以 .yml 结尾
+    let fileName = name || promptGroup.dslFiles[fileIndex].name;
+    if (!fileName.toLowerCase().endsWith('.yml')) {
+      fileName = fileName.replace(/\.[^/.]+$/, '') + '.yml';
+    }
+
+    // 更新文件属性
+    if (name) promptGroup.dslFiles[fileIndex].name = fileName;
+    if (fileData) promptGroup.dslFiles[fileIndex].fileData = fileData;
+    if (mimeType) promptGroup.dslFiles[fileIndex].mimeType = mimeType;
+    
+    // 更新时间
+    promptGroup.updatedAt = new Date();
+
+    const updatedPromptGroup = await promptGroup.save();
+    logger.info(`DSL文件更新成功: ${fileName}`);
+    
+    res.json(updatedPromptGroup.dslFiles[fileIndex]);
+  } catch (error) {
+    logger.error(`更新DSL文件时出错: ${error.message}`);
+    if (!res.statusCode || res.statusCode === 200) {
+      res.status(500);
+    }
+    throw error;
   }
 });
 
@@ -242,5 +325,6 @@ module.exports = {
   updatePrompt,
   deletePrompt,
   addDslFileToGroup,
+  updateDslFile,
   deleteDslFile,
 };

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;