/*
* 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<String, String> {
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<? extends String, ? extends String> m) {
for (Object o : m.entrySet())
if (o instanceof Map.Entry) {
Map.Entry<String, String> 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();
}
}