115 lines
3.4 KiB
JavaScript
115 lines
3.4 KiB
JavaScript
|
'use client';
|
|||
|
|
|
|||
|
|
import { useTranslation } from 'react-i18next';
|
|||
|
|
import { toast } from 'sonner';
|
|||
|
|
import axios from 'axios';
|
|||
|
|
|
|||
|
|
const useQuestionExport = projectId => {
|
|||
|
|
const { t } = useTranslation();
|
|||
|
|
|
|||
|
|
// 导出问题集
|
|||
|
|
const exportQuestions = async exportOptions => {
|
|||
|
|
try {
|
|||
|
|
const apiUrl = `/api/projects/${projectId}/questions/export`;
|
|||
|
|
const requestBody = {
|
|||
|
|
format: exportOptions.format || 'json'
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 如果有选中的问题 ID，传递 ID 列表
|
|||
|
|
if (exportOptions.selectedIds && exportOptions.selectedIds.length > 0) {
|
|||
|
|
requestBody.selectedIds = exportOptions.selectedIds;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果有筛选条件，传递筛选参数
|
|||
|
|
if (exportOptions.filters) {
|
|||
|
|
requestBody.filters = exportOptions.filters;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const response = await axios.post(apiUrl, requestBody);
|
|||
|
|
const questions = response.data;
|
|||
|
|
|
|||
|
|
// 处理和下载数据
|
|||
|
|
await processAndDownloadData(questions, exportOptions);
|
|||
|
|
|
|||
|
|
toast.success(t('questions.exportSuccess'));
|
|||
|
|
return true;
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('Export failed:', error);
|
|||
|
|
toast.error(error.message || t('questions.exportFailed'));
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 处理和下载数据的通用函数
|
|||
|
|
const processAndDownloadData = async (data, exportOptions) => {
|
|||
|
|
const format = exportOptions.format || 'json';
|
|||
|
|
let content;
|
|||
|
|
let filename;
|
|||
|
|
let mimeType;
|
|||
|
|
|
|||
|
|
const timestamp = new Date().toISOString().split('T')[0];
|
|||
|
|
|
|||
|
|
switch (format) {
|
|||
|
|
case 'json':
|
|||
|
|
content = JSON.stringify(data, null, 2);
|
|||
|
|
filename = `questions-${projectId}-${timestamp}.json`;
|
|||
|
|
mimeType = 'application/json';
|
|||
|
|
break;
|
|||
|
|
|
|||
|
|
case 'jsonl':
|
|||
|
|
content = data.map(item => JSON.stringify(item)).join('\n');
|
|||
|
|
filename = `questions-${projectId}-${timestamp}.jsonl`;
|
|||
|
|
mimeType = 'application/jsonl';
|
|||
|
|
break;
|
|||
|
|
|
|||
|
|
case 'txt':
|
|||
|
|
content = data.map(item => item.question).join('\n\n');
|
|||
|
|
filename = `questions-${projectId}-${timestamp}.txt`;
|
|||
|
|
mimeType = 'text/plain';
|
|||
|
|
break;
|
|||
|
|
|
|||
|
|
case 'csv':
|
|||
|
|
// CSV 格式
|
|||
|
|
const headers = Object.keys(data[0] || {});
|
|||
|
|
const csvRows = [headers.join(',')];
|
|||
|
|
data.forEach(item => {
|
|||
|
|
const values = headers.map(header => {
|
|||
|
|
const value = item[header] || '';
|
|||
|
|
// 处理包含逗号或引号的值
|
|||
|
|
if (typeof value === 'string' && (value.includes(',') || value.includes('"') || value.includes('\n'))) {
|
|||
|
|
return `"${value.replace(/"/g, '""')}"`;
|
|||
|
|
}
|
|||
|
|
return value;
|
|||
|
|
});
|
|||
|
|
csvRows.push(values.join(','));
|
|||
|
|
});
|
|||
|
|
content = csvRows.join('\n');
|
|||
|
|
filename = `questions-${projectId}-${timestamp}.csv`;
|
|||
|
|
mimeType = 'text/csv';
|
|||
|
|
break;
|
|||
|
|
|
|||
|
|
default:
|
|||
|
|
content = JSON.stringify(data, null, 2);
|
|||
|
|
filename = `questions-${projectId}-${timestamp}.json`;
|
|||
|
|
mimeType = 'application/json';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建下载链接
|
|||
|
|
const blob = new Blob([content], { type: mimeType });
|
|||
|
|
const url = URL.createObjectURL(blob);
|
|||
|
|
const link = document.createElement('a');
|
|||
|
|
link.href = url;
|
|||
|
|
link.download = filename;
|
|||
|
|
document.body.appendChild(link);
|
|||
|
|
link.click();
|
|||
|
|
document.body.removeChild(link);
|
|||
|
|
URL.revokeObjectURL(url);
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
exportQuestions
|
|||
|
|
};
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
export default useQuestionExport;
|