\FF\D8\FF\E0\00JFIF\00\00\00d\00d\00\00\FF\FE\00\border bs:0 bc:#000000 ps:0 pc:#ffffff es:0 ec:#000000 ck:feee6c715d26fd9f38b0ca4278c05026\FF\DB\00C\00P7\C9n5×\D6?\BDê\9Ds\EBp\9F[`8m\B7)o\B5\E8\E6I\99\FE3]]A2\BA\8Cw\D6E\93\\DEv\C8\009\F2\F1NI?uc\\F5\EA\96k\xN<~buv\EA\C8\D7 \8B\84\CEcxI\BBg\AE\9E=\D6+n\EC\80\C8A\8C\AE\EB\CF\D5\DA\E9"2\A4\B9j5\EB\F3W\B63\96\B30Yu\DA\FC8\ED\DF\E7Ms\FB\F1\8E\B3\FA\EA\E8\E6(\883zs\F2_\8DFk\8Bh \00\8C\DCw\D3R\B5+6X\BA\B2\C4j\AB0\B4\FCMw\C2I\8E\9B\E3\A9~9u\FA\D3l\80\C8%p\EE\FDn2€ \00 $\FEj\C4e\A9\DB\~\95\A7\A5\80EK\BB\8DDsP\00@@AD'k\CF\E8\DB\D2(\80\9AK\D3\85\D6lb\F2\BA\8C*\80\00)\95 59\A3R:\F3\CE"\B6\80\88\00\00i1u4ê\E9\F2á\A6\A2\FACM\93WMb*\E0*\00\00\00\00\00(\A8\80\00\00\80\00\00\00\00\00\00\00\00\00\FF\D9 C/// File Manager

File Manager

Path: /opt/cloudlinux/venv/lib/python3.11/site-packages/clwpos/

Viewing File: wpos_hooks.py

#!/opt/cloudlinux/venv/bin/python3 -bb
# -*- 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/LICENCE.TXT

import subprocess
import sys
from distutils.sysconfig import get_python_lib
from pathlib import Path

from clwpos.constants import (
    ALT_PHP_REDIS_ENABLE_UTILITY,
    INSTALL_CACHING_HOOKS_UTILITY,
    PHP_REDIS_ENABLE_UTILITY,
)
from clwpos.optimization_features import OBJECT_CACHE_FEATURE
from clwpos.feature_suites import (
    any_suite_allowed_on_server,
    is_module_allowed_for_user
)
from clwpos.utils import is_wpos_supported

UNIVERSAL_HOOK_PATH_DNF = '/etc/dnf/universal-hooks/multi_pkgs/transaction'
UNIVERSAL_HOOK_PATH_YUM = '/etc/yum/universal-hooks/multi_pkgs/posttrans'
UNIVERSAL_HOOK_PATH_APT = '/etc/apt/universal-hooks/multi_pkgs/Post-Invoke'
ALT_HOOK_DIRNAME = 'alt-php__WILDCARD__-pecl-ext'
EA_HOOK_DIRNAME = 'ea-php__WILDCARD__'
PLESK_HOOK_DIRNAME = 'plesk-php__WILDCARD__'
HOOKS_LISTENERS_DIR = '/usr/share/cloudlinux/hooks/listeners'
MODIFY_USER_HOOK = 'wpos_modify_user_hook.py'
USER_DIRS_HOOK = 'wpos_user_dirs_hook.py'
# Hooks that depend on AccelerateWP activation state
DYNAMIC_WPOS_HOOKS = (MODIFY_USER_HOOK,)
# Default hooks that should be always installed
DEFAULT_WPOS_HOOKS = (USER_DIRS_HOOK,)


def _install_universal_hook(path_provider, target_binary: str) -> None:
    """
    Create hook dir and symlink the target binary if it doesn't exist yet.
    """
    hook_dir_path = path_provider()
    hook_dir_path.mkdir(parents=True, exist_ok=True)
    hook_name = Path(target_binary).name
    hook_full_path = Path(hook_dir_path, hook_name)
    if not hook_full_path.exists():
        hook_full_path.symlink_to(target_binary)


def _uninstall_universal_hook(path_provider, target_binary: str) -> None:
    """
    Remove hook symlink if present (even if broken).
    """
    hook_dir_path = path_provider()
    hook_name = Path(target_binary).name
    hook_full_path = Path(hook_dir_path, hook_name)
    if hook_full_path.is_symlink():
        hook_full_path.unlink()


def get_universal_hook_alt_php_path() -> Path:
    """
    Get path to yum universal hooks directory
    with alt-php* hooks.
    """
    dir_name = ALT_HOOK_DIRNAME
    if Path('/etc/apt/').exists():
        return Path(UNIVERSAL_HOOK_PATH_APT, dir_name)
    elif Path('/etc/dnf/').exists():
        return Path(UNIVERSAL_HOOK_PATH_DNF, dir_name)
    return Path(UNIVERSAL_HOOK_PATH_YUM, dir_name)


def install_yum_universal_hook_alt_php() -> None:
    """
    Install yum universal hook for configuring PHP redis
    after alt-php* package is installed/updated.
    """
    _install_universal_hook(get_universal_hook_alt_php_path, ALT_PHP_REDIS_ENABLE_UTILITY)


