v2fly/main/commands/all/engineering/reversepb.go

package engineering

import (
	"bytes"
	"flag"
	"io"
	"os"

	"github.com/golang/protobuf/proto"

	core "github.com/v2fly/v2ray-core/v5"
	"github.com/v2fly/v2ray-core/v5/infra/conf/jsonpb"
	"github.com/v2fly/v2ray-core/v5/infra/conf/v2jsonpb"
	"github.com/v2fly/v2ray-core/v5/main/commands/base"
)

var cmdReversePb = &base.Command{
	UsageLine: "{{.Exec}} engineering reversepb [-f format]",
	Flag: func() flag.FlagSet {
		fs := flag.NewFlagSet("", flag.ExitOnError)
		configFormat = fs.String("f", "v2jsonpb", "")
		return *fs
	}(),
	Run: func(cmd *base.Command, args []string) {
		cmd.Flag.Parse(args)
		configIn := bytes.NewBuffer(nil)
		io.Copy(configIn, os.Stdin)
		var conf core.Config
		if err := proto.Unmarshal(configIn.Bytes(), &conf); err != nil {
			base.Fatalf("%s", err)
		}
		switch *configFormat {
		case "jsonpb":
			if err := jsonpb.DumpJSONPb(&conf, os.Stdout); err != nil {
				base.Fatalf("%s", err)
			}
		case "v2jsonpb":
			if value, err := v2jsonpb.DumpV2JsonPb(&conf); err != nil {
				base.Fatalf("%s", err)
			} else {
				io.Copy(os.Stdout, bytes.NewReader(value))
			}
		}
	},
}