src/main/java/de/uapcore/lightpit/viewmodel/VersionInfo.java

package de.uapcore.lightpit.viewmodel;

import de.uapcore.lightpit.entities.Issue;
import de.uapcore.lightpit.entities.IssueSummary;
import de.uapcore.lightpit.entities.Version;

import java.util.ArrayList;
import java.util.List;

public class VersionInfo {

    private final Version version;

    private final IssueSummary reportedTotal = new IssueSummary();
    private final IssueSummary scheduledTotal = new IssueSummary();
    private final IssueSummary resolvedTotal = new IssueSummary();

    private final List<Issue> reported = new ArrayList<>();
    private final List<Issue> scheduled = new ArrayList<>();
    private final List<Issue> resolved = new ArrayList<>();

    public VersionInfo(Version version) {
        this.version = version;
    }

    public Version getVersion() {
        return version;
    }

    public void addReported(Issue issue) {
        reportedTotal.add(issue);
        reported.add(issue);
    }

    public void addScheduled(Issue issue) {
        scheduledTotal.add(issue);
        scheduled.add(issue);
    }

    public void addResolved(Issue issue) {
        resolvedTotal.add(issue);
        resolved.add(issue);
    }

    public IssueSummary getReportedTotal() {
        return reportedTotal;
    }

    public IssueSummary getScheduledTotal() {
        return scheduledTotal;
    }

    public IssueSummary getResolvedTotal() {
        return resolvedTotal;
    }

    public List<Issue> getReported() {
        return reported;
    }

    public List<Issue> getScheduled() {
        return scheduled;
    }

    public List<Issue> getResolved() {
        return resolved;
    }

    public void collectIssues(List<Issue> issues) {
        for (Issue issue : issues) {
            if (issue.getAffectedVersions().contains(version)) {
                addReported(issue);
            }
            if (issue.getScheduledVersions().contains(version)) {
                addScheduled(issue);
            }
            if (issue.getResolvedVersions().contains(version)) {
                addResolved(issue);
            }
        }
    }
}