Sindbad~EG File Manager

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

from __future__ import absolute_import

import os
from functools import wraps

from clcommon.public_hooks.lib import ModifyPackageHook
import cldetectlib
from clcontrollib import detect_panelclass
from clveconfig.ve_config import get_xml_config, save_xml

OLD_ADMIN_KEY = 'OLD_ADMIN_TEMPLATE'
NEW_ADMIN_KEY = 'NEW_ADMIN_TEMPLATE'
OLD_DOMAIN_KEY = 'OLD_DOMAIN_TEMPLATE'
NEW_DOMAIN_KEY = 'NEW_DOMAIN_TEMPLATE'


def is_plesk_or_exit(f):
    """
    Check that function runs on plesk, or return None
    """
    @wraps(f)
    def inner(*args, **kwargs):
        if not cldetectlib.is_plesk():
            return None
        else:
            return f(*args, **kwargs)
    return inner


class PleskRenameHook(ModifyPackageHook):
    """
    Sync plesk DB package id with ve.cfg package name
    """
    @is_plesk_or_exit
    def post_modify_package(self, *args, **kwargs):
        """
        Synchronize plesk DB and ve.cfg package names by plesk DB id
        """
        panel = detect_panelclass()

        package_id = os.environ.get(OLD_ADMIN_KEY)
        new_id = os.environ.get(NEW_ADMIN_KEY)
        if not package_id:
            package_id = os.environ.get(OLD_DOMAIN_KEY)
            new_id = os.environ.get(NEW_DOMAIN_KEY)
        assert package_id and package_id == new_id     # this id should be equal in case of modify\rename

        new_package_name = panel.get_package_name_by_id_from_plesk_db(int(package_id))
        xml, _ = get_xml_config(False)
        packages = xml.getElementsByTagName('package')
        try:
            el_to_rename = next(filter(     # pylint: disable=W1639
                lambda x: x.getAttribute('plesk_id') == package_id,
                packages
            ))
        except StopIteration:
            return
        el_to_rename.setAttribute('id', new_package_name)
        save_xml(xml)