ok
Direktori : /opt/cloudlinux/venv/lib/python3.11/site-packages/clwpos/user/ |
Current File : //opt/cloudlinux/venv/lib/python3.11/site-packages/clwpos/user/wpos_user.py |
# -*- coding: utf-8 -*- # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2021 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT """ Implementation of clwpos-user CLI utility """ import argparse import os import pwd from itertools import chain from typing import Dict, List, Optional, Tuple from clcommon.clpwd import drop_user_privileges from clcommon.const import Feature as ModuleFeature from clcommon.cpapi import is_panel_feature_supported from clwpos.optimization_features.features import WP_BROKEN_CORE_REPAIR_TIP from packaging.version import parse as parse_version from clwpos import constants from clwpos import gettext as _ from clwpos.cl_wpos_exceptions import WposError from clwpos.daemon import WposDaemon from clwpos.data_collector_utils import ( get_cached_php_installed_versions, get_user_info, ) from clwpos.feature_suites import ( get_allowed_modules, get_allowed_features_dict ) from clwpos.feature_suites.configurations import get_visible_modules, get_visible_features_dict from clwpos.logsetup import ( setup_logging, USER_LOGFILE_PATH, init_wpos_sentry_safely ) from clwpos.object_cache.redis_utils import ( get_cached_php_versions_with_redis_loaded, reload_redis ) from clwpos.optimization_features import ( OBJECT_CACHE_FEATURE, SITE_OPTIMIZATION_FEATURE, CDN_FEATURE, ALL_OPTIMIZATION_FEATURES, CompatibilityIssue, Feature, UniqueId, convert_feature_list_to_interface, disable_without_config_affecting, enable_without_config_affecting, DomainName ) from clwpos.papi import get_subscriptions_info, is_feature_hidden_server_wide from clwpos.parse import ArgumentParser, CustomFormatter from clwpos.scoped_cache import enable_caching from clwpos.user.config import UserConfig, LicenseApproveStatus from clwpos.user.progress_check import ( CommandProgress, track_command_progress, update_command_progress, ) from clwpos.user.redis_lib import RedisLibUser from clwpos.user.website_check import post_site_check, RollbackException from clwpos.utils import ( USER_WPOS_DIR, check_license_decorator, print_data, catch_error, error_and_exit, daemon_communicate, home_dir, user_name, user_uid, check_domain, is_run_under_user, supported_php_handlers, _run_clwpos_as_user_in_cagefs, is_redis_configuration_running, redis_is_running, get_server_wide_options ) from clwpos.wp_utils import wp_get_constant, php_info, update_redis_disable_banners_constant, WordpressError logger = setup_logging(__name__) LSWS_INCOMPATIBLE_FEATURES = (OBJECT_CACHE_FEATURE, CDN_FEATURE) parser = ArgumentParser( "/usr/bin/clwpos-user", "Utility for control CL AccelerateWP under user", formatter_class=CustomFormatter ) if not is_run_under_user(): parser.add_argument('--user', default=None, required=True) class MaxCacheMemory(int): """ Class to validate format and values of cache memory setted by user. """ def __new__(cls, *args, **kwargs): try: instance = super(MaxCacheMemory, cls).__new__(cls, *args, **kwargs) except ValueError: raise argparse.ArgumentTypeError("invalid value type, must be integer") min_memory = constants.MINIMUM_MAX_CACHE_MEMORY max_memory = constants.MAXIMUM_MAX_CACHE_MEMORY if not min_memory <= instance <= max_memory: raise argparse.ArgumentTypeError( f"value must be in range: [{min_memory}, {max_memory}]") return instance class CloudlinuxWposUser: """ Class for run clwpos-user utility commands """ COMMAND_RELOAD_DICT = {"command": "reload"} def __init__(self): self._is_json = False self._opts = None self._wp_config_suffix = "wp-config.php" self._is_redis_running = None self._supported_handlers = None self._supported_php_versions = None self._minimum_wordpress_versions = None self._php_versions_with_redis_not_loaded = None self.command_progress: Optional[CommandProgress] = None init_wpos_sentry_safely(logger) @property def redis_socket_path(self): return os.path.join(home_dir(), USER_WPOS_DIR, 'redis.sock') @property def supported_handlers(self): if self._supported_handlers is None: self._supported_handlers = supported_php_handlers() return self._supported_handlers @property def is_redis_running(self): if self._is_redis_running is None: self._is_redis_running = redis_is_running() return self._is_redis_running @property def supported_php_versions(self): """ Return dict where key is optimization feature and values is list of supported php version (for object-cache with redis enabled). """ if self._supported_php_versions is None: supported_php_versions = { OBJECT_CACHE_FEATURE: [ php_version.identifier for php_version in get_cached_php_versions_with_redis_loaded() if OBJECT_CACHE_FEATURE.is_php_supported(php_version) ], SITE_OPTIMIZATION_FEATURE: [ php_version.identifier for php_version in get_cached_php_installed_versions() if SITE_OPTIMIZATION_FEATURE.is_php_supported(php_version) ] } self._supported_php_versions = supported_php_versions return self._supported_php_versions @catch_error def run(self, argv): """ Run command action :param argv: sys.argv[1:] :return: clwpos-user utility retcode """ self._opts = self._parse_args(argv) self._is_json = True if not is_run_under_user(): _run_clwpos_as_user_in_cagefs(self._opts.user) else: _run_clwpos_as_user_in_cagefs() if not is_run_under_user() and not is_panel_feature_supported(ModuleFeature.CAGEFS): drop_user_privileges(self._opts.user, effective_or_real=False, set_env=True) result = getattr(self, self._opts.command.replace("-", "_"))() print_data(self._is_json, result) def _parse_args(self, argv): raise NotImplementedError def _check_monitoring_daemon_and_exit(self, feature): """ Ensures monitoring socket is present, otherwise - exit """ if feature == OBJECT_CACHE_FEATURE \ and not os.path.exists(constants.WPOS_DAEMON_SOCKET_FILE): error_and_exit( self._is_json, { "result": _("Unable to find monitoring daemon socket %(daemon_file)s. Please, contact your " "system administrator to ensure service %(service_name)s is currently running"), "context": {"daemon_file": constants.WPOS_DAEMON_SOCKET_FILE, "service_name": constants.MONIROTING_SERVICE}, } ) def _check_redis_configuration_process_and_exit(self, feature): """ Ensures redis configuration processes are not in progress. """ if feature == OBJECT_CACHE_FEATURE and is_redis_configuration_running(): error_and_exit( self._is_json, { "result": _("Configuration of PHP redis extension is in progress. " "Please wait until the end of this process to use AccelerateWP."), } ) def _is_redis_daemon_reload_needed(self, user_config: UserConfig, feature: Feature, before_config_modified=False) -> bool: """ Determine if it is needed to send reload command to Wpos redis daemon. Such reload should be done after user has enabled object caching for the first of all his sites, or disable for the last. """ command = self._opts.command if feature != OBJECT_CACHE_FEATURE: return False sites_count_with_enabled_module = user_config.get_enabled_sites_count_by_modules([feature]) force_reload = before_config_modified and (command == 'enable' and sites_count_with_enabled_module == 0) if force_reload \ or (not before_config_modified and command == 'enable' and sites_count_with_enabled_module == 1) \ or (command == 'disable' and sites_count_with_enabled_module == 0): logger.info('Redis needs reloading, command: "%s", sites count: "%s", force reload: %s', command, str(sites_count_with_enabled_module), str(force_reload)) return True return False @staticmethod def _get_php_versions_handlers_pair_for_docroot(domains_per_docroot: List, php_data: List) -> Tuple: """ Returns pair (all_php_versions, all_php_handlers) for domains in docroot """ versions, handlers = set(), set() for item in php_data: domain = item['vhost'] if domain not in domains_per_docroot: continue versions.add(item['version']) handlers.add(item['handler']) return versions, handlers def collect_general_issues(self, module_name: str): """ Collects general (not depending on worpress/docroot setup) server misconfigurations/incompatibilities with WPOS """ issues = [] return issues @update_command_progress def collect_docroot_issues(self, module: Feature, doc_root_info: Dict, php_info: List, visible_features: List): """ Collects incompatibilities related to docroot (non-supported handler, etc) """ issues = [] domains_per_docroot = doc_root_info['domains'] versions, handlers = self._get_php_versions_handlers_pair_for_docroot(domains_per_docroot, php_info) if len(versions) > 1: issues.append( CompatibilityIssue( header=_('Different PHP versions for domains'), description=_('Those domains: %(domains)s are in same docroot, ' 'but using different PHP version'), fix_tip=_('Set or ask your system administrator to set same PHP version on those domains'), context={ 'domains': ', '.join(domains_per_docroot) }, unique_id=UniqueId.PHP_MISCONFIGURATION, telemetry=dict( reason='PHP_VERSION_BADLY_CONFIGURED', domains=domains_per_docroot ) ) ) if len(handlers) > 1: issues.append( CompatibilityIssue( header=_('Different PHP handlers for domains'), description=_('Those domains: %(domains)s are in same docroot, ' 'but using different PHP handlers'), fix_tip=_('Set or ask your system administrator to set same PHP handler on those domains'), context={ 'domains': ', '.join(domains_per_docroot) }, unique_id=UniqueId.PHP_MISCONFIGURATION, telemetry=dict( reason='PHP_HANDLER_BADLY_CONFIGURED', domains=domains_per_docroot ) ) ) additional_issues = module.collect_docroot_issues(self, doc_root_info, visible_features=visible_features) issues.extend(additional_issues) return issues @staticmethod def _is_our_roc_plugin(wp_path: str) -> bool: """ Checks that WP's WP_REDIS_PATH is defined and contains our path """ wp_redis_path = wp_get_constant(wp_path, 'WP_REDIS_PATH', raise_exception=True) if wp_redis_path and "/.clwpos/redis.sock" in wp_redis_path: return True return False def check_installed_roc_plugin(self, wp_path: str) -> bool: """ Checks that ROC plugin was installed by us, or not exists at all Returns False if the plugin was found but our config modifications was not """ if os.path.exists(os.path.join(wp_path, 'wp-content', 'plugins', 'redis-cache')): if not self._is_our_roc_plugin(wp_path): return False return True @update_command_progress def collect_wordpress_issues( self, module_name: Feature, wordpress_info: Dict, docroot: str, module_is_enabled: bool ): """ Collects incompatibilities related to wordpress setup (conflicting plugin enabled, etc) """ issues = [] # Implicitly assume a module's version string is properly formatted minimum_supported_wp_version = parse_version(module_name.minimum_supported_wp_version()) wp_version = wordpress_info.get("version") if not wp_version: issues.append( CompatibilityIssue( header=_('Failed to determine WordPress version'), description=_( 'Optimization feature couldn\'t be applied because ' 'WordPress installation at %(wp_path)s seems to be ' 'damaged or misconfigured.\n' 'Details: %(error_details)s', ), fix_tip=WP_BROKEN_CORE_REPAIR_TIP, context=dict( wp_path=docroot, error_details=wordpress_info.get( 'version_missing_reason', # We expect version is None -> version_missing_reason is not None, # but no guaranties on other side. # FIXME: turn "version" into proper algebraic type once the code # is properly typed 'wp-include/version.php file is invalid', ), ), unique_id=UniqueId.MISCONFIGURED_WORDPRESS, telemetry=dict(minimum_supported_wp_version=str(minimum_supported_wp_version)), ) ) elif wp_version < minimum_supported_wp_version: issues.append( CompatibilityIssue( header=_('Unsupported WordPress version'), description=_('Optimization feature is incompatible with ' 'WordPress version %(wp_version)s being used.'), fix_tip=_('The minimal supported WordPress version is %(minimum_supported_wp_version)s. ' 'Upgrade your WordPress installation or reinstall it from the scratch.'), context=dict( minimum_supported_wp_version=str(minimum_supported_wp_version), wp_version=str(wp_version) ), unique_id=UniqueId.UNCOMPATIBLE_WORDPRESS_VERSION, telemetry=dict( minimum_supported_wp_version=str(minimum_supported_wp_version), wp_version=str(wp_version) ) )) issues.extend(module_name.collect_wordpress_issues(self, wordpress_info, docroot, module_is_enabled)) return issues @parser.argument('--listen', action='store_true', required=True) @parser.argument('--domain', required=False, default=None) @parser.argument('--wp-path', required=False, default=None) @parser.argument('--feature', required=True) @parser.argument('--ignore-errors', required=False, default=False, action='store_true') @parser.argument('--advice-id', type=int, required=False, default=None) @parser.command() def subscription(self): if any((self._opts.domain, self._opts.wp_path is not None)) \ and not all((self._opts.domain, self._opts.wp_path is not None)): error_and_exit(self._is_json, {'result': "Both domain and wp_path arguments " "are requited if one of them is set"}) features = self._opts.feature.split(',') for f in features: if f not in [feature.to_interface_name() for feature in ALL_OPTIMIZATION_FEATURES]: error_and_exit(self._is_json, {'result': "Unsupported feature passed"}) daemon_communicate({ 'command': WposDaemon.DAEMON_REGISTER_UPGRADE_ATTEMPT, 'domain': self._opts.domain, 'wp_path': self._opts.wp_path, # comma separated str: "object_cache,critical_css,.." 'feature': self._opts.feature, 'ignore_errors': self._opts.ignore_errors, 'advice_id': self._opts.advice_id, }) return {} @parser.argument('--status', '-s', help='Status of license agreements', action='store_true') @parser.argument('--approve', '-a', help='Approve license agreement') @parser.argument('--text', '-t', help='Read and approve license agreement') @parser.command() def agreement(self): if self._opts.status: uc = UserConfig(user_name()) result = {} for feature in ALL_OPTIMIZATION_FEATURES: result[feature.NAME] = uc.get_license_approve_status(feature).name return {'licenses': result} elif self._opts.approve: feature = Feature(self._opts.approve) uc = UserConfig(user_name()) uc.approve_license_agreement(feature) return { 'status': "License agreement accepted" } elif self._opts.text: feature = Feature(self._opts.text) if not feature.HAS_LICENSE_TERMS: return { 'result': "NO_LICENSE_TERMS" } return { 'text': open(feature.LICENSE_TERMS_PATH).read() } else: raise NotImplementedError('Unknown command') @track_command_progress def _get(self): allowed_features_dict = get_allowed_features_dict(user_uid()) visible_features_dict = get_visible_features_dict(user_uid()) visible_features = list(chain(*visible_features_dict.values())) converted_allowed_features = {feature_set: convert_feature_list_to_interface(features) for feature_set, features in allowed_features_dict.items()} converted_visible_features = {feature_set: convert_feature_list_to_interface(features) for feature_set, features in visible_features_dict.items()} with enable_caching(): phpinfo = php_info() if not phpinfo: if visible_features: # empty php_info with any module allowed means that something went wrong error_and_exit( self._is_json, {"result": _("Failed to retrieve data about php version which is currently used. " "Contact your administrator if the issue persists.")}, ) # no modules allowed and the fact that we cant retrieve php_info # probably means that Wpos was never allowed for user return {"docroots": [], "allowed_features": converted_allowed_features, "visible_features": converted_visible_features} self.command_progress.update() user_info = self._get_user_info() visible_modules = get_visible_modules(user_uid()) self.command_progress.recalculate_number_of_total_stages(user_info) self.command_progress.update() # Collect issues which depends only on a feature general_feat_issues = {} for feat in ALL_OPTIMIZATION_FEATURES: if feat in get_server_wide_options().hidden_features: logger.warning('Optimization feature "%s" is skipped, because it is hidden by administrator') continue general_feat_issues[feat] = self.collect_general_issues(feat) # Collect issues which depends on a docroot and feature docroots_features_issues = { docroot: { feat: self.collect_docroot_issues( feat, doc_root_info, phpinfo, visible_features, ) for feat in general_feat_issues.keys() } for docroot, doc_root_info in user_info.items() } # Collect WordPress installation issues for each feature and gather result docroots = [] for docroot, doc_root_info in user_info.items(): wps_out = [] for wp in doc_root_info["wps"]: feats = {} for feat in ALL_OPTIMIZATION_FEATURES: is_enabled = self._is_enabled(doc_root_info["domains"][0], wp["path"], feat) module_info = {"enabled": is_enabled, "visible": feat in visible_modules} wordpress_issues = self.collect_wordpress_issues( feat, wp, docroot, module_is_enabled=is_enabled ) issues = [ *general_feat_issues[feat], *docroots_features_issues[docroot][feat], *wordpress_issues, ] if issues: module_info.update({"issues": [ issue.dict_repr for issue in issues ]}) feats[feat.to_interface_name()] = module_info wps_out += [ { "path": wp["path"], "version": wp["version"] or "UNKNOWN", "features": feats, }, ] docroots += [{ **doc_root_info, "wps": wps_out, # convert output to old defined format "php_version": doc_root_info["php_version"].identifier, }] subscriptions = get_subscriptions_info(user_name()) return { "docroots": docroots, "allowed_features": converted_allowed_features, "visible_features": converted_visible_features, "upgrade_url": { feature.lower(): self._get_upgrade_url(feature.lower()) for feature in {OBJECT_CACHE_FEATURE.NAME, CDN_FEATURE.NAME} }, "subscription": subscriptions, "used_memory": RedisLibUser(self.redis_socket_path).get_redis_used_memory(), "max_cache_memory": UserConfig(user_name()).get_config().get( "max_cache_memory", UserConfig.DEFAULT_MAX_CACHE_MEMORY ) } def _get_upgrade_url(self, feature): try: upgrade_url = daemon_communicate({ "command": WposDaemon.DAEMON_GET_UPGRADE_LINK_COMMAND, "feature": feature })["upgrade_url"] except WposError: upgrade_url = None return upgrade_url @parser.argument('--website', type=str, help='Website to scan', required=True) @parser.command() def scan(self): """ ATTENTION: --website = actually domain, -> pass domain name instead of website !!! """ phpinfo = php_info() u_id = user_uid() allowed_modules, visible_modules = get_allowed_modules(u_id), get_visible_modules(u_id) if not phpinfo: logger.error('Incompatibilities will not be collected! ' 'PHP info was not found for domain: %s', self._opts.website) return {"issues": {}} user_info = self._get_user_info() wps = {} for module_name in ALL_OPTIMIZATION_FEATURES: general_issues = self.collect_general_issues(module_name) if module_name not in visible_modules: general_issues.append( CompatibilityIssue( header=_('Feature was not made visible'), description=_('Optimization feature was not made visible to user'), fix_tip=_(f'Contact Administrator to allow using optimization feature'), unique_id=UniqueId.FEATURE_NOT_MADE_VISIBLE, telemetry=dict(reason='FEATURE_NOT_MADE_VISIBLE') ) ) if is_feature_hidden_server_wide(module_name): general_issues.append( CompatibilityIssue( header=_('Feature is hidden server wide'), description=_('Optimization feature is hidden in server wide settings'), fix_tip=_(f'Contact Administrator to make non-hidden'), unique_id=UniqueId.FEATURE_HIDDEN_SERVER_WIDE, telemetry=dict(reason='FEATURE_HIDDEN_SERVER_WIDE') ) ) for docroot, doc_root_info in user_info.items(): if self._opts.website not in doc_root_info["domains"]: continue docroot_issues = self.collect_docroot_issues(module_name, doc_root_info, phpinfo, allowed_modules) for wp in doc_root_info["wps"]: is_enabled = self._is_enabled(doc_root_info["domains"][0], wp["path"], module_name) if is_enabled: wps.setdefault(wp['path'], []).append({ 'type': UniqueId.CLOUDLINUX_MODULE_ALREADY_ENABLED, 'context': dict(), 'advice_type': module_name.NAME }) else: wordpress_issues = self.collect_wordpress_issues( module_name, wp, docroot, module_is_enabled=is_enabled) incompatibilities = [*general_issues, *docroot_issues, *wordpress_issues] wps.setdefault(wp['path'], []).extend([ { 'type': issue.unique_id, 'context': issue.telemetry, 'advice_type': module_name.NAME } for issue in incompatibilities if isinstance(issue, CompatibilityIssue) ]) # convert output to old defined format for document_root in user_info.values(): document_root["php_version"] = document_root["php_version"].identifier return { "issues": wps } def _get_user_info(self, docroot: str = None): def reformat_php_data(info): """""" php = info.pop("php") info["php_version"] = php["version"] info["php_handler"] = php["handler"] if docroot is None: user_info = get_user_info() for doc_root_info in user_info.values(): reformat_php_data(doc_root_info) else: user_info = get_user_info()[docroot] reformat_php_data(user_info) return user_info def get_current_issues_by_docroot( self, module_name: Feature, wordpress_path: str, docroot: str, is_module_enabled: bool ): """ Obtains issues for special docroot and wordpress """ wordpress_issues = [] allowed_modules = get_allowed_modules(user_uid()) general_issues = self.collect_general_issues(module_name) phpinfo = php_info() user_info_by_docroot = self._get_user_info(docroot) docroot_issues = self.collect_docroot_issues(module_name, user_info_by_docroot, phpinfo, allowed_modules) for wp in user_info_by_docroot["wps"]: if wp['path'] != wordpress_path: continue wordpress_issues = self.collect_wordpress_issues(module_name, wp, docroot, module_is_enabled=is_module_enabled) return [issue.dict_repr for issue in (*general_issues, *docroot_issues, *wordpress_issues)] def _is_enabled(self, domain: str, wp_path: str, module: str) -> bool: uc = UserConfig(user_name()) return uc.is_module_enabled(domain, wp_path, module) and module in get_allowed_modules(user_uid()) @parser.argument('--max_cache_memory', help='Maximum cache memory to use in MB', type=MaxCacheMemory, default=constants.DEFAULT_MAX_CACHE_MEMORY) @parser.command() @check_license_decorator def set(self): uc = UserConfig(user_name()) # TODO: this method must probably UPDATE config and not REPLACE it completely params = {"max_cache_memory": f"{self._opts.max_cache_memory}mb"} uc.set_params(params) daemon_communicate(self.COMMAND_RELOAD_DICT) return params @track_command_progress def _disable(self): username, doc_root = check_domain(self._opts.domain) wp_path = self._opts.wp_path.strip("/") uc = UserConfig(user_name()) errors = [] feature_name = self._opts.feature.optimization_feature() if not uc.is_module_enabled(self._opts.domain, wp_path, feature_name): return {"warning": _('Optimization feature %(feature)s is already disabled on the domain %(domain)s. ' 'Nothing to be done.'), "context": {"domain": self._opts.domain, "feature": self._opts.feature}} self._check_monitoring_daemon_and_exit(feature=feature_name) self.command_progress.update() last_error = disable_without_config_affecting( DomainName(self._opts.domain), wp_path, module=feature_name) if last_error: logger.error(last_error.message % last_error.context) errors.append(last_error) else: # do not change config values in case if disable reported errors try: uc.disable_module(self._opts.domain, wp_path, feature_name) except Exception as e: logger.exception("unable to disable module in config") errors.append(e) self.command_progress.update() if self._is_redis_daemon_reload_needed(uc, feature=feature_name): try: reload_redis() except WposError as e: s_details = e.details % e.context logger.exception("CLWPOS daemon error: '%s'; details: '%s'", e.message, s_details) errors.append(e) except Exception as e: logger.exception("unable to reload cache backend") errors.append(e) self.command_progress.update() is_module_enabled = self._is_enabled(self._opts.domain, wp_path, feature_name) is_module_visible = feature_name in get_visible_modules(user_uid()) try: last_error = errors.pop(-1) except IndexError: last_error = None if is_module_enabled: if last_error: raise WposError( message=_("Optimization feature disabling failed because one or more steps reported error. " "Caching is still active, but may work unstable. Try disabling it again. " "Contact your system administrator if this issue persists. " "Detailed information you can find in log file '%(log_path)s'"), details=last_error.message, context={ 'log_path': USER_LOGFILE_PATH.format(homedir=home_dir()), **getattr(last_error, 'context', {}) } ) else: error_and_exit( self._is_json, {"result": _("WordPress caching module is still enabled, but no errors happened. " "Try again and contact your system administrator if this issue persists.")}, ) else: response = { "feature": { "enabled": is_module_enabled, "visible": is_module_visible }, "used_memory": RedisLibUser(self.redis_socket_path).get_redis_used_memory() } if last_error: response.update( { "warning": _("Optimization feature disabled, but one or more steps reported error. " "Detailed information you can find in log file '%(log_path)s'"), "context": { "log_path": USER_LOGFILE_PATH.format(homedir=home_dir()) } } ) else: issues = self.get_current_issues_by_docroot(feature_name, wp_path, doc_root, is_module_enabled) if issues: response["feature"]["issues"] = issues return response @parser.argument('--feature', type=str, help='List of AccelerateWP optimization features separated by commas ' 'on which to perform an action', choices=ALL_OPTIMIZATION_FEATURES, default=OBJECT_CACHE_FEATURE) @parser.argument('--action', type=str, help='Action to perform on AccelerateWP optimization features', required=True) @parser.command() def do_action(self): """ Perform action on selected AccelerateWP features """ action = self._opts.action feature = self._opts.feature # dictionary where key - action, value - function that provides that action action_to_func = {"purge": self.redis_purge} if action not in action_to_func: error_and_exit( self._is_json, { "result": f'Invalid action "{action}", currently only "purge" action is supported' }, ) return action_to_func[action](feature) # perform action on module def redis_purge(self, *args): """ Clean entire redis cache for user. """ return RedisLibUser(self.redis_socket_path).purge_redis() @check_license_decorator @track_command_progress def _enable(self): """ Enable object_cache for user with compliance with end-user spec. :return: """ wp_path = self._opts.wp_path.strip("/") username, doc_root = check_domain(self._opts.domain) make_web_check = not any([self._opts.ignore_errors, self._opts.skip_dns_check]) abs_wp_path = os.path.join(doc_root, wp_path) uc = UserConfig(username) feature_name: Feature = self._opts.feature.optimization_feature() # bypass to allow approvements in UI which has checkbox # asking "do you agree" right before we install the feature if self._opts.approve_license_terms: uc.approve_license_agreement(feature_name) if uc.get_license_approve_status(feature_name) == LicenseApproveStatus.NOT_APPROVED: raise WposError('License approve required to use this feature. ' 'Open AccelerateWP plugin in your control panel, enable feature and ' 'accept terms and conditions to proceed with installation.') included_features = feature_name.included_optimization_features() is_module_enabled = all([self._is_enabled(self._opts.domain, wp_path, feature) for feature in included_features]) if is_module_enabled: return {"warning": _("The %(feature)s optimization feature is already enabled on " "the domain %(domain)s. Nothing to be done."), "context": {"domain": self._opts.domain, "feature": self._opts.feature}} if feature_name in get_server_wide_options().hidden_features: error_and_exit( self._is_json, { "result": _("Optimization feature %(feature)s is hidden by administrator."), "context": {"feature": self._opts.feature}, } ) is_feature_visible = all([feature in get_visible_modules(user_uid()) for feature in included_features]) is_feature_allowed = all([feature in get_allowed_modules(user_uid()) for feature in included_features]) # check that enabling of module is allowed by admin if not is_feature_allowed: if is_feature_visible: from clwpos.daemon import WposDaemon error_and_exit( self._is_json, { "result": _("PAYMENT_REQUIRED"), "upgrade_url": daemon_communicate({ "command": WposDaemon.DAEMON_GET_UPGRADE_LINK_COMMAND, 'feature': feature_name })["upgrade_url"], # keep it here so UI can still highlight it "context": {"feature": self._opts.feature}, } ) else: error_and_exit( self._is_json, { "result": _("Usage of the optimization feature %(feature)s is prohibited by admin."), "context": {"feature": self._opts.feature}, } ) self._check_redis_configuration_process_and_exit(feature=feature_name) self._check_monitoring_daemon_and_exit(feature=feature_name) self.command_progress.update() with enable_caching(): # check that user's wordpress fits to requirements issues = self.get_current_issues_by_docroot( feature_name, wp_path, doc_root, is_module_enabled) if issues: error_and_exit( self._is_json, {'feature': {'enabled': is_module_enabled, 'issues': issues}, 'result': _('Website "%(domain)s/%(wp_path)s" has compatibility ' 'issues and optimization feature cannot be enabled.'), 'context': {'domain': self._opts.domain, 'wp_path': wp_path}}, ) # check user's site before installation if make_web_check: self._website_check(abs_wp_path, self._opts.domain, uc, wp_path, feature_name) self.command_progress.update() # # # redis object cache plugin checks connectivity to redis during activation if self._is_redis_daemon_reload_needed(uc, feature=feature_name, before_config_modified=True): reload_redis(uid=pwd.getpwnam(username).pw_uid, force='yes') self.command_progress.update() # try to enable module with wp-cli without adding info into user's wpos config ok, enable_failed_info = enable_without_config_affecting( DomainName(self._opts.domain), wp_path, module=feature_name, ignore_errors=self._opts.ignore_errors ) if not ok: # reload once again, because plugin activation failed if self._is_redis_daemon_reload_needed(uc, feature=feature_name, before_config_modified=True): reload_redis() if 'result' in enable_failed_info: error_and_exit(self._is_json, message=enable_failed_info) raise WposError(**enable_failed_info) for feature in included_features: uc.enable_module(self._opts.domain, wp_path, feature) self.command_progress.update() # check user's site after installation and enabling if make_web_check: self._website_check(abs_wp_path, self._opts.domain, uc, wp_path, feature_name, rollback=True) self.command_progress.update() is_module_enabled = all([self._is_enabled(self._opts.domain, wp_path, feature) for feature in included_features]) with enable_caching(): issues = self.get_current_issues_by_docroot(feature_name, wp_path, doc_root, is_module_enabled) module_data = { 'enabled': is_module_enabled } if issues: module_data.update({ 'issues': issues }) return { 'feature': module_data, 'used_memory': RedisLibUser(self.redis_socket_path).get_redis_used_memory() } def _website_check(self, abs_wp_path, domain, uc, wp_path, module, rollback=False): """ Performs website availability checks and raises errors in case of problems :param abs_wp_path: absolute path to wordpress installation :param domain: domain that wordpress installation belongs to :param uc: user config instance :param wp_path: relative path to wordpress installation :param rollback: whether to roll redis and plugin changes back :return: """ try: post_site_check(domain, wp_path, abs_wp_path) except WposError as e: if rollback: uc.disable_module(domain, wp_path, module) reload_redis() Feature(module).disable(abs_wp_path) if isinstance(e, RollbackException): raise error_and_exit( is_json=True, message={ "context": {}, "result": _("Unexpected error occurred during plugin installation for WordPress. " "Try again and contact your system administrator if the issue persists."), "details": str(e) }, ) @parser.command() def get_progress(self): """ Return current progress of the currently performed command. Provides amount of: - total stages - how many stages the command execution process is divided into, - completed stages - how many stages have been already completed. """ return CommandProgress.get_status() def _object_cache_banner(self): disable_object_cache_banners = False if self._opts.disable: disable_object_cache_banners = True if self._opts.all: user_info = self._get_user_info() for docroot, doc_root_info in user_info.items(): for wp in doc_root_info["wps"]: self._object_cache_banner_toggle(disable_object_cache_banners, docroot, wp["path"], doc_root_info["domains"][0]) return {} elif self._opts.domain: domain = self._opts.domain try: username, docroot = check_domain(domain) except Exception as e: raise WposError( message=_("Can't find docroot for domain '%(domain)s'"), context={"domain": domain} ) self._object_cache_banner_toggle(disable_object_cache_banners, docroot, self._opts.wp_path, self._opts.domain) return {} else: raise NotImplementedError('Unknown command') def _object_cache_banner_toggle(self, disable_banner: bool, abs_wp_path: str, wp_path: str, domain: str): module_enabled = self._is_enabled(domain, wp_path, OBJECT_CACHE_FEATURE) if module_enabled is False: constant_value = None else: constant_value = str(disable_banner).lower() abs_wp_path = os.path.join(abs_wp_path, wp_path) # abs_wp_path is <docroot>+<wp_name> res = update_redis_disable_banners_constant(abs_wp_path, constant_value) if isinstance(res, WordpressError): raise WposError(message=res.message, context=res.context)