oont-contents/plugins/jetpack-boost/app/modules/class-modules-setup.php
2025-04-06 08:34:48 +02:00

269 lines
7.6 KiB
PHP

<?php
namespace Automattic\Jetpack_Boost\Modules;
use Automattic\Jetpack\Schema\Schema;
use Automattic\Jetpack\WP_JS_Data_Sync\Data_Sync;
use Automattic\Jetpack_Boost\Contracts\Changes_Output_After_Activation;
use Automattic\Jetpack_Boost\Contracts\Feature;
use Automattic\Jetpack_Boost\Contracts\Has_Data_Sync;
use Automattic\Jetpack_Boost\Contracts\Has_Setup;
use Automattic\Jetpack_Boost\Data_Sync\Modules_State_Entry;
use Automattic\Jetpack_Boost\Lib\Setup;
use Automattic\Jetpack_Boost\Lib\Status;
use Automattic\Jetpack_Boost\REST_API\Contracts\Has_Always_Available_Endpoints;
use Automattic\Jetpack_Boost\REST_API\Contracts\Has_Endpoints;
use Automattic\Jetpack_Boost\REST_API\REST_API;
class Modules_Setup implements Has_Setup, Has_Data_Sync {
/**
* @var Module[]
*/
protected $available_modules;
/**
* @var Module[]
*/
protected $available_submodules;
public function __construct() {
$this->available_modules = $this->get_available_modules();
$this->available_submodules = $this->get_available_submodules();
}
public function get_available_modules() {
$available_modules = array();
foreach ( Features_Index::FEATURES as $feature ) {
$module = new Module( new $feature() );
if ( $module->is_available() ) {
$available_modules[ $feature::get_slug() ] = $module;
}
}
return $available_modules;
}
public function get_available_submodules() {
$available_submodules = array();
foreach ( Features_Index::SUB_FEATURES as $feature ) {
$module = new Module( new $feature() );
if ( $module->is_available() ) {
$available_submodules[ $feature::get_slug() ] = $module;
}
}
return $available_submodules;
}
public function get_available_modules_and_submodules() {
return array_merge( $this->available_modules, $this->available_submodules );
}
/**
* Get modules that are currently active and optimizing the site.
*
* @return string[] Slugs of optimization modules that are currently active and serving.
*/
public function get_ready_active_optimization_modules() {
$working_modules = array();
foreach ( $this->get_available_modules_and_submodules() as $slug => $module ) {
if ( $module->is_optimizing() ) {
$working_modules[] = $slug;
}
}
return $working_modules;
}
public function get_status() {
$status = array();
foreach ( $this->get_available_modules_and_submodules() as $slug => $module ) {
$status[ $slug ] = $module->is_enabled();
}
return $status;
}
/**
* Used to register endpoints that will be available even
* if the module is not enabled.
*
* @return bool|void
*/
public function register_always_available_endpoints() {
foreach ( Features_Index::get_all_features() as $feature_class ) {
$feature = new $feature_class();
if ( ! $feature instanceof Has_Always_Available_Endpoints || ! $feature instanceof Feature ) {
continue;
}
if ( empty( $feature->get_always_available_endpoints() ) ) {
return false;
}
$module = new Module( $feature );
if ( ! $module->is_available() ) {
continue;
}
REST_API::register( $feature->get_always_available_endpoints() );
}
}
private function setup_features_data_sync() {
foreach ( Features_Index::get_all_features() as $feature_class ) {
$feature = new $feature_class();
if ( ! $feature instanceof Has_Data_Sync ) {
continue;
}
$feature->register_data_sync( Data_Sync::get_instance( JETPACK_BOOST_DATASYNC_NAMESPACE ) );
}
}
private function register_endpoints( $feature ) {
if ( ! $feature instanceof Has_Endpoints ) {
return false;
}
if ( empty( $feature->get_endpoints() ) ) {
return false;
}
REST_API::register( $feature->get_endpoints() );
}
public function load_modules() {
$this->init_modules( $this->available_modules );
}
/**
* Registers general data sync for the modules.
*/
public function register_data_sync( $instance ) {
$modules_state_schema = Schema::as_array(
Schema::as_assoc_array(
array(
'active' => Schema::as_boolean()->fallback( false ),
'available' => Schema::as_boolean()->nullable(),
)
)
)->fallback( array() );
$entry = new Modules_State_Entry( array_merge( Features_Index::FEATURES, Features_Index::SUB_FEATURES ) );
$instance->register( 'modules_state', $modules_state_schema, $entry );
}
/**
* Initialize the modules.
*
* @param Module[] $modules The modules to initialize.
*/
private function init_modules( array $modules ) {
foreach ( $modules as $slug => $module ) {
if ( ! $module->is_enabled() ) {
continue;
}
Setup::add( $module->feature );
$submodules = $module->get_available_submodules();
if ( ! empty( $submodules ) ) {
$this->init_modules( $submodules );
}
$this->register_endpoints( $module->feature );
do_action( "jetpack_boost_{$slug}_initialized", $this );
}
}
/**
* @inheritDoc
*/
public function setup() {
// We need to setup data sync outside of plugins_loaded to prevent side effects on other classes that are loaded from other actions earlier.
self::register_data_sync( Data_Sync::get_instance( JETPACK_BOOST_DATASYNC_NAMESPACE ) );
$this->setup_features_data_sync();
$this->register_always_available_endpoints();
add_action( 'plugins_loaded', array( $this, 'load_modules' ) );
add_action( 'jetpack_boost_module_status_updated', array( $this, 'on_module_status_update' ), 10, 2 );
// Add a hook to fire page output changed action when a module that Changes_Output_After_Activation indicates something has changed.
foreach ( $this->get_available_modules_and_submodules() as $module ) {
$this->notice_page_output_change_of_module( $module );
}
}
private function notice_page_output_change_of_module( $module ) {
if ( ! $module->is_enabled() ) {
return;
}
$feature = $module->feature;
if ( ! ( $feature instanceof Changes_Output_After_Activation ) ) {
return;
}
$action_names = $feature::get_change_output_action_names();
if ( empty( $action_names ) ) {
return;
}
foreach ( $action_names as $action ) {
add_action( $action, array( $module, 'indicate_page_output_changed' ), 10, 1 );
}
}
/**
* Handle module status changes.
*
* @param string $module_slug The module slug.
* @param bool $is_activated The new status.
*/
public function on_module_status_update( $module_slug, $is_activated ) {
$modules = $this->get_available_modules_and_submodules();
if ( ! isset( $modules[ $module_slug ] ) ) {
return;
}
$module = $modules[ $module_slug ];
if ( ! $module ) {
return;
}
$status = new Status( $module_slug );
$status->on_update( $is_activated );
if ( $is_activated ) {
$module->on_activate();
} else {
$module->on_deactivate();
}
// Now run the activation/deactivation for all submodules that are effected by this modules status change.
$submodules = $module->get_available_submodules();
if ( is_array( $submodules ) ) {
foreach ( $submodules as $submodule ) {
// Only worry about submodules that are enabled.
if ( ! $submodule->is_enabled() ) {
continue;
}
$active_parent_modules = $submodule->get_active_parent_modules();
if ( $is_activated && count( $active_parent_modules ) === 1 ) {
// If current module is the only active parent module, run activation on the submodule.
// If this submodule has other parent modules, we can assume they are already activated.
$submodule->on_activate();
}
// If submodule has no active parent modules left, run deactivate on the submodule.
// If this submodule still has other parent modules, we can assume they are not ready to be deactivated.
if ( ! $is_activated && empty( $active_parent_modules ) ) {
$submodule->on_deactivate();
}
}
}
}
}