OwlCyberSecurity - MANAGER

Edit File: plesk_sdk.py

"""
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.

This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>
"""
import json
import re
from subprocess import PIPE, check_output

class PleskSdk:
    """PHP wrapper for Plesk SDK"""

_cmd = (
        "plesk",
        "php",
        "-r",
    )
    _php = """
        require_once("sdk.php");
        pm_Context::init("revisium-antivirus");
        {};
        """

@classmethod
    def _join(cls, php: str, strip_whitespaces=True) -> str:
        """
        Append a piece of code to the header and make it a single-line string
        """
        code = cls._php.format(php.strip())
        r = "".join(filter(None, (line.strip() for line in code.splitlines())))
        if strip_whitespaces:
            r = re.sub(r"(\W)\s+|\s+(\W)", r"\1\2", r)
        return r

@classmethod
    def exec(cls, php, *args):
        cmd = (*cls._cmd, cls._join(php), *args)
        return check_output(cmd, stderr=PIPE, text=True)

@classmethod
    def context__get_var_dir(cls):
        return cls.exec("echo pm_Context::getVarDir()")

@classmethod
    def settings__get(cls, key, default):
        return cls.exec(
            "echo pm_Settings::get($argv[1], $argv[2])",
            key,
            json.dumps(default),
        )

@classmethod
    def domain__get_all_domains(cls):
        php = """
            echo json_encode(
                array_map(
                    function ($domain) {
                        $props = [
                            "id" => ["getId", []],
                            "name" => ["getName", []],
                            "document_root" => ["getDocumentRoot", []],
                            "owner" => ["getSysUserLogin", []],
                            "imav_cleanup_allowed" => ["hasPermission", ["ra_cleanup_allowed"]],
                        ];
                        $result = (object)[];
                        foreach ($props as $property => [$method, $args]) {
                            try {
                                $result->{$property} = call_user_func_array([$domain, $method], $args);
                            } catch (Throwable $ignored) {}
                        }
                        return $result;
                    },
                    pm_Domain::getAllDomains()
                )
            );
            """
        domains = json.loads(cls.exec(php))
        # because when there are no domains -> it returns empty LIST instead of dict,
        # but further code relies on DICT
        return domains if domains else {}