AWorkspaceAction.java

package org.freeplane.plugin.workspace.actions;

import java.awt.Component;
import java.awt.event.ActionEvent;
import java.net.URL;

import javax.swing.Icon;
import javax.swing.ImageIcon;
import javax.swing.JPopupMenu;
import javax.swing.JTree;
import javax.swing.tree.TreePath;

import org.freeplane.core.resources.ResourceController;
import org.freeplane.core.ui.AFreeplaneAction;
import org.freeplane.core.util.LogUtils;
import org.freeplane.core.util.TextUtils;
import org.freeplane.plugin.workspace.WorkspaceController;
import org.freeplane.plugin.workspace.components.menu.CheckEnableOnPopup;
import org.freeplane.plugin.workspace.components.menu.WorkspacePopupMenu;
import org.freeplane.plugin.workspace.model.AWorkspaceTreeNode;
import org.freeplane.plugin.workspace.model.project.AWorkspaceProject;

public abstract class AWorkspaceAction extends AFreeplaneAction {

	/**
	 * 
	 */
	private static final long serialVersionUID = 1L;

	public AWorkspaceAction(String key, String title, Icon icon) {
		super(key, title, icon);
	}

	public AWorkspaceAction(String key) {
		super(key, TextUtils.getRawText(getTextKey(key)), null);
		setIcon();
	}
	
	public void setEnabledFor(AWorkspaceTreeNode node, TreePath[] selectedPaths) {
		setEnabled();
	}

	public void setEnabled() {
		setEnabled(true);
	}
	
	public void setSelectedFor(AWorkspaceTreeNode node, TreePath[] selectedPaths) {
		super.setSelected();
	}
	
	public String getTextKey() {
		return getTextKey(getKey());
	}
	
	private static String getTextKey(String key) {
		return key + ".label";
	}
	
	public void afterMapChange(final Object newMap) {
	}
	
	private void setIcon() {
		final String iconResource = ResourceController.getResourceController().getProperty(getIconKey(), null);
		if (iconResource != null) {
			// look in this package
			URL url = this.getClass().getResource(iconResource);
			if (url != null) {
				final ImageIcon icon = new ImageIcon(url);
				putValue(SMALL_ICON, icon);
				return;
			}
			//  look in workspace package
			url = WorkspaceController.class.getResource(iconResource);
			if (url != null) {
				final ImageIcon icon = new ImageIcon(url);
				putValue(SMALL_ICON, icon);
				return;
			}
			// look in freeplane package
			url = ResourceController.class.getResource(iconResource);
			if (url != null) {
				final ImageIcon icon = new ImageIcon(url);
				putValue(SMALL_ICON, icon);
				return;
			}
			
			LogUtils.severe("can not load icon '" + iconResource + "'");
		}
	}
	
	protected AWorkspaceTreeNode getNodeFromActionEvent(ActionEvent e) {
		JTree tree = null;
		TreePath path = null;
		
		if(e.getSource() instanceof JTree) {
			tree = (JTree) e.getSource();
			path = tree.getSelectionPath();
		} 
		else { 
			WorkspacePopupMenu pop = getRootPopupMenu((Component) e.getSource());
			if(pop == null) {
				return null;
			}
			tree = (JTree)pop.getInvoker();
			int x = pop.getInvokerLocation().x;
			int y = pop.getInvokerLocation().y;
			
			path = tree.getClosestPathForLocation(x, y);
		}
		
		if(path == null) {
			return null;
		}
		return (AWorkspaceTreeNode) path.getLastPathComponent();
	}
	
	protected AWorkspaceTreeNode[] getSelectedNodes(ActionEvent e) {
		JTree tree = null;
		if(e.getSource() instanceof JTree) {
			tree = (JTree) e.getSource();
		} 
		else { 
			WorkspacePopupMenu pop = getRootPopupMenu((Component) e.getSource());
			if(pop == null) {
				return null;
			}
			tree = (JTree)pop.getInvoker();
		}
		AWorkspaceTreeNode[] nodes = new AWorkspaceTreeNode[tree.getSelectionPaths().length]; 
		int i = 0;
		for(TreePath path : tree.getSelectionPaths()) {
			nodes[i++] = (AWorkspaceTreeNode) path.getLastPathComponent();
		}
		return nodes;
	}
	
	
	
	protected AWorkspaceProject getProjectFromActionEvent(ActionEvent e) {
		AWorkspaceTreeNode node = getNodeFromActionEvent(e);
		if(node == null) {
			return null;
		}
		return WorkspaceController.getSelectedProject(node);
	}
	
	public WorkspacePopupMenu getRootPopupMenu(Component component) {
		Component parent = component;
		while(!(parent instanceof WorkspacePopupMenu) && parent != null) {
			if(parent.getParent() == null && parent instanceof JPopupMenu) {
				parent = getRootPopupMenu(((JPopupMenu) parent).getInvoker());
				break;
			} 
			else  {
				parent = parent.getParent();
			}
		}
		return (WorkspacePopupMenu) parent;
	}
	
	protected Component getComponentFromActionEvent(ActionEvent e) {
		WorkspacePopupMenu pop = getRootPopupMenu((Component) e.getSource()); //(WorkspacePopupMenu)((Component) e.getSource()).getParent();		
		JTree tree = (JTree)pop.getInvoker();
		return tree.getComponentAt(pop.getInvokerLocation());
	}
	
	static public boolean checkEnabledOnPopup(final AFreeplaneAction action) {
		if(action instanceof AWorkspaceAction) {	
			final CheckEnableOnPopup annotation = action.getClass().getAnnotation(CheckEnableOnPopup.class);
			if (annotation != null) {
				return true;
			}
		}
		return false;
	}
}