Нема описа

SubscriberSubscribeController.php 8.6KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250
  1. <?php
  2. namespace MailPoet\Subscribers;
  3. if (!defined('ABSPATH')) exit;
  4. use MailPoet\Entities\FormEntity;
  5. use MailPoet\Form\FormsRepository;
  6. use MailPoet\Form\Util\FieldNameObfuscator;
  7. use MailPoet\NotFoundException;
  8. use MailPoet\Settings\SettingsController;
  9. use MailPoet\Statistics\StatisticsFormsRepository;
  10. use MailPoet\Subscription\Captcha;
  11. use MailPoet\Subscription\CaptchaSession;
  12. use MailPoet\Subscription\SubscriptionUrlFactory;
  13. use MailPoet\Subscription\Throttling as SubscriptionThrottling;
  14. use MailPoet\UnexpectedValueException;
  15. use MailPoet\WP\Functions as WPFunctions;
  16. class SubscriberSubscribeController {
  17. /** @var FormsRepository */
  18. private $formsRepository;
  19. /** @var Captcha */
  20. private $subscriptionCaptcha;
  21. /** @var CaptchaSession */
  22. private $captchaSession;
  23. /** @var SubscriptionUrlFactory */
  24. private $subscriptionUrlFactory;
  25. /** @var FieldNameObfuscator */
  26. private $fieldNameObfuscator;
  27. /** @var SettingsController */
  28. private $settings;
  29. /** @var RequiredCustomFieldValidator */
  30. private $requiredCustomFieldValidator;
  31. /** @var SubscriberActions */
  32. private $subscriberActions;
  33. /** @var WPFunctions */
  34. private $wp;
  35. /** @var SubscriptionThrottling */
  36. private $throttling;
  37. /** @var StatisticsFormsRepository */
  38. private $statisticsFormsRepository;
  39. public function __construct(
  40. Captcha $subscriptionCaptcha,
  41. CaptchaSession $captchaSession,
  42. SubscriberActions $subscriberActions,
  43. SubscriptionUrlFactory $subscriptionUrlFactory,
  44. SubscriptionThrottling $throttling,
  45. FieldNameObfuscator $fieldNameObfuscator,
  46. RequiredCustomFieldValidator $requiredCustomFieldValidator,
  47. SettingsController $settings,
  48. FormsRepository $formsRepository,
  49. StatisticsFormsRepository $statisticsFormsRepository,
  50. WPFunctions $wp
  51. ) {
  52. $this->formsRepository = $formsRepository;
  53. $this->subscriptionCaptcha = $subscriptionCaptcha;
  54. $this->captchaSession = $captchaSession;
  55. $this->subscriptionUrlFactory = $subscriptionUrlFactory;
  56. $this->requiredCustomFieldValidator = $requiredCustomFieldValidator;
  57. $this->fieldNameObfuscator = $fieldNameObfuscator;
  58. $this->settings = $settings;
  59. $this->subscriberActions = $subscriberActions;
  60. $this->wp = $wp;
  61. $this->throttling = $throttling;
  62. $this->statisticsFormsRepository = $statisticsFormsRepository;
  63. }
  64. public function subscribe(array $data): array {
  65. $form = $this->getForm($data);
  66. if (!empty($data['email'])) {
  67. throw new UnexpectedValueException(__('Please leave the first field empty.', 'mailpoet'));
  68. }
  69. $captchaSettings = $this->settings->get('captcha');
  70. $data = $this->initCaptcha($captchaSettings, $form, $data);
  71. $data = $this->deobfuscateFormPayload($data);
  72. try {
  73. $this->requiredCustomFieldValidator->validate($data, $form);
  74. } catch (\Exception $e) {
  75. throw new UnexpectedValueException($e->getMessage());
  76. }
  77. $segmentIds = $this->getSegmentIds($form, $data['segments'] ?? []);
  78. unset($data['segments']);
  79. $meta = $this->validateCaptcha($captchaSettings, $data);
  80. if (isset($meta['error'])) {
  81. return $meta;
  82. }
  83. // only accept fields defined in the form
  84. $formFieldIds = array_filter(array_map(function (array $formField): ?string {
  85. if (!isset($formField['id'])) {
  86. return null;
  87. }
  88. return is_numeric($formField['id']) ? "cf_{$formField['id']}" : $formField['id'];
  89. }, $form->getBlocksByTypes(FormEntity::FORM_FIELD_TYPES)));
  90. $data = array_intersect_key($data, array_flip($formFieldIds));
  91. // make sure we don't allow too many subscriptions with the same ip address
  92. $timeout = $this->throttling->throttle();
  93. if ($timeout > 0) {
  94. $timeToWait = $this->throttling->secondsToTimeString($timeout);
  95. $meta['refresh_captcha'] = true;
  96. $meta['error'] = sprintf(__('You need to wait %s before subscribing again.', 'mailpoet'), $timeToWait);
  97. return $meta;
  98. }
  99. $subscriber = $this->subscriberActions->subscribe($data, $segmentIds);
  100. if (!empty($captchaSettings['type']) && $captchaSettings['type'] === Captcha::TYPE_BUILTIN) {
  101. // Captcha has been verified, invalidate the session vars
  102. $this->captchaSession->reset();
  103. }
  104. // record form statistics
  105. $this->statisticsFormsRepository->record($form, $subscriber);
  106. $formSettings = $form->getSettings();
  107. if (!empty($formSettings['on_success'])) {
  108. if ($formSettings['on_success'] === 'page') {
  109. // redirect to a page on a success, pass the page url in the meta
  110. $meta['redirect_url'] = $this->wp->getPermalink($formSettings['success_page']);
  111. } else if ($formSettings['on_success'] === 'url') {
  112. $meta['redirect_url'] = $formSettings['success_url'];
  113. }
  114. }
  115. return $meta;
  116. }
  117. private function deobfuscateFormPayload($data): array {
  118. return $this->fieldNameObfuscator->deobfuscateFormPayload($data);
  119. }
  120. private function initCaptcha(?array $captchaSettings, FormEntity $form, array $data): array {
  121. if (!$captchaSettings || !isset($captchaSettings['type'])) {
  122. return $data;
  123. }
  124. if ($captchaSettings['type'] === Captcha::TYPE_BUILTIN) {
  125. $captchaSessionId = isset($data['captcha_session_id']) ? $data['captcha_session_id'] : null;
  126. $this->captchaSession->init($captchaSessionId);
  127. if (!isset($data['captcha'])) {
  128. // Save form data to session
  129. $this->captchaSession->setFormData(array_merge($data, ['form_id' => $form->getId()]));
  130. } elseif ($this->captchaSession->getFormData()) {
  131. // Restore form data from session
  132. $data = array_merge($this->captchaSession->getFormData(), ['captcha' => $data['captcha']]);
  133. }
  134. // Otherwise use the post data
  135. }
  136. return $data;
  137. }
  138. private function validateCaptcha($captchaSettings, $data): array {
  139. if (empty($captchaSettings['type'])) {
  140. return [];
  141. }
  142. $meta = [];
  143. $isBuiltinCaptchaRequired = false;
  144. if ($captchaSettings['type'] === Captcha::TYPE_BUILTIN) {
  145. $isBuiltinCaptchaRequired = $this->subscriptionCaptcha->isRequired(isset($data['email']) ? $data['email'] : '');
  146. if ($isBuiltinCaptchaRequired && empty($data['captcha'])) {
  147. $meta['redirect_url'] = $this->subscriptionUrlFactory->getCaptchaUrl($this->captchaSession->getId());
  148. $meta['error'] = __('Please fill in the CAPTCHA.', 'mailpoet');
  149. return $meta;
  150. }
  151. }
  152. if ($captchaSettings['type'] === Captcha::TYPE_RECAPTCHA && empty($data['recaptcha'])) {
  153. return ['error' => __('Please check the CAPTCHA.', 'mailpoet')];
  154. }
  155. if ($captchaSettings['type'] === Captcha::TYPE_RECAPTCHA) {
  156. $response = empty($data['recaptcha']) ? $data['recaptcha-no-js'] : $data['recaptcha'];
  157. $response = $this->wp->wpRemotePost('https://www.google.com/recaptcha/api/siteverify', [
  158. 'body' => [
  159. 'secret' => $captchaSettings['recaptcha_secret_token'],
  160. 'response' => $response,
  161. ],
  162. ]);
  163. if (is_wp_error($response)) {
  164. return ['error' => __('Error while validating the CAPTCHA.', 'mailpoet')];
  165. }
  166. $response = json_decode(wp_remote_retrieve_body($response));
  167. if (empty($response->success)) {
  168. return ['error' => __('Error while validating the CAPTCHA.', 'mailpoet')];
  169. }
  170. } elseif ($captchaSettings['type'] === Captcha::TYPE_BUILTIN && $isBuiltinCaptchaRequired) {
  171. $captchaHash = $this->captchaSession->getCaptchaHash();
  172. if (empty($captchaHash)) {
  173. $meta['error'] = __('Please regenerate the CAPTCHA.', 'mailpoet');
  174. } elseif (!hash_equals(strtolower($data['captcha']), strtolower($captchaHash))) {
  175. $this->captchaSession->setCaptchaHash(null);
  176. $meta['refresh_captcha'] = true;
  177. $meta['error'] = __('The characters entered do not match with the previous CAPTCHA.', 'mailpoet');
  178. }
  179. }
  180. return $meta;
  181. }
  182. private function getSegmentIds(FormEntity $form, array $segmentIds): array {
  183. // If form contains segment selection blocks allow only segments ids configured in those blocks
  184. $segmentBlocksSegmentIds = $form->getSegmentBlocksSegmentIds();
  185. if (!empty($segmentBlocksSegmentIds)) {
  186. $segmentIds = array_intersect($segmentIds, $segmentBlocksSegmentIds);
  187. } else {
  188. $segmentIds = $form->getSettingsSegmentIds();
  189. }
  190. if (empty($segmentIds)) {
  191. throw new UnexpectedValueException(__('Please select a list.', 'mailpoet'));
  192. }
  193. return $segmentIds;
  194. }
  195. private function getForm(array $data): FormEntity {
  196. $formId = (isset($data['form_id']) ? (int)$data['form_id'] : false);
  197. $form = $this->formsRepository->findOneById($formId);
  198. if (!$form) {
  199. throw new NotFoundException(__('Please specify a valid form ID.', 'mailpoet'));
  200. }
  201. return $form;
  202. }
  203. }