22

✏️ Editing: mac_pkgutil.py

"""
Installer support for macOS.

Installer is the native .pkg/.mpkg package manager for macOS.
"""

import os.path
import urllib

import salt.utils.itertools
import salt.utils.mac_utils
import salt.utils.path
import salt.utils.platform
from salt.exceptions import SaltInvocationError

# Don't shadow built-in's.
__func_alias__ = {"list_": "list"}

__virtualname__ = "pkgutil"

def __virtual__():
    if not salt.utils.platform.is_darwin():
        return (False, "Only available on Mac OS systems")

if not salt.utils.path.which("pkgutil"):
        return (False, "Missing pkgutil binary")

return __virtualname__

def list_():
    """
    List the installed packages.

:return: A list of installed packages
    :rtype: list

CLI Example:

.. code-block:: bash

salt '*' pkgutil.list
    """
    cmd = "pkgutil --pkgs"
    ret = salt.utils.mac_utils.execute_return_result(cmd)
    return ret.splitlines()

def is_installed(package_id):
    """
    Returns whether a given package id is installed.

:return: True if installed, otherwise False
    :rtype: bool

CLI Example:

.. code-block:: bash

salt '*' pkgutil.is_installed com.apple.pkg.gcc4.2Leo
    """
    return package_id in list_()

def _install_from_path(path):
    """
    Internal function to install a package from the given path
    """
    if not os.path.exists(path):
        msg = f"File not found: {path}"
        raise SaltInvocationError(msg)

cmd = f'installer -pkg "{path}" -target /'
    return salt.utils.mac_utils.execute_return_success(cmd)

def install(source, package_id):
    """
    Install a .pkg from an URI or an absolute path.

:param str source: The path to a package.

:param str package_id: The package ID

:return: True if successful, otherwise False
    :rtype: bool

CLI Example:

.. code-block:: bash

salt '*' pkgutil.install source=/vagrant/build_essentials.pkg package_id=com.apple.pkg.gcc4.2Leo
    """
    if is_installed(package_id):
        return True

uri = urllib.parse.urlparse(source)
    if not uri.scheme == "":
        msg = f"Unsupported scheme for source uri: {uri.scheme}"
        raise SaltInvocationError(msg)

_install_from_path(source)

return is_installed(package_id)

def forget(package_id):
    """
    .. versionadded:: 2016.3.0

Remove the receipt data about the specified package. Does not remove files.

.. warning::
        DO NOT use this command to fix broken package design

:param str package_id: The name of the package to forget

:return: True if successful, otherwise False
    :rtype: bool

CLI Example:

.. code-block:: bash

salt '*' pkgutil.forget com.apple.pkg.gcc4.2Leo
    """
    cmd = f"pkgutil --forget {package_id}"
    salt.utils.mac_utils.execute_return_success(cmd)
    return not is_installed(package_id)

MR SKB