AlkantarClanX12
Current Path : /opt/cloudlinux/venv/lib/python3.11/site-packages/clcommon/cpapi/plugins/ |
Current File : //opt/cloudlinux/venv/lib/python3.11/site-packages/clcommon/cpapi/plugins/vendors.py |
import os from typing import Tuple, List, Optional, Any, Dict # NOQA from vendors_api import CONFIG_PATH from vendors_api.exceptions import NotFound, NotImplementedByVendor from clcommon.cpapi.GeneralPanel import GeneralPanelPluginV1, PHPDescription from clcommon.cpapi.cpapiexceptions import NoDBAccessData, NotSupported, NoDomain from vendors_api.parser import PublicApi from clcommon.features import ALL_CL_FEATURES, Feature class PanelPlugin(GeneralPanelPluginV1): def __init__(self): super().__init__() self._api = PublicApi() def getCPName(self): if not os.path.exists(CONFIG_PATH): return None return self._api.panel_info().name def get_cp_description(self): if not os.path.exists(CONFIG_PATH): return None info = self._api.panel_info() return {'name': info.name, 'version': info.version, 'additional_info': None} def admin_packages(self, raise_exc=False): main_admin = self._api.admins(is_main=True)[0] return [pack.name for pack in self._api.packages(owner=main_admin.name)] def resellers_packages(self, raise_exc=False): result = {} main_admin = self._api.admins(is_main=True)[0] for pack in self._api.packages(): if pack.owner == main_admin.name: continue if pack.owner not in result: result[pack.owner] = [] result[pack.owner].append(pack.name) return result def get_uids_list_by_package(self, package_name, reseller_name=None): main_admin = self._api.admins(is_main=True)[0] try: users = self._api.users( package_name=package_name, package_owner=reseller_name or main_admin.name, fields=['id'] ) except NotFound: return [] return [str(u.id) for u in users] def admins(self): admins = self._api.admins() return [admin.unix_user for admin in admins if admin.unix_user] def resellers(self): resellers = self._api.resellers() return tuple(r.name for r in resellers) def is_reseller(self, username): return username in self.resellers() def db_access(self): try: db = self._api.db_info().mysql except NotImplementedByVendor as e: raise NoDBAccessData(f'db_info is not implemented by vendor: `{e}`') from e if db is None: raise NoDBAccessData('db_access is not supported by this control panel') access = {} access['login'] = db.access.login access['pass'] = db.access.password access['host'] = db.access.host access['port'] = db.access.port return access def dblogin_cplogin_pairs(self, cplogin_lst=None, with_system_users=False): try: db = self._api.db_info().mysql except NotImplementedByVendor as e: raise NotSupported(f'db_info is not implemented by vendor: `{e}`') from e if db is None: raise NotSupported('dblogin_cplogin_pairs is not supported by this control panel') result = [] for sys_user, db_users in db.mapping.items(): if cplogin_lst and sys_user not in cplogin_lst: continue for db_user in db_users: result.append([db_user, sys_user]) return tuple(result) def _convert_by_mapping(self, objects, mapping, keyls): def _get_key_nested(token, obj): keys = token.split('.') for key in keys: obj = getattr(obj, key, None) return obj result = [] for user in objects: as_array = [] for key in keyls: if key not in mapping: value = None else: value = _get_key_nested(mapping[key], user) as_array.append(value) result.append(as_array) return result def _sys_users_info(self, sys_login, keyls): # type: (Optional[str], Tuple[str]) -> List[Tuple] mapping = { 'cplogin': 'username', 'mail': 'email', 'reseller': 'owner', 'dns': 'domain', 'locale': 'locale_code', 'package': 'package.name' } try: users = self._api.users(filter_names=sys_login) except NotFound: return [] return self._convert_by_mapping(users, mapping, keyls) def _resellers_info(self, sys_login, keyls): # type: (Optional[str], Tuple[str]) -> List[Tuple] mapping = { 'cplogin': 'name', 'mail': 'email', 'locale': 'locale_code', } try: resellers = self._api.resellers(filter_names=sys_login) except NotFound: resellers = [] try: admins = self._api.admins(filter_names=sys_login) except NotFound: admins = [] return self._convert_by_mapping(resellers + admins, mapping, keyls) def cpinfo(self, cpuser=None, keyls=('cplogin', 'package', 'mail', 'reseller', 'dns'), search_sys_users=True): if search_sys_users: return self._sys_users_info(cpuser, keyls) else: return self._resellers_info(cpuser, keyls) def list_users(self, raise_exc=False): users = self._api.users(fields=['id', 'package', 'owner']) result = {} for user in users: result[user.id] = { 'package': getattr(user.package, 'name', None), 'reseller': user.owner } return result def get_reseller_users(self, reseller): try: users = self._api.users(owner=reseller, fields=['id', 'package', 'owner']) except NotFound: return {} return { user.id: {'package': getattr(user.package, 'name', None), 'reseller': user.owner} for user in users } def list_all(self, raise_exc=False): users = self._api.users(fields=['id', 'package']) return { user.id: getattr(user.package, 'name', None) for user in users } def reseller_package_by_uid(self, user_id): try: user = self._api.users(unix_id=user_id, fields=['owner', 'package'])[0] except (NotFound, IndexError): return '', '' return user.owner, getattr(user.package, 'name', None) def get_admin_emails_list(self): # see get_admin_email in __init__ # we do not care how much admins exist # in control panel now # so just return one main_admin = self._api.admins(is_main=True)[0] return [main_admin.email] def docroot(self, domain): try: domain = self._api.domains(name=domain)[domain] except (NotFound, KeyError) as e: raise NoDomain(f"Can't obtain document root for domain '{domain}'") from e return domain.document_root, domain.owner @staticmethod def useraliases(cpuser, domain): """ Return aliases from user domain :param str|unicode cpuser: user login :param str|unicode domain: :return list of aliases """ return [] def userdomains(self, cpuser): try: domains = self._api.domains(owner=cpuser) except NotFound: return [] result = [] # main domain must be first sorted_domains = sorted(list(domains.items()), key=lambda __d: not __d[1].is_main) for domain, info in sorted_domains: result.append((domain, info.document_root)) return result def reseller_users(self, resellername=None): try: return [ user.username for user in self._api.users( owner=resellername, fields=['username']) ] except NotFound: return [] def reseller_domains(self, resellername=None): try: users = self.reseller_users(resellername) return dict(self.cpinfo(users, keyls=('cplogin', 'dns'))) except NotFound: return {} def get_user_login_url(self, domain): url_template = self._api.panel_info().user_login_url if url_template is None: return url_template return url_template.format(domain=domain) def get_reseller_id_pairs(self): return {r.name: r.id for r in self._api.resellers()} def get_admin_locale(self): main_admin = self._api.admins(is_main=True)[0] return main_admin.locale_code def get_unsupported_cl_features(self) -> tuple[Feature, ...]: feature_is_enabled_map = self._api.panel_info().supported_cl_features # when hoster does not define list of supported features # we assume that all of them are supported if feature_is_enabled_map is None: if self.is_feature_lve_supported(): return tuple() return (Feature.LVE, ) disabled_features = set( feature for feature in ALL_CL_FEATURES # we should show only feature that explicitly enabled by hoster # that is why we use default False value here if not feature_is_enabled_map.get(feature, False) ) # LVE is critical for many other modules, so there is live-check if self.is_feature_lve_supported(): if Feature.LVE in disabled_features: disabled_features.remove(Feature.LVE) else: disabled_features.add(Feature.LVE) # to not make all vendors edit theis custom scripts with new parameter if Feature.AUTOTRACING in disabled_features and Feature.XRAY not in disabled_features: disabled_features.remove(Feature.AUTOTRACING) return tuple(disabled_features) def is_feature_lve_supported(self): from clcommon.lib.cledition import is_container # pylint: disable=import-outside-toplevel from clcommon.lib.cledition import is_cl_solo_edition # pylint: disable=import-outside-toplevel is_binary_exists = os.path.exists('/usr/sbin/lvectl') return all([ is_binary_exists, not is_container(), not is_cl_solo_edition(skip_jwt_check=True), ]) def get_domains_php_info(self): domains = self._api.domains(with_php=True) domains_php_info = {} for domain, domain_info in domains.items(): handler_type = 'fpm' if domain_info.php.fpm \ else 'cgi' if domain_info.php.handler: handler_type = domain_info.php.handler php_version_full = domain_info.php.php_version_id if not php_version_full: continue domains_php_info[domain] = { 'username': domain_info.owner, 'php_version_id': php_version_full, 'handler_type': handler_type, 'display_version': php_version_full } return domains_php_info def get_installed_php_versions(self): phps = self._api.php() php_description = [] for php in phps: php_description.append(PHPDescription( identifier=php.identifier, version=f'{php.identifier[-2]}.{php.identifier[-1]}', dir=php.dir, modules_dir=php.modules_dir, bin=php.bin, ini=php.ini, )) return php_description