CustomerExtractor.php 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. <?php
  2. /**
  3. *
  4. * Copyright © Magento, Inc. All rights reserved.
  5. * See COPYING.txt for license details.
  6. */
  7. namespace Magento\Customer\Model;
  8. use Magento\Customer\Api\CustomerMetadataInterface;
  9. use Magento\Customer\Api\Data\CustomerInterface;
  10. use Magento\Customer\Api\GroupManagementInterface;
  11. use Magento\Framework\App\RequestInterface;
  12. class CustomerExtractor
  13. {
  14. /**
  15. * @var \Magento\Customer\Model\Metadata\FormFactory
  16. */
  17. protected $formFactory;
  18. /**
  19. * @var \Magento\Customer\Api\Data\CustomerInterfaceFactory
  20. */
  21. protected $customerFactory;
  22. /**
  23. * @var \Magento\Store\Model\StoreManagerInterface
  24. */
  25. protected $storeManager;
  26. /**
  27. * @var GroupManagementInterface
  28. */
  29. protected $customerGroupManagement;
  30. /**
  31. * @var \Magento\Framework\Api\DataObjectHelper
  32. */
  33. protected $dataObjectHelper;
  34. /**
  35. * @param Metadata\FormFactory $formFactory
  36. * @param \Magento\Customer\Api\Data\CustomerInterfaceFactory $customerFactory
  37. * @param \Magento\Store\Model\StoreManagerInterface $storeManager
  38. * @param GroupManagementInterface $customerGroupManagement
  39. * @param \Magento\Framework\Api\DataObjectHelper $dataObjectHelper
  40. */
  41. public function __construct(
  42. \Magento\Customer\Model\Metadata\FormFactory $formFactory,
  43. \Magento\Customer\Api\Data\CustomerInterfaceFactory $customerFactory,
  44. \Magento\Store\Model\StoreManagerInterface $storeManager,
  45. GroupManagementInterface $customerGroupManagement,
  46. \Magento\Framework\Api\DataObjectHelper $dataObjectHelper
  47. ) {
  48. $this->formFactory = $formFactory;
  49. $this->customerFactory = $customerFactory;
  50. $this->storeManager = $storeManager;
  51. $this->customerGroupManagement = $customerGroupManagement;
  52. $this->dataObjectHelper = $dataObjectHelper;
  53. }
  54. /**
  55. * @param string $formCode
  56. * @param RequestInterface $request
  57. * @param array $attributeValues
  58. * @return CustomerInterface
  59. */
  60. public function extract(
  61. $formCode,
  62. RequestInterface $request,
  63. array $attributeValues = []
  64. ) {
  65. $customerForm = $this->formFactory->create(
  66. CustomerMetadataInterface::ENTITY_TYPE_CUSTOMER,
  67. $formCode,
  68. $attributeValues
  69. );
  70. $customerData = $customerForm->extractData($request);
  71. $customerData = $customerForm->compactData($customerData);
  72. $allowedAttributes = $customerForm->getAllowedAttributes();
  73. $isGroupIdEmpty = isset($allowedAttributes['group_id']);
  74. $customerDataObject = $this->customerFactory->create();
  75. $this->dataObjectHelper->populateWithArray(
  76. $customerDataObject,
  77. $customerData,
  78. \Magento\Customer\Api\Data\CustomerInterface::class
  79. );
  80. $store = $this->storeManager->getStore();
  81. if ($isGroupIdEmpty) {
  82. $customerDataObject->setGroupId(
  83. $this->customerGroupManagement->getDefaultGroup($store->getId())->getId()
  84. );
  85. }
  86. $customerDataObject->setWebsiteId($store->getWebsiteId());
  87. $customerDataObject->setStoreId($store->getId());
  88. return $customerDataObject;
  89. }
  90. }