^^&&^^

Edit File: cli.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 logging
import pwd

from defence360agent.utils import (
    check_run,
    CheckRunError,
)
from imav.model.wordpress import WPSite
from imav.wordpress import (
    PLUGIN_PATH,
    PLUGIN_SLUG,
    utils,
)

logger = logging.getLogger(__name__)

async def plugin_install(site: WPSite):
    """Install the Imunify Security WordPress plugin on given WordPress site."""
    username = pwd.getpwuid(site.uid).pw_name
    php_path = await utils.get_php_binary_path(site, username)

args = [
        *utils.wp_wrapper(php_path, site.docroot),
        "plugin",
        "install",
        str(PLUGIN_PATH),
        "--activate",
        "--force",
    ]

command = utils.build_command_for_user(username, args)

logger.info(f"Installing wp plugin {command}")

await check_run(command)

async def plugin_update(site: WPSite):
    """
    Update the Imunify Security WordPress plugin on given WordPress site.

Currently, this is the same as install, but in the future it may differ.
    """
    username = pwd.getpwuid(site.uid).pw_name
    php_path = await utils.get_php_binary_path(site, username)

args = [
        *utils.wp_wrapper(php_path, site.docroot),
        "plugin",
        "install",
        str(PLUGIN_PATH),
        "--activate",
        "--force",
    ]

command = utils.build_command_for_user(username, args)

logger.info(f"Updating wp plugin {command}")

await check_run(command)

async def plugin_uninstall(site: WPSite):
    """Uninstall the imunify-security wp plugin from given wp site."""
    username = pwd.getpwuid(site.uid).pw_name
    php_path = await utils.get_php_binary_path(site, username)

args = [
        *utils.wp_wrapper(php_path, site.docroot),
        "plugin",
        "uninstall",
        PLUGIN_SLUG,
        "--deactivate",
    ]

command = utils.build_command_for_user(username, args)

logger.info(f"Uninstalling wp plugin {command}")

await check_run(command)

async def get_plugin_version(site: WPSite):
    """Get the version of the imunify-security wp plugin installed on given WordPress site."""
    username = pwd.getpwuid(site.uid).pw_name
    php_path = await utils.get_php_binary_path(site, username)
    args = [
        *utils.wp_wrapper(php_path, site.docroot),
        "plugin",
        "get",
        PLUGIN_SLUG,
        "--field=version",
    ]
    command = utils.build_command_for_user(username, args)

logger.info(f"Getting wp plugin version {command}")

try:
        result = await check_run(command)
        output = result.decode("utf-8").strip()
        if not output:
            return ""
        return output.split(maxsplit=1)[0]
    except CheckRunError as e:
        logger.error(
            "Failed to get wp plugin version. Return code: %s",
            e.returncode,
        )
        return None
    except UnicodeDecodeError as e:
        logger.error("Failed to decode wp plugin version output: %s", e)
        return None

async def is_plugin_installed(site: WPSite):
    """Check if the imunify-security wp plugin is installed on given WordPress site."""
    username = pwd.getpwuid(site.uid).pw_name
    php_path = await utils.get_php_binary_path(site, username)
    args = [
        *utils.wp_wrapper(php_path, site.docroot),
        "plugin",
        "is-installed",
        PLUGIN_SLUG,
    ]
    command = utils.build_command_for_user(username, args)

logger.info(f"Checking if wp plugin is installed {command}")

try:
        await check_run(command)
    except CheckRunError:
        # exit code other than 0 means plugin is not installed or there is an error
        return False

# exit code 0 means plugin is installed
    return True

async def is_wordpress_installed(site: WPSite):
    """Check if WordPress is installed and given site is accessible using WP CLI."""
    username = pwd.getpwuid(site.uid).pw_name
    php_path = await utils.get_php_binary_path(site, username)
    args = [
        *utils.wp_wrapper(php_path, site.docroot),
        "core",
        "is-installed",
    ]
    command = utils.build_command_for_user(username, args)

logger.info(f"Checking if WordPress is installed {command}")

try:
        # exit code other than 0 means WordPress is not installed or there is an error
        await check_run(command)
    except CheckRunError:
        return False

# exit code 0 means WordPress is installed
    return True