feat: add composite type support for field resolvers

v2/pkg/ast/ast_selection.go

@@ -212,6 +212,18 @@ func (d *Document) SelectionSetHasFieldSelectionWithExactName(set int, name []by
 	return false, InvalidRef
 }
 
+// SelectionSetFieldRefs returns a list of field refs in the selection set.
+// It traverses through the field selections of the selection set and returns the actual field refs.
+func (d *Document) SelectionSetFieldRefs(set int) (refs []int) {
+	for _, selectionRef := range d.SelectionSets[set].SelectionRefs {
+		if d.Selections[selectionRef].Kind == SelectionKindField {
+			refs = append(refs, d.Selections[selectionRef].Ref)
+		}
+	}
+	return
+}
+
+// SelectionSetFieldSelections returns a list of field selection refs in the selection set.
 func (d *Document) SelectionSetFieldSelections(set int) (refs []int) {
 	for _, selectionRef := range d.SelectionSets[set].SelectionRefs {
 		if d.Selections[selectionRef].Kind == SelectionKindField {

v2/pkg/astvisitor/visitor.go

@@ -4021,6 +4021,7 @@ func (w *Walker) FieldDefinitionDirectiveArgumentValueByName(field int, directiv
 }
 
 // InRootField returns true if the current field is a root field.
+// This helper function can be used in EnterField and LeaveField.
 func (w *Walker) InRootField() bool {
 	return w.CurrentKind == ast.NodeKindField &&
 		len(w.Ancestors) == 2 &&