Metacat

/**

 *  '$RCSfile$'

 *    Purpose: A Class that implements org.xml.sax.EntityResolver interface

 *             for resolving external entities

 *  Copyright: 2000 Regents of the University of California and the

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: Jivka Bojilova, Matt Jones

 *    Release: @release@

 *

 *   '$Author: bojilova $'

 *     '$Date: 2000-12-12 13:29:29 -0800 (Tue, 12 Dec 2000) $'

 * '$Revision: 599 $'

 */

package edu.ucsb.nceas.metacat;

import org.xml.sax.*;

import org.xml.sax.helpers.DefaultHandler;

import java.sql.*;

import java.io.File;

import java.io.Reader;

import java.io.BufferedReader;

import java.io.FileWriter;

import java.io.BufferedWriter;

import java.io.InputStream;

import java.io.IOException;

import java.net.URL;

import java.net.URLConnection;

import java.net.MalformedURLException;

/** 

 * A database aware Class implementing EntityResolver interface for the SAX 

 * parser to call when processing the XML stream and intercepting any 

 * external entities (including the external DTD subset and external 

 * parameter entities, if any) before including them.

 */

public class DBEntityResolver implements EntityResolver

{

  private Connection conn = null;

  private DefaultHandler handler = null;

  private String docname = null;

  private String doctype = null;

  private String systemid = null;

  private Reader dtdtext = null;

  /** 

   * Construct an instance of the DBEntityResolver class

   *

   * @param conn the JDBC connection to which information is written

   */

  public DBEntityResolver(Connection conn)

  {

    this.conn = conn;

  }

  /** 

   * Construct an instance of the DBEntityResolver class

   *

   * @param conn the JDBC connection to which information is written

   * @param handler the SAX handler to determine parsing context

   * @param dtd Reader of new dtd to be uploaded on server's file system

   */

  public DBEntityResolver(Connection conn, DefaultHandler handler, Reader dtd)

  {

    this.conn = conn;

    this.handler = handler;

    this.dtdtext = dtd;

  }

  /** 

   * The Parser call this method before opening any external entity 

   * except the top-level document entity (including the external DTD subset,

   * external entities referenced within the DTD, and external entities 

   * referenced within the document element)

   */

  public InputSource resolveEntity (String publicId, String systemId)

                     throws SAXException

  {

    String dbSystemID;

    String doctype = null;

    // Won't have a handler under all cases

    if ( handler != null ) {

      if ( handler instanceof DBSAXHandler ) {

        DBSAXHandler dhandler = null;

        dhandler = (DBSAXHandler)handler;

        if ( dhandler.processingDTD() ) {

          // public ID is doctype

          if (publicId != null) {   

            doctype = publicId;

          // assume public ID (doctype) is docname

          } else if (systemId != null) {

            doctype = dhandler.getDocname();

          }

        }

      } else if ( handler instanceof AccessControlList ) {

        AccessControlList ahandler = null;

        ahandler = (AccessControlList)handler;

        //if ( ahandler.processingDTD() ) {

          // public ID is doctype

          if (publicId != null) {   

            doctype = publicId;

          // assume public ID (doctype) is docname

          } else if (systemId != null) {

            doctype = ahandler.getDocname();

          }

        //}

      }

    }

    // get System ID for doctype

    if (doctype != null) {

      // look at db XML Catalog for System ID 

      dbSystemID = getDTDSystemID(doctype);

      boolean doctypeIsInDB = true;

      // no System ID found in db XML Catalog

      if (dbSystemID == null) {

        doctypeIsInDB = false;

        // use the provided System ID

        if (systemId != null) {

          dbSystemID = systemId;

        }

      }

      // there are dtd text provided; try to upload on Metacat

      if ( dtdtext != null ) {

        dbSystemID = uploadDTD(dbSystemID);

      }

      // check URLConnection first

      InputStream istream = checkURLConnection(dbSystemID);

      // need to register System ID in db XML Catalog if not yet

      if ( !doctypeIsInDB ) {

        registerDTD(doctype, dbSystemID);

      }

      // return a byte-input stream for use

      InputSource is = new InputSource(dbSystemID); 

      //is.setPublicId(doctype);

      //is.setSystemId(dbSystemID);

      is.setByteStream(istream);

      return is;

    } else {

      // use the provided systemId for other cases

      InputStream istream = checkURLConnection(systemId);

      return null;

    }

  }

  /** 

   * Look at db XML Catalog to get System ID (if any) for @doctype.

   * Return null if there are no System ID found for @doctype

   */

  private String getDTDSystemID( String doctype )

                 throws SAXException

  {

    String systemid = null;

    Statement stmt;

    try {

      stmt = conn.createStatement();

      stmt.execute("SELECT system_id FROM xml_catalog " + 

                   "WHERE entry_type = 'DTD' AND public_id = '" +

                   doctype + "'");

      ResultSet rs = stmt.getResultSet();

      boolean tableHasRows = rs.next();

      if (tableHasRows) {

        systemid = rs.getString(1);

      }

      stmt.close();

    } catch (SQLException e) {

      throw new SAXException

      ("DBEntityResolver.getDTDSystemID(): " + e.getMessage());

    }

    // return the selected System ID

    return systemid;

  }

  /** 

   * Register new DTD identified by @systemId in Metacat XML Catalog 

   * . make a reference with @systemId for @doctype in Metacat DB

   */

  private void registerDTD ( String doctype, String systemId )

