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 .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{})) }