fix: trammel: promisify

Author: coderaiser

lib/trammel.js

@@ -1,5 +1,6 @@
 'use strict';
 
+const {promisify} = require('node:util');
 const {lstat, readdir} = require('fs/promises');
 const path = require('path');
 const {EventEmitter} = require('events');
@@ -12,18 +13,20 @@ const tryToCatch = require('try-to-catch');
     shall return information about the link, while stat() shall return
     information about the file the link references.
 */
-module.exports = async (dir, options, callback) => {
+module.exports = async (dir, options = {}) => {
     const emitter = new EventEmitter();
 
-    let total = 0;
-    let type;
+    const [total] = await Promise.all([
+        subscribe(emitter, options),
+        processDir(dir, options, emitter),
+    ]);
 
-    if (!callback) {
-        callback = options;
-        options = {};
-    } else {
-        type = options.type;
-    }
+    return total;
+};
+
+const subscribe = promisify((emitter, options, callback) => {
+    const {type} = options;
+    let total = 0;
 
     emitter.on('file', (file, stat) => {
         total += stat.size;
@@ -37,9 +40,7 @@ module.exports = async (dir, options, callback) => {
 
         callback(null, format.size(total));
     });
-    
-    await processDir(dir, options, emitter);
-};
+});
 
 async function processDir(dir, options, emitter) {
     const {stopOnError} = options;