                 throws SAXException

  {

    // make a reference in db catalog table with @systemId for @doctype

    try {

      PreparedStatement pstmt;

      pstmt = conn.prepareStatement(

             "INSERT INTO xml_catalog " +

             "(catalog_id, entry_type, public_id, system_id) " +

             "VALUES (null, 'DTD', ?, ?)");

      // Bind the values to the query

      pstmt.setString(1, doctype);

      pstmt.setString(2, systemId);

      // Do the insertion

      pstmt.execute();

      pstmt.close();

    } catch (SQLException e) {

      throw new SAXException

      ("DBEntityResolver.registerDTD(): " + e.getMessage());

    }

  }

  /** 

   * Upload new DTD text identified by @systemId on Metacat file system

   */

  private String uploadDTD ( String systemId )

                 throws SAXException

  {

    MetaCatUtil util = new MetaCatUtil();

    String dtdPath = util.getOption("dtdPath");

    String dtdURL = util.getOption("dtdURL");

    // get filename from systemId

    String filename = systemId;

    int slash = Math.max(filename.lastIndexOf('/'), filename.lastIndexOf('\\'));

    if ( slash > -1 ) {

      filename = filename.substring(slash + 1);

    }

    // writing dtd text on Metacat file system as filename

    try {

      // create a buffering character-input stream

      // that uses a default-sized input buffer

      BufferedReader in = new BufferedReader(dtdtext);

      // open file writer to write the input into it

      //String dtdPath = "/opt/tomcat/webapps/bojilova/dtd/";

      File f = new File(dtdPath, filename);

      try {

        if ( f.exists() && !f.canWrite() ) {

          throw new IOException("Not writable: " + f.getCanonicalFile());

        }

      } catch (SecurityException se) {

        // if a security manager exists,

        // its checkRead method is called for f.exist()

        // or checkWrite method is called for f.canWrite()

        throw se;

      }

      // create a buffered character-output stream

      // that uses a default-sized output buffer

      FileWriter fw = new FileWriter(f);

      BufferedWriter out = new BufferedWriter(fw);

      // read the input and write into the file writer

	    String inputLine;

	    while ( (inputLine = in.readLine()) != null ) {

	      out.write(inputLine, 0, inputLine.length());

  	    out.newLine(); //instead of out.write('\r\n');

	    }

      // the input and the output streams must be closed

	    in.close();

	    out.flush();

	    out.close();

	    fw.close();

    } catch (MalformedURLException e) {

      throw new SAXException

      ("DBEntityResolver.uploadDTD(): " + e.getMessage());

    } catch (IOException e) {

      throw new SAXException

      ("DBEntityResolver.uploadDTD(): " + e.getMessage());

    } catch (SecurityException e) {

      throw new SAXException

      ("DBEntityResolver.uploadDTD(): " + e.getMessage());

    }

    //String dtdURL = "http://dev.nceas.ucsb.edu/bojilova/dtd/";

    return  dtdURL + filename;

  }

  /** 

   * Check URL Connection for @systemId, and return an InputStream

   * that can be used to read from the systemId URL.  The parser ends

   * up using this via the InputSource to read the DTD.

   *

   * @param systemId a URI (in practice URL) to be checked and opened

   */

  private InputStream checkURLConnection (String systemId)

                      throws SAXException

  {

    try {

      return (new URL(systemId).openStream());

    } catch (MalformedURLException e) {

      throw new SAXException

      ("DBEntityResolver.checkURLConnection(): " + e.getMessage());

    } catch (IOException e) {

      throw new SAXException

      ("DBEntityResolver.checkURLConnection(): " + e.getMessage());

    }    

  }   

}

/**

 * '$Log$

 * 'Revision 1.15  2000/08/14 20:53:33  jones

 * 'Added "release" keyword to all metacat source files so that the release

 * 'number will be evident in software distributions.

 * '

 * 'Revision 1.14  2000/06/30 00:52:04  jones

 * 'changed char stream to byte stream

 * '

 * 'Revision 1.13  2000/06/29 23:27:08  jones

 * 'Fixed bug in DBEntityResolver so that it now properly delegates to

 * 'the system id found inthe database.

 * 'Changed DBValidate to use DBEntityResolver, rather than the OASIS

 * 'catalog, and to return validation results in XML format.

 * '

 * 'Revision 1.12  2000/06/27 04:31:07  jones

 * 'Fixed bugs associated with the new UPDATE and DELETE functions of

 * 'DBWriter.  There were problematic interactions between some static

 * 'variables used in DBEntityResolver and the way in which the

 * 'Servlet objects are re-used across multiple client invocations.

 * '

 * 'Generally cleaned up error reporting.  Now all errors and success

 * 'results are reported as XML documents from MetaCatServlet.  Need

 * 'to make the command line tools do the same.

 * '

 * 'Revision 1.11  2000/06/26 10:35:04  jones

 * 'Merged in substantial changes to DBWriter and associated classes and to

 * 'the MetaCatServlet in order to accomodate the new UPDATE and DELETE

 * 'functions.  The command line tools and the parameters for the

 * 'servlet have changed substantially.

 * '

 * 'Revision 1.10.2.2  2000/06/25 23:38:16  jones

 * 'Added RCSfile keyword

 * '

 * 'Revision 1.10.2.1  2000/06/25 23:34:17  jones

 * 'Changed documentation formatting, added log entries at bottom of source files

 * ''

 */