iconfig.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. package config
  2. import (
  3. "errors"
  4. "fmt"
  5. "github.com/urfave/cli/v2"
  6. "gopkg.in/yaml.v3"
  7. "os"
  8. "reflect"
  9. )
  10. const cliMetaKey = "cli.config"
  11. type IConfig interface {
  12. Invalidate() error
  13. }
  14. func Invalidate(config IConfig) error {
  15. if config == nil {
  16. return errors.New("config must not be a nil")
  17. }
  18. val := reflect.ValueOf(config)
  19. if val.Kind() != reflect.Struct {
  20. if val.IsZero() {
  21. return errors.New("config is empty")
  22. }
  23. val = reflect.ValueOf(config).Elem()
  24. }
  25. for i := 0; i < val.NumField(); i++ {
  26. if !val.Field(i).CanInterface() || val.Field(i).IsZero() {
  27. continue
  28. }
  29. if elm, ok := val.Field(i).Interface().(IConfig); ok {
  30. if er := elm.Invalidate(); er != nil {
  31. return er
  32. }
  33. }
  34. }
  35. return nil
  36. }
  37. func Read[T IConfig](path string) (*T, error) {
  38. var (
  39. cfg = new(T)
  40. fd []byte
  41. err error
  42. )
  43. if fd, err = os.ReadFile(path); err != nil {
  44. return nil, err
  45. }
  46. if err = yaml.Unmarshal(fd, cfg); err != nil {
  47. return nil, err
  48. }
  49. if err = (*cfg).Invalidate(); err != nil {
  50. return nil, err
  51. }
  52. if err = Invalidate(*cfg); err != nil {
  53. return nil, err
  54. }
  55. return cfg, nil
  56. }
  57. func Load[T IConfig](ctx *cli.Context, path cli.Path) error {
  58. cfg, err := Read[T](path)
  59. if err != nil {
  60. return fmt.Errorf("could not load config from: %s; %v", path, err)
  61. }
  62. if cfg == nil {
  63. return fmt.Errorf("could not load config from: %s", path)
  64. }
  65. ctx.App.Metadata[cliMetaKey] = cfg
  66. return nil
  67. }
  68. func Get[T IConfig](ctx *cli.Context) *T {
  69. cfg, ok := ctx.App.Metadata[cliMetaKey].(*T)
  70. if !ok {
  71. return nil
  72. }
  73. return cfg
  74. }