bundles/org.eclipse.wst.xsd.ui/src-common/org/eclipse/wst/xsd/ui/internal/common/commands/AddXSDModelGroupCommand.java

/*******************************************************************************
 * Copyright (c) 2001, 2006 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.common.commands;

import org.eclipse.xsd.XSDComplexTypeDefinition;
import org.eclipse.xsd.XSDCompositor;
import org.eclipse.xsd.XSDConcreteComponent;
import org.eclipse.xsd.XSDElementDeclaration;
import org.eclipse.xsd.XSDFactory;
import org.eclipse.xsd.XSDModelGroup;
import org.eclipse.xsd.XSDParticle;
import org.eclipse.xsd.XSDSimpleTypeDefinition;
import org.eclipse.xsd.util.XSDSchemaBuildingTools;

public class AddXSDModelGroupCommand extends BaseCommand
{
  XSDConcreteComponent parent;
  XSDCompositor xsdCompositor;
  XSDModelGroup newModelGroup;

  public AddXSDModelGroupCommand(String label, XSDConcreteComponent parent, XSDCompositor xsdCompositor)
  {
    super(label);
    this.parent = parent;
    this.xsdCompositor = xsdCompositor;
  }

  public void execute()
  {
    try
    {
      beginRecording(parent.getElement());
      XSDConcreteComponent owner = getOwner();
      if (owner != null)
      {
        XSDParticle particle = XSDFactory.eINSTANCE.createXSDParticle();
        newModelGroup = createModelGroup();
        particle.setContent(newModelGroup);

        XSDComplexTypeDefinition ctd = (XSDComplexTypeDefinition) owner;
        ctd.setContent(particle);
      }
      formatChild(parent.getElement());
    }
    finally
    {
      endRecording();
    }
  }
  
  public void undo()
  {
    super.undo();
    
    if (parent instanceof XSDModelGroup)
    {
      XSDModelGroup model = (XSDModelGroup) parent;
      model.getContents().remove(newModelGroup.getContainer());
    }
  }
  
  private XSDConcreteComponent getOwner()
  {
    XSDConcreteComponent owner = null;
    if (parent instanceof XSDElementDeclaration)
    {
      XSDElementDeclaration ed = (XSDElementDeclaration)parent;      
      if (ed.getTypeDefinition() != null) 
      {
        if (ed.getAnonymousTypeDefinition() == null)
        {
          ed.setTypeDefinition(null);
          XSDComplexTypeDefinition td = XSDFactory.eINSTANCE.createXSDComplexTypeDefinition();
          ed.setAnonymousTypeDefinition(td);
          owner = ed.getTypeDefinition();
        }
        else
        {
          XSDComplexTypeDefinition td = XSDFactory.eINSTANCE.createXSDComplexTypeDefinition();
          ed.setAnonymousTypeDefinition(td);
          owner = td;        
        }
      }        
      else if (ed.getAnonymousTypeDefinition() == null)
      {
        XSDComplexTypeDefinition td = XSDFactory.eINSTANCE.createXSDComplexTypeDefinition();
        ed.setAnonymousTypeDefinition(td);
        owner = td;        
      }
      else if (ed.getAnonymousTypeDefinition() instanceof XSDComplexTypeDefinition)
      {
        owner = ed.getAnonymousTypeDefinition();
      }
      else if (ed.getAnonymousTypeDefinition() instanceof XSDSimpleTypeDefinition)
      {
        XSDComplexTypeDefinition td = XSDFactory.eINSTANCE.createXSDComplexTypeDefinition();
        ed.setAnonymousTypeDefinition(td);
        owner = td;        
      }
    }
    else if (parent instanceof XSDModelGroup)
    {
      newModelGroup = createModelGroup();
      ((XSDModelGroup) parent).getContents().add(newModelGroup.getContainer());
    }
    else if (parent instanceof XSDComplexTypeDefinition)
    {
      XSDComplexTypeDefinition ct = (XSDComplexTypeDefinition)parent;
      owner = parent;
      if (ct.getContent() instanceof XSDParticle)
      {
        XSDParticle particle = (XSDParticle)ct.getContent();
        if (particle.getContent() instanceof XSDModelGroup)
        {
          owner = null;
          newModelGroup = createModelGroup();
          XSDModelGroup newParent = (XSDModelGroup)particle.getContent();
          newParent.getContents().add(newModelGroup.getContainer());
        }
        
      }
    }
    return owner;
  }
  

  protected boolean adopt(XSDConcreteComponent model)
  {
    return false;
  }
  
  protected XSDModelGroup createModelGroup()
  {
    
    XSDFactory factory = XSDSchemaBuildingTools.getXSDFactory();
    XSDParticle particle = factory.createXSDParticle();
    XSDModelGroup modelGroup = factory.createXSDModelGroup();
    modelGroup.setCompositor(xsdCompositor);
    particle.setContent(modelGroup);
    addedXSDConcreteComponent = modelGroup;
    return modelGroup;
  }
}