com.gs.dmn.serialization.AbstractNamespacePrefixMapper Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016 Goldman Sachs.
 *
 * 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 com.gs.dmn.serialization;

import com.sun.xml.bind.marshaller.NamespacePrefixMapper;
import org.apache.commons.lang3.StringUtils;

import java.util.LinkedHashMap;
import java.util.Map;

import static com.gs.dmn.serialization.DMNConstants.XSD_NS;
import static com.gs.dmn.serialization.DMNConstants.XSD_PREFIX;

public abstract class AbstractNamespacePrefixMapper extends NamespacePrefixMapper {
    private final String userNamespace;
    private final String userPrefix;

    protected final Map namespaceMap = new LinkedHashMap<>();

    public AbstractNamespacePrefixMapper() {
        this(null, null, null);
    }

    public AbstractNamespacePrefixMapper(String namespace, String prefix, Map otherNamespaces) {
        this.userNamespace = namespace;
        this.userPrefix = prefix;
        this.namespaceMap.put(XSD_NS, XSD_PREFIX);
        if (otherNamespaces != null) {
            this.namespaceMap.putAll(otherNamespaces);
        }
        if (!StringUtils.isEmpty(userNamespace) && !StringUtils.isEmpty(userPrefix)) {
            this.namespaceMap.put(userNamespace, userPrefix);
        }
    }

    @Override
    public String getPreferredPrefix(String namespaceUri, String suggestion, boolean requirePrefix) {
        return namespaceMap.getOrDefault(namespaceUri, suggestion);
    }

    @Override
    public String[] getPreDeclaredNamespaceUris() {
        return namespaceMap.keySet().toArray(new String[]{});
    }

    public String getUserNamespace() {
        return userNamespace;
    }

    public String getUserPrefix() {
        return userPrefix;
    }
}