index.js 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240
  1. // Store setTimeout reference so promise-polyfill will be unaffected by
  2. // other code modifying setTimeout (like sinon.useFakeTimers())
  3. var setTimeoutFunc = setTimeout;
  4. function noop() {}
  5. // Polyfill for Function.prototype.bind
  6. function bind(fn, thisArg) {
  7. return function() {
  8. fn.apply(thisArg, arguments);
  9. };
  10. }
  11. function Promise(fn) {
  12. if (!(this instanceof Promise))
  13. throw new TypeError('Promises must be constructed via new');
  14. if (typeof fn !== 'function') throw new TypeError('not a function');
  15. this._state = 0;
  16. this._handled = false;
  17. this._value = undefined;
  18. this._deferreds = [];
  19. doResolve(fn, this);
  20. }
  21. function handle(self, deferred) {
  22. while (self._state === 3) {
  23. self = self._value;
  24. }
  25. if (self._state === 0) {
  26. self._deferreds.push(deferred);
  27. return;
  28. }
  29. self._handled = true;
  30. Promise._immediateFn(function() {
  31. var cb = self._state === 1 ? deferred.onFulfilled : deferred.onRejected;
  32. if (cb === null) {
  33. (self._state === 1 ? resolve : reject)(deferred.promise, self._value);
  34. return;
  35. }
  36. var ret;
  37. try {
  38. ret = cb(self._value);
  39. } catch (e) {
  40. reject(deferred.promise, e);
  41. return;
  42. }
  43. resolve(deferred.promise, ret);
  44. });
  45. }
  46. function resolve(self, newValue) {
  47. try {
  48. // Promise Resolution Procedure: https://github.com/promises-aplus/promises-spec#the-promise-resolution-procedure
  49. if (newValue === self)
  50. throw new TypeError('A promise cannot be resolved with itself.');
  51. if (
  52. newValue &&
  53. (typeof newValue === 'object' || typeof newValue === 'function')
  54. ) {
  55. var then = newValue.then;
  56. if (newValue instanceof Promise) {
  57. self._state = 3;
  58. self._value = newValue;
  59. finale(self);
  60. return;
  61. } else if (typeof then === 'function') {
  62. doResolve(bind(then, newValue), self);
  63. return;
  64. }
  65. }
  66. self._state = 1;
  67. self._value = newValue;
  68. finale(self);
  69. } catch (e) {
  70. reject(self, e);
  71. }
  72. }
  73. function reject(self, newValue) {
  74. self._state = 2;
  75. self._value = newValue;
  76. finale(self);
  77. }
  78. function finale(self) {
  79. if (self._state === 2 && self._deferreds.length === 0) {
  80. Promise._immediateFn(function() {
  81. if (!self._handled) {
  82. Promise._unhandledRejectionFn(self._value);
  83. }
  84. });
  85. }
  86. for (var i = 0, len = self._deferreds.length; i < len; i++) {
  87. handle(self, self._deferreds[i]);
  88. }
  89. self._deferreds = null;
  90. }
  91. function Handler(onFulfilled, onRejected, promise) {
  92. this.onFulfilled = typeof onFulfilled === 'function' ? onFulfilled : null;
  93. this.onRejected = typeof onRejected === 'function' ? onRejected : null;
  94. this.promise = promise;
  95. }
  96. /**
  97. * Take a potentially misbehaving resolver function and make sure
  98. * onFulfilled and onRejected are only called once.
  99. *
  100. * Makes no guarantees about asynchrony.
  101. */
  102. function doResolve(fn, self) {
  103. var done = false;
  104. try {
  105. fn(
  106. function(value) {
  107. if (done) return;
  108. done = true;
  109. resolve(self, value);
  110. },
  111. function(reason) {
  112. if (done) return;
  113. done = true;
  114. reject(self, reason);
  115. }
  116. );
  117. } catch (ex) {
  118. if (done) return;
  119. done = true;
  120. reject(self, ex);
  121. }
  122. }
  123. Promise.prototype['catch'] = function(onRejected) {
  124. return this.then(null, onRejected);
  125. };
  126. Promise.prototype.then = function(onFulfilled, onRejected) {
  127. var prom = new this.constructor(noop);
  128. handle(this, new Handler(onFulfilled, onRejected, prom));
  129. return prom;
  130. };
  131. Promise.prototype['finally'] = function(callback) {
  132. var constructor = this.constructor;
  133. return this.then(
  134. function(value) {
  135. return constructor.resolve(callback()).then(function() {
  136. return value;
  137. });
  138. },
  139. function(reason) {
  140. return constructor.resolve(callback()).then(function() {
  141. return constructor.reject(reason);
  142. });
  143. }
  144. );
  145. };
  146. Promise.all = function(arr) {
  147. return new Promise(function(resolve, reject) {
  148. if (!arr || typeof arr.length === 'undefined')
  149. throw new TypeError('Promise.all accepts an array');
  150. var args = Array.prototype.slice.call(arr);
  151. if (args.length === 0) return resolve([]);
  152. var remaining = args.length;
  153. function res(i, val) {
  154. try {
  155. if (val && (typeof val === 'object' || typeof val === 'function')) {
  156. var then = val.then;
  157. if (typeof then === 'function') {
  158. then.call(
  159. val,
  160. function(val) {
  161. res(i, val);
  162. },
  163. reject
  164. );
  165. return;
  166. }
  167. }
  168. args[i] = val;
  169. if (--remaining === 0) {
  170. resolve(args);
  171. }
  172. } catch (ex) {
  173. reject(ex);
  174. }
  175. }
  176. for (var i = 0; i < args.length; i++) {
  177. res(i, args[i]);
  178. }
  179. });
  180. };
  181. Promise.resolve = function(value) {
  182. if (value && typeof value === 'object' && value.constructor === Promise) {
  183. return value;
  184. }
  185. return new Promise(function(resolve) {
  186. resolve(value);
  187. });
  188. };
  189. Promise.reject = function(value) {
  190. return new Promise(function(resolve, reject) {
  191. reject(value);
  192. });
  193. };
  194. Promise.race = function(values) {
  195. return new Promise(function(resolve, reject) {
  196. for (var i = 0, len = values.length; i < len; i++) {
  197. values[i].then(resolve, reject);
  198. }
  199. });
  200. };
  201. // Use polyfill for setImmediate for performance gains
  202. Promise._immediateFn =
  203. (typeof setImmediate === 'function' &&
  204. function(fn) {
  205. setImmediate(fn);
  206. }) ||
  207. function(fn) {
  208. setTimeoutFunc(fn, 0);
  209. };
  210. Promise._unhandledRejectionFn = function _unhandledRejectionFn(err) {
  211. if (typeof console !== 'undefined' && console) {
  212. console.warn('Possible Unhandled Promise Rejection:', err); // eslint-disable-line no-console
  213. }
  214. };
  215. export default Promise;