Refactor addSubscriberField to Doctrine

[MAILPOET-3628]
This commit is contained in:
Pavel Dohnal
2021-09-22 09:10:03 +02:00
committed by Veljko V
parent b368a76593
commit acc7ab043d

View File

@ -4,7 +4,7 @@ namespace MailPoet\API\MP\v1;
use MailPoet\CustomFields\ApiDataSanitizer; use MailPoet\CustomFields\ApiDataSanitizer;
use MailPoet\CustomFields\CustomFieldsRepository; use MailPoet\CustomFields\CustomFieldsRepository;
use MailPoet\Models\CustomField; use MailPoet\Entities\CustomFieldEntity;
class CustomFields { class CustomFields {
/** @var ApiDataSanitizer */ /** @var ApiDataSanitizer */
@ -64,15 +64,20 @@ class CustomFields {
} }
public function addSubscriberField(array $data = []): array { public function addSubscriberField(array $data = []): array {
$customField = CustomField::createOrUpdate($this->customFieldsDataSanitizer->sanitize($data)); try {
$errors = $customField->getErrors(); $customField = $this->customFieldsRepository->createOrUpdate($this->customFieldsDataSanitizer->sanitize($data));
if (!empty($errors)) { } catch (\Exception $e) {
throw new APIException('Failed to save a new subscriber field ' . join(', ', $errors), APIException::FAILED_TO_SAVE_SUBSCRIBER_FIELD); throw new APIException('Failed to save a new subscriber field ' . join(', ', $e->getMessage()), APIException::FAILED_TO_SAVE_SUBSCRIBER_FIELD);
} }
$customField = CustomField::findOne($customField->id); $customField = $this->customFieldsRepository->findOneById($customField->getId());
if (!$customField instanceof CustomField) { if (!$customField instanceof CustomFieldEntity) {
throw new APIException('Failed to create a new subscriber field', APIException::FAILED_TO_SAVE_SUBSCRIBER_FIELD); throw new APIException('Failed to create a new subscriber field', APIException::FAILED_TO_SAVE_SUBSCRIBER_FIELD);
} }
return $customField->asArray(); return [
'id' => 'cf_' . $customField->getId(),
'name' => $customField->getName(),
'type' => $customField->getType(),
'params' => $customField->getParams(),
];
} }
} }