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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package mergers
 
import (
    "fmt"
    "io"
 
    "github.com/v2fly/v2ray-core/v5/common/cmdarg"
    "github.com/v2fly/v2ray-core/v5/infra/conf/merge"
)
 
type jsonConverter func(v []byte) ([]byte, error)
 
// makeMerger makes a merger who merge the format by converting it to JSON
func makeMerger(name string, extensions []string, converter jsonConverter) *Merger {
    return &Merger{
        Name:       name,
        Extensions: extensions,
        Merge:      makeToJSONMergeFunc(converter),
    }
}
 
// makeToJSONMergeFunc makes a merge func who merge the format by converting it to JSON
func makeToJSONMergeFunc(converter func(v []byte) ([]byte, error)) MergeFunc {
    return func(input interface{}, target map[string]interface{}) error {
        if target == nil {
            panic("merge target is nil")
        }
        switch v := input.(type) {
        case string:
            err := loadFile(v, target, converter)
            if err != nil {
                return err
            }
        case []string:
            err := loadFiles(v, target, converter)
            if err != nil {
                return err
            }
        case cmdarg.Arg:
            err := loadFiles(v, target, converter)
            if err != nil {
                return err
            }
        case []byte:
            err := loadBytes(v, target, converter)
            if err != nil {
                return err
            }
        case io.Reader:
            err := loadReader(v, target, converter)
            if err != nil {
                return err
            }
        default:
            return newError("unknown merge input type")
        }
        return nil
    }
}
 
func loadFiles(files []string, target map[string]interface{}, converter func(v []byte) ([]byte, error)) error {
    for _, file := range files {
        err := loadFile(file, target, converter)
        if err != nil {
            return err
        }
    }
    return nil
}
 
func loadFile(file string, target map[string]interface{}, converter func(v []byte) ([]byte, error)) error {
    bs, err := cmdarg.LoadArgToBytes(file)
    if err != nil {
        return fmt.Errorf("fail to load %s: %s", file, err)
    }
    if converter != nil {
        bs, err = converter(bs)
        if err != nil {
            return fmt.Errorf("error convert to json '%s': %s", file, err)
        }
    }
    _, err = merge.ToMap(bs, target)
    return err
}
 
func loadReader(reader io.Reader, target map[string]interface{}, converter func(v []byte) ([]byte, error)) error {
    bs, err := io.ReadAll(reader)
    if err != nil {
        return err
    }
    return loadBytes(bs, target, converter)
}
 
func loadBytes(bs []byte, target map[string]interface{}, converter func(v []byte) ([]byte, error)) error {
    var err error
    if converter != nil {
        bs, err = converter(bs)
        if err != nil {
            return fmt.Errorf("fail to convert to json: %s", err)
        }
    }
    _, err = merge.ToMap(bs, target)
    return err
}