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
package command 
 
import ( 
    "context" 
    "encoding/base64" 
 
    "google.golang.org/grpc" 
 
    core "github.com/v2fly/v2ray-core/v5" 
    "github.com/v2fly/v2ray-core/v5/common" 
    "github.com/v2fly/v2ray-core/v5/features/extension" 
) 
 
type service struct { 
    UnimplementedInstanceManagementServiceServer 
 
    instman extension.InstanceManagement 
} 
 
func (s service) ListInstance(ctx context.Context, req *ListInstanceReq) (*ListInstanceResp, error) { 
    instanceNames, err := s.instman.ListInstance(ctx) 
    if err != nil { 
        return nil, err 
    } 
    return &ListInstanceResp{Name: instanceNames}, nil 
} 
 
func (s service) AddInstance(ctx context.Context, req *AddInstanceReq) (*AddInstanceResp, error) { 
    configContent, err := base64.StdEncoding.DecodeString(req.ConfigContentB64) 
    if err != nil { 
        return nil, err 
    } 
    err = s.instman.AddInstance(ctx, req.Name, configContent, req.ConfigType) 
    if err != nil { 
        return nil, err 
    } 
    return &AddInstanceResp{}, nil 
} 
 
func (s service) StartInstance(ctx context.Context, req *StartInstanceReq) (*StartInstanceResp, error) { 
    err := s.instman.StartInstance(ctx, req.Name) 
    if err != nil { 
        return nil, err 
    } 
    return &StartInstanceResp{}, nil 
} 
 
func (s service) Register(server *grpc.Server) { 
    RegisterInstanceManagementServiceServer(server, s) 
} 
 
func init() { 
    common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, cfg interface{}) (interface{}, error) { 
        s := core.MustFromContext(ctx) 
        sv := &service{} 
        err := s.RequireFeatures(func(instman extension.InstanceManagement) { 
            sv.instman = instman 
        }) 
        if err != nil { 
            return nil, err 
        } 
        return sv, nil 
    })) 
}