package dmo.fs.db.postgres;

import dmo.fs.db.DodexDatabaseBase;
import dmo.fs.util.DodexUtil;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.persistence.EntityManagerFactory;
import jakarta.persistence.Persistence;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.hibernate.Session;
import org.hibernate.SessionFactory;

import java.io.Serial;
import java.io.Serializable;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Map;

@ApplicationScoped
public class DodexDatabasePostgres extends DodexDatabaseBase implements Serializable {
    @Serial
    private static final long serialVersionUID = 1L;
    private final static Logger logger = LogManager.getLogger(DodexDatabasePostgres.class.getName());

    private static EntityManagerFactory emf;

    public DodexDatabasePostgres() {
        super();
    }

    @Override
    public void entityManagerSetup() {
        if (emf == null) {
            emf = Persistence.createEntityManagerFactory("postgres." + DodexUtil.getMode());
        }

        entityManager = emf.createEntityManager();

        Session session = entityManager.unwrap(Session.class);
        SessionFactory sessionFactory = session.getSessionFactory();
        Map<String, Object> properties = sessionFactory.getProperties();
        Object sessionDbname = properties.get("dbname");
        dbName = sessionDbname != null ? sessionDbname.toString() : "test";
        Object mode = properties.get("mode");
    }

    @Override
    public void entityManagerSetup(EntityManagerFactory emf) {

    }

    @Override
    public void configDatabase() {
        String[] types = {"TABLE"};
    /*
      dbName must be consistent with the datasource defined in microprofile-config.properties
     */
        Session session = entityManager.unwrap(Session.class);
        session.doWork(connection -> {
            try {
                String user = connection.getMetaData().getUserName();
                connection.setAutoCommit(false);
                DatabaseMetaData databaseMetaData = connection.getMetaData();
                boolean didCreateATable = false;
                ResultSet rsDbname = databaseMetaData.getTables(dbName, null, "users", types);

                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("USERS")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Users Table Created.");
                }
                rsDbname = databaseMetaData.getTables(dbName, null, "messages", types);
                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("MESSAGES")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Messages Table Created.");
                }
                rsDbname = databaseMetaData.getTables(dbName, null, "undelivered", types);
                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("UNDELIVERED")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Undelivered Table Created.");
                }
                rsDbname = databaseMetaData.getTables(dbName, null, "groups", types);
                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("GROUPS")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Groups Table Created.");
                }
                rsDbname = databaseMetaData.getTables(dbName, null, "member", types);
                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("MEMBER")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Member Table Created.");
                }
                rsDbname = databaseMetaData.getTables(dbName, null, "golfer", types);
                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("GOLFER")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Golfer Table Created.");
                }
                rsDbname = databaseMetaData.getTables(dbName, null, "course", types);
                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("COURSE")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Course Table Created.");
                }
                rsDbname = databaseMetaData.getTables(dbName, null, "ratings", types);
                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("RATINGS")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Ratings Table Created.");
                }
                rsDbname = databaseMetaData.getTables(dbName, null, "scores", types);
                if (!rsDbname.next()) {
                    connection.createStatement().execute(getDatabaseBuild().getCreateTable("SCORES")
                      .replaceAll("dummy", user));
                    didCreateATable = true;
                    logger.info("Scores Table Created.");
                }
                if (didCreateATable) {
                    ResultSet rsSchemas = databaseMetaData.getCatalogs();
                    while (rsSchemas.next()) {
                        if (dbName.equals(rsSchemas.getString(1))) {
                            logger.warn("Used database: '{}' to Create Tables.", dbName);
                        }
                    }
                }
                rsDbname.close();
                connection.commit();
                connection.close();
            } catch (SQLException se) {
                se.printStackTrace();
            }
        });
        setupSql(entityManager);
    }

    public EntityManagerFactory getEmf() {
        return emf;
    }
}