$57 GRAYBYTE WORDPRESS FILE MANAGER $45

#!/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, ModifyUserHook
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, ModifyUserHook):
    """
    Keep the LVD on-disk state (domains.json + binary id-registry)
    in sync with control-panel domain and user events so that
    per-domain limits keep working.
    """
    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 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(),
            )

    @lve_supported_or_exit
    def post_modify_user(self, username, new_name=None,
                         new_owner=None, **kwargs):
        if new_name is None:
            return

        if not registry_path_by_username(new_name):
            return

        cmd = [
            self.LVECTL, 'regenerate-domains',
            '--username', new_name,
        ]
        result = subprocess.run(cmd, capture_output=True, check=False)
        if result.returncode != 0:
            log.warning(
                'regenerate-domains after user rename failed (rc=%d): %s',
                result.returncode,
                result.stderr.decode(errors='replace').strip(),
            )