bundles/org.eclipse.wst.xsd.ui/src/org/eclipse/wst/xsd/ui/internal/actions/CreateElementAction.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.actions;
import java.util.List;

import org.eclipse.gef.GraphicalEditPart;
import org.eclipse.gef.ui.parts.AbstractEditPartViewer;
import org.eclipse.jface.action.Action;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.jface.viewers.ISelectionProvider;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.swt.widgets.Display;
import org.eclipse.wst.sse.core.internal.format.IStructuredFormatProcessor;
import org.eclipse.wst.xml.core.internal.document.DocumentImpl;
import org.eclipse.wst.xml.core.internal.provisional.document.IDOMModel;
import org.eclipse.wst.xml.core.internal.provisional.document.IDOMNode;
import org.eclipse.wst.xml.core.internal.provisional.format.FormatProcessorXML;
import org.eclipse.wst.xsd.ui.internal.graph.editparts.ComplexTypeDefinitionEditPart;
import org.eclipse.wst.xsd.ui.internal.graph.editparts.ElementDeclarationEditPart;
import org.eclipse.wst.xsd.ui.internal.graph.editparts.ModelGroupDefinitionEditPart;
import org.eclipse.wst.xsd.ui.internal.graph.editparts.TopLevelComponentEditPart;
import org.eclipse.wst.xsd.ui.internal.util.XSDDOMHelper;
import org.eclipse.xsd.XSDAttributeDeclaration;
import org.eclipse.xsd.XSDAttributeGroupDefinition;
import org.eclipse.xsd.XSDAttributeUse;
import org.eclipse.xsd.XSDComplexTypeDefinition;
import org.eclipse.xsd.XSDConcreteComponent;
import org.eclipse.xsd.XSDElementDeclaration;
import org.eclipse.xsd.XSDModelGroupDefinition;
import org.eclipse.xsd.XSDNamedComponent;
import org.eclipse.xsd.XSDSchema;
import org.eclipse.xsd.util.XSDConstants;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.Text;

public class CreateElementAction extends Action
{
  protected String description;
  protected Element parentNode;

  protected ISelectionProvider selectionProvider;
  protected XSDSchema xsdSchema;

  protected Object sourceContext;

  /**
   * Constructor for CreateElementAction.
   */
  public CreateElementAction()
  {
    super();
  }
  /**
   * Constructor for CreateElementAction.
   * @param text
   */
  public CreateElementAction(String text)
  {
    super(text);
  }
  /**
   * Constructor for CreateElementAction.
   * @param text
   * @param image
   */
  public CreateElementAction(String text, ImageDescriptor image)
  {
    super(text, image);
  }

  public void setXSDSchema(XSDSchema xsdSchema)
  {
    this.xsdSchema = xsdSchema;
  }
  
  public void setSelectionProvider(ISelectionProvider selectionProvider)
  {
    this.selectionProvider = selectionProvider;
  }

  public void setSourceContext(Object sourceContext)
  {
    this.sourceContext = sourceContext;
  }
  
  /**
   * Gets the parentNode.
   * @return Returns a Element
   */
  public Element getParentNode()
  {
    return parentNode;
  }

  /**
   * Sets the parentNode.
   * @param parentNode The parentNode to set
   */
  public void setParentNode(Element parentNode)
  {
    this.parentNode = parentNode;
  }

  boolean isGlobal = false;
  
  public void setIsGlobal(boolean isGlobal)
  {
    this.isGlobal = isGlobal;
  }
  
  public boolean getIsGlobal()
  {
    return isGlobal;
  }

  protected Node relativeNode;
  protected String elementTag;
  public void setElementTag(String elementTag)
  {
    this.elementTag = elementTag;
  }
  
  public DocumentImpl getDocument()
  {
    return (DocumentImpl) getParentNode().getOwnerDocument();
  }
    
  public void beginRecording(String description)
  {
    getDocument().getModel().beginRecording(this, description);
  }
  
  public void endRecording()
  {
    DocumentImpl doc = (DocumentImpl) getDocument();
    
    doc.getModel().endRecording(this);    
  }
  
  public Element createAndAddNewChildElement()
  {
    String prefix = parentNode.getPrefix();
    prefix = (prefix == null) ? "" : (prefix + ":");
    Element childNode = getDocument().createElementNS(XSDConstants.SCHEMA_FOR_SCHEMA_URI_2001, prefix + elementTag);
    if (getAttributes() != null)
    {
      List attributes = getAttributes();
      for (int i = 0; i < attributes.size(); i++)
      {
        DOMAttribute attr = (DOMAttribute) attributes.get(i);
        childNode.setAttribute(attr.getName(), attr.getValue());
      }
    }
    if (getRelativeNode() == null)
    {
      parentNode.appendChild(childNode);
    }
    else
    {
      ((Element)parentNode).insertBefore(childNode,getRelativeNode());
    }
    
    if (isGlobal && getRelativeNode() == null)
    {
      Text textNode = getDocument().createTextNode("\n\n");
      parentNode.appendChild(textNode);
    }
    else if (isGlobal && getRelativeNode() != null)
    {
      Text textNode = getDocument().createTextNode("\n\n");
      parentNode.insertBefore(textNode, getRelativeNode());
    }

    formatChild(childNode);
    
    return childNode;
  }    
    
