org.eclipse.wst.sse.sieditor.model/api/org/eclipse/wst/sse/sieditor/command/emf/wsdl/AddOperationCommand.java

/*******************************************************************************
 * Copyright (c) 2010 SAP AG.
 * 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:
 *    Emil Simeonov - initial API and implementation.
 *    Dimitar Donchev - initial API and implementation.
 *    Dimitar Tenev - initial API and implementation.
 *    Nevena Manova - initial API and implementation.
 *    Georgi Konstantinov - initial API and implementation.
 *    Keshav Veerapaneni - initial API and implementation.
 *******************************************************************************/
package org.eclipse.wst.sse.sieditor.command.emf.wsdl;

import static org.eclipse.wst.sse.sieditor.model.utils.EmfWsdlUtils.getWSDLFactory;
import static org.eclipse.wst.sse.sieditor.model.utils.EmfWsdlUtils.makeMessageName;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.wst.wsdl.Definition;
import org.eclipse.wst.wsdl.Input;
import org.eclipse.wst.wsdl.Operation;
import org.eclipse.wst.wsdl.Output;
import org.eclipse.wst.wsdl.PortType;

import org.eclipse.wst.sse.sieditor.command.common.AbstractWSDLNotificationOperation;
import org.eclipse.wst.sse.sieditor.command.emf.xsd.AddStructureTypeCommand;
import org.eclipse.wst.sse.sieditor.model.api.IWsdlModelRoot;
import org.eclipse.wst.sse.sieditor.model.api.IXSDModelRoot;
import org.eclipse.wst.sse.sieditor.model.i18n.Messages;
import org.eclipse.wst.sse.sieditor.model.utils.EmfWsdlUtils;
import org.eclipse.wst.sse.sieditor.model.utils.NameGenerator;
import org.eclipse.wst.sse.sieditor.model.utils.StatusUtils;
import org.eclipse.wst.sse.sieditor.model.wsdl.api.IServiceInterface;
import org.eclipse.wst.sse.sieditor.model.wsdl.api.OperationType;
import org.eclipse.wst.sse.sieditor.model.wsdl.impl.Description;
import org.eclipse.wst.sse.sieditor.model.wsdl.impl.OperationParameter;
import org.eclipse.wst.sse.sieditor.model.wsdl.impl.ServiceInterface;
import org.eclipse.wst.sse.sieditor.model.wsdl.impl.ServiceOperation;
import org.eclipse.wst.sse.sieditor.model.xsd.api.ISchema;
import org.eclipse.wst.sse.sieditor.model.xsd.impl.AbstractType;
import org.eclipse.wst.sse.sieditor.model.xsd.impl.Schema;

/**
 * Command for adding a new operaiton
 * 
 * 
 * 
 */
public class AddOperationCommand extends AbstractWSDLNotificationOperation {
    private String _name;
    private ServiceOperation _modelOperation;
    private OperationType _operationType;
    private AddMessageCommand addInputMessageCommand;
    private SetParameterTypeCommand setInParameterTypeCommand;
    private SetParameterTypeCommand setOutParameterTypeCommand;

    public AddOperationCommand(IWsdlModelRoot root, IServiceInterface object, String name, OperationType operationType) {
        super(root, object, Messages.AddOperationCommand_add_new_operation_command_label);
        this._name = name;
        this._operationType = operationType;
    }

    @Override
    public boolean canExecute() {
        return !(getModelRoot() == null || modelObject == null || _name == null)
                && ((ServiceInterface) modelObject).getOperation(_name).isEmpty();
    }

