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
package formats
 
import (
    "bytes"
 
    core "github.com/v2fly/v2ray-core/v5"
    "github.com/v2fly/v2ray-core/v5/common"
    "github.com/v2fly/v2ray-core/v5/infra/conf/merge"
    "github.com/v2fly/v2ray-core/v5/infra/conf/mergers"
    "github.com/v2fly/v2ray-core/v5/infra/conf/serial"
)
 
func init() {
    for _, formatName := range mergers.GetAllNames() {
        loader, err := makeMergeLoader(formatName)
        if err != nil {
            panic(err)
        }
        if formatName == core.FormatAuto {
            loader.Extension = nil
        }
        common.Must(core.RegisterConfigLoader(loader))
    }
}
 
func makeMergeLoader(formatName string) (*core.ConfigFormat, error) {
    extensions, err := mergers.GetExtensions(formatName)
    if err != nil {
        return nil, err
    }
    return &core.ConfigFormat{
        Name:      []string{formatName},
        Extension: extensions,
        Loader:    makeLoaderFunc(formatName),
    }, nil
}
 
func makeLoaderFunc(formatName string) core.ConfigLoader {
    return func(input interface{}) (*core.Config, error) {
        m := make(map[string]interface{})
        err := mergers.MergeAs(formatName, input, m)
        if err != nil {
            return nil, err
        }
        data, err := merge.FromMap(m)
        if err != nil {
            return nil, err
        }
        r := bytes.NewReader(data)
        cf, err := serial.DecodeJSONConfig(r)
        if err != nil {
            return nil, err
        }
        return cf.Build()
    }
}