bundles/org.eclipse.wst.xsd.ui/src/org/eclipse/wst/xsd/ui/internal/refactor/rename/GlobalSimpleOrComplexTypeRenamer.java

/*******************************************************************************
 * Copyright (c) 2001, 2004 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.wst.xsd.ui.internal.refactor.rename;

import java.util.Iterator;

import org.eclipse.wst.xsd.ui.internal.util.XSDDOMHelper;
import org.eclipse.xsd.XSDAttributeDeclaration;
import org.eclipse.xsd.XSDAttributeGroupContent;
import org.eclipse.xsd.XSDAttributeUse;
import org.eclipse.xsd.XSDComplexTypeDefinition;
import org.eclipse.xsd.XSDElementDeclaration;
import org.eclipse.xsd.XSDNamedComponent;
import org.eclipse.xsd.XSDTypeDefinition;
import org.eclipse.xsd.util.XSDConstants;
import org.w3c.dom.Element;


public class GlobalSimpleOrComplexTypeRenamer extends BaseRenamer
{
  /**
   * Constructor for GlobalSimpleOrComplexTypeRenamer.
   * @param globalComponent
   */
  public GlobalSimpleOrComplexTypeRenamer(XSDNamedComponent globalComponent, String newName)
  {
    super(globalComponent, newName);
  }
  
  
  /**
   * @see XSDVisitor#visitElementDeclaration(XSDElementDeclaration)
   */
  public void visitElementDeclaration(XSDElementDeclaration element)
  {
    if (!element.isElementDeclarationReference() &&
        globalComponent.equals(element.getTypeDefinition()))
    {
      element.getElement().setAttribute(XSDConstants.TYPE_ATTRIBUTE, getNewQName());
    }

    super.visitElementDeclaration(element);
  }


  /**
   * @see org.eclipse.wst.xsd.ui.internal.refactor.XSDVisitor#visitComplexTypeDefinition(XSDComplexTypeDefinition)
   */
  public void visitComplexTypeDefinition(XSDComplexTypeDefinition type)
  {
    super.visitComplexTypeDefinition(type);
    if (type.getAttributeContents() != null)
    {
      for (Iterator iter = type.getAttributeContents().iterator(); iter.hasNext(); )
      {
        XSDAttributeGroupContent attrGroupContent = (XSDAttributeGroupContent) iter.next();
        if (attrGroupContent instanceof XSDAttributeUse)
        {
          XSDAttributeUse attrUse = (XSDAttributeUse) attrGroupContent;
          XSDAttributeDeclaration attrDecl = attrUse.getContent();
          
          // now is this a reference?
          if (!attrDecl.isAttributeDeclarationReference() &&
              globalComponent.equals(attrDecl.getTypeDefinition()))
          {
            attrDecl.getElement().setAttribute(XSDConstants.TYPE_ATTRIBUTE, getNewQName());
          }
        }
      }
    }

    XSDTypeDefinition base = type.getBaseTypeDefinition();
    if (base.equals(globalComponent))
    {
      XSDDOMHelper helper = new XSDDOMHelper();
      Element derivedByNode = helper.getDerivedByElementFromComplexType(type.getElement());
      if (derivedByNode != null)
      {
        derivedByNode.setAttribute(XSDConstants.BASE_ATTRIBUTE, getNewQName());
      }
    }
  }
}