git add

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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package api 
 
import ( 
    "context" 
    "fmt" 
    "os" 
    "strings" 
    "time" 
 
    "google.golang.org/grpc" 
    "google.golang.org/protobuf/encoding/protojson" 
    "google.golang.org/protobuf/proto" 
 
    core "github.com/v2fly/v2ray-core/v5" 
    "github.com/v2fly/v2ray-core/v5/main/commands/base" 
) 
 
var ( 
    apiServerAddrPtr string 
    apiTimeout       int 
    apiJSON          bool 
    // APIConfigFormat is an internal variable 
    APIConfigFormat string 
    // APIConfigRecursively is an internal variable 
    APIConfigRecursively bool 
) 
 
// SetSharedFlags is an internal API 
func SetSharedFlags(cmd *base.Command) { 
    setSharedFlags(cmd) 
} 
 
func setSharedFlags(cmd *base.Command) { 
    cmd.Flag.StringVar(&apiServerAddrPtr, "s", "127.0.0.1:8080", "") 
    cmd.Flag.StringVar(&apiServerAddrPtr, "server", "127.0.0.1:8080", "") 
    cmd.Flag.IntVar(&apiTimeout, "t", 3, "") 
    cmd.Flag.IntVar(&apiTimeout, "timeout", 3, "") 
    cmd.Flag.BoolVar(&apiJSON, "json", false, "") 
} 
 
// SetSharedConfigFlags is an internal API 
func SetSharedConfigFlags(cmd *base.Command) { 
    setSharedConfigFlags(cmd) 
} 
 
func setSharedConfigFlags(cmd *base.Command) { 
    cmd.Flag.StringVar(&APIConfigFormat, "format", core.FormatAuto, "") 
    cmd.Flag.BoolVar(&APIConfigRecursively, "r", false, "") 
} 
 
// SetSharedFlags is an internal API 
func DialAPIServer() (conn *grpc.ClientConn, ctx context.Context, close func()) { 
    return dialAPIServer() 
} 
 
func dialAPIServer() (conn *grpc.ClientConn, ctx context.Context, close func()) { 
    ctx, cancel := context.WithTimeout(context.Background(), time.Duration(apiTimeout)*time.Second) 
    conn = dialAPIServerWithContext(ctx) 
    close = func() { 
        cancel() 
        conn.Close() 
    } 
    return 
} 
 
func dialAPIServerWithoutTimeout() (conn *grpc.ClientConn, ctx context.Context, close func()) { 
    ctx = context.Background() 
    conn = dialAPIServerWithContext(ctx) 
    close = func() { 
        conn.Close() 
    } 
    return 
} 
 
func dialAPIServerWithContext(ctx context.Context) (conn *grpc.ClientConn) { 
    conn, err := grpc.DialContext(ctx, apiServerAddrPtr, grpc.WithInsecure(), grpc.WithBlock()) 
    if err != nil { 
        base.Fatalf("failed to dial %s", apiServerAddrPtr) 
    } 
    return 
} 
 
func protoToJSONString(m proto.Message, prefix, indent string) (string, error) { // nolint: unparam 
    return strings.TrimSpace(protojson.MarshalOptions{Indent: indent}.Format(m)), nil 
} 
 
func showJSONResponse(m proto.Message) { 
    output, err := protoToJSONString(m, "", "") 
    if err != nil { 
        fmt.Fprintf(os.Stdout, "%v\n", m) 
        base.Fatalf("error encode json: %s", err) 
    } 
    fmt.Println(output) 
}