package io.delphiplatform.api.v3.rdb.service.amazon;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Sets;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import io.delphiplatform.api.util.CollectionUtils;
import io.delphiplatform.api.v3.model.IncludeChart;
import io.delphiplatform.api.v3.model.amazon.AmazonChartTrackPosition;
import io.delphiplatform.api.v3.rdb.entity.amazon.AmazonChartTrackEntity;
import io.delphiplatform.api.v3.rdb.entity.amazon.AmazonTrackEntity;
import io.delphiplatform.api.v3.rdb.repository.amazon.AmazonTrackDailyRepository;
import io.delphiplatform.api.v3.rdb.service.EntityService;
import io.delphiplatform.api.v3.rdb.service.dto.JpaPropertyPath;
import io.delphiplatform.api.v3.rdb.service.specification.SpecificationProvider;
import io.delphiplatform.api.v3.view.util.Params;

import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.AMAZON_ALBUM;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.AMAZON_GENRE;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.AMAZON_GENRE_NAME;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.CHART;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.CHART_ID;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.COUNTRY_CODE;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.CURRENT_POSITION;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.DATE_PROPERTY;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.ISRC;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.IS_EXIT;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.IS_SONY;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.REGION;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.TRACK;
import static io.delphiplatform.api.v3.rdb.entity.CommonJpaPropertyNames.TRACK_ID;

@Service
@Transactional(readOnly = true)
public class AmazonChartTrackPositionService extends EntityService<AmazonChartTrackEntity> {

    private static final ImmutableMap<String, String> SORT_BY_TO_ENTITY_FIELD_MAP = ImmutableMap
        .<String, String>builder()
        .put("chart.chart_id", "chart_chart_id")
        .put("chart.total_positions", "chart_total_positions")
        .put("chart.country_code", "chart_country_code")
        .put("chart.genre", "chart_genre_name")
        .put("position.current", "current_position")
        .put("position.date", "date")
        .put("position.date_gap", "date_gap")
        .put("position.is_entry", "is_chart_entry")
        .put("position.is_exit", "is_exit")
        .put("position.is_reentry", "is_chart_reentry")
        .put("position.previous_date", "prev_chart_date")
        .put("position.previous_position", "previous_position")
        .put("position.trend", "trend")
        .put("position.total_days", "total_days")
        .put("track.name", "track_name")
        .put("track.amazon_track_id", "track_amazon_track_id")
        .put("track.is_sony", "track_is_sony")
        .put("track.isrc", "track_isrc")
        .put("track.release_date", "track_amazon_album_release_date")
        .put("isrc", "track_isrc") //because isrc for response model is taken from track
        .build();
    private final SpecificationProvider<AmazonChartTrackEntity> specificationProvider;
    private final AmazonArtistService artistService;

    @Autowired
    public AmazonChartTrackPositionService(AmazonTrackDailyRepository repository,
        SpecificationProvider<AmazonChartTrackEntity> specificationProvider,
        AmazonArtistService artistService) {
        super(repository, SORT_BY_TO_ENTITY_FIELD_MAP);
        this.specificationProvider = specificationProvider;
        this.artistService = artistService;
    }

    public List<AmazonChartTrackPosition> getAmazonChartTrackPositions(Params params) {
        boolean includeTrack = CollectionUtils.contains(params.getIncludeChart(), IncludeChart.TRACK);

        Set<JpaPropertyPath> requestFetches = Sets.newHashSet(JpaPropertyPath.ofJoins(CHART, AMAZON_GENRE));

        if (includeTrack) {
            requestFetches.add(JpaPropertyPath.ofJoins(TRACK, AMAZON_ALBUM));
        } else {
            requestFetches.add(JpaPropertyPath.ofJoins(TRACK));
        }

        Specification<AmazonChartTrackEntity> spec = and(Arrays.asList(
            specificationProvider.greaterThanOrEqualToValueSpec(DATE_PROPERTY, params.getStartDate()),
            specificationProvider.lessThanOrEqualToValueSpec(DATE_PROPERTY, params.getEndDate()),
            specificationProvider.multiValueSpec(List.of(CHART, CHART_ID), params.getChartId()),
            specificationProvider.singleValueSpec(List.of(TRACK, IS_SONY), params.getIsSony()),
            specificationProvider.singleValueSpec(IS_EXIT, params.getIsExit()),
            specificationProvider.multiValueSpec(List.of(TRACK, ISRC), params.getIsrc()),
            specificationProvider.multiValueSpec(TRACK_ID, params.getTrackId()),
            specificationProvider.multiTextIgnoreCaseValueSpec(List.of(CHART, AMAZON_GENRE, AMAZON_GENRE_NAME), params.getGenreName()),
            specificationProvider.greaterThanOrEqualToValueSpec(CURRENT_POSITION, params.getMinPosition()),
            specificationProvider.lessThanOrEqualToValueSpec(CURRENT_POSITION, params.getMaxPosition()),
            specificationProvider.multiValueSpec(List.of(CHART, REGION, COUNTRY_CODE), params.getCountryCode()),

            specificationProvider.fetchesSpecOfPaths(requestFetches)
        ));

        List<AmazonChartTrackEntity> entities = find(spec, getPageRequestSortByToCamelCase(params));

        Set<String> trackIds = entities.stream()
            .map(AmazonChartTrackEntity::getTrack)
            .map(AmazonTrackEntity::getAmazonTrackId)
            .collect(Collectors.toSet());

        Map<String, String> trackToArtist = artistService.findByTrackIdToArtistNames(trackIds);

        return entities.stream().map(elem -> AmazonChartTrackPosition.of(elem, includeTrack,
                trackToArtist.get(elem.getTrack().getAmazonTrackId())))
            .collect(Collectors.toList());
    }
}