Plex/src/main/java/dev/plex/command/impl/NotesCMD.java

190 lines
7.6 KiB
Java
Raw Normal View History

2022-04-07 02:00:28 +00:00
package dev.plex.command.impl;
import dev.plex.cache.DataUtils;
import dev.plex.command.PlexCommand;
import dev.plex.command.annotation.CommandParameters;
import dev.plex.command.annotation.CommandPermissions;
import dev.plex.player.PlexPlayer;
import dev.plex.punishment.extra.Note;
import dev.plex.rank.enums.Rank;
2022-04-07 07:37:31 +00:00
import dev.plex.storage.StorageType;
2022-04-07 02:00:28 +00:00
import dev.plex.util.PlexUtils;
import net.kyori.adventure.text.Component;
import net.kyori.adventure.text.format.NamedTextColor;
import net.kyori.adventure.text.format.TextDecoration;
2022-04-07 02:00:28 +00:00
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
2022-04-07 07:37:31 +00:00
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.atomic.AtomicReference;
2022-04-07 02:00:28 +00:00
@CommandParameters(name = "notes", description = "Manage notes for a player", usage = "/<command> <player> <list | add <note> | remove <id> | clear>")
@CommandPermissions(level = Rank.ADMIN, permission = "plex.notes")
public class NotesCMD extends PlexCommand
{
private static final DateTimeFormatter DATE_FORMAT = DateTimeFormatter.ofPattern("MM/dd/yyyy 'at' hh:mm:ss a");
@Override
protected Component execute(@NotNull CommandSender sender, @Nullable Player playerSender, String[] args)
{
if (args.length < 2)
{
return usage();
}
Player player = getNonNullPlayer(args[0]);
PlexPlayer plexPlayer = getPlexPlayer(player);
switch (args[1].toLowerCase())
{
case "list":
{
2022-04-07 07:37:31 +00:00
if (plugin.getStorageType() != StorageType.MONGODB)
2022-04-07 02:00:28 +00:00
{
2022-04-07 07:37:31 +00:00
plugin.getSqlNotes().getNotes(plexPlayer.getUuid()).whenComplete((notes, ex) ->
{
if (notes.size() == 0)
{
send(sender, mmString("<red>This player has no notes!"));
return;
}
readNotes(sender, plexPlayer, notes);
2022-04-07 07:37:31 +00:00
});
}
else
2022-04-07 07:37:31 +00:00
{
List<Note> notes = plexPlayer.getNotes();
if (notes.size() == 0)
{
2022-04-07 07:37:31 +00:00
return mmString("<red>This player has no notes!");
}
readNotes(sender, plexPlayer, notes);
2022-04-07 07:37:31 +00:00
}
2022-04-07 02:00:28 +00:00
return null;
}
case "add":
{
if (args.length < 3)
{
return usage();
}
String content = StringUtils.join(ArrayUtils.subarray(args, 2, args.length), " ");
if (playerSender != null)
{
Note note = new Note(plexPlayer.getUuid(), content, playerSender.getUniqueId(), LocalDateTime.now());
2022-04-07 02:00:28 +00:00
plexPlayer.getNotes().add(note);
2022-04-07 07:37:31 +00:00
if (plugin.getStorageType() != StorageType.MONGODB)
{
plugin.getSqlNotes().addNote(note);
}
else
2022-04-07 07:37:31 +00:00
{
DataUtils.update(plexPlayer);
}
2022-04-07 02:00:28 +00:00
return Component.text("Note added.").color(NamedTextColor.GREEN);
}
}
case "remove":
{
int id;
try
{
id = Integer.parseInt(args[2]);
}
catch (NumberFormatException ignored)
{
return Component.text("Invalid number: " + args[2]).color(NamedTextColor.RED);
}
2022-04-07 07:37:31 +00:00
if (plugin.getStorageType() != StorageType.MONGODB)
{
2022-04-07 07:37:31 +00:00
plugin.getSqlNotes().getNotes(plexPlayer.getUuid()).whenComplete((notes, ex) ->
{
2022-04-07 07:37:31 +00:00
for (Note note : notes)
{
2022-04-07 07:37:31 +00:00
if (note.getId() == id)
{
plugin.getSqlNotes().deleteNote(id, plexPlayer.getUuid()).whenComplete((notes1, ex1) ->
send(sender, Component.text("Removed note with ID: " + id).color(NamedTextColor.GREEN)));
}
else
2022-04-07 07:37:31 +00:00
{
send(sender, mmString("<red>A note with this ID could not be found"));
}
}
2022-04-07 07:37:31 +00:00
plexPlayer.getNotes().removeIf(note -> note.getId() == id);
});
}
else
2022-04-07 07:37:31 +00:00
{
if (plexPlayer.getNotes().removeIf(note -> note.getId() == id))
{
return Component.text("Removed note with ID: " + id).color(NamedTextColor.GREEN);
}
2022-04-07 07:37:31 +00:00
return mmString("<red>A note with this ID could not be found");
}
2022-04-07 02:00:28 +00:00
}
case "clear":
{
2022-04-07 07:37:31 +00:00
if (plugin.getStorageType() != StorageType.MONGODB)
2022-04-07 02:00:28 +00:00
{
2022-04-07 07:37:31 +00:00
plugin.getSqlNotes().getNotes(plexPlayer.getUuid()).whenComplete((notes, ex) ->
{
2022-04-07 07:37:31 +00:00
for (Note note : notes)
{
plugin.getSqlNotes().deleteNote(note.getId(), plexPlayer.getUuid());
}
plexPlayer.getNotes().clear();
send(sender, Component.text("Cleared " + notes.size() + " note(s).").color(NamedTextColor.GREEN));
});
}
else
{
2022-04-07 07:37:31 +00:00
int count = plexPlayer.getNotes().size();
plexPlayer.getNotes().clear();
DataUtils.update(plexPlayer);
return Component.text("Cleared " + count + " note(s).").color(NamedTextColor.GREEN);
}
return null;
2022-04-07 02:00:28 +00:00
}
default:
{
return usage();
}
}
}
private void readNotes(@NotNull CommandSender sender, PlexPlayer plexPlayer, List<Note> notes)
{
AtomicReference<Component> noteList = new AtomicReference<>(Component.text("Player notes for: " + plexPlayer.getName()).color(NamedTextColor.GREEN));
for (Note note : notes)
{
Component noteLine = Component.text(note.getId() + " - Written by: " + DataUtils.getPlayer(note.getWrittenBy()).getName() + " on " + DATE_FORMAT.format(note.getTimestamp())).color(NamedTextColor.YELLOW).decoration(TextDecoration.ITALIC, false);
noteLine = noteLine.append(Component.text(note.getNote())).color(NamedTextColor.GOLD).decoration(TextDecoration.ITALIC, true);
noteList.set(noteList.get().append(Component.newline()));
noteList.set(noteList.get().append(noteLine));
}
send(sender, noteList.get());
}
2022-04-07 02:00:28 +00:00
@Override
public @NotNull List<String> tabComplete(@NotNull CommandSender sender, @NotNull String alias, @NotNull String[] args) throws IllegalArgumentException
{
if (args.length == 1)
{
return PlexUtils.getPlayerNameList();
}
if (args.length == 2)
{
return Arrays.asList("list", "add", "remove", "clear");
}
return Collections.emptyList();
}
}