source: trunk/gsdl3/src/java/org/greenstone/gsdl3/gs3build/notifier/NotifierManager.java@ 8633

package org.greenstone.gsdl3.gs3build.notifier;



import java.io.*;
import java.net.*;
import java.sql.*;
import java.util.Calendar;
import java.util.Date;
import java.text.DateFormat;
import java.text.SimpleDateFormat;

import org.greenstone.gsdl3.gs3build.CollectionManager;
import org.greenstone.gsdl3.gs3build.util.GS3SQLConnection;

public class NotifierManager {

    public NotifierManager() {
    }
    
    public void detectEvents(CollectionManager collManager) {
        
        String collectionName = collManager.getCollectionName();
        String host = "http://localhost:8080/soap/servlet/rpcrouter";
        
        String lastBuild = dateToSqlString(collManager.getBuildDate());
        Statement statement = collManager.getDatabase().createStatement();
        try {
            // detect all new documents. A document is new if and only if
            // AccessionDate >= CollectionLastRebuiltDate
            ResultSet results = statement.executeQuery("SELECT DocID FROM document WHERE DocType != 'GSMETADATA' AND AccessionDate >= " + lastBuild);
            notify(collectionName, host, results, collManager.getDatabase());
            
            // detect modified documents. A document is modified if and only if
            // AccessionDate < CollectionLastRebuiltDate (ie, it is not new) and 
            // ModifiedDate >= IndexedDate
            results = statement.executeQuery("SELECT DocID FROM document WHERE DocType != 'GSMETADATA' AND AccessionDate < " + lastBuild + " AND ModifiedDate >= IndexedDate");
            notify(collectionName, host, results, collManager.getDatabase());
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    /**
     * @param collName
     * @param hostURL
     * @param results
     * @throws MalformedURLException
     * @throws IOException
     * @throws ProtocolException
     * @throws SQLException
     */
    private void notify(String collName, String hostURL, ResultSet results, GS3SQLConnection database) {
        try {
            URL url = new URL("http://localhost:8080/alerting/service?receive");
            System.out.println("trying to POST to " + url);
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            conn.setDoOutput(true);
            conn.setRequestMethod("POST");
            conn.setUseCaches(false);
            while(results.next()) {
                System.out.println("new document: " + results.getString("DocID"));
                try {
                    conn.connect();
                    OutputStream toReceiver = conn.getOutputStream();
                    OutputStreamWriter writeToReceiver = new OutputStreamWriter(toReceiver);
                    
                    // TODO HTTP Post -> SOAP
                    String documentID = results.getString("DocID");
                    
                    writeToReceiver.write("type=new_document&");
                    writeToReceiver.write("collectionID=" + collName + "&");
                    writeToReceiver.write("host_url=" + hostURL + "&");
                    writeToReceiver.write("document_title=" + getDocumentTitle(documentID, database) + "&");
                    writeToReceiver.write("documentID=" + documentID + "\n");
                    writeToReceiver.close();
                    
                    System.out.println(conn.getResponseCode());
                    InputStream fromReceiver = conn.getInputStream();
                    int i;
                    while ((i = fromReceiver.read()) != -1) System.out.write(i);
                    conn.disconnect();
                } catch (IOException e1) {
                    // TODO Auto-generated catch block
                    System.out.println("Error talking to server.");
                    e1.printStackTrace();
                }
            }
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    /**
     * @param documentID
     * @return
     * @throws SQLException
     */
    private String getDocumentTitle(String documentID, GS3SQLConnection database) throws SQLException {
        Statement statement = database.createStatement();
        ResultSet results = statement.executeQuery("select mdvalues.value " +
                "from mdvalues join namespaces on (mdvalues.namespaceref = namespaces.namespaceid) " +
                "join metadata on (metadata.metadataref = namespaces.metadataref) " +
                "where label like 'Title' and metadata.docID = '" + documentID +"';");
        return results.getString(0);
    }

    /**
     * @param date
     * @return
     */
    private String dateToSqlString(Date date) {
        return new SimpleDateFormat("yyyyMMddHHmmss").format(date);
    }
}