feat: introduce flag noAdditionalProperties

This will default `additionalProperties` to `false` for all objects in the schema if set to `true`. The `additionalProperties` annotation, `schemaRoot.additionalProperties` and additionalProperties set via other ways (nested in `itemProperties`, `patternProperties` or similar) will take precedence over this setting, so it's fully backwards compatible. Closes #95.
losisin · Oct 24, 2024 · 114bd03 · 114bd03
1 parent 6d1e5b0
commit 114bd03
Show file tree

Hide file tree

Showing 9 changed files with 144 additions and 42 deletions.
diff --git a/README.md b/README.md
@@ -116,6 +116,8 @@ Usage: helm schema [options...] <arguments>
     	Multiple yaml files as inputs (comma-separated)
   -output string
     	Output file path (default "values.schema.json")
+  -noAdditionalProperties value
+         Default additionalProperties to false for all objects in the schema (true/false)
   -schemaRoot.additionalProperties value
     	JSON schema additional properties (true/false)
   -schemaRoot.description string

diff --git a/pkg/cmd.go b/pkg/cmd.go
@@ -20,6 +20,7 @@ func ParseFlags(progname string, args []string) (*Config, string, error) {
 	flags.StringVar(&conf.OutputPath, "output", "values.schema.json", "Output file path")
 	flags.IntVar(&conf.Draft, "draft", 2020, "Draft version (4, 6, 7, 2019, or 2020)")
 	flags.IntVar(&conf.Indent, "indent", 4, "Indentation spaces (even number)")
+	flags.Var(&conf.NoAdditionalProperties, "noAdditionalProperties", "Default additionalProperties to false for all objects in the schema")
 
 	// Nested SchemaRoot flags
 	flags.StringVar(&conf.SchemaRoot.ID, "schemaRoot.id", "", "JSON schema ID")
@@ -88,6 +89,10 @@ func MergeConfig(fileConfig, flagConfig *Config) *Config {
 	if flagConfig.IndentSet || mergedConfig.Indent == 0 {
 		mergedConfig.Indent = flagConfig.Indent
 	}
+
+	if flagConfig.NoAdditionalProperties.IsSet() {
+		mergedConfig.NoAdditionalProperties = flagConfig.NoAdditionalProperties
+	}
 	if flagConfig.SchemaRoot.ID != "" {
 		mergedConfig.SchemaRoot.ID = flagConfig.SchemaRoot.ID
 	}

diff --git a/pkg/generator.go b/pkg/generator.go
@@ -89,14 +89,16 @@ func GenerateJsonSchema(config *Config) error {
 	}
 
 	// Convert merged Schema into a JSON Schema compliant map
-	jsonSchemaMap, err := convertSchemaToMap(mergedSchema)
+	jsonSchemaMap, err := convertSchemaToMap(mergedSchema, config.NoAdditionalProperties.value)
 	if err != nil {
 		return err
 	}
 	jsonSchemaMap["$schema"] = schemaURL // Include the schema draft version
 
 	if config.SchemaRoot.AdditionalProperties.IsSet() {
 		jsonSchemaMap["additionalProperties"] = config.SchemaRoot.AdditionalProperties.Value()
+	} else if config.NoAdditionalProperties.value {
+		jsonSchemaMap["additionalProperties"] = false
 	}
 
 	// If validation is successful, marshal the schema and save to the file

diff --git a/pkg/generator_test.go b/pkg/generator_test.go
@@ -11,41 +11,68 @@ import (
 )
 
 func TestGenerateJsonSchema(t *testing.T) {
-	config := &Config{
-		Input: []string{
-			"../testdata/full.yaml",
-			"../testdata/empty.yaml",
+	tests := []struct {
+		name               string
+		config             *Config
+		templateSchemaFile string
+	}{
+		{
+			name: "full json schema",
+			config: &Config{
+				Input: []string{
+					"../testdata/full.yaml",
+					"../testdata/empty.yaml",
+				},
+				OutputPath: "../testdata/output.json",
+				Draft:      2020,
+				Indent:     4,
+				SchemaRoot: SchemaRoot{
+					ID:                   "https://example.com/schema",
+					Ref:                  "schema/product.json",
+					Title:                "Helm Values Schema",
+					Description:          "Schema for Helm values",
+					AdditionalProperties: BoolFlag{set: true, value: true},
+				},
+			},
+			templateSchemaFile: "../testdata/full.schema.json",
 		},
-		OutputPath: "../testdata/output.json",
-		Draft:      2020,
-		Indent:     4,
-		SchemaRoot: SchemaRoot{
-			ID:                   "https://example.com/schema",
-			Ref:                  "schema/product.json",
-			Title:                "Helm Values Schema",
-			Description:          "Schema for Helm values",
-			AdditionalProperties: BoolFlag{set: true, value: true},
+		{
+			name: "full json schema",
+			config: &Config{
+				Draft:                  2020,
+				Indent:                 4,
+				NoAdditionalProperties: BoolFlag{set: true, value: true},
+				Input: []string{
+					"../testdata/noAdditionalProperties.yaml",
+				},
+				OutputPath: "../testdata/output1.json",
+			},
+			templateSchemaFile: "../testdata/noAdditionalProperties.schema.json",
 		},
 	}
 
-	err := GenerateJsonSchema(config)
-	assert.NoError(t, err)
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			err := GenerateJsonSchema(tt.config)
+			assert.NoError(t, err)
 
-	generatedBytes, err := os.ReadFile(config.OutputPath)
-	assert.NoError(t, err)
+			generatedBytes, err := os.ReadFile(tt.config.OutputPath)
+			assert.NoError(t, err)
 
-	templateBytes, err := os.ReadFile("../testdata/full.schema.json")
-	assert.NoError(t, err)
+			templateBytes, err := os.ReadFile(tt.templateSchemaFile)
+			assert.NoError(t, err)
 
-	var generatedSchema, templateSchema map[string]interface{}
-	err = json.Unmarshal(generatedBytes, &generatedSchema)
-	assert.NoError(t, err)
-	err = json.Unmarshal(templateBytes, &templateSchema)
-	assert.NoError(t, err)
+			var generatedSchema, templateSchema map[string]interface{}
+			err = json.Unmarshal(generatedBytes, &generatedSchema)
+			assert.NoError(t, err)
+			err = json.Unmarshal(templateBytes, &templateSchema)
+			assert.NoError(t, err)
 
-	assert.Equal(t, templateSchema, generatedSchema, "Generated JSON schema does not match the template")
+			assert.Equal(t, templateSchema, generatedSchema, "Generated JSON schema does not match the template")
 
-	os.Remove(config.OutputPath)
+			os.Remove(tt.config.OutputPath)
+		})
+	}
 }
 
 func TestGenerateJsonSchema_Errors(t *testing.T) {
@@ -149,6 +176,7 @@ func TestGenerateJsonSchema_AdditionalProperties(t *testing.T) {
 		name                    string
 		additionalPropertiesSet bool
 		additionalProperties    bool
+		noAdditionalProperties  bool
 		expected                interface{}
 	}{
 		{
@@ -168,22 +196,42 @@ func TestGenerateJsonSchema_AdditionalProperties(t *testing.T) {
 			additionalPropertiesSet: false,
 			expected:                nil,
 		},
+		{
+			name:                    "AdditionalProperties not set, but NoAdditionalProperties set",
+			additionalPropertiesSet: false,
+			noAdditionalProperties:  true,
+			expected:                false,
+		},
+		{
+			name:                    "NoAdditionalProperties set, but AdditionalProperties set to true",
+			additionalPropertiesSet: true,
+			additionalProperties:    true,
+			noAdditionalProperties:  true,
+			expected:                true,
+		},
 	}
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			additionalPropertiesFlag := &BoolFlag{}
+			noAdditionalPropertiesFlag := &BoolFlag{}
 			if tt.additionalPropertiesSet {
 				if err := additionalPropertiesFlag.Set(fmt.Sprintf("%t", tt.additionalProperties)); err != nil {
 					t.Fatalf("Failed to set additionalPropertiesFlag: %v", err)
 				}
 			}
+			if tt.noAdditionalProperties {
+				if err := noAdditionalPropertiesFlag.Set(fmt.Sprintf("%t", tt.noAdditionalProperties)); err != nil {
+					t.Fatalf("Failed to set noAdditionalPropertiesFlag: %v", err)
+				}
+			}
 
 			config := &Config{
-				Input:      []string{"../testdata/empty.yaml"},
-				OutputPath: "../testdata/empty.schema.json",
-				Draft:      2020,
-				Indent:     4,
+				Input:                  []string{"../testdata/empty.yaml"},
+				OutputPath:             "../testdata/empty.schema.json",
+				Draft:                  2020,
+				Indent:                 4,
+				NoAdditionalProperties: *noAdditionalPropertiesFlag,
 				SchemaRoot: SchemaRoot{
 					ID:                   "",
 					Title:                "",

diff --git a/pkg/parser.go b/pkg/parser.go
@@ -103,11 +103,11 @@ func mergeSchemas(dest, src *Schema) *Schema {
 	return dest
 }
 
-func convertSchemaToMap(schema *Schema) (map[string]interface{}, error) {
-	return convertSchemaToMapRec(schema, make(map[uintptr]bool))
+func convertSchemaToMap(schema *Schema, noAdditionalProperties bool) (map[string]interface{}, error) {
+	return convertSchemaToMapRec(schema, make(map[uintptr]bool), noAdditionalProperties)
 }
 
-func convertSchemaToMapRec(schema *Schema, visited map[uintptr]bool) (map[string]interface{}, error) {
+func convertSchemaToMapRec(schema *Schema, visited map[uintptr]bool, noAdditionalProperties bool) (map[string]interface{}, error) {
 	if schema == nil {
 		return nil, nil
 	}
@@ -178,6 +178,8 @@ func convertSchemaToMapRec(schema *Schema, visited map[uintptr]bool) (map[string
 	}
 	if schema.AdditionalProperties != nil {
 		schemaMap["additionalProperties"] = *schema.AdditionalProperties
+	} else if noAdditionalProperties && schema.Type == "object" {
+		schemaMap["additionalProperties"] = false
 	}
 	if schema.ID != "" {
 		schemaMap["$id"] = schema.ID
@@ -196,7 +198,7 @@ func convertSchemaToMapRec(schema *Schema, visited map[uintptr]bool) (map[string
 
 	// Nested Schemas
 	if schema.Items != nil {
-		itemsMap, err := convertSchemaToMapRec(schema.Items, visited)
+		itemsMap, err := convertSchemaToMapRec(schema.Items, visited, noAdditionalProperties)
 		if err != nil {
 			return nil, err
 		}
@@ -205,7 +207,7 @@ func convertSchemaToMapRec(schema *Schema, visited map[uintptr]bool) (map[string
 	if schema.Properties != nil {
 		propertiesMap := make(map[string]interface{})
 		for propName, propSchema := range schema.Properties {
-			propMap, err := convertSchemaToMapRec(propSchema, visited)
+			propMap, err := convertSchemaToMapRec(propSchema, visited, noAdditionalProperties)
 			if err != nil {
 				return nil, err
 			}

diff --git a/pkg/parser_test.go b/pkg/parser_test.go
@@ -271,7 +271,7 @@ func TestConvertSchemaToMap(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			got, err := convertSchemaToMap(tt.schema)
+			got, err := convertSchemaToMap(tt.schema, false)
 			if (err != nil) != tt.wantErr {
 				t.Errorf("convertSchemaToMap() error = %v, wantErr %v", err, tt.wantErr)
 				return
@@ -318,7 +318,7 @@ func TestConvertSchemaToMapFail(t *testing.T) {
 
 	for _, tc := range tests {
 		t.Run(tc.name, func(t *testing.T) {
-			_, err := convertSchemaToMap(tc.schema)
+			_, err := convertSchemaToMap(tc.schema, false)
 			tc.expectedErr(t, err)
 		})
 	}

diff --git a/pkg/utils.go b/pkg/utils.go
@@ -17,10 +17,11 @@ type SchemaRoot struct {
 
 // Save values of parsed flags in Config
 type Config struct {
-	Input      multiStringFlag `yaml:"input"`
-	OutputPath string          `yaml:"output"`
-	Draft      int             `yaml:"draft"`
-	Indent     int             `yaml:"indent"`
+	Input                  multiStringFlag `yaml:"input"`
+	OutputPath             string          `yaml:"output"`
+	Draft                  int             `yaml:"draft"`
+	Indent                 int             `yaml:"indent"`
+	NoAdditionalProperties BoolFlag        `yaml:"noAdditionalProperties"`
 
 	SchemaRoot SchemaRoot `yaml:"schemaRoot"`
 

diff --git a/testdata/noAdditionalProperties.schema.json b/testdata/noAdditionalProperties.schema.json
@@ -0,0 +1,38 @@
+{
+    "$schema": "https://json-schema.org/draft/2020-12/schema",
+    "additionalProperties": false,
+    "properties": {
+        "object": {
+            "additionalProperties": false,
+            "properties": {},
+            "type": "object"
+        },
+        "objectOfObjects": {
+            "additionalProperties": false,
+            "patternProperties": {
+                "^.*$": {
+                    "type": "object"
+                }
+            },
+            "properties": {},
+            "type": "object"
+        },
+        "objectOfObjectsWithInnerAdditionalPropertiesAllowed": {
+            "additionalProperties": false,
+            "patternProperties": {
+                "^.*$": {
+                    "additionalProperties": true,
+                    "type": "object"
+                }
+            },
+            "properties": {},
+            "type": "object"
+        },
+        "objectWithAdditionalPropertiesAllowed": {
+            "additionalProperties": true,
+            "properties": {},
+            "type": "object"
+        }
+    },
+    "type": "object"
+}
diff --git a/testdata/noAdditionalProperties.yaml b/testdata/noAdditionalProperties.yaml
@@ -0,0 +1,4 @@
+object: {}
+objectWithAdditionalPropertiesAllowed: {} # @schema additionalProperties: true
+objectOfObjects: {} # @schema patternProperties: { "^.*$": {"type": "object" } }
+objectOfObjectsWithInnerAdditionalPropertiesAllowed: {} # @schema patternProperties: { "^.*$": {"type": "object", "additionalProperties": true }}