dros/src/pages/patient/components/ActionPanel.tsx

import React from 'react';
import { Button, Tooltip, Modal, message } from 'antd';
import { useDispatch, useSelector } from 'react-redux';
import {
  deleteWorkThunk,
  lockWorkInWorklistThunk,
} from '@/states/patient/worklist/slices/workSlice';
import { deleteWorkThunk as deleteWorkThunkFromHistory, lockWorkInhistorylistThunk } from '@/states/patient/worklist/slices/history';
import { switchToSendPanel } from '@/states/patient/worklist/slices/historyPanelSwitchSlice';
import { FormattedMessage } from 'react-intl';
import { AppDispatch, RootState, useAppSelector } from '@/states/store';
import Icon from '@/components/Icon';
import DiagnosticReport from '../DiagnosticReport';
import { Popup } from 'antd-mobile';
import { setVisible } from '@/states/patient/DiagnosticReport/slice';
import EditTaskModal from './EditTaskModal';
import { openEditModal } from '@/states/patient/edit/editFormSlice';
import { setBusinessFlow } from '@/states/BusinessFlowSlice';
import { setSourceTask, setRegisterInfo } from '@/states/patient/reregister/reregisterSlice';
import { mapTaskToRegisterInfo } from '@/domain/patient/taskToRegister';
import { showNotImplemented } from '@/utils/notificationHelper';
import { string } from 'zod';

interface ActionButtonProps {
  icon: React.ReactNode;
  tooltip: React.ReactNode;
  onClick?: () => void;
  'data-testid'?: string;
}

const ActionButton: React.FC<ActionButtonProps> = ({
  icon,
  tooltip,
  onClick,
  'data-testid': dataTestId,
}) => (
  <Tooltip title={tooltip}>
    <Button icon={icon} onClick={onClick} style={{ width: '2.5rem' }} data-testid={dataTestId} />
  </Tooltip>
);

