org.opentripplanner.ext.restapi.mapping.FareMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of application Show documentation
The newest version!
package org.opentripplanner.ext.restapi.mapping;

import com.google.common.collect.Multimap;
import java.util.Collection;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Optional;
import javax.annotation.Nullable;
import org.opentripplanner.ext.restapi.model.ApiCurrency;
import org.opentripplanner.ext.restapi.model.ApiFareProduct;
import org.opentripplanner.ext.restapi.model.ApiFareQualifier;
import org.opentripplanner.ext.restapi.model.ApiItineraryFares;
import org.opentripplanner.ext.restapi.model.ApiLegProducts;
import org.opentripplanner.ext.restapi.model.ApiMoney;
import org.opentripplanner.model.fare.FareMedium;
import org.opentripplanner.model.fare.FareProduct;
import org.opentripplanner.model.fare.FareProductUse;
import org.opentripplanner.model.fare.RiderCategory;
import org.opentripplanner.model.plan.Itinerary;
import org.opentripplanner.model.plan.Leg;
import org.opentripplanner.transit.model.basic.Money;

public class FareMapper {

  private final Locale locale;

  public FareMapper(Locale locale) {
    this.locale = locale;
  }

  public ApiItineraryFares mapFare(Itinerary itinerary) {
    var fares = itinerary.getFares();

    return new ApiItineraryFares(
      Map.of(),
      Map.of(),
      toApiFareProducts(fares.getItineraryProducts()),
      toApiLegProducts(itinerary, fares.getLegProducts())
    );
  }

  private List toApiLegProducts(
    Itinerary itinerary,
    Multimap legProducts
  ) {
    if (legProducts.isEmpty()) {
      return null;
    } else {
      return legProducts
        .keySet()
        .stream()
        .map(leg -> {
          var index = itinerary.getLegIndex(leg);
          // eventually we want to implement products that span multiple legs (but not the entire itinerary)
          return new ApiLegProducts(
            List.of(index),
            instancesToApiFareProducts(legProducts.get(leg))
          );
        })
        .toList();
    }
  }

  private static ApiFareQualifier toApiFareQualifier(@Nullable FareMedium nullable) {
    return Optional.ofNullable(nullable)
      .map(c -> new ApiFareQualifier(c.id().getId(), c.name()))
      .orElse(null);
  }

  private static ApiFareQualifier toApiFareQualifier(@Nullable RiderCategory nullable) {
    return Optional.ofNullable(nullable)
      .map(c -> new ApiFareQualifier(c.id().getId(), c.name()))
      .orElse(null);
  }

  private List instancesToApiFareProducts(Collection product) {
    return toApiFareProducts(product.stream().map(FareProductUse::product).toList());
  }

  private List toApiFareProducts(Collection product) {
    if (product.isEmpty()) return null;
    else {
      return product
        .stream()
        .map(p ->
          new ApiFareProduct(
            p.id().toString(),
            p.name(),
            toApiMoney(p.price()),
            toApiFareQualifier(p.medium()),
            toApiFareQualifier(p.category())
          )
        )
        .toList();
    }
  }

  private ApiMoney toApiMoney(Money m) {
    var c = m.currency();
    return new ApiMoney(
      m.minorUnitAmount(),
      new ApiCurrency(
        c.getCurrencyCode(),
        c.getDefaultFractionDigits(),
        c.getCurrencyCode(),
        c.getSymbol(locale)
      )
    );
  }
}