package com.infoclinika.pdx.projection;

import java.util.HashSet;
import java.util.Objects;
import java.util.Set;

/**
 * Projection with results of expression data upload.
 */
public class ExpressionDataUploadResponse {
    private final GeneMappingResponse geneMappingResponse;
    private final Set<String> missingSamples;
    private final int totalSamples;
    private final DataUploadStatus status;

    /**
     * Default constructor.
     */
    public ExpressionDataUploadResponse() {
        this.geneMappingResponse = new GeneMappingResponse();
        this.missingSamples = new HashSet<>();
        this.totalSamples = 0;
        this.status = DataUploadStatus.SUCCESS;
    }

    /**
     * Constructor to with gene mapping response.
     */
    public ExpressionDataUploadResponse(GeneMappingResponse geneMappingResponse, DataUploadStatus status) {
        this.geneMappingResponse = geneMappingResponse;
        this.missingSamples = new HashSet<>();
        this.totalSamples = 0;
        this.status = status;
    }

    /**
     * Constructor to with missing sample data.
     */
    public ExpressionDataUploadResponse(Set<String> samples, int total, DataUploadStatus status) {
        this.geneMappingResponse = new GeneMappingResponse();
        this.missingSamples = samples;
        this.totalSamples = total;
        this.status = status;
    }

    public GeneMappingResponse getGeneMappingResponse() {
        return geneMappingResponse;
    }

    public Set<String> getMissingSamples() {
        return missingSamples;
    }

    public int getTotalSamples() {
        return totalSamples;
    }

    public DataUploadStatus getStatus() {
        return status;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }
        final ExpressionDataUploadResponse that = (ExpressionDataUploadResponse) o;
        return getTotalSamples() == that.getTotalSamples()
               && Objects.equals(getGeneMappingResponse(), that.getGeneMappingResponse())
               && Objects.equals(getMissingSamples(), that.getMissingSamples())
               && getStatus() == that.getStatus();
    }

    @Override
    public int hashCode() {
        return Objects.hash(getGeneMappingResponse(), getMissingSamples(), getTotalSamples(), getStatus());
    }

    @Override
    public String toString() {
        return "ExpressionDataUploadResponse{"
               + "geneMappingResponse=" + geneMappingResponse
               + ", missingSamples=" + missingSamples
               + ", totalSamples=" + totalSamples
               + ", status=" + status
               + '}';
    }
}