korrekt søk, men dessverre for treigt

[cristin-ws.git] / src / main / java / no / cristin / ws / FinnForskerId.java

package no.cristin.ws;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Hashtable;
import java.util.List;

import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.Status;

import org.jboss.resteasy.annotations.cache.Cache;
import org.json.JSONException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Person RPC-tjeneste.
 */
@Path("/finnForsker")
public class FinnForskerId {

	private static final Logger log = LoggerFactory.getLogger(FinnForskerId.class.toString());
    @Context private HttpServletRequest servletRequest;

    // NB! hardkoding av cristin-url som kjem til å forandre seg!
    private static final String presURL = "http://www.cristin.no/as/WebObjects/cristin.woa/wa/personVis?type=PERSON&pnr=";

	/**
	 * Returnerer liste av personer som har matchende navn og institusjon.
	 * 
	 * @param navn Forskerens navn
	 * @param institusjon Forskerens institusjon
	 * @param format
	 *            "html", "xml" eller "json" for å angi output format
	 * @return liste over forskere som matcher input
	 */
	@GET
	@Cache(maxAge = 0)
	public Response finnForsker(
	        @QueryParam("navn") String navn,
			@QueryParam("inst") String institusjon,
			@QueryParam("format") String format) throws JSONException {
	    log.info("finnForsker " + navn + ":" + institusjon + ":" + format);

	    List<String> errors = new ArrayList<String>();

	    if (format == null) format = "xml";
        MediaType outputFormat = MediaType.APPLICATION_XML_TYPE;
        switch (format) {
            case "json":
                outputFormat = MediaType.APPLICATION_JSON_TYPE;
                break;
            case "html":
                outputFormat = MediaType.TEXT_HTML_TYPE;
                break;
            default:
                outputFormat = MediaType.APPLICATION_XML_TYPE;
        }

		if (navn == null) {
		   return Response.status(Status.BAD_REQUEST).entity("navn er null").build();
		}
		
		String sql = "select p.personlopenr as lopenr, p.presentasjonslopenr as presnr, p.fornavn ||' '|| p.etternavn as navn, "
                + "case when p.bilde is not null then 'J' else 'N' end as bilde, "
                + "i.institusjonsakronym as akronym, "
                + "i.institusjonsnavn_bokmal as instnavnb, i.institusjonsnavn_nynorsk as instnavnn, i.institusjonsnavn_engelsk as instnavne, "
                + "s.stednavn_bokmal as stednavnb, s.stednavn_nynorsk as stednavnn, s.stednavn_engelsk as stednavne "
                + "from frida.person p "
                + "left outer join frida.ansettelse a on a.personlopenr = p.personlopenr "
                + "left outer join frida.sted s on s.institusjonsnr = a.institusjonsnr and s.avdnr = a.avdnr and "
                + "                                 s.undavdnr = a.undavdnr and s.gruppenr = a.gruppenr "
                + "left outer join frida.institusjon i on i.institusjonsnr = a.institusjonsnr "
            + " where regexp_like(p.fornavn||' '||p.etternavn, ?, 'i') ";

        List<String> parameters = new ArrayList<String>();
        
        // legg til where for navn
        parameters.add(lagRegexp(navn.trim().toLowerCase()));
        
        if (institusjon != null) {
            sql += "and ( ";
            institusjon = institusjon.trim().toLowerCase();
            if (institusjon.indexOf(' ') == -1) {
                sql += "lower (i.institusjonsakronym) = ? or ";
                parameters.add(institusjon);
            }
            sql += "regexp_like(i.institusjonsnavn_bokmal||' '||i.institusjonsnavn_nynorsk||' '||i.institusjonsnavn_engelsk, ?,'i')) ";
            parameters.add(lagRegexp(institusjon));
        }


        Forskere forskere = new Forskere();
        try (
                Connection conn = Database.getConnection();
                PreparedStatement stmt = conn.prepareStatement(sql);
                ) {

            int index = 1;
            for (int idx = 1; idx <= parameters.size(); idx++) {
                log.info("finnForsker legger til param " + idx + ": " + parameters.get(idx - 1));
                stmt.setString(idx, parameters.get(idx - 1));
            }
            log.info("finnForsker " + sql + ", parametre: " + index);
            ResultSet rs = stmt.executeQuery();
            int results = 0;
            final String bildeURL = servletRequest.getRequestURL().substring(0,servletRequest.getRequestURL().indexOf("finnFo"));
            while (rs.next()) {
                forskere.leggTilEllerOppdater(
                        rs.getInt("lopenr"),
                        rs.getInt("presnr") == 0 ? null:  presURL + rs.getInt("presnr"),
                        rs.getString("navn"),
                        "N".equals(rs.getString("bilde")) ? null : bildeURL,
                        rs.getString("akronym"),
                        rs.getString("instnavnb"),
                        rs.getString("instnavnn"),
                        rs.getString("instnavne"),
                        rs.getString("stednavnb"),
                        rs.getString("stednavnn"),
                        rs.getString("stednavne")
                        );
            }
            
        } catch (SQLException e) {
            log.error("finnForsker error",e);
        }
            

        
        return Response
                .status(200)
                .entity(forskere.format(outputFormat))
                .header(HttpHeaders.CONTENT_TYPE, outputFormat+"; charset=UTF-8")
                .build();


	}

