v2ray-core/main/confloader/external/external.go

49 lines
1.0 KiB
Go
Raw Normal View History

2019-12-16 01:33:41 +00:00
package external
import (
"io"
"os"
"strings"
"v2ray.com/core/common/buf"
"v2ray.com/core/common/platform/ctlcmd"
"v2ray.com/core/main/confloader"
)
2020-01-01 17:23:56 +00:00
//go:generate errorgen
2019-12-16 01:33:41 +00:00
2020-01-01 17:23:56 +00:00
func loadConfigFile(configFile string) (io.ReadCloser, error) {
if configFile == "stdin:" {
return os.Stdin, nil
2019-12-16 01:33:41 +00:00
}
2020-01-01 17:23:56 +00:00
if strings.HasPrefix(configFile, "http://") || strings.HasPrefix(configFile, "https://") {
content, err := ctlcmd.Run([]string{"fetch", configFile}, nil)
if err != nil {
return nil, err
}
return &buf.MultiBufferContainer{
MultiBuffer: content,
}, nil
2019-12-16 01:33:41 +00:00
}
2020-01-01 17:23:56 +00:00
fixedFile := os.ExpandEnv(configFile)
file, err := os.Open(fixedFile)
2019-12-16 01:33:41 +00:00
if err != nil {
2020-01-01 17:23:56 +00:00
return nil, newError("config file not readable").Base(err)
2019-12-16 01:33:41 +00:00
}
2020-01-01 17:23:56 +00:00
defer file.Close()
2019-12-16 01:33:41 +00:00
2020-01-01 17:23:56 +00:00
content, err := buf.ReadFrom(file)
2019-12-16 01:33:41 +00:00
if err != nil {
2020-01-01 17:23:56 +00:00
return nil, newError("failed to load config file: ", fixedFile).Base(err).AtWarning()
2019-12-16 01:33:41 +00:00
}
2020-01-01 17:23:56 +00:00
return &buf.MultiBufferContainer{
MultiBuffer: content,
}, nil
2019-12-16 01:33:41 +00:00
}
func init() {
2020-01-01 17:23:56 +00:00
confloader.EffectiveConfigFileLoader = loadConfigFile
2019-12-16 01:33:41 +00:00
}