from threading import Event

CMD_ESP_TEST = "TEST"
CMD_ESP_LOAD_STEPS = "eSteps_load"
CMD_ESP_START_EXTRUSION = "eSteps_startExtrusion"
CMD_ESP_SAVE = "eSteps_save"

class API(octoprint.plugin.SimpleApiPlugin):

    @staticmethod
    def apiCommands():
        return {
            CMD_ESP_SAVE: [],
            CMD_ESP_LOAD_STEPS: [],
            CMD_ESP_START_EXTRUSION: []
        }

    def apiGateWay(self, command, data):
        self._logger.debug("api command [%s] received payload [%s]", command, data)
        if command == CMD_ESP_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, {
                    "msg": "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()
            return flask.jsonify({
                "data": self.data["steps"]
            })
        if command == CMD_ESP_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, {
                    "msg": "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 the tool
            self._printer.commands("M104 S%(extrudeTemp)s" % data)
            return

        if command == CMD_ESP_SAVE:
            eStepsSettings = self._settings.get(['eSteps'])
            userControlsTemp = eStepsSettings.get("userControlsTemp")
            turnOffHotend = eStepsSettings.get("turnOffHotend")
            self._printer.commands(["M92 E%(newESteps)s" % data, "M500"] + ["M104 S0"] if turnOffHotend and not userControlsTemp else  [])
            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, event):
        reg = re.compile("echo:\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()