React 相关

import { forwardRef, useImperativeHandle, useRef, useState } from "react"
import { Modal } from "antd"

export type ModalPayload = {
  onOk?: () => void
  onCancel?: () => void
}

export interface ModalProps {
  show: (payload: ModalPayload) => void
}

const TestModal = forwardRef<ModalProps>((_props, ref) => {
  const [open, setOpen] = useState(false)
  const payloadRef = useRef<EStockOrderModalPayload>({})

  useImperativeHandle(
    ref,
    () => ({
      show(payload) {
        payloadRef.current = payload
        setOpen(true)
      },
    })
  )

  const wrapWithClose = (method?: () => void) => () => {
    setOpen(false)
    if (method) {
      method()
    }
  }

  return (
    <>
      <Modal
        open={open}
        onOk={wrapWithClose(payloadRef.current.onOk)}
        onCancel={wrapWithClose(payloadRef.current.onCancel)}
        closable={false}
      >
        <div>123123213</div>
      </Modal>
    </>
  )
})

TestModal.displayName = 'TestModal'
export default TestModal