ok
Mini Shell
# -*- 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
from __future__ import absolute_import
import json
import os
import pwd
import re
import subprocess
from dataclasses import dataclass
from typing import Optional, Union, List
from functools import lru_cache
from typing_extensions import TypedDict
from clwpos import gettext as _
from clwpos.cl_wpos_exceptions import WposError, WpCliCommandError
from clwpos.constants import RedisRequiredConstants, WP_CLI, WP_CLI_SKIP_PLUGINS_ENV
from clwpos.data_collector_utils import php_info
from clwpos.logsetup import (
setup_logging,
ADMIN_LOGFILE_PATH,
USER_LOGFILE_PATH
)
from clwpos.php.base import PHP
from clwpos.scoped_cache import cached_in_scope
from clwpos.user.cache import wp_config_cache
from clwpos.user.website_check import RollbackException, WebsiteCheckError
from clwpos.user.website_check.errors import (
WebsiteCheckBadHttpCode,
CDNActivationFailed,
JSCssCheckBadHttpCode
)
from clwpos.utils import (
WposUser,
is_run_under_user,
wp_cli_compatibility_check,
run_in_cagefs_if_needed,
user_name,
litespeed_is_running
)
_logger = setup_logging(__name__)
@dataclass
class WordpressError:
message: str
context: dict
code: Optional[int] = None
def get_php_version(abs_wp_path: str) -> PHP:
"""
Return php_version that will be used for calling wp-cli commands.
If 'CLWPOS_USE_SAVED_PHP_VERSION' envar is defined,
try to get this version from a previously saved file.
"""
use_saved_php_version = bool(os.environ.get("CLWPOS_USE_SAVED_PHP_VERSION"))
if use_saved_php_version:
php_version = _get_saved_php_version(abs_wp_path) or _get_php_version(abs_wp_path)
else:
php_version = _get_php_version(abs_wp_path)
return php_version
@lru_cache(maxsize=None)
def _get_php_version(abs_wp_path: str) -> PHP:
"""Return PHP version."""
result = php_info()
items = []
for item in result:
if abs_wp_path.startswith(item["documentroot"]):
items.append((item["documentroot"], item["version"]))
items.sort(reverse=True)
return items[0][1]
def _get_saved_php_version(abs_wp_path: str) -> Optional[PHP]:
"""
Get domain's php version from a previously saved file.
"""
if not is_run_under_user():
raise WposError('Internal Error. Contact CloudLinux support')
php_file_id = os.environ.get("CLWPOS_PHP_FILE_ID")
php_info_file = WposUser(user_name()).php_info.format(file_id=php_file_id)
if not os.path.exists(php_info_file):
return None
try:
with open(php_info_file) as f:
_php_info = json.load(f)
except (OSError, json.decoder.JSONDecodeError) as e:
_logger.exception("Error during reading of \".php_info\" file: %s", e)
return None
php_versions = []
for vhost_info in _php_info:
if abs_wp_path.startswith(vhost_info["documentroot"]):
php_versions.append((vhost_info["documentroot"], vhost_info["version"]))
if not php_versions:
return None
return PHP(**sorted(
php_versions,
key=lambda item: item[1]['identifier'],
reverse=True
)[0][1])
def wordpress(path: str, command: str, subcommand: str, *args, env=None) -> Union[str, WordpressError]:
"""
Helper to execute wp commands, for example
wp --path=<path> plugin install redis-cache
wp --path=<path> plugin activate redis-cache
wp --path=<path> redis enable
wp --path=<path> plugin deactivate redis-cache
wp --path=<path> plugin uninstall redis-cache
@return: stderr if error was happened.
"""
php_version = get_php_version(path)
php_bin_path = str(php_version.bin)
if not os.path.exists(php_bin_path):
_logger.exception("Error during wp-cli command execution \"%s\": "
"invalid path to binary file \"%s\"",
command, php_bin_path)
return WordpressError(
message=_("Error during resolving path to php binary file:\n"
"got non-existent path \"%(path)s\"."),
context={"path": php_bin_path}
)
# [attention] compatibility check may raise WpCliUnsupportedException exception
wp_cli_compatibility_check(php_version)
command_part = [command, subcommand, *args]
full_command = [
WP_CLI,
php_bin_path,
path,
*command_part
]
environment = env or {}
try:
output = run_in_cagefs_if_needed(full_command, check=True, env=environment)
except subprocess.CalledProcessError as error:
post_check_flag = 'Post-check failed' in error.stderr
command = ' '.join(full_command)
_logger.exception("Error during command execution: \n%s\n"
"stdout=%s\n"
"stderr=%s",
command, error.stdout, error.stderr)
logger_path = ADMIN_LOGFILE_PATH if not os.getuid() else USER_LOGFILE_PATH.format(
homedir=pwd.getpwuid(os.getuid()).pw_dir)
if post_check_flag:
_process_post_check(error, logger_path)
else:
message = _("Unexpected error happened during command execution: '%(command)s'.\n"
"Event is logged to file: '%(logger_path)s' with stdout and stderr recorded.")
context = {
"command": command,
"logger_path": logger_path,
"error_desc": error.stderr
}
return WordpressError(
message=message,
context=context,
code=error.returncode
)
return output.stdout
def _process_post_check(
result: subprocess.CalledProcessError,
log_path: str):
"""
In some rare cases plugin installation (CDN) may return
post-check errors which we should handle properly.
"""
# case 1: plugin did not replace static links on website
# with cdn urls for some reason
cnd_not_active_msg = "CDN url not found on page"
if cnd_not_active_msg in result.stderr:
raise RollbackException(CDNActivationFailed(log_path))
# case 2: website returned error code during check
# URL ' . $this->url . ' returned unexpected response status (' . $this->url_code . ‘)’
url_match = 'URL '
unexpected_error_match = 'returned unexpected response status'
if url_match in result.stderr and unexpected_error_match in result.stderr:
# "URL https://.../ ", " status (300)"
url_part, code_part = result.stderr.split(unexpected_error_match)
url = url_part.split()[1]
code = code_part.strip("( )’'")
raise RollbackException(
WebsiteCheckBadHttpCode(url, code)
)
# case 3: website itself works fine, but plugin was not able to reach static files
# CDN url returned unexpected response status (' . $this->static_code . ’)
unexpected_error_static_content = 'CDN url returned unexpected response status'
if unexpected_error_static_content in result.stderr:
# "URL https://.../ ", " status (300)"
code_part = result.stderr.split('(')[1]
code = code_part.strip("( )’'")
raise RollbackException(
JSCssCheckBadHttpCode(code)
)
# case 4: unknown error
message = _("WordPress plugin failed to activate correctly. "
"Changes were reverted and caching module is now disabled.\n"
"Error reported from plugin: \n"
"%(error_desc)s.\n\n"
"Event is logged to file: '%(logger_path)s' with stdout and stderr recorded.")
context = {
"command": result.cmd,
"logger_path": log_path,
"error_desc": result.stdout + result.stderr
}
raise RollbackException(WebsiteCheckError(
header='Post check failed',
description=message,
context=context,
fix_tip='',
))
def is_multisite(path: str) -> bool:
marker = 'cl_multisite_detected'
command = 'if ( is_multisite() ) { echo "%s"; } else { echo "not_multisite"; }' % marker
with wp_config_cache('is_multisite', path=path) as record:
if record.data:
return marker in record.data
result = wordpress(path, 'eval', command, env={WP_CLI_SKIP_PLUGINS_ENV: '1'})
if isinstance(result, WordpressError):
raise WposError(message=result.message, context=result.context)
record.data = result
return marker in result
def wp_get_constant(wp_path: str, constant: str, raise_exception=False) -> Optional[str]:
"""
Get:
- defined constant value
- None in case of error
- empty string if no such constant found
"""
command = "if (defined('%(const)s')) { echo %(const)s; }" % {'const': constant}
with wp_config_cache('const.' + constant, path=wp_path) as record:
if record.data:
return record.data
result = wordpress(wp_path, 'eval', command, env={WP_CLI_SKIP_PLUGINS_ENV: '1'})
if isinstance(result, WordpressError):
if raise_exception:
raise WpCliCommandError(message=result.message, context=result.context)
_logger.error('Error during get WP constant: %s', result)
return None
record.data = result
return result
def diagnose_redis_connection_constants(docroot: str, wordpress_path: str):
"""
Check required constants for redis connection establishment
"""
redis_schema = wp_get_constant(os.path.join(docroot, wordpress_path), RedisRequiredConstants.WP_REDIS_SCHEME.name,
raise_exception=True)
if not redis_schema and redis_schema != RedisRequiredConstants.WP_REDIS_SCHEME.val:
raise WposError('WordPress constant "%(constant)s" is not defined or defined with wrong value %(value)s',
context={'constant': RedisRequiredConstants.WP_REDIS_SCHEME.name, 'value': redis_schema})
socket = wp_get_constant(os.path.join(docroot, wordpress_path), RedisRequiredConstants.WP_REDIS_PATH.name,
raise_exception=True)
if not socket:
raise WposError('WordPress constant "%(constant)s" is not defined',
context={'constant': RedisRequiredConstants.WP_REDIS_PATH.name})
if not os.path.exists(socket):
raise WposError('Redis socket %(socket)s does not exist in the system',
context={'socket': socket})
def obtain_wp_cli_env(abs_wp_path: str):
"""
Returns needed envars for wp-cli
"""
env = None
if os.path.exists(os.path.join(abs_wp_path, '.htaccess')):
server_software = 'LiteSpeed' if litespeed_is_running() else 'Apache'
env = {'SERVER_SOFTWARE': server_software}
return env
class PluginInfo(TypedDict):
# e.g. "clsop",
name: str
# e.g. "active",
status: str
# e.g. "none",
update: str
# e.g. "3.12.6.1-1-2"
version: str
@cached_in_scope
def _cached_plugin_list(abs_wp_path: str, wp_cli_env=None) -> List[PluginInfo]:
if wp_cli_env:
wp_cli_env = dict(wp_cli_env)
wp_cli_env.update({WP_CLI_SKIP_PLUGINS_ENV: '1'})
else:
wp_cli_env = {WP_CLI_SKIP_PLUGINS_ENV: '1'}
result = wordpress(abs_wp_path, "plugin", "list", "--json", env=wp_cli_env)
# TODO: raise exceptions instead of silent errors
if isinstance(result, WordpressError):
return []
# Try to find json, when output contains php errors
def _maybe_json(output):
_result = None
pattern = r'(\[{.*?}\])'
matches = re.search(pattern, output)
if matches:
try:
json_string = matches.group(1)
_result = json.loads(json_string)
except Exception:
pass
return _result
try:
return json.loads(result)
except (ValueError, TypeError, json.JSONDecodeError) as e:
maybe_json_result = _maybe_json(result)
if maybe_json_result is not None:
return maybe_json_result
raise WposError(
message=_(
'Malformed plugins information received from wp-cli. '
'Raw response is %(response)s.'),
context={'response': result},
details=str(e)
)
def plugin_list(abs_wp_path: str, wp_cli_env=None) -> List[PluginInfo]:
return _cached_plugin_list(os.path.normpath(abs_wp_path), tuple(wp_cli_env.items()) if wp_cli_env else None)
def list_active_plugins(abs_wp_path: str, wp_cli_env=None):
return [
item for item in plugin_list(abs_wp_path, wp_cli_env)
if item['status'] == 'active'
]
def get_plugin_data(wordpress_abs_path, plugin_name) -> List[PluginInfo]:
plugins_data = plugin_list(abs_wp_path=wordpress_abs_path)
if isinstance(plugins_data, WordpressError):
raise WposError(message=plugins_data.message,
context=plugins_data.context)
return [item for item in plugins_data if item['name'] == plugin_name]
def is_plugin_activated(abs_wp_path: str, plugin_name: str, wp_cli_env=None) -> bool:
return any(item['name'] == plugin_name and item['status'] == 'active'
for item in plugin_list(abs_wp_path, wp_cli_env))
def is_plugin_installed(abs_wp_path: str, plugin_name: str, wp_cli_env=None) -> bool:
return any(item['name'] == plugin_name for item in plugin_list(abs_wp_path, wp_cli_env))
def update_constant(abs_wp_path: str, constant_name: str, constant_value=None) -> Optional[WordpressError]:
if constant_value is None:
# Remove constant
result = wordpress(abs_wp_path, "config", "has", constant_name, "--type=constant",
env={WP_CLI_SKIP_PLUGINS_ENV: '1'})
# Skip, because not zero return code means that constant doesn't exist
if isinstance(result, WordpressError):
result = None
else:
result = wordpress(abs_wp_path, "config", "delete", constant_name, "--type=constant", "--quiet",
env={WP_CLI_SKIP_PLUGINS_ENV: '1'})
else:
# Update constant
result = wordpress(abs_wp_path, "config", "set", constant_name, constant_value, "--raw", "--type=constant",
'--anchor=// End of CloudLinux generated section', "--placement=before", "--quiet",
env={WP_CLI_SKIP_PLUGINS_ENV: '1'})
return result
def update_redis_disable_banners_constant(abs_wp_path: str, constant_value=None) -> Optional[WordpressError]:
return update_constant(abs_wp_path, RedisRequiredConstants.WP_REDIS_DISABLE_BANNERS.name, constant_value)
Zerion Mini Shell 1.0