external.go 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. package external
  2. import (
  3. "io"
  4. "os"
  5. "strings"
  6. "v2ray.com/core/common/buf"
  7. "v2ray.com/core/common/platform/ctlcmd"
  8. "v2ray.com/core/main/confloader"
  9. )
  10. //go:generate errorgen
  11. func loadConfigFile(configFile string) (io.ReadCloser, error) {
  12. if configFile == "stdin:" {
  13. return os.Stdin, nil
  14. }
  15. if strings.HasPrefix(configFile, "http://") || strings.HasPrefix(configFile, "https://") {
  16. content, err := ctlcmd.Run([]string{"fetch", configFile}, nil)
  17. if err != nil {
  18. return nil, err
  19. }
  20. return &buf.MultiBufferContainer{
  21. MultiBuffer: content,
  22. }, nil
  23. }
  24. fixedFile := os.ExpandEnv(configFile)
  25. file, err := os.Open(fixedFile)
  26. if err != nil {
  27. return nil, newError("config file not readable").Base(err)
  28. }
  29. defer file.Close()
  30. content, err := buf.ReadFrom(file)
  31. if err != nil {
  32. return nil, newError("failed to load config file: ", fixedFile).Base(err).AtWarning()
  33. }
  34. return &buf.MultiBufferContainer{
  35. MultiBuffer: content,
  36. }, nil
  37. }
  38. func init() {
  39. confloader.EffectiveConfigFileLoader = loadConfigFile
  40. }