first-update

easy-dataset-main/app/api/projects/[projectId]/custom-prompts/route.js

@@ -0,0 +1,105 @@
+import { NextResponse } from 'next/server';
+import {
+  getCustomPrompts,
+  getCustomPrompt,
+  saveCustomPrompt,
+  deleteCustomPrompt,
+  batchSaveCustomPrompts,
+  toggleCustomPrompt,
+  getPromptTemplates
+} from '@/lib/db/custom-prompts';
+
+// 获取项目的自定义提示词
+export async function GET(request, { params }) {
+  try {
+    const { projectId } = params;
+    const { searchParams } = new URL(request.url);
+    const promptType = searchParams.get('promptType');
+    const language = searchParams.get('language');
+
+    if (!projectId) {
+      return NextResponse.json({ error: 'Project ID is required' }, { status: 400 });
+    }
+
+    const customPrompts = await getCustomPrompts(projectId, promptType, language);
+    const templates = await getPromptTemplates();
+
+    return NextResponse.json({
+      success: true,
+      customPrompts,
+      templates
+    });
+  } catch (error) {
+    console.error('获取自定义提示词失败:', error);
+    return NextResponse.json({ error: error.message }, { status: 500 });
+  }
+}
+
+// 保存自定义提示词
+export async function POST(request, { params }) {
+  try {
+    const { projectId } = params;
+    const body = await request.json();
+
+    if (!projectId) {
+      return NextResponse.json({ error: 'Project ID is required' }, { status: 400 });
+    }
+
+    // 批量保存
+    if (body.prompts && Array.isArray(body.prompts)) {
+      const results = await batchSaveCustomPrompts(projectId, body.prompts);
+      return NextResponse.json({
+        success: true,
+        results
+      });
+    }
+
+    // 单个保存
+    const { promptType, promptKey, language, content } = body;
+    if (!promptType || !promptKey || !language || content === undefined) {
+      return NextResponse.json(
+        {
+          error: 'promptType, promptKey, language and content are required'
+        },
+        { status: 400 }
+      );
+    }
+
+    const result = await saveCustomPrompt(projectId, promptType, promptKey, language, content);
+    return NextResponse.json({
+      success: true,
+      result
+    });
+  } catch (error) {
+    console.error('保存自定义提示词失败:', error);
+    return NextResponse.json({ error: error.message }, { status: 500 });
+  }
+}
+
+// 删除自定义提示词
+export async function DELETE(request, { params }) {
+  try {
+    const { projectId } = params;
+    const { searchParams } = new URL(request.url);
+    const promptType = searchParams.get('promptType');
+    const promptKey = searchParams.get('promptKey');
+    const language = searchParams.get('language');
+
+    if (!projectId || !promptType || !promptKey || !language) {
+      return NextResponse.json(
+        {
+          error: 'projectId, promptType, promptKey and language are required'
+        },
+        { status: 400 }
+      );
+    }
+
+    const success = await deleteCustomPrompt(projectId, promptType, promptKey, language);
+    return NextResponse.json({
+      success
+    });
+  } catch (error) {
+    console.error('删除自定义提示词失败:', error);
+    return NextResponse.json({ error: error.message }, { status: 500 });
+  }
+}