Mercurial > hg > MetaDataManagerRestlet

view src/de/mpiwg/itgroup/nimanager/edit/Editor.java @ 4:aae47a713589

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
editing
author dwinter
date Mon, 02 Jan 2012 12:42:28 +0100
parents
children
line wrap: on
line source

package de.mpiwg.itgroup.nimanager.edit;

import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Level;
import org.apache.log4j.Logger;
import org.apache.log4j.lf5.util.StreamUtils;
import org.openrdf.model.Statement;
import org.openrdf.repository.RepositoryException;
import org.openrdf.repository.RepositoryResult;
import org.restlet.Context;
import org.restlet.data.ClientInfo;
import org.restlet.data.Form;
import org.restlet.data.MediaType;
import org.restlet.data.Status;
import org.restlet.engine.component.ChildContext;
import org.restlet.representation.Representation;
import org.restlet.representation.StringRepresentation;
import org.restlet.resource.Get;
import org.restlet.resource.Post;
import org.restlet.resource.ServerResource;
import org.restlet.security.User;

import de.mpiwg.itgroup.metaDataManagerRestlet.RestServer;
import de.mpiwg.itgroup.triplestoremanager.exceptions.TripleStoreHandlerException;
import de.mpiwg.itgroup.triplestoremanager.owl.MetaDataHandler;
import de.mpiwg.itgroup.triplestoremanager.owl.TripleStoreHandler;
import de.mpiwg.itgroup.triplestoremanager.owl.TripleStoreHandler.LiteralQuadruple;
import de.mpiwg.itgroup.triplestoremanager.owl.TripleStoreHandler.Quadruple;
import edu.stanford.smi.protegex.owl.model.impl.DefaultRDFProperty;
import edu.stanford.smi.protege.model.DefaultCls;
/**
 * @author dwinter
 *
 * Edit an Entity defined by OWL modell
 */
public class Editor extends ServerResource {
		
