import agent_util
import sys
import os
import platform
from agent_util import float

def build_sensor_dict():
    cmd = """sensors -u"""
    ret, out = agent_util.execute_command(cmd)
    lines = out.splitlines()
    current_sensor_package = ''
    current_sensor = ''
    sensors_dict = {}
    for l in lines:
        l = l.lower()
        if not l or l == '' or 'adapter' in l: continue
        if ':' in l:
            if not l or l == '' or 'crit' in l or 'max' in l: continue
            line = l.strip().split(':')
            if not line[1]:
                current_sensor = line[0].replace(' ', '_')
            else:
                sens_type = 'temperature'
                if 'fan' in current_sensor:
                    sens_type = 'fan_speed'

                textkey = "%s.%s.%s" % (current_sensor_package, current_sensor, line[0])
                if sens_type not in sensors_dict:
                    sensors_dict[sens_type] = {}
                sensors_dict[sens_type][textkey] = float(line[1])

        else:
            current_sensor_package = l
    return sensors_dict


class LMSensorsPlugin(agent_util.Plugin):
    textkey = "lm_sensors"
    label = "Hardware Sensors"

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

        if not agent_util.which("sensors", exc=False):
            self.log.info("lm_sensors binary not found")
            status = agent_util.UNSUPPORTED
            msg = "lm_sensors binary not found"
            return {}

        sensors = build_sensor_dict()
        self.log.debug("Found sensor data:\n%s" % sensors)
        data = {}

        if 'temperature' in sensors.keys():
            temp_options = sorted(sensors['temperature'].keys())
            data['temperature'] = {
                "label": "Sensor temperature",
                "options": temp_options,
                "status": status,
                "error_message": msg,
                "unit": "Celsius"
            }

        if 'fan_speed' in sensors.keys():
            fan_options = sorted(sensors['fan_speed'].keys())
            data['fan_speed'] = {
                "label": "Fan speed",
                "options": fan_options,
                "status": status,
                "error_message": msg,
                "unit": "RPM"
            }

        return data


    def check(self, textkey, option, config):
        sensors = build_sensor_dict()
        value = sensors.get(textkey, {}).get(option)
        if value == None:
            return None
        return float(value)