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

import {
  postDeviceGroupGetDeviceGroupList,
  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 type { ActionType } from '@ant-design/pro-components';
import {
  PageContainer,
  ProCard,
  ProColumns,
  ProDescriptions,
  ProList,
  ProTable,
} from '@ant-design/pro-components';
import { useAccess, useIntl } from '@umijs/max';
import { Button, Tag, message } from 'antd';
import { 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<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,
    // },
  ];

  // TODO 左侧图标目前写死，需替换
  // 操作左侧网点数据节点， 项目列表数据变更
  function fetchProjectByGroupId(record: Record<string, any>) {
    // console.log('clickTreeNode', projectCardListRef.current)
    postProjectGetProjectByGroupId({ id: record.id }).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: 'transparent' }}
              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>
          ),
        }));
        // 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="项目模型列表"
              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={
          <ProTable<API.DeviceGroup>
            // headerTitle={intl.formatMessage({
            //   id: 'pages.searchTable.title',
            //   defaultMessage: '$$$',
            // })}
            cardProps={{
              bodyStyle: {
                padding: 0,
                margin: 0,
              },
            }}
            options={{
              search: false, // TODO 1113 搜索需要对接，比对UI调整交互
              fullScreen: false,
              setting: false,
              density: false,
              reload: false,
            }}
            actionRef={actionRef}
            search={false}
            rowKey="key"
            onRow={(record) => {
              return {
                onClick: () => {
                  if (record.id) {
                    setCurrentRow(record);
                    fetchProjectByGroupId(record);
                  }
                },
              };
            }}
            pagination={{
              showSizeChanger: false,
              pageSize: 10,
            }}
            columnsState={{
              persistenceKey: 'device_group_list',
              persistenceType: 'localStorage',
            }}
            request={async (params = {}, sort) => {
              const { current, ...rest } = params;
              const reqParams = {
                page: current,
                desc: false,
                orderKey: '',
                ...rest,
              };
              if (sort && Object.keys(sort).length) {
                reqParams.orderKey = Object.keys(sort)[0];
                let sort_select = sort[reqParams.orderKey];
                reqParams.desc = sort_select === 'descend';
              }
              let resp = await postDeviceGroupGetDeviceGroupList({
                ...reqParams,
                name: params?.keyword || '',
              });
              return {
                data: resp.data.list.map((v: API.DeviceGroup) => {
                  return { ...v, key: v.id };
                }),
                success: resp.success,
                total: resp.data.total,
                current: resp.data.page,
                pageSize: resp.data.pageSize,
              };
            }}
            columns={columns1}
            rowSelection={
              {
                // onChange: (_, selectedRows) => {
                //   // setSelectedRows(selectedRows);
                // },
              }
            }
          />
        }
        rightDom={
          <>
            <ProCard bodyStyle={{ paddingTop: 24 }} colSpan="80%">
              <ProDescriptions
                column={3}
                dataSource={currentRow}
                title="网点详细信息"
                tooltip="包含了从服务器请求，columns等功能"
                columns={columns}
              ></ProDescriptions>
            </ProCard>
            <ProCard
              wrap
              tabs={{
                activeKey: tab,
                items: tabs,
                onChange: (key) => {
                  changeProjectTab(key);
                },
              }}
            ></ProCard>
          </>
        }
      ></TreeAndTableList>
      {/* <ProCard split="vertical">
        <ProCard title="网点选择" colSpan="24%" bodyStyle={{ padding: 0, margin: 0 }}></ProCard>
        <ProCard title="" headerBordered split={'horizontal'}></ProCard>
      </ProCard> */}
      <UpdateForm
        updateModalOpen={updateModalOpen}
        handleModal={handleUpdateModal}
        groupId={currentRow?.id || 0}
        reload={actionRef.current?.reload}
        projectModelConfigId={projectConfigId}
      />
    </PageContainer>
  );
};

export default AlgorithmSetting;