WorkspaceController.java example

Explorer
Desktop-master
package org.freeplane.plugin.workspace;

import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.net.URL;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import org.freeplane.core.extension.IExtension;
import org.freeplane.core.resources.IFreeplanePropertyListener;
import org.freeplane.core.resources.OptionPanelController;
import org.freeplane.core.resources.OptionPanelController.PropertyLoadListener;
import org.freeplane.core.resources.ResourceBundles;
import org.freeplane.core.resources.components.IPropertyControl;
import org.freeplane.core.ui.AFreeplaneAction;
import org.freeplane.core.ui.FreeplaneActionCascade;
import org.freeplane.core.util.Compat;
import org.freeplane.core.util.LogUtils;
import org.freeplane.features.map.MapModel;
import org.freeplane.features.mode.AController;
import org.freeplane.features.mode.Controller;
import org.freeplane.features.mode.ModeController;
import org.freeplane.features.mode.mindmapmode.MModeController;
import org.freeplane.plugin.workspace.features.AWorkspaceModeExtension;
import org.freeplane.plugin.workspace.features.ModeControlAlreadyRegisteredException;
import org.freeplane.plugin.workspace.features.WorkspaceMapModelExtension;
import org.freeplane.plugin.workspace.features.WorkspaceModelExtensionWriterReader;
import org.freeplane.plugin.workspace.io.FileSystemManager;
import org.freeplane.plugin.workspace.mindmapmode.MModeWorkspaceController;
import org.freeplane.plugin.workspace.model.AWorkspaceTreeNode;
import org.freeplane.plugin.workspace.model.WorkspaceModel;
import org.freeplane.plugin.workspace.model.project.AWorkspaceProject;

public final class WorkspaceController implements IExtension {
	/**
	 * @deprecated -- use PROJECT_RESOURCE_URL_PROTOCOL
	 */
	@Deprecated
	public static final String WORKSPACE_RESOURCE_URL_PROTOCOL = "workspace";
	public static final String PROJECT_RESOURCE_URL_PROTOCOL = "project";
	public static final String PROPERTY_RESOURCE_URL_PROTOCOL = "property";
	public static final String WORKSPACE_VERSION = "1.0";
	
	private static WorkspaceController self;
	private static Map<Class<? extends ModeController>, Class<? extends AWorkspaceModeExtension>> modeWorkspaceCtrlMap = new HashMap<Class<? extends ModeController>, Class<? extends AWorkspaceModeExtension>>();
	
	private WorkspaceController(Controller controller) {
		self = this;
	}
	
	public static void addAction(final AFreeplaneAction action) {
		addAction(Controller.getCurrentController(), action);
	}
	
	public static void addAction(AController controller, final AFreeplaneAction action) {
		if(action == null) {
			return;
		}
		try {
			controller.addAction(action);
		} catch (Exception e) {
			LogUtils.info(WorkspaceController.class + ".addAction(): action "+ action.getKey() +" not added! ("+e.getMessage()+")");
		}	
	}
	
	public static void addCascadingAction(final AFreeplaneAction action) {
		if(action == null) {
			return;
		}
		try {
			FreeplaneActionCascade.addAction(action);
		} catch (Exception e) {
			LogUtils.info(WorkspaceController.class + ".addCascadingAction(): action "+ action.getKey() +" not added! ("+e.getMessage()+")");
		}	
	}

	public static void replaceAction(final AController controller, final AFreeplaneAction action) {
		AFreeplaneAction previousAction = getAction(controller, action.getKey());
		if(previousAction != null) {
			removeAction(controller, action.getKey());
		}
		addAction(controller, action);		
	}
	
	public static void replaceAction(final AFreeplaneAction action) {
		replaceAction(Controller.getCurrentController(), action);
	}
	
	public static AFreeplaneAction getAction(final String key) {
		return getAction(Controller.getCurrentController(), key);
	}
	
	public static AFreeplaneAction getAction(final AController controller, final String key) {
		try {
			return controller.getAction(key);
		} catch (Exception e) {
			LogUtils.info(WorkspaceController.class + ".getAction(): action "+ key +" not found!");
		}		
		return null;
	}

