Inlife/nexrender

const EventEmitter = require('events');

const NEXRENDER_JOB_POLLING = process.env.NEXRENDER_JOB_POLLING || 10 * 1000;

const withEventEmitter = (wrappedFetch, job, polling = NEXRENDER_JOB_POLLING) => {
    const emitter  = new EventEmitter();
    const interval = setInterval(async () => {
        try {
            const updatedJob = await wrappedFetch(`/jobs/${job.uid}/status`)

            // Support updating render progress throughout rendering process
            if (updatedJob.state == 'render:dorender' && updatedJob.renderProgress) {
                emitter.emit('progress', updatedJob, updatedJob.renderProgress);
            }

            if (updatedJob.state == 'finished') {
                emitter.emit('progress', updatedJob, 100);
            }

            // push only strict state changes
            if (job.state != updatedJob.state) {
                job.state = updatedJob.state;
                emitter.emit(job.state, updatedJob, wrappedFetch);
            }

            if (job.state == 'finished' || job.state == 'error') {
                clearInterval(interval);
            }
        } catch (err) {
            clearInterval(interval);
            emitter.emit('error', err);
        }

    }, polling);

    /* trigger first callback */
    setImmediate(() => emitter.emit('created', job))

    return emitter;
}

module.exports = (wrappedFetch, polling) => ({
    listJobs: async () => await wrappedFetch(`/jobs`),
    fetchJob: async id => await wrappedFetch(`/jobs/${id}`),
    pickupJob: async (selector) => await wrappedFetch(`/jobs/pickup${ selector ? `/${selector}` : '' }`),

    addJob: async data =>
        withEventEmitter(wrappedFetch, await wrappedFetch(`/jobs`, {
            method: 'post',
            headers: { 'content-type': 'application/json' },
            body: JSON.stringify(data),
        }), polling),

    resumeJob: async id =>
        withEventEmitter(wrappedFetch, await wrappedFetch(`/jobs/${id}`), polling),

    updateJob: async (id, data) =>
        await wrappedFetch(`/jobs/${id}`, {
            method: 'put',
            headers: { 'content-type': 'application/json' },
            body: JSON.stringify(data),
        }),

    removeJob: async id =>
        await wrappedFetch(`/jobs/${id}`, {
            method: 'delete'
        }),
})

/* deprecated method name */
/* keeping for now for, backward compatibility */
module.exports.removejob = async id => {
    console.warn("`removejob()` has been deprecated and will be removed in a future version. Please use `removeJob()` instead.")
    return module.exports.removeJob(id)
}
module.exports.getJob = async id => {
    console.warn("`getJob()` has been deprecated and will be removed in a future version. Please use `resumeJob()` instead.")
    return module.exports.resumeJob(id)
}