You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
268 lines
6.8 KiB
268 lines
6.8 KiB
'use strict'; |
|
|
|
const toString = Object.prototype.toString; |
|
const colors = require('ansi-colors'); |
|
let called = false; |
|
let fns = []; |
|
|
|
const complements = { |
|
'yellow': 'blue', |
|
'cyan': 'red', |
|
'green': 'magenta', |
|
'black': 'white', |
|
'blue': 'yellow', |
|
'red': 'cyan', |
|
'magenta': 'green', |
|
'white': 'black' |
|
}; |
|
|
|
exports.longest = (arr, prop) => { |
|
return arr.reduce((a, v) => Math.max(a, prop ? v[prop].length : v.length), 0); |
|
}; |
|
|
|
exports.hasColor = str => !!str && colors.hasColor(str); |
|
|
|
const isObject = exports.isObject = val => { |
|
return val !== null && typeof val === 'object' && !Array.isArray(val); |
|
}; |
|
|
|
exports.nativeType = val => { |
|
return toString.call(val).slice(8, -1).toLowerCase().replace(/\s/g, ''); |
|
}; |
|
|
|
exports.isAsyncFn = val => { |
|
return exports.nativeType(val) === 'asyncfunction'; |
|
}; |
|
|
|
exports.isPrimitive = val => { |
|
return val != null && typeof val !== 'object' && typeof val !== 'function'; |
|
}; |
|
|
|
exports.resolve = (context, value, ...rest) => { |
|
if (typeof value === 'function') { |
|
return value.call(context, ...rest); |
|
} |
|
return value; |
|
}; |
|
|
|
exports.scrollDown = (choices = []) => [...choices.slice(1), choices[0]]; |
|
exports.scrollUp = (choices = []) => [choices.pop(), ...choices]; |
|
|
|
exports.reorder = (arr = []) => { |
|
let res = arr.slice(); |
|
res.sort((a, b) => { |
|
if (a.index > b.index) return 1; |
|
if (a.index < b.index) return -1; |
|
return 0; |
|
}); |
|
return res; |
|
}; |
|
|
|
exports.swap = (arr, index, pos) => { |
|
let len = arr.length; |
|
let idx = pos === len ? 0 : pos < 0 ? len - 1 : pos; |
|
let choice = arr[index]; |
|
arr[index] = arr[idx]; |
|
arr[idx] = choice; |
|
}; |
|
|
|
exports.width = (stream, fallback = 80) => { |
|
let columns = (stream && stream.columns) ? stream.columns : fallback; |
|
if (stream && typeof stream.getWindowSize === 'function') { |
|
columns = stream.getWindowSize()[0]; |
|
} |
|
if (process.platform === 'win32') { |
|
return columns - 1; |
|
} |
|
return columns; |
|
}; |
|
|
|
exports.height = (stream, fallback = 20) => { |
|
let rows = (stream && stream.rows) ? stream.rows : fallback; |
|
if (stream && typeof stream.getWindowSize === 'function') { |
|
rows = stream.getWindowSize()[1]; |
|
} |
|
return rows; |
|
}; |
|
|
|
exports.wordWrap = (str, options = {}) => { |
|
if (!str) return str; |
|
|
|
if (typeof options === 'number') { |
|
options = { width: options }; |
|
} |
|
|
|
let { indent = '', newline = ('\n' + indent), width = 80 } = options; |
|
let spaces = (newline + indent).match(/[^\S\n]/g) || []; |
|
width -= spaces.length; |
|
let source = `.{1,${width}}([\\s\\u200B]+|$)|[^\\s\\u200B]+?([\\s\\u200B]+|$)`; |
|
let output = str.trim(); |
|
let regex = new RegExp(source, 'g'); |
|
let lines = output.match(regex) || []; |
|
lines = lines.map(line => line.replace(/\n$/, '')); |
|
if (options.padEnd) lines = lines.map(line => line.padEnd(width, ' ')); |
|
if (options.padStart) lines = lines.map(line => line.padStart(width, ' ')); |
|
return indent + lines.join(newline); |
|
}; |
|
|
|
exports.unmute = color => { |
|
let name = color.stack.find(n => colors.keys.color.includes(n)); |
|
if (name) { |
|
return colors[name]; |
|
} |
|
let bg = color.stack.find(n => n.slice(2) === 'bg'); |
|
if (bg) { |
|
return colors[name.slice(2)]; |
|
} |
|
return str => str; |
|
}; |
|
|
|
exports.pascal = str => str ? str[0].toUpperCase() + str.slice(1) : ''; |
|
|
|
exports.inverse = color => { |
|
if (!color || !color.stack) return color; |
|
let name = color.stack.find(n => colors.keys.color.includes(n)); |
|
if (name) { |
|
let col = colors['bg' + exports.pascal(name)]; |
|
return col ? col.black : color; |
|
} |
|
let bg = color.stack.find(n => n.slice(0, 2) === 'bg'); |
|
if (bg) { |
|
return colors[bg.slice(2).toLowerCase()] || color; |
|
} |
|
return colors.none; |
|
}; |
|
|
|
exports.complement = color => { |
|
if (!color || !color.stack) return color; |
|
let name = color.stack.find(n => colors.keys.color.includes(n)); |
|
let bg = color.stack.find(n => n.slice(0, 2) === 'bg'); |
|
if (name && !bg) { |
|
return colors[complements[name] || name]; |
|
} |
|
if (bg) { |
|
let lower = bg.slice(2).toLowerCase(); |
|
let comp = complements[lower]; |
|
if (!comp) return color; |
|
return colors['bg' + exports.pascal(comp)] || color; |
|
} |
|
return colors.none; |
|
}; |
|
|
|
exports.meridiem = date => { |
|
let hours = date.getHours(); |
|
let minutes = date.getMinutes(); |
|
let ampm = hours >= 12 ? 'pm' : 'am'; |
|
hours = hours % 12; |
|
let hrs = hours === 0 ? 12 : hours; |
|
let min = minutes < 10 ? '0' + minutes : minutes; |
|
return hrs + ':' + min + ' ' + ampm; |
|
}; |
|
|
|
/** |
|
* Set a value on the given object. |
|
* @param {Object} obj |
|
* @param {String} prop |
|
* @param {any} value |
|
*/ |
|
|
|
exports.set = (obj = {}, prop = '', val) => { |
|
return prop.split('.').reduce((acc, k, i, arr) => { |
|
let value = arr.length - 1 > i ? (acc[k] || {}) : val; |
|
if (!exports.isObject(value) && i < arr.length - 1) value = {}; |
|
return (acc[k] = value); |
|
}, obj); |
|
}; |
|
|
|
/** |
|
* Get a value from the given object. |
|
* @param {Object} obj |
|
* @param {String} prop |
|
*/ |
|
|
|
exports.get = (obj = {}, prop = '', fallback) => { |
|
let value = obj[prop] == null |
|
? prop.split('.').reduce((acc, k) => acc && acc[k], obj) |
|
: obj[prop]; |
|
return value == null ? fallback : value; |
|
}; |
|
|
|
exports.mixin = (target, b) => { |
|
if (!isObject(target)) return b; |
|
if (!isObject(b)) return target; |
|
for (let key of Object.keys(b)) { |
|
let desc = Object.getOwnPropertyDescriptor(b, key); |
|
if (desc.hasOwnProperty('value')) { |
|
if (target.hasOwnProperty(key) && isObject(desc.value)) { |
|
let existing = Object.getOwnPropertyDescriptor(target, key); |
|
if (isObject(existing.value)) { |
|
target[key] = exports.merge({}, target[key], b[key]); |
|
} else { |
|
Reflect.defineProperty(target, key, desc); |
|
} |
|
} else { |
|
Reflect.defineProperty(target, key, desc); |
|
} |
|
} else { |
|
Reflect.defineProperty(target, key, desc); |
|
} |
|
} |
|
return target; |
|
}; |
|
|
|
exports.merge = (...args) => { |
|
let target = {}; |
|
for (let ele of args) exports.mixin(target, ele); |
|
return target; |
|
}; |
|
|
|
exports.mixinEmitter = (obj, emitter) => { |
|
let proto = emitter.constructor.prototype; |
|
for (let key of Object.keys(proto)) { |
|
let val = proto[key]; |
|
if (typeof val === 'function') { |
|
exports.define(obj, key, val.bind(emitter)); |
|
} else { |
|
exports.define(obj, key, val); |
|
} |
|
} |
|
}; |
|
|
|
exports.onExit = callback => { |
|
const onExit = (quit, code) => { |
|
if (called) return; |
|
|
|
called = true; |
|
fns.forEach(fn => fn()); |
|
|
|
if (quit === true) { |
|
process.exit(128 + code); |
|
} |
|
}; |
|
|
|
if (fns.length === 0) { |
|
process.once('SIGTERM', onExit.bind(null, true, 15)); |
|
process.once('SIGINT', onExit.bind(null, true, 2)); |
|
process.once('exit', onExit); |
|
} |
|
|
|
fns.push(callback); |
|
}; |
|
|
|
exports.define = (obj, key, value) => { |
|
Reflect.defineProperty(obj, key, { value }); |
|
}; |
|
|
|
exports.defineExport = (obj, key, fn) => { |
|
let custom; |
|
Reflect.defineProperty(obj, key, { |
|
enumerable: true, |
|
configurable: true, |
|
set(val) { |
|
custom = val; |
|
}, |
|
get() { |
|
return custom ? custom() : fn(); |
|
} |
|
}); |
|
};
|
|
|