001    /*
002     * JBoss, Home of Professional Open Source.
003     * Copyright 2008, Red Hat Middleware LLC, and individual contributors
004     * as indicated by the @author tags. See the copyright.txt file in the
005     * distribution for a full listing of individual contributors. 
006     *
007     * This is free software; you can redistribute it and/or modify it
008     * under the terms of the GNU Lesser General Public License as
009     * published by the Free Software Foundation; either version 2.1 of
010     * the License, or (at your option) any later version.
011     *
012     * This software is distributed in the hope that it will be useful,
013     * but WITHOUT ANY WARRANTY; without even the implied warranty of
014     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
015     * Lesser General Public License for more details.
016     *
017     * You should have received a copy of the GNU Lesser General Public
018     * License along with this software; if not, write to the Free
019     * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
020     * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
021     */
022    package org.jboss.dna.common.xml;
023    
024    import java.util.HashMap;
025    import java.util.Iterator;
026    import java.util.Map;
027    import javax.xml.namespace.NamespaceContext;
028    
029    /**
030     * @author Randall Hauch
031     */
032    public class SimpleNamespaceContext implements NamespaceContext {
033    
034        private final Map<String, String> prefixToNamespace = new HashMap<String, String>();
035    
036        /**
037         * {@inheritDoc}
038         */
039        public String getNamespaceURI( String prefix ) {
040            return this.prefixToNamespace.get(prefix);
041        }
042    
043        /**
044         * {@inheritDoc}
045         */
046        public String getPrefix( String namespaceURI ) {
047            for (Map.Entry<String, String> entry : this.prefixToNamespace.entrySet()) {
048                if (entry.getValue().equals(namespaceURI)) return entry.getKey();
049            }
050            return null;
051        }
052    
053        public SimpleNamespaceContext setNamespace( String prefix,
054                                                    String namespaceUri ) {
055            this.prefixToNamespace.put(prefix, namespaceUri);
056            return this;
057        }
058    
059        /**
060         * {@inheritDoc}
061         */
062        public Iterator<String> getPrefixes( String namespaceURI ) {
063            return this.prefixToNamespace.keySet().iterator();
064        }
065    
066    }