Simplified message model and refactored some class

This commit is contained in:
Anthony Robert committed 2018-11-04 20:17:41 -05:00
1 parent 66957f0eeb
commit 2750f6f71d
5 files changed
+16 -15

No files matched your search

@@ -59,6 +59,8 @@ public class DatabaseCommand extends AbstractCommand {
@Override @Override
public String execute(MessageReceivedEvent event, String args) { public String execute(MessageReceivedEvent event, String args) {
//TODO: Clean up the mess
event.getMessage().delete().queue();
if (!event.getAuthor().getDiscriminator().equals(Settings.getCreator().getDiscriminator())) if (!event.getAuthor().getDiscriminator().equals(Settings.getCreator().getDiscriminator()))
return null; return null;
@@ -69,7 +71,7 @@ public class DatabaseCommand extends AbstractCommand {
case "message": case "message":
if (countArgs(args) > 1) { if (countArgs(args) > 1) {
event.getMessage().delete().queue();
MMessage message = ((DBMessage) DBController.get("message")).getMessage(parseArgs(args, 1)); MMessage message = ((DBMessage) DBController.get("message")).getMessage(parseArgs(args, 1));
if (message != null) { if (message != null) {
EmbedBuilder embed = new EmbedBuilder(); EmbedBuilder embed = new EmbedBuilder();
@@ -97,14 +99,7 @@ public class DatabaseCommand extends AbstractCommand {
} }
embed.setFooter("Guild : " + event.getGuild().getName(), null); embed.setFooter("Guild : " + event.getGuild().getName(), null);
event.getAuthor().openPrivateChannel().queue(new Consumer<PrivateChannel>() { event.getAuthor().openPrivateChannel().queue(t -> t.sendMessage(embed.build()).queue());
@Override
public void accept(PrivateChannel t) {
t.sendMessage(embed.build()).queue();
}
});
} }
} }
@@ -67,7 +67,7 @@ public class EvalMethod extends AbstractCommand {
se.put("music", MusicManager.get(event.getGuild())); se.put("music", MusicManager.get(event.getGuild()));
try { try {
event.getTextChannel().sendMessage(" Evaluated Successfully:\n```\n"+se.eval("(function() {with (imports) {\nreturn " + args + "\n}})();") + " ```").queue(); event.getTextChannel().sendMessage(" Evaluated Successfully:\n```"+se.eval("(function() {with (imports) {\nreturn " + args + "\n}})();") + " ```").queue();
} }
catch(Exception e) { catch(Exception e) {
event.getTextChannel().sendMessage(" An exception was thrown:\n```\n"+e+" ```").queue(); event.getTextChannel().sendMessage(" An exception was thrown:\n```\n"+e+" ```").queue();
@@ -47,7 +47,7 @@ public class DBMessage extends AbstractController {
ResultSet rs = database.prepareAndExecute("SELECT * FROM log_message WHERE messageId = ?", messageId); ResultSet rs = database.prepareAndExecute("SELECT * FROM log_message WHERE messageId = ?", messageId);
MTableModel table = MysqlDatabase.convertToTableModel(rs); MTableModel table = MysqlDatabase.convertToTableModel(rs);
if (table.getNbRows() > 0) { if (table.getNbRows() > 0) {
return new MMessage(table.getRowModel(0).getItem(0), table.getRowModel(0).getItem(1), table.getRowModel(0).getItem(2), table.getRowModel(0).getItem(3), table.getRowModel(0).getItem(4), table.getRowModel(0).getItem(5)); return new MMessage(table.getRowModel(0));
} }
@@ -61,7 +61,7 @@ public class DBMessage extends AbstractController {
MTableModel table = MysqlDatabase.convertToTableModel(rs); MTableModel table = MysqlDatabase.convertToTableModel(rs);
if (table.getNbRows() > 0) { if (table.getNbRows() > 0) {
for (int row = 0; row < max; row++) { for (int row = 0; row < max; row++) {
messages.add(new MMessage(table.getRowModel(row).getItem(0), table.getRowModel(row).getItem(1), table.getRowModel(row).getItem(2), table.getRowModel(row).getItem(3), table.getRowModel(row).getItem(4), table.getRowModel(row).getItem(5))); messages.add(new MMessage(table.getRowModel(row)));
} }
} }
@@ -21,8 +21,14 @@ public class MMessage {
this.editedMessage = ""; this.editedMessage = "";
} }
public MMessage() { public MMessage(MRowModel row) {
this.serverName = (String) row.getItem(0);
this.channelName = (String) row.getItem(1);
this.sender = (String) row.getItem(2);
this.dateSent = (String) row.getItem(3);
this.messageId = (String) row.getItem(4);
this.message = (String) row.getItem(5);
this.editedMessage = "";
} }
public MMessage(Object serverName, Object channelName, Object sender, Object dateSent, Object messageId, Object message) { public MMessage(Object serverName, Object channelName, Object sender, Object dateSent, Object messageId, Object message) {
@@ -13,7 +13,7 @@ public class LogHandler {
public static void log(Exception e) { public static void log(Exception e) {
MusicBot.LOGGER.error(e.getMessage()); MusicBot.LOGGER.error(e.getMessage());
Settings.getLogChannel().sendMessage("Error : " + e.getMessage()).queue(); Settings.getLogChannel().sendMessage("Error : " + e).queue();
} }
} }