Xray-core/main/toml/toml.go

51 lines
1.3 KiB
Go
Raw Normal View History

2020-12-24 19:11:32 +00:00
package toml
import (
"context"
2020-12-24 19:11:32 +00:00
"io"
"github.com/xtls/xray-core/common"
"github.com/xtls/xray-core/common/cmdarg"
"github.com/xtls/xray-core/common/errors"
2020-12-24 19:11:32 +00:00
"github.com/xtls/xray-core/core"
"github.com/xtls/xray-core/infra/conf"
"github.com/xtls/xray-core/infra/conf/serial"
"github.com/xtls/xray-core/main/confloader"
)
func init() {
common.Must(core.RegisterConfigLoader(&core.ConfigFormat{
Name: "TOML",
Extension: []string{"toml"},
Loader: func(input interface{}) (*core.Config, error) {
switch v := input.(type) {
case cmdarg.Arg:
cf := &conf.Config{}
for i, arg := range v {
errors.LogInfo(context.Background(), "Reading config: ", arg)
2020-12-24 19:11:32 +00:00
r, err := confloader.LoadConfig(arg)
2020-12-25 10:53:17 +00:00
if err != nil {
return nil, errors.New("failed to read config: ", arg).Base(err)
2020-12-25 10:53:17 +00:00
}
2020-12-24 19:11:32 +00:00
c, err := serial.DecodeTOMLConfig(r)
2020-12-25 10:53:17 +00:00
if err != nil {
return nil, errors.New("failed to decode config: ", arg).Base(err)
2020-12-25 10:53:17 +00:00
}
2020-12-24 19:11:32 +00:00
if i == 0 {
// This ensure even if the muti-json parser do not support a setting,
// It is still respected automatically for the first configure file
*cf = *c
continue
}
cf.Override(c, arg)
}
return cf.Build()
case io.Reader:
return serial.LoadTOMLConfig(v)
default:
return nil, errors.New("unknow type")
2020-12-24 19:11:32 +00:00
}
},
}))
}