# SPDX-License-Identifier: GPL-3.0-or-later

################################################################################
# Practical Libraries
################################################################################
# Logger library compatible with multiprocessing
from loguru import logger

import os, time

# Library for starting processes
import multiprocessing

# Basic planktoscope communication libraries
import planktoscope.mqtt

logger.info("planktoscope.module is loaded")

################################################################################
# Main Segmenter class
################################################################################
class ModuleProcess(multiprocessing.Process):
    """This class contains the main definitions for the module of the PlanktoScope"""

    @logger.catch
    def __init__(self, event):
        """Initialize the Module class

        Args:
            event (multiprocessing.Event): shutdown event
        """
        super(ModuleProcess, self).__init__(name="light")

        logger.info("planktoscope.module is initialising")

        # Do all your initialisation here

        logger.success("planktoscope.mdule is initialised and ready to go!")

    @logger.catch
    def treat_message(self):
        action = ""
        if self.module_client.new_message_received():
            logger.info("We received a new message")
            last_message = self.module_client.msg["payload"]
            logger.debug(last_message)
            self.module_client.read_message()
            if "action" not in last_message:
                logger.error(
                    f"The received message has the wrong argument {last_message}"
                )
                self.module_client.client.publish("status/module", '{"status":"Error"}')
                return
            action = last_message["action"]
        if action == "on":
            # Treat the received messages here
            pass
        elif action != "":
            logger.warning(
                f"We did not understand the received request {action} - {last_message}"
            )

    ################################################################################
    # While loop for capturing commands from Node-RED
    ################################################################################
    @logger.catch
    def run(self):
        """This is the function that needs to be started to create a thread"""
        logger.info(
            f"The module control thread has been started in process {os.getpid()}"
        )

        # MQTT Service connection
        self.module_client = planktoscope.mqtt.MQTT_Client(
            topic="module/#", name="module_client"
        )

        # Publish the status "Ready" to via MQTT to Node-RED
        self.module_client.client.publish("status/module", '{"status":"Ready"}')

        logger.success("Module is READY!")

        # This is the loop
        while not self.stop_event.is_set():
            self.treat_message()
            time.sleep(0.1)

        logger.info("Shutting down the module process")
        # Do your deinit and ressources cleanup here
        self.module_client.client.publish("status/module", '{"status":"Dead"}')
        self.module_client.shutdown()
        logger.success("Module process shut down! See you!")


# This is called if this script is launched directly
if __name__ == "__main__":
    # This should be tests of your module
    pass