TFGuilds/src/main/java/me/totalfreedom/tfguilds/sql/SQLWarpData.java

package me.totalfreedom.tfguilds.sql;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import me.totalfreedom.tfguilds.TFGuilds;
import me.totalfreedom.tfguilds.guild.GuildWarp;
import org.bukkit.World;
import org.bukkit.entity.Player;

public class SQLWarpData
{
    private static final TFGuilds plugin = TFGuilds.getPlugin();
    private static final String TABLE_NAME = "warps";

    private final String SELECT = "SELECT * FROM `" + TABLE_NAME + "` WHERE identifier=? AND warp_name=?";
    private final String SELECT_ALL = "SELECT * FROM `" + TABLE_NAME + "`";
    private final String UPDATE = "UPDATE `" + TABLE_NAME + "` WHERE identifier=? AND warp_name=? SET x=?, y=?, z=?, world=?";
    private final String DELETE = "DELETE FROM `" + TABLE_NAME + "` WHERE identifier=? AND warp_name=?";
    private final String INSERT = "INSERT INTO `" + TABLE_NAME + "` (`identifier`, `warp_name`, `x`, `y`, `z`, `world`) VALUES (?, ?, ?, ?, ?, ?);";

    public boolean exists(String identifier, String warpName)
    {
        GuildWarp warp = plugin.warps.get(identifier);
        return warp != null && warp.getWarpName().equalsIgnoreCase(warpName);
    }

    public GuildWarp get(String identifier, String warpName)
    {
        Connection connection = plugin.sql.getConnection();
        try
        {
            PreparedStatement statement = connection.prepareStatement(SELECT);
            statement.setString(1, identifier);
            statement.setString(2, warpName);
            ResultSet set = statement.executeQuery();
            double x = set.getDouble("x");
            double y = set.getDouble("y");
            double z = set.getDouble("z");
            World world = plugin.worldData.getWorld(set.getInt("world"));
            return new GuildWarp(identifier, warpName, x, y, z, world);
        }
        catch (SQLException ex)
        {
            ex.printStackTrace();
        }
        return null;
    }

    public void getAll()
    {
        Connection connection = plugin.sql.getConnection();
        try
        {
            PreparedStatement statement = connection.prepareStatement(SELECT_ALL);
            ResultSet set = statement.executeQuery();
            while (set.next())
            {
                String identifier = set.getString("identifier");
                plugin.warps.put(identifier, get(identifier, set.getString("warp_name")));
            }
        }
        catch (SQLException ex)
        {
            ex.printStackTrace();
        }
    }

    public GuildWarp create(String identifier, String warpName, Player player)
    {
        Connection connection = plugin.sql.getConnection();
        try
        {
            PreparedStatement statement = connection.prepareStatement(INSERT);
            statement.setString(1, identifier);
            statement.setString(2, warpName);
            statement.setDouble(3, player.getLocation().getX());
            statement.setDouble(4, player.getLocation().getY());
            statement.setDouble(5, player.getLocation().getZ());
            statement.setInt(6, plugin.worldData.getWorldID(player.getWorld()));
            statement.execute();
            GuildWarp warp = new GuildWarp(identifier, warpName, player.getLocation().getX(), player.getLocation().getY(),
                    player.getLocation().getZ(), player.getWorld());
            plugin.warps.put(identifier, warp);
            return warp;
        }
        catch (SQLException ex)
        {
            ex.printStackTrace();
        }
        return null;
    }

    public void save(GuildWarp warp)
    {
        Connection connection = plugin.sql.getConnection();
        try
        {
            PreparedStatement statement = connection.prepareStatement(UPDATE);
            statement.setString(1, warp.getIguild());
            statement.setString(2, warp.getWarpName());
            statement.setDouble(3, warp.getX());
            statement.setDouble(4, warp.getY());
            statement.setDouble(5, warp.getZ());
            statement.setInt(6, plugin.worldData.getWorldID(warp.getWorld()));
            statement.execute();
            plugin.warps.put(warp.getIguild(), warp);
        }
        catch (SQLException ex)
        {
            ex.printStackTrace();
        }
    }

    public void delete(GuildWarp warp)
    {
        Connection connection = plugin.sql.getConnection();
        try
        {
            PreparedStatement statement = connection.prepareStatement(DELETE);
            statement.setString(1, warp.getIguild());
            statement.setString(2, warp.getWarpName());
            statement.execute();
            plugin.warps.remove(warp.getIguild());
        }
        catch (SQLException ex)
        {
            ex.printStackTrace();
        }
    }
}