Entity.js

const yaml = require('js-yaml');
const _ = require('lodash');

class Entity {
    // get all the information about the entity
    getInfo() {
        return this.info;
    }

    // default Columns means all the fields we want to show at when using get and the user is not specify which kind of output he want (not include a lot of information)
    toDefault() {
        return this.extractValues(this.defaultColumns);
    }

    // wide Columns means all the fields we want to show when using get with -o wide option (return more information then the defualt)
    toWide() {
        return this.extractValues(this.wideColumns);
    }

    toJson() {
        return JSON.stringify(this.info, null, 4);
    }

    toYaml() {
        return yaml.dump(this.info);
    }

    toName() {
        return String(`${this.entityType}/${this.name || this.info.name}`);
    }

    toId() {
        return String(this.id || this.info.id);
    }

    extractValues(columns) {
        const values = {};
        columns.forEach((key) => {
            if (_.get(this, key)) {
                values[key.replace(/^.*\./, '')] = _.get(this, key);
            } else {
                values[key.replace(/^.*\./, '')] = _.get(this.info, key);
            }
        });
        return values;
    }
}

module.exports = Entity;