# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, print_function, unicode_literals

import re
from threading import Event

import flask
import octoprint.plugin

CMD_ESTEPS_LOAD_STEPS = "eSteps_load"
CMD_ESTEPS_START_EXTRUSION = "eSteps_startExtrusion"
CMD_ESTEPS_SAVE = "eSteps_save"

class API(octoprint.plugin.SimpleApiPlugin):

    @staticmethod
    def apiCommands():
        return {
            CMD_ESTEPS_SAVE: [],
            CMD_ESTEPS_LOAD_STEPS: [],
            CMD_ESTEPS_START_EXTRUSION: ['extrudeLength','extrudeSpeed','extrudeTemp']
        }

    def apiGateWay(self, command, data):
        if command == CMD_ESTEPS_LOAD_STEPS:
            self._logger.debug("Load steps from EEPROM")
            if not self._printer.is_ready():
                self._logger.warning("Printer not ready, operation canceled")
                return flask.abort(503, 'Printer not ready, operation canceled')

            # Register listener waiting response for M92 command
            m92Event = Event()
            self.registerGCodeWaiter("M92", self.m92GCodeResponse, m92Event)

            # Issue M92 command
            self._printer.commands("M92")

            m92Event.wait(5)

            return flask.jsonify({
                "data": self.data["steps"]
            })
        if command == CMD_ESTEPS_START_EXTRUSION:
            self._logger.debug("Heating the extruder [%s]", data)
            if not self._printer.is_ready():
                self._logger.warning("Printer not ready, operation canceled")
                return flask.abort(503,"Printer not ready, operation canceled")

            # Register event to be trigger when temp is achieved
            self.registerEventTemp("T0", int(data["extrudeTemp"]), self.startExtrusion, data["extrudeLength"], data["extrudeSpeed"])
            # Heating HotEnd
            self._printer.commands("M104 S%(extrudeTemp)s" % data)

        if command == CMD_ESTEPS_SAVE:
            if (("newESteps" not in data or data["newESteps"] <= 0)
                and
                (("newXSteps" not in data or data["newXSteps"] <= 0) or
                 ("newYSteps" not in data or data["newYSteps"] <= 0) or
                 ("newZSteps" not in data or data["newZSteps"] <= 0))):
                return flask.abort(400, "Invalid arguments. No value provided for X,Y, Z or E steps")
            stopHeater = []
            if "newESteps" in data:
                eStepsSettings = self._settings.get(['eSteps'])
                userControlsTemp = eStepsSettings.get("userControlsTemp")
                turnOffHotend = eStepsSettings.get("turnOffHotend")
                stopHeater = ["M104 S0"] if turnOffHotend and not userControlsTemp else  []

            steps = ("M92 " +
                    ("E%(newESteps)s " % data if "newESteps" in data else "") +
                    ("X%(newXSteps)s " % data if "newXSteps" in data else "") +
                    ("Y%(newYSteps)s " % data if "newYSteps" in data else "") +
                    ("Z%(newYSteps)s " % data if "newZSteps" in data else ""))

            #save data to EEPROM and cool-down
            self._printer.commands([steps, "M500"] + stopHeater)
            return


############## HANDLERS ##############
    @staticmethod
    def startExtrusion(self, temps, extrudeLength, extrudeSpeed, *args):
        self._logger.debug("Temperature achieved, extrusion started [temps:%s, extrudeLength:%s, extrudeSpeed:%s, args:%s]",
        temps, extrudeLength, extrudeSpeed, args)

        # Extrude
        self._printer.extrude(amount=int(extrudeLength), speed=int(extrudeSpeed))

    @staticmethod
    def m92GCodeResponse(self, line, regex, event):
        reg = re.compile(".*\s*(?P<command>(?P<gCode>M\d{1,3}) X(?P<xVal>\d{1,3}.\d{1,3}) Y(?P<yVal>\d{1,3}.\d{1,3}) Z(?P<zVal>\d{1,3}.\d{1,3}) E(?P<eVal>\d{1,3}.\d{1,3}))")
        isM92command = reg.match(line)
        if isM92command:
            command = isM92command.group("command")
            if isM92command.group("gCode") == "M92":
                self.data["steps"]["X"] = float(isM92command.group("xVal"))
                self.data["steps"]["Y"] = float(isM92command.group("yVal"))
                self.data["steps"]["Z"] = float(isM92command.group("zVal"))
                self.data["steps"]["E"] = float(isM92command.group("eVal"))

                # Send the new data to the UI to be reloaded
                self._logger.debug("gCode: %s", command)
                self._logger.debug("X: %s, Y:%s, Z:%s, E:%s", self.data["steps"]["X"], self.data["steps"]["Y"], self.data["steps"]["Z"], self.data["steps"]["E"])
                self._logger.debug("Finished data collection")
                self.collectCommand = False
        if event:
            event.set()