signal-desktop/ts/textsecure/TaskWithTimeout.ts

// tslint:disable no-default-export

export default function createTaskWithTimeout(
  task: () => Promise<any>,
  id: string,
  options: { timeout?: number } = {}
) {
  const timeout = options.timeout || 1000 * 60 * 2; // two minutes

  const errorForStack = new Error('for stack');

  return async () =>
    new Promise((resolve, reject) => {
      let complete = false;
      let timer: any = setTimeout(() => {
        if (!complete) {
          const message = `${id ||
            ''} task did not complete in time. Calling stack: ${
            errorForStack.stack
          }`;

          window.log.error(message);
          reject(new Error(message));

          return;
        }

        return null;
      }, timeout);
      const clearTimer = () => {
        try {
          const localTimer = timer;
          if (localTimer) {
            timer = null;
            clearTimeout(localTimer);
          }
        } catch (error) {
          window.log.error(
            id || '',
            'task ran into problem canceling timer. Calling stack:',
            errorForStack.stack
          );
        }
      };

      const success = (result: any) => {
        clearTimer();
        complete = true;
        resolve(result);

        return;
      };
      const failure = (error: Error) => {
        clearTimer();
        complete = true;
        reject(error);

        return;
      };

      let promise;
      try {
        promise = task();
      } catch (error) {
        clearTimer();
        throw error;
      }
      if (!promise || !promise.then) {
        clearTimer();
        complete = true;
        resolve(promise);

        return;
      }

      return promise.then(success, failure);
    });
}