<?php namespace MailPoet\Config; if (!defined('ABSPATH')) exit; use MailPoet\Cron\ActionScheduler\ActionScheduler as CronActionScheduler; use MailPoet\Cron\CronTrigger; use MailPoet\InvalidStateException; use MailPoet\Settings\SettingsController; use MailPoet\WP\Functions as WPFunctions; class Activator { public const TRANSIENT_ACTIVATE_KEY = 'mailpoet_activator_activate'; private const TRANSIENT_EXPIRATION = 120; // seconds /** @var SettingsController */ private $settings; /** @var Populator */ private $populator; /** @var WPFunctions */ private $wp; /** @var Migrator */ private $migrator; /** @var CronActionScheduler */ private $cronActionSchedulerRunner; public function __construct( SettingsController $settings, Populator $populator, WPFunctions $wp, Migrator $migrator, CronActionScheduler $cronActionSchedulerRunner ) { $this->settings = $settings; $this->populator = $populator; $this->wp = $wp; $this->migrator = $migrator; $this->cronActionSchedulerRunner = $cronActionSchedulerRunner; } public function activate() { $isRunning = $this->wp->getTransient(self::TRANSIENT_ACTIVATE_KEY); if ($isRunning === false) { $this->lockActivation(); try { $this->processActivate(); } finally { $this->unlockActivation(); } } else { throw new InvalidStateException(__('MailPoet version update is in progress, please refresh the page in a minute.', 'mailpoet')); } } private function lockActivation(): void { $this->wp->setTransient(self::TRANSIENT_ACTIVATE_KEY, '1', self::TRANSIENT_EXPIRATION); } private function unlockActivation(): void { $this->wp->deleteTransient(self::TRANSIENT_ACTIVATE_KEY); } private function processActivate(): void { $this->migrator->up(); $this->deactivateCronActions(); $this->populator->up(); $this->updateDbVersion(); $caps = new Capabilities(); $caps->setupWPCapabilities(); $localizer = new Localizer(); $localizer->forceInstallLanguagePacks($this->wp); } public function deactivate() { $this->lockActivation(); $this->migrator->down(); $caps = new Capabilities(); $caps->removeWPCapabilities(); $this->unlockActivation(); } /** * Deactivate action scheduler cron actions when the migration run. * This should prevent processing actions during migrations. * They are later re-activated in CronTrigger * * @return void */ private function deactivateCronActions(): void { $currentMethod = $this->settings->get(CronTrigger::SETTING_NAME . '.method'); if ($currentMethod !== CronTrigger::METHOD_ACTION_SCHEDULER) { return; } $this->cronActionSchedulerRunner->unscheduleAllCronActions(); } public function updateDbVersion() { try { $currentDbVersion = $this->settings->get('db_version'); } catch (\Exception $e) { $currentDbVersion = null; } $this->settings->set('db_version', Env::$version); // if current db version and plugin version differ, log an update if (version_compare((string)$currentDbVersion, Env::$version) !== 0) { $updatesLog = (array)$this->settings->get('updates_log', []); $updatesLog[] = [ 'previous_version' => $currentDbVersion, 'new_version' => Env::$version, 'date' => date('Y-m-d H:i:s'), ]; $this->settings->set('updates_log', $updatesLog); } } }