org.entur.gbfs.mapper.DiscoveryFileAdditionalMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gbfs-mapper-java Show documentation

Enables mapping between different versions of GBFS (bidirectional)

There is a newer version: 2.0.16

package org.entur.gbfs.mapper;

import org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeedName;
import org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeeds;
import org.mobilitydata.gbfs.v3_0.gbfs.GBFSFeed;
import org.mapstruct.Context;

import java.net.URI;
import java.util.List;
import java.util.Map;

public class DiscoveryFileAdditionalMapper {
    private static GBFSFeedNameMapper feedNameMapper = GBFSFeedNameMapper.INSTANCE;

    public org.mobilitydata.gbfs.v3_0.gbfs.GBFSData map(Map source, @Context String sourceLanguage) {
        List mappedFeeds = source.get(sourceLanguage).getFeeds().stream()
                .filter(this::filterLegacySourceFeeds)
                .map(this::map)
                .toList();

        return new org.mobilitydata.gbfs.v3_0.gbfs.GBFSData()
                .withFeeds(mappedFeeds);
    }

    public Map map(org.mobilitydata.gbfs.v3_0.gbfs.GBFSData source, @Context String targetLanguage) {
        org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeeds mappedFeeds = new org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeeds();
        mappedFeeds.setFeeds(source.getFeeds().stream().map(this::map).toList());

        return Map.of(
                targetLanguage,
                mappedFeeds
        );
    }

    private boolean filterLegacySourceFeeds(org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeed sourceFeed) {
        return !List.of(
                GBFSFeedName.SystemHours,
                GBFSFeedName.SystemCalendar
        ).contains(sourceFeed.getName());
    }

    private org.mobilitydata.gbfs.v3_0.gbfs.GBFSFeed map(org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeed sourceFeed) {
        org.mobilitydata.gbfs.v3_0.gbfs.GBFSFeed targetFeed = new org.mobilitydata.gbfs.v3_0.gbfs.GBFSFeed();
        targetFeed.setName(feedNameMapper.map(sourceFeed.getName()));
        targetFeed.setUrl(sourceFeed.getUrl().toString());
        return targetFeed;
    }

    private org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeed map(org.mobilitydata.gbfs.v3_0.gbfs.GBFSFeed sourceFeed) {
        org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeed targetFeed = new org.mobilitydata.gbfs.v2_3.gbfs.GBFSFeed();
        targetFeed.setName(feedNameMapper.map(sourceFeed.getName()));
        targetFeed.setUrl(URI.create(sourceFeed.getUrl()));
        return targetFeed;
    }
}