From 6e20faa737bfd51b685c95c475a32958700a1875 Mon Sep 17 00:00:00 2001
From: Peter Hutterer <peter.hutterer@who-t.net>
Date: Fri, 9 Aug 2019 14:01:02 +1000
Subject: [PATCH] protocol: add a timeout argument to the new protocol handlers

The default timeout is too short for registering a button, let's control this
from the message.

Signed-off-by: Peter Hutterer <peter.hutterer@who-t.net>
---
 tuhi/protocol.py | 8 +++++---
 tuhi/wacom.py    | 5 +++--
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/tuhi/protocol.py b/tuhi/protocol.py
index 0b7297c..e6f4732 100644
--- a/tuhi/protocol.py
+++ b/tuhi/protocol.py
@@ -17,11 +17,11 @@
 # opcodes and message formats. So the entry point is the Protocol class, and
 # the messages can be accessed by key.
 #
-#     def my_callback(request, requires_reply=True, userdata=None):
+#     def my_callback(request, requires_reply=True, userdata=None, timeout=5):
 #        if request is not None:
 #            send_to_device(request)
 #        if requires_reply:
-#            return read_from device()
+#            return read_from device() or eventually_timeout()
 #
 #     p = Protocol(ProtocolVersion.INTUOS_PRO, my_callback, userdata)
 #     m1 = P.get(Interactions.GET_NAME)
@@ -378,7 +378,7 @@ class Msg(object):
        removed in later versions but to keep the caller stack simpler, we
        just provide a noop Msg.'''
 
-    def __init__(self, callback, userdata=None):
+    def __init__(self, callback, userdata=None, timeout=None):
         super().__init__()
         assert self.opcode is not None
         assert self.protocol is not None
@@ -386,6 +386,7 @@ class Msg(object):
         self._callback = callback
         self.errorcode = None
         self.userdata = userdata
+        self.timeout = timeout
         self.args = [0x00]  # Empty messages don't exist
 
     @property
@@ -425,6 +426,7 @@ class Msg(object):
         self.request = NordicData([self.opcode, len(self.args or []), *(self.args or [])])
         self.reply = self._callback(request=self.request if self.requires_request else None,
                                     requires_reply=self.requires_reply,
+                                    timeout=self.timeout or None,
                                     userdata=self.userdata)
         if self.requires_reply:
             try:
diff --git a/tuhi/wacom.py b/tuhi/wacom.py
index 07c5e8c..65153b7 100644
--- a/tuhi/wacom.py
+++ b/tuhi/wacom.py
@@ -534,11 +534,12 @@ class WacomProtocolLowLevelComm(GObject.Object):
         return args
 
     # The callback used by the protocol messages
-    def nordic_data_exchange(self, request, requires_reply=False, userdata=None):
+    def nordic_data_exchange(self, request, requires_reply=False,
+                             userdata=None, timeout=None):
         if request is not None:
             self.send_nordic_command(request.opcode, request)
         if requires_reply:
-            return self.wait_nordic_data(expected_opcode=None, timeout=5)
+            return self.wait_nordic_data(expected_opcode=None, timeout=timeout or 5)
 
 
 class WacomRegisterHelper(WacomProtocolLowLevelComm):