mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-09 11:51:02 -05:00
109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package mergers
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/v2fly/v2ray-core/v5/common/cmdarg"
|
|
"github.com/v2fly/v2ray-core/v5/common/errors"
|
|
"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 input == nil {
|
|
return nil
|
|
}
|
|
if target == nil {
|
|
return errors.New("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
|
|
}
|