com.bluexml.side.form.clazz.AddRefAction.java Source code

Introduction

Here is the source code for com.bluexml.side.form.clazz.AddRefAction.java

Source

/*
Copyright (C) 2007-2011  BlueXML - www.bluexml.com
    
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
    
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
    
You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
    
*/

package com.bluexml.side.form.clazz;

import java.util.Iterator;
import java.util.Random;

import org.eclipse.emf.common.command.Command;
import org.eclipse.emf.common.command.CompoundCommand;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.edit.command.AddCommand;
import org.eclipse.emf.edit.domain.EditingDomain;
import org.eclipse.emf.edit.domain.IEditingDomainProvider;
import org.eclipse.jface.action.Action;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.ui.IWorkbenchPart;

import com.bluexml.side.clazz.Clazz;
import com.bluexml.side.form.FormClass;
import com.bluexml.side.form.FormFactory;
import com.bluexml.side.form.FormPackage;
import com.bluexml.side.form.Reference;
import com.bluexml.side.form.clazz.utils.ClassInitialization;
import com.bluexml.side.form.common.utils.FormDiagramUtils;

public class AddRefAction extends Action implements ISelectionChangedListener {

    protected EObject selectedObject;
    private EditingDomain domain;
    private Clazz c;
    private Reference ref;

    public AddRefAction(Clazz p_c, Reference p_ref) {
        super();
        c = p_c;
        ref = p_ref;
    }

    public void selectionChanged(SelectionChangedEvent event) {
        if (event.getSelection() instanceof IStructuredSelection) {
            setEnabled(updateSelection((IStructuredSelection) event.getSelection()));
        } else {
            setEnabled(false);
        }
    }

    public boolean updateSelection(IStructuredSelection selection) {
        selectedObject = null;
        for (Iterator<?> objects = selection.iterator(); objects.hasNext();) {
            Object object = objects.next();
            if (object instanceof Reference) {
                selectedObject = (EObject) object;
            } else {
                return false;
            }
        }

        return selectedObject != null;
    }

    @Override
    public void run() {
        super.run();
        doAction();
    }

    @SuppressWarnings("unchecked")
    private void doAction() {
        if (c != null && ref != null) {
            //FormContainer form =  FormFactory.eINSTANCE.createFormClass();

            FormClass formClass = FormFactory.eINSTANCE.createFormClass();
            //form.setRoot(formClass);

            formClass.setReal_class(c);
            Random random = new Random();
            int pick = random.nextInt(Integer.MAX_VALUE);
            formClass.setId(c.getName() + "_" + pick);
            if (c.getTitle() != null && c.getTitle().length() > 0) {
                formClass.setLabel(c.getTitle());
                //form.setName(c.getTitle() + " ref from " + ((FormClass)ref.eContainer()).getLabel() + " (" + ref.getLabel() + ")");
            } else {
                formClass.setLabel(c.getName());
                //form.setName(c.getName() + " ref from " + ((FormClass)ref.eContainer()).getLabel() + " (" + ref.getLabel() + ")");
            }
            ;
            Command setCmd = AddCommand.create(domain, ref, FormPackage.eINSTANCE.getModelChoiceField_Target(),
                    formClass);
            //ref.getTarget().add(formClass);
            // Commands :
            // Add the Form
            Command addFormCmd = AddCommand.create(domain, FormDiagramUtils.getParentFormCollection(ref),
                    FormPackage.eINSTANCE.getFormCollection_Forms(), formClass);
            // Add the reference
            CompoundCommand cc = new CompoundCommand();
            cc.append(addFormCmd);
            cc.append(setCmd);
            cc.append(ClassInitialization.initializeClass(formClass, domain));
            domain.getCommandStack().execute(cc);
        }
    }

    @Override
    public String getText() {
        String label = ((c.getTitle() != null && c.getTitle().length() > 0) ? c.getTitle() : c.getName());
        if (c.isAbstract()) {
            label += " (Abstract Type)";
        }
        return label;
    }

    public void setActiveWorkbenchPart(IWorkbenchPart workbenchPart) {
        if (workbenchPart instanceof IEditingDomainProvider) {
            domain = ((IEditingDomainProvider) workbenchPart).getEditingDomain();
        }
    }
}