telegram-mtproto/lib/smart-timeout.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.smartTimeout = undefined;

var _bluebird = require('bluebird');

var _bluebird2 = _interopRequireDefault(_bluebird);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

const cancelToken = Symbol('cancel token');

const timeoutRefs = new WeakSet();

const pause = delay => new _bluebird2.default(r => setTimeout(r, delay));

const smartTimeout = exports.smartTimeout = (fn, delay = 0, ...args) => {
  const newToken = Symbol('cancel id');
  const checkRun = () => {
    if (timeoutRefs.has(newToken)) {
      timeoutRefs.delete(newToken);
      return fn(...args);
    } else return false;
  };
  const promise = pause(delay).then(checkRun);
  promise[cancelToken] = newToken;
  return promise;
};

smartTimeout.cancel = promise => {
  if (!promise || !promise[cancelToken]) return false;
  const token = promise[cancelToken];
  return timeoutRefs.has(token) ? timeoutRefs.delete(token) : false;
};

smartTimeout.immediate = (fn, ...args) => _bluebird2.default.resolve().then(() => fn(...args));

smartTimeout.promise = (fn, delay = 0, ...args) => pause(delay).then(() => fn(...args));

exports.default = smartTimeout;
//# sourceMappingURL=smart-timeout.js.map