import os
import time

from PyQt5.QtCore import QThread, pyqtSignal, QCoreApplication
from PyQt5.QtGui import QIcon
from PyQt5.QtWidgets import QSystemTrayIcon, QMenu

from fpakman.core import resource
from fpakman.core.controller import FlatpakController
from fpakman.view.qt.window import ManageWindow


class UpdateCheck(QThread):

    signal = pyqtSignal(int)

    def __init__(self, check_interval: int, controller: FlatpakController, parent=None):
        super(UpdateCheck, self).__init__(parent)
        self.controller = controller
        self.check_interval = check_interval

    def run(self):

        while True:

            apps = self.controller.refresh()

            updates = len([app for app in apps if app['update']])

            self.signal.emit(updates)

            time.sleep(self.check_interval)


class TrayIcon(QSystemTrayIcon):

    def __init__(self, locale_keys: dict, controller: FlatpakController, check_interval: int = 60, parent=None):
        self.locale_keys = locale_keys
        self.controller = controller

        self.icon_default = QIcon(resource.get_path('img/flathub_45.svg'))
        self.icon_update = QIcon(resource.get_path('img/update_logo.svg'))
        QSystemTrayIcon.__init__(self, self.icon_default, parent)

        self.menu = QMenu(parent)
        self.action_manage = self.menu.addAction(self.locale_keys['tray.action.manage'])
        self.action_manage.triggered.connect(self.show_manage_window)
        self.action_exit = self.menu.addAction(self.locale_keys['tray.action.exit'])
        self.action_exit.triggered.connect(lambda: QCoreApplication.exit())
        self.setContextMenu(self.menu)

        self.manage_window = ManageWindow(locale_keys=self.locale_keys, controller=controller, tray_icon=self)
        self.check_thread = UpdateCheck(check_interval=check_interval, controller=self.controller)
        self.check_thread.signal.connect(self.notify_updates)
        self.check_thread.start()

    def notify_updates(self, updates: int):
        if updates > 0:
            if self.icon().cacheKey() != self.icon_update.cacheKey():
                self.setIcon(self.icon_update)

                msg = '{}: {}'.format(self.locale_keys['notification.new_updates'], updates)
                self.setToolTip(msg)

                if bool(os.getenv('FPAKMAN_UPDATE_NOTIFICATION', 1)):
                    os.system("notify-send -i {} '{}'".format(resource.get_path('img/flathub_45.svg'), msg))

                if self.manage_window:
                    self.manage_window.refresh()

        else:
            self.setIcon(self.icon_default)
            self.setToolTip(None)

    def show_manage_window(self):

        if not self.manage_window:
            self.manage_window = ManageWindow(controller=self.controller)

        self.manage_window.refresh()
        self.manage_window.show()