123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113 |
- 'use strict';
- const processFn = (fn, options, proxy, unwrapped) => function (...arguments_) {
- const P = options.promiseModule;
- return new P((resolve, reject) => {
- if (options.multiArgs) {
- arguments_.push((...result) => {
- if (options.errorFirst) {
- if (result[0]) {
- reject(result);
- } else {
- result.shift();
- resolve(result);
- }
- } else {
- resolve(result);
- }
- });
- } else if (options.errorFirst) {
- arguments_.push((error, result) => {
- if (error) {
- reject(error);
- } else {
- resolve(result);
- }
- });
- } else {
- arguments_.push(resolve);
- }
- const self = this === proxy ? unwrapped : this;
- Reflect.apply(fn, self, arguments_);
- });
- };
- const filterCache = new WeakMap();
- module.exports = (input, options) => {
- options = {
- exclude: [/.+(?:Sync|Stream)$/],
- errorFirst: true,
- promiseModule: Promise,
- ...options
- };
- const objectType = typeof input;
- if (!(input !== null && (objectType === 'object' || objectType === 'function'))) {
- throw new TypeError(`Expected \`input\` to be a \`Function\` or \`Object\`, got \`${input === null ? 'null' : objectType}\``);
- }
- const filter = (target, key) => {
- let cached = filterCache.get(target);
- if (!cached) {
- cached = {};
- filterCache.set(target, cached);
- }
- if (key in cached) {
- return cached[key];
- }
- const match = pattern => (typeof pattern === 'string' || typeof key === 'symbol') ? key === pattern : pattern.test(key);
- const desc = Reflect.getOwnPropertyDescriptor(target, key);
- const writableOrConfigurableOwn = (desc === undefined || desc.writable || desc.configurable);
- const included = options.include ? options.include.some(match) : !options.exclude.some(match);
- const shouldFilter = included && writableOrConfigurableOwn;
- cached[key] = shouldFilter;
- return shouldFilter;
- };
- const cache = new WeakMap();
- const proxy = new Proxy(input, {
- apply(target, thisArg, args) {
- const cached = cache.get(target);
- if (cached) {
- return Reflect.apply(cached, thisArg, args);
- }
- const pified = options.excludeMain ? target : processFn(target, options, proxy, target);
- cache.set(target, pified);
- return Reflect.apply(pified, thisArg, args);
- },
- get(target, key) {
- const property = target[key];
- // eslint-disable-next-line no-use-extend-native/no-use-extend-native
- if (!filter(target, key) || property === Function.prototype[key]) {
- return property;
- }
- const cached = cache.get(property);
- if (cached) {
- return cached;
- }
- if (typeof property === 'function') {
- const pified = processFn(property, options, proxy, target);
- cache.set(property, pified);
- return pified;
- }
- return property;
- }
- });
- return proxy;
- };
|