nz.co.testamation.common.util.PrefixMapNamespaceContext Maven / Gradle / Ivy

Go to download
/*
* Copyright 2016 Ratha Long
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package nz.co.testamation.common.util;

import com.google.common.base.Function;
import com.google.common.base.Predicate;
import com.google.common.collect.Iterables;
import org.apache.commons.lang3.StringUtils;

import javax.xml.XMLConstants;
import javax.xml.namespace.NamespaceContext;
import java.util.Iterator;
import java.util.Map;

public class PrefixMapNamespaceContext implements NamespaceContext {

    final Map prefixNamespaceMap;

    public PrefixMapNamespaceContext( Map prefixNamespaceMap ) {
        this.prefixNamespaceMap = prefixNamespaceMap;
    }

    public String getNamespaceURI( String prefix ) {
        switch ( prefix ) {
            case XMLConstants.XML_NS_PREFIX:
                return XMLConstants.XML_NS_URI;
            case XMLConstants.XMLNS_ATTRIBUTE:
                return XMLConstants.XMLNS_ATTRIBUTE_NS_URI;
            default:
                return StringUtils.defaultString( prefixNamespaceMap.get( prefix ), XMLConstants.NULL_NS_URI );
        }
    }

    public String getPrefix( String namespaceURI ) {
        Iterator prefixes = getPrefixes( namespaceURI );
        if ( prefixes.hasNext() ) {
            return prefixes.next();
        }
        return null;
    }

    public Iterator getPrefixes( final String namespaceURI ) {
        return Iterables.transform(
            Iterables.filter(
                prefixNamespaceMap.entrySet(),
                new Predicate>() {
                    @Override
                    public boolean apply( Map.Entry entry ) {
                        return namespaceURI.equals( entry.getValue() );
                    }
                }
            ), new Function, String>() {
                @Override
                public String apply( Map.Entry entry ) {
                    return entry.getKey();
                }
            }
        ).iterator();

    }

}