const ActionPanel: React.FC = () => {
  const dispatch = useDispatch<AppDispatch>();

  const workSelectedIds = useSelector(
    (state: RootState) => state.workSelection.selectedIds
  );
  const historySelectedIds = useSelector(
    (state: RootState) => state.historySelection.selectedIds
  );
  const visible = useSelector(
    (state: RootState) => state.diagnosticReport.visible
  );
  const themeType = useAppSelector((state) => state.theme.themeType);
  const currentKey = useSelector(
    (state: RootState) => state.BusinessFlow.currentKey
  );
  const workEntities = useSelector(
    (state: RootState) => state.workEntities.data
  );
  const workEntitiesFromHistory = useSelector(
    (state: RootState) => state.historyEntities.data
  );

  const getSelectedWorkIds = () => {
    const selectedIds =
      currentKey === 'worklist' ? workSelectedIds : historySelectedIds;
    return selectedIds;
  };
  const getDeleteThunk = (): typeof deleteWorkThunk => {
    return currentKey === 'worklist'
      ? deleteWorkThunk
      : deleteWorkThunkFromHistory;
  };
  const getLockThunk = () => {
    return currentKey === 'worklist'
      ? lockWorkInWorklistThunk
      : lockWorkInhistorylistThunk;
  };

  const getSelectedWorks = () => {
    const selectedIds = getSelectedWorkIds();
    if (currentKey === 'worklist') {
      return workEntities.find((w) => selectedIds.includes(w.StudyID));
    } else if (currentKey === 'historylist') {
      return workEntitiesFromHistory.find((w) =>
        selectedIds.includes(w.StudyID)
      );
    }
  };
  // 使用 worklist 或 history 的选中项，取决于哪个有值
  const selectedIds = getSelectedWorkIds();

  const handleDelete = () => {
    if (selectedIds.length === 0) {
      message.warning('请先选择要删除的项目');
      return;
    }
    //判断是否锁定
    const selectedWork = getSelectedWorks();
    if (selectedWork?.StudyLock === 'Locked') {
      message.warning('锁定状态不可删除');
      return;
    }

    Modal.confirm({
      title: '确认删除',
      content: `确定要删除选中的 ${selectedIds.length} 个项目吗？此操作不可撤销。`,
      okText: '确认删除',
      cancelText: '取消',
      okButtonProps: {
        danger: true,
        'data-testid': 'modal-confirm-delete'
      },
      cancelButtonProps: {
        'data-testid': 'modal-cancel-delete'
      },
      centered: true,
      onOk: () => {
        const delThunk = getDeleteThunk();
        dispatch(delThunk(selectedIds));
      },
    });
  };

  const handleSend = () => {
    dispatch(switchToSendPanel());
  };

  const handleShowReport = () => {
    dispatch(setVisible(true));
  };

  const getWorksFromWorklistOrHistory = () => {
    return currentKey === 'worklist' ? workEntities : workEntitiesFromHistory;
  };
  
  const handleEdit = () => {
    const selectedIds = getSelectedWorkIds();

    if (selectedIds.length === 0) {
      message.warning('请先选择要编辑的任务');
      return;
    }

    if (selectedIds.length > 1) {
      message.warning('只能编辑一个任务');
      return;
    }

    const works = getWorksFromWorklistOrHistory();
    const task = works.find((item) => item.StudyID === selectedIds[0]);

    if (task) {
      // 通过 dispatch action 传递数据到 slice
      dispatch(openEditModal(task));
    }
  };
  
  const handleReRegister = async () => {
    const selectedIds = getSelectedWorkIds();
    
    // 验证选择
    if (selectedIds.length === 0) {
      message.warning('请先选择要重新注册的项目');
      return;
    }
    
    if (selectedIds.length > 1) {
      message.warning('只能重新注册一个项目');
      return;
    }
    
    const works = getWorksFromWorklistOrHistory();
    const task = works.find((item) => item.StudyID === selectedIds[0]);
    
    // 验证任务数据
    if (!task) {
      message.error('找不到选中的任务数据');
      return;
    }
    
    // 验证必要字段
    if (!task.PatientName || !task.PatientID) {
      message.error('任务数据不完整，缺少必要信息');
      return;
    }
    
    try {
      // 设置源任务数据
      await dispatch(setSourceTask(task));
      
      // 将Task数据映射为RegisterInfo
      const registerInfo = mapTaskToRegisterInfo(task);
      console.log(`study转换后的注册信息：${JSON.stringify(registerInfo)}`)
      // 验证映射后的数据
      if (!registerInfo.patient_name || !registerInfo.patient_id) {
        throw new Error('数据映射失败，缺少必要信息');
      }
      console.log(`开始设置注册信息`);
      // 设置注册信息
      await dispatch(setRegisterInfo(registerInfo));
      console.log(`开始切换到注册页面`);
      // 切换到注册页面
      await dispatch(setBusinessFlow('register'));
      
      message.success('已切换到注册页面，表单已预填充');
    } catch (error) {
      console.error('ReRegister error:', error);
      const errorMessage = error instanceof Error ? error.message : '重新注册失败，请重试';
      message.error(`重新注册失败: ${errorMessage}`);
    }
  };

  const handleLock = () => {
    const selectedIds = getSelectedWorkIds();

    // 2. 检查是否有选中项
    if (selectedIds.length === 0) {
      message.warning('请先选择要锁定/解锁的项目');
      return;
    }
    // 3. 获取第一个选中项的锁定状态
    const works = getWorksFromWorklistOrHistory();
    const selectedItem = works.find((item) => item.StudyID === selectedIds[0]);
    if (!selectedItem) return;

    // 4. 根据当前状态切换
    const newLockState =
      selectedItem.StudyLock === 'Locked' ? 'Unlocked' : 'Locked';
    const lockThunk = getLockThunk();
    // 为每个选中项执行锁定/解锁操作
    selectedIds.forEach((studyId) => {
      console.log(
        `锁定，触发action ，目标 studyid是 ${studyId}，新状态是 ${newLockState}`
      );
      dispatch(lockThunk({ studyId, lock: newLockState }));
    });
  };

  return (
    <div className="flex flex-wrap gap-2 w-full">
      <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="Delete"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.deleteTask"
            defaultMessage="actionPanel.deleteTask"
          />
        }
        onClick={handleDelete}
        data-testid="delete-button"
      />
      <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="EditPatient"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.editPatient"
            defaultMessage="actionPanel.editPatient"
          />
        }
        onClick={handleEdit}
      />
      <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="Protect"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.lockTask"
            defaultMessage="actionPanel.lockTask"
          />
        }
        onClick={handleLock}
      />
      <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="RIS"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.risSync"
            defaultMessage="actionPanel.risSync"
          />
        }
        onClick={() => showNotImplemented('')}
      />
      {currentKey === 'historylist' && (
      <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="ReRegister"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.reRegister"
            defaultMessage="actionPanel.reRegister"
          />
        }
        onClick={handleReRegister}
      />)}


      <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="Sort"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.sortList"
            defaultMessage="actionPanel.sortList"
          />
        }
        onClick={() => showNotImplemented('')}
      />

      <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="Swap"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.imageExchange"
            defaultMessage="actionPanel.imageExchange"
          />
        }
        onClick={() => showNotImplemented('')}
      />
      {currentKey === 'historylist' && (
        <ActionButton
          icon={
            <Icon
              module="module-patient"
              name="Send"
              userId="base"
              theme="default"
              size="2x"
              state="normal"
            />
          }
          tooltip={
            <FormattedMessage
              id="actionPanel.send"
              defaultMessage="actionPanel.send"
            />
          }
          onClick={handleSend}
        />
      )}
      {/* 我们不需要导出功能 */}
      {/* <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="Export"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.export"
            defaultMessage="actionPanel.export"
          />
        }
        onClick={() => showNotImplemented('')}
      /> */}
      {/* 我们不需要导入功能 */}
      {/* <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="Import"
            userId="base"
            theme="default"
            size="2x"
            state="normal"
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.import"
            defaultMessage="actionPanel.import"
          />
        }
        onClick={() => showNotImplemented('')}
      /> */}
      <ActionButton
        icon={
          <Icon
            module="module-patient"
            name="report"
            userId="base"
            theme={themeType}
            size="2x"
            state="normal"
            width={40}
            height={40}
          />
        }
        tooltip={
          <FormattedMessage
            id="actionPanel.showReport"
            defaultMessage="actionPanel.showReport"
          />
        }
        onClick={handleShowReport}
      />
      <Popup
        visible={visible}
        onMaskClick={() => dispatch(setVisible(false))}
        position="right"
        bodyStyle={{ width: '100vw', height: '100vh' }}
      >
        <DiagnosticReport />
      </Popup>
      <EditTaskModal />
    </div>
  );
};

export default ActionPanel;