Sindbad~EG File Manager

# -*- 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
from __future__ import division
from __future__ import print_function
import os
import re
import logging
import subprocess
from pwd import getpwnam

from clcommon.cpapi import cpinfo, userdomains, get_main_username_by_uid
from clcommon.public_hooks import POST_MODIFY_USER, POST_MODIFY_DOMAIN, POST_MODIFY_PACKAGE

PLESK_CONFIG_PATH = '/etc/psa/psa.conf'
logger = logging.getLogger(__name__)


def domain_updated(old_domain_name, new_domain_name=None):
    """
    Triggered after any domain change
    """
    # TODO: probably we can use cpapi here
    # username = domain_owner(new_domain_name)
    with open(PLESK_CONFIG_PATH, 'r') as f:
        config = f.read()
    match = re.search(r'^HTTPD_VHOSTS_D\s+(.*)$', config, re.MULTILINE)
    if match is None:
        logger.error("Plesk config file does not contain `HTTDP_VHOSTS_D` setting")
        return 1
    vhosts_path = match.group(1)
    domain_path = os.path.join(vhosts_path, new_domain_name)
    uid = os.stat(domain_path).st_uid
    system_user = get_main_username_by_uid(uid)
    # Emulate getpwuid error exception
    if system_user == 'N/A':
        raise KeyError(f'getpwuid(): uid not found: {uid}')

    args = [
        POST_MODIFY_DOMAIN, 'modify',
        '--username', system_user, '--domain', old_domain_name]
    if old_domain_name != new_domain_name:
        args += ['--new-domain', new_domain_name]
    return subprocess.call(args)


def physical_hosting_created(new_system_user=None, new_domain_name=None):
    """
    Triggered after creation new subscription in Plesk.

    This function is called in two situations:
     - when user with main domain is created
     - and when additional domain created
    """
    if not new_system_user or not new_domain_name:
        logger.warning("ph_created but NEW_SYSTEM_USER or NEW_DOMAIN_NAME is empty, do nothing")
        return 0

    number_of_owned_domains = len(userdomains(new_system_user))
    # user must have at least one domain, because it is in his home dir path
    if number_of_owned_domains == 1:
        owner = cpinfo(cpuser=new_system_user, keyls=('reseller', ))[0][0]
        return subprocess.call([
            POST_MODIFY_USER, 'create',
            '--username', new_system_user, '--owner', owner])
    # in case when this method is called more than once -> additional domain created
    elif number_of_owned_domains > 1:
        return subprocess.call([
            POST_MODIFY_DOMAIN, 'create',
            '--username', new_system_user, '--domain', new_domain_name])
    # impossible situation (I hope)
    else:
        raise ValueError('user %s does not own any domains!' % new_system_user)


def physical_hosting_deleted(system_user=None, domain_name=None):
    """
    Triggered after deletion physical hosting in Plesk.
    - means deletion of physical hosting on server
      (BUT NOT DOMAIN ITSELF -> domain can e.g. forward something)
    - when there are no more domains owned by system user -> unix
      user is destroyed here
    """
    if not system_user or not domain_name:
        logger.warning("ph_deleted but OLD_SYSTEM_USER or OLD_DOMAIN_NAME"
                       " is empty, do nothing")
        return 0

    try:
        getpwnam(system_user)
    except KeyError:
        return subprocess.call([
            POST_MODIFY_USER, 'delete',
            '--username', system_user])
    else:
        return subprocess.call([
            POST_MODIFY_DOMAIN, 'delete',
            '--username', system_user, '--domain', domain_name])


def physical_hosting_updated(old_system_user=None, new_system_user=None):
    """
    Triggered after any subscription change in plesk.
    """
    if not old_system_user:
        logger.warning('ph_updated but OLD_SYSTEM_USER is empty, do nothing')
        return 0

    args = [
        POST_MODIFY_USER, 'modify',
        '--username', old_system_user]
    if new_system_user != old_system_user:
        args += ['--new-username', new_system_user]

    return subprocess.call(args)


def plan_renamed():
    args = [POST_MODIFY_PACKAGE, 'rename']
    return subprocess.call(args)