path: root/plugins/org.eclipse.emf.cdo.transfer.ui/src/org/eclipse/emf/cdo/transfer/ui/TransferDropAdapter.java

/*
 * Copyright (c) 2004 - 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;

import org.eclipse.emf.cdo.transfer.CDOTransfer;
import org.eclipse.emf.cdo.transfer.CDOTransferElement;
import org.eclipse.emf.cdo.transfer.CDOTransferSystem;
import org.eclipse.emf.cdo.transfer.spi.ui.TransferUIProvider;
import org.eclipse.emf.cdo.transfer.spi.ui.TransferUIProvider.Factory;
import org.eclipse.emf.cdo.transfer.ui.TransferDialog.InitializationState;

import org.eclipse.net4j.util.container.IManagedContainer;
import org.eclipse.net4j.util.container.IPluginContainer;
import org.eclipse.net4j.util.ui.dnd.DNDDropAdapter;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.jface.viewers.StructuredViewer;
import org.eclipse.swt.dnd.DND;
import org.eclipse.swt.dnd.Transfer;
import org.eclipse.swt.widgets.Shell;

import java.util.ArrayList;
import java.util.List;

/**
 * @author Eike Stepper
 */
public class TransferDropAdapter extends DNDDropAdapter<Object>
{
  private TransferUIProvider[] uiProviders;

  public TransferDropAdapter(StructuredViewer viewer)
  {
    super(viewer);
    uiProviders = getUIProviders();

    List<Transfer> transfers = new ArrayList<Transfer>();
    for (int i = 0; i < uiProviders.length; i++)
    {
      TransferUIProvider uiProvider = uiProviders[i];
      uiProvider.addSupportedTransfers(transfers);
    }

    setTransfers(transfers.toArray(new Transfer[transfers.size()]));
  }

  protected IManagedContainer getContainer()
  {
    return IPluginContainer.INSTANCE;
  }

  protected TransferUIProvider[] getUIProviders()
  {
    return Factory.getAll(getContainer());
  }

  @Override
  protected boolean validateTarget(Object target, int operation)
  {
    CDOTransferElement targetElement = getTargetElement(target);
    if (targetElement != null && targetElement.isDirectory())
    {
      overrideOperation(DND.DROP_COPY);
      return true;
    }

    return false;
  }

  @Override
  protected boolean performDrop(Object data, Object target)
  {
    if (data == null)
    {
      return false;
    }

    final List<CDOTransferElement> sourceElements = getSourceElements(data);
    if (sourceElements == null || sourceElements.isEmpty())
    {
      return false;
    }

    final CDOTransferElement targetElement = getTargetElement(target);
    if (targetElement == null || !targetElement.isDirectory())
    {
      return false;
    }

    CDOTransferSystem sourceSystem = sourceElements.get(0).getSystem();
    CDOTransferSystem targetSystem = targetElement.getSystem();

    final CDOTransfer transfer = new CDOTransfer(sourceSystem, targetSystem);
    transfer.getRootMapping().setRelativePath(targetElement.getPath());

    Shell shell = getViewer().getControl().getShell();
    final TransferDialog dialog = new TransferDialog(shell, transfer);

    new Job("Mapping elements")
    {
      @Override
      protected IStatus run(IProgressMonitor monitor)
      {
        monitor.beginTask(getName(), sourceElements.size());

        try
        {
          for (CDOTransferElement sourceElement : sourceElements)
          {
            if (monitor.isCanceled() || dialog.getInitializationState() == InitializationState.CANCELED)
            {
              dialog.setInitializationState(InitializationState.CANCELED);
              return Status.OK_STATUS;
            }

            transfer.map(sourceElement);
            monitor.worked(1);
          }

          dialog.setInitializationState(InitializationState.MAPPED);
          return Status.OK_STATUS;
        }
        catch (RuntimeException ex)
        {
          dialog.setInitializationState(InitializationState.FAILED);
          throw ex;
        }
        finally
        {
          monitor.done();
        }
      }
    }.schedule();

    if (dialog.open() == TransferDialog.OK)
    {
      transfer.perform();
      return true;
    }

    return false;
  }

  protected boolean performTransfer(CDOTransfer transfer)
  {
    Shell shell = getViewer().getControl().getShell();
    TransferDialog dialog = new TransferDialog(shell, transfer);
    if (dialog.open() == TransferDialog.OK)
    {
      transfer.perform();
      return true;
    }

    return false;
  }

  protected List<CDOTransferElement> getSourceElements(Object data)
  {
    for (int i = 0; i < uiProviders.length; i++)
    {
      TransferUIProvider uiProvider = uiProviders[i];

      List<CDOTransferElement> elements = uiProvider.convertTransferData(data);
      if (elements != null)
      {
        return elements;
      }
    }

    return null;
  }

  protected CDOTransferElement getTargetElement(Object target)
  {
    for (int i = 0; i < uiProviders.length; i++)
    {
      TransferUIProvider uiProvider = uiProviders[i];

      CDOTransferElement element = uiProvider.convertTransferTarget(target);
      if (element != null)
      {
        return element;
      }
    }

    return null;
  }

  public static TransferDropAdapter support(StructuredViewer viewer)
  {
    TransferDropAdapter dropAdapter = new TransferDropAdapter(viewer);
    Transfer[] transfers = dropAdapter.getTransfers();
    viewer.addDropSupport(DND.DROP_COPY | DND.DROP_MOVE | DND.DROP_DEFAULT, transfers, dropAdapter);
    return dropAdapter;
  }
}