fields.go

     1  // Copyright 2024 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package astutil
     6  
     7  import (
     8  	"go/ast"
     9  	"iter"
    10  )
    11  
    12  // FlatFields 'flattens' an ast.FieldList, returning an iterator over each
    13  // (name, field) combination in the list. For unnamed fields, the identifier is
    14  // nil.
    15  func FlatFields(list *ast.FieldList) iter.Seq2[*ast.Ident, *ast.Field] {
    16  	return func(yield func(*ast.Ident, *ast.Field) bool) {
    17  		if list == nil {
    18  			return
    19  		}
    20  
    21  		for _, field := range list.List {
    22  			if len(field.Names) == 0 {
    23  				if !yield(nil, field) {
    24  					return
    25  				}
    26  			} else {
    27  				for _, name := range field.Names {
    28  					if !yield(name, field) {
    29  						return
    30  					}
    31  				}
    32  			}
    33  		}
    34  	}
    35  }
    36
View as plain text