com.amalto.core.history.accessor.UnaryFieldAccessor.java Source code

Introduction

Here is the source code for com.amalto.core.history.accessor.UnaryFieldAccessor.java

Source

/*
 * Copyright (C) 2006-2016 Talend Inc. - www.talend.com
 * 
 * This source code is available under agreement available at
 * %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
 * 
 * You should have received a copy of the agreement along with this program; if not, write to Talend SA 9 rue Pages
 * 92150 Suresnes, France
 */

package com.amalto.core.history.accessor;

import javax.xml.XMLConstants;

import com.amalto.core.history.DOMMutableDocument;
import org.apache.commons.lang.StringUtils;
import org.w3c.dom.Attr;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.w3c.dom.Text;

import com.amalto.core.history.action.FieldUpdateAction;
import com.amalto.core.schema.validation.SkipAttributeDocumentBuilder;

/**
 *
 */
class UnaryFieldAccessor implements DOMAccessor {

    private final DOMAccessor parent;

    private final String fieldName;

    private final DOMMutableDocument document;

    public UnaryFieldAccessor(DOMAccessor parent, String fieldName, DOMMutableDocument document) {
        this.parent = parent;
        this.fieldName = fieldName;
        this.document = document;
    }

    private Element getElement() {
        Element element = null;
        Node parentNode = parent.getNode();
        if (parentNode != null) {
            Node current = parentNode.getFirstChild();
            while (current != null) {
                if (fieldName.equals(current.getNodeName())) {
                    element = (Element) current;
                    break;
                }
                current = current.getNextSibling();
            }
        }
        return element;
    }

    private void internalSet(String value, Element element) {
        element.setTextContent(value);
    }

    private Element internalCreate() {
        parent.create();
        Document domDocument = document.asDOM();
        Element element = getElement();
        if (element == null) {
            Element newElement = domDocument.createElementNS(domDocument.getNamespaceURI(), fieldName);
            Node parentNode = parent.getNode();
            Node lastAccessedNode = document.getLastAccessedNode();
            if (parentNode == lastAccessedNode) {
                parentNode.insertBefore(newElement, parentNode.getFirstChild());
            } else if (lastAccessedNode != null && lastAccessedNode.getParentNode() == parentNode) {
                parentNode.insertBefore(newElement, lastAccessedNode.getNextSibling());
            } else {
                parentNode.appendChild(newElement);
            }
            element = newElement;
            document.setLastAccessedNode(element);
        }
        return element;
    }

    @Override
    public void set(String value) {
        Element element = getElement();
        internalSet(value, element);
    }

    @Override
    public String get() {
        Element element = getElement();
        Node textChild = element.getFirstChild();
        if (textChild != null) {
            if (textChild instanceof Text) {
                return element.getTextContent();// get node value can not handle bracket well
            } else {
                return textChild.getNodeValue();
            }
        } else {
            return StringUtils.EMPTY;
        }
    }

    @Override
    public void touch() {
        document.setLastAccessedNode(getElement());
    }

    @Override
    public Node getNode() {
        return getElement();
    }

    @Override
    public void create() {
        internalCreate();
    }

    @Override
    public void insert() {
        create();
    }

    @Override
    public void createAndSet(String value) {
        Element element = internalCreate();
        internalSet(value, element);
    }

    @Override
    public void delete() {
        while (exist()) {
            Element element = getElement();
            element.getParentNode().removeChild(element);
        }

        // if the parent is exist and have no the child, will remove.
        if (parent.exist() && isEmptyChildForXSIType()) {
            Node parentNode = parent.getNode();
            parentNode.getParentNode().removeChild(parentNode);
        }
    }

    @Override
    public boolean exist() {
        return parent.exist() && getElement() != null;
    }

    public boolean isEmptyChildForXSIType() {
        Node parentNode = parent.getNode();
        NodeList childs = parentNode.getChildNodes();
        return childs.getLength() == 0;
    }

    @Override
    public void markModified(Marker marker) {
        Document domDocument = document.asDOM();
        Element element = getElement();
        if (element != null) {
            Attr newAttribute = domDocument.createAttribute(MODIFIED_MARKER_ATTRIBUTE);
            switch (marker) {
            case ADD:
                newAttribute.setValue(FieldUpdateAction.MODIFY_ADD_MARKER_VALUE);
                break;
            case UPDATE:
                newAttribute.setValue(FieldUpdateAction.MODIFY_UPDATE_MARKER_VALUE);
                break;
            case REMOVE:
                newAttribute.setValue(FieldUpdateAction.MODIFY_REMOVE_MARKER_VALUE);
                break;
            default:
                throw new IllegalArgumentException("No support for marker " + marker); //$NON-NLS-1$
            }
            element.getAttributes().setNamedItem(newAttribute);
        }
    }

    @Override
    public void markUnmodified() {
        Node parentNode = getElement();
        NamedNodeMap attributes = parentNode.getAttributes();
        if (attributes.getNamedItem(MODIFIED_MARKER_ATTRIBUTE) != null) {
            attributes.removeNamedItem(MODIFIED_MARKER_ATTRIBUTE);
        }
    }

    @Override
    public int size() {
        if (!exist()) {
            return 0;
        }
        Node node = parent.getNode();
        if (node instanceof Element) {
            Element parentElement = (Element) node;
            return parentElement.getElementsByTagName(fieldName).getLength();
        } else {
            return 1;
        }
    }

    @Override
    public String getActualType() {
        Attr type = ((Element) getNode()).getAttributeNodeNS(XMLConstants.W3C_XML_SCHEMA_INSTANCE_NS_URI, "type"); //$NON-NLS-1$
        if (type == null) {
            type = ((Element) getNode()).getAttributeNodeNS(SkipAttributeDocumentBuilder.TALEND_NAMESPACE, "type"); //$NON-NLS-1$
        }
        if (type == null) {
            return StringUtils.EMPTY;
        } else {
            return type.getValue();
        }
    }

    @Override
    public int compareTo(Accessor accessor) {
        if (exist() != accessor.exist()) {
            return -1;
        }
        if (exist()) {
            return get().equals(accessor.get()) ? 0 : -1;
        }
        return -1;
    }
}