first-update

easy-dataset-main/app/projects/[projectId]/questions/hooks/useQuestionTemplates.js

@@ -0,0 +1,122 @@
+import { useState, useEffect } from 'react';
+import axios from 'axios';
+import { toast } from 'sonner';
+import { useTranslation } from 'react-i18next';
+
+/**
+ * 问题模板管理 Hook
+ * @param {string} projectId - 项目ID
+ * @param {string} sourceType - 数据源类型: 'image' | 'text' | null (null表示获取所有)
+ */
+export function useQuestionTemplates(projectId, sourceType = null) {
+  const { t } = useTranslation();
+  const [templates, setTemplates] = useState([]);
+  const [loading, setLoading] = useState(false);
+
+  // 获取模板列表
+  const fetchTemplates = async () => {
+    try {
+      setLoading(true);
+      const params = sourceType ? `?sourceType=${sourceType}` : '';
+      const response = await axios.get(`/api/projects/${projectId}/questions/templates${params}`);
+      if (response.data.success) {
+        setTemplates(response.data.templates);
+      }
+    } catch (error) {
+      console.error('Failed to fetch templates:', error);
+      toast.error(t('questions.fetchTemplatesFailed'));
+    } finally {
+      setLoading(false);
+    }
+  };
+
+  // 创建模板
+  const createTemplate = async data => {
+    try {
+      const response = await axios.post(`/api/projects/${projectId}/questions/templates`, data);
+      if (response.data.success) {
+        const { template, generation } = response.data;
+
+        // 显示模板创建成功消息
+        toast.success(t('questions.createTemplateSuccess'));
+
+        // 如果有自动生成结果，显示相应消息
+        if (generation) {
+          if (generation.success) {
+            if (generation.successCount > 0) {
+              toast.success(
+                t('questions.template.autoGenerateSuccess', {
+                  count: generation.successCount
+                })
+              );
+            }
+            if (generation.failCount > 0) {
+              toast.warning(
+                t('questions.template.autoGeneratePartialFail', {
+                  success: generation.successCount,
+                  fail: generation.failCount
+                })
+              );
+            }
+          } else {
+            toast.error(generation.message || t('questions.template.autoGenerateFailed'));
+          }
+        }
+
+        fetchTemplates();
+        return template;
+      }
+    } catch (error) {
+      console.error('Failed to create template:', error);
+      toast.error(t('questions.createTemplateFailed'));
+      throw error;
+    }
+  };
+
+  // 更新模板
+  const updateTemplate = async (templateId, data) => {
+    try {
+      const response = await axios.put(`/api/projects/${projectId}/questions/templates/${templateId}`, data);
+      if (response.data.success) {
+        toast.success(t('questions.updateTemplateSuccess'));
+        fetchTemplates();
+        return response.data.template;
+      }
+    } catch (error) {
+      console.error('Failed to update template:', error);
+      toast.error(t('questions.updateTemplateFailed'));
+      throw error;
+    }
+  };
+
+  // 删除模板
+  const deleteTemplate = async templateId => {
+    try {
+      const response = await axios.delete(`/api/projects/${projectId}/questions/templates/${templateId}`);
+      if (response.data.success) {
+        toast.success(t('questions.deleteTemplateSuccess'));
+        fetchTemplates();
+      }
+    } catch (error) {
+      console.error('Failed to delete template:', error);
+      toast.error(t('questions.deleteTemplateFailed'));
+      throw error;
+    }
+  };
+
+  // 初始加载
+  useEffect(() => {
+    if (projectId) {
+      fetchTemplates();
+    }
+  }, [projectId, sourceType]);
+
+  return {
+    templates,
+    loading,
+    createTemplate,
+    updateTemplate,
+    deleteTemplate,
+    refetch: fetchTemplates
+  };
+}