groovy.util.slurpersupport.NamespaceAwareHashMap Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of groovy-all Show documentation

Groovy: A powerful, dynamic language for the JVM

There is a newer version: 3.0.22

/*
 * Copyright 2003-2014 the original author or authors.
 *
 * 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 groovy.util.slurpersupport;

import groovy.xml.QName;

import java.util.HashMap;
import java.util.Map;

public class NamespaceAwareHashMap extends HashMap {
    public void setNamespaceTagHints(Map namespaceTagHints) {
        this.namespaceTagHints = namespaceTagHints;
    }

    private Map namespaceTagHints = null;

    public Map getNamespaceTagHints() {
        return namespaceTagHints;
    }

    @Override
    public String get(Object key) {
        key = adjustForNamespaceIfNeeded(key);
        return super.get(key);
    }

    @Override
    public String remove(Object key) {
        key = adjustForNamespaceIfNeeded(key).toString();
        return super.remove(key);
    }

    @Override
    public boolean containsKey(Object key) {
        key = adjustForNamespaceIfNeeded(key).toString();
        return super.containsKey(key);
    }

    @Override
    public String put(String key, String value) {
        key = adjustForNamespaceIfNeeded(key).toString();
        return super.put(key, value);
    }

    @Override
    public void putAll(Map m) {
        for (Object o : m.entrySet())
            if (o instanceof Map.Entry) {
                Map.Entry e = (Map.Entry) o;
                put(e.getKey(), e.getValue());
            }
    }

    private Object adjustForNamespaceIfNeeded(Object key) {
        String keyString = key.toString();
        if (keyString.contains("{") || namespaceTagHints == null || namespaceTagHints.isEmpty() || keyString.contains("{") || !keyString.contains(":")) {
            return key;
        }
        final int i = keyString.indexOf(":");
        return new QName(namespaceTagHints.get(keyString.substring(0, i)).toString(), keyString.substring(i + 1)).toString();
    }
}