package model;

import java.io.IOException;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;

import javax.naming.NamingException;
import tool.DriverNotFound;

public class DatabaseManagement {
	
	private static final String TYPE_JDBC = "jdbc";
	private static final String TYPE_DS = "ds";
	private static final String TYPE = "type";
	private static final String USERID = "userid";
	private static final String PASSWORD = "password";
	private static final String DRIVER = "driver";
	private static final String URL = "url";
	private static final String DATASOURCE = "datasource";
	private static final String PROVIDER_URL = "provider_url";
	private static final String INITIAL_CONTEXT_FACTORY = "initial_context_factory";

	private static final String file = "databaseInformation.properties";
	
	private static String type = null;
	private static String userid = null;
	private static String password = null;
	private static String driver = null;
	private static String url = null;
	private static String datasource = null;
	private static String initialContextFactory = null;
	private static String providerUrl = null;

	static {
		refresh();
	}

	public static void refresh() {
		try {
			InputStream inputStream = DatabaseManagement.class.getClassLoader().getResourceAsStream(file);
			if (inputStream != null) { 
				Properties properties = new Properties();
				properties.load(inputStream);
				type = properties.getProperty(TYPE, type);
				if (type.equalsIgnoreCase(TYPE_JDBC)) {
					userid = properties.getProperty(USERID, userid);
					password = properties.getProperty(PASSWORD, password);
					driver = properties.getProperty(DRIVER, driver);
					url = properties.getProperty(URL, url);
				}
				if (type.equalsIgnoreCase(TYPE_DS)) {
					userid = properties.getProperty(USERID, userid);
					password = properties.getProperty(PASSWORD, password);
					datasource = properties.getProperty(DATASOURCE, datasource);
					providerUrl = properties.getProperty(PROVIDER_URL, providerUrl);
					initialContextFactory = properties.getProperty(INITIAL_CONTEXT_FACTORY, initialContextFactory);
				}
			} 
			inputStream.close();
		} catch (IOException e) {
			e.printStackTrace();
		} 
	}

	public static Connection getConnection() throws SQLException, DriverNotFound {
		ConnectionFactory factory = null;
		try {
			if (type.equalsIgnoreCase(TYPE_JDBC)) {
				factory = new ConnectionFactory(driver, url, userid, password);
			} else {
				factory = new ConnectionFactory(userid, password, datasource, providerUrl, initialContextFactory);
			}
		} catch (ClassNotFoundException e) {
			throw new DriverNotFound(driver);
		} catch (NamingException e) {
			e.printStackTrace();
		}
		return factory.getConnection();
	}

}