forked from v2fly/v2ray-core
-
Notifications
You must be signed in to change notification settings - Fork 0
/
reversepb.go
45 lines (41 loc) · 1.08 KB
/
reversepb.go
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))
}
}
},
}