Files
piratepoet/lib/Subscription/Throttling.php
Jan Lysý 4d56020071 Remove usage old model from Throttling
[MAILPOET-3032]
2021-04-12 14:26:36 +02:00

74 lines
2.5 KiB
PHP

<?php
namespace MailPoet\Subscription;
use MailPoet\Entities\SubscriberIPEntity;
use MailPoet\Subscribers\SubscriberIPsRepository;
use MailPoet\Util\Helpers;
use MailPoet\WP\Functions as WPFunctions;
class Throttling {
/** @var SubscriberIPsRepository */
private $subscriberIPsRepository;
/** @var WPFunctions */
private $wp;
public function __construct(SubscriberIPsRepository $subscriberIPsRepository, WPFunctions $wp) {
$this->wp = $wp;
$this->subscriberIPsRepository = $subscriberIPsRepository;
}
public function throttle() {
$subscriptionLimitEnabled = $this->wp->applyFilters('mailpoet_subscription_limit_enabled', true);
$subscriptionLimitWindow = (int)$this->wp->applyFilters('mailpoet_subscription_limit_window', DAY_IN_SECONDS);
$subscriptionLimitBase = (int)$this->wp->applyFilters('mailpoet_subscription_limit_base', MINUTE_IN_SECONDS);
$subscriberIp = Helpers::getIP();
if ($subscriptionLimitEnabled && !$this->wp->isUserLoggedIn()) {
if (!empty($subscriberIp)) {
$subscriptionCount = $this->subscriberIPsRepository->getCountByIPAndCreatedAtAfterTimeInSeconds($subscriberIp, $subscriptionLimitWindow);
if ($subscriptionCount > 0) {
$timeout = $subscriptionLimitBase * pow(2, $subscriptionCount - 1);
$existingUser = $this->subscriberIPsRepository->findOneByIPAndCreatedAtAfterTimeInSeconds($subscriberIp, $timeout);
if (!empty($existingUser)) {
return $timeout;
}
}
}
}
if ($subscriberIp !== null) {
$ip = new SubscriberIPEntity($subscriberIp);
$existingIp = $this->subscriberIPsRepository->findOneBy(['ip' => $ip->getIP(), 'createdAt' => $ip->getCreatedAt()]);
if (!$existingIp) {
$this->subscriberIPsRepository->persist($ip);
$this->subscriberIPsRepository->flush();
}
}
$this->purge();
return false;
}
public function purge(): void {
$interval = $this->wp->applyFilters('mailpoet_subscription_purge_window', MONTH_IN_SECONDS);
$this->subscriberIPsRepository->deleteCreatedAtBeforeTimeInSeconds($interval);
}
public function secondsToTimeString($seconds): string {
$hrs = floor($seconds / 3600);
$min = floor($seconds % 3600 / 60);
$sec = $seconds % 3600 % 60;
$result = [
'hours' => $hrs ? sprintf(__('%d hours', 'mailpoet'), $hrs) : '',
'minutes' => $min ? sprintf(__('%d minutes', 'mailpoet'), $min) : '',
'seconds' => $sec ? sprintf(__('%d seconds', 'mailpoet'), $sec) : '',
];
return join(' ', array_filter($result));
}
}