-
Notifications
You must be signed in to change notification settings - Fork 80
/
utils.js
65 lines (58 loc) · 1.21 KB
/
utils.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
"use strict";
const chalk = require("chalk");
let nextTick;
if (typeof setImmediate === "function") {
nextTick = setImmediate;
} else if (typeof process === "object" && process && process.nextTick) {
nextTick = process.nextTick;
} else {
nextTick = cb => {
setTimeout(cb, 0);
};
}
function nodeify(promise, cb) {
if (typeof cb !== "function") {
return promise;
}
return promise
.then(res => {
nextTick(() => {
cb(null, res);
});
})
.catch(err => {
cb(err);
});
}
function nodeifySelf(cb) {
return nodeify(this, cb);
}
Object.assign(Promise.prototype, {
nodeify: nodeifySelf
});
const co = genfun => {
const gen = genfun();
const next = value => {
const ret = gen.next(value);
if (ret.done) return;
ret.value((err, val) => {
if (err) {
return console.error(err);
}
return next(val);
});
};
next();
};
const log = (color, args) => console.log(chalk[color](args));
const print = Object.create(null);
print.warn = (...args) => log("yellow", args);
print.err = (...args) => log("red", args);
print.info = (...args) => log("green", args);
const noop = () => {};
module.exports = {
nodeify,
co,
print,
noop
};