Metacat

/**

 *        Name: DBEntityResolver.java

 *     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

 *

 *     Version: '$Id: DBEntityResolver.java 109 2000-05-31 00:56:56Z bojilova $'

 */

package edu.ucsb.nceas.metacat;

import org.xml.sax.*;

import java.sql.*;

import java.net.URL;

import java.net.MalformedURLException;

import java.util.Stack;

import java.util.EmptyStackException;

/** 

 * 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

{

   static String doctype = null;

   private Connection conn = null;

   private int pIdCounter = 0;

   private long currentElementNo;

   /** Construct an instance of the DBEntityResolver clas

    *

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

    */

   public DBEntityResolver(Connection conn)

   {

      this.conn = conn;

   }

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

   {

     String dbSystemId;

     currentElementNo = DBSAXHandler.elementNo;

     if (publicId != null) {

        pIdCounter += 1;

        System.out.println("from DBEntityResolver: current element is " + DBSAXHandler.elementNo);

        System.out.println("from DBEntityResolver: " + pIdCounter + " " + publicId);

        // look at the db XML Catalog and get dbSystemId by this publicId

        if (currentElementNo == 0) {

            doctype = publicId;

            dbSystemId = getDTDSystemID (conn, publicId);

            if (dbSystemId == "")

                // register publicId in db and use the provided systemId

                if (systemId != "") {

                    new URL(systemId);

                    registerDTDSystemID (conn, doctype, publicId, systemId);

                    return null;

                }

            new URL(dbSystemId);

            return new InputSource(dbSystemId);

        } 

        /*else {

            // look at the db XML Catalog and get dbSystemId by this publicId for a given doctype

            dbSystemId = getEntitySystemID (conn, doctype, publicId);

            if (dbSystemId == "")

                // register publicId in db for a given doctype and use the provided systemId

                if (systemId != "") {

                    new URL(systemId);

                    registerEntityPublicID (conn, doctype, publicId, systemId);

                    return null;

                }

            new URL(dbSystemId);

            return new InputSource(dbSystemId);

        }*/  

     }

     // publicId is null => doctype is null => doctype = docname

     if ( systemId != null) {

        if (currentElementNo == 0) {

            doctype = DBSAXHandler.docname;

            dbSystemId = getDTDSystemID (conn, doctype);

            if (dbSystemId == "") {

                new URL(systemId);

                registerDTDSystemID (conn, doctype, doctype, systemId);

                return null;

            }    

            new URL(dbSystemId);

            return new InputSource(dbSystemId);

        }

     }

     // use the default behaviour

     return null;

   }

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

     * Return empty string if there are not */

   private String getDTDSystemID (Connection conn, String doctype)  {

        String system_id = "";

        Statement stmt;

        try {

          stmt = conn.createStatement();

          System.out.println("DOCTYPE:" + doctype);

          stmt.execute("SELECT system_id FROM xml_catalog " + 

                       "WHERE entity_type = 'DTD' AND public_id = '" + doctype + "'");

          try {

            ResultSet rs = stmt.getResultSet();

            try {

              boolean tableHasRows = rs.next();

              if (tableHasRows) {

                try {

                  system_id = rs.getString(1);

                } catch (SQLException e) {

                  System.out.println("DBEntityResolver.getDTDSystemID() - Error with getString: " + e.getMessage());

                }

              }

            } catch (SQLException e) {

              System.out.println("DBEntityResolver.getDTDSystemID() - Error with next: " + e.getMessage());

            }

          } catch (SQLException e) {

            System.out.println("DBEntityResolver.getDTDSystemID() - Error with getrset: " + e.getMessage());

          }

          stmt.close();

        } catch (SQLException e) {

          System.out.println("DBEntityResolver.getDTDSystemID() - Error getting id: " + e.getMessage());

          System.exit(1);

        }

        // return the selected System ID

        return system_id;

   }

   /** Register DTD System ID in db XML Catalog */

   private void registerDTDSystemID (Connection conn, String doctype, String publicId, String systemId)

   {

        try {

            java.net.URLConnection urlConn = (new URL(systemId)).openConnection();

            urlConn.connect();

        } catch (java.io.IOException e) {

            System.out.println("IOException: " + e.getMessage());

            return;

        }    

        try {

          conn.setAutoCommit(false);

          PreparedStatement pstmt;

          pstmt = conn.prepareStatement(

                "INSERT INTO xml_catalog (entity_id, entity_type, source_doctype, public_id, system_id) " +

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

          // Bind the values to the query

          pstmt.setString(1, doctype);

          pstmt.setString(2, publicId);

          pstmt.setString(3, systemId);

          // Do the insertion

          pstmt.execute();

          pstmt.close();

          conn.commit();

          conn.setAutoCommit(true);

        } catch (SQLException e) {

          System.out.println(e.getMessage());

        }

   }

   /** Look at db XML Catalog to get System ID (if any) for that Public ID and doctype.

     * Return empty string if there are not */

   private String getEntitySystemID (Connection conn, String doctype, String publicId)

   {

        String system_id = "";

        Statement stmt;

        try {

          stmt = conn.createStatement();

          stmt.execute("SELECT system_id FROM xml_catalog " + 

                       "WHERE entity_type = 'ENTITY' AND source_doctype = '" + doctype + "' AND public_id = '" + publicId + "'");

          try {

            ResultSet rs = stmt.getResultSet();

            try {

              boolean tableHasRows = rs.next();

              if (tableHasRows) {

                try {

                  system_id = rs.getString(1);

                } catch (SQLException e) {

                  System.out.println("DBEntityResolver.getEntitySystemID() - Error with getString: " + e.getMessage());

                }

              }

            } catch (SQLException e) {

              System.out.println("DBEntityResolver.getEntitySystemID() - Error with next: " + e.getMessage());

            }

          } catch (SQLException e) {

            System.out.println("DBEntityResolver.getEntitySystemID() - Error with getrset: " + e.getMessage());

          }

          stmt.close();

        } catch (SQLException e) {

          System.out.println("DBEntityResolver.getEntitySystemID() - Error getting id: " + e.getMessage());

        }

        // return the selected System ID number

        return system_id;

   }

   /** Register Public ID in db XML Catalog */

   private void registerEntityPublicID (Connection conn, String doctype, String publicId, String systemId)

   {

        try {

          conn.setAutoCommit(false);

          PreparedStatement pstmt;

          pstmt = conn.prepareStatement(

                "INSERT INTO xml_catalog (entity_id, entity_name, entity_type, source_doctype, public_id, system_id) " +

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

          // Bind the values to the query

          pstmt.setString(1, doctype);

          pstmt.setString(2, publicId);

          pstmt.setString(3, systemId);

          // Do the insertion

          pstmt.execute();

          pstmt.close();

          conn.commit();

          conn.setAutoCommit(true);

        } catch (SQLException e) {

          System.out.println(e.getMessage());

        }

   }

}