mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2026-04-23 06:49:37 +08:00
e1732a364fed
541 lines
12 KiB
Go
541 lines
12 KiB
Go
//go:build !nocli
|
|
|
|
package main
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/e1732a364fed/v2ray_simple/machine"
|
|
"github.com/e1732a364fed/v2ray_simple/netLayer"
|
|
"github.com/e1732a364fed/v2ray_simple/proxy"
|
|
"github.com/e1732a364fed/v2ray_simple/utils"
|
|
"github.com/manifoldco/promptui"
|
|
)
|
|
|
|
//cli = command line interface
|
|
|
|
type CliCmd struct {
|
|
Name string
|
|
f func()
|
|
}
|
|
|
|
func (cc CliCmd) String() string {
|
|
return cc.Name
|
|
}
|
|
|
|
func flist(list []*CliCmd) (result []func()) {
|
|
for _, v := range list {
|
|
result = append(result, v.f)
|
|
}
|
|
return
|
|
}
|
|
|
|
// cliCmdList 包含所有交互模式中可执行的命令；
|
|
// 本文件 中添加的 CliCmd都是直接返回运行结果的、无需进一步交互的命令
|
|
var cliCmdList = []*CliCmd{
|
|
{
|
|
"查询当前状态", func() {
|
|
mainM.PrintAllState(os.Stdout, false)
|
|
},
|
|
}, {
|
|
"打印当前版本所支持的所有协议", printSupportedProtocols,
|
|
}, {
|
|
"生成随机ssl证书", generateRandomSSlCert,
|
|
}, {
|
|
"生成一个随机的uuid供你参考", generateAndPrintUUID,
|
|
}, {
|
|
"下载geosite文件夹", tryDownloadGeositeSource,
|
|
}, {
|
|
"下载geoip文件(GeoLite2-Country.mmdb)", tryDownloadMMDB,
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
flag.BoolVar(&interactive_mode, "i", false, "enable interactive commandline mode")
|
|
|
|
//cli.go 中添加的 CliCmd都是需进一步交互的命令
|
|
|
|
cliCmdList = append(cliCmdList, &CliCmd{
|
|
"【生成分享链接】<-当前的配置", func() {
|
|
sc := mainM.DumpStandardConf()
|
|
interactively_generate_share(&sc)
|
|
},
|
|
}, &CliCmd{
|
|
"【导出标准配置文件】<-当前全部配置", interactively_exportVsConf,
|
|
}, &CliCmd{
|
|
"【交互生成配置】，超级强大", func() { generateConfigFileInteractively(mainM) },
|
|
}, &CliCmd{
|
|
"热删除配置", func() { interactively_hotRemoveServerOrClient(mainM) },
|
|
}, &CliCmd{
|
|
"【热加载】新配置文件", func() { interactively_hotLoadConfigFile(mainM) },
|
|
}, &CliCmd{
|
|
"【热加载】新配置url", func() { interactively_hotLoadUrlConfig(mainM) },
|
|
}, &CliCmd{
|
|
"调节日志等级", interactively_adjust_loglevel,
|
|
})
|
|
|
|
runCli = runCli_func
|
|
}
|
|
|
|
// 交互式命令行用户界面
|
|
//
|
|
// 阻塞，可按ctrl+C退出或回退到上一级
|
|
func runCli_func() {
|
|
defer func() {
|
|
utils.PrintStr("Interactive Mode exited. \n")
|
|
if ce := utils.CanLogInfo("Interactive Mode exited"); ce != nil {
|
|
ce.Write()
|
|
}
|
|
|
|
savePerferences()
|
|
}()
|
|
|
|
loadPreferences()
|
|
|
|
searcher := func(input string, index int) bool {
|
|
pepper := cliCmdList[index]
|
|
name := strings.Replace(strings.ToLower(pepper.Name), " ", "", -1)
|
|
input = strings.Replace(strings.ToLower(input), " ", "", -1)
|
|
|
|
return strings.Contains(name, input)
|
|
}
|
|
|
|
for {
|
|
Select := promptui.Select{
|
|
Label: "请选择想执行的功能",
|
|
Items: cliCmdList,
|
|
Searcher: searcher,
|
|
}
|
|
|
|
i, result, err := Select.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("你选择了 %s\n", result)
|
|
|
|
if f := cliCmdList[i].f; f != nil {
|
|
f()
|
|
}
|
|
|
|
if cp := currentUserPreference.Cli; cp != nil {
|
|
if cp.AutoArrange {
|
|
updateMostRecentCli(i)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func generateConfigFileInteractively(m *machine.M) {
|
|
|
|
rootLevelList := []string{
|
|
"【打印】当前缓存的配置",
|
|
"【开始交互生成】配置",
|
|
"【清除】此次缓存的配置",
|
|
"【写到文件】<-将该缓存的配置 (client.toml和 server.toml)",
|
|
"【生成分享】链接url <-该缓存的配置",
|
|
"【投入运行】（热加载) <-将此次生成的配置",
|
|
}
|
|
|
|
confClient := proxy.StandardConf{}
|
|
confServer := proxy.StandardConf{}
|
|
|
|
var clientStr, serverStr string
|
|
|
|
for {
|
|
Select := promptui.Select{
|
|
Label: "请选择想为你的配置文件做的事情",
|
|
Items: rootLevelList,
|
|
}
|
|
|
|
i, result, err := Select.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("你选择了 %s\n", result)
|
|
|
|
generateConfStr := func() {
|
|
|
|
confClient.Route = []*netLayer.RuleConf{{
|
|
DialTag: "direct",
|
|
Domains: []string{"geosite:cn"},
|
|
}}
|
|
|
|
var vsConfClient machine.VSConf = machine.VSConf{
|
|
AppConf: &machine.AppConf{MyCountryISO_3166: "CN"},
|
|
StandardConf: confClient,
|
|
}
|
|
|
|
clientStr, err = utils.GetPurgedTomlStr(&vsConfClient)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
serverStr, err = utils.GetPurgedTomlStr(confServer)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
switch i {
|
|
case 0: //print
|
|
|
|
generateConfStr()
|
|
|
|
buf := utils.GetBuf()
|
|
|
|
buf.WriteString("#客户端配置\n")
|
|
buf.WriteString(clientStr)
|
|
buf.WriteString("\n")
|
|
|
|
buf.WriteString("#服务端配置\n")
|
|
buf.WriteString(serverStr)
|
|
buf.WriteString("\n")
|
|
utils.PrintStr(buf.String())
|
|
|
|
utils.PutBuf(buf)
|
|
|
|
case 1:
|
|
interactively_generateConf(&confClient, &confServer)
|
|
|
|
case 2: //clear
|
|
confClient = proxy.StandardConf{}
|
|
confServer = proxy.StandardConf{}
|
|
clientStr = ""
|
|
serverStr = ""
|
|
case 3: //output
|
|
|
|
generateConfStr()
|
|
|
|
var clientFile *os.File
|
|
clientFile, err = os.OpenFile("client.toml", os.O_WRONLY|os.O_CREATE, 0666)
|
|
if err != nil {
|
|
fmt.Println("Can't create client.toml", err)
|
|
return
|
|
}
|
|
clientFile.WriteString(clientStr)
|
|
clientFile.Close()
|
|
|
|
var serverFile *os.File
|
|
serverFile, err = os.OpenFile("server.toml", os.O_WRONLY|os.O_CREATE, 0666)
|
|
if err != nil {
|
|
fmt.Println("Can't create server.toml", err)
|
|
return
|
|
}
|
|
serverFile.WriteString(serverStr)
|
|
serverFile.Close()
|
|
|
|
utils.PrintStr("生成成功！请查看文件\n")
|
|
case 4: //share url
|
|
if len(confClient.Dial) > 0 {
|
|
|
|
utils.PrintStr("生成的分享链接如下：\n")
|
|
|
|
interactively_generate_share(&confClient)
|
|
|
|
} else {
|
|
utils.PrintStr("请先进行配置\n")
|
|
|
|
}
|
|
case 5: //hot load
|
|
utils.PrintStr("因为本次同时生成了服务端和客户端配置, 请选择要热加载的是哪一个\n")
|
|
selectHot := promptui.Select{
|
|
Label: "加载客户端配置还是服务端配置？",
|
|
Items: []string{
|
|
"服务端",
|
|
"客户端",
|
|
},
|
|
}
|
|
ihot, result, err := selectHot.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("你选择了 %s\n", result)
|
|
|
|
switch ihot {
|
|
case 0:
|
|
|
|
m.LoadDialConf(confServer.Dial)
|
|
m.LoadListenConf(confServer.Listen, true)
|
|
|
|
case 1:
|
|
m.LoadDialConf(confClient.Dial)
|
|
m.LoadListenConf(confClient.Listen, true)
|
|
}
|
|
|
|
utils.PrintStr("加载成功！你可以回退(ctrl+c)到上级来使用 【查询当前状态】来查询新增的配置\n")
|
|
|
|
} // switch i case 1
|
|
} //for
|
|
}
|
|
|
|
// 热删除配置
|
|
func interactively_hotRemoveServerOrClient(m *machine.M) {
|
|
utils.PrintStr("即将开始热删除配置步骤, 删除正在运行的配置可能有未知风险，谨慎操作\n")
|
|
utils.PrintStr("【当前所有配置】为：\n")
|
|
utils.PrintStr(delimiter)
|
|
m.PrintAllState(os.Stdout, false)
|
|
|
|
var items []string
|
|
if m.ServerCount() > 0 {
|
|
items = append(items, "listen")
|
|
}
|
|
if m.ClientCount() > 0 {
|
|
items = append(items, "dial")
|
|
}
|
|
if len(items) == 0 {
|
|
utils.PrintStr("没listen也没dial! 一个都没有，没法删!\n")
|
|
return
|
|
}
|
|
|
|
Select := promptui.Select{
|
|
Label: "请选择你想删除的是dial还是listen",
|
|
Items: items,
|
|
}
|
|
|
|
i, result, err := Select.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
var will_delete_listen, will_delete_dial bool
|
|
|
|
var will_delete_index int
|
|
|
|
fmt.Printf("你选择了 %s\n", result)
|
|
switch i {
|
|
case 0:
|
|
if m.ServerCount() > 0 {
|
|
will_delete_listen = true
|
|
|
|
} else if m.ClientCount() > 0 {
|
|
will_delete_dial = true
|
|
}
|
|
|
|
case 1:
|
|
if m.ServerCount() > 0 {
|
|
will_delete_dial = true
|
|
|
|
}
|
|
}
|
|
|
|
var theInt int64
|
|
|
|
if (will_delete_dial && m.ClientCount() > 1) || (will_delete_listen && m.ServerCount() > 1) {
|
|
|
|
validateFunc := func(input string) error {
|
|
theInt, err = strconv.ParseInt(input, 10, 64)
|
|
if err != nil || theInt < 0 {
|
|
return utils.ErrInvalidNumber
|
|
}
|
|
|
|
if will_delete_dial && int(theInt) >= m.ClientCount() {
|
|
return errors.New("must with in len of dial array")
|
|
}
|
|
|
|
if will_delete_listen && int(theInt) >= m.ServerCount() {
|
|
return errors.New("must with in len of listen array")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
utils.PrintStr("请输入你想删除的序号\n")
|
|
|
|
promptIdx := promptui.Prompt{
|
|
Label: "序号",
|
|
Validate: validateFunc,
|
|
}
|
|
|
|
_, err = promptIdx.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("你输入了 %d\n", theInt)
|
|
|
|
}
|
|
|
|
will_delete_index = int(theInt)
|
|
|
|
if will_delete_dial {
|
|
m.HotDeleteClient(will_delete_index)
|
|
}
|
|
if will_delete_listen {
|
|
m.HotDeleteServer(will_delete_index)
|
|
|
|
}
|
|
|
|
utils.PrintStr("删除成功！当前状态：\n")
|
|
utils.PrintStr(delimiter)
|
|
m.PrintAllState(os.Stdout, false)
|
|
}
|
|
|
|
func interactively_hotLoadUrlConfig(m *machine.M) {
|
|
utils.PrintStr("即将开始热添加url配置\n")
|
|
Select := promptui.Select{
|
|
Label: "请选择你的url的格式类型",
|
|
Items: []string{
|
|
"vs标准url格式",
|
|
"协议官方url格式(视代理协议不同而不同)",
|
|
},
|
|
}
|
|
i, result, err := Select.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("你选择了 %s\n", result)
|
|
|
|
switch i {
|
|
case 1:
|
|
fmt.Printf("目前暂不支持")
|
|
return
|
|
|
|
case 0:
|
|
Select := promptui.Select{
|
|
Label: "请选择该url是用于dial还是listen",
|
|
Items: []string{
|
|
"dial",
|
|
"listen",
|
|
},
|
|
}
|
|
i, result, err := Select.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
fmt.Printf("你选择了 %s\n", result)
|
|
|
|
fmt.Printf("请输入你的配置url\n")
|
|
|
|
var theUrlStr string
|
|
|
|
fmt.Scanln(&theUrlStr)
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
if i == 0 {
|
|
m.HotLoadDialUrl(theUrlStr, proxy.UrlFormat)
|
|
|
|
} else {
|
|
m.HotLoadListenUrl(theUrlStr, proxy.UrlFormat)
|
|
|
|
}
|
|
return
|
|
|
|
}
|
|
}
|
|
|
|
// 热添加配置文件
|
|
func interactively_hotLoadConfigFile(m *machine.M) {
|
|
utils.PrintStr("即将开始热添加配置文件\n")
|
|
utils.PrintStr("【注意】我们交互模式只支持热添加listen和dial, 对于dns/route/fallback的热增删, 请期待api server未来的实现.\n")
|
|
utils.PrintStr("【当前所有配置】为：\n")
|
|
utils.PrintStr(delimiter)
|
|
m.PrintAllState(os.Stdout, false)
|
|
|
|
utils.PrintStr("请输入你想添加的文件名称\n")
|
|
|
|
promptFile := promptui.Prompt{
|
|
Label: "配置文件",
|
|
Validate: func(s string) error {
|
|
|
|
if err := utils.IsFilePath(s); err != nil {
|
|
return err
|
|
}
|
|
if !utils.FileExist(utils.GetFilePath(s)) {
|
|
return os.ErrNotExist
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
|
|
fpath, err := promptFile.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("你输入了 %s\n", fpath)
|
|
|
|
_, err = mainM.LoadConfig(fpath, "", "")
|
|
if err != nil {
|
|
|
|
log.Printf("can not load standard config file: %s\n", err)
|
|
return
|
|
}
|
|
|
|
//listen, dial, dns, route, fallbacks 这几项都可以选择性加载
|
|
|
|
//但是route和fallback的话，动态增删很麻烦，因为route/fallback可能配置相当多条;
|
|
|
|
//而dns的话,没法简单增删, 而是会覆盖。
|
|
|
|
//因此我们交互模式暂且只支持 listen和dial的热加载。 dns/route/fallback的热增删可以用apiServer实现.
|
|
|
|
//也就是说，理论上要写一个比较好的前端，才能妥善解决 复杂条目的热增删问题。
|
|
|
|
m.LoadStandardConf()
|
|
|
|
utils.PrintStr("添加成功！当前状态：\n")
|
|
utils.PrintStr(delimiter)
|
|
m.PrintAllState(os.Stdout, false)
|
|
}
|
|
|
|
func interactively_adjust_loglevel() {
|
|
fmt.Println("当前日志等级为：", utils.LogLevelStr(utils.LogLevel))
|
|
|
|
list := utils.LogLevelStrList()
|
|
Select := promptui.Select{
|
|
Label: "请选择你调节为点loglevel",
|
|
Items: list,
|
|
}
|
|
|
|
i, result, err := Select.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("你选择了 %s\n", result)
|
|
|
|
if i < len(list) && i >= 0 {
|
|
utils.LogLevel = i
|
|
utils.InitLog("")
|
|
|
|
utils.PrintStr("调节 日志等级完毕. 现在等级为\n")
|
|
utils.PrintStr(list[i])
|
|
utils.PrintStr("\n")
|
|
|
|
}
|
|
}
|