|
1 | 1 | /* Copyright (c) 2012 Rod Vagg <@rvagg> */ |
2 | | -const fs = require('fs') |
3 | | -const path = require('path') |
4 | | -const after = require('after') |
| 2 | +import fs from 'fs/promises' |
| 3 | +import path from 'path' |
5 | 4 |
|
6 | | -function readfiletree (dir, obj, callback) { |
7 | | - if (typeof obj === 'function') { |
8 | | - callback = obj |
| 5 | +export async function readfiletree (dir, obj) { |
| 6 | + if (obj == null) { |
9 | 7 | obj = {} |
10 | 8 | } |
11 | | - |
12 | | - fs.readdir(dir, afterReaddir) |
13 | | - |
14 | | - function afterReaddir (err, list) { |
15 | | - if (err) { |
16 | | - return callback(err) |
17 | | - } |
18 | | - |
19 | | - const done = after(list.length, (err) => { |
20 | | - if (err) { |
21 | | - return callback(err) |
22 | | - } |
23 | | - callback(null, obj) |
24 | | - }) |
25 | | - list.forEach((f) => eachFile(f, done)) |
26 | | - } |
27 | | - |
28 | | - function eachFile (f, callback) { |
29 | | - var p = path.join(dir, f) |
30 | | - fs.stat(p, (err, stat) => { |
31 | | - if (err) { |
32 | | - return callback(err) |
33 | | - } |
34 | | - if (stat.isDirectory()) { |
35 | | - obj[f] = {} |
36 | | - return readfiletree(p, obj[f], callback) |
37 | | - } else if (stat.isFile()) { |
38 | | - fs.readFile(p, 'utf8', (err, data) => { |
39 | | - if (err) { |
40 | | - return callback(err) |
41 | | - } |
42 | | - obj[f] = data |
43 | | - callback() |
44 | | - }) |
45 | | - } |
46 | | - }) |
47 | | - } |
48 | | -} |
49 | | - |
50 | | -module.exports = function maybePromiseWrap (dir, callback) { |
51 | | - if (typeof callback === 'function') { |
52 | | - return readfiletree(dir, callback) |
53 | | - } |
54 | | - |
55 | | - return new Promise((resolve, reject) => { |
56 | | - callback = (err, data) => { |
57 | | - if (err) { |
58 | | - return reject(err) |
59 | | - } |
60 | | - resolve(data) |
| 9 | + const list = await fs.readdir(dir) |
| 10 | + await Promise.all(list.map(async (f) => { |
| 11 | + const p = path.join(dir, f) |
| 12 | + const stat = await fs.stat(p) |
| 13 | + if (stat.isDirectory()) { |
| 14 | + obj[f] = {} |
| 15 | + await readfiletree(p, obj[f]) |
| 16 | + } else if (stat.isFile()) { |
| 17 | + obj[f] = await fs.readFile(p, 'utf8') |
61 | 18 | } |
62 | | - |
63 | | - readfiletree(dir, callback) |
64 | | - }) |
| 19 | + })) |
| 20 | + return obj |
65 | 21 | } |
0 commit comments