1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 18:17:52 -05:00
v2fly/app/instman/command/command.go
2021-10-28 18:34:19 +08:00

65 lines
1.6 KiB
Go

package command
import (
"context"
"encoding/base64"
"google.golang.org/grpc"
core "github.com/v2fly/v2ray-core/v4"
"github.com/v2fly/v2ray-core/v4/common"
"github.com/v2fly/v2ray-core/v4/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
}))
}