/**
* Resets all counts for a given player
*
* @param playerName player to reset the stats for
*/
public void resetAll(String playerName) {
int playerId = getPlayerId(playerName);
for (MessageNode node : MessageNode.values()) {
if (node.getColumnName() != null) {
set(node, playerId, 0);
}
}
}
/**
* Set the count of a certain message to a certain value
*
* @param node node to set the count for
* @param playerId player for whom we are tracking the count
* @param value value to set
*/
private void set(MessageNode node, int playerId, int value) {
Validate.isTrue(value >= 0, "Count has to be positive");
Connection conn = null;
Statement statement = null;
try {
conn = retrieveConnection();
statement = conn.createStatement();
// Set the count to the provided value
String setQuery =
String.format(
"UPDATE %s SET %s = %s WHERE id = %s",
msgTable, node.getColumnName(), value, playerId);
statement.execute(setQuery);
} catch (SQLException e) {
e.printStackTrace();
} finally {
try {
if (conn != null) conn.close();
if (statement != null) statement.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
}
/**
* Get how often a message has been displayed
*
* @param node to get the count for
* @param playerId id of the player to get the count for
* @return how often this message has been displayed
*/
private int getCountFor(MessageNode node, int playerId) {
Connection conn = null;
Statement statement = null;
ResultSet result = null;
int value = 0;
try {
conn = retrieveConnection();
statement = conn.createStatement();
String select = String.format("SELECT * FROM %s WHERE %s = %s", msgTable, "id", playerId);
result = statement.executeQuery(select);
if (result.next()) value = result.getInt(node.getColumnName());
else // create the missing row
{
String newPlayerDataQuery =
String.format( // empty row in messages
"INSERT INTO %s (%s) VALUES (%s)", msgTable, "id", playerId);
conn.createStatement().executeUpdate(newPlayerDataQuery);
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
try {
if (conn != null) conn.close();
if (statement != null) statement.close();
if (result != null) result.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
return value;
}
/** Creates tables if they do not exist. */
private void initializeTables() {
Connection conn = null;
Statement statement = null;
try {
conn = retrieveConnection();
statement = conn.createStatement();
statement.setQueryTimeout(30);
// One table holding the playername id relation
String playerQuery =
String.format(
"CREATE TABLE IF NOT EXISTS %s (id INTEGER PRIMARY KEY AUTOINCREMENT, %s STRING)",
playerTable, "name");
statement.executeUpdate(playerQuery);
// One column for every message
StringBuilder columns = new StringBuilder();
for (MessageNode node : MessageNode.getMessageNodes()) {
MsgCategory cat = messages.getCat(node);
if (node.getColumnName() != null
&& (cat == MsgCategory.TUTORIAL || cat == MsgCategory.ONE_TIME)) {
columns.append(',');
columns.append(node.getColumnName());
}
}
String msgQuery =
String.format(
"CREATE TABLE IF NOT EXISTS %s (id INTEGER PRIMARY KEY UNIQUE %s)",
msgTable, columns);
statement.executeUpdate(msgQuery);
// Check if all columns are present
DatabaseMetaData dmd = conn.getMetaData();
// Add missing columns
for (MessageNode node : MessageNode.getMessageNodes()) {
MsgCategory cat = messages.getCat(node);
if (cat == MsgCategory.TUTORIAL || cat == MsgCategory.ONE_TIME) {
ResultSet set = dmd.getColumns(null, null, msgTable, node.getColumnName());
if (!set.next()) {
String updateQuery =
String.format("ALTER TABLE %s ADD COLUMN %s", msgTable, node.getColumnName());
statement.executeUpdate(updateQuery);
}
}
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
try {
if (conn != null) conn.close();
if (statement != null) statement.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
}
private void send(Player player, MessageNode node, String message) {
switch (messages.getCat(node)) {
case NOTIFICATION:
if (player == null) {
plugin.getLogger().warning("Could not send the following message: " + message);
} else {
// FEATURE: don't spam messages
PlayerData playerData =
plugin.getModuleForClass(DataStoreModule.class).getPlayerData(player.getName());
long now = Calendar.getInstance().getTimeInMillis();
if (!node.equals(playerData.lastMessageSent)
|| now - playerData.lastMessageTimestamp > 30000) {
if (popupsAreEnabled(MsgCategory.NOTIFICATION))
sendPopup(player, MsgCategory.NOTIFICATION, message);
else player.sendMessage(message);
playerData.lastMessageSent = node;
playerData.lastMessageTimestamp = now;
}
}
break;
case TUTORIAL:
Validate.notNull(player);
if (persistModule.getCountFor(node, player.getName()) < messages.getMsgCount(node)) {
long now = Calendar.getInstance().getTimeInMillis();
if (!timeouts.contains(player.getName(), node)
|| now - timeouts.get(player.getName(), node) > 120000) // only if contains
{
timeouts.put(player.getName(), node, now);
String msgText = messages.getString(node);
if (manager != null) sendPopup(player, MsgCategory.TUTORIAL, msgText);
else
player.sendMessage(
ChatColor.DARK_RED + plugin.getTag() + ChatColor.WHITE + " " + msgText);
persistModule.increment(node, player.getName());
}
} else timeouts.remove(player, message);
break;
case BROADCAST:
plugin.getServer().broadcastMessage(message);
break;
default:
throw new UnsupportedOperationException(messages.getCat(node) + " not implemented");
}
}