nl.tue.gale.ae.processor.xmlmodule.RepositoryModule.java Source code

Introduction

Here is the source code for nl.tue.gale.ae.processor.xmlmodule.RepositoryModule.java

Source

/*
    
   This file is part of GALE (Generic Adaptation Language and Engine).
    
GALE 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 (at your option) 
any later version.
    
GALE 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.
    
You should have received a copy of the GNU Lesser General Public License
along with GALE. If not, see <http://www.gnu.org/licenses/>.
    
 */
/**
 * RepositoryModule.java
 * Last modified: $Date$
 * In revision:   $Revision$
 * Modified by:   $Author$
 *
 * Copyright (c) 2008-2011 Eindhoven University of Technology.
 * All Rights Reserved.
 *
 * This software is proprietary information of the Eindhoven University
 * of Technology. It may be used according to the GNU LGPL license.
 */
package nl.tue.gale.ae.processor.xmlmodule;

import static nl.tue.gale.common.GaleUtil.createHTMLElement;

import java.util.List;

import nl.tue.gale.ae.GaleContext;
import nl.tue.gale.ae.ProcessorException;
import nl.tue.gale.ae.Resource;
import nl.tue.gale.common.GaleUtil;

import org.dom4j.Element;
import org.dom4j.Node;

import com.google.common.collect.ImmutableList;

public class RepositoryModule extends AbstractModule {
    private ImmutableList<String> mimeToHandle = ImmutableList.of("text/xml", "application/xml");

    @Override
    public List<String> getMimeToHandle() {
        return mimeToHandle;
    }

    public void setMimeToHandle(List<String> mimeToHandle) {
        this.mimeToHandle = ImmutableList.copyOf(mimeToHandle);
    }

    @Override
    public Element traverse(Element element, Resource resource) throws ProcessorException {
        GaleContext gale = GaleContext.of(resource);
        removeTextNodes(element);
        resource.put("mime", "text/xhtml");
        if (gale.isObject()) {
            processor.traverseChildren(element, resource);
            Element span = GaleUtil.createHTMLElement("span");
            moveContent(element, span);
            return (Element) GaleUtil.replaceNode(element, span);
        }
        Element result = createHTMLElement("html");
        Element body = result.addElement("body");
        moveContent(element, body);
        processor.traverse(result, resource);
        GaleUtil.replaceNode(element, result);
        return null;
    }

    private void moveContent(Element source, Element target) {
        @SuppressWarnings("unchecked")
        List<Node> content = (List<Node>) source.content();
        for (Node node : ImmutableList.copyOf(content)) {
            node.detach();
            target.add(node);
        }
    }

    private void removeTextNodes(Element element) {
        @SuppressWarnings("unchecked")
        List<Node> content = (List<Node>) element.content();
        for (Node node : ImmutableList.copyOf(content))
            if (node.getNodeType() == Node.TEXT_NODE)
                content.remove(node);
    }
}