ForestAdmin/toolbelt

const agent = require('superagent');
const ProgressBar = require('progress');
const { promisify } = require('util');
const Context = require('@forestadmin/context');

const jobDeserializer = require('../deserializers/job');

const setTimeoutAsync = promisify(setTimeout);

function JobStateChecker(message, oclifExit) {
  const { assertPresent, authenticator, env, logger } = Context.inject();
  assertPresent({ authenticator, env, logger });

  const bar = new ProgressBar(`${message} [:bar] :percent `, { total: 100 });
  bar.update(0);

  this.check = async jobId => {
    try {
      const jobResponse = await agent
        .get(`${env.FOREST_SERVER_URL}/api/jobs/${jobId}`)
        .set('Authorization', `Bearer ${authenticator.getAuthToken()}`)
        .then(response => jobDeserializer.deserialize(response.body));

      if (jobResponse && jobResponse.state) {
        let isBarComplete = false;
        if (jobResponse.progress) {
          bar.update(jobResponse.progress / 100);
          isBarComplete = bar.complete;
        }
        if (jobResponse.state !== 'inactive' && jobResponse.state !== 'active') {
          if (jobResponse.state === 'complete' && !isBarComplete) {
            bar.update(1);
          }
          return jobResponse.state !== 'failed';
        }
      }
    } catch (error) {
      if (!error.status) {
        throw error;
      }

      logger.error(`HTTP ${error.status}: ${error.message}`);
      oclifExit(100);
    }

    await setTimeoutAsync(1000);
    return this.check(jobId);
  };
}

module.exports = JobStateChecker;