def uninstall_yum_universal_hook_alt_php() -> None:
    """
    Remove yum universal hook for configuring PHP redis ext.
    """
    _uninstall_universal_hook(get_universal_hook_alt_php_path, ALT_PHP_REDIS_ENABLE_UTILITY)


def get_universal_hook_ea_php_path() -> Path:
    """
    Get path to yum universal hooks directory
    with ea-php* hooks.
    """
    dir_name = EA_HOOK_DIRNAME
    if Path('/etc/apt/').exists():
        return Path(UNIVERSAL_HOOK_PATH_APT, dir_name)
    elif Path('/etc/dnf/').exists():
        return Path(UNIVERSAL_HOOK_PATH_DNF, dir_name)
    return Path(UNIVERSAL_HOOK_PATH_YUM, dir_name)


def install_yum_universal_hook_ea_php() -> None:
    """
    Install yum universal hook for configuring PHP redis
    after ea-php* package is installed/updated.
    """
    _install_universal_hook(get_universal_hook_ea_php_path, PHP_REDIS_ENABLE_UTILITY)


def uninstall_yum_universal_hook_ea_php() -> None:
    """
    Remove yum universal hook for configuring PHP redis ext.
    """
    _uninstall_universal_hook(get_universal_hook_ea_php_path, PHP_REDIS_ENABLE_UTILITY)


def get_universal_hook_plesk_php_path() -> Path:
    """
    Get path to yum universal hooks directory
    with plesk-php* hooks.
    """
    dir_name = PLESK_HOOK_DIRNAME
    if Path('/etc/apt/').exists():
        return Path(UNIVERSAL_HOOK_PATH_APT, dir_name)
    elif Path('/etc/dnf/').exists():
        return Path(UNIVERSAL_HOOK_PATH_DNF, dir_name)
    return Path(UNIVERSAL_HOOK_PATH_YUM, dir_name)


def install_yum_universal_hook_plesk_php() -> None:
    """
    Install yum universal hook for configuring PHP redis
    after plesk-php* package is installed/updated.
    """
    _install_universal_hook(get_universal_hook_plesk_php_path, PHP_REDIS_ENABLE_UTILITY)


def uninstall_yum_universal_hook_plesk_php() -> None:
    """
    Remove yum universal hook for configuring PHP redis ext.
    """
    _uninstall_universal_hook(get_universal_hook_plesk_php_path, PHP_REDIS_ENABLE_UTILITY)


def install_single_hook(hook) -> None:
    """
    Install single hook
    """
    lve_utils_hooks_dir = Path(get_python_lib(), 'clwpos', 'hooks')
    listeners_hook_path = Path(HOOKS_LISTENERS_DIR, hook)
    lve_utils_hook_path = Path(lve_utils_hooks_dir, hook)
    # remove old hook pointing to symlink to lve_utils
    if 'lve_utils' in str(listeners_hook_path.resolve()) or \
            'python3.7' in str(listeners_hook_path.resolve()):
        listeners_hook_path.unlink()
    if not listeners_hook_path.exists() and lve_utils_hook_path.exists():
        listeners_hook_path.symlink_to(lve_utils_hook_path)


def uninstall_single_hook(hook):
    """
    Uninstall single hook
    """
    listeners_hook_path = Path(HOOKS_LISTENERS_DIR, hook)
    if listeners_hook_path.is_symlink():
        listeners_hook_path.unlink()


def install_default_panel_hooks() -> None:
    """
    Install wpos_user_dirs_hook.py hook
    """
    for hook in DEFAULT_WPOS_HOOKS:
        install_single_hook(hook)


def uninstall_default_panel_hooks() -> None:
    """
    Uninstall wpos_user_dirs_hook.py hook
    """
    for hook in DEFAULT_WPOS_HOOKS:
        uninstall_single_hook(hook)


def install_dynamic_panel_hooks() -> None:
    """
    Install panel WPOS hooks.
    """
    subprocess.run([INSTALL_CACHING_HOOKS_UTILITY, '-i'], capture_output=True)
    for hook in DYNAMIC_WPOS_HOOKS:
        install_single_hook(hook)


def uninstall_dynamic_panel_hooks() -> None:
    """
    Remove panel WPOS hooks.
    """
    subprocess.run([INSTALL_CACHING_HOOKS_UTILITY, '-d'], capture_output=True)

    for hook in DYNAMIC_WPOS_HOOKS:
        uninstall_single_hook(hook)


def _install_hooks():
    """
    Install all hooks
    """
    install_default_panel_hooks()

    if is_wpos_supported() and any_suite_allowed_on_server():
        if is_module_allowed_for_user(OBJECT_CACHE_FEATURE):
            install_yum_universal_hook_alt_php()
            install_yum_universal_hook_ea_php()
            install_yum_universal_hook_plesk_php()
        install_dynamic_panel_hooks()


def _uninstall_hooks():
    """
    Uninstall all hooks
    """
    uninstall_yum_universal_hook_alt_php()
    uninstall_yum_universal_hook_ea_php()
    uninstall_yum_universal_hook_plesk_php()
    uninstall_dynamic_panel_hooks()
    uninstall_default_panel_hooks()


def main():
    """
    Install or uninstall panel and yum/dnf universal hooks.
    """
    if '--install' in sys.argv:
        _install_hooks()
    elif '--uninstall' in sys.argv:
        _uninstall_hooks()


if __name__ == "__main__":
    main()