mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-10-31 16:27:41 -04:00
62 lines
1.9 KiB
Go
62 lines
1.9 KiB
Go
package v4
|
|
|
|
import (
|
|
"google.golang.org/protobuf/types/known/anypb"
|
|
"strings"
|
|
|
|
"github.com/jhump/protoreflect/desc"
|
|
"github.com/jhump/protoreflect/dynamic"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/app/commander"
|
|
loggerservice "github.com/v2fly/v2ray-core/v4/app/log/command"
|
|
observatoryservice "github.com/v2fly/v2ray-core/v4/app/observatory/command"
|
|
handlerservice "github.com/v2fly/v2ray-core/v4/app/proxyman/command"
|
|
routerservice "github.com/v2fly/v2ray-core/v4/app/router/command"
|
|
statsservice "github.com/v2fly/v2ray-core/v4/app/stats/command"
|
|
"github.com/v2fly/v2ray-core/v4/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
|
|
}
|