honjeng-build-cli/bin/generator.js

一个mini版的脚手架

const path = require('path');
const fs = require('fs-extra');
const ora = require('ora');
const inquirer = require('inquirer');

const downloadGitRepo = require('download-git-repo');

const util = require('util');

const { getRepolist } = require('./http');

async function wrapLoading(fn, message, ...args) {
    const spinner = ora(message);

    spinner.start();

    try {
        const result = await fn(...args);
        spinner.succeed();
        return result;
    } catch (e) {
        spinner.fail('Request failed ....');
    }
}

class Generator {
    constructor(name, target, ask) {
        this.name = name;
        this.target = target;
        this.ask = ask;
        this.downloadGitRepo = util.promisify(downloadGitRepo);
    }

    async getRepo() {
        const repoList = await wrapLoading(getRepolist, 'waiting fetch template');
        if (!repoList) return;

        const repos = repoList.map(item => item.name);

        const { repo } = await inquirer.prompt({
            name: 'repo',
            type: 'list',
            choices: repos,
            message: 'Please choose a template'
        })

        return repo;
    }

    async download(repo, tag) {
        const requestUrl = `yuan-cli/${repo}`;

        await wrapLoading(
            this.downloadGitRepo,
            'waiting download template',
            requestUrl,
            path.resolve(process.cwd(), this.target),
        )
    }

    async create() {
        const repo = await this.getRepo();
        console.log("用户选择了", repo);

        await this.download(repo);

        let targetPath = path.resolve(process.cwd(), this.target);

        let jsonPath = path.join(targetPath, 'package.json');

        if (fs.existsSync(jsonPath)) {
            const data = fs.readFileSync(jsonPath).toString();

            let json = JSON.parse(data);
            json.name = this.name;

            Object.keys(this.ask).forEach((item) => {
                json[item] = this.ask[item];
            });

            fs.writeFileSync(jsonPath, JSON.stringify(json, null, '\t'), 'utf-8');
        }
    }
}

module.exports = Generator;