formats.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. package formats
  2. import (
  3. "bytes"
  4. core "github.com/v2fly/v2ray-core/v5"
  5. "github.com/v2fly/v2ray-core/v5/common"
  6. "github.com/v2fly/v2ray-core/v5/infra/conf/merge"
  7. "github.com/v2fly/v2ray-core/v5/infra/conf/mergers"
  8. "github.com/v2fly/v2ray-core/v5/infra/conf/serial"
  9. )
  10. func init() {
  11. for _, formatName := range mergers.GetAllNames() {
  12. loader, err := makeMergeLoader(formatName)
  13. if err != nil {
  14. panic(err)
  15. }
  16. if formatName == core.FormatAuto {
  17. loader.Extension = nil
  18. }
  19. common.Must(core.RegisterConfigLoader(loader))
  20. }
  21. }
  22. func makeMergeLoader(formatName string) (*core.ConfigFormat, error) {
  23. extensions, err := mergers.GetExtensions(formatName)
  24. if err != nil {
  25. return nil, err
  26. }
  27. return &core.ConfigFormat{
  28. Name: []string{formatName},
  29. Extension: extensions,
  30. Priority: -2,
  31. Loader: makeLoaderFunc(formatName),
  32. }, nil
  33. }
  34. func makeLoaderFunc(formatName string) core.ConfigLoader {
  35. return func(input interface{}) (*core.Config, error) {
  36. m := make(map[string]interface{})
  37. err := mergers.MergeAs(formatName, input, m)
  38. if err != nil {
  39. return nil, err
  40. }
  41. data, err := merge.FromMap(m)
  42. if err != nil {
  43. return nil, err
  44. }
  45. r := bytes.NewReader(data)
  46. cf, err := serial.DecodeJSONConfig(r)
  47. if err != nil {
  48. return nil, err
  49. }
  50. return cf.Build()
  51. }
  52. }