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
package helpers
 
import (
    "bytes"
    "os"
 
    "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"
    v4 "github.com/v2fly/v2ray-core/v5/infra/conf/v4"
)
 
// LoadConfig load config files to *conf.Config, it will:
// - resolve folder to files
// - try to read stdin if no file specified
func LoadConfig(files []string, format string, recursively bool) (*v4.Config, error) {
    m, err := LoadConfigToMap(files, format, recursively)
    if err != nil {
        return nil, err
    }
    bs, err := merge.FromMap(m)
    if err != nil {
        return nil, err
    }
    r := bytes.NewReader(bs)
    return serial.DecodeJSONConfig(r)
}
 
// LoadConfigToMap load config files to map, it will:
// - resolve folder to files
// - try to read stdin if no file specified
func LoadConfigToMap(files []string, format string, recursively bool) (map[string]interface{}, error) {
    var err error
    if len(files) > 0 {
        var extensions []string
        extensions, err := mergers.GetExtensions(format)
        if err != nil {
            return nil, err
        }
        files, err = ResolveFolderToFiles(files, extensions, recursively)
        if err != nil {
            return nil, err
        }
    }
    m := make(map[string]interface{})
    if len(files) == 0 {
        err = mergers.MergeAs(format, os.Stdin, m)
    } else {
        err = mergers.MergeAs(format, files, m)
    }
    if err != nil {
        return nil, err
    }
    return m, nil
}