com.bluexml.side.view.edit.ui.actions.AddLinkedFieldAction.java Source code

Introduction

Here is the source code for com.bluexml.side.view.edit.ui.actions.AddLinkedFieldAction.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.view.edit.ui.actions;

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

import org.eclipse.emf.common.command.Command;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.plugin.EcorePlugin;
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.Association;
import com.bluexml.side.clazz.Attribute;
import com.bluexml.side.view.AbstractView;
import com.bluexml.side.view.Col;
import com.bluexml.side.view.DataTable;
import com.bluexml.side.view.Field;
import com.bluexml.side.view.FieldElement;
import com.bluexml.side.view.ViewFactory;
import com.bluexml.side.view.ViewPackage;
import com.bluexml.side.view.edit.ui.utils.InternalModification;
import com.bluexml.side.view.edit.ui.utils.model.ClassUtils;

public class AddLinkedFieldAction extends Action implements ISelectionChangedListener {
    protected EObject toAdd;
    protected AbstractView parent;
    private EditingDomain domain;
    protected List<Association> path;

    public AddLinkedFieldAction(EObject a, List<Association> path, AbstractView parent) {
        super();
        toAdd = a;
        this.path = path;
        this.parent = parent;
    }

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

    public boolean updateSelection(IStructuredSelection selection) {

        for (Iterator<?> objects = selection.iterator(); objects.hasNext();) {
            Object object = objects.next();
            if (object instanceof Attribute) {
                toAdd = (Attribute) object;
            } else {
                return false;
            }
        }

        return toAdd != null;
    }

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

    private void doAction() {
        try {
            if (toAdd instanceof Attribute) {
                FieldElement child = null;
                Field f = ClassUtils.getFieldForAttribute((Attribute) toAdd);
                f.getPath().addAll(path);

                if (parent instanceof DataTable) {
                    Col col = ViewFactory.eINSTANCE.createCol();
                    col.getChildren().add(f);
                    col.setName(f.getName());
                    child = col;
                } else {
                    child = f;
                }
                Command cmd = AddCommand.create(domain, parent, ViewPackage.eINSTANCE.getFieldContainer_Children(),
                        child);
                domain.getCommandStack().execute(cmd);
            } else {
                EcorePlugin.INSTANCE.log("Error : Add Linked Field launch on non attribute item.");
            }
        } catch (Exception e) {
            e.printStackTrace();
            EcorePlugin.INSTANCE.log("Get Linked Field : " + e.getMessage());
            e.printStackTrace();
            InternalModification.moveToDisabled();
        }
    }

    @Override
    public String getText() {
        String text = "";
        if (toAdd != null && toAdd instanceof Association) {
            Association asso = (Association) toAdd;
            //System.err.println("text (" + text + ") for" + asso);
            text = asso.getTitle();
            if (text == null || text.length() == 0) {
                text = asso.getName();
            }
        } else if (toAdd != null && toAdd instanceof Attribute) {
            Attribute att = (Attribute) toAdd;
            //System.err.println("text (" + text + ") for" + att);
            text = att.getTitle();
            if (text == null || text.length() == 0) {
                text = att.getName();
            }
        }

        return text;
    }

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