	public static AFreeplaneAction removeAction(final AController controller, final String key) {
		try {
			return controller.removeAction(key);
		} catch (Exception e) {
			LogUtils.info(WorkspaceController.class + "removeAction(): action "+ key +" not found!");
		}		
		return null;
	}

	public static void install(Controller controller) {
		if(self == null) {
			new WorkspaceController(controller);
			self.setupLanguage(controller);
		}
		controller.addExtension(WorkspaceController.class, self);
	}
	
	public static WorkspaceController getController() {
		return self;
	}
	
	public static void registerWorkspaceModeExtension(Class<? extends ModeController> modeController, Class<? extends AWorkspaceModeExtension> modeWorkspaceCtrl) throws ModeControlAlreadyRegisteredException {
		synchronized (modeWorkspaceCtrlMap) {
			//WORKSPACE - INFO: allow overwrite?
			if(modeWorkspaceCtrlMap.containsKey(modeController)) {
				throw new ModeControlAlreadyRegisteredException(modeController);
			}
			modeWorkspaceCtrlMap.put(modeController, modeWorkspaceCtrl);
		}
	}
	
	public static void removeWorkspaceModeExtension(Class<? extends ModeController> modeController) {
		synchronized (modeWorkspaceCtrlMap) {
			modeWorkspaceCtrlMap.remove(modeController);
		}
	}

	public boolean installMode(ModeController modeController) {
		AWorkspaceModeExtension modeCtrl = modeController.getExtension(AWorkspaceModeExtension.class);
		if(modeCtrl == null) {
			Class<? extends AWorkspaceModeExtension> clazz = modeWorkspaceCtrlMap.get(modeController.getClass());
			if(clazz == null) {
				return false;
			}
			try {
				modeCtrl = clazz.getConstructor(ModeController.class).newInstance(modeController);
				modeController.addExtension(AWorkspaceModeExtension.class, modeCtrl);
				WorkspaceModelExtensionWriterReader.register(modeController);
				return true;
			} catch (Exception e) {
				e.printStackTrace();
			}			
		}
		else {
			return true;
		}
		return false;
		
	}
	
	private void setupLanguage(Controller controller) {
		setLanguage();
		
		final OptionPanelController optionController = controller.getOptionPanelController();
		
		optionController.addPropertyLoadListener(new PropertyLoadListener() {			
			public void propertiesLoaded(Collection<IPropertyControl> properties) {
				setLanguage();
			}
		});
		
		controller.getResourceController().addPropertyChangeListener(new IFreeplanePropertyListener() {
			
			public void propertyChanged(String propertyName, String newValue, String oldValue) {
				if(propertyName.equalsIgnoreCase("language")){
					setLanguage();
				}
			}
		});
		try {
			WorkspaceController.registerWorkspaceModeExtension(MModeController.class, MModeWorkspaceController.class);
		} catch (ModeControlAlreadyRegisteredException e) {
			e.printStackTrace();
		}
		
	}
	
	private void setLanguage() {
		final String DEFAULT_LANGUAGE = "en";
		ResourceBundles resBundle = ((ResourceBundles)Controller.getCurrentController().getResourceController().getResources());
		String lang = resBundle.getLanguageCode();
		if (lang == null || lang.equals(ResourceBundles.LANGUAGE_AUTOMATIC)) {
			lang = DEFAULT_LANGUAGE;
		}
		
		URL res = this.getClass().getResource("/translations/Resources_"+lang+".properties");
		if (res == null) {
			lang = DEFAULT_LANGUAGE;
			res = this.getClass().getResource("/translations/Resources_"+lang+".properties");
		}
		
		if (res == null) {
			return;
		}
					
		resBundle.addResources(resBundle.getLanguageCode(), res);
	}
	
	public void startModeExtension(ModeController modeController) {
		AWorkspaceModeExtension modeCtrl = modeController.getExtension(AWorkspaceModeExtension.class);
		if(modeCtrl == null) {
			return;
		}
		modeCtrl.start(modeController);
		
	}
	
