LawTreeView.java

package gov.nysenate.openleg.client.view.law;

import gov.nysenate.openleg.client.view.base.ViewObject;
import gov.nysenate.openleg.model.law.LawDocument;
import gov.nysenate.openleg.model.law.LawTree;
import gov.nysenate.openleg.model.law.LawTreeNode;
import gov.nysenate.openleg.service.law.data.LawDocumentNotFoundEx;

import java.time.LocalDate;
import java.util.List;
import java.util.Map;
import java.util.Optional;

public class LawTreeView implements ViewObject
{
    protected LawVersionIdView lawVersion;
    protected LawInfoView info;
    protected List<LocalDate> publishedDates;
    protected LawNodeView documents;

    public LawTreeView(LawTree lawTree, String fromLocation, Integer depth) {
        this(lawTree, fromLocation, depth, null);
    }

    public LawTreeView(LawTree lawTree, String fromLocation, Integer depth, Map<String, LawDocument> docMap) {
        if (lawTree != null) {
            lawVersion = new LawVersionIdView(lawTree.getLawVersionId());
            info = new LawInfoView(lawTree.getLawInfo());
            publishedDates = lawTree.getPublishedDates();
            if (fromLocation != null && !fromLocation.isEmpty()) {
                Optional<LawTreeNode> fromNode = lawTree.getRootNode().findNode(info.lawId + fromLocation, false);
                if (fromNode.isPresent()) {
                    documents = new LawNodeView(fromNode.get(), depth, docMap);
                }
                else {
                    throw new LawDocumentNotFoundEx(info.lawId, LocalDate.now(),
                            "The location " + fromLocation + " does not exist for this law tree,");
                }
            }
            else {
                documents = new LawNodeView(lawTree.getRootNode(), depth, docMap);
            }
        }
    }

    @Override
    public String getViewType() {
        return "law-tree";
    }

    public LawVersionIdView getLawVersion() {
        return lawVersion;
    }

    public LawInfoView getInfo() {
        return info;
    }

    public List<LocalDate> getPublishedDates() {
        return publishedDates;
    }

    public LawNodeView getDocuments() {
        return documents;
    }
}