Merge pull request #28 from billyshambrook/fix/any-type-const

Support multiple types for const
dadav · May 29, 2024 · bdf7656 · bdf7656
2 parents 2cafc75 + 3390b8b
commit bdf7656
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/pkg/schema/schema.go b/pkg/schema/schema.go
@@ -121,7 +121,7 @@ type Schema struct {
 	Maximum              *int                  `yaml:"maximum,omitempty"              json:"maximum,omitempty"`
 	Else                 *Schema               `yaml:"else,omitempty"                 json:"else,omitempty"`
 	Pattern              string                `yaml:"pattern,omitempty"              json:"pattern,omitempty"`
-	Const                string                `yaml:"const,omitempty"                json:"const,omitempty"`
+	Const                interface{}           `yaml:"const,omitempty"                json:"const,omitempty"`
 	Ref                  string                `yaml:"$ref,omitempty"                 json:"$ref,omitempty"`
 	Schema               string                `yaml:"$schema,omitempty"              json:"$schema,omitempty"`
 	Id                   string                `yaml:"$id,omitempty"                  json:"$id,omitempty"`
@@ -233,7 +233,7 @@ func (s Schema) Validate() error {
 		return fmt.Errorf("cant use items if type is %s. Use type=array", s.Type)
 	}
 
-	if s.Const != "" && !s.Type.IsEmpty() {
+	if s.Const != nil && !s.Type.IsEmpty() {
 		return errors.New("if your are using const, you can't use type")
 	}
 

diff --git a/pkg/schema/schema_test.go b/pkg/schema/schema_test.go
@@ -45,6 +45,20 @@ func TestValidate(t *testing.T) {
 		{
 			comment: `
 # @schema
+# const: "hello"
+# @schema`,
+			expectedValid: true,
+		},
+		{
+			comment: `
+# @schema
+# const: true
+# @schema`,
+			expectedValid: true,
+		},
+		{
+			comment: `
+# @schema
 # format: ipv4
 # @schema`,
 			expectedValid: true,