# -*- coding: utf-8 -*-
"""
Watch for pkgs that have upgrades, then fire an event.

.. versionadded:: 2016.3.0
"""

# Import python libs
from __future__ import absolute_import, unicode_literals

import logging

__virtualname__ = "pkg"

log = logging.getLogger(__name__)


def __virtual__():
    """
    Only load if strace is installed
    """
    return __virtualname__ if "pkg.upgrade_available" in __salt__ else False


def validate(config):
    """
    Validate the beacon configuration
    """
    # Configuration for pkg beacon should be a list
    if not isinstance(config, list):
        return False, ("Configuration for pkg beacon must be a list.")

    # Configuration for pkg beacon should contain pkgs
    pkgs_found = False
    pkgs_not_list = False
    for config_item in config:
        if "pkgs" in config_item:
            pkgs_found = True
            if isinstance(config_item["pkgs"], list):
                pkgs_not_list = True

    if not pkgs_found or not pkgs_not_list:
        return False, "Configuration for pkg beacon requires list of pkgs."
    return True, "Valid beacon configuration"


def beacon(config):
    """
    Check if installed packages are the latest versions
    and fire an event for those that have upgrades.

    .. code-block:: yaml

        beacons:
          pkg:
            - pkgs:
                - zsh
                - apache2
            - refresh: True
    """
    ret = []

    _refresh = False
    pkgs = []
    for config_item in config:
        if "pkgs" in config_item:
            pkgs += config_item["pkgs"]
        if "refresh" in config and config["refresh"]:
            _refresh = True

    for pkg in pkgs:
        _installed = __salt__["pkg.version"](pkg)
        _latest = __salt__["pkg.latest_version"](pkg, refresh=_refresh)
        if _installed and _latest:
            _pkg = {"pkg": pkg, "version": _latest}
            ret.append(_pkg)
    return ret

"""
Watch for pkgs that have upgrades, then fire an event.

.. versionadded:: 2016.3.0
"""
import logging

__virtualname__ = "pkg"

log = logging.getLogger(__name__)


def __virtual__():
    """
    Only load if strace is installed
    """
    return __virtualname__ if "pkg.upgrade_available" in __salt__ else False


def validate(config):
    """
    Validate the beacon configuration
    """
    # Configuration for pkg beacon should be a list
    if not isinstance(config, list):
        return False, ("Configuration for pkg beacon must be a list.")

    # Configuration for pkg beacon should contain pkgs
    pkgs_found = False
    pkgs_not_list = False
    for config_item in config:
        if "pkgs" in config_item:
            pkgs_found = True
            if isinstance(config_item["pkgs"], list):
                pkgs_not_list = True

    if not pkgs_found or not pkgs_not_list:
        return False, "Configuration for pkg beacon requires list of pkgs."
    return True, "Valid beacon configuration"


def beacon(config):
    """
    Check if installed packages are the latest versions
    and fire an event for those that have upgrades.

    .. code-block:: yaml

        beacons:
          pkg:
            - pkgs:
                - zsh
                - apache2
            - refresh: True
    """
    ret = []

    _refresh = False
    pkgs = []
    for config_item in config:
        if "pkgs" in config_item:
            pkgs += config_item["pkgs"]
        if "refresh" in config and config["refresh"]:
            _refresh = True

    for pkg in pkgs:
        _installed = __salt__["pkg.version"](pkg)
        _latest = __salt__["pkg.latest_version"](pkg, refresh=_refresh)
        if _installed and _latest:
            _pkg = {"pkg": pkg, "version": _latest}
            ret.append(_pkg)
    return ret

Drop Py2 and six on salt/beacons/pkg.py