bundles/org.eclipse.wst.wsdl.ui/src-asd-wsdl11/org/eclipse/wst/wsdl/ui/internal/adapters/commands/W11AddSchemaCommand.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.wsdl.ui.internal.adapters.commands;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.wst.wsdl.Definition;
import org.eclipse.wst.wsdl.Types;
import org.eclipse.wst.wsdl.XSDSchemaExtensibilityElement;
import org.eclipse.wst.wsdl.ui.internal.asd.Messages;
import org.eclipse.wst.wsdl.ui.internal.asd.actions.IASDAddCommand;
import org.eclipse.wst.wsdl.ui.internal.commands.AddXSDSchemaCommand;
import org.eclipse.wst.wsdl.ui.internal.util.NameUtil;

public class W11AddSchemaCommand extends W11TopLevelElementCommand implements IASDAddCommand {
	private Object schema;
	
	public W11AddSchemaCommand(Definition definition) {
    super(Messages._UI_ACTION_ADD_SCHEMA, definition);
	}
	
	public void execute() {
		try {
			beginRecording(definition.getElement());
			super.execute();
    
			String tns = definition.getTargetNamespace();
			List existingNamespaces = new ArrayList();
			Types types = definition.getETypes();
	
			if (types != null) {
				Iterator eeIt = types.getEExtensibilityElements().iterator();
				while (eeIt.hasNext()) {
					Object item = eeIt.next();
					if (item instanceof XSDSchemaExtensibilityElement) {
						XSDSchemaExtensibilityElement eeElement = (XSDSchemaExtensibilityElement) item;
						if (eeElement.getSchema() != null) {
							String ns = eeElement.getSchema().getTargetNamespace();
							existingNamespaces.add(ns);
						}
					}
				} 
			}
			tns = NameUtil.getUniqueNameHelper(tns, existingNamespaces);
			
			AddXSDSchemaCommand command = new AddXSDSchemaCommand(definition, tns);
			command.run();
			formatChild(command.getWSDLElement().getElement());
			schema = command.getWSDLElement();
		}
		finally {
			endRecording(definition.getElement());
		}
	}
	
	public Object getNewlyAddedComponent() {
		return schema;
	}
}