	private String lagRegexp(String navn) {
	    log.info("lagRegexp "+ navn);
	    StringBuilder sb = new StringBuilder();
	    String[] split = navn.split(" ");
	    sb.append("^(.+ )*");
	    sb.append(split[0]);
	    for (int i = 1; i < split.length; i++) {
	        sb.append(" (.+[[:space:]])*");
	        sb.append(split[i]);
	    }
        sb.append("( .+)*$");
        log.info("lagRegexp "+ sb.toString());
        return sb.toString();
    }

    private static final String escapeXML(String input) {
        return input.replace("&", "&amp;");
    }

    final private class Forskere {
        Hashtable<Integer,Forsker> forskere = new Hashtable<Integer,Forsker>();
        Object format(MediaType outputFormat) throws JSONException {
            StringBuilder sb = new StringBuilder(4000);
            sb.append("<forskere>");
            for (Forsker f: forskere.values()) {
                f.toXml(sb);
            }
            sb.append("</forskere>");
            if (outputFormat.equals(MediaType.APPLICATION_JSON_TYPE)) {
                
                String json = org.json.XML.toJSONObject(sb.toString()).toString();
                return json;
            }
            return sb.toString();
        }
        public void leggTilEllerOppdater(int lopenr, String presentasjonsURL, String navn,
                String bildeURL, String akronym,
                String instavnBokmål, String instavnNynorsk, String instavnEngelsk,
                String stednavnBokmål, String stednavnNynorsk, String stednavnEngelsk) {
            Sted s = akronym == null ? null : new Sted(akronym, instavnBokmål, instavnNynorsk, instavnEngelsk, stednavnBokmål, stednavnNynorsk, stednavnEngelsk);
            if (forskere.containsKey(lopenr)) {
                if (s != null) forskere.get(lopenr).tilhorigheter.add(s);
            } else {
                Forsker f = new Forsker(lopenr, presentasjonsURL, navn, bildeURL);
                if (s != null) f.tilhorigheter.add(s);
                forskere.put(lopenr, f);
            }
            
        }
        
    }
	
	final static private class Forsker {
	    private int lopenr;
        private String presentasjonsURL;
	    private String navn;
        private String bildeURL;
	    private List<Sted> tilhorigheter = new ArrayList<Sted>();

	    public Forsker (int lopenr, String presentasjonsURL, String navn, String bildeURL) {
	        this.lopenr = lopenr;
	        this.presentasjonsURL = presentasjonsURL;
	        this.navn = navn;
	        this.bildeURL = bildeURL;
	    }

        public void toXml(StringBuilder sb) {
            sb.append("<forsker ");
            sb.append("id=\"");
            sb.append(lopenr);
            sb.append("\" navn=\"");
            sb.append(navn);
            sb.append("\" ");
            if (presentasjonsURL != null) {
                sb.append("presentasjon=\"");
                sb.append(escapeXML(presentasjonsURL));
                sb.append(lopenr);
                sb.append("\" ");                
            }
            if (bildeURL != null) {
                sb.append("bilde=\"");
                sb.append(bildeURL);
                sb.append("hentPersonBilde?lopenr=");
                sb.append(lopenr);
                sb.append("\"");
            }
            sb.append(">");
            for (Sted s: tilhorigheter) {
                s.toXml(sb);
            }
            sb.append("</forsker>");
        }

	}
	final static private class Sted {
	    private String akronym;
        private String instnavnBokmål;
        private String instnavnNynorsk;
        private String instnavnEngelsk;
        private String stednavnBokmål;
        private String stednavnNynorsk;
        private String stednavnEngelsk;

        public Sted(String akronym, String instnavnBokmål, String instnavnNynorsk, String instnavnEngelsk,
                String navnBokmål, String navnNynorsk, String navnEngelsk) {
	        this.akronym = akronym;
            this.instnavnBokmål = instnavnBokmål;
            this.instnavnNynorsk = instnavnNynorsk;
            this.instnavnEngelsk = instnavnEngelsk;
	        this.stednavnBokmål = navnBokmål;
	        this.stednavnNynorsk = navnNynorsk;
	        this.stednavnEngelsk = navnEngelsk;
	    }

        public void toXml(StringBuilder sb) {
            sb.append("<sted akronym=\"");
            sb.append(escapeXML(akronym));
            sb.append("\" ");
            sb.append("instnavnb=\"");
            sb.append(escapeXML(instnavnBokmål));
            sb.append("\" ");
            if (instnavnNynorsk != null) sb.append(" instnavnn=\"" + escapeXML(instnavnNynorsk) + "\" ");
            if (instnavnEngelsk != null) sb.append(" instnavne=\"" + escapeXML(instnavnEngelsk) + "\" ");
            sb.append("stednavnb=\"");
            sb.append(escapeXML(stednavnBokmål));
            sb.append("\" ");
            if (stednavnNynorsk != null) sb.append(" stednavnn=\"" + escapeXML(stednavnNynorsk) + "\" ");
            if (stednavnEngelsk != null) sb.append(" stednavne=\"" + escapeXML(stednavnEngelsk) + "\" ");
            
            sb.append("/>");
        }
	}
	
}