package de.mpg.mpiwg.berlin.mpdl.lt.morph.db;

import java.io.File;
import java.util.HashMap;

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 DbEnvMorphSup {
  private String dataDir;
  private File envPath;
  private Environment env;
  private EnvironmentConfig envConfig;
  private DatabaseConfig dbConfig;
  private HashMap<String, Database> morphSupDBs = new HashMap<String, Database>();

  public DbEnvMorphSup() {
  }

  public void setDataDir(String dataDir) {
    this.dataDir = dataDir;
  }
  
  public void initReadOnly() throws ApplicationException {
    try {
      envConfig = new EnvironmentConfig();
      dbConfig = new DatabaseConfig();
      envPath = new File(dataDir);
      env = new Environment(envPath, envConfig);
    } catch (DatabaseException e) {
      throw new ApplicationException(e);
    }
  }

  public void initReadWrite() 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);
      envPath = new File(dataDir);
      env = new Environment(envPath, envConfig);
    } catch (DatabaseException e) {
      throw new ApplicationException(e);
    }
  }

  public void openDatabase(String morphSupName) throws ApplicationException {
    try {
      Database lexDB = morphSupDBs.get(morphSupName);
      if (lexDB == null) {
        Database morphSupDB = env.openDatabase(null, morphSupName + ".db", dbConfig);
        morphSupDBs.put(morphSupName, morphSupDB);
      }
    } catch (DatabaseException e) {
      throw new ApplicationException(e);
    }
  }
  
  public void closeDatabase(String morphSupName) throws ApplicationException {
    try {
      if (morphSupDBs != null) {
        Database morphSupDB = morphSupDBs.get(morphSupName);
        if (morphSupDB != null)
          morphSupDB.close();
      }
    } catch (DatabaseException e) {
      throw new ApplicationException(e);
    }
  }
  
  public Environment getEnv() {
    return env;
  }

  public Database getMorphSupDB(String morphSupName) {
    Database morphSupDB = morphSupDBs.get(morphSupName);
    return morphSupDB;
  }

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