CaptureHandler.java

/**
 * Copyright 2010 The University of North Carolina at Chapel Hill
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package unc.lib.cdr.workbench.capture;

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

import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.jface.viewers.ISelectionProvider;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.window.Window;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IWorkbenchPart;
import org.eclipse.ui.handlers.HandlerUtil;
import org.eclipse.ui.progress.IWorkbenchSiteProgressService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import unc.lib.cdr.workbench.originals.OriginalFileStore;

public class CaptureHandler extends AbstractHandler {

	@SuppressWarnings("unused")
	private static final Logger LOG = LoggerFactory.getLogger(CaptureHandler.class);

	/*
	 * if parent folders have been arranged, they will stay in their places, reset arrangement is separate!
	 *
	 * @see org.eclipse.core.commands.IHandler#execute(org.eclipse.core.commands. ExecutionEvent)
	 */
	@Override
	public Object execute(ExecutionEvent event) throws ExecutionException {
		List<OriginalFileStore> toCapture = new ArrayList<OriginalFileStore>();
		ISelectionProvider sp = HandlerUtil.getActiveSite(event).getSelectionProvider();
		IStructuredSelection s = (IStructuredSelection) sp.getSelection();
		toCapture.addAll(s.toList());
		

		CaptureJob job = new CaptureJob("Capturing " + Integer.toString(toCapture.size()) + " items...", toCapture);
		
		String filter = event.getParameter("workbench_plugin.commandParameterCaptureFilter");
		if("true".equals(filter)) {
			SelectFileExtensionsDialog dialog = new SelectFileExtensionsDialog(HandlerUtil.getActiveShell(event), toCapture);
			dialog.create();
			if (dialog.open() == Window.OK) {
			  job.setIncludedFileExtensions(dialog.getSelectedFileExtensions());
			} else {
				return null;
			}
		}

		IWorkbenchPart part = HandlerUtil.getActivePart(event);
		if (part != null) {
			IWorkbenchSiteProgressService siteService = (IWorkbenchSiteProgressService) part.getSite().getAdapter(
					IWorkbenchSiteProgressService.class);
			Shell shell = HandlerUtil.getActiveShell(event);
			siteService.showInDialog(shell, job);
			siteService.schedule(job);
		} else {
			job.schedule();
		}
		return null;
	}

}