# coding=utf-8

################################################################################
### RPi sysfs Plugin for Plotty Temp Graph plugin
### Based on plugin structure provided by jneilliii
###
### Jeffrey J. Kosowsky
### August 7, 2022
###
### Note: configure from cli using:
###       /home/kosowsky/octoprint/venv/bin/octoprint config set [-bool|-int] plugins.RPiSysGraph.<config_variable> <config_value>
###
################################################################################

from __future__ import absolute_import

import octoprint.plugin
from octoprint.util import RepeatedTimer

# Config default values
SENSOR_NAME = "RPiSys"
SENSOR_PATH = "/sys/class/thermal/thermal_zone0/temp"
SENSOR_DIVISOR = 1000
OUTPUT_PRECISION = 1 #Output precision in digits
UPDATE_PERIOD = 5 #Seconds

class RPiSysGraph(octoprint.plugin.StartupPlugin, octoprint.plugin.SettingsPlugin):
        def __init__(self):
                self.last_temps = dict()
                self.poll_temps = None

                self.sensor_name = None
                self.sensor_path = None
                self.sensor_divisor = None
                self.output_precision = None
                self.update_period = None
                self.convertTo_celsius = None
                self.convertTo_fahrenheit = None

        def get_settings_defaults(self):
                return {
                        "sensor_name": SENSOR_NAME,
                        "sensor_path": SENSOR_PATH,
                        "sensor_divisor": SENSOR_DIVISOR,
                        "output_precision": 1,
                        "convertTo_celsius": False,
                        "convertTo_fahrenheit": False,
                        "update_period": UPDATE_PERIOD,
                }

        def on_after_startup(self):
                self.sensor_name = self._settings.get(["sensor_name"])
                self.sensor_path = self._settings.get(["sensor_path"])
                self.sensor_divisor = self._settings.get(["sensor_divisor"])
                self.output_precision = self._settings.get_int(["output_precision"])
                self.convertTo_celsius = self._settings.get_boolean(["convertTo_celsius"])
                self.convertTo_fahrenheit = self._settings.get_boolean(["convertTo_fahrenheit"])
                self.update_period = self._settings.get_int(["update_period"])

                # start repeated timer for checking temp from sensor, runs every 5 seconds
                self.poll_temps = RepeatedTimer(self.update_period, self.read_temp)
                self.poll_temps.start()

        def read_temp(self):
                currtemp = None

                try:
                        with open(r"/sys/class/thermal/thermal_zone0/temp") as File:
                                        currtemp = float(File.readline())/self.sensor_divisor

                        if currtemp is not None:
                                if self.convertTo_fahrenheit:
                                        currtemp = currtemp * 1.8 + 32
                                elif self.convertTo_celsius:
                                        currtemp = (currtemp - 32) * 5/9
                                currtemp = round(currtemp, self.output_precision)
                                self.last_temps[self.sensor_name] = (currtemp, None)
#                                self._logger.warning("JJK: {} : {}".format(self.sensor_name, currtemp)) #JJKDEBUG
                except:
                        self._logger.debug("There was an error getting temperature from the SystemCmd temperature plugin")
#                        self._logger.warning ("There was an error getting temperature from the SystemCmd temperature plugin") #JJKDEBUG

        def temp_callback(self, comm, parsed_temps):
                parsed_temps.update(self.last_temps)
                return parsed_temps

__plugin_name__ = "RPiSys Plotly Temp Graph Integration"
__plugin_pythoncompat__ = ">=3,<4"
__plugin_version__ = "0.1.0"
__plugin_implementation__ = RPiSysGraph()
__plugin_hooks__ = {
        "octoprint.comm.protocol.temperatures.received": __plugin_implementation__.temp_callback
}