	private  String NAMED_ENTITIES_ONTOLOGIE_URL = "http://ontologies.mpiwg-berlin.mpg.de/authorities/namedIdentities";
	private  String VIRTUOSO_PW;
	private  String VIRTUOSO_USER;
	private  String virtuoso_server_url;
	private  String TRIPLE_INDEX_PATH;
	private  Logger rl = Logger.getRootLogger();
	private  MetaDataHandler mh;
	private  TripleStoreHandler th;
	private String editGraph="file:///personEdits.rdf";
	
	
	public Editor (){
		ChildContext context = (ChildContext)Context.getCurrent();
		virtuoso_server_url = context.getParameters().getFirstValue("de.mpwig.itgroup.personSearch.virtuoso.url");
		VIRTUOSO_PW = context.getParameters().getFirstValue("de.mpwig.itgroup.personSearch.virtuoso.pw");
		VIRTUOSO_USER = context.getParameters().getFirstValue("de.mpwig.itgroup.personSearch.virtuoso.user");
		rl.setLevel(Level.DEBUG);
		try {
			th = TripleStoreHandler.getInstance(virtuoso_server_url, VIRTUOSO_USER, VIRTUOSO_PW);

			//th = new TripleStoreHandler(virtuoso_server_url, VIRTUOSO_USER, VIRTUOSO_PW);
		} catch (TripleStoreHandlerException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
		mh = MetaDataHandler.getInstance();
	}
	
	public Editor(String serverUrl,String User, String PW){
		virtuoso_server_url = serverUrl;
		VIRTUOSO_PW =  PW;
		VIRTUOSO_USER = User;
		
		try {
			th = TripleStoreHandler.getInstance(virtuoso_server_url, VIRTUOSO_USER, VIRTUOSO_PW);
			//th = new TripleStoreHandler(virtuoso_server_url, VIRTUOSO_USER, VIRTUOSO_PW);
		} catch (TripleStoreHandlerException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
		mh = MetaDataHandler.getInstance();
	}
	/**
	 * Hole alle Properties eines Instance wie definiert in OWL.
	 * @param instance
	 * @return
	 */
	private HashMap<String, Object> getProperties(String instance, List<String> NSFilter){
		RepositoryResult<Statement> stmts;
		HashMap<String, Object> retMap = new HashMap<String, Object>();
		try {
			stmts= th.getStatements(instance, "http://www.w3.org/1999/02/22-rdf-syntax-ns#type", null); 
		} catch (RepositoryException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			return null;
		}
		try {
			while(stmts.hasNext()){
				Statement stmt = stmts.next();
				addPropsOfType(retMap,stmt.getObject().stringValue(),instance, NSFilter);
			}
		} catch (RepositoryException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			return null;
		}
		
		
		return retMap;
	}
	private void addPropsOfType(HashMap<String, Object> retMap,
			String cls, String instance, List<String> NSfilter) {
		
//		Set props = mh.getAssociatedPropsForClass(cls);
//		Iterator propsIt = props.iterator();
//		while (propsIt.hasNext()){
//			Object prop = propsIt.next();
//			
//			
//		}
		Collection props = mh.getAssociatedPropsForClass(cls);
	
		// filter namespaces
		if (NSfilter!=null){
			
		Object[] propsCopy = props.toArray();
		for (Object prop:propsCopy){
			if (DefaultRDFProperty.class.isInstance(prop)){
				DefaultRDFProperty rdfProp = (DefaultRDFProperty)prop;
				rl.debug(rdfProp.getNamespace());
				if (!NSfilter.contains(rdfProp.getNamespace())){ // nicht im filter
						props.remove(prop);
				}
		}
		}
		}
	
		Map<String, Object> newProps;
		try {
			newProps = th.getJenaRDFValues(props, instance, mh.getOwlModel(),true);
		} catch (RepositoryException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			return;
		}
		
		retMap.putAll(newProps);
		
		
	}
	
	@Post
	public Representation postHTML(Representation entity){
		User user = handleBasicAuthentification(entity);
		if (user==null){
			setStatus(Status.CLIENT_ERROR_UNAUTHORIZED);
			return null;
		}
		
		Form form = new Form(entity);
		
		String id = form.getFirstValue("id");
		if (id==null){
			setStatus(Status.CLIENT_ERROR_BAD_REQUEST);
			return new StringRepresentation("Parameter ID missing");
		}
		
		String type = form.getFirstValue("type");
		if (type==null){
			setStatus(Status.CLIENT_ERROR_BAD_REQUEST);
			return new StringRepresentation("Parameter type missing");
		}

		
		ArrayList<Quadruple> quads = new ArrayList<Quadruple>();
		ArrayList<Quadruple> deleteQuads = new ArrayList<Quadruple>();
		Map<String,String> formMap = form.getValuesMap();
		for (String key: formMap.keySet()){
			if (isDataTypeProperty(key,type)){
				String value=formMap.get(key);
				
				//remove white space
				value=StringUtils.stripEnd(value, null);
				value=StringUtils.stripStart(value, null);
				
				if (value!=null && !value.equals("")){ //not empty
					deleteQuads.add(new LiteralQuadruple(id, key, null, editGraph));
					quads.add(new LiteralQuadruple(id, key, formMap.get(key), editGraph));
				}
			}
		}
		try {
			th.remove(deleteQuads);
			th.write(quads);
		} catch (RepositoryException e) {
			setStatus(Status.SERVER_ERROR_INTERNAL);
			e.printStackTrace();
			return null;
		} catch (TripleStoreHandlerException e) {
			setStatus(Status.SERVER_ERROR_INTERNAL);
			e.printStackTrace();
			return null;
		}
		
		return new StringRepresentation("written");
		
	}
	
	/**
	 * Checks if the 
	 * @param key
	 * @param value 
	 * @return
	 */
	private boolean isDataTypeProperty(String key,String type) {
		
	
		Collection props = mh.getAssociatedPropsForClass(type);
		
			
		Object[] propsCopy = props.toArray();
		for (Object prop:propsCopy){
			if (DefaultRDFProperty.class.isInstance(prop)){
				DefaultRDFProperty rdfProp = (DefaultRDFProperty)prop;
				rl.debug(rdfProp.getNamespace());
				if (rdfProp.getName().equals(key)){
					rl.debug(rdfProp);
					return true;
				}
			}
		}
		return false;
	}
	

	@Get("html")
	public Representation getHTML(Representation entity){
		Form form = getRequest().getResourceRef().getQueryAsForm();
		
		
		String id  =form.getValuesMap().get("id");			
		try {
			id = URLDecoder.decode(id, "utf-8");
		} catch (UnsupportedEncodingException e) {
			e.printStackTrace();
			e.printStackTrace();
			setStatus(Status.SERVER_ERROR_INTERNAL, "REPOSITORY ERROR");
			return new StringRepresentation(
					"<xml><error>Unsupported encoding</error></xml>",
					MediaType.TEXT_HTML);
			
		}
		id=StringUtils.stripEnd(id, null);
		id=StringUtils.stripStart(id, null);
		
		ArrayList<String> NSFilter = new ArrayList<String>();
		NSFilter.add("http://xmlns.com/foaf/0.1/"); // only foaf
		HashMap<String, Object> props = getProperties(id, NSFilter);
		
		
		ArrayList<String> NSFilterRDF = new ArrayList<String>();
		NSFilterRDF.add("http://www.w3.org/1999/02/22-rdf-syntax-ns#"); // only foaf
		HashMap<String, Object> propsRDF = getProperties(id, NSFilterRDF);
		
		String RDFtype = ((DefaultCls)(propsRDF.get("http://www.w3.org/1999/02/22-rdf-syntax-ns#type"))).getName();
	
		
		String retString="<html><body><form method=\"post\" class=\"editEntity\">";
		retString+=String.format("<input type=\"hidden\" name=\"id\" value=\"%s\" len=\"50\">", id);
		retString+=String.format("<input type=\"hidden\" name=\"type\" value=\"%s\" len=\"50\">", RDFtype);
		for (String prop: props.keySet()){
			
			retString+=String.format("<div class=\"labelEntity\">%s<input type=\"text\" name=\"%s\" value=\"%s\" len=\"50\"></div>", 
					prop,prop,String.valueOf(props.get(prop)));
		}
		
		retString+="<input type=\"submit\"/>";
		retString+="</form></body></html>";
		return new StringRepresentation(retString,MediaType.TEXT_HTML);
		
	}
	
	
	private User handleBasicAuthentification(Representation entity) {
		RestServer restServer = (RestServer) getApplication();
		if (!restServer.authenticate(getRequest(), getResponse())) {
			// Not authenticated
			return null;
		}

		ClientInfo ci = getRequest().getClientInfo();
		rl.debug(ci);
		return getRequest().getClientInfo().getUser();

	}
	
	public static void main(String args[]) {
		Logger.getRootLogger().setLevel(Level.DEBUG);
		Editor ed = new Editor("jdbc:virtuoso://virtuoso.mpiwg-berlin.mpg.de:1111","dba","wa55er");
		ArrayList<String> NSFilter = new ArrayList<String>();
		NSFilter.add("http://xmlns.com/foaf/0.1/"); // only foaf
		HashMap<String, Object> props = ed.getProperties("http://ontologies.mpiwg-berlin.mpg.de/tempObjects/person/370868",NSFilter);
		System.out.println(props);
		
	}
}