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: tao $'

 *     '$Date: 2002-05-23 20:48:20 -0700 (Thu, 23 May 2002) $'

 * '$Revision: 1131 $'

 *

 * 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.

 *

 * 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.

 *

 * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 */

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.BufferedInputStream;

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()

  {

    //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(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);

      }

      // open URLConnection to check first

      InputStream istream = checkURLConnection(dbSystemID);

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

      if ( !doctypeIsInDB ) {

        // new DTD from outside URL location; try to upload on Metacat

        if ( dtdtext == null ) {

          dbSystemID = uploadDTDFromURL(istream, dbSystemID);

        }

        registerDTD(doctype, dbSystemID);

      }

      // return a byte-input stream for use

      InputSource is = new InputSource(dbSystemID); 

      // close and open URLConnection again

      try {

        istream.close();

      } catch (IOException e) {

        throw new SAXException 

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

      }    

      istream = checkURLConnection(dbSystemID);

      is.setByteStream(istream);

      return is;

    } else {

      // use provided systemId for the 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 = null;

    DBConnection conn = null;

    int serialNumber = -1;

    try {

      //check out DBConnection

      conn=DBConnectionPool.getDBConnection("DBEntityResolver.getDTDSystemID");

      serialNumber=conn.getCheckOutSerialNumber();

      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());

    }

    finally

    {

      DBConnectionPool.returnDBConnection(conn, serialNumber);

    }//finally

    // 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

  {

    DBConnection conn = null;

    int serialNumber = -1;

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

    try {

      //check out DBConnection

      conn=DBConnectionPool.getDBConnection("DBEntityResolver.registerDTD");

      serialNumber=conn.getCheckOutSerialNumber();

      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());

    }

    finally

    {

      DBConnectionPool.returnDBConnection(conn, serialNumber);

    }

  }

  /** 

   * Upload new DTD text identified by @systemId to 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);

     synchronized (f) { 

      try {

        if ( f.exists() ) {

          throw new IOException("File already exist: " + f.getCanonicalFile());

        //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();

     } // end of synchronized      

    } 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;

  }

  /** 

   * Upload new DTD located at outside URL to Metacat file system

   */

  private String uploadDTDFromURL(InputStream istream, 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

      BufferedInputStream in = new BufferedInputStream(istream);

      // open file writer to write the input into it

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

      File f = new File(dtdPath, filename);

     synchronized (f) { 

      try {

        if ( f.exists() ) {

          throw new IOException("File already exist: " + f.getCanonicalFile());

        //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

	    int inputByte;

	    while ( (inputByte = in.read()) != -1 ) {

	      out.write(inputByte);

  	    //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();

     } // end of synchronized

    } catch (MalformedURLException e) {

      throw new SAXException

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

    } catch (IOException e) {

      throw new SAXException

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

    } catch (SecurityException e) {

      throw new SAXException

      ("DBEntityResolver.uploadDTDFromURL(): " + 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());

    }    

  }   

}