vial/src/main/python/main_window.py

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

from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import QWidget, QComboBox, QToolButton, QHBoxLayout, QVBoxLayout, QMainWindow, QAction, qApp, \
    QFileDialog, QDialog, QTabWidget

import json

from firmware_flasher import FirmwareFlasher
from keyboard import Keyboard
from layout_editor import LayoutEditor
from util import tr, find_vial_keyboards, open_device


class MainWindow(QMainWindow):

    def __init__(self):
        super().__init__()
        self.keyboard = None
        self.devices = []
        self.sideload_json = None
        self.sideload_vid = self.sideload_pid = -1

        self.combobox_devices = QComboBox()
        self.combobox_devices.currentIndexChanged.connect(self.on_device_selected)

        btn_refresh_devices = QToolButton()
        btn_refresh_devices.setToolButtonStyle(Qt.ToolButtonTextOnly)
        btn_refresh_devices.setText(tr("MainWindow", "Refresh"))
        btn_refresh_devices.clicked.connect(self.on_click_refresh)

        layout_combobox = QHBoxLayout()
        layout_combobox.addWidget(self.combobox_devices)
        layout_combobox.addWidget(btn_refresh_devices)

        self.layout_editor = LayoutEditor()
        flasher = FirmwareFlasher()

        tabs = QTabWidget()
        for container, lbl in [(self.layout_editor, "Layout"), (flasher, "Firmware updater")]:
            w = QWidget()
            w.setLayout(container)
            tabs.addTab(w, tr("MainWindow", lbl))

        layout = QVBoxLayout()
        layout.addLayout(layout_combobox)
        layout.addWidget(tabs)
        w = QWidget()
        w.setLayout(layout)
        self.setCentralWidget(w)

        self.init_menu()

        # make sure initial state is valid
        self.on_click_refresh()

    def init_menu(self):
        layout_load_act = QAction(tr("MenuFile", "Load saved layout..."), self)
        layout_load_act.setShortcut("Ctrl+O")
        layout_load_act.triggered.connect(self.on_layout_load)

        layout_save_act = QAction(tr("MenuFile", "Save current layout..."), self)
        layout_save_act.setShortcut("Ctrl+S")
        layout_save_act.triggered.connect(self.on_layout_save)

        sideload_json_act = QAction(tr("MenuFile", "Sideload VIA JSON..."), self)
        sideload_json_act.triggered.connect(self.on_sideload_json)

        exit_act = QAction(tr("MenuFile", "Exit"), self)
        exit_act.setShortcut("Ctrl+Q")
        exit_act.triggered.connect(qApp.exit)

        menubar = self.menuBar()
        file_menu = self.menuBar().addMenu(tr("Menu", "File"))
        file_menu.addAction(layout_load_act)
        file_menu.addAction(layout_save_act)
        file_menu.addSeparator()
        file_menu.addAction(sideload_json_act)
        file_menu.addSeparator()
        file_menu.addAction(exit_act)

    def on_layout_load(self):
        dialog = QFileDialog()
        dialog.setDefaultSuffix("vil")
        dialog.setAcceptMode(QFileDialog.AcceptOpen)
        dialog.setNameFilters(["Vial layout (*.vil)"])
        if dialog.exec_() == QDialog.Accepted:
            with open(dialog.selectedFiles()[0], "rb") as inf:
                data = inf.read()
            self.layout_editor.restore_layout(data)

    def on_layout_save(self):
        dialog = QFileDialog()
        dialog.setDefaultSuffix("vil")
        dialog.setAcceptMode(QFileDialog.AcceptSave)
        dialog.setNameFilters(["Vial layout (*.vil)"])
        if dialog.exec_() == QDialog.Accepted:
            with open(dialog.selectedFiles()[0], "wb") as outf:
                outf.write(self.layout_editor.save_layout())

    def on_click_refresh(self):
        self.devices = find_vial_keyboards(self.sideload_vid, self.sideload_pid)
        self.combobox_devices.clear()

        for dev in self.devices:
            title = "{} {}".format(dev["manufacturer_string"], dev["product_string"])
            if dev["vendor_id"] == self.sideload_vid and dev["product_id"] == self.sideload_pid:
                title += " [sideload]"
            self.combobox_devices.addItem(title)

    def on_device_selected(self):
        if self.keyboard is not None:
            self.keyboard.dev.close()
        idx = self.combobox_devices.currentIndex()
        if idx >= 0:
            dev = self.devices[idx]
            self.keyboard = Keyboard(open_device(dev))
            if dev["vendor_id"] == self.sideload_vid and dev["product_id"] == self.sideload_pid:
                self.keyboard.reload(self.sideload_json)
            else:
                self.keyboard.reload()
            self.layout_editor.rebuild(self.keyboard)

    def on_sideload_json(self):
        dialog = QFileDialog()
        dialog.setDefaultSuffix("json")
        dialog.setAcceptMode(QFileDialog.AcceptOpen)
        dialog.setNameFilters(["VIA layout JSON (*.json)"])
        if dialog.exec_() == QDialog.Accepted:
            with open(dialog.selectedFiles()[0], "rb") as inf:
                data = inf.read()
            self.sideload_json = json.loads(data)
            self.sideload_vid = int(self.sideload_json["vendorId"], 16)
            self.sideload_pid = int(self.sideload_json["productId"], 16)
            self.on_click_refresh()