import { useState, useContext } from 'react'
import PropTypes from 'prop-types'
import { ApiServerContext } from '../../context/ApiServerContext'
import { message } from 'antd'
import MessageDialogView from '../../common/MessageDialogView.jsx'

const CancelShipment = ({ onOk, objectData }) => {
  const [cancelLoading, setCancelLoading] = useState(false)
  const { sendObjectFunction } = useContext(ApiServerContext)

  const handleCancel = async () => {
    setCancelLoading(true)
    try {
      const result = await sendObjectFunction(
        objectData._id,
        'Shipment',
        'cancel'
      )
      if (result) {
        message.success('Shipment cancelled successfully')
        onOk(result)
      }
    } catch (error) {
      console.error('Error cancelling shipment:', error)
    } finally {
      setCancelLoading(false)
    }
  }

  return (
    <MessageDialogView
      title={'Are you sure you want to cancel this shipment?'}
      description={`Cancelling shipment ${objectData?.name || objectData?._reference || objectData?._id} will update its status to cancelled.`}
      onOk={handleCancel}
      okText='Cancel'
      okLoading={cancelLoading}
    />
  )
}

CancelShipment.propTypes = {
  onOk: PropTypes.func.isRequired,
  objectData: PropTypes.object
}

export default CancelShipment
