import os

try:
    import json
except ImportError:
    import simplejson as json
import agent_util
import datetime
import time
import sys
from agent_util import float


class UptimePlugin(agent_util.Plugin):
    textkey = "uptime"
    label = "Machine Uptime"
    platform = sys.platform

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

        if "sunos" in self.platform:
            psr = agent_util.which("psrinfo", exc=True)
            if not psr:
                status = agent_util.MISCONFIGURED
                msg = "Unable to collect uptime data, please make sure psrinfo is installed."
        else:
            who = agent_util.which("who", exc=True)
            if not who:
                status = agent_util.MISCONFIGURED
                msg = (
                    "Unable to collect uptime data, please make sure Who is installed."
                )

        metadata = {
            "time_since_last_boot": {
                "label": "Time since last boot",
                "options": None,
                "status": status,
                "error_message": msg,
                "unit": "minutes",
            },
        }
        return metadata

    def check(self, textkey, data, config):
        if textkey == "time_since_last_boot":
            lb = ""
            if "sunos" in self.platform:
                psrinfo = agent_util.which("psrinfo", exc=True)
                retcode, output = agent_util.execute_command(psrinfo)
                lb = output.strip()
                lb = lb.split()[-2:]
                lb = " ".join(lb)
                lb = datetime.datetime.strptime(lb, "%m/%d/%Y %H:%M:%S")

            elif "darwin" in self.platform:
                sysctl = agent_util.which("sysctl", exc=True)
                retcode, output = agent_util.execute_command(
                    "%s -n kern.boottime" % sysctl
                )
                fields = output.strip().split()
                return time.time() - int(fields[3].strip(","))

            elif "aix" in self.platform:
                uptime = agent_util.which("uptime", exc=True)
                retcode, output = agent_util.execute_command(uptime)
                self.log.debug("####\n%s" % str(output))
                lb = re.sub(",|days|day", "", str(output))
                self.log.debug("####\n%s" % str(lb))
                if "day" in output:
                    if "min" in output:
                        lb = re.split(" ", str(lb))[4:7]
                        lb = datetime.timedelta(days=int(lb[0]), minutes=int(lb[-1]))
                    elif "hr" in output:
                        lb = re.split(" ", str(lb))[4:7]
                        lb = datetime.timedelta(days=int(lb[0]), hours=int(lb[-1]))
                    else:
                        lb = re.split(" |:", str(lb))[5:10]
                        lb = datetime.timedelta(
                            days=int(lb[0]), hours=int(lb[-2]), minutes=int(lb[-1])
                        )

                else:
                    if "min" in output:
                        lb = re.split(" ", str(lb))[4]
                        lb = datetime.timedelta(minutes=int(lb))
                    elif "hr" in output:
                        lb = re.split(" ", str(lb))[4]
                        lb = datetime.timedelta(hours=int(lb))
                    else:
                        return 60

            else:
                uptime = open("/proc/uptime").read().split()[0]
                return int(float(uptime))

            ct = datetime.datetime.now()
            diff = ct - lb
            return int(diff.total_seconds())

        return 0