coveo-search-ui/src/misc/PromisesShim.ts

Coveo JavaScript Search Framework

/* istanbul ignore next */
export function shim() {
  const doShim = promiseInstance => {
    if (typeof promiseInstance.prototype['finally'] != 'function') {
      promiseInstance.prototype['finally'] = function finallyPolyfill(callback) {
        let constructor = this.constructor;
        return this.then(
          function(value) {
            return constructor.resolve(callback()).then(function() {
              return value;
            });
          },
          function(reason) {
            return constructor.resolve(callback()).then(function() {
              throw reason;
            });
          }
        );
      };
    }

    let rethrowError = self => {
      self.then(null, function(err) {
        setTimeout(function() {
          throw err;
        }, 0);
      });
    };

    if (typeof promiseInstance.prototype['done'] !== 'function') {
      promiseInstance.prototype['done'] = function(onFulfilled, onRejected) {
        let self = arguments.length ? this.then.apply(this, arguments) : this;
        rethrowError(self);
        return this;
      };
    }

    if (typeof promiseInstance.prototype['fail'] !== 'function') {
      promiseInstance.prototype['fail'] = function(onFulfilled, onRejected) {
        let self = arguments.length ? this.catch.apply(this, arguments) : this;
        rethrowError(self);
        return this;
      };
    }
  };

  const globalPromise = window['Promise'];
  const localPromise = Promise;
  if (globalPromise) {
    doShim(globalPromise);
  }
  if (localPromise) {
    doShim(localPromise);
  }
}