package com.pretzel.dev.villagertradelimiter.database;

import com.pretzel.dev.villagertradelimiter.VillagerTradeLimiter;
import com.pretzel.dev.villagertradelimiter.data.Cooldown;
import com.pretzel.dev.villagertradelimiter.data.PlayerData;
import com.pretzel.dev.villagertradelimiter.lib.Util;
import org.bukkit.configuration.ConfigurationSection;

import java.util.UUID;

public class DatabaseManager {
    private static final String CREATE_TABLE_COOLDOWN =
            "CREATE TABLE IF NOT EXISTS vtl_cooldown("+
                    "uuid CHAR(36) NOT NULL,"+
                    "item VARCHAR(255) NOT NULL,"+
                    "time BIGINT NOT NULL,"+
                    "PRIMARY KEY(uuid, item));";
    private static final String SELECT_ITEMS = "SELECT * FROM vtl_cooldown;";
    private static final String INSERT_ITEM = "INSERT OR IGNORE INTO vtl_cooldown(uuid,item,time) VALUES?;"; //INSERT IGNORE INTO for MySQL
    private static final String DELETE_ITEMS = "DELETE FROM vtl_cooldown WHERE uuid='?';";

    private final VillagerTradeLimiter instance;
    private Database database;

    public DatabaseManager(final VillagerTradeLimiter instance) {
        this.instance = instance;
    }

    public void load() {
        final ConfigurationSection cfg = instance.getCfg().getConfigurationSection("database");
        if(cfg == null) {
            Util.consoleMsg("Database settings missing from config.yml!");
            this.database = null;
            return;
        }
        boolean mysql = cfg.getBoolean("mysql", false);
        if(this.database != null && ((mysql && this.database.isMySQL()) || (!mysql && !this.database.isMySQL()))) this.database.load(cfg);
        else this.database = (mysql?new MySQL(instance, cfg):new SQLite(instance));
        this.database.execute(CREATE_TABLE_COOLDOWN, false);

        //Loads all the data
        this.database.execute(SELECT_ITEMS, true, (result,args) -> {
            if(result != null) {
                for(String row : result) {
                    final String[] tokens = row.split(",");

                    UUID uuid = UUID.fromString(tokens[0]);
                    String item = tokens[1];
                    long time = Long.parseLong(tokens[2]);

                    PlayerData playerData = instance.getPlayerData().get(uuid);
                    if(playerData == null) {
                        playerData = new PlayerData();
                        instance.getPlayerData().put(uuid, playerData);
                    }

                    String cooldown = instance.getCfg().getString("Overrides."+item+".Cooldown", null);
                    if(cooldown != null && System.currentTimeMillis() < time + Cooldown.parseTime(cooldown)) {
                        playerData.getTradingCooldowns().put(item, time);
                    }
                }
            }
        });
    }

    public void savePlayer(final UUID uuid, boolean async) {
        if(this.database == null) return;

        //Delete existing rows for player
        final String uuidStr = uuid.toString();
        if(async) this.database.execute(DELETE_ITEMS.replace("?", uuidStr), false, (result,args) -> save(uuid, true));
        else {
            this.database.execute(DELETE_ITEMS.replace("?", uuidStr), false);
            save(uuid, false);
        }
    }
    private void save(final UUID uuid, boolean async) {
        //Insert new rows for player pages
        final PlayerData playerData = instance.getPlayerData().get(uuid);
        if(playerData == null) return;

        String values = "";
        for(String item : playerData.getTradingCooldowns().keySet()) {
            long time = playerData.getTradingCooldowns().get(item);

            if(!values.isEmpty()) values += ",";
            values += "('"+uuid+"','"+item+"','"+time+"')";
        }
        if(values.isEmpty()) return;
        String sql = INSERT_ITEM.replace("?", values);
        if(this.database.isMySQL()) sql = sql.replace(" OR ", " ");
        if(async) this.database.execute(sql, false, (result,args) -> {});
        else this.database.execute(sql, false);
    }
}