DocearRecommendationsMapController.java example

Explorer
Docear-master
package org.docear.plugin.services.recommendations.mode;

import java.io.InputStreamReader;
import java.net.UnknownHostException;
import java.rmi.UnexpectedException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

import javax.swing.JProgressBar;
import javax.swing.ProgressMonitor;
import javax.swing.SwingUtilities;
import javax.ws.rs.core.MultivaluedMap;

import org.docear.plugin.core.util.CoreUtils;
import org.docear.plugin.services.ServiceController;
import org.docear.plugin.services.communications.CommunicationsController;
import org.docear.plugin.services.communications.features.DocearServiceResponse;
import org.docear.plugin.services.communications.features.DocearServiceResponse.Status;
import org.docear.plugin.services.recommendations.RecommendationEntry;
import org.docear.plugin.services.xml.DocearXmlBuilder;
import org.docear.plugin.services.xml.DocearXmlElement;
import org.docear.plugin.services.xml.DocearXmlRootElement;
import org.freeplane.core.ui.components.UITools;
import org.freeplane.core.util.TextUtils;
import org.freeplane.features.map.IMapLifeCycleListener;
import org.freeplane.features.map.MapController;
import org.freeplane.features.map.MapModel;
import org.freeplane.features.map.NodeModel;
import org.freeplane.features.mode.Controller;
import org.freeplane.features.mode.mindmapmode.MModeController;
import org.freeplane.features.ui.IMapViewManager;
import org.freeplane.n3.nanoxml.IXMLParser;
import org.freeplane.n3.nanoxml.IXMLReader;
import org.freeplane.n3.nanoxml.StdXMLReader;
import org.freeplane.n3.nanoxml.XMLParserFactory;
import org.freeplane.view.swing.map.MapView;

import com.sun.jersey.core.util.StringKeyStringValueIgnoreCaseMultivaluedMap;



public class DocearRecommendationsMapController extends MapController {

	private DocearRecommendationsModeController modeController;
	private MapView currentMapView;

	public DocearRecommendationsMapController(DocearRecommendationsModeController modeController) {
		super(modeController);
		this.modeController = modeController;
		addMapLifeCycleListener(new IMapLifeCycleListener() {
			public void onSavedAs(MapModel map) {
			}

			public void onSaved(MapModel map) {
			}

			public void onRemove(MapModel map) {
				currentMapView = null;
				if(Controller.getCurrentController().getMapViewManager().getMaps().size() <= 1) {
					SwingUtilities.invokeLater(new Runnable() {
						
						public void run() {
							Controller.getCurrentController().selectMode(MModeController.MODENAME);
						}
					});
				}
			}

			public void onCreate(MapModel map) {
			}
		});
	}

	public DocearRecommendationsModeController getModeController() {
		return modeController;
	}

	public MapModel newMap() {			
		DocearRecommendationsMapModel mapModel = null;
		final ProgressMonitor monitor = new ProgressMonitor(UITools.getFrame(), TextUtils.getText("recommendations.request.wait.text"), null, 0, 100);
		monitor.setMillisToDecideToPopup(0);
		monitor.setMillisToPopup(0);
		if (ServiceController.getController().isRecommendationsAllowed()) {
			new Thread() {			
				public void run() {
					DocearRecommendationsMapModel mapModel = null;	
					try {					
						monitor.setProgress(1);
						((JProgressBar)monitor.getAccessibleContext().getAccessibleChild(1)).setIndeterminate(true);
						long l = System.currentTimeMillis();
						Collection<RecommendationEntry> recommendations = getNewRecommendations(true);
						System.out.println("exec time: "+(System.currentTimeMillis()-l));
						mapModel = new DocearRecommendationsMapModel(recommendations);
					}
					catch(Exception e) {
						mapModel = getExceptionModel(e);
					}
					SwingUtilities.invokeLater(getNewMapRunnable(monitor, mapModel));									
					
				}

				private Runnable getNewMapRunnable(final ProgressMonitor monitor, final DocearRecommendationsMapModel mapModel) {
					return new Runnable() {
						
						public void run() {
							fireMapCreated(mapModel);
							newMapView(mapModel);
							monitor.close();
						}
					};
				}
				
			}.start();
			//mapModel = getProgressModel();
		} 
		else {
			mapModel = new DocearRecommendationsMapModel(null);
			fireMapCreated(mapModel);
			newMapView(mapModel);
		}
		
		
		return mapModel;
	}
	
