#!/usr/bin/env node

const { program } = require('commander')
const PKG = require('../package.json') //引入package json
const registries = require('../registries.json'); //引入初始源
const inquirer = require('inquirer');
const { exec, execSync } = require('child_process') //子線程用于執行shell命令
const ping = require('node-http-ping') //ping網址的一個庫
const fs = require('fs')
const chalk = require("chalk"); //console 變顏色的一個庫
const path = require('path')
program.version(PKG.version) //設置版本默認命令 -V --version

//讀取源地址方便設置*
const getOrigin = async () => {
    return await execSync('npm get registry', { encoding: "utf-8" })
}

//列出所有的源，如果當前有在使用前面加上*
program.command('ls').description('查看鏡像').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() ? ('* ' + k) : ('  ' + k)
        const Arr = new Array(...newK)
        Arr.length = max;
        const prefix = Array.from(Arr).map(v => v ? v : '-').join('')

        message.push(prefix + '  ' + registries[k].registry)
    })
    console.log(message.join('\n'))
})

//切換源
program.command('use').description('請選擇鏡像').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('切換錯誤', err)
            } else {
                console.log('切換成功')
            }
        })
    })
})



//獲取當前源
program.command('current').description('查看當前源').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('當前源:', v))
})

//ping 源
program.command('ping').description('測試鏡像地址速度').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('GG')))

    })
})

//添加源 讀寫registries.json 文件實現
program.command('add').description('自定義鏡像').action(() => {
    inquirer.prompt([
        {
            type: "input",
            name: "name",
            message: "請輸入鏡像名稱",
            validate(answer) {
                const keys = Object.keys(registries)
                if (keys.includes(answer)) {
                    return `不能起名${answer}跟保留字沖突`
                }
                if (!answer) {
                    return '名稱不能為空'
                }
                return true
            }
        },
        {
            type: "input",
            name: "url",
            message: "請輸入鏡像地址",
            validate(answer) {
                if (!answer) {
                    return `url不能為空`
                }
                return true
            }       
        }
    ]).then(result => {

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

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



program.parse(process.argv)