package gov.uspto.patent.doc.sgml.fragments;

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

import org.dom4j.Document;
import org.dom4j.Node;

import gov.uspto.parser.dom4j.DOMFragmentReader;
import gov.uspto.patent.doc.sgml.items.DocNode;
import gov.uspto.patent.model.DocumentId;
import gov.uspto.patent.model.DocumentIdType;

public class PctRegionalIdNode extends DOMFragmentReader<List<DocumentId>> {
    private static final String FRAGMENT_PATH = "/PATDOC/SDOBI/B800/B860/B861/DOC"; // PCT or regional authority filing
    private static final String FRAGMENT_PATH2 = "/PATDOC/SDOBI/B800/B870/B871/DOC"; // PCT or regional authority publication

    public PctRegionalIdNode(Document document) {
        super(document);
    }

    @Override
    public List<DocumentId> read() {
        List<DocumentId> pctDocIds = new ArrayList<DocumentId>();

        Node fragmentNode = document.selectSingleNode(FRAGMENT_PATH);
        if (fragmentNode != null) {
            DocumentId documentId = new DocNode(fragmentNode).read();
            if (documentId != null) {
                documentId.setType(DocumentIdType.REGIONAL_FILING);
            }

            pctDocIds.add(documentId);
        }

        Node fragmentNode2 = document.selectSingleNode(FRAGMENT_PATH2);
        if (fragmentNode2 != null) {
            DocumentId documentId = new DocNode(fragmentNode2).read();
            if (documentId != null) {
                documentId.setType(DocumentIdType.REGIONAL_PUBLICATION);
            }
            pctDocIds.add(documentId);
        }

        return pctDocIds;
    }

}