import agent_util
import os
import re
import sys
from agent_util import float


class EntropyPlugin(agent_util.Plugin):
    textkey = "entropy"
    label = "Entropy"

    @classmethod
    def get_metadata(self, config):
        status = agent_util.SUPPORTED
        msg = None
        if not os.path.exists("/proc/sys/kernel/random/entropy_avail"):
            status = agent_util.UNSUPPORTED
            msg = "/proc/sys/kernel/random/entropy_avail does not exist."

        data = {
            "entropy.avail": {
                "label": "Entropy level",
                "options": None,
                "status": status,
                "error_message": msg,
            }
        }
        return data

    def check(self, textkey, data, config):
        if textkey == "entropy.avail":
            return float(
                open("/proc/sys/kernel/random/entropy_avail", "r").read().strip()
            )
        return 0