feat(devices): add onekey support

OneKeyHQ · Dec 5, 2024 · be03010 · be03010
1 parent 9ebf888
commit be03010
Show file tree

Hide file tree

Showing 12 changed files with 795 additions and 23 deletions.
diff --git a/.github/actions/install-sim/action.yml b/.github/actions/install-sim/action.yml
@@ -17,7 +17,14 @@ runs:
         apt-get update
         apt-get install -y libsdl2-image-2.0-0 libusb-1.0-0
         tar -xvf trezor-firmware.tar.gz
-
+    
+    - if: startsWith(inputs.device, 'onekey-')
+      shell: bash
+      run: |
+        apt-get update
+        apt-get install -y libsdl2-image-2.0-0 libusb-1.0-0
+        tar -xvf onekey-firmware.tar.gz
+        
     - if: inputs.device == 'coldcard'
       shell: bash
       run: |

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -154,6 +154,8 @@ jobs:
           - { name: 'ledger',   archive: 'speculos',         paths: 'test/work/speculos' }
           - { name: 'keepkey',  archive: 'keepkey-firmware', paths: 'test/work/keepkey-firmware/bin' }
           - { name: 'bitbox02', archive: 'bitbox02',         paths: 'test/work/bitbox02-firmware/build-build/bin/simulator' }
+          - { name: 'onekey-1', archive: 'onekey-firmware',  paths: 'test/work/onekey-firmware' }
+          - { name: 'onekey-t', archive: 'onekey-firmware',  paths: 'test/work/onekey-firmware' }
 
     steps:
       - uses: actions/checkout@v4
@@ -221,6 +223,8 @@ jobs:
           - 'ledger-legacy'
           - 'keepkey'
           - 'bitbox02'
+          - 'onekey-1'
+          - 'onekey-t'
         script:
           - name: 'Wheel'
             install: 'pip install dist/*.whl'
@@ -292,6 +296,8 @@ jobs:
           - 'ledger-legacy'
           - 'keepkey'
           - 'bitbox02'
+          - 'onekey-1'
+          - 'onekey-t'
         interface: [ 'library', 'cli', 'stdin' ]
 
     container: python:${{ matrix.python-version }}

diff --git a/hwilib/devices/__init__.py b/hwilib/devices/__init__.py
@@ -5,5 +5,6 @@
     'digitalbitbox',
     'coldcard',
     'bitbox02',
-    'jade'
+    'jade',
+    'onekey'
 ]
diff --git a/hwilib/devices/keepkey.py b/hwilib/devices/keepkey.py
@@ -160,8 +160,7 @@ def __init__(self, path: str, password: Optional[str] = None, expert: bool = Fal
         if path.startswith("udp"):
             model.default_mapping.register(KeepkeyDebugLinkState)
 
-        super(KeepkeyClient, self).__init__(path, password, expert, chain, KEEPKEY_HID_IDS, KEEPKEY_WEBUSB_IDS, KEEPKEY_SIMULATOR_PATH, model)
-        self.type = 'Keepkey'
+        super(KeepkeyClient, self).__init__(path, password, expert, chain, KEEPKEY_HID_IDS, KEEPKEY_WEBUSB_IDS, KEEPKEY_SIMULATOR_PATH, model, device_type="Keepkey")
 
     def can_sign_taproot(self) -> bool:
         """