| /******************************************************************************* |
| * Copyright (c) 2001, 2008 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 java.util.ArrayList; |
| import java.util.List; |
| |
| import org.eclipse.wst.xsd.ui.internal.common.util.XSDCommonUIUtils; |
| import org.eclipse.xsd.XSDComplexTypeContent; |
| 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.XSDModelGroupDefinition; |
| import org.eclipse.xsd.XSDParticle; |
| import org.eclipse.xsd.XSDSchema; |
| import org.eclipse.xsd.XSDSimpleTypeDefinition; |
| import org.eclipse.xsd.util.XSDSchemaBuildingTools; |
| import org.w3c.dom.Text; |
| |
| public class AddXSDElementCommand extends BaseCommand |
| { |
| XSDComplexTypeDefinition xsdComplexTypeDefinition; |
| XSDModelGroupDefinition xsdModelGroupDefinition; |
| XSDModelGroup xsdModelGroup; |
| XSDSchema xsdSchema; |
| boolean isReference; |
| private String nameToAdd; |
| // The index of the currently selected item. If no item is selected, index will be less than 0 |
| private int index = -1; |
| // Determines where the element should be inserted based on the currently selected element. If no |
| // element is selected, use the default behaviour of appending the element to the end |
| private String addElementLocation; |
| |
| public AddXSDElementCommand() |
| { |
| super(); |
| } |
| |
| public AddXSDElementCommand(String label, XSDComplexTypeDefinition xsdComplexTypeDefinition) |
| { |
| super(label); |
| this.xsdComplexTypeDefinition = xsdComplexTypeDefinition; |
| xsdModelGroup = getModelGroup(xsdComplexTypeDefinition); |
| } |
| |
| public AddXSDElementCommand(String label, XSDModelGroupDefinition xsdModelGroupDefinition) |
| { |
| super(label); |
| this.xsdModelGroupDefinition = xsdModelGroupDefinition; |
| xsdModelGroup = getModelGroup(xsdModelGroupDefinition); |
| } |
| |
| public AddXSDElementCommand(String label, XSDModelGroup xsdModelGroup, XSDComplexTypeDefinition xsdComplexTypeDefinition) |
| { |
| super(label); |
| this.xsdModelGroup = xsdModelGroup; |
| this.xsdComplexTypeDefinition = xsdComplexTypeDefinition; |
| } |
| |
| public AddXSDElementCommand(String label, XSDModelGroup xsdModelGroup) |
| { |
| super(label); |
| this.xsdModelGroup = xsdModelGroup; |
| } |
| |
| public AddXSDElementCommand(String label, XSDModelGroup xsdModelGroup, String ID, int index) |
| { |
| super(label); |
| this.xsdModelGroup = xsdModelGroup; |
| this.index = index; |
| this.addElementLocation = ID; |
| } |
| |
| public AddXSDElementCommand(String label, XSDSchema xsdSchema) |
| { |
| super(label); |
| this.xsdSchema = xsdSchema; |
| } |
| |
| public void setReference(boolean isReference) |
| { |
| this.isReference = isReference; |
| } |
| |
| public void setNameToAdd(String name) |
| { |
| nameToAdd = name; |
| } |
| |
| public void setAddElementLocation(String LocationType) |
| { |
| addElementLocation = LocationType; |
| } |
| |
| protected int getInsertionIndex() |
| { |
| if (index < 0) |
| return -1; |
| |
| if (addElementLocation.equals(org.eclipse.wst.xsd.ui.internal.common.actions.AddXSDElementAction.BEFORE_SELECTED_ID)) |
| { |
| return index; |
| } |
| else if (addElementLocation.equals(org.eclipse.wst.xsd.ui.internal.common.actions.AddXSDElementAction.AFTER_SELECTED_ID)) |
| { |
| index++; |
| return index; |
| } |
| else |
| { |
| return -1; |
| } |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.eclipse.gef.commands.Command#execute() |
| */ |
| public void execute() |
| { |
| try |
| { |
| if (xsdSchema != null) |
| { |
| beginRecording(xsdSchema.getElement()); |
| XSDElementDeclaration element = createGlobalXSDElementDeclaration(); |
| Text textNode = xsdSchema.getDocument().createTextNode("\n"); //$NON-NLS-1$ |
| xsdSchema.getElement().appendChild(textNode); |
| xsdSchema.getContents().add(element); |
| addedXSDConcreteComponent = element; |
| } |
| else if (xsdModelGroupDefinition != null) |
| { |
| beginRecording(xsdModelGroupDefinition.getElement()); |
| if (xsdModelGroup == null) |
| { |
| XSDFactory factory = XSDSchemaBuildingTools.getXSDFactory(); |
| XSDParticle particle = factory.createXSDParticle(); |
| xsdModelGroup = factory.createXSDModelGroup(); |
| xsdModelGroup.setCompositor(XSDCompositor.SEQUENCE_LITERAL); |
| particle.setContent(xsdModelGroup); |
| xsdModelGroupDefinition.setModelGroup(xsdModelGroup); |
| } |
| xsdSchema = xsdModelGroupDefinition.getSchema(); |
| if (!isReference) |
| { |
| xsdModelGroup.getContents().add(createXSDElementDeclarationForModelGroupDefinitions()); |
| } |
| else |
| { |
| xsdModelGroup.getContents().add(createXSDElementReference()); |
| } |
| formatChild(xsdModelGroupDefinition.getElement()); |
| } |
| else if (xsdModelGroup != null && (xsdComplexTypeDefinition == null || xsdModelGroupDefinition == null) ) |
| { |
| xsdSchema = xsdModelGroup.getSchema(); |
| beginRecording(xsdSchema.getElement()); |
| if (!isReference) |
| { |
| index = getInsertionIndex(); |
| if(index >= 0 && index < xsdModelGroup.getContents().size()) |
| { |
| xsdModelGroup.getContents().add(index,createXSDElementDeclaration()); |
| } |
| else |
| { |
| xsdModelGroup.getContents().add(createXSDElementDeclaration()); |
| } |
| } |
| else |
| { |
| xsdModelGroup.getContents().add(createXSDElementReference()); |
| } |
| formatChild(xsdModelGroup.getElement()); |
| } |
| else |
| { |
| xsdSchema = xsdComplexTypeDefinition.getSchema(); |
| beginRecording(xsdSchema.getElement()); |
| if (xsdModelGroup == null) |
| { |
| XSDFactory factory = XSDSchemaBuildingTools.getXSDFactory(); |
| XSDParticle particle = factory.createXSDParticle(); |
| xsdModelGroup = factory.createXSDModelGroup(); |
| xsdModelGroup.setCompositor(XSDCompositor.SEQUENCE_LITERAL); |
| particle.setContent(xsdModelGroup); |
| xsdComplexTypeDefinition.setContent(particle); |
| } |
| if (!isReference) |
| { |
| xsdModelGroup.getContents().add(createXSDElementDeclarationForComplexType()); |
| } |
| else |
| { |
| xsdModelGroup.getContents().add(createXSDElementReference()); |
| } |
| formatChild(xsdComplexTypeDefinition.getElement()); |
| } |
| } |
| finally |
| { |
| endRecording(); |
| } |
| } |
| |
| protected XSDParticle createXSDElementDeclaration() |
| { |
| XSDSimpleTypeDefinition type = xsdModelGroup.getSchema().getSchemaForSchema().resolveSimpleTypeDefinition("string"); //$NON-NLS-1$ |
| |
| XSDElementDeclaration element = XSDFactory.eINSTANCE.createXSDElementDeclaration(); |
| |
| XSDConcreteComponent comp = xsdModelGroup.getContainer(); |
| ArrayList usedAttributeNames = new ArrayList(); |
| XSDCommonUIUtils.resetVisitedGroupsStack(); |
| usedAttributeNames.addAll(XSDCommonUIUtils.getChildElements(xsdModelGroup)); |
| while (comp != null) |
| { |
| if (comp instanceof XSDModelGroupDefinition) |
| { |
| usedAttributeNames.addAll(XSDCommonUIUtils.getAllAttributes((XSDModelGroupDefinition)comp)); |
| break; |
| } |
| else if (comp instanceof XSDComplexTypeDefinition) |
| { |
| usedAttributeNames.addAll(XSDCommonUIUtils.getAllAttributes((XSDComplexTypeDefinition)comp)); |
| usedAttributeNames.addAll(XSDCommonUIUtils.getInheritedAttributes((XSDComplexTypeDefinition)comp)); |
| break; |
| } |
| comp = comp.getContainer(); |
| } |
| element.setName(XSDCommonUIUtils.createUniqueElementName( |
| nameToAdd == null ? "NewElement" : nameToAdd , usedAttributeNames)); //$NON-NLS-1$ |
| element.setTypeDefinition(type); |
| |
| XSDParticle particle = XSDFactory.eINSTANCE.createXSDParticle(); |
| particle.setContent(element); |
| addedXSDConcreteComponent = element; |
| return particle; |
| } |
| |
| protected XSDParticle createXSDElementReference() |
| { |
| List list = xsdModelGroup.getSchema().getElementDeclarations(); |
| XSDElementDeclaration referencedElement = null; |
| if (list.size() > 0) |
| { |
| referencedElement = (XSDElementDeclaration)list.get(0); |
| } |
| else |
| { |
| referencedElement = createGlobalXSDElementDeclaration(); |
| Text textNode = xsdSchema.getDocument().createTextNode("\n"); //$NON-NLS-1$ |
| xsdSchema.getElement().appendChild(textNode); |
| xsdSchema.getContents().add(referencedElement); |
| } |
| |
| XSDElementDeclaration element = XSDFactory.eINSTANCE.createXSDElementDeclaration(); |
| |
| element.setResolvedElementDeclaration(referencedElement); |
| XSDParticle particle = XSDFactory.eINSTANCE.createXSDParticle(); |
| particle.setContent(element); |
| addedXSDConcreteComponent = element; |
| return particle; |
| } |
| |
| protected XSDParticle createXSDElementDeclarationForComplexType() |
| { |
| XSDSimpleTypeDefinition type = xsdModelGroup.getSchema().getSchemaForSchema().resolveSimpleTypeDefinition("string"); //$NON-NLS-1$ |
| |
| XSDElementDeclaration element = XSDFactory.eINSTANCE.createXSDElementDeclaration(); |
| |
| ArrayList usedAttributeNames = new ArrayList(); |
| usedAttributeNames.addAll(XSDCommonUIUtils.getAllAttributes(xsdComplexTypeDefinition)); |
| usedAttributeNames.addAll(XSDCommonUIUtils.getInheritedAttributes(xsdComplexTypeDefinition)); |
| element.setName(XSDCommonUIUtils.createUniqueElementName( |
| nameToAdd == null ? "NewElement" : nameToAdd , usedAttributeNames)); //$NON-NLS-1$ |
| element.setTypeDefinition(type); |
| |
| XSDParticle particle = XSDFactory.eINSTANCE.createXSDParticle(); |
| particle.setContent(element); |
| addedXSDConcreteComponent = element; |
| return particle; |
| } |
| |
| protected XSDParticle createXSDElementDeclarationForModelGroupDefinitions() |
| { |
| XSDSimpleTypeDefinition type = xsdModelGroup.getSchema().getSchemaForSchema().resolveSimpleTypeDefinition("string"); //$NON-NLS-1$ |
| |
| XSDElementDeclaration element = XSDFactory.eINSTANCE.createXSDElementDeclaration(); |
| |
| ArrayList usedAttributeNames = new ArrayList(); |
| usedAttributeNames.addAll(XSDCommonUIUtils.getAllAttributes(xsdModelGroupDefinition)); |
| element.setName(XSDCommonUIUtils.createUniqueElementName( |
| nameToAdd == null ? "NewElement" : nameToAdd , usedAttributeNames)); //$NON-NLS-1$ |
| element.setTypeDefinition(type); |
| |
| XSDParticle particle = XSDFactory.eINSTANCE.createXSDParticle(); |
| particle.setContent(element); |
| addedXSDConcreteComponent = element; |
| return particle; |
| } |
| |
| protected XSDElementDeclaration createGlobalXSDElementDeclaration() |
| { |
| ensureSchemaElement(xsdSchema); |
| XSDSimpleTypeDefinition type = xsdSchema.getSchemaForSchema().resolveSimpleTypeDefinition("string"); //$NON-NLS-1$ |
| XSDFactory factory = XSDSchemaBuildingTools.getXSDFactory(); |
| XSDElementDeclaration element = factory.createXSDElementDeclaration(); |
| |
| element.setName(XSDCommonUIUtils.createUniqueElementName( |
| nameToAdd == null ? "NewElement" : nameToAdd , xsdSchema.getElementDeclarations())); //$NON-NLS-1$ |
| element.setTypeDefinition(type); |
| |
| return element; |
| } |
| |
| public XSDModelGroup getModelGroup(XSDModelGroupDefinition modelGroupDef) |
| { |
| return modelGroupDef.getModelGroup(); |
| } |
| |
| //PORT |
| public XSDModelGroup getModelGroup(XSDComplexTypeDefinition cType) |
| { |
| XSDParticle particle = null; |
| |
| XSDComplexTypeContent xsdComplexTypeContent = cType.getContent(); |
| if (xsdComplexTypeContent instanceof XSDParticle) |
| { |
| particle = (XSDParticle)xsdComplexTypeContent; |
| } |
| |
| if (particle == null) |
| { |
| return null; |
| } |
| |
| Object particleContent = particle.getContent(); |
| XSDModelGroup group = null; |
| |
| if (particleContent instanceof XSDModelGroupDefinition) |
| { |
| group = ((XSDModelGroupDefinition) particleContent).getResolvedModelGroupDefinition().getModelGroup(); |
| } |
| else if (particleContent instanceof XSDModelGroup) |
| { |
| group = (XSDModelGroup) particleContent; |
| } |
| |
| if (group == null) |
| { |
| return null; |
| } |
| |
| // if (group.getContents().isEmpty() || group.eResource() != cType.eResource()) |
| // { |
| // if (cType.getBaseType() != null) |
| // { |
| // XSDComplexTypeContent content = cType.getContent(); |
| // if (content instanceof XSDParticle) |
| // { |
| // group = (XSDModelGroup) ((XSDParticle) content).getContent(); |
| // } |
| // } |
| // } |
| |
| return group; |
| } |
| |
| public XSDConcreteComponent getAddedComponent() |
| { |
| return super.getAddedComponent(); |
| } |
| |
| } |