package dmo.fs.spa.db.sqlite3;

import com.fasterxml.jackson.databind.JsonNode;
import dmo.fs.spa.db.SpaDbConfiguration;
import dmo.fs.spa.db.SqlBuilder;
import dmo.fs.spa.utils.SpaLogin;
import dmo.fs.spa.utils.SpaLoginImpl;
import dmo.fs.utils.DodexUtil;
import dmo.fs.vertx.Server;
import io.reactivex.rxjava3.core.Completable;
import io.reactivex.rxjava3.core.Single;
import io.reactivex.rxjava3.disposables.Disposable;
import io.vertx.core.Future;
import io.vertx.core.Promise;
import io.vertx.jdbcclient.JDBCConnectOptions;
import io.vertx.rxjava3.core.Vertx;
import io.vertx.rxjava3.jdbcclient.JDBCPool;
import io.vertx.rxjava3.sqlclient.Row;
import io.vertx.rxjava3.sqlclient.RowSet;
import io.vertx.sqlclient.PoolOptions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.sql.SQLException;
import java.util.Map;
import java.util.Properties;
import java.util.concurrent.ConcurrentHashMap;

public class SpaDatabaseSqlite3 extends DbSqlite3 {
	private final static Logger logger =
			LoggerFactory.getLogger(SpaDatabaseSqlite3.class.getName());
	protected Disposable disposable;
	protected Properties dbProperties = new Properties();
	protected Map<String, String> dbOverrideMap = new ConcurrentHashMap<>();
	protected Map<String, String> dbMap = new ConcurrentHashMap<>();
	protected JsonNode defaultNode;
	protected String webEnv = System.getenv("VERTXWEB_ENVIRONMENT");
	protected DodexUtil dodexUtil = new DodexUtil();
	private Vertx vertx;

	public SpaDatabaseSqlite3(Map<String, String> dbOverrideMap, Properties dbOverrideProps)
			throws InterruptedException, IOException, SQLException {
		super();

		defaultNode = dodexUtil.getDefaultNode();

		webEnv = webEnv == null || "prod".equals(webEnv) ? "prod" : "dev";

		dbMap = dodexUtil.jsonNodeToMap(defaultNode, webEnv);
		dbProperties = dodexUtil.mapToProperties(dbMap);

		if (dbOverrideProps != null && dbOverrideProps.size() > 0) {
			this.dbProperties = dbOverrideProps;
		}
		if (dbOverrideMap != null) {
			this.dbOverrideMap = dbOverrideMap;
		}

		dbProperties.setProperty("foreign_keys", "true");

		SpaDbConfiguration.mapMerge(dbMap, dbOverrideMap);
		databaseSetup();
	}

	public SpaDatabaseSqlite3() throws InterruptedException, IOException, SQLException {
		super();

		defaultNode = dodexUtil.getDefaultNode();
		webEnv = webEnv == null || "prod".equals(webEnv) ? "prod" : "dev";

		dbMap = dodexUtil.jsonNodeToMap(defaultNode, webEnv);
		dbProperties = dodexUtil.mapToProperties(dbMap);

		dbProperties.setProperty("foreign_keys", "true");

		// databaseSetup();
	}

	@Override
	public Future<Void> databaseSetup() throws InterruptedException, SQLException {
		if ("dev".equals(webEnv)) {
			SpaDbConfiguration.configureTestDefaults(dbMap, dbProperties);
		} else {
			SpaDbConfiguration.configureDefaults(dbMap, dbProperties);
		}

		PoolOptions poolOptions =
				new PoolOptions().setMaxSize(Runtime.getRuntime().availableProcessors() * 5);

		JDBCConnectOptions connectOptions;

		connectOptions = new JDBCConnectOptions()
				.setJdbcUrl(dbMap.get("url") + dbMap.get("filename") + "?foreign_keys=on;")
				.setIdleTimeout(1);

		vertx = Server.getRxVertx();

		pool = JDBCPool.pool(vertx, connectOptions, poolOptions);

		Completable completable = pool.rxGetConnection()
				.flatMapCompletable(conn -> conn.rxBegin().flatMapCompletable(
						tx -> conn.query(CHECKLOGINSQL).rxExecute().doOnSuccess(rows -> {
							if (rows.size() == 0) {
								final String usersSql = getCreateTable("LOGIN");

								Single<RowSet<Row>> crow =
										conn.query(usersSql).rxExecute().doOnError(err -> {
											logger.info(String.format("Login Table Error: %s",
													err.getMessage()));
										}).doOnSuccess(result -> {
											logger.info("Login Table Added.");
										});

								crow.subscribe(result -> {
									//
								}, err -> {
									logger.info(String.format("Login Table Error: %s",
											err.getMessage()));
								});
							}
						}).doOnError(err -> {
							logger.info(String.format("Login Table Error: %s", err.getMessage()));

						}).flatMapCompletable(res -> tx.rxCommit())));

		Promise<Void> setupPromise = Promise.promise();
		completable.subscribe(() -> {
			try {
				SqlBuilder.setupSql(pool);
				setupPromise.complete();
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}, err -> {
			logger.info(String.format("Tables Create Error: %s", err.getMessage()));
		});

		return setupPromise.future();
	}

	@Override
	public SpaLogin createSpaLogin() {
		return new SpaLoginImpl();
	}

	@Override
	@SuppressWarnings("unchecked")
	public <T> T getPool() {
		return (T) pool;
	}

	@Override
	public Vertx getVertx() {
		return vertx;
	}

	@Override
	public void setVertx(Vertx vertx) {
		this.vertx = vertx;
	}
}