ExtensionAttributesProcessor.php 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. <?php
  2. /**
  3. * Copyright © Magento, Inc. All rights reserved.
  4. * See COPYING.txt for license details.
  5. */
  6. declare(strict_types=1);
  7. namespace Magento\Sales\Model\Order\ShipmentDocumentFactory;
  8. use Magento\Framework\Reflection\ExtensionAttributesProcessor as AttributesProcessor;
  9. use Magento\Sales\Api\Data\ShipmentCreationArgumentsExtensionInterface;
  10. use Magento\Sales\Api\Data\ShipmentCreationArgumentsInterface;
  11. use Magento\Sales\Api\Data\ShipmentExtensionFactory;
  12. use Magento\Sales\Api\Data\ShipmentExtensionInterface;
  13. use Magento\Sales\Api\Data\ShipmentInterface;
  14. /**
  15. * Build and set extension attributes for shipment.
  16. */
  17. class ExtensionAttributesProcessor
  18. {
  19. /**
  20. * @var ShipmentExtensionFactory
  21. */
  22. private $shipmentExtensionFactory;
  23. /**
  24. * @var ExtensionAttributesProcessor
  25. */
  26. private $extensionAttributesProcessor;
  27. /**
  28. * @param ShipmentExtensionFactory $shipmentExtensionFactory
  29. * @param AttributesProcessor $extensionAttributesProcessor
  30. */
  31. public function __construct(
  32. ShipmentExtensionFactory $shipmentExtensionFactory,
  33. AttributesProcessor $extensionAttributesProcessor
  34. ) {
  35. $this->shipmentExtensionFactory = $shipmentExtensionFactory;
  36. $this->extensionAttributesProcessor = $extensionAttributesProcessor;
  37. }
  38. /**
  39. * @param ShipmentInterface $shipment
  40. * @param ShipmentCreationArgumentsInterface $arguments
  41. * @return void
  42. */
  43. public function execute(
  44. ShipmentInterface $shipment,
  45. ShipmentCreationArgumentsInterface $arguments
  46. ): void {
  47. $shipmentExtensionAttributes = [];
  48. if (null !== $shipment->getExtensionAttributes()) {
  49. $shipmentExtensionAttributes = $this->extensionAttributesProcessor->buildOutputDataArray(
  50. $shipment->getExtensionAttributes(),
  51. ShipmentExtensionInterface::class
  52. );
  53. }
  54. $argumentsExtensionAttributes = [];
  55. if (null !== $arguments->getExtensionAttributes()) {
  56. $argumentsExtensionAttributes = $this->extensionAttributesProcessor->buildOutputDataArray(
  57. $arguments->getExtensionAttributes(),
  58. ShipmentCreationArgumentsExtensionInterface::class
  59. );
  60. }
  61. $mergedExtensionAttributes = $this->shipmentExtensionFactory->create([
  62. 'data' => array_merge($shipmentExtensionAttributes, $argumentsExtensionAttributes)
  63. ]);
  64. $shipment->setExtensionAttributes($mergedExtensionAttributes);
  65. }
  66. }