package com.endava.thucydides.pas.helpers;

import java.sql.Connection;

import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

public class DatabaseSourse {
	private static String server;
	private static String dbName;
	private static String user;
	private static String password;

	private static Connection connection;

	private final static String CONNECTION_URL_PATTERN = "jdbc:sqlserver://%s;databaseName=%s;user=%s;password=%s";

	private static PropertiesReader pr = new PropertiesReader("dbconnection.properties");

	public static void establishDBConnection() throws SQLException {
		server = System.getProperty("dbServer");
		dbName = System.getProperty("dbName");

		if(server == null) server = pr.getProperty("dbServer");
		if(dbName == null) dbName = pr.getProperty("dbName");
		user = pr.getProperty("user");
		password = pr.getProperty("password");

		String connectionURL = String.format(CONNECTION_URL_PATTERN, server, dbName, user, password);

		connection = DriverManager.getConnection(connectionURL);
	}

	public static List<HashMap<String,String>> executeQuery(String query, String[] parameters) throws SQLException {
		Statement stmt = connection.createStatement();
		ResultSet rs = stmt.executeQuery(String.format(query, parameters));
		ResultSetMetaData rsmd = rs.getMetaData();
		List<HashMap<String,String>> result = new ArrayList<HashMap<String,String>>();
		HashMap<String, String> map = null;
		while (rs.next()) {
			map = new HashMap<String, String>();
			for(int i = 1 ; i <= rsmd.getColumnCount() ; i++){
				map.put(rsmd.getColumnName(i), rs.getString(i));
			}
			result.add(map);
		}
		return result;
	}

}