Hunter0x7c7
2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
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))
            }
        }
    },
}