org.apache.hdt.ui/src/org/apache/hdt/ui/internal/hdfs/DiscardDownloadResourceAction.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.hdt.ui.internal.hdfs;

import java.io.File;
import java.util.Iterator;

import org.apache.hdt.core.internal.hdfs.HDFSFileStore;
import org.apache.hdt.core.internal.hdfs.HDFSManager;
import org.apache.hdt.core.internal.hdfs.UploadFileJob;
import org.apache.log4j.Logger;
import org.eclipse.core.filesystem.EFS;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.jface.action.IAction;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.ui.IObjectActionDelegate;
import org.eclipse.ui.IWorkbenchPart;

public class DiscardDownloadResourceAction implements IObjectActionDelegate {

	private final static Logger logger = Logger.getLogger(DiscardDownloadResourceAction.class);
	private ISelection selection;
	private IWorkbenchPart targetPart;

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.ui.IActionDelegate#run(org.eclipse.jface.action.IAction)
	 */
	@Override
	public void run(IAction action) {
		if (this.selection != null && !this.selection.isEmpty()) {
			IStructuredSelection sSelection = (IStructuredSelection) this.selection;
			@SuppressWarnings("rawtypes")
			Iterator itr = sSelection.iterator();
			while (itr.hasNext()) {
				Object object = itr.next();
				if (object instanceof IResource) {
					IResource r = (IResource) object;
					discardDownloadResource(r);
				}
			}
		}
	}

	/**
	 * @param r
	 */
	private void discardDownloadResource(IResource r) {
		try {
			HDFSFileStore store = (HDFSFileStore) EFS.getStore(r.getLocationURI());
			switch (r.getType()) {
			case IResource.FOLDER:
				IFolder folder = (IFolder) r;
				IResource[] members = folder.members();
				if (members != null) {
					for (int mc = 0; mc < members.length; mc++) {
						discardDownloadResource(members[mc]);
					}
				}
			case IResource.FILE:
				if (store.isLocalFile()) {
					File file = store.getLocalFile();
					HDFSManager.INSTANCE.startServerOperation(store.toURI().toString());
					try{
						if (file.exists()) {
							file.delete();
							UploadFileJob.deleteFoldersIfEmpty(file.getParentFile());
						}
						r.getParent().refreshLocal(IResource.DEPTH_ONE, new NullProgressMonitor());
					}finally{
						HDFSManager.INSTANCE.stopServerOperation(store.toURI().toString());
					}
				}
			}
		} catch (CoreException e) {
			MessageDialog.openError(targetPart.getSite().getShell(), "Upload HDFS Resources", "Error uploading resource to " + r.getLocationURI() + ": "
					+ e.getMessage());
			logger.warn(e.getMessage(), e);
		}
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * org.eclipse.ui.IActionDelegate#selectionChanged(org.eclipse.jface.action
	 * .IAction, org.eclipse.jface.viewers.ISelection)
	 */
	@Override
	public void selectionChanged(IAction action, ISelection selection) {
		this.selection = selection;
		boolean enabled = true;
		if (this.selection != null && !this.selection.isEmpty()) {
			IStructuredSelection sSelection = (IStructuredSelection) this.selection;
			@SuppressWarnings("rawtypes")
			Iterator itr = sSelection.iterator();
			while (itr.hasNext()) {
				Object object = itr.next();
				if (object instanceof IResource) {
					IResource r = (IResource) object;
					try {
						HDFSFileStore store = (HDFSFileStore) EFS.getStore(r.getLocationURI());
						enabled = store.isLocalFile();
					} catch (Throwable t) {
						enabled = false;
					}
				} else
					enabled = false;
			}
		} else
			enabled = false;
		action.setEnabled(enabled);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * org.eclipse.ui.IObjectActionDelegate#setActivePart(org.eclipse.jface.
	 * action.IAction, org.eclipse.ui.IWorkbenchPart)
	 */
	@Override
	public void setActivePart(IAction action, IWorkbenchPart targetPart) {
		this.targetPart = targetPart;

	}

}