org.opentripplanner.ext.transmodelapi.support.GqlUtil Maven / Gradle / Ivy

Go to download
package org.opentripplanner.ext.transmodelapi.support;

import graphql.Scalars;
import graphql.introspection.Introspection.DirectiveLocation;
import graphql.schema.DataFetchingEnvironment;
import graphql.schema.GraphQLDirective;
import graphql.schema.GraphQLFieldDefinition;
import graphql.schema.GraphQLInputObjectField;
import graphql.schema.GraphQLList;
import graphql.schema.GraphQLNonNull;
import graphql.schema.GraphQLObjectType;
import graphql.schema.GraphQLScalarType;
import org.opentripplanner.ext.transmodelapi.TransmodelRequestContext;
import org.opentripplanner.ext.transmodelapi.mapping.ServiceDateMapper;
import org.opentripplanner.ext.transmodelapi.mapping.TransitIdMapper;
import org.opentripplanner.ext.transmodelapi.model.scalars.DateScalarFactory;
import org.opentripplanner.ext.transmodelapi.model.scalars.DateTimeScalarFactory;
import org.opentripplanner.ext.transmodelapi.model.scalars.DoubleFunctionScalarFactory;
import org.opentripplanner.ext.transmodelapi.model.scalars.LocalTimeScalarFactory;
import org.opentripplanner.ext.transmodelapi.model.scalars.TimeScalarFactory;
import org.opentripplanner.routing.RoutingService;

import java.util.List;
import java.util.TimeZone;

/**
 * Provide some of the commonly used "chain" of methods. Like all ids should be created
 * the same wayThis
 */
public class GqlUtil {
  public final GraphQLScalarType dateTimeScalar;
  public final GraphQLScalarType dateScalar;
  public final GraphQLScalarType doubleFunctionScalar;
  public final GraphQLScalarType localTimeScalar;
  public final GraphQLObjectType timeScalar;
  public final ServiceDateMapper serviceDateMapper;
  public final GraphQLDirective timingData;

  /** private to prevent util class from instantiation */
  public GqlUtil(TimeZone timeZone) {
    this.dateTimeScalar = DateTimeScalarFactory.createMillisecondsSinceEpochAsDateTimeStringScalar(timeZone);
    this.dateScalar = DateScalarFactory.createSecondsSinceEpochAsDateStringScalar(timeZone);
    this.doubleFunctionScalar = DoubleFunctionScalarFactory.createDoubleFunctionScalar();
    this.localTimeScalar = LocalTimeScalarFactory.createLocalTimeScalar();
    this.timeScalar = TimeScalarFactory.createSecondsSinceMidnightAsTimeObject();
    this.serviceDateMapper =  new ServiceDateMapper(timeZone);
    this.timingData = GraphQLDirective.newDirective()
            .name("timingData")
            .description("Add timing data to prometheus, if Actuator API is enabled")
            .validLocation(DirectiveLocation.FIELD_DEFINITION)
            .build();
  }

  public static RoutingService getRoutingService(DataFetchingEnvironment environment) {
    return ((TransmodelRequestContext) environment.getContext()).getRoutingService();
  }

  public static GraphQLFieldDefinition newTransitIdField() {
    return GraphQLFieldDefinition
        .newFieldDefinition()
        .name("id")
        .type(new GraphQLNonNull(Scalars.GraphQLID))
        .dataFetcher(env -> TransitIdMapper.mapEntityIDToApi(env.getSource()))
        .build();
  }

  public static GraphQLInputObjectField newIdListInputField(String name, String description) {
    return GraphQLInputObjectField.newInputObjectField()
        .name(name)
        .description(description)
        .type(new GraphQLList(Scalars.GraphQLID))
        .defaultValue(List.of())
        .build();
  }

  public static boolean hasArgument(DataFetchingEnvironment environment, String name) {
    return environment.containsArgument(name) && environment.getArgument(name) != null;
  }

  public static  List listOfNullSafe(T element) {
    return element == null ? List.of(): List.of(element);
  }
}