1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 18:17:52 -05:00
v2fly/infra/conf/v4/api.go
2022-01-02 15:16:23 +00:00

62 lines
1.9 KiB
Go

package v4
import (
"strings"
"github.com/jhump/protoreflect/desc"
"github.com/jhump/protoreflect/dynamic"
"google.golang.org/protobuf/types/known/anypb"
"github.com/v2fly/v2ray-core/v5/app/commander"
loggerservice "github.com/v2fly/v2ray-core/v5/app/log/command"
observatoryservice "github.com/v2fly/v2ray-core/v5/app/observatory/command"
handlerservice "github.com/v2fly/v2ray-core/v5/app/proxyman/command"
routerservice "github.com/v2fly/v2ray-core/v5/app/router/command"
statsservice "github.com/v2fly/v2ray-core/v5/app/stats/command"
"github.com/v2fly/v2ray-core/v5/common/serial"
)
type APIConfig struct {
Tag string `json:"tag"`
Services []string `json:"services"`
}
func (c *APIConfig) Build() (*commander.Config, error) {
if c.Tag == "" {
return nil, newError("API tag can't be empty.")
}
services := make([]*anypb.Any, 0, 16)
for _, s := range c.Services {
switch strings.ToLower(s) {
case "reflectionservice":
services = append(services, serial.ToTypedMessage(&commander.ReflectionConfig{}))
case "handlerservice":
services = append(services, serial.ToTypedMessage(&handlerservice.Config{}))
case "loggerservice":
services = append(services, serial.ToTypedMessage(&loggerservice.Config{}))
case "statsservice":
services = append(services, serial.ToTypedMessage(&statsservice.Config{}))
case "observatoryservice":
services = append(services, serial.ToTypedMessage(&observatoryservice.Config{}))
case "routingservice":
services = append(services, serial.ToTypedMessage(&routerservice.Config{}))
default:
if !strings.HasPrefix(s, "#") {
continue
}
message, err := desc.LoadMessageDescriptor(s[1:])
if err != nil || message == nil {
return nil, newError("Cannot find API", s, "").Base(err)
}
serviceConfig := dynamic.NewMessage(message)
services = append(services, serial.ToTypedMessage(serviceConfig))
}
}
return &commander.Config{
Tag: c.Tag,
Service: services,
}, nil
}