From ca9bf60a1becce063ff74724bbd24ec3b8f24f10 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A1szl=C3=B3=20Monda?= <laci@monda.hu>
Date: Tue, 30 Jan 2018 21:46:39 +0100
Subject: [PATCH] Extract updateDeviceFirmware()

---
 packages/usb/update-device-firmware.js | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/packages/usb/update-device-firmware.js b/packages/usb/update-device-firmware.js
index d8f37a77..9772522c 100755
--- a/packages/usb/update-device-firmware.js
+++ b/packages/usb/update-device-firmware.js
@@ -3,26 +3,30 @@ const uhk = require('./uhk');
 const program = require('commander');
 require('shelljs/global');
 
-const extension = '.hex';
 config.fatal = true;
+const extension = '.hex';
 
 program
     .usage(`firmwareImage${extension}`)
     .parse(process.argv)
 
 const firmwareImage = program.args[0];
-const usbDir = `${__dirname}`;
-const blhost = uhk.getBlhostCmd(uhk.enumerationNameToProductId.bootloader);
 
-uhk.checkFirmwareImage(firmwareImage, extension);
+async function updateDeviceFirmware(firmwareImage, extension) {
+    const usbDir = `${__dirname}`;
+    const blhost = uhk.getBlhostCmd(uhk.enumerationNameToProductId.bootloader);
 
-(async function() {
+    uhk.checkFirmwareImage(firmwareImage, extension);
     config.verbose = true;
+
     await uhk.reenumerate('bootloader');
     exec(`${blhost} flash-security-disable 0403020108070605`);
     exec(`${blhost} flash-erase-region 0xc000 475136`);
     exec(`${blhost} flash-image ${firmwareImage}`);
     exec(`${blhost} reset`);
+
     config.verbose = false;
     echo('Firmware updated successfully.');
-})();
+};
+
+updateDeviceFirmware(firmwareImage, extension);