iconfig.go 1.5 KB

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