$16 GRAYBYTE WORDPRESS FILE MANAGER $92

#!/opt/cloudlinux/venv/bin/python3 -bb
# coding=utf-8
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT

import logging
import subprocess

from clcommon.public_hooks.lib import ModifyDomainHook
from clcommon.lib.cledition import lve_supported_or_exit
from websiteisolation.id_registry import registry_path_by_username

log = logging.getLogger(__name__)


class DomainLimitsHook(ModifyDomainHook):
    """
    After a domain rename or document-root change, update the LVD
    on-disk state (domains.json + binary id-registry) so that
    per-domain limits keep working for the affected domain.
    """
    LVECTL = '/usr/sbin/lvectl'

    @lve_supported_or_exit
    def post_modify_domain(
            self, username, domain,
            new_domain=None, include_subdomains=None,
            old_docroot=None, new_docroot=None, **kwargs):

        # if there is not registry for the user - domain limits are NOT enabled
        if not registry_path_by_username(username):
            return

        current_domain = new_domain or domain

        cmd = [
            self.LVECTL, 'regenerate-domains',
            '--username', username,
            '--domain', current_domain,
        ]
        if new_domain:
            cmd += ['--old-domain', domain]
        if old_docroot:
            cmd += ['--old-docroot', old_docroot]

        result = subprocess.run(cmd, capture_output=True, check=False)
        if result.returncode != 0:
            log.warning(
                'regenerate-domains failed (rc=%d): %s',
                result.returncode,
                result.stderr.decode(errors='replace').strip(),
            )