  protected void formatChild(Element child)
  {
    if (child instanceof IDOMNode)
    {
      IDOMModel model = ((IDOMNode)child).getModel();
      try
      {
        // tell the model that we are about to make a big model change
        model.aboutToChangeModel();
        
	      IStructuredFormatProcessor formatProcessor = new FormatProcessorXML();
		    formatProcessor.formatNode(child);
      }
      finally
      {
        // tell the model that we are done with the big model change
        model.changedModel(); 
      }
    }
  }
  /*
   * @see IAction#run()
   */
  public void run()
  {
    beginRecording(getDescription());
    final Element child = createAndAddNewChildElement();
    endRecording();

    if (selectionProvider != null)
    {
      final XSDConcreteComponent comp = xsdSchema.getCorrespondingComponent(child);
//      selectionProvider.setSelection(new StructuredSelection(comp));
      
    Runnable runnable = new Runnable()
    {
      public void run()
      {
        if (comp instanceof XSDAttributeDeclaration)
        {
          if (((XSDAttributeDeclaration)comp).getContainer() instanceof XSDAttributeUse)
          {
            if (comp.getContainer().getContainer() instanceof XSDAttributeGroupDefinition)
            {
              selectionProvider.setSelection(new StructuredSelection(comp.getContainer()));
            }
            else if (comp.getContainer().getContainer() instanceof XSDComplexTypeDefinition)
            {
              if (XSDDOMHelper.inputEquals((Element)child, XSDConstants.ATTRIBUTE_ELEMENT_TAG, true))
              {
                selectionProvider.setSelection(new StructuredSelection(comp.getContainer()));
              }
              else
              {
                selectionProvider.setSelection(new StructuredSelection(comp));
              }
            }
            else
            {
              selectionProvider.setSelection(new StructuredSelection(comp));
            }
          }
          else
          {
            selectionProvider.setSelection(new StructuredSelection(comp));
          }
        }
        else
        {
          selectionProvider.setSelection(new StructuredSelection(comp));
        }
        if (comp instanceof XSDNamedComponent)
        {
          if (sourceContext instanceof AbstractEditPartViewer)
          {
            AbstractEditPartViewer viewer = (AbstractEditPartViewer)sourceContext;
          
            Object obj = viewer.getSelectedEditParts().get(0);
            
            if (obj instanceof GraphicalEditPart)
            {
              if (obj instanceof ElementDeclarationEditPart)
              {
                XSDElementDeclaration elem = ((ElementDeclarationEditPart)obj).getXSDElementDeclaration();
                if (!elem.isElementDeclarationReference())
                {
                  ((ElementDeclarationEditPart)obj).doEditName();
                }
              }
              else if (obj instanceof ModelGroupDefinitionEditPart)
              {
                XSDModelGroupDefinition group = ((ModelGroupDefinitionEditPart)obj).getXSDModelGroupDefinition();
                if (!group.isModelGroupDefinitionReference())
                {
                  ((ModelGroupDefinitionEditPart)obj).doEditName();
                }
              }
              else if (obj instanceof ComplexTypeDefinitionEditPart)
              {
                XSDComplexTypeDefinition ct = ((ComplexTypeDefinitionEditPart)obj).getXSDComplexTypeDefinition();
                if (ct.getName() != null)
                {
                  ((ComplexTypeDefinitionEditPart)obj).doEditName();
                }
              }
              else if (obj instanceof TopLevelComponentEditPart)
              {
                ((TopLevelComponentEditPart)obj).doEditName();
              }
            }

          }
        }
      }
    };
    Display.getDefault().timerExec(50,runnable);
    }
  }

  /**
   * Gets the relativeNode.
   * @return Returns a Element
   */
  public Node getRelativeNode()
  {
    return relativeNode;
  }

  /**
   * Sets the relativeNode.
   * @param relativeNode The relativeNode to set
   */
  public void setRelativeNode(Node relativeNode)
  {
    this.relativeNode = relativeNode;
  }

  /**
   * Gets the description.
   * @return Returns a String
   */
  public String getDescription()
  {
    if (description == null)
    {
      return getText();
    }
    return description;
  }

  /**
   * Sets the description.
   * @param description The description to set
   */
  public void setDescription(String description)
  {
    this.description = description;
  }

  protected List attributes;
  /**
   * Gets the nameAttribute.
   * @return Returns a String
   */
  public List getAttributes()
  {
    return attributes;
  }

  /**
   * Sets the attributes.
   * @param attributes The attributes to set
   */
  public void setAttributes(List attributes)
  {
    this.attributes = attributes;
  }
}