From 2dadce739117cd84126e79fa11885df95f778642 Mon Sep 17 00:00:00 2001
From: WubzyGD <wubzygd@gmail.com>
Date: Thu, 27 Oct 2022 02:13:15 -0600
Subject: [PATCH] ora promise wrap

---
 src/util/log/ora.js | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/src/util/log/ora.js b/src/util/log/ora.js
index 7b0f6ca..230f4ea 100644
--- a/src/util/log/ora.js
+++ b/src/util/log/ora.js
@@ -1,12 +1,18 @@
 const ora = require('ora');
 
 module.exports = async (text, promise, callback) => {
-    let spin = ora(text).start();
-    await promise
-        .catch(e => callback(spin, e))
-        .then((...args) => {
-            spin.stop();
-            spin.clear();
-            callback(spin, args);
-        });
+    return new Promise(async (resolve, reject) => {
+        let spin = ora(text).start();
+        return await promise
+            .catch(e => {
+                spin.stop();
+                spin.clear();
+                return reject(e)
+            })
+            .then((...args) => {
+                spin.stop();
+                spin.clear();
+                return callback ? resolve(callback(spin, args)) : resolve(true);
+            });
+    });
 };
\ No newline at end of file