timeout-promise.ts

class TPromise<T> extends Promise<T> {
  static TimeoutError = class extends Error {};

  constructor(
    f: (
      resolve: (value: T | PromiseLike<T>) => void,
      reject: (reason?: any) => void
    ) => void,
    timeout: number = Infinity
  ) {
    let resolve, reject;
    super((res, rej) => {
      resolve = res;
      reject = rej;
    });

    const timer =
      timeout === Infinity
        ? null
        : setTimeout(() => {
            reject(
              new TPromise.TimeoutError(`Load timeout for promise: ${timeout}`)
            );
          }, timeout);

    f(
      (val: T | PromiseLike<T>) => {
        resolve(val);
        clearTimeout(timer);
      },
      (err) => {
        reject(err);
        clearTimeout(timer);
      }
    );
  }
}