org.eclipse.emf.cdo.transfer.ui.swt.TransferTreeComposite.java Source code

Introduction

Here is the source code for org.eclipse.emf.cdo.transfer.ui.swt.TransferTreeComposite.java

Source

/*
 * Copyright (c) 2012 Eike Stepper (Berlin, Germany) 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:
 *    Eike Stepper - initial API and implementation
 */
package org.eclipse.emf.cdo.transfer.ui.swt;

import org.eclipse.emf.cdo.transfer.CDOTransfer;
import org.eclipse.emf.cdo.transfer.CDOTransferMapping;
import org.eclipse.emf.cdo.transfer.ui.TransferContentProvider;
import org.eclipse.emf.cdo.transfer.ui.TransferLabelProvider;

import org.eclipse.net4j.ui.shared.SharedIcons;

import org.eclipse.jface.action.Action;
import org.eclipse.jface.action.IMenuListener;
import org.eclipse.jface.action.IMenuManager;
import org.eclipse.jface.action.MenuManager;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.TreeViewerColumn;
import org.eclipse.swt.SWT;
import org.eclipse.swt.layout.FillLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Menu;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeColumn;

/**
 * A {@link Composite composite} that contains a {@link #getViewer() tree viewer} for the {@link CDOTransferMapping mappings} of a {@link #getTransfer() transfer}.
 *
 * @author Eike Stepper
 * @since 4.2
 */
public class TransferTreeComposite extends Composite {
    private CDOTransfer transfer;

    private TreeViewer viewer;

    public TransferTreeComposite(Composite parent, int style, CDOTransfer transfer) {
        super(parent, style);
        this.transfer = transfer;

        setLayout(new FillLayout(SWT.VERTICAL));
        viewer = new TreeViewer(this, SWT.FULL_SELECTION);

        Tree tree = viewer.getTree();
        tree.setLinesVisible(true);
        tree.setHeaderVisible(true);

        TreeViewerColumn sourceViewerColumn = new TreeViewerColumn(viewer, SWT.NONE);
        TreeColumn sourceColumn = sourceViewerColumn.getColumn();
        sourceColumn.setWidth(350);
        sourceColumn.setText("From " + transfer.getSourceSystem());

        TreeViewerColumn typeViewerColumn = new TreeViewerColumn(viewer, SWT.NONE);
        TreeColumn typeColumn = typeViewerColumn.getColumn();
        typeColumn.setWidth(100);
        typeColumn.setText("Type");

        TreeViewerColumn targetViewerColumn = new TreeViewerColumn(viewer, SWT.NONE);
        TreeColumn targetColumn = targetViewerColumn.getColumn();
        targetColumn.setWidth(450);
        targetColumn.setText("To " + transfer.getTargetSystem());

        TreeViewerColumn statusViewerColumn = new TreeViewerColumn(viewer, SWT.NONE);
        TreeColumn statusColumn = statusViewerColumn.getColumn();
        statusColumn.setWidth(100);
        statusColumn.setText("Status");

        viewer.setAutoExpandLevel(2);
        viewer.setContentProvider(new TransferContentProvider());
        viewer.setLabelProvider(new TransferLabelProvider(transfer));
        viewer.setInput(transfer.getRootMapping());

        MenuManager manager = new MenuManager();
        Menu menu = manager.createContextMenu(tree);
        manager.addMenuListener(new IMenuListener() {
            public void menuAboutToShow(IMenuManager manager) {
                IStructuredSelection selection = (IStructuredSelection) viewer.getSelection();
                CDOTransferMapping mapping = (CDOTransferMapping) selection.getFirstElement();
                manager.add(new UnmapAction(mapping));
            }
        });

        manager.setRemoveAllWhenShown(true);
        tree.setMenu(menu);
    }

    public CDOTransfer getTransfer() {
        return transfer;
    }

    public TreeViewer getViewer() {
        return viewer;
    }

    public CDOTransferMapping getSelectedMapping() {
        IStructuredSelection selection = (IStructuredSelection) viewer.getSelection();
        return (CDOTransferMapping) selection.getFirstElement();
    }

    @Override
    public boolean setFocus() {
        return viewer.getTree().setFocus();
    }

    /**
     * An {@link Action action} to {@link CDOTransferMapping#unmap() unmap} a {@link CDOTransferMapping transfer mapping}.
     *
     * @author Eike Stepper
     */
    public static class UnmapAction extends Action {
        private CDOTransferMapping mapping;

        public UnmapAction(CDOTransferMapping mapping) {
            super("Unmap", SharedIcons.getDescriptor(SharedIcons.ETOOL_DELETE));
            this.mapping = mapping;
        }

        public CDOTransferMapping getMapping() {
            return mapping;
        }

        @Override
        public void run() {
            mapping.unmap();
        }
    }
}