Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. // Copyright 2022 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package migration
  4. import (
  5. "fmt"
  6. "os"
  7. "strings"
  8. "time"
  9. "code.gitea.io/gitea/modules/json"
  10. "code.gitea.io/gitea/modules/log"
  11. "github.com/santhosh-tekuri/jsonschema/v5"
  12. "gopkg.in/yaml.v3"
  13. )
  14. // Load project data from file, with optional validation
  15. func Load(filename string, data any, validation bool) error {
  16. isJSON := strings.HasSuffix(filename, ".json")
  17. bs, err := os.ReadFile(filename)
  18. if err != nil {
  19. return err
  20. }
  21. if validation {
  22. err := validate(bs, data, isJSON)
  23. if err != nil {
  24. return err
  25. }
  26. }
  27. return unmarshal(bs, data, isJSON)
  28. }
  29. func unmarshal(bs []byte, data any, isJSON bool) error {
  30. if isJSON {
  31. return json.Unmarshal(bs, data)
  32. }
  33. return yaml.Unmarshal(bs, data)
  34. }
  35. func getSchema(filename string) (*jsonschema.Schema, error) {
  36. c := jsonschema.NewCompiler()
  37. c.LoadURL = openSchema
  38. return c.Compile(filename)
  39. }
  40. func validate(bs []byte, datatype any, isJSON bool) error {
  41. var v any
  42. err := unmarshal(bs, &v, isJSON)
  43. if err != nil {
  44. return err
  45. }
  46. if !isJSON {
  47. v, err = toStringKeys(v)
  48. if err != nil {
  49. return err
  50. }
  51. }
  52. var schemaFilename string
  53. switch datatype := datatype.(type) {
  54. case *[]*Issue:
  55. schemaFilename = "issue.json"
  56. case *[]*Milestone:
  57. schemaFilename = "milestone.json"
  58. default:
  59. return fmt.Errorf("file_format:validate: %T has not a validation implemented", datatype)
  60. }
  61. sch, err := getSchema(schemaFilename)
  62. if err != nil {
  63. return err
  64. }
  65. err = sch.Validate(v)
  66. if err != nil {
  67. log.Error("migration validation with %s failed:\n%#v", schemaFilename, err)
  68. }
  69. return err
  70. }
  71. func toStringKeys(val any) (any, error) {
  72. var err error
  73. switch val := val.(type) {
  74. case map[string]any:
  75. m := make(map[string]any)
  76. for k, v := range val {
  77. m[k], err = toStringKeys(v)
  78. if err != nil {
  79. return nil, err
  80. }
  81. }
  82. return m, nil
  83. case []any:
  84. l := make([]any, len(val))
  85. for i, v := range val {
  86. l[i], err = toStringKeys(v)
  87. if err != nil {
  88. return nil, err
  89. }
  90. }
  91. return l, nil
  92. case time.Time:
  93. return val.Format(time.RFC3339), nil
  94. default:
  95. return val, nil
  96. }
  97. }