fix #12

openmove · Feb 23, 2023 · b51b013 · b51b013
1 parent 05b1882
commit b51b013
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/src/main/java/org/opentripplanner/api/parameter/QualifiedModeSet.java b/src/main/java/org/opentripplanner/api/parameter/QualifiedModeSet.java
@@ -6,6 +6,7 @@
 import org.opentripplanner.routing.core.TraverseModeSet;
 
 import java.io.Serializable;
+import java.util.List;
 import java.util.Set;
 
 /**
@@ -28,6 +29,12 @@ public QualifiedModeSet(String s) {
             qModes.add(new QualifiedMode(qMode));
         }
     }
+
+    public QualifiedModeSet(List<String> modeList) {
+        for (String qMode : modeList) {
+            qModes.add(new QualifiedMode(qMode));
+        }
+    }
 
     /**
      * Modify an existing routing request, setting fields to reflect these qualified modes.

diff --git a/src/main/java/org/opentripplanner/index/GraphQlPlanner.java b/src/main/java/org/opentripplanner/index/GraphQlPlanner.java
@@ -233,6 +233,12 @@ private RoutingRequest createRequest(DataFetchingEnvironment environment) {
             new QualifiedModeSet((String)environment.getArgument("modes")).applyToRoutingRequest(request);
             request.setModes(request.modes);
         }
+
+        if(hasArgument(environment, "transportModes")) {
+        	List<String> transportModes = environment.getArgument("transportModes");
+        	new QualifiedModeSet(transportModes).applyToRoutingRequest(request);
+            request.setModes(request.modes);
+        }
 
         if (request.allowBikeRental && !hasArgument(environment, "bikeSpeed")) {
             //slower bike speed for bike sharing, based on empirical evidence from DC.

diff --git a/src/main/java/org/opentripplanner/index/IndexGraphQLSchema.java b/src/main/java/org/opentripplanner/index/IndexGraphQLSchema.java
@@ -705,7 +705,7 @@ public IndexGraphQLSchema(GraphIndex index) {
                 .argument(GraphQLArgument.newArgument()
                         .name("transportModes")
                         .description("List of transportation modes that the user is willing to use. Default: `[\"WALK\",\"TRANSIT\"]`")
-                        .type(new GraphQLList(transportModeInputType))
+                        .type(new GraphQLList(Scalars.GraphQLString))
                         .build())
                 .argument(GraphQLArgument.newArgument()
                         .name("modeWeight")