手把手带你开发一个node切换源小工具

前端技术 03/18 JS

node怎么切换源？下面本篇文章带大家手搓一个node切换源小工具，希望对大家有所帮助！

手把手带你开发一个node切换源小工具

嗨嗨嗨，又到了写轮子环节了，为什么要写这个东西呢？

应为npm自带的源下载东西灰常慢

目前已经有一款工具了nrm 也是做切换源的例如tabao源，腾讯源，下载依赖包的时候能加速，那有这么多的源nrm可以帮我们管理起来随时切换。

第一个朋友安装nrm很麻烦还需要用源码的方式安装

第二个朋友公司私服多，自己又懒得手动切换

于是我们想自己写一个小工具实现简易的切换npm源【相关教程推荐：】

思路1，调用命令设置源

npm config set registry  源地址

思路2 使用查看命令获取源地址

npm config get registry

主要就是这两步操作

代码实现

commander

commander是一个nodejs的模块可以解析我们输入的命令，常用于各种脚手架如vue vite等，

例如 xxx -V查看版本 xxx use执行脚本 xxx -h查看帮助等都可以使用 commander实现

inquirer

inquirer也是nodejs的一个模块，常用于命令交互，如vue的cli，vite等，react脚手架等

例如这种选项，还有输入框，多选等

registries.json

这个文件里面放一些初始的源,从nrm的github偷的ping是我自己加的

{
    "npm": {
        "home": "https://www.npmjs.org",
        "registry": "https://registry.npmjs.org/",
        "ping": "https://registry.npmjs.org"
    },
    "yarn": {
        "home": "https://yarnpkg.com",
        "registry": "https://registry.yarnpkg.com/",
        "ping": "https://registry.yarnpkg.com"
    },
    "tencent": {
        "home": "https://mirrors.cloud.tencent.com/npm/",
        "registry": "https://mirrors.cloud.tencent.com/npm/",
        "ping": "https://mirrors.cloud.tencent.com/npm"
    },
    "cnpm": {
        "home": "https://cnpmjs.org",
        "registry": "https://r.cnpmjs.org/",
        "ping": "https://r.cnpmjs.org"
    },
    "taobao": {
        "home": "https://npmmirror.com",
        "registry": "https://registry.npmmirror.com/",
        "ping": "https://registry.npmmirror.com"
    },
    "npmMirror": {
        "home": "https://skimdb.npmjs.com/",
        "registry": "https://skimdb.npmjs.com/registry/",
        "ping": "https://skimdb.npmjs.com/registry"
    }
}

#!/usr/bin/env node

const { program } = require(&#39;commander&#39;)
const PKG = require(&#39;../package.json&#39;) //引入package json
const registries = require(&#39;../registries.json&#39;); //引入初始源
const inquirer = require(&#39;inquirer&#39;);
const { exec, execSync } = require(&#39;child_process&#39;) //子线程用于执行shell命令
const ping = require(&#39;node-http-ping&#39;) //ping网址的一个库
const fs = require(&#39;fs&#39;)
const chalk = require("chalk"); //console 变颜色的一个库
const path = require(&#39;path&#39;)
program.version(PKG.version) //设置版本默认命令 -V --version

//读取源地址方便设置*
const getOrigin = async () => {
    return await execSync(&#39;npm get registry&#39;, { encoding: "utf-8" })
}

//列出所有的源，如果当前有在使用前面加上*
program.command(&#39;ls&#39;).description(&#39;查看镜像&#39;).action(async () => {

    const res = await getOrigin()

    const keys = Object.keys(registries)

    const message = []

    //填充横线算法npm------  yarn------
    const max = Math.max(...keys.map(v => v.length)) + 3
    keys.forEach(k => {

        const newK = registries[k].registry == res.trim() ? (&#39;* &#39; + k) : (&#39;  &#39; + k)
        const Arr = new Array(...newK)
        Arr.length = max;
        const prefix = Array.from(Arr).map(v => v ? v : &#39;-&#39;).join(&#39;&#39;)

        message.push(prefix + &#39;  &#39; + registries[k].registry)
    })
    console.log(message.join(&#39;
&#39;))
})

//切换源
program.command(&#39;use&#39;).description(&#39;请选择镜像&#39;).action(() => {
    inquirer.prompt([
        {
            type: "list",
            name: "sel",
            message: "请选择镜像",
            choices: Object.keys(registries)
        }
    ]).then(result => {

        const reg = registries[result.sel].registry

        exec(`npm config set registry ${reg}`, null, (err, stdout, stderr) => {

            if (err) {
                console.error(&#39;切换错误&#39;, err)
            } else {
                console.log(&#39;切换成功&#39;)
            }
        })
    })
})



//获取当前源
program.command(&#39;current&#39;).description(&#39;查看当前源&#39;).action(async () => {
    const reg = await getOrigin()
    const v = Object.keys(registries).find(k => {
        if (registries[k].registry === reg.trim()) {
            return k;
        }
    })
    console.log(chalk.blue(&#39;当前源:&#39;, v))
})

//ping 源
program.command(&#39;ping&#39;).description(&#39;测试镜像地址速度&#39;).action(() => {
    inquirer.prompt([
        {
            type: "list",
            name: "sel",
            message: "请选择镜像",
            choices: Object.keys(registries)
        }
    ]).then(result => {

        const url = registries[result.sel].ping.trim()

        ping(url).then(time => console.log(chalk.blue(`响应时长: ${time}ms`)))
            .catch(() => console.log(chalk.red(&#39;GG&#39;)))

    })
})

//添加源 读写registries.json 文件实现
program.command(&#39;add&#39;).description(&#39;自定义镜像&#39;).action(() => {
    inquirer.prompt([
        {
            type: "input",
            name: "name",
            message: "请输入镜像名称",
            validate(answer) {
                const keys = Object.keys(registries)
                if (keys.includes(answer)) {
                    return `不能起名${answer}跟保留字冲突`
                }
                if (!answer) {
                    return &#39;名称不能为空&#39;
                }
                return true
            }
        },
        {
            type: "input",
            name: "url",
            message: "请输入镜像地址",
            validate(answer) {
                if (!answer) {
                    return `url不能为空`
                }
                return true
            }       
        }
    ]).then(result => {

        const del = (url) => {
            const arr = url.split(&#39;&#39;)
            //本来想用at 16 以下不支持
            return arr[arr.length - 1] == &#39;/&#39; ? (arr.pop() && arr.join(&#39;&#39;)) : arr.join(&#39;&#39;)
        }

        registries[result.name] = {
            home: result.url.trim(),
            registry: result.url.trim(),
            ping: del(result.url.trim()), //去掉末尾/ 不然无法ping
        }
        fs.writeFileSync(path.join(__dirname, &#39;../registries.json&#39;), JSON.stringify(registries, null, 4))
        console.log(chalk.blue(&#39;添加完成&#39;))
    })
})



program.parse(process.argv)

使用方式