package com.infoclinika.pdx.projection;

import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Set;

import static java.util.stream.Collectors.toSet;

/**
 * Projection with results of gene-synonyms mapping.
 */
public class GeneMappingResponse {
    private final Set<String> missingGenes;
    private final Set<String> averagedGenes;
    private final Map<String, Set<String>> multiGeneSynonyms;

    /**
     * Default constructor.
     */
    public GeneMappingResponse() {
        this.missingGenes = new HashSet<>();
        this.averagedGenes = new HashSet<>();
        this.multiGeneSynonyms = new HashMap<>();
    }

    /**
     * Initializes all fields.
     */
    public GeneMappingResponse(Set<String> missingGenes, Set<String> averagedGenes,
                               Map<String, Set<String>> multiGeneSynonyms) {
        this.missingGenes = missingGenes;
        this.averagedGenes = averagedGenes;
        this.multiGeneSynonyms = multiGeneSynonyms;
    }

    /**
     * Generates single response from the list of responses.
     *
     * @param responses List of gene mapping responses.
     * @return Aggregated gene mapping response.
     */
    public static GeneMappingResponse listToResponse(List<GeneMappingResponse> responses) {
        final Set<String> missingGenes = responses.stream()
                                                  .flatMap(response -> response.missingGenes.stream())
                                                  .collect(toSet());
        final Set<String> averagedGenes = responses.stream()
                                                   .flatMap(response -> response.averagedGenes.stream())
                                                   .collect(toSet());
        final Map<String, Set<String>> multiGeneSynonyms = new HashMap<>();
        responses.forEach(response -> multiGeneSynonyms.putAll(response.multiGeneSynonyms));

        return new GeneMappingResponse(missingGenes, averagedGenes, multiGeneSynonyms);
    }

    public Set<String> getMissingGenes() {
        return missingGenes;
    }

    public Set<String> getAveragedGenes() {
        return averagedGenes;
    }

    public Map<String, Set<String>> getMultiGeneSynonyms() {
        return multiGeneSynonyms;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }
        final GeneMappingResponse that = (GeneMappingResponse) o;
        return Objects.equals(getMissingGenes(), that.getMissingGenes())
               && Objects.equals(getAveragedGenes(), that.getAveragedGenes())
               && Objects.equals(getMultiGeneSynonyms(), that.getMultiGeneSynonyms());
    }

    @Override
    public int hashCode() {
        return Objects.hash(getMissingGenes(), getAveragedGenes(), getMultiGeneSynonyms());
    }

    @Override
    public String toString() {
        return "GeneMappingResponse{"
               + "missingGenes=" + missingGenes
               + ", averagedGenes=" + averagedGenes
               + ", multiGeneSynonyms=" + multiGeneSynonyms
               + '}';
    }
}