walker/schemacontext: Introduce nesting level

hashicorp · Nov 6, 2023 · 0d22b9d · 0d22b9d
1 parent 8715001
commit 0d22b9d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/decoder/internal/walker/walker.go b/decoder/internal/walker/walker.go
@@ -24,6 +24,12 @@ type Walker interface {
 func Walk(ctx context.Context, node hclsyntax.Node, nodeSchema schema.Schema, w Walker) hcl.Diagnostics {
 	var diags hcl.Diagnostics
 
+	blkNestingLvl, ok := schemacontext.BlockNestingLevel(ctx)
+	if !ok {
+		ctx = schemacontext.WithBlockNestingLevel(ctx, 0)
+		blkNestingLvl = 0
+	}
+
 	switch nodeType := node.(type) {
 	case *hclsyntax.Body:
 		bodyCtx := ctx
@@ -105,6 +111,7 @@ func Walk(ctx context.Context, node hclsyntax.Node, nodeSchema schema.Schema, w
 			blockBodySchema = mergedSchema
 		}
 
+		blockCtx = schemacontext.WithBlockNestingLevel(blockCtx, blkNestingLvl+1)
 		diags = diags.Extend(Walk(blockCtx, nodeType.Body, blockBodySchema, w))
 
 		// TODO: case hclsyntax.Expression

diff --git a/schemacontext/context.go b/schemacontext/context.go
@@ -8,6 +8,7 @@ import "context"
 type unknownSchemaCtxKey struct{}
 type foundBlocksCtxKey struct{}
 type dynamicBlocksCtxKey struct{}
+type blockNestingLevelCtxKey struct{}
 
 // WithUnknownSchema attaches a flag indicating that the schema being passed
 // is not wholly known.
@@ -42,3 +43,12 @@ func WithDynamicBlocks(ctx context.Context, dynamicBlocks map[string]uint64) con
 func DynamicBlocks(ctx context.Context) map[string]uint64 {
 	return ctx.Value(dynamicBlocksCtxKey{}).(map[string]uint64)
 }
+
+func WithBlockNestingLevel(ctx context.Context, nestingLvl uint64) context.Context {
+	return context.WithValue(ctx, blockNestingLevelCtxKey{}, nestingLvl)
+}
+
+func BlockNestingLevel(ctx context.Context) (uint64, bool) {
+	lvl, ok := ctx.Value(blockNestingLevelCtxKey{}).(uint64)
+	return lvl, ok
+}