    @Override
    public IStatus run(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
        final ServiceInterface intf = (ServiceInterface) modelObject;
        final Operation newOperation = getWSDLFactory().createOperation();
        newOperation.setName(_name);
        newOperation
                .setStyle(_operationType == null || _operationType == OperationType.REQUEST_RESPONSE ? javax.wsdl.OperationType.REQUEST_RESPONSE
                        : javax.wsdl.OperationType.ONE_WAY);

        final Description description = (Description) getModelRoot().getDescription();
        final Definition definition = description.getComponent();

        final PortType portType = intf.getComponent();
        portType.addOperation(newOperation);
        newOperation.setEnclosingDefinition(definition);

        // Add Input
        AddInputCommand addInputCommand = new AddInputCommand(getModelRoot(), newOperation, null, _operationType);
        IStatus status = getModelRoot().getEnv().execute(addInputCommand);
        if (!StatusUtils.canContinue(status)) {
            return status;
        }

        final Input input = newOperation.getEInput();

        // Create a new message
        String msgName = makeMessageName(newOperation.getName(), ServiceOperation.OPERATION_INMSG_SUFFIX, definition);
        addInputMessageCommand = new AddMessageCommand(description, input, msgName, null);
        status = getModelRoot().getEnv().execute(addInputMessageCommand);
        if (!StatusUtils.canContinue(status)) {
            return status;
        }

        input.getEMessage();

        if (null == _operationType || _operationType == OperationType.REQUEST_RESPONSE) {
            // Create Output
            final AddOutputCommand addOutputCommand = new AddOutputCommand(getModelRoot(), newOperation, null, _operationType);
            status = getModelRoot().getEnv().execute(addOutputCommand);
            if (!StatusUtils.canContinue(status)) {
                return status;
            }

            final Output output = newOperation.getEOutput();

            // Create a new message
            msgName = makeMessageName(newOperation.getName(), ServiceOperation.OPERATION_OUTMSG_SUFFIX, definition);
            AddMessageCommand addOutputMessageCommand = new AddMessageCommand(description, output, msgName, null);
            status = getModelRoot().getEnv().execute(addOutputMessageCommand);
            if (!StatusUtils.canContinue(status)) {
                return status;
            }

            output.getEMessage();
        }
        _modelOperation = new ServiceOperation(getModelRoot(), newOperation, intf);

        ISchema schema = EmfWsdlUtils.getDefaultSchema(description);
        if (schema == null) { // Create new schema with target namespace
            AddNewSchemaCommand addNewSchemaCommand = new AddNewSchemaCommand(getModelRoot(), definition.getTargetNamespace());
            status = getModelRoot().getEnv().execute(addNewSchemaCommand);
            if (!StatusUtils.canContinue(status)) {
                return status;
            }
            schema = addNewSchemaCommand.getNewSchema();
        }

        AddStructureTypeCommand addGlobalElementCommand = new AddStructureTypeCommand((IXSDModelRoot) schema.getModelRoot(),
                schema, NameGenerator.getNewElementDefaultName(schema), true, (AbstractType) Schema.getDefaultSimpleType());
        status = getModelRoot().getEnv().execute(addGlobalElementCommand);
        if (!StatusUtils.canContinue(status)) {
            return status;
        }

        OperationParameter parameter = (OperationParameter) _modelOperation.getAllInputParameters().iterator().next();
        setInParameterTypeCommand = new SetParameterTypeCommand(parameter, addGlobalElementCommand
                .getStructureType());
        getModelRoot().getEnv().execute(setInParameterTypeCommand);
        if (!StatusUtils.canContinue(status)) {
            return status;
        }

        if (_operationType == OperationType.REQUEST_RESPONSE) {
            addGlobalElementCommand = new AddStructureTypeCommand((IXSDModelRoot) schema.getModelRoot(), schema, NameGenerator
                    .getNewElementDefaultName(schema), true, (AbstractType) Schema.getDefaultSimpleType());
            status = getModelRoot().getEnv().execute(addGlobalElementCommand);
            if (!StatusUtils.canContinue(status)) {
                return status;
            }

            parameter = (OperationParameter) _modelOperation.getAllOutputParameters().iterator().next();
            setOutParameterTypeCommand = new SetParameterTypeCommand(parameter, addGlobalElementCommand
                    .getStructureType());
            getModelRoot().getEnv().execute(setOutParameterTypeCommand);
            if (!StatusUtils.canContinue(status)) {
                return status;
            }
        }

        return Status.OK_STATUS;
    }

    public ServiceOperation getOperation() {
        return _modelOperation;
    }

}