From b2e6381decfe92aeec25080e98373198d05b872e Mon Sep 17 00:00:00 2001
From: Neucrack <CZD666666@gmail.com>
Date: Thu, 3 Mar 2022 23:30:45 +0800
Subject: [PATCH] optimize protocol shortcut show

---
 COMTool/assets/qss/style-dark.qss |  1 -
 COMTool/assets/qss/style.qss      |  1 -
 COMTool/plugins/protocol.py       | 25 ++++++++++++++++++-------
 3 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/COMTool/assets/qss/style-dark.qss b/COMTool/assets/qss/style-dark.qss
index 4f8d1f5..6a51f40 100644
--- a/COMTool/assets/qss/style-dark.qss
+++ b/COMTool/assets/qss/style-dark.qss
@@ -282,7 +282,6 @@ QPushButton:default {
 .remark {
     min-width: 10px;
     min-height: 10px;
-    width: 20px;
     height: 20px;
 }
 
diff --git a/COMTool/assets/qss/style.qss b/COMTool/assets/qss/style.qss
index 1927a79..2c16876 100644
--- a/COMTool/assets/qss/style.qss
+++ b/COMTool/assets/qss/style.qss
@@ -282,7 +282,6 @@ QPushButton:default {
 .remark {
     min-width: 10px;
     min-height: 10px;
-    width: 20px;
     height: 20px;
 }
 
diff --git a/COMTool/plugins/protocol.py b/COMTool/plugins/protocol.py
index 4773aae..bc28c73 100644
--- a/COMTool/plugins/protocol.py
+++ b/COMTool/plugins/protocol.py
@@ -236,7 +236,7 @@ def onInit(self, config, plugins):
         self.encodeMethod = lambda x:x
         self.decodeMethod = lambda x:x
         self.pressedKeys = []
-        self.keyModeClickTimes = 0
+        self.keyModeClickTime = 0
 
     def print(self, *args, **kw_args):
         end = "\n"
@@ -297,12 +297,12 @@ def onWidgetMain(self, parent, rootWindow):
         customSendItemsLayoutWrapper.setContentsMargins(0,0,0,0)
         cutomSendItemsWraper.setLayout(customSendItemsLayoutWrapper)
         # items container
-        customItems = QWidget()
+        self.customItems = QWidget()
         self.customSendItemsLayout = QVBoxLayout()
         self.customSendItemsLayout.setContentsMargins(0,0,0,0)
-        customItems.setLayout(self.customSendItemsLayout)
+        self.customItems.setLayout(self.customSendItemsLayout)
 
-        customSendItemsLayoutWrapper.addWidget(customItems)
+        customSendItemsLayoutWrapper.addWidget(self.customItems)
         customSendItemsLayoutWrapper.addWidget(self.addButton)
         customSendItemsLayoutWrapper.addStretch(0)
 
@@ -320,16 +320,26 @@ def clearReceived():
         def keyModeOn(event):
             self.keyModeBtn.setProperty("class", "deleteBtn")
             utils_ui.updateStyle(self.mainWidget, self.keyModeBtn)
-            self.keyModeClickTimes = time.time()
+            self.keyModeClickTime = time.time()
+            # show all shortcut
+            widgets = self.customItems.findChildren(QPushButton, "editRemark")
+            for i, w in enumerate(widgets):
+                shortcut = "+".join((name for v, name in self.config["customSendItems"][i]["shortcut"]))
+                w.setText(shortcut)
+                utils_ui.updateStyle(self.mainWidget, w)
 
         def keyModeOff(event):
             self.keyModeBtn.setProperty("class", "")
             utils_ui.updateStyle(self.mainWidget, self.keyModeBtn)
-            self.keyModeClickTimes = 0
+            self.keyModeClickTime = 0
+            # hide all shortcut
+            widgets = self.customItems.findChildren(QPushButton, "editRemark")
+            for w in widgets:
+                w.setText("")
 
         def keyModeTuggle():
             if self.keyModeBtn.property("class") == "deleteBtn":
-                if time.time() - self.keyModeClickTimes < 0.2:
+                if time.time() - self.keyModeClickTime < 0.2:
                     return
                 else:
                     self.keyModeBtn.clearFocus()
@@ -550,6 +560,7 @@ def insertSendItem(self, item = None, load = False):
             item["shortcut"] = []
         utils_ui.setButtonIcon(send, item["icon"])
         editRemark = QPushButton("")
+        editRemark.setObjectName("editRemark")
         utils_ui.setButtonIcon(editRemark, "ei.pencil")
         editRemark.setProperty("class", "remark")
         cmd.setToolTip(text)