XstreamTreeWriter.java

/**
 * Copyright (c) Codice Foundation
 * <p/>
 * This is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser
 * General Public License as published by the Free Software Foundation, either version 3 of the
 * License, or any later version.
 * <p/>
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
 * even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details. A copy of the GNU Lesser General Public License
 * is distributed along with this program and can be found at
 * <http://www.gnu.org/licenses/lgpl.html>.
 **/

package org.codice.ddf.spatial.ogc.csw.catalog.converter;

import java.util.regex.Pattern;

import org.apache.commons.lang.StringUtils;

import com.thoughtworks.xstream.io.path.Path;
import com.thoughtworks.xstream.io.path.PathTracker;
import com.thoughtworks.xstream.io.path.PathTrackingWriter;

public class XstreamTreeWriter {

    private static final Pattern NORMALIZE_NODE = Pattern.compile("\\[[0-9]+\\]");

    private PathTrackingWriter writer = null;

    private PathTracker tracker = null;

    private XstreamPathValueTracker pathValues = null;

    XstreamTreeWriter(PathTrackingWriter writer, PathTracker tracker,
            XstreamPathValueTracker pathValues) {
        this.writer = writer;
        this.tracker = tracker;
        this.pathValues = pathValues;
    }

    void startVisit(final String node) {
        if (!isAttributeNode(node)) {
            writer.startNode(normalizeNode(node));
        }

        Path currentPath = tracker.getPath();

        for (Path path : pathValues.getPaths()) {
            String value = pathValues.getFirstValue(path);

            if (value != null) {

                if (currentPath.isAncestor(path) && path.toString()
                        .endsWith(node)) {

                    if (isAttributePath(path) && isAttributeNode(node)) {

                        writer.addAttribute(getAttributeNameFromPath(path), value);
                    }
                }
            }
        }

    }

    private String normalizeNode(String node) {
        return NORMALIZE_NODE.matcher(node)
                .replaceAll("");
    }

    void endVisit(final String node) {
        if (!isAttributeNode(node)) {

            Path currentPath = tracker.getPath();

            for (Path path : pathValues.getPaths()) {
                String value = pathValues.getFirstValue(path);

                if (value != null) {

                    if (currentPath.isAncestor(path) && path.toString()
                            .endsWith(node)) {

                        if (!isAttributeNode(node) && !isAttributePath(path)) {

                            writer.setValue(value);
                        }
                    }
                }
            }

            writer.endNode();
        }
    }

    private boolean isAttributeNode(final String node) {
        return node.trim()
                .startsWith("@");
    }

    public boolean isAttributePath(final Path path) {
        return path.toString()
                .contains("@");
    }

    public String getAttributeNameFromPath(final Path path) {
        return StringUtils.substringAfterLast(path.toString(), "@");
    }

}