package de.mpg.mpiwg.berlin.mpdl.lt.doc.regularization;

import java.io.File;

import com.sleepycat.je.Database;
import com.sleepycat.je.DatabaseConfig;
import com.sleepycat.je.DatabaseException;
import com.sleepycat.je.Environment;
import com.sleepycat.je.EnvironmentConfig;

import de.mpg.mpiwg.berlin.mpdl.exception.ApplicationException;

public class DbEnvRegularization {
  private String dataDir;
  private File envPath;
  private Environment env;
  private EnvironmentConfig envConfig;
  private DatabaseConfig dbConfig;
  private Database origDB;
  private Database normDB;

  public DbEnvRegularization() {
  }

  public void setDataDir(String dataDir) {
    this.dataDir = dataDir;
  }
  
  public void init() throws ApplicationException {
    try {
      envConfig = new EnvironmentConfig();
      dbConfig = new DatabaseConfig();
      envConfig.setReadOnly(false);
      dbConfig.setReadOnly(false);
      envConfig.setAllowCreate(true);
      dbConfig.setAllowCreate(true);
      envConfig.setTransactional(true);
      dbConfig.setTransactional(true);
      // allow duplicates for keys
      dbConfig.setSortedDuplicates(true);
      envPath = new File(dataDir);
      env = new Environment(envPath, envConfig);
    } catch (DatabaseException e) {
      throw new ApplicationException(e);
    }
  }

  public void openDatabases() throws ApplicationException {
    try {
      // open databases (and create them if they do not exist)
      origDB = env.openDatabase(null, "OrigDB", dbConfig);
      normDB = env.openDatabase(null, "NormDB", dbConfig);
    } catch (DatabaseException e) {
      throw new ApplicationException(e);
    }
  }
  
  public void removeDatabases() throws ApplicationException {
    try {
      if (origDB != null)
        origDB.close();
      if (normDB != null)
        normDB.close();
      env.removeDatabase(null, "OrigDB");    
      env.removeDatabase(null, "NormDB");
      origDB = null;
      normDB = null;
    } catch (DatabaseException e) {
      throw new ApplicationException(e);
    }
  }
  
  public Environment getEnv() {
    return env;
  }

  public Database getNormDB() {
    return normDB;
  }

  public Database getOrigDB() {
    return origDB;
  }

  public void close() throws ApplicationException {
    if (env != null) {
      try {
        if (origDB != null)
          origDB.close();
        if (normDB != null)
          normDB.close();
        if (env != null)
          env.close();
      } catch (DatabaseException e) {
        throw new ApplicationException(e);
      }
    }
  }
}