	public MapModel newMap(Collection<RecommendationEntry> recommendations) {			
		DocearRecommendationsMapModel mapModel = new DocearRecommendationsMapModel(recommendations);
		
		fireMapCreated(mapModel);
		newMapView(mapModel);
		
		return mapModel;
	}

	private DocearRecommendationsMapModel getExceptionModel(Exception e) {
		DocearRecommendationsMapModel mapModel = new DocearRecommendationsMapModel();
		String message = "";
		if (e instanceof UnknownHostException) {
			message = TextUtils.getText("recommendations.error.no_connection");
		}
		else {
			message = TextUtils.getText("recommendations.error.unknown");
		}
		mapModel.setRoot(DocearRecommendationsNodeModel.getNoRecommendationsNode(mapModel, message));
		mapModel.getRootNode().setFolded(false);
		return mapModel;
	}
	
	public NodeModel newNode(final Object userObject, final MapModel map) {
		throw new UnsupportedOperationException();
	}

	public void toggleFolded(final NodeModel node) {
		if (hasChildren(node) && !node.isRoot()) {
			setFolded(node, !isFolded(node));
		}
	}

	public void newMapView(MapModel map) {
		closeCurrentMapView();
		this.currentMapView = createMapView(map);
		IMapViewManager manager = Controller.getCurrentController().getMapViewManager();
		manager.changeToMapView(this.currentMapView);
		manager.updateMapViewName();
	}

	public void refreshRecommendations() {
		newMap();
	}
	
	private void closeCurrentMapView() {
		if (this.currentMapView != null) {
			Controller.getCurrentController().getMapViewManager().changeToMapView(this.currentMapView);
			Controller.getCurrentController().getMapViewManager().close(true);
			this.currentMapView = null;
		}
	}

	private MapView createMapView(MapModel map) {
		DocearRecommendationsMapView mapView = new DocearRecommendationsMapView(map, this.getModeController());
		return mapView;
	}

	public static Collection<RecommendationEntry> getNewRecommendations(boolean userRequest) throws UnknownHostException, UnexpectedException {		
		String name = CommunicationsController.getController().getUserName();
		if (!CoreUtils.isEmpty(name)) {
			MultivaluedMap<String,String> params = new StringKeyStringValueIgnoreCaseMultivaluedMap();
			if(!userRequest) {
				params.add("auto", "true");
			}
			DocearServiceResponse response = CommunicationsController.getController().get("/user/" + name + "/recommendations/documents", params);

			if (response.getStatus() == Status.OK) {
				try {
					DocearXmlBuilder xmlBuilder = new DocearXmlBuilder();
					IXMLReader reader = new StdXMLReader(new InputStreamReader(response.getContent(), "UTF8"));
					IXMLParser parser = XMLParserFactory.createDefaultXMLParser();
					parser.setBuilder(xmlBuilder);
					parser.setReader(reader);
					parser.parse();
					DocearXmlRootElement result = (DocearXmlRootElement) xmlBuilder.getRoot();
					Collection<DocearXmlElement> documents = result.findAll("document");
					List<RecommendationEntry> recommendations = new ArrayList<RecommendationEntry>();
					for (DocearXmlElement document : documents) {
						String title = document.find("title").getContent();
						String url = document.find("sourceid").getContent();
						String click = document.getParent().getAttributeValue("fulltext");
						recommendations.add(new RecommendationEntry(title, url, click));
					}

					return recommendations;
				}
				catch (Exception e) {
					e.printStackTrace();
				}
			}
			else if (response.getStatus() == Status.NO_CONTENT) {
				return null;
			}
			else if (response.getStatus() == Status.UNKNOWN_HOST) {
				throw new UnknownHostException("no connection");
			}			
			else {
				throw new UnexpectedException("unkown");
			}
		}
		else {
			throw new IllegalStateException("no username set");
		}
		return Collections.emptyList();
	}
}