import BasicInformation from "./components/BasicInformation";
import {
  queryDatasetByIdUsingGet,
  updateDatasetByIdUsingPut,
} from "../dataset.api";
import { useCallback, useEffect, useState } from "react";
import { Dataset, DatasetType } from "../dataset.model";
import { App, Button, Form, Modal } from "antd";

export default function EditDataset({
  open,
  data,
  onClose,
  onRefresh,
}: {
  open: boolean;
  data: Dataset | null;
  onClose: () => void;
  onRefresh?: (showMessage?: boolean) => void;
}) {
  const [form] = Form.useForm();
  const { message } = App.useApp();

  const [newDataset, setNewDataset] = useState({
    name: "",
    description: "",
    datasetType: DatasetType.TEXT,
    tags: [],
    parentDatasetId: "",
  });
  const fetchDataset = useCallback(async () => {
    if (!open) return;
    // 如果有id，说明是编辑模式
    if (data && data.id) {
      const { data: newData } = await queryDatasetByIdUsingGet(data.id);
      const updatedDataset = {
        ...newData,
        type: newData.type,
        tags: newData.tags.map((tag) => tag.name) || [],
        parentDatasetId: newData.parentDatasetId || "",
      };
      setNewDataset(updatedDataset);
      form.setFieldsValue(updatedDataset);
    }
  }, [data, form, open]);

  useEffect(() => {
    fetchDataset();
  }, [fetchDataset]);

  const handleValuesChange = (_, allValues) => {
    setNewDataset((prev) => ({ ...prev, ...allValues }));
  };

  const handleSubmit = async () => {
    const formValues = await form.validateFields();

    const params = {
      ...formValues,
      files: undefined,
    };
    try {
      await updateDatasetByIdUsingPut(data?.id, params);
      onClose();
      message.success("数据集更新成功");
      onRefresh?.(false);
    } catch (error) {
      console.error(error);
      message.error("数据集更新失败，请重试");
      return;
    }
  };

  return (
    <Modal
      title={`编辑数据集${data?.name}`}
      onCancel={onClose}
      open={open}
      width={600}
      maskClosable={false}
      footer={
        <>
          <Button onClick={onClose}>取消</Button>
          <Button type="primary" onClick={handleSubmit}>
            确定
          </Button>
        </>
      }
    >
      <Form
        form={form}
        initialValues={newDataset}
        onValuesChange={handleValuesChange}
        layout="vertical"
      >
        <BasicInformation
          data={newDataset}
          setData={setNewDataset}
          hidden={["datasetType", "dataSource"]}
        />
      </Form>
    </Modal>
  );
}
