PingChuan
93 lines
2.5 KiB
TypeScript
93 lines
2.5 KiB
TypeScript
import { useState, useEffect, useCallback } from 'react';
|
|
import { message } from 'antd';
|
|
import type { FormInstance } from 'antd';
|
|
import type { Dataset } from '~/api/dify-dataset/type/datasetTypes';
|
|
import { updateDatasetName } from '~/api/dify-dataset/api/datasetApi';
|
|
|
|
/**
|
|
* 知识库设置状态管理 Hook
|
|
*/
|
|
export function useDatasetSettings(
|
|
dataset: Dataset | null,
|
|
form: FormInstance,
|
|
onDatasetUpdated: (dataset: Dataset) => void
|
|
) {
|
|
const [saving, setSaving] = useState(false);
|
|
const [hasChanges, setHasChanges] = useState(false);
|
|
|
|
// 初始化表单数据
|
|
useEffect(() => {
|
|
if (dataset) {
|
|
form.setFieldsValue({
|
|
name: dataset.name,
|
|
description: dataset.description || '',
|
|
});
|
|
setHasChanges(false);
|
|
}
|
|
}, [dataset, form]);
|
|
|
|
/**
|
|
* 处理表单值变化
|
|
*/
|
|
const handleValuesChange = useCallback(() => {
|
|
const values = form.getFieldsValue();
|
|
const changed =
|
|
values.name !== dataset?.name ||
|
|
values.description !== (dataset?.description || '');
|
|
setHasChanges(changed);
|
|
}, [form, dataset]);
|
|
|
|
/**
|
|
* 保存设置
|
|
*/
|
|
const handleSave = useCallback(async () => {
|
|
if (!dataset) {
|
|
message.error('知识库不存在');
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const values = await form.validateFields();
|
|
setSaving(true);
|
|
|
|
// 目前只支持修改名称
|
|
const updatedDataset = await updateDatasetName(dataset.id, values.name);
|
|
|
|
message.success('保存成功');
|
|
onDatasetUpdated(updatedDataset);
|
|
setHasChanges(false);
|
|
} catch (err: any) {
|
|
console.error('保存设置失败:', err);
|
|
message.error(err.message || '保存失败');
|
|
} finally {
|
|
setSaving(false);
|
|
}
|
|
}, [dataset, form, onDatasetUpdated]);
|
|
|
|
/**
|
|
* 重置表单
|
|
*/
|
|
const handleReset = useCallback(() => {
|
|
if (dataset) {
|
|
form.setFieldsValue({
|
|
name: dataset.name,
|
|
description: dataset.description || '',
|
|
});
|
|
setHasChanges(false);
|
|
}
|
|
}, [dataset, form]);
|
|
|
|
return {
|
|
// 状态
|
|
saving,
|
|
hasChanges,
|
|
|
|
// 方法
|
|
handleValuesChange,
|
|
handleSave,
|
|
handleReset,
|
|
};
|
|
}
|
|
|
|
export type UseDatasetSettingsReturn = ReturnType<typeof useDatasetSettings>;
|