package ru.prisonlife.PrisonAPI.template;

import org.bukkit.entity.Player;
import ru.prisonlife.PrisonAPI.connect.DataBasePrison4Life;

import java.sql.*;

public class Prisoner extends DataBasePrison4Life {
    private Player me;
    private Wallet wallet;
    private Level level;
    private PreparedStatement state;
    private ResultSet r_set;

    public Prisoner(Player player) throws SQLException {
        me = player;
        setInResUUID();
        try {
            if(r_set.next())
                wallet = new Wallet(this);
        } catch(NullPointerException z) {}
    }

    public Player getPlayer() { return me; }

    public Wallet getWallet() { return this.wallet; }

    public void setWallet() throws SQLException {
        setInResUUID();
        if(!r_set.next()) {
            state.execute("INSERT INTO ECONOMY(NAME, UUID, WALLET) VALUES (" +
                    "'" + getPlayer().getName() + "', '" + me.getUniqueId() + "', 0);");
            wallet = new Wallet(this);
        }
        disconnect();
    }

    public void delWallet() throws SQLException {
        setInResUUID();
        if(r_set.first())
            state.execute("DELETE FROM ECONOMY WHERE UUID = '" + me.getUniqueId() + "'");
        disconnect();
    }

    public Level getPrisonLevel() throws SQLException {
        setInResUUID();
        if(r_set.first())
            return this.level;
        disconnect();
        return null;
    }

    public void setPrisonLevel() throws SQLException {
        setInResUUID();
        if(!r_set.next()) {
            state.execute("INSERT INTO PLAYERS_LEVEL(NAME, UUID, POINTS, LEVEL) VALUES (" +
                    "'" + me.getName() + "', '" + me.getUniqueId() + "', 0, 1);");
            level = new Level(this);
        }
        disconnect();
    }

    public void delPrisonLevel() throws SQLException {
        setInResUUID();
        if(r_set.first()) {
            state.execute("DELETE FROM PLAYERS_LEVEL WHERE UUID = '" + me.getUniqueId() + "'");
        }
    }

    private void setInResUUID() throws SQLException {
        state = connection.prepareStatement("SELECT * UUID FROM ECONOMY WHERE UUID = ?;");
        state.setString(1, me.getUniqueId().toString());
        r_set = state.executeQuery("SELECT UUID FROM ECONOMY WHERE UUID = '" + me.getUniqueId() + "'");
    }

    private void disconnect() throws SQLException {
        r_set.close();
        state.close();
    }
}