Mini Shell

#!/opt/cloudlinux/venv/bin/python3 -bb

# 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

import subprocess
import sys
import json
from typing import List, Set, Tuple, Optional
from xml.dom import minidom as xml

from clveconfig import ve_config
from clcommon.lib.cledition import lve_supported_or_exit

LVECTL = '/usr/sbin/lvectl'


def lvectl_list_packages() -> Set[Tuple[str, Optional[str]]]:
    """
    Parse packages list from `lvectl all-package-list` command
    Ex: {('pack1', 'reseller1'), ('pack2', None), ...}
    """
    res = []
    _, response = subprocess.getstatusoutput(f'{LVECTL} all-package-list --json')
    data = json.loads(response).get('data', [])
    for package in data:
        _reseller = package.get('RESELLER')
        if not _reseller or _reseller == 'N/A':
            _reseller = None
        res.append(
            (package.get('ID'), _reseller)
        )
    return set(
        filter(lambda x: x[0] not in {'VE_DEFAULT'}, res)
    )


def xml_get_packages(xml_element: xml.Element) -> List[Tuple[str, Optional[str]]]:
    """
    Parse packages list from ve.cfg
    """
    res = []
    packages = xml_element.getElementsByTagName('package')
    for package in packages:
        res.append(
            (package.getAttribute('id'), package.getAttribute('reseller') or None)
        )
    return res


@lve_supported_or_exit
def main():
    """
    Check that ve.cfg contains no orphan package elements
    Otherwise delete these packages
    """
    changed = False
    xml_config, _ = ve_config.get_xml_config(use_cache=False)
    xml_packages = xml_get_packages(xml_config)
    raw_xml_packages = xml_config.getElementsByTagName('package')
    lvectl_packages = lvectl_list_packages()
    for package in xml_packages:
        package_id, package_reseller = package
        if package not in lvectl_packages:
            try:
                package_element = next(
                    filter(
                        lambda x, pid=package_id, pres=package_reseller: x.getAttribute('id') == pid
                        and (
                            (x.hasAttribute('reseller') is False and pres is None)
                            or x.getAttribute('reseller') == str(pres)
                        ),
                        raw_xml_packages,
                    )
                )
                package_element.parentNode.removeChild(package_element)
                changed = True
            except StopIteration:
                pass
    if changed:
        ve_config.save_xml(xml_config)
    return 0


if __name__ == '__main__':
    sys.exit(main())