package org.eclipse.wst.xsd.ui.internal.common.properties.sections.appinfo;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import org.eclipse.wst.xsd.ui.internal.common.util.XSDCommonUIUtils;
import org.eclipse.xsd.XSDAnnotation;
import org.eclipse.xsd.XSDConcreteComponent;
import org.eclipse.xsd.util.XSDConstants;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

public class XSDExtensionTreeContentProvider extends DOMExtensionTreeContentProvider
{  
  public Object[] getElements(Object inputElement)
  {
    if (inputElement instanceof XSDConcreteComponent)
    {
      XSDConcreteComponent component = (XSDConcreteComponent) inputElement;
      List elementsAndAttributes = new ArrayList();
      XSDAnnotation xsdAnnotation = XSDCommonUIUtils.getInputXSDAnnotation(component, false);
      if (xsdAnnotation != null)
      {
        // Bug 137856 - Extension page of properties shouldn't display imported elements
        // Added this if statement
        if (xsdAnnotation.getSchema() == component.getSchema())
        {
          List appInfoList = xsdAnnotation.getApplicationInformation();
          Element appInfoElement = null;
          if (appInfoList.size() > 0)
          {
            appInfoElement = (Element) appInfoList.get(0);
          }
          if (appInfoElement != null)
          {
            for (Iterator it = appInfoList.iterator(); it.hasNext();)
            {
              Object obj = it.next();
              if (obj instanceof Element)
              {
                Element appInfo = (Element) obj;
                NodeList nodeList = appInfo.getChildNodes();
                int length = nodeList.getLength();
                for (int i = 0; i < length; i++)
                {
                  Node node = nodeList.item(i);
                  if (node instanceof Element)
                  {
                    elementsAndAttributes.add(node);
                  }
                }
              }
            }
            
            /** Construct attributes list */
            NamedNodeMap attributes = appInfoElement.getAttributes();
            if (attributes != null)
            {
              // String defaultNamespace =
              // (String)component.getSchema().getQNamePrefixToNamespaceMap().get("");
              int length = attributes.getLength();
              for (int i = 0; i < length; i++)
              {
                Node oneAttribute = attributes.item(i);
                if (!isXmlnsAttribute(oneAttribute))
                {
                  String namespace = oneAttribute.getNamespaceURI();
                  boolean isExtension = true;
                  if (namespace == null && oneAttribute.getPrefix() == null)
                  {
                    isExtension = false;
                  }
                  else if (!XSDConstants.SCHEMA_FOR_SCHEMA_URI_2001.equals(namespace))
                  {
                    isExtension = true;
                  }
                  if (isExtension)
                  {
                    elementsAndAttributes.add(oneAttribute);
                  }
                }
              }
            }
          }
        }
      }
      return elementsAndAttributes.toArray();
    }
    return Collections.EMPTY_LIST.toArray();
  }

  private static boolean isXmlnsAttribute(Node attribute)
  {
    String prefix = attribute.getPrefix();
    if (prefix != null)
    {
      // this handle the xmlns:foo="blah" case
      return "xmlns".equals(prefix); //$NON-NLS-1$
    }
    else
    {
      // this handles the xmlns="blah" case
      return "xmlns".equals(attribute.getNodeName()); //$NON-NLS-1$
    }
  }
}