/*
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF licenses this file
 *  to you 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(":")) {
            return key;
        }
        final int i = keyString.indexOf(":");
        return new QName(namespaceTagHints.get(keyString.substring(0, i)).toString(), keyString.substring(i + 1)).toString();
    }
}
