import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;




public class Vote {


    private static Connection con = null;
    private static Statement stmt;

    public static void createConnection() {
        try {
            Class.forName("com.mysql.jdbc.Driver").newInstance();
            String IP="WEB HOST IP";
            String DB="DATABASE NAME";
            String User="DATABASE USERNAME";
            String Pass="DATABASE PASSWORD";
            con = DriverManager.getConnection("jdbc:mysql://"+IP+"/"+DB, User, Pass);
            stmt = con.createStatement();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void claimVote(Player player) {
        createConnection();
        if (checkVote(player.getUsername())) {
            int tokens = getVotes(player.getUsername()); //Number of times this user has voted. Number of unclaimed votes.

            //methods for giving rewards to players goes here.
            //for example, I might give a user 50,000 coins per 1 time they voted, so I would do:
            player.getInventory().add(new Item(995, 50000 * tokens)); // adds 50,000 * token amount of coins to the player's inventory.


            //remove votes after user as claimed.
            removeVotes(player.getUsername());
        } else {
            player.sendMessage("You either have not voted, already claimed your reward, or there was an error.");
            player.sendMessage("Either try again in a few minutes, or contact a staff member.");
        }
    }

    private static ResultSet query(String s) throws SQLException {
        try {
            if (s.toLowerCase().startsWith("select")) {
                ResultSet rs = stmt.executeQuery(s);
                return rs;
            } else {
                stmt.executeUpdate(s);
            }
            return null;
        } catch (Exception e) {
            destroyConnection();
            createConnection();
        }
        return null;
    }


    private static void destroyConnection() {
        try {
            stmt.close();
            con.close();
        } catch (Exception e) {


        }
    }


    private static boolean checkVote(String playerName) {
        try {
            String name2 = playerName.replaceAll("_", " ");
            Statement statement = con.createStatement();
            String query = "SELECT * FROM players WHERE username = '" + name2 + "'";
            ResultSet results = statement.executeQuery(query);
                while(results.next()) {
                    int tokens = results.getInt("tokens");
                    if(tokens >= 0) {
                         return true;
                    }
                }
        } catch(SQLException e) {
            e.printStackTrace();
        }
        return false;
    }

    private static int getVotes(String playerName) {
        try {
            String name2 = playerName.replaceAll("_", " ");
            Statement statement = con.createStatement();
            String query = "SELECT * FROM players WHERE username = '" + name2 + "'";
            ResultSet results = statement.executeQuery(query);
            while(results.next()) {
                int tokens = results.getInt("tokens");
                if(tokens >= 1) {
                    return tokens;
                }
            }
            } catch(SQLException e) {
                e.printStackTrace();
            }
            return 0;
    }

    private static boolean removeVotes(String playerName) {
        try {
            String name2 = playerName.replaceAll("_", " ");
            query("DELETE FROM `players` WHERE username = '"+name2+"';");
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
        return true;
    }

}