org.opentripplanner.apis.support.mapping.StreetNoteMapper 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.apis.support.mapping;

import java.util.List;
import java.util.Locale;
import java.util.Set;
import java.util.stream.Collectors;
import org.opentripplanner.ext.restapi.model.ApiAlert;
import org.opentripplanner.street.model.note.StreetNote;

public class StreetNoteMapper {

  private final Locale locale;

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

  public List mapToApi(Set newAlerts) {
    // Using {@code null} and not an empty set will minimize the JSON removing the
    // {@code alerts} from the result.
    if (newAlerts == null || newAlerts.isEmpty()) {
      return null;
    }

    return newAlerts.stream().map(this::mapToApi).collect(Collectors.toList());
  }

  ApiAlert mapToApi(StreetNote domain) {
    ApiAlert api = new ApiAlert();
    if (domain.note != null) {
      api.alertHeaderText = domain.note.toString(locale);
    }

    if (domain.descriptionText != null) {
      api.alertDescriptionText = domain.descriptionText.toString(locale);
    }

    if (domain.url != null) {
      api.alertUrl = domain.url;
    }

    api.effectiveStartDate = domain.effectiveStartDate;
    api.effectiveEndDate = domain.effectiveEndDate;

    return api;
  }
}