source: trunk/gsdl3/src/java/org/greenstone/gsdl3/action/QueryAction.java@ 3568

package org.greenstone.gsdl3.action;

import org.greenstone.gsdl3.core.ModuleInterface;
import org.greenstone.gsdl3.util.*;
// XML classes
import org.w3c.dom.Node; 
import org.w3c.dom.NodeList; 
import org.w3c.dom.Text; 
import org.w3c.dom.Document; 
import org.w3c.dom.Element; 

import java.util.HashMap;
import java.util.Map;
import java.util.Iterator;
import java.io.File;

/** action class for queries */
public class QueryAction extends Action {
    
    public Element process (Element message) {

    // get the request - assume there is only one
    Element request = (Element)GSXML.getChildByTagName(message, GSXML.REQUEST_ELEM);

    // create the return page tree
    Element page = doc_.createElement(GSXML.PAGE_ELEM);
    page.setAttribute(GSXML.LANG_ATT, request.getAttribute(GSXML.LANG_ATT));
    // add the lang stuff from message
    page.appendChild(doc_.importNode(GSXML.getChildByTagName(message, GSXML.DISPLAY_ELEM), true));
    // add the system stuff from message
       page.appendChild(doc_.importNode(GSXML.getChildByTagName(message, GSXML.CONFIGURATION_ELEM), true));

    // query type is teh subaction
    String query_type = request.getAttribute(GSXML.SUBACTION_ATT);
    
    if (query_type.equals("")) {
        query_type = "text";
    }

    if (query_type.equals("text")) {
        return textQuery(page, request);
    }
    if (query_type.equals("field")) {
        return fieldQuery(page, request);
    }
    return unknownQuery(page, request, query_type);

    
    }
    
    // we may be able to generalize this for all query types?
    protected Element textQuery(Element page, Element request) {

    // check that the stylesheet is present - cant output a page without one. we may adapt this to use unknownquery stylesheet? - or ask for one from the MR
    String stylesheet = GSFile.stylesheetFile(config_, "textquery.xsl");
    if (stylesheet==null) {
        System.err.println("QueryAction Error: textquery stylesheet not found!");
        return null;
    }

    // extract the params from the cgi-request, and check that we have a coll specified
    Element cgi_param_list = (Element)GSXML.getChildByTagName(request, GSXML.PARAM_ELEM+GSXML.LIST_MODIFIER);
    HashMap params = GSXML.extractParams(cgi_param_list);

    String collection = (String)params.get("collection");
    if (collection == null || collection.equals("")) {
        System.err.println("QueryAction Error: no colleciton was specified!");
        return null; 
    }

    // get the service info from the MR - this will probably need to be cached somehow later on. and add the service node to the page 
    Element mr_info_message = doc_.createElement(GSXML.MESSAGE_ELEM);
    Element mr_info_request = doc_.createElement(GSXML.REQUEST_ELEM);
    mr_info_message.appendChild(mr_info_request);
    mr_info_request.setAttribute(GSXML.TYPE_ATT, GSXML.REQUEST_TYPE_DESCRIBE);
    mr_info_request.setAttribute(GSXML.LANG_ATT, page.getAttribute(GSXML.LANG_ATT));

    String to = collection;
    to = GSPath.appendLink(to, "TextQuery");
    mr_info_request.setAttribute(GSXML.TO_ATT, to);

    Element mr_info_response = (Element) mr_.process(mr_info_message);

    String path = GSXML.RESPONSE_ELEM;
    path = GSPath.appendLink(path, GSXML.SERVICE_ELEM);
    Element description = (Element)doc_.importNode(GSXML.getNodeByPath(mr_info_response, path), true);
    Element pl = (Element)GSXML.getChildByTagName(description, GSXML.PARAM_ELEM+GSXML.LIST_MODIFIER);
    
    if (pl !=null) {
        // add short names to the params in the param list
        cgi_.addShortNames(pl);

        // for each param in the description, overwrite teh default value with the currently set value if present 
        Element param = (Element)pl.getFirstChild();
        while (param !=null) {
        if (param.getNodeName().equals(GSXML.PARAM_ELEM)) { // just in case
            String name = param.getAttribute(GSXML.NAME_ATT);
            String current = (String)params.get(name);
            if (current !=null && !current.equals("")) {
            param.setAttribute(GSXML.DEFAULT_ATT, current);
            }
        }
        param = (Element)param.getNextSibling();
        }
    }
    
    // part of the data for the description is the cgi-params
    // if we have this here, do we need to do the previous step?
    Element cgi_request = (Element)doc_.importNode(request, true);
    page.appendChild(cgi_request);

    page.appendChild(description);
    
    // check the query string
    String query = (String)params.get(GSXML.REQUEST_TYPE_QUERY);

    if (query!=null&& !query.equals("")) {
        // do the query
        
        Element mr_query_message = doc_.createElement(GSXML.MESSAGE_ELEM);
        Element mr_query_request = doc_.createElement(GSXML.REQUEST_ELEM);
        mr_query_message.appendChild(mr_query_request);

        mr_query_request.setAttribute(GSXML.TYPE_ATT, GSXML.REQUEST_TYPE_QUERY);
        mr_query_request.setAttribute(GSXML.TO_ATT, to);
        mr_query_request.setAttribute(GSXML.LANG_ATT, page.getAttribute(GSXML.LANG_ATT));
        // paramList
        // should we remove the query from the param list?
        Element query_param_list = (Element)doc_.importNode(cgi_param_list, true);
        mr_query_request.appendChild(query_param_list);
        
        // content - contains the query string
        Element query_content = GSXML.createTextElement(doc_, GSXML.CONTENT_ELEM, query);
        mr_query_request.appendChild(query_content);
        
        Element mr_query_response = (Element)mr_.process(mr_query_message);
        
        // this result is the list of docs.
        // now take that list, and get the Titles
        // for now, just create a whole new request
        
        // check that there are some resources - for now check the list, but later should use a numdocs metadata elem
        path = GSXML.RESPONSE_ELEM;
        path = GSPath.appendLink(path, GSXML.CONTENT_ELEM);
        path = GSPath.appendLink(path, GSXML.RESOURCE_ELEM+GSXML.LIST_MODIFIER);
        
        Element resource_list = (Element)GSXML.getNodeByPath(mr_query_response,
                                 path); // resourceList not present if no docs found
        if (resource_list == null) {
        
        Element result_response = (Element)GSXML.getChildByTagName(mr_query_response, GSXML.RESPONSE_ELEM);
        
        page.appendChild(doc_.importNode(result_response, true));
        
        Document style_doc = converter_.getDOM(new File(stylesheet));
        GSXSLT.absoluteIncludePaths(style_doc, config_);
        return (Element)transformer_.transform(style_doc, page);    
        
        }
        
        // we have a doc list, so get the metadata - for now, get title.
        // can we dynamically decide what metadata to get?
        Element mr_metadata_message = doc_.createElement(GSXML.MESSAGE_ELEM);
        Element mr_metadata_request = doc_.createElement(GSXML.REQUEST_ELEM);
        mr_metadata_message.appendChild(mr_metadata_request);
        
        mr_metadata_request.setAttribute(GSXML.TYPE_ATT, GSXML.REQUEST_TYPE_QUERY);
        mr_metadata_request.setAttribute(GSXML.LANG_ATT, page.getAttribute(GSXML.LANG_ATT));
        to = GSPath.appendLink(collection, "MetadataRetrieve");
        mr_metadata_request.setAttribute(GSXML.TO_ATT, to);
        
        Element meta_content = doc_.createElement(GSXML.CONTENT_ELEM);
        mr_metadata_request.appendChild(meta_content);
        
        // the first part of the content is the doc list
        meta_content.appendChild(doc_.importNode(resource_list, true));
        
        // the second part of the content is the metadata list
        Element metadata_list = doc_.createElement(GSXML.METADATA_ELEM+GSXML.LIST_MODIFIER);
        Element title = doc_.createElement(GSXML.METADATA_ELEM);
        title.setAttribute(GSXML.NAME_ATT, "Title");
        metadata_list.appendChild(title);
        meta_content.appendChild(metadata_list);

        Element mr_metadata_response = (Element)mr_.process(mr_metadata_message);
        
        Element result_response = (Element)GSXML.getChildByTagName(mr_metadata_response, GSXML.RESPONSE_ELEM);
        
        page.appendChild(doc_.importNode(result_response, true));
    
    } // do teh query

    
    // output the page
    // process using the stylesheet
    Document style_doc = converter_.getDOM(new File(stylesheet));
    GSXSLT.absoluteIncludePaths(style_doc, config_);
    return (Element)transformer_.transform(style_doc, page);    
    
    }

    protected Element fieldQuery(Element page, Element request) {
    String p =  GSHTML.errorPage("field query not implemented");

    return converter_.getDOM(p).getDocumentElement();
    }
    protected Element unknownQuery(Element page, Element request, String query_type) {

    String p =  GSHTML.errorPage("unknown query subtype - "+query_type+" query not implemented");
    return null;//converter_.getDOM(p).getDocumentElement();
    }
}