fix: skip invalid locations

v2/pkg/engine/resolve/const.go

@@ -2,6 +2,10 @@ package resolve
 
 import "errors"
 
+const (
+	locationsField = "locations"
+)
+
 var (
 	lBrace                    = []byte("{")
 	rBrace                    = []byte("}")
@@ -16,7 +20,7 @@ var (
 	literalFalse              = []byte("false")
 	literalErrors             = []byte("errors")
 	literalMessage            = []byte("message")
-	literalLocations          = []byte("locations")
+	literalLocations          = []byte(locationsField)
 	literalPath               = []byte("path")
 	literalUnderscoreEntities = []byte("_entities")
 	literalExtensions         = []byte("extensions")
@@ -35,19 +39,3 @@ var (
 	errHeaderPathInvalid           = errors.New("invalid header path: header variables must be of this format: .request.header.{{ key }} ")
 	ErrUnableToResolve             = errors.New("unable to resolve operation")
 )
-
-var (
-	errorPaths = [][]string{
-		{"message"},
-		{"locations"},
-		{"path"},
-		{"extensions"},
-	}
-)
-
-const (
-	errorsMessagePathIndex    = 0
-	errorsLocationsPathIndex  = 1
-	errorsPathPathIndex       = 2
-	errorsExtensionsPathIndex = 3
-)

v2/pkg/engine/resolve/loader.go

@@ -936,12 +936,48 @@ func (l *Loader) optionallyOmitErrorFields(values []*astjson.Value) {
 
 // optionallyOmitErrorLocations removes the "locations" object from all values.
 func (l *Loader) optionallyOmitErrorLocations(values []*astjson.Value) {
-	if !l.omitSubgraphErrorLocations {
-		return
-	}
 	for _, value := range values {
-		if value.Exists("locations") {
-			value.Del("locations")
+		// If the flag is set, delete all locations
+		if !value.Exists(locationsField) || l.omitSubgraphErrorLocations {
+			value.Del(locationsField)
+			continue
+		}
+
+		locations := value.Get(locationsField)
+		if locations.Type() != astjson.TypeArray {
+			continue
+		}
+
+		locationsArray := locations.GetArray()
+		locationsClone := slices.Clone(locationsArray)
+
+		deletedEntries := 0
+		locationsArrayLength := len(locationsArray)
+
+		// We loop on a clone since we delete elements inline
+		for i, loc := range locationsClone {
+			line := loc.Get("line")
+			column := loc.Get("column")
+
+			// Skip invalid locations: nil values or values <= 0
+			if line == nil || column == nil {
+				locations.Del(strconv.Itoa(i - deletedEntries))
+				deletedEntries++
+				continue
+			}
+
+			// Keep location only if both line and column are > 0
+			// In case it is not an int, 0 will be returned which is invalid anyway
+			isValid := line.GetInt() > 0 && column.GetInt() > 0
+			if !isValid {
+				locations.Del(strconv.Itoa(i - deletedEntries))
+				deletedEntries++
+			}
+		}
+
+		// If all locations were invalid, delete the locations field
+		if locationsArrayLength == deletedEntries {
+			value.Del(locationsField)
 		}
 	}
 }