	public void shutdown() {
		for (String modeName : Controller.getCurrentController().getModes()) {
			ModeController modeController = Controller.getCurrentController().getModeController(modeName);
			AWorkspaceModeExtension modeExt = modeController.getExtension(AWorkspaceModeExtension.class);
			if(modeExt == null) {
				continue;
			}
			modeExt.shutdown();
		}
		
	}
	
	public static WorkspaceModel getCurrentModel() {
		return getCurrentModeExtension().getModel();
	}

	public static AWorkspaceModeExtension getModeExtension(ModeController modeController) {
		if(modeController == null) {
			return null;
		}
		return modeController.getExtension(AWorkspaceModeExtension.class);
	}
	
	public static AWorkspaceModeExtension getCurrentModeExtension() {
		return getModeExtension(Controller.getCurrentModeController());
	}
	
	public static FileSystemManager getFileSystemMgr() {
		return new FileSystemManager(getCurrentModeExtension().getFileTypeManager());
	}

	public static void loadProject(AWorkspaceProject project) throws IOException {
		getCurrentModeExtension().getProjectLoader().loadProject(project);
		indexProject(project);
	}

	public static URI getDefaultProjectHome() {		
		return getCurrentModeExtension().getDefaultProjectHome();
	}

	public static URI getApplicationHome() {
		String appName = Controller.getCurrentController().getResourceController().getProperty("ApplicationName");
		String homePath = System.getProperty("user.home")+ File.separator + appName;
		return new File(homePath).toURI();
	}
	
	public static URI getApplicationSettingsHome() {
		File home = new File(Compat.getApplicationUserDirectory());
		return home.toURI();
	}
		
	public static AWorkspaceProject getMapProject() {
		return getMapProject(Controller.getCurrentController().getMap());       
	}
	
	public static AWorkspaceProject getMapProject(MapModel map) {
		if (map != null) {
			WorkspaceMapModelExtension wmme = getMapModelExtension(map);
			return wmme.getProject();
		}
		return null;
	}
	
	public static AWorkspaceProject getSelectedProject() {
		return getCurrentModeExtension().getSelectedProject();
	}
	
	public static AWorkspaceProject getSelectedProject(AWorkspaceTreeNode node) {
		return getCurrentModel().getProject(node.getModel());
	}

	public static WorkspaceMapModelExtension getMapModelExtension(MapModel map) {
		return getMapModelExtension(map, true);
	}
	
	public static WorkspaceMapModelExtension getMapModelExtension(MapModel map, boolean createIfNotExists) {
		if(map == null) {
			return null;
		}
		WorkspaceMapModelExtension wmme = map.getExtension(WorkspaceMapModelExtension.class);
		if(createIfNotExists && wmme == null) {
			wmme = new WorkspaceMapModelExtension();
			map.addExtension(WorkspaceMapModelExtension.class, wmme);
		}
		return wmme;
	}
	
	public static AWorkspaceProject addMapToProject(MapModel map, AWorkspaceProject project) {
		return addMapToProject(map, project, true);
	}
	
	public static AWorkspaceProject addMapToProject(MapModel map, AWorkspaceProject project, boolean overwrite) {
		if(map == null || project == null) {
			throw new IllegalArgumentException("NULL");
		}
		WorkspaceMapModelExtension wmme = getMapModelExtension(map);
		AWorkspaceProject oldProject = wmme.getProject();
		if(overwrite || oldProject == null) {
			wmme.setProject(project);
		}
		return oldProject;
	}
	
	public static void indexProject(AWorkspaceProject project) {
		if(project == null) {
			return;
		}
		getCurrentModeExtension().getModel().indexProject(project);
	}
	
	public static AWorkspaceProject getCachedProjectByID(String projectID) {
		if(projectID == null) {
			return null;
		}
		try {
			return getCurrentModeExtension().getModel().getCachedProjectByID(projectID);
		}
		catch (Exception e) {
			LogUtils.info("Exception in org.freeplane.plugin.workspace.WorkspaceController.getCachedProjectByID(projectID): "+ e.getMessage());
			return null;
		}
	}

	public static void save() {
		getCurrentModeExtension().save();
	}

	public static void load() {
		getCurrentModeExtension().load();
	}
	
	public static void clear() {
		getCurrentModeExtension().clear();
	}
	
}