123 lines
3.6 KiB
JavaScript
123 lines
3.6 KiB
JavaScript
|
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
|
||
|
|
};
|
||
|
|
}
|