gub/src/index.js

> 接入成本极低，快速从原有项目(可运行项目)中clone出新项目，并重写package.json，安装依赖

import program from "commander"
import nanoid from "nanoid"
import log from "./log"
import fs from "fs-extra"
import path from "path"
import username from "git-username"
import inquirer from "inquirer"
import execa from "execa"
import chalk from "chalk"
import mkdirp from "mkdirp"
import { Spinner } from "cli-spinner"
import getenv from "getenv"
import escapeStringRegexp from "escape-string-regexp"
import {
    createAliasViewer,
    selectAliasViewer,
    updateAliasViewer,
    removeAliasViewer,
    addAlias
} from "./alias"
import parseGithubUrl from "parse-github-url"

const banner =
    "\n\n|||||||||||||||||||||||||||||||||||||||||||||||||||||\n\n Welcome use gub to init your npm package.\n\n  https://github.com/janryWang/gub\n\n|||||||||||||||||||||||||||||||||||||||||||||||||||||\n\n"

const required = input => {
    if (input) return true
    else return "This field is required"
}

const getGloabNodeModulesPath = () => {
    return String(getenv("PATH"))
        .split(":")
        .filter(path => path.indexOf("node") > -1 && path[0] !== ".")[0]
}

const hasTnpm = async () => {
    const npm = getGloabNodeModulesPath()
    if (npm) {
        return await fs.exists(npm + "/tnpm")
    } else {
        return false
    }
}

const promiseCall = (fn, ...args) => {
    return new Promise((resolve, reject) => {
        fn(...args, (err, res) => {
            if (!err) {
                resolve(res)
            } else {
                reject(err)
            }
        })
    })
}

const transform = async (file_path, fn) => {
    const file = await fs.readFile(file_path, "utf8")
    return await fs.writeFile(file_path, fn(file), "utf8")
}

/**
 *
 * 1. clone文件到当前目录
 * 2.
 *
 */

const getPkgInfo = async cwd => {
    const name = cwd.replace(/.*\/([^\/]+)$/g, "$1")
    return await inquirer.prompt([
        {
            type: "input",
            name: "name",
            validate: required,
            default: name,
            message: "Please input the package name"
        },
        {
            type: "input",
            name: "description",
            message: "Please input the package description"
        },
        {
            type: "input",
            name: "version",
            default: "0.1.0-alpha.0",
            message: "Please input the package version"
        },
        {
            type: "input",
            name: "author",
            default: username({ cwd }),
            message: "Please input the pacakge author"
        },
        {
            type: "input",
            name: "license",
            default: "MIT",
            message: "Please input the pacakge license"
        }
    ])
}

const traverseFile = async (dir, traverse) => {
    const files = await fs.readdir(dir)
    for (let i = 0; i < files.length; i++) {
        try {
            let file_path = path.join(dir, files[i])
            let status = await fs.stat(file_path)
            if (status) {
                if (status.isFile()) {
                    let content = await fs.readFile(file_path, "utf-8")
                    if (typeof traverse === "function") {
                        let newContent = traverse(content)
                        if (newContent !== undefined)
                            await fs.writeFile(file_path, newContent)
                    }
                } else {
                    await traverseFile(file_path, traverse)
                }
            }
        } catch (e) {}
    }
}

const cloneRepo = async (repos, branch, dir = process.cwd()) => {
    const tmp_path = `/tmp/gub_repos/${nanoid()}`
    const cwd = process.cwd()
    const spinner = new Spinner(chalk.yellow("Cloning your repository... %s"))
    spinner.setSpinnerString("|/-\\")
    const pkg = await getPkgInfo(cwd)
    console.log("\n\n")
    spinner.start()
    try {
        if (!fs.existsSync(tmp_path)) {
            await promiseCall(mkdirp, tmp_path)
        }
        await execa.shell(
            `git clone ${branch ? `-b ${branch}` : ""} ${repos} ${tmp_path}`
        )

        await fs.copy(tmp_path, dir, {
            overwrite: true,
            filter: src => {
                if (src.indexOf(".gitignore") > -1) return true
                return src.indexOf(".git") == -1
            }
        })

        let oldPkgName = "",
            newPkgName = pkg.name

        await transform(path.resolve(dir, "./package.json"), function(file) {
            const _pkg = file ? JSON.parse(file) : {}
            oldPkgName = _pkg.name
            if (file) {
                return JSON.stringify(Object.assign(_pkg, pkg), null, 2)
            } else {
                return JSON.stringify(pkg, null, 2)
            }
        })

        await traverseFile(dir, target => {
            return target.replace(
                new RegExp(escapeStringRegexp(oldPkgName), "ig"),
                newPkgName
            )
        })

        const tnpm = await hasTnpm()
        await execa.shell(`${tnpm ? "tnpm" : "npm"} install`, { cwd: dir })
        spinner.stop(true)
        log.success("🎉🎉 Gub init success!")
    } catch (e) {
        spinner.stop(true)
        throw e
    }
}

program
    .command("init [repos] [dir]")
    .option("-b, --branch <branch>", "Init with git branch")
    .action(async (repos, dir, options) => {
        try {
            log.flat(banner)
            let no_repos = !repos
            if (!repos) {
                repos = await selectAliasViewer()
            }
            await cloneRepo(repos, options.branch, dir)
            if (no_repos) {
                await addAlias(repos, parseGithubUrl(repos).name)
            }
        } catch (e) {
            if (e) {
                log.error(e.Error || e.Message || e.message)
            } else {
                log.error("Operation Failed!")
            }
        }
    })

program
    .command("alias")
    .option("-c, --create", "Add an repository alias")
    .option("-u, --update", "Update the repository alias")
    .option("-r, --remove", "Remove the repository alias")
    .option("-d, --delete", "Remove the repository alias")
    .action(async cmd => {
        if (cmd.create) {
            await createAliasViewer()
        } else if (cmd.update) {
            await updateAliasViewer()
        } else if (cmd.remove) {
            await removeAliasViewer()
        } else if (cmd.delete) {
            await createAliasViewer()
        }
    })

program.parse(process.argv)