package org.argeo.jcr.ui.explorer.commands;

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

import javax.jcr.Node;

import org.argeo.eclipse.ui.ErrorFeedback;
import org.argeo.jcr.JcrUtils;
import org.argeo.jcr.ui.explorer.JcrExplorerPlugin;
import org.argeo.jcr.ui.explorer.model.SingleJcrNode;
import org.argeo.jcr.ui.explorer.model.WorkspaceNode;
import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.handlers.HandlerUtil;

/** Opens the generic node editor. */
public class GetNodeSize extends AbstractHandler {
	// private final static Log log = LogFactory.getLog(GetNodeSize.class);

	public final static String ID = JcrExplorerPlugin.ID + ".getNodeSize";

	// public final static String DEFAULT_ICON_REL_PATH = "icons/getSize.gif";
	// public final static String DEFAULT_LABEL = JcrExplorerPlugin
	// .getMessage("getNodeSizeCmdLbl");

	public Object execute(ExecutionEvent event) throws ExecutionException {
		// JcrUtils.getRepositoryByAlias(repositoryRegister,
		// ArgeoJcrConstants.ALIAS_NODE);

		ISelection selection = HandlerUtil.getActiveWorkbenchWindow(event)
				.getActivePage().getSelection();

		if (selection != null && !selection.isEmpty()
				&& selection instanceof IStructuredSelection) {

			// IStructuredSelection iss = (IStructuredSelection) selection;
			// if (iss.size() > 1)
			// ErrorFeedback.show(JcrExplorerPlugin
			// .getMessage("warningInvalidMultipleSelection"), null);

			long size = 0;

			Iterator<?> it = ((IStructuredSelection) selection).iterator();

			// as the size method is recursive, we keep track of nodes for which
			// we already have computed size so that we don't count them twice.
			// In a first approximation, we assume that the structure selection
			// keep the nodes ordered.
			// TODO : enhance that.
			List<String> importedPathes = new ArrayList<String>();
			try {
				nodesIt: while (it.hasNext()) {
					Object obj = it.next();
					String curPath;
					Node node;
					if (obj instanceof SingleJcrNode) {
						node = ((SingleJcrNode) obj).getNode();
						curPath = node.getSession().getWorkspace().getName();
						curPath += "/" + node.getPath();
					} else if (obj instanceof WorkspaceNode) {
						node = ((WorkspaceNode) obj).getRootNode();
						curPath = node.getSession().getWorkspace().getName();
					} else
						// unvalid object type
						continue nodesIt;

					Iterator<String> itPath = importedPathes.iterator();
					while (itPath.hasNext()) {
						String refPath = itPath.next();
						if (curPath.startsWith(refPath))
							// Already done : skip node
							continue nodesIt;
					}
					size += JcrUtils.getNodeApproxSize(node);
					importedPathes.add(curPath);
				}
			} catch (Exception e) {
				ErrorFeedback.show("Cannot Get size of selected node ", e);
			}

			String[] labels = { "OK" };
			Shell shell = HandlerUtil.getActiveWorkbenchWindow(event)
					.getShell();
			MessageDialog md = new MessageDialog(shell, "Node size", null,
					"Node size is: " + size / 1024 + " KB",
					MessageDialog.INFORMATION, labels, 0);
			md.open();
		}
		return null;
	}
}