SfExtension.js

const Generator = require('yeoman-generator');

/**
 * @typedef {import('@salesforce/command').SfdxCommand} SfdxCommand
 */

/**
 * @typedef {object} SfExtensionConfig
 * @param {object} config oclif/sfdx runtime configs
 * @param {SfdxCommand} command the sfdx command to which this extension is currently attached
 * @param {string} lifecycle 'before' or 'after' which indicates when this extension is running in relation to the sfdxCommand
 */

class SfExtension extends Generator{
    /**
     * constructor
     * @param {Array<string>} args yeoman args
     * @param {SfExtensionConfig} config
     */
    constructor(yoArgs, sfContext){
        super(yoArgs, sfContext);
        // default desintationRoot to process.cwd()
        this.destinationRoot(process.cwd());
        // set sfContext
        let { config, Command, lifecycle, yargs } = sfContext;
        this.sfContext = {
            config, 
            Command, 
            lifecycle,
            yargs
        };
    }

    set forceOverwrite(forceOverwrite){
        if(!this.env.conflicter) throw new Error('Must set forceOverwrite during Yeoman running context');
        if(forceOverwrite && typeof forceOverwrite === 'boolean'){
            this.env.conflicter.force = forceOverwrite;
        }
    }
}

module.exports = SfExtension;