General-AI-Platform-Web/src/pages/Setting/AlgorithmSetting.tsx

import {
  postDeviceGroupGetDeviceGroupById,
  postDeviceGroupGetDeviceGroupTree,
} from '@/services/device/DeviceGroup';

import TreeAndTableList, { ProCardTypeProps } from '@/layouts/treeAndTableList';
import { postProjectGetProjectByGroupId } from '@/services/project/Project';
import { postModelVersionGetModelVersionListByIds } from '@/services/resource/ModelVersion';
import { FormattedMessage } from '@@/exports';
import { CaretRightOutlined, PauseOutlined } from '@ant-design/icons';
import type { ActionType } from '@ant-design/pro-components';
import { PageContainer, ProCard, ProDescriptions, ProList } from '@ant-design/pro-components';
import { useAccess, useIntl } from '@umijs/max';
import { Button, Tag, message } from 'antd';
import Tree, { DataNode } from 'antd/es/tree';
import React, { useEffect, useRef, useState } from 'react';
import ProjectCard from './components/ProjectCard';
import UpdateForm from './components/UpdateForm';

const AlgorithmSetting: React.FC = () => {
  /**
   * @en-US Pop-up window of new window
   * @zh-CN 新建窗口的弹窗
   *  */

  /**
   * @en-US International configuration
   * @zh-CN 国际化配置
   * */
  const access = useAccess();
  const intl = useIntl();
  const projectCardListRef = useRef<any>(null);
  const actionRef = useRef<ActionType>();
  const [updateModalOpen, setUpdateModalOpen] = useState<boolean>(false);
  const [hasInit, setHasInit] = useState<boolean>(false);
  const [nodeTreeData, setNodeTreeData] = React.useState<DataNode[]>([]);
  const [selectNodes, setSelectNodes] = React.useState<any[]>([]);
  const [currentRow, setCurrentRow] = useState<API.DeviceGroup>();
  const [currentCard, setCurrentCard] = useState();

  const [selectedRowsState, setSelectedRows] = useState<API.DeviceGroup[]>([]);
  const [tabs, setTabs] = useState([]);
  const [cardActionProps, setCardActionProps] = useState<'actions' | 'extra'>('extra');
  const [modelVersionData, setModelVersionData] = useState<any[]>([]);
  const [projectData, setProjectData] = useState<Record<string,any>>({});
  const [targetKeys, setTargetKeys] = useState<string[]>([]);
  const [projectConfigId, setProjectConfigId] = useState<number>(0);
  const handleUpdateModal = () => {
    if (updateModalOpen) {
      setUpdateModalOpen(false);
      setProjectConfigId(0);
    } else {
      setUpdateModalOpen(true);
    }
  };

  const [tab, setTab] = useState<string>('');
  // const columns1: ProColumns<API.DeviceGroup>[] = [
  //   {
  //     title: '全选',
  //     dataIndex: 'name',
  //     hideInSearch: true,
  //   },
  // ];

  const columns = [
    // {
    //   title: <FormattedMessage id="device.device_group.table.list.id" defaultMessage="id" />,
    //   dataIndex: 'id',
    //   sorter: true,
    // },

    {
      title: <FormattedMessage id="device.device_group.table.list.name" defaultMessage="$$$" />,
      dataIndex: 'name',
      hideInSearch: true,
    },

    {
      title: <FormattedMessage id="device.device_group.table.list.isEnable" defaultMessage="$$$" />,
      dataIndex: 'isEnable',
      filters: true,
      onFilter: true,
      hideInSearch: true,
      valueType: 'switch',
    },

    {
      title: (
        <FormattedMessage id="device.device_group.table.list.createTime" defaultMessage="$$$" />
      ),
      dataIndex: 'createTime',
      sorter: true,
      hideInSearch: true,
      valueType: 'dateTime',
    },

    {
      title: (
        <FormattedMessage id="device.device_group.table.list.updateTime" defaultMessage="$$$" />
      ),
      dataIndex: 'updateTime',
      sorter: true,
      hideInSearch: true,
      valueType: 'dateTime',
    },

    {
      title: <FormattedMessage id="device.device_group.table.list.code" defaultMessage="$$$" />,
      dataIndex: 'code',
      hideInSearch: true,
    },

    {
      title: <FormattedMessage id="device.device_group.table.list.address" defaultMessage="$$$" />,
      dataIndex: 'address',
      hideInSearch: true,
    },

    // {
    //   title: (
    //     <FormattedMessage id="device.device_group.table.list.telephone" defaultMessage="$$$" />
    //   ),
    //   dataIndex: 'telephone',
    //   hideInSearch: true,
    // },

    // {
    //   title: <FormattedMessage id="device.device_group.table.list.lon" defaultMessage="$$$" />,
    //   dataIndex: 'lon',
    //   hideInSearch: true,
    // },

    // {
    //   title: <FormattedMessage id="device.device_group.table.list.lat" defaultMessage="$$$" />,
    //   dataIndex: 'lat',
    //   hideInSearch: true,
    // },

    // {
    //   title: (
    //     <FormattedMessage id="device.device_group.table.list.managerName" defaultMessage="$$$" />
    //   ),
    //   dataIndex: 'managerName',
    //   hideInSearch: true,
    // },

    // {
    //   title: (
    //     <FormattedMessage id="device.device_group.table.list.managerPhone" defaultMessage="$$$" />
    //   ),
    //   dataIndex: 'managerPhone',
    //   hideInSearch: true,
    // },

    // {
    //   title: <FormattedMessage id="device.device_group.table.list.remark" defaultMessage="$$$" />,
    //   dataIndex: 'remark',
    //   hideInSearch: true,
    // },
  ];

  // 项目信息展示
  async function fetchProjectInfoByGroupId(record: Record<string, any>) {
    const resp = await postDeviceGroupGetDeviceGroupById({ id: record.key });
    console.log('fetchProjectInfoByGroupId_resp', resp.data.deviceGroup);
    setCurrentRow(resp.data.deviceGroup);
  }

  // TODO 左侧图标目前写死，需替换
  // 操作左侧网点数据节点， 项目列表数据变更
  function fetchProjectByGroupId(record: Record<string, any>) {
    // console.log('clickTreeNode', projectCardListRef.current)
    postProjectGetProjectByGroupId({ id: record.key }).then((resp) => {
      let tab_data = (resp?.data?.list || []).map((v: any) => {
        console.log(resp.data.list);
        let model_data = ([...v?.models] || []).map((item: any) => ({
          content: (
            <ProCard
              style={{ backgroundColor: 'rgba(21, 91, 212, 0.03)' }}
              bodyStyle={{ margin: 0, padding: 0 }}
            >
              <div
                style={{
                  display: 'flex',
                  alignItems: 'center',
                  padding: 15,
                  justifyContent: 'space-between',
                }}
              >
                <div>
                  <img src="/images/icons/logo.svg" alt="" style={{ width: '64px' }} />
                </div>
                <Button
                  type="link"
                  size="small"
                  key={item.configId}
                  style={{ padding: 0, fontWeight: 700 }}
                  onClick={() => {
                    setProjectData(item)
                    console.log(item,'setProjectData_item', projectData)
                    setProjectConfigId(item.configId);
                    handleUpdateModal();
                  }}
                >
                  配置参数
                </Button>
              </div>
              <ProjectCard info={item}></ProjectCard>
            </ProCard>
          ),
        }));
        // console.log(88, model_data);
        // setProjectData({...projectData, [v.id]: v})
        return {
          label: `项目：${v.name}`,
          key: v.id,
          children: (
            <ProList<any>
              className="gn"
              ghost={true}
              itemCardProps={{
                ghost: true,
                bodyStyle: { padding: 0, margin: 0 },
              }}
              pagination={{
                defaultPageSize: 6,
                showSizeChanger: false,
              }}
              showActions="hover"
              grid={{ gutter: 8, xs: 1, md: 2, xl: 3, xxl: 4 }}
              metas={{
                type: {},
                content: {},
                actions: {
                  cardActionProps,
                },
              }}
              headerTitle="项目模型列表"
              toolBarRender={(action, rows) => {
                console.log(action, rows, 'toolBarRender');
                // TODO 需要对接接口
                const isProcess: boolean = false;
                return isProcess ? (
                  <Button type="primary" danger>
                    <PauseOutlined style={{ fontSize: 14, lineHeight: 1.5 }} />
                    结束运行
                  </Button>
                ) : (
                  <Button type="primary">
                    <CaretRightOutlined style={{ fontSize: 14, lineHeight: 1.5 }} />
                    开始运行
                  </Button>
                );
              }}
              dataSource={model_data}
            />
          ),
        };
      });
      if (tab_data.length) {
        setTab(tab_data[0].key);
      }
      setTabs(tab_data);
    });
  }

  // TODO 需联调
  // 操作项目列表数据类别切换， 项目列表数据变更
  function changeProjectTab(key: React.SetStateAction<string>) {
    let model_ids = projectData[key]?.inferConfig?.models || [];
    postModelVersionGetModelVersionListByIds({ ids: model_ids }).then((resp) => {
      let version_model_data = (resp?.data?.list || []).map((item: Record<string, any>) => {
        return {
          subTitle: <Tag color="#5BD8A6">经典算法</Tag>,
          avatar: 'https://gw.alipayobjects.com/zos/antfincdn/UCSiy1j6jx/xingzhuang.svg',
          content: (
            <ProCard bodyStyle={{ margin: 0, padding: 0 }}>
              <div
                style={{
                  display: 'flex',
                  alignItems: 'center',
                  padding: 15,
                  justifyContent: 'space-between',
                }}
              >
                <div>
                  <div>{JSON.stringify(item.configId)}</div>
                  <img src="/images/model.png" alt="" style={{ width: '50px' }} />
                </div>
                <Button
                  type="link"
                  size="small"
                  key={item.configId}
                  style={{ padding: 0 }}
                  onClick={() => {
                    setProjectConfigId(item.configId);
                    handleUpdateModal();
                  }}
                >
                  配置
                </Button>
              </div>
              <ProjectCard info={item}></ProjectCard>
            </ProCard>
          ),
        };
      });
      setModelVersionData(version_model_data);
      console.log(99, modelVersionData);
    });
    setTab(key);
  }

  useEffect(() => {
    postDeviceGroupGetDeviceGroupTree()
      .then((resp) => {
        setNodeTreeData(resp.data.tree);
        setHasInit(true);
      })
      .catch(() => {
        message.error(intl.formatMessage({ id: 'common.failure', defaultMessage: '$$$' }));
      });
  }, []);

  return (
    <PageContainer>
      <TreeAndTableList
        leftCard={
          {
            title: '网点选择',
          } as ProCardTypeProps
        }
        leftDom={
          hasInit && (
            <Tree
              checkable={false}
              defaultExpandAll={true}
              selectedKeys={selectNodes}
              treeData={nodeTreeData}
              onSelect={(selectKeys, info) => {
                // TODO_1 此处接口需要重新联调
                console.log(info.node, 'onSelect_info');
                setSelectNodes(selectKeys);
                fetchProjectInfoByGroupId(info?.node);
                fetchProjectByGroupId(info?.node);
                actionRef.current?.reload();
              }}
              checkStrictly={false}
            />
          )
        }
        rightDom={
          <div className="gn">
            <ProCard
              title="网点详细信息"
              headStyle={{borderBottom: '1px solid #E0E0E0', paddingTop: 0,  paddingBottom:12 }}
              style={{ background: 'transparent' , paddingTop: 24 }}
              bodyStyle={{  paddingBottom: 0 }}
              colSpan="80%"
            >
              <ProDescriptions
                column={3}
                dataSource={currentRow}
                columns={columns}
              ></ProDescriptions>
            </ProCard>
            <ProCard
              wrap
              tabs={{
                activeKey: tab,
                items: tabs,
                onChange: (key) => {
                  changeProjectTab(key);
                },
              }}
            ></ProCard>
          </div>
        }
      ></TreeAndTableList>
      {/* <ProCard split="vertical">
        <ProCard title="网点选择" colSpan="24%" bodyStyle={{ padding: 0, margin: 0 }}></ProCard>
        <ProCard title="" headerBordered split={'horizontal'}></ProCard>
      </ProCard> */}
      <UpdateForm
        projectData={projectData as Record<string, any>}
        updateModalOpen={updateModalOpen}
        handleModal={handleUpdateModal}
        groupId={currentRow?.id || 0}
        reload={actionRef.current?.reload}
        projectModelConfigId={projectConfigId}
      />
    </PageContainer>
  );
};

export default AlgorithmSetting;