XSDNewGroupFromTypeAction.java

// ============================================================================
//
// Copyright (C) 2006-2016 Talend Inc. - www.talend.com
//
// This source code is available under agreement available at
// %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
//
// You should have received a copy of the agreement
// along with this program; if not, write to Talend SA
// 9 rue Pages 92150 Suresnes, France
//
// ============================================================================
package com.amalto.workbench.actions;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.jface.window.Window;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.events.SelectionListener;
import org.eclipse.swt.widgets.Event;
import org.eclipse.xsd.XSDComplexTypeDefinition;
import org.eclipse.xsd.XSDCompositor;
import org.eclipse.xsd.XSDElementDeclaration;
import org.eclipse.xsd.XSDFactory;
import org.eclipse.xsd.XSDModelGroup;
import org.eclipse.xsd.XSDParticle;
import org.eclipse.xsd.util.XSDSchemaBuildingTools;

import com.amalto.workbench.dialogs.NewGroupDialog;
import com.amalto.workbench.editors.DataModelMainPage;
import com.amalto.workbench.i18n.Messages;
import com.amalto.workbench.image.ImageCache;

public class XSDNewGroupFromTypeAction extends UndoAction implements SelectionListener {

    private static Log log = LogFactory.getLog(XSDNewGroupFromTypeAction.class);

    private NewGroupDialog dialog = null;

    private XSDModelGroup group = null;

    private boolean isChoice;

    private boolean isAll;

    private int minOccurs;

    private int maxOccurs;

    public XSDNewGroupFromTypeAction(DataModelMainPage page) {
        super(page);
        setImageDescriptor(ImageCache.getImage("icons/add_obj.gif"));//$NON-NLS-1$
        setText(Messages.XSDNewGroupFromTypeAction_AddGroup);
        setToolTipText(Messages.XSDNewGroupFromTypeAction_AddGroupAtTopXX);
    }

    public IStatus doAction() {
        try {

            IStructuredSelection selection = (IStructuredSelection) page.getTreeViewer().getSelection();
            if (selection.getFirstElement() instanceof XSDComplexTypeDefinition) {
                XSDComplexTypeDefinition ctd = (XSDComplexTypeDefinition) selection.getFirstElement();
                if (!(ctd.getContent() instanceof XSDParticle))
                    return Status.CANCEL_STATUS;
                if (!(((XSDParticle) ctd.getContent()).getTerm() instanceof XSDModelGroup))
                    return Status.CANCEL_STATUS;
                group = (XSDModelGroup) ((XSDParticle) ctd.getContent()).getTerm();
            } else if (selection.getFirstElement() instanceof XSDParticle) {
                group = (XSDModelGroup) ((XSDParticle) selection.getFirstElement()).getTerm();
            } else {
                return Status.CANCEL_STATUS;
            }

            dialog = new NewGroupDialog(this, page.getSite().getShell());
            dialog.setBlockOnOpen(true);
            int ret = dialog.open();
            if (ret == Window.CANCEL) {
                return Status.CANCEL_STATUS;
            }

            XSDFactory factory = XSDSchemaBuildingTools.getXSDFactory();

            // add an element declaration
            XSDElementDeclaration subElement = factory.createXSDElementDeclaration();
            subElement.setName("subelement");//$NON-NLS-1$
            subElement.setTypeDefinition(schema.resolveSimpleTypeDefinition(schema.getSchemaForSchemaNamespace(), "string"));//$NON-NLS-1$

            XSDParticle subParticle = factory.createXSDParticle();
            subParticle.setMinOccurs(1);
            subParticle.setMaxOccurs(1);
            subParticle.setContent(subElement);
            subParticle.updateElement();

            XSDModelGroup newGroup = factory.createXSDModelGroup();
            if (isChoice)
                newGroup.setCompositor(XSDCompositor.CHOICE_LITERAL);
            else if (isAll)
                newGroup.setCompositor(XSDCompositor.ALL_LITERAL);
            else
                newGroup.setCompositor(XSDCompositor.SEQUENCE_LITERAL);
            newGroup.getContents().add(0, subParticle);
            newGroup.updateElement();

            XSDParticle particle = factory.createXSDParticle();
            particle.setContent(newGroup);
            particle.setMinOccurs(this.minOccurs);
            particle.setMaxOccurs(this.maxOccurs);

            group.getContents().add(0, particle);
            group.updateElement();

            page.refresh();
            page.getTreeViewer().setSelection(new StructuredSelection(particle), true);
            page.markDirty();

        } catch (Exception e) {
            log.error(e.getMessage(), e);
            MessageDialog.openError(page.getSite().getShell(), Messages._Error,
                    Messages.bind(Messages.XSDNewGroupFromTypeAction_ErrorMsg, e.getLocalizedMessage()));
            return Status.CANCEL_STATUS;
        }

        return Status.OK_STATUS;
    }

    public void runWithEvent(Event event) {
        super.runWithEvent(event);
    }

    /********************************
     * Listener to input dialog
     */
    public void widgetDefaultSelected(SelectionEvent e) {
    }

    public void widgetSelected(SelectionEvent e) {
        if (dialog.getReturnCode() == -1)
            return; // there was a validation error
        isChoice = dialog.isChoice();
        isAll = dialog.isAll();
        minOccurs = dialog.getMinOccurs();
        maxOccurs = dialog.getMaxOccurs();
        dialog.close();
    }

}