Add extensions data to ErrorContext class

(cherry picked from commit e15cc8b)

Author: Tomasz Fijalkowski

graphql-kickstart-spring-support/src/main/java/graphql/kickstart/spring/error/ErrorContext.java

@@ -0,0 +1,40 @@
+package graphql.kickstart.spring.error;
+
+import graphql.ErrorClassification;
+import graphql.language.SourceLocation;
+
+import java.util.List;
+import java.util.Map;
+
+public class ErrorContext {
+    private final List<SourceLocation> locations;
+    private final List<Object> path;
+    private final Map<String, Object> extensions;
+    private final ErrorClassification errorType;
+
+    public ErrorContext(List<SourceLocation> locations,
+                        List<Object> path,
+                        Map<String, Object> extensions,
+                        ErrorClassification errorType) {
+        this.locations = locations;
+        this.path = path;
+        this.extensions = extensions;
+        this.errorType = errorType;
+    }
+
+    public List<SourceLocation> getLocations() {
+        return locations;
+    }
+
+    public List<Object> getPath() {
+        return path;
+    }
+
+    public Map<String, Object> getExtensions() {
+        return extensions;
+    }
+
+    public ErrorClassification getErrorType() {
+        return errorType;
+    }
+}

graphql-kickstart-spring-support/src/main/java/graphql/kickstart/spring/error/GraphQLErrorFromExceptionHandler.java

@@ -32,7 +32,13 @@ protected List<GraphQLError> filterGraphQLErrors(List<GraphQLError> errors) {
   }
 
   private Collection<GraphQLError> transform(GraphQLError error) {
-    return extractException(error).map(this::transform)
+    ErrorContext errorContext = new ErrorContext(
+            error.getLocations(),
+            error.getPath(),
+            error.getExtensions(),
+            error.getErrorType()
+    );
+    return extractException(error).map(throwable -> transform(throwable, errorContext))
         .orElse(singletonList(new GenericGraphQLError(error.getMessage())));
   }
 
@@ -47,13 +53,13 @@ private Optional<Throwable> extractException(GraphQLError error) {
     return Optional.empty();
   }
 
-  private Collection<GraphQLError> transform(Throwable throwable) {
+  private Collection<GraphQLError> transform(Throwable throwable, ErrorContext errorContext) {
     Map<Class<? extends Throwable>, GraphQLErrorFactory> applicables = new HashMap<>();
     factories.forEach(factory -> factory.mostConcrete(throwable).ifPresent(t -> applicables.put(t, factory)));
     return applicables.keySet().stream()
         .min(new ThrowableComparator())
         .map(applicables::get)
-        .map(factory -> factory.create(throwable))
+        .map(factory -> factory.create(throwable, errorContext))
         .orElse(singletonList(new ThrowableGraphQLError(throwable)));
   }