Add bitbox01 downloader

Author: achow101

hwilib/firmware.py

@@ -122,6 +122,35 @@ def keepkey_id_formatted(id_str):
 
     return {'success': True, 'message': 'Keepkey firmware version {} downloaded to {}'.format(version, downloaded_file)}
 
+def digitalbitbox_01_download(version=None, bitcoinonly=False):
+    versions = feedparser.parse('https://github.com/digitalbitbox/mcu/tags.atom')
+    releases = versions.entries
+
+    def id_formatted(id_str):
+        tag = id_str.split('/')[-1]
+        p = re.compile(r'^v\d+.\d+.\d$')
+        return bool(p.match(tag))
+
+    releases = [r for r in releases if id_formatted(r['id'])]
+    releases.sort(key=lambda r: r["updated_parsed"], reverse=True)
+
+    version_info = {}
+    if version is None:
+        version_info = releases[0]
+        version = version_info['id'].split('/')[-1][1:]
+    else:
+        for r in releases:
+            if r['id'].split('/')[-1][1:] == version:
+                version_info = r
+                break
+        else:
+            raise BadArgumentError('{} is not available'.format(version))
+
+    url = 'https://github.com/digitalbitbox/mcu/releases/download/v{}/firmware.deterministic.{}.signed.bin'.format(version, version)
+    downloaded_file = _download_file(url)
+
+    return {'success': True, 'message': 'Digital Bitbox01 firmware version {} downloaded to {}'.format(version, downloaded_file)}
+
 def download_firmware(model, version, bitcoinonly=False):
     dev_model = model.lower()
     func_name = dev_model + '_download'