formats.go 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  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. Loader: makeLoaderFunc(formatName),
  31. }, nil
  32. }
  33. func makeLoaderFunc(formatName string) core.ConfigLoader {
  34. return func(input interface{}) (*core.Config, error) {
  35. m := make(map[string]interface{})
  36. err := mergers.MergeAs(formatName, input, m)
  37. if err != nil {
  38. return nil, err
  39. }
  40. data, err := merge.FromMap(m)
  41. if err != nil {
  42. return nil, err
  43. }
  44. r := bytes.NewReader(data)
  45. cf, err := serial.DecodeJSONConfig(r)
  46. if err != nil {
  47. return nil, err
  48. }
  49. return cf.Build()
  50. }
  51. }