package de.gnmyt.SQLToolkit.manager;

import de.gnmyt.SQLToolkit.drivers.MySQLConnection;
import org.slf4j.Logger;

import java.sql.ResultSet;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.HashMap;

public class ResultManager {

    private final Logger LOG = MySQLConnection.LOG;

    private final ResultSet resultSet;

    /**
     * Basic constructor for the ResultManager
     * @param resultSet Existing ResultSet
     */
    public ResultManager(ResultSet resultSet) {
        this.resultSet = resultSet;
    }

    /**
     * Getting the ResultSet
     * @return ResultSet
     */
    public ResultSet getResultSet() {
        return resultSet;
    }

    /**
     * Get the Object of an Result
     * @param column Table column
     * @return Object
     */
    public Object getObject(String column) {
        try { while (resultSet.next()) { return resultSet.getObject(column); }
        } catch (Exception err) { LOG.error(err.getMessage()); }
        return null;
    }

    /**
     * Get the String of an Result
     * @param column Table column
     * @return String
     */
    public String getString(String column) {
        try { while (resultSet.next()) { return resultSet.getString(column); }
        } catch (Exception err) { LOG.error(err.getMessage()); }
        return "";
    }

    /**
     * Get the Integer of an Result
     * @param column Table column
     * @return Integer
     */
    public Integer getInteger(String column) {
        try { while (resultSet.next()) { return resultSet.getInt(column); }
        } catch (Exception err) { LOG.error(err.getMessage()); }
        return 0;
    }

    /**
     * Get the Boolean of an Result
     * @param column Table column
     * @return Boolean
     */
    public Boolean getBoolean(String column) {
        try { while (resultSet.next()) { return resultSet.getBoolean(column); }
        } catch (Exception err) { LOG.error(err.getMessage()); }
        return false;
    }

    public Timestamp getTimestamp(String column) {
        try { while (resultSet.next()) { return resultSet.getTimestamp(column); }
        } catch (Exception err) { LOG.error(err.getMessage()); }
        return null;
    }

    /**
     * Get the count of the Result
     * @return Integer (Count)
     */
    public Integer getRowCount() {
        int count=0;
        try { while (resultSet.next()) { count++; } } catch (Exception err) { LOG.error(err.getMessage()); }
        return count;
    }

    /**
     * Get a List of all Results
     * @param column Table column
     * @return ArrayList of Result
     */
    public ArrayList<String> getList(String column) {
        ArrayList<String> results = new ArrayList<>();
        try { while (resultSet.next()) { results.add(resultSet.getString(column)); }
        } catch (Exception err) { LOG.error(err.getMessage()); }
        return results;
    }

    /**
     * Get a List of all Results for 2 columns
     * @param column Table column #1
     * @param column2 Table column #2
     * @return HashMap of Result
     */
    public HashMap<String, String> getMultipleList(String column, String column2) {
        HashMap<String, String> results = new HashMap<>();
        try { while (resultSet.next()) {
            results.put(resultSet.getString(column), resultSet.getString(column2)); }
        } catch (Exception err) { LOG.error(err.getMessage()); }
        return results;
    }

    /**
     * Get a list of all results with all columns
     * @return ArrayList with the result
     */
    public ArrayList<HashMap<String, Object>> getList() {
        ArrayList<HashMap<String, Object>> results = new ArrayList<>();
        try {
            while (resultSet.next()) {
                HashMap<String, Object> result = new HashMap<>();
                for (int i=0; i < resultSet.getMetaData().getColumnCount(); i++) {
                    String columnName = resultSet.getMetaData().getColumnName(i+1);
                    result.put(columnName, resultSet.getObject(columnName));
                }
                results.add(result);
            }
        } catch (Exception err) { LOG.error(err.getMessage()); }
        return results;
    }

}