package at.retrophil.freeforall.mysql;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.function.Consumer;

import org.bukkit.Bukkit;
import org.bukkit.plugin.Plugin;

public class AsyncMySQL {

	private ExecutorService executor;
	private Plugin plugin;
	private MySQL sql;

	public AsyncMySQL(Plugin owner, String host, int port, String user, String password, String database) {
		try {
			sql = new MySQL(host, port, user, password, database);
			executor = Executors.newCachedThreadPool();
			plugin = owner;
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public void update(PreparedStatement statement) {
		executor.execute(() -> sql.queryUpdate(statement));
	}

	public void update(String statement) {
		executor.execute(() -> sql.queryUpdate(statement));
	}

	public void query(PreparedStatement statement, Consumer<ResultSet> consumer) {
		executor.execute(() -> {
			ResultSet result = sql.query(statement);
			Bukkit.getScheduler().runTask(plugin, () -> consumer.accept(result));
		});
	}

	public void query(String statement, Consumer<ResultSet> consumer) {
		executor.execute(() -> {
			ResultSet result = sql.query(statement);
			Bukkit.getScheduler().runTask(plugin, () -> consumer.accept(result));
		});
	}

	public PreparedStatement prepare(String query) {
		try {
			return sql.getConnection().prepareStatement(query);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return null;
	}

	public static class MySQL {

		private String host, user, password, database;
		private int port;
		private Connection conn;

		public MySQL(String host, int port, String user, String password, String database) throws Exception {
			this.host = host;
			this.port = port;
			this.user = user;
			this.password = password;
			this.database = database;

			this.openConnection();
		}

		public void queryUpdate(String query) {
			checkConnection();
			try (PreparedStatement statement = conn.prepareStatement(query)) {
				queryUpdate(statement);
			} catch (Exception e) {
				e.printStackTrace();
			}
		}

		public void queryUpdate(PreparedStatement statement) {
			checkConnection();
			try {
				statement.executeUpdate();
			} catch (Exception e) {
				e.printStackTrace();
			} finally {
				try {
					statement.close();
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
		}

		public ResultSet query(String query) {
			checkConnection();
			try {
				return query(conn.prepareStatement(query));
			} catch (Exception e) {
				e.printStackTrace();
			}
			return null;
		}

		public ResultSet query(PreparedStatement statement) {
			checkConnection();
			try {
				return statement.executeQuery();
			} catch (Exception e) {
				e.printStackTrace();
			}
			return null;
		}

		public Connection openConnection() throws Exception {
			Class.forName("com.mysql.jdbc.Driver");
			return this.conn = DriverManager.getConnection("jdbc:mysql://" + this.host + ":" + this.port + "/" + this.database, this.user, this.password);
		}

		public Connection getConnection() {
			return this.conn;
		}

		private void checkConnection() {
			try {
				if (this.conn == null || !this.conn.isValid(10) || this.conn.isClosed()) openConnection();
			} catch (Exception e) {
				e.printStackTrace();
			}
		}

		public void closeConnection() {
			try {
				this.conn.close();
			} catch (SQLException e) {
				e.printStackTrace();
			} finally {
				this.conn = null;
			}
		}
	}
}