@codefresh-io/pipeline-autobot/analyzer/analyzer.context.js

const _ = require('lodash');

const Github = require('../providers/github');
const { comment, emptyLine } = require('../json-to-yaml');

class AnalyzerContext {

    constructor(context){
        this.steps = {};
        this.groups = {};
        this.context = context;
        this.headers = {
            [comment()]: 'More examples of Codefresh YAML can be found at',
            [comment()]: 'https://codefresh.io/docs/docs/yaml-examples/examples/',
            [emptyLine()]: true,
            version: '1.0',
            [comment()]: 'Stages can help you organize your steps in stages'
        };
        const { repo: { repoOwner, repoName, branch } } = this.context;
        this.github = new Github(context.credentials, repoOwner, repoName, branch);
    }

    async loadRepoInfo(){
        this.repoInfo = await this.github.get();
        //something go wrong
        if(this.repoInfo.languageStatistics.message){
            throw new Error(this.repoInfo.languageStatistics.message);
        }
        this.context.repoInfo = this.repoInfo;
    }

    getSteps(){
        return this.steps;
    }

    getGroup(key) {
        return this.groups[key];
    }

    getHeaders(){
        return this.headers;
    }


    getContext(){
        return this.context;
    }

    addStep(info){
        this._checkForCommentsAndAdd(info.comments, this.steps);

        if(_.isArray(info)){
            info.forEach(i => {
                this.steps[i.key] = i.step;
                this.groups[i.key] = i.group;
            })
        }
        else {
            this.steps[info.key] = info.step;
            this.groups[info.key] = info.group;
        }
    }

    addHeader(info){
        this._checkForCommentsAndAdd(info.comments, this.headers);

        this.headers[info.key] = info.header;
    }

    getGithub(){
        return this.github;
    }

    /**
     * Check 'info' object for 'comments' array. If comments exist insert them before object.
     * @param {*} comments - step/header comments
     * @param {*} partition - 'headers' or 'steps' object
     */
    _checkForCommentsAndAdd(comments, partition) {
        if (_.isArray(comments)) {
            comments.forEach(c => {
                partition[comment()] = c;
            })
        }
    }

}
module.exports = AnalyzerContext;