mirror of https://github.com/v2ray/v2ray-core
89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
package control
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
"v2ray.com/core/common"
|
|
"v2ray.com/core/infra/conf"
|
|
"v2ray.com/core/infra/conf/serial"
|
|
)
|
|
|
|
// ConfigCommand is the json to pb convert struct
|
|
type ConfigCommand struct{}
|
|
|
|
// Name for cmd usage
|
|
func (c *ConfigCommand) Name() string {
|
|
return "config"
|
|
}
|
|
|
|
// Description for help usage
|
|
func (c *ConfigCommand) Description() Description {
|
|
return Description{
|
|
Short: "merge multiple json config",
|
|
Usage: []string{"v2ctl config config.json c1.json c2.json <url>.json"},
|
|
}
|
|
}
|
|
|
|
// Execute real work here.
|
|
func (c *ConfigCommand) Execute(args []string) error {
|
|
if len(args) < 1 {
|
|
return newError("empty config list")
|
|
}
|
|
|
|
conf := &conf.Config{}
|
|
for _, arg := range args {
|
|
ctllog.Println("Read config: ", arg)
|
|
r, err := c.LoadArg(arg)
|
|
common.Must(err)
|
|
c, err := serial.DecodeJSONConfig(r)
|
|
if err != nil {
|
|
ctllog.Fatalln(err)
|
|
}
|
|
conf.Override(c, arg)
|
|
}
|
|
|
|
pbConfig, err := conf.Build()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
bytesConfig, err := proto.Marshal(pbConfig)
|
|
if err != nil {
|
|
return newError("failed to marshal proto config").Base(err)
|
|
}
|
|
|
|
if _, err := os.Stdout.Write(bytesConfig); err != nil {
|
|
return newError("failed to write proto config").Base(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// LoadArg loads one arg, maybe an remote url, or local file path
|
|
func (c *ConfigCommand) LoadArg(arg string) (out io.Reader, err error) {
|
|
|
|
var data []byte
|
|
if strings.HasPrefix(arg, "http://") || strings.HasPrefix(arg, "https://") {
|
|
data, err = FetchHTTPContent(arg)
|
|
} else if arg == "stdin:" {
|
|
data, err = ioutil.ReadAll(os.Stdin)
|
|
} else {
|
|
data, err = ioutil.ReadFile(arg)
|
|
}
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
out = bytes.NewBuffer(data)
|
|
return
|
|
}
|
|
|
|
func init() {
|
|
common.Must(RegisterCommand(&ConfigCommand{}))
|
|
}
|