source: gli/trunk/src/org/greenstone/gatherer/greenstone3/ServletConfiguration.java@ 14538

/**
 *#########################################################################
 *
 * A component of the Gatherer application, part of the Greenstone digital
 * library suite from the New Zealand Digital Library Project at the
 * University of Waikato, New Zealand.
 *
 * <BR><BR>
 *
 * Author: John Thompson, Greenstone Digital Library, University of Waikato
 *
 * <BR><BR>
 *
 * Copyright (C) 1999 New Zealand Digital Library Project
 *
 * <BR><BR>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * <BR><BR>
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * <BR><BR>
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *########################################################################
 */
package org.greenstone.gatherer.greenstone3;

import java.awt.*;
import java.io.*;
import java.lang.ref.*;
import java.lang.Thread;
import java.net.*;
import java.util.*;
import javax.swing.*;
import javax.swing.plaf.*;
import org.greenstone.gatherer.Gatherer;
import org.greenstone.gatherer.DebugStream;
import org.greenstone.gatherer.util.ArrayTools;
import org.greenstone.gatherer.util.StaticStrings;
import org.greenstone.gatherer.util.Utility;
import org.greenstone.gatherer.util.XMLTools;
import org.greenstone.gatherer.remote.RemoteGreenstoneServer;
import org.w3c.dom.*;

/** This class stores the mapping between sites and servlets available for them. The info is read in from the web.xml file.
 * @author Katherine Don, New Zealand Digital Library, University of Waikato
 * @version 
 */
public class ServletConfiguration {
    
    static public String ADD_COMMAND = "?a=s&sa=a&st=collection&sn=";
    static public String DEACTIVATE_COMMAND = "?a=s&sa=d&st=collection&sn=";
    private String gsdl3_path = "";
    private ArrayList sites = null;
    private HashMap mappings = null;

    public ServletConfiguration(String gsdl3_path) throws InterruptedException {
    
    this.gsdl3_path = gsdl3_path;

    if (Gatherer.isGsdlRemote){
        // if (RemoteGreenstoneServer.downloadWebXMLFile().equals("")) {
        //System.err.println("Error: Could not download web.xml.");
        //System.exit(0);
        //}
        System.out.println("Starting download web.xml.");
        long startTime = System.currentTimeMillis();
        long patience = 1000 * 10;
        Thread t = new Thread(new downloadWebXMl());
        t.start();

        System.out.println("Waiting for download thread to finish");
        while (t.isAlive()) {
        System.out.println("Still waiting...");
        //Wait maximum of 1 second for MessageLoop thread to
            //finish.
        t.join(1000);
        if (((System.currentTimeMillis() - startTime) > patience) &&
                    t.isAlive()) {
            //t.interrupt();
            //Shouldn't be long now -- wait indefinitely
            //t.join();
            System.out.print("Have been waiting 10 seconds for connecting to the server, do you want to terminate the connection?(y/n)");
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            try {
            if (br.readLine().toLowerCase().startsWith("y")){
                 System.out.println("Terminate the connection! ");
                 System.exit(0);
            }else{
                startTime = System.currentTimeMillis();
            };
            } catch (IOException ioe) {
            System.out.println("IO error trying to read your answer!");
            System.exit(1);
            }
        }
        }
        System.out.println("The web.xml file has been downloaded");
    }
    
    //String web_xml_path = gsdl3_path + File.separator + "web" + File.separator + "WEB-INF"+ File.separator + "web.xml";
    File web_xml;
    if (Gatherer.isGsdlRemote){
        web_xml = new File(gsdl3_path + "web.xml");
    }else{
        web_xml = new File(gsdl3_path + "WEB-INF"+ File.separator + "web.xml");
    }
    if (!web_xml.exists()) {
        DebugStream.println("Error: no web.xml found at "+web_xml.toString());
        return;
    }

    this.sites = new ArrayList();
    if (Gatherer.isGsdlRemote){
        String sites_on_server = RemoteGreenstoneServer.getSiteNames();
        if (sites_on_server.equals("")) {
        // !! Something went wrong : could not get names of the sites
        System.err.println("Error: Could not get names of the sites.");
        System.exit(0);
        }
       
        String[] sites_arr=sites_on_server.split("-----");
        for (int i=0; i<sites_arr.length; i++){
        if (!(sites_arr[i].trim().equals(""))){
            this.sites.add(sites_arr[i].trim());
        }
        }
    }else{
        // find the sites
        File start = new File(Utility.getSitesDir(gsdl3_path));
        File possible_sites[] = start.listFiles();
        ArrayTools.sort(possible_sites);
        for (int i=0; possible_sites != null && i < possible_sites.length; i++) {
        File site_config = new File(possible_sites[i], "siteConfig.xml");
        if (site_config.exists()) {
            this.sites.add(possible_sites[i].getName());
        }
        }
    }

    this.mappings = new HashMap();
    Document web_config = XMLTools.parseXMLFile(web_xml.getAbsolutePath(), false);
       
    NodeList servlet_mappings = web_config.getElementsByTagName("servlet-mapping");
    // make a little map class
    HashMap url_mappings = new HashMap();
    for (int i=0; i<servlet_mappings.getLength(); i++) {
        Element map = (Element)servlet_mappings.item(i);
        String name = XMLTools.getValue(XMLTools.getNodeFromNamed(map, "servlet-name"));
        String pattern = XMLTools.getValue(XMLTools.getNodeFromNamed(map, "url-pattern"));
        url_mappings.put(name, pattern);

    }

    NodeList servlets = web_config.getElementsByTagName("servlet");
    for (int i=0; i<servlets.getLength(); i++) {
        Element servlet = (Element)servlets.item(i);
        String name = XMLTools.getValue(XMLTools.getNodeFromNamed(servlet, "servlet-name"));
        String description = XMLTools.getValue(XMLTools.getNodeFromNamed(servlet, "description"));
        String site = getServletSite(servlet);
           
        if (site != null) {
        ArrayList this_site = (ArrayList)mappings.get(site);
        if (this_site == null) {
            this_site = new ArrayList();
        }
        String url = (String)url_mappings.get(name);
        this_site.add(url);
        mappings.put(site, this_site);
        }
        
    }

    web_config = null;
    url_mappings = null;
    }
    
    public ArrayList getSites() {
    return this.sites;
    }

    public ArrayList getServletsForSite(String site) {
    return (ArrayList)this.mappings.get(site);
    }
    public String getServletPath(String site) {
    
    // for now, get the first one
    ArrayList servlets = (ArrayList)mappings.get(site);
    if(servlets == null) {
        return null;
    }
    return (String)servlets.get(0);
    }

    private String getServletSite(Element servlet) {
    
    NodeList params = servlet.getElementsByTagName("init-param");
    if ( params == null || params.getLength() == 0) {
        return null;
    }
    String site = null;
    for (int i=0; i<params.getLength()&& site == null; i++) {
        String p_name = XMLTools.getValue(XMLTools.getNodeFromNamed(params.item(i), "param-name"));
        if (p_name.equals("site_name")) {
        site = XMLTools.getValue(XMLTools.getNodeFromNamed(params.item(i), "param-value"));
        }
    }
    return site;
    }

    private static class downloadWebXMl implements Runnable {
    public void run() {
        if (RemoteGreenstoneServer.downloadWebXMLFile().equals("")) {
        System.err.println("Error: Could not download web.xml.");
        System.exit(0);
        }
    }
    }

}