PromiseTracker.ts

import Deferred from '@/base/deferred'

class PromiseTracker {
    private readonly deferred = new Deferred()
    constructor() {
        this.deferred.catch()
    }

    public track<T extends object>(obj: T): T {
        return this._track(obj)
    }

    public rejectAll() {
        this.deferred.reject('rejected by tracker')
    }

    private _track(obj: any): any {
        const _this = this
        return new Proxy(obj, {
            get(target, key, receiver) {
                return _this._track(Reflect.get(target, key, receiver))
            },
            apply(target, thisArg, argArray) {
                const ret = Reflect.apply(target, thisArg, argArray)
                if (ret instanceof Promise) {
                    const d = new Deferred()
                    ret.then(r => {
                        d.resolve(_this._track(r))
                    }).catch(err => {
                        d.reject(_this._track(err))
                    })

                    _this.deferred
                        .catch(err => {
                            d.reject(_this._track(err))
                        })
                    return _this._track(d.then())
                }
                return _this._track(ret)
            }
        })
    }
}