/* Copyright (c) 2008 Google Inc.
 *
 * 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.google.gdata.wireformats;

import com.google.common.collect.Maps;
import com.google.gdata.util.common.xml.XmlNamespace;
import com.google.gdata.model.Attribute;
import com.google.gdata.model.AttributeKey;
import com.google.gdata.model.AttributeMetadata;
import com.google.gdata.model.Element;
import com.google.gdata.model.ElementMetadata;
import com.google.gdata.model.QName;

import java.util.Iterator;
import java.util.Map;

/**
 * Utility functions shared across generators.
 *
 * 
 */
public class GeneratorUtils {

  /**
   * Calculate the set of namespaces on an element.  This will find all
   * namespaces declared on the element or child elements, ordered in
   * depth-first order.
   */
  public static Map<String, XmlNamespace> calculateNamespaces(
      Element root, ElementMetadata<?, ?> metadata) {
    Map<String, XmlNamespace> namespaceMap = Maps.newHashMap();
    calculateNamespaces(namespaceMap, root, metadata);
    return namespaceMap;
  }

  /**
   * Calculate the namespaces on an element using the given visitor to store
   * the namespaces.  We cheat by using an attribute visitor for even visiting
   * the element names, because all we're doing with the visitor is adding a
   * QName.
   */
  private static void calculateNamespaces(Map<String, XmlNamespace> namespaces,
      Element e, ElementMetadata<?, ?> metadata) {
    QName name = (metadata == null) ? e.getElementId() : metadata.getName();
    addNamespace(namespaces, name);

    Iterator<Attribute> attIter = e.getAttributeIterator(metadata);
    while (attIter.hasNext()) {
      Attribute att = attIter.next();
      AttributeKey<?> attKey = att.getAttributeKey();
      AttributeMetadata<?> attMeta = (metadata == null) ? null
          : metadata.bindAttribute(attKey);
      name = (attMeta == null) ? attKey.getId() : attMeta.getName();
      addNamespace(namespaces, name);
    }

    Iterator<Element> childIter = e.getElementIterator(metadata);
    while (childIter.hasNext()) {
      Element child = childIter.next();
      ElementMetadata<?, ?> childMeta = (metadata == null) ? null
          : metadata.bindElement(child.getElementKey());
      calculateNamespaces(namespaces, child, childMeta);
    }
  }

  /**
   * Add a qualified name to the map by URI.  Only the first namespace with a
   * given URI is added to the map.
   */
  private static void addNamespace(Map<String, XmlNamespace> namespaces,
      QName name) {
    if (name == null) {
      return;
    }
    XmlNamespace ns = name.getNs();
    if (ns == null) {
      return;
    }
    String alias = ns.getAlias();
    if (alias == null) {
      return;
    }
    String uri = ns.getUri();
    if (namespaces.containsKey(uri)) {
      return;
    }
    namespaces.put(uri, ns);
  }
}