import agent_util
import logging


logger = logging.getLogger(__name__)


class UsersPlugin(agent_util.Plugin):
    textkey = "users"
    label = "Users"

    @classmethod
    def get_metadata(self, config):
        status = agent_util.SUPPORTED
        msg = None

        options = []

        data = {
            "users.unique_login_count": {
                "label": "Users unique login count",
                "options": options,
                "status": status,
                "error_message": msg,
            },
            "users.total_login_count": {
                "label": "Users total login count",
                "options": options,
                "status": status,
                "error_message": msg,
            },
        }
        return data

    def check(self, textkey, data, config):
        query = ""
        if textkey == "users.unique_login_count":
            query = "who |cut -c 1-9 |sort -u |wc -l"

        if textkey == "users.total_login_count":
            query = "who | wc -l"

        ret, output = agent_util.execute_command("%s" % query)
        return int(output)