haroldtreen/epub-press

class StatusTracker {
    constructor() {
        this.statuses = {};
        this.keys = [];
    }

    reset() {
        this.statuses = {};
        this.keys = [];
    }

    setStatus(id, statusType) {
        this.statuses[id] = StatusTracker.buildStatus(statusType);
        if (this.keys.indexOf(id) < 0) {
            this.keys.push(id);
        }
        limitStatuses(this); // eslint-disable-line
    }

    getStatus(id) {
        return this.statuses[id];
    }

    clearStatus(id) {
        delete this.statuses[id];
    }
}

function limitStatuses(tracker) {
    if (tracker.keys.length < StatusTracker.MAX_STATUSES) {
        return;
    }

    tracker.clearStatus(tracker.keys.shift());
}

StatusTracker.buildStatus = (statusType) =>
    StatusTracker.STATUSES[statusType] || StatusTracker.STATUSES.DEFAULT;

StatusTracker.MAX_STATUSES = 100;
StatusTracker.STATUSES = {
    DEFAULT: { message: 'Unknown Status', progress: 0 },
    PUBLISHING: { message: 'Publishing...', progress: 5 },
    FETCHING_HTML: { message: 'Fetching HTML...', progress: 10 },
    EXTRACTING_CONTENT: { message: 'Extracting Content...', progress: 30 },
    FETCHING_IMAGES: { message: 'Fetching Images...', progress: 70 },
    FORMATTING_HTML: { message: 'Formatting HTML...', progress: 80 },
    CREATING_COVER: { message: 'Creating cover...', progress: 85 },
    WRITING_EBOOK: { message: 'Writing Ebook...', progress: 90 },
    DONE: { message: 'Done!', progress: 100 },
    FAILED: { message: 'Publish Failed :(', progress: 100, httpStatus: 500 },
};
StatusTracker.STATUS_TYPES = Object.keys(StatusTracker.STATUSES).reduce((types, statusKey) => {
    // eslint-disable-next-line
    types[statusKey] = statusKey;
    return types;
}, {});

module.exports = StatusTracker;