update

Hunter0x7c7

2022-08-11 b8230139fb40edea387617b6accd8371e37eda58

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
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 
}