EnunciateJaxbNamespacePrefixMapper.java

/**
 * Copyright © 2006-2016 Web Cohesion (info@webcohesion.com)
 *
 * 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.webcohesion.enunciate.rt;

import com.sun.xml.bind.marshaller.NamespacePrefixMapper;

import java.util.Properties;

/**
 * A namespace prefix mapper.
 *
 * @author Ryan Heaton
 */
public class EnunciateJaxbNamespacePrefixMapper extends NamespacePrefixMapper {

  private final String defaultNs;
  private final Properties ns2prefix;

  public EnunciateJaxbNamespacePrefixMapper(String defaultNs, Properties ns2prefix) {
    this.defaultNs = defaultNs;
    this.ns2prefix = new Properties();
    if (ns2prefix != null) {
      this.ns2prefix.putAll(ns2prefix);
    }
  }

  public String getPreferredPrefix(String nsuri, String suggestion, boolean requirePrefix) {
    String prefix = this.ns2prefix.containsKey(nsuri) ? this.ns2prefix.getProperty(nsuri) : suggestion;
    if (!requirePrefix && nsuri.equals(this.defaultNs)) {
      prefix = "";
    }
    return prefix;
  }
}