package org.opensync.xmldb;
   
   import org.apache.xindice.client.xmldb.services.CollectionManager;
   import org.apache.xindice.util.XindiceException;
   import org.apache.xindice.xml.dom.DOMParser;
   import org.xmldb.api.base.Database;
   import org.xmldb.api.base.Resource;
   import org.xmldb.api.base.ResourceIterator;
   import org.xmldb.api.base.ResourceSet;
  import org.xmldb.api.base.XMLDBException;
  import org.xmldb.api.modules.XMLResource;
  import org.xmldb.api.modules.XPathQueryService;
  
  /***
   * <p>Title: DatabaseManager</p>
   * <p>Description: XML:DB API database Manager (cf. http://www.xmldb.org).</p>
   */
  
  public class DatabaseManager {
  
    String driver = "org.apache.xindice.client.xmldb.DatabaseImpl";
    Database database;
  
    public DatabaseManager() throws ClassNotFoundException, IllegalAccessException, InstantiationException, XMLDBException {
  
      Class c = Class.forName(driver);
      database = (Database) c.newInstance();
      org.xmldb.api.DatabaseManager.registerDatabase(database);
    }
  
    public void closeDatabase() throws ClassNotFoundException, IllegalAccessException, InstantiationException, XMLDBException {
      org.xmldb.api.DatabaseManager.deregisterDatabase(database);
    }
  
    public org.xmldb.api.base.Collection tx_start(org.xmldb.api.base.Collection col) throws XindiceException, XMLDBException {
  
      // Build up a Collection XML configuration.
      String collectionName = "tx_"+col.getName();
      if (this.getCollection(collectionName) != null) this.removeCollection(collectionName);
      org.xmldb.api.base.Collection tx_col = this.createCollection(collectionName);
      return tx_col;
    }
  
    public org.xmldb.api.base.Collection tx_commit(org.xmldb.api.base.Collection col) throws XindiceException, XMLDBException {
  
      // Build up a Collection XML configuration.
      String tx_collectionName = "tx_"+col.getName();
      org.xmldb.api.base.Collection tx_col = null;
      if ( (tx_col = this.getCollection(tx_collectionName)) != null) {
        //Copy tx_col to col
        col = this.copyCollection(tx_col, col);
      }
      return col;
    }
  
    public org.xmldb.api.base.Collection tx_rollback(org.xmldb.api.base.Collection col) throws XindiceException, XMLDBException {
  
      // Just remove the tx_col
       String tx_collectionName = "tx_"+col.getName();
      org.xmldb.api.base.Collection tx_col = null;
      if ( (tx_col = this.getCollection(tx_collectionName)) != null) {
        this.removeCollection(tx_collectionName);
      }
  
      return col;
    }
  
    public org.xmldb.api.base.Collection copyCollection(org.xmldb.api.base.Collection src_col, org.xmldb.api.base.Collection dest_col) throws XindiceException, XMLDBException {
      String[] res = src_col.listResources();
      for (int i=0; i < res.length; i++) {
        dest_col.storeResource(src_col.getResource(res[i]));
      }
      return dest_col;
    }
  
    public org.xmldb.api.base.Collection createCollection(String collectionName) throws XindiceException, XMLDBException {
  
      org.xmldb.api.base.Collection col = null;
      col = org.xmldb.api.DatabaseManager.getCollection("xmldb:xindice:///db/");
      CollectionManager service = (CollectionManager) col.getService("CollectionManager", "1.0");
  
      // Build up the Collection XML configuration.
      String collectionConfig =
  	"<collection compressed=\"true\" name=\"" + collectionName + "\">" +
  	"   <filer class=\"org.apache.xindice.core.filer.BTreeFiler\" gzip=\"true\"/>" +
  	"</collection>";
      service.createCollection(collectionName, DOMParser.toDocument(collectionConfig));
      //System.out.println("Collection " + collectionName + " created.");
      col = this.getCollection(collectionName);
      return col;
    }
  
    public org.xmldb.api.base.Collection getCollection(String collectionName) throws XindiceException, XMLDBException {
  
      org.xmldb.api.base.Collection col = null;
      col = org.xmldb.api.DatabaseManager.getCollection("xmldb:xindice:///db/"+collectionName);
      /*
      if (col != null) System.out.println("Collection " + collectionName + " found.");
      else System.out.println("Collection " + collectionName + " not found.");
     */
     return col;
   }
 
   public void removeCollection(String collectionName) throws XindiceException, XMLDBException {
     org.xmldb.api.base.Collection col = null;
     col = org.xmldb.api.DatabaseManager.getCollection("xmldb:xindice:///db/");
     CollectionManager service = (CollectionManager) col.getService("CollectionManager", "1.0");
     service.removeCollection(collectionName);
   }
 
   public void addDocument(org.xmldb.api.base.Collection col, String id, String xmlData) throws XindiceException, XMLDBException {
 
    XMLResource document = (XMLResource) col.createResource(id, "XMLResource");
    document.setContent(xmlData);
    col.storeResource(document);
    //System.out.println("Document " + xmlData + " inserted as " + document.getId());
   }
 
   public void removeDocument(org.xmldb.api.base.Collection col, String xmlData) throws XindiceException, XMLDBException {
 
     Resource document = col.getResource(xmlData);
     col.removeResource(document);
     //System.out.println("Document " + xmlData + " removed");
   }
 
   public XMLResource retrieveDocument(org.xmldb.api.base.Collection col, String id) throws XindiceException, XMLDBException {
     XMLResource document = null;
 
     document = (XMLResource) col.getResource(id);
     if (document != null) {
       //System.out.println("Document " + id);
       //System.out.println(document.getContent());
     }
     else {
       //System.out.println("Document not found");
     }
     return document;
   }
 
   public ResourceSet xpathQuery(org.xmldb.api.base.Collection col, String xpath) throws XindiceException, XMLDBException {
 
     XPathQueryService service = (XPathQueryService) col.getService("XPathQueryService", "1.0");
     ResourceSet resultSet = service.query(xpath);
 
     return resultSet;
   }
 
   public void close(org.xmldb.api.base.Collection col) {
     try {
       if (col != null) {
 	col.close();
       }
     } catch (XMLDBException e) {
 
     }
   }
 
   public void displayResults(ResourceSet resultSet) throws XMLDBException {
       ResourceIterator results = resultSet.getIterator();
       while (results.hasMoreResources()) {
 	Resource res = results.nextResource();
 	//System.out.println((String) res.getContent());
       }
   }
 
   public static void main(String[] args) throws Exception {
 
     DatabaseManager dbMgr = null;
     org.xmldb.api.base.Collection col = null;
     org.xmldb.api.base.Collection tx_col = null;
     try {
       dbMgr = new DatabaseManager();
       if ( (col = dbMgr.getCollection("test1")) == null) col = dbMgr.createCollection("test1");
 
       tx_col = dbMgr.tx_start(col);
 
 
       String view = "<view keyword=\"theme4\" value=\"val4\"/>";
       // Does the view exists in the current collection ?
       if (dbMgr.retrieveDocument(col, "theme4") == null) {
 	// The view doesn't exist: add it !
 	dbMgr.addDocument(tx_col, "theme4", view);
       }
 
 
       /*
       dbMgr.addDocument(col, "theme1", "<view keyword=\"theme1\" value=\"val1\"/>");
       dbMgr.addDocument(col, "theme2", "<view keyword=\"theme2\" value=\"val2\"/>");
       dbMgr.addDocument(col, "theme3", "<view keyword=\"theme3\" value=\"val3\"/>");
       */
 
       dbMgr.tx_commit(col);
 
       ResourceSet resultSet = dbMgr.xpathQuery(col, "/view[@keyword='theme4' and @value='val4']");
       dbMgr.displayResults(resultSet);
 
     } catch(Exception e) {
       System.out.println(e.getLocalizedMessage());
     }
     finally {
       dbMgr.close(col);
     }
   }
 }