path: root/extras/UI_Handeler.py

"""
   Copyright 2023 Suchinton Chakravarty

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
"""

from main import *
from PyQt5 import QtCore
from PyQt5.QtCore import QPropertyAnimation
from PyQt5.QtWidgets import QWidget
from PyQt5.QtCore import QEasingCurve
from PyQt5.QtWidgets import QGraphicsOpacityEffect
from kuksa_client.grpc import Field, SubscribeEntry, View
from kuksa_client.grpc.aio import VSSClient
from PyQt5.QtCore import pyqtSignal
import asyncio
from PyQt5.QtCore import QThread
import pathlib
import logging
import json

from . import Kuksa_Instance as kuksa_instance
from Widgets import settings

# Global variables
subscribed = False


class GrpcSubscriptionThread(QThread):
    updateReceived = pyqtSignal(str, str)

    def __init__(self):
        QThread.__init__(self)
        self.client = None

    def run(self):        
        config = kuksa_instance.KuksaClientSingleton.instance().get_config()
        token = kuksa_instance.KuksaClientSingleton.instance().get_token()

        SUBSCRIPTION_ENTRIES = [
            SubscribeEntry('Vehicle.Speed', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Powertrain.CombustionEngine.Speed', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Body.Lights.DirectionIndicator.Left.IsSignaling', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Body.Lights.DirectionIndicator.Right.IsSignaling', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Body.Lights.Hazard.IsSignaling', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Powertrain.FuelSystem.Level', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Powertrain.CombustionEngine.ECT', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Powertrain.Transmission.SelectedGear', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Cabin.HVAC.Station.Row1.Left.Temperature', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Cabin.HVAC.Station.Row1.Left.FanSpeed', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Cabin.HVAC.Station.Row1.Right.Temperature', View.FIELDS, (Field.VALUE,)),
            SubscribeEntry('Vehicle.Cabin.HVAC.Station.Row1.Right.FanSpeed', View.FIELDS, (Field.VALUE,)),
        ]

        async def grpc_subscription(client):
            try:
                await client.connect()
                async for updates in client.subscribe(entries=SUBSCRIPTION_ENTRIES):
                    for update in updates:
                        if update.entry.value is not None:
                            self.updateReceived.emit(str(update.entry.path),
                                                     str(update.entry.value.value))
                client.disconnect()
            except Exception as e:
                logging.error(f"Error during gRPC subscription: {e}")

        try:
            client = VSSClient(host=config['ip'], 
                               port=config['port'], 
                               token=token,
                               root_certificates=pathlib.Path(config['cacertificate']), 
                               tls_server_name=config['tls_server_name'])
            asyncio.set_event_loop(asyncio.new_event_loop())
            loop = asyncio.get_event_loop()
            loop.run_until_complete(grpc_subscription(client))
        except Exception as e:
            logging.error(f"Error during gRPC subscription: {e}")


class UI_Handeler(MainWindow):    

    def fullscreen(self):
        self.headerContainer.hide()
        self.setAttribute(QtCore.Qt.WA_TranslucentBackground, False)
        self.showFullScreen()

    def block_updates():
        global block_subscription_updates
        block_subscription_updates = True

    def unblock_updates():
        global block_subscription_updates
        block_subscription_updates = False

    def Hide_Navbar(self, bool_arg):
        """
        This method hides the navigation bar of the UI.

        Args:
        - bool_arg: A boolean value indicating whether to hide the navigation bar or not.
        """
        height = self.BottomMenuSubContainer.height()
        heightExtended = 75 if bool_arg else 0

        self.animation = QPropertyAnimation(
            self.BottomMenuSubContainer, b"minimumHeight")
        self.animation.setDuration(400)
        self.animation.setStartValue(height)
        self.animation.setEndValue(heightExtended)
        self.animation.setEasingCurve(QtCore.QEasingCurve.InOutQuart)
        self.animation.start()

    def animateSwitch(self, index):
        """
        This method animates the switching of pages for QstackedWidget with the animation being a fade in and out.

        Args:
        - index: The index of the page to switch to.
        """
        self.fader_widget = FaderWidget(
            self.stackedWidget.currentWidget(), self.stackedWidget.widget(index))
        self.stackedWidget.setCurrentIndex(index)

    def toggleMaximized(self):
        if self.isMaximized():
            self.showNormal()
        else:
            self.showMaximized()

    def moveWindow(self, event):
        if event.buttons() == QtCore.Qt.LeftButton:
            self.move(self.pos() + event.globalPos() - self.clickPosition)
            self.clickPosition = event.globalPos()
            event.accept()

    def mousePressEvent(self, event):
        self.clickPosition = event.globalPos()
        event.accept()

    def mouseReleaseEvent(self, event):
        self.clickPosition = None
        event.accept()

    def set_instance(self):
        self.kuksa = kuksa_instance.KuksaClientSingleton.instance()
        self.client = self.kuksa.get_client()
        if self.client is not None and self.client.checkConnection():
            return True
        else:
            logging.error("Kuksa client is not connected, try reconnecting")
            return False

    def stop_client(self):
        if self.client is not None and self.client.checkConnection():
            self.client.stop()

    def subscribe_VSS_Signals(self):
        """
        This method subscribes to the VSS signals from Kuksa.
        """
        global subscribed
        if not subscribed:
            self.kuksa = kuksa_instance.KuksaClientSingleton.instance()
            self.client = self.kuksa.get_client()
            if self.client is not None and self.client.checkConnection():
                signals = [
                    "Vehicle.Speed",
                    "Vehicle.Powertrain.CombustionEngine.Speed",
                    "Vehicle.Body.Lights.DirectionIndicator.Left.IsSignaling",
                    "Vehicle.Body.Lights.DirectionIndicator.Right.IsSignaling",
                    "Vehicle.Body.Lights.Hazard.IsSignaling",
                    "Vehicle.Powertrain.FuelSystem.Level",
                    "Vehicle.Powertrain.CombustionEngine.ECT",
                    "Vehicle.Powertrain.Transmission.SelectedGear",
                    "Vehicle.Cabin.HVAC.Station.Row1.Left.Temperature",
                    "Vehicle.Cabin.HVAC.Station.Row1.Left.FanSpeed",
                    "Vehicle.Cabin.HVAC.Station.Row1.Right.Temperature",
                    "Vehicle.Cabin.HVAC.Station.Row1.Right.FanSpeed"]

                if settings.Protocol == "ws":
                    for signal in signals:
                        self.client.subscribe(
                            signal, lambda data: UI_Handeler.VSS_callback(self, data), 'value')
                if settings.Protocol == "grpc":
                    self.worker = GrpcSubscriptionThread()
                    self.worker.updateReceived.connect(
                        lambda path, value: UI_Handeler.VSS_callback(self=self, path=path, value=value))
                    self.worker.start()
                subscribed = True
            else:
                subscribed = False
                logging.error(
                    "Kuksa client is not connected, try reconnecting")

    def VSS_callback(self, data=None, path=None, value=None):
        """
        This method is the callback function for the VSS signals from Kuksa.

        Args:
        - data: The data received from the signal.
        """

        IC_Page = self.stackedWidget.widget(1)
        HVAC_Page = self.stackedWidget.widget(2)

        if data is not None:
            info = json.loads(data)
            path = info.get('data', {}).get('path')
            value = info.get('data', {}).get('dp', {}).get('value')

        if path == "Vehicle.Speed" and int(float(value)):
            # block connection updates for IC_Page.Speed_slider.
            IC_Page.Speed_slider.blockSignals(True)
            IC_Page.Speed_slider.setValue(int(float(value)))
            IC_Page.Speed_slider.blockSignals(False)
            IC_Page.Speed_monitor.display(int(IC_Page.Speed_slider.value()))

        if path == "Vehicle.Powertrain.CombustionEngine.Speed" and int(float(value)):
            # block connection updates for IC_Page.RPM_slider.
            IC_Page.RPM_slider.blockSignals(True)
            IC_Page.RPM_monitor.display(int(IC_Page.RPM_slider.value()))
            IC_Page.RPM_slider.setValue(int(float(value)))
            IC_Page.RPM_slider.blockSignals(False)

        if path == "Vehicle.Body.Lights.DirectionIndicator.Left.IsSignaling":
            IC_Page.leftIndicatorBtn.blockSignals(True)
            IC_Page.leftIndicatorBtn.setChecked(bool(value))
            IC_Page.leftIndicatorBtn.blockSignals(False)

        if path == "Vehicle.Body.Lights.DirectionIndicator.Right.IsSignaling":
            IC_Page.rightIndicatorBtn.blockSignals(True)
            IC_Page.rightIndicatorBtn.setChecked(bool(value))
            IC_Page.rightIndicatorBtn.blockSignals(False)

        if path == "Vehicle.Body.Lights.Hazard.IsSignaling":
            IC_Page.hazardBtn.blockSignals(True)
            IC_Page.hazardBtn.setChecked(bool(value))
            IC_Page.hazardBtn.blockSignals(False)

        if path == "Vehicle.Powertrain.FuelSystem.Level":
            IC_Page.fuelLevel_slider.blockSignals(True)
            IC_Page.fuelLevel_slider.setValue(int(float(value)))
            IC_Page.fuelLevel_slider.blockSignals(False)

        if path == "Vehicle.Powertrain.CombustionEngine.ECT":
            IC_Page.coolantTemp_slider.blockSignals(True)
            IC_Page.coolantTemp_slider.setValue(int(float(value)))
            IC_Page.coolantTemp_slider.blockSignals(False)

        if path == "Vehicle.Powertrain.Transmission.SelectedGear":
            if int(float(value)) == 127:
                IC_Page.driveBtn.blockSignals(True)
                IC_Page.driveBtn.setChecked(True)
                IC_Page.driveBtn.blockSignals(False)
            elif int(float(value)) == 126:
                IC_Page.parkBtn.blockSignals(True)
                IC_Page.parkBtn.setChecked(True)
                IC_Page.parkBtn.blockSignals(False)
            elif int(float(value)) == -1:
                IC_Page.reverseBtn.blockSignals(True)
                IC_Page.reverseBtn.setChecked(True)
                IC_Page.reverseBtn.blockSignals(False)
            elif int(float(value)) == 0:
                IC_Page.neutralBtn.blockSignals(True)
                IC_Page.neutralBtn.setChecked(True)
                IC_Page.neutralBtn.blockSignals(False)

        if path == "Vehicle.Cabin.HVAC.Station.Row1.Left.Temperature":
            HVAC_Page.leftTempList.blockSignals(True)
            item = HVAC_Page.leftTempList.findItems(
                str(int(float(value))) + "°C", QtCore.Qt.MatchExactly)[0]
            HVAC_Page.leftTempList.setCurrentItem(item)
            HVAC_Page.leftTempList.scrollToItem(item, 1)
            HVAC_Page.leftTempList.blockSignals(False)

        if path == "Vehicle.Cabin.HVAC.Station.Row1.Left.FanSpeed":
            HVAC_Page.leftFanSpeed_slider.blockSignals(True)
            HVAC_Page.leftFanSpeed_slider.setValue(int(float(value)))
            HVAC_Page.leftFanSpeed_slider.blockSignals(False)

        if path == "Vehicle.Cabin.HVAC.Station.Row1.Right.Temperature":
            HVAC_Page.rightTempList.blockSignals(True)
            item = HVAC_Page.leftTempList.findItems(
                str(int(float(value))) + "°C", QtCore.Qt.MatchExactly)[0]
            HVAC_Page.leftTempList.setCurrentItem(item)
            HVAC_Page.rightTempList.scrollToItem(item, 1)
            HVAC_Page.rightTempList.blockSignals(False)

        if path == "Vehicle.Cabin.HVAC.Station.Row1.Right.FanSpeed":
            HVAC_Page.rightFanSpeed_slider.blockSignals(True)
            HVAC_Page.rightFanSpeed_slider.setValue(int(float(value)))
            HVAC_Page.rightFanSpeed_slider.blockSignals(False)


class FaderWidget(QWidget):
    def __init__(self, old_widget, new_widget):
        super().__init__(new_widget)

        self.old_widget = old_widget
        self.new_widget = new_widget

        self.effect = QGraphicsOpacityEffect()
        self.new_widget.setGraphicsEffect(self.effect)

        self.animation = QPropertyAnimation(self.effect, b"opacity")
        self.animation.setDuration(300)
        self.animation.setStartValue(0)
        self.animation.setEndValue(1)
        self.animation.setEasingCurve(QEasingCurve.OutCubic)
        self.animation.finished.connect(self.close)

        self.animate()

    def animate(self):
        self.animation.start()

    def close(self):
        self.old_widget.close()
        self.new_widget.show()
        super().close()