# SPDX-FileCopyrightText: 2020 Andrius Štikonas <andrius@stikonas.eu>
# SPDX-License-Identifier: LGPL-3.0-or-later

import asyncio
import json
import logging

class NeoHub:
    def __init__(self):
        self._logger = logging.getLogger('neohub')
        self._firmware_version = 0
        pass


    async def connect(self, host='Neo-Hub', port='4242'):
        self._host = host
        self._port = port


    async def _send(self, message):
        reader, writer = await asyncio.open_connection(self._host, self._port)
        encoded_message = bytearray(json.dumps(message) + "\0\r", "utf-8")
        self._logger.debug(f"Sending message: {encoded_message}")
        writer.write(encoded_message)
        await writer.drain()

        data = await reader.read(4096)
        json_string = data.decode('utf-8')
        self._logger.debug(f"Received message: {json_string}")
        writer.close()
        await writer.wait_closed()
        print(json_string)
        return json.loads(json_string)


    async def firmware(self):
        """
        NeoHub firmware version
        """
        message = {"FIRMWARE": 0}

        result = await self._send(message)
        firmware_version = int(result['firmware version'])
        return firmware_version


    async def reset(self):
        """
        Reboot neohub

        Returns True if Restart is initiated
        """
        message = {"RESET": 0}

        firmware_version = await self.firmware()
        result = ""
        if firmware_version >= 2027:
            result = await self._send(message)
            return result['Restarting'] == 1
        else:
            return False


    async def get_system(self):
        """
        Get system wide variables
        """
        message = {"GET_SYSTEM": 0}

        data = await self._send(message)
        return data