Add set subcommand to the tags admin command

broccolai · Feb 21, 2021 · a273f7c · a273f7c
1 parent 1af1598
commit a273f7c
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 1 deletion.
diff --git a/api/src/main/java/broccolai/tags/api/service/MessageService.java b/api/src/main/java/broccolai/tags/api/service/MessageService.java
@@ -25,6 +25,8 @@ public interface MessageService extends Service {
 
     Component commandAdminList(@NonNull Collection<Tag> tags);
 
+    Component commandAdminSet(@NonNull Tag tag, @NonNull TagsUser target);
+
     Component commandErrorUserNotFound(@NonNull String input);
 
     Component commandErrorTagNotFound(@NonNull String input);

diff --git a/core/src/main/java/broccolai/tags/core/commands/TagsAdminCommand.java b/core/src/main/java/broccolai/tags/core/commands/TagsAdminCommand.java
@@ -12,10 +12,11 @@
 import cloud.commandframework.CommandManager;
 import cloud.commandframework.context.CommandContext;
 import com.google.inject.Inject;
-import org.checkerframework.checker.nullness.qual.NonNull;
 
 import java.util.Collection;
 
+import org.checkerframework.checker.nullness.qual.NonNull;
+
 public final class TagsAdminCommand implements PluginCommand {
 
     private final @NonNull CloudArgumentFactory argumentFactory;
@@ -65,6 +66,14 @@ public void register(
                 .argument(this.argumentFactory.user("target", false))
                 .handler(this::handleList)
         );
+
+        commandManager.command(tagsCommand
+                .literal("set")
+                .permission("tags.command.admin.set")
+                .argument(this.argumentFactory.user("target", true))
+                .argument(this.argumentFactory.tag("tag", TagParserMode.TARGET))
+                .handler(this::handleSet)
+        );
     }
 
     private void handleGive(final @NonNull CommandContext<CommandUser> context) {
@@ -94,4 +103,13 @@ private void handleList(final @NonNull CommandContext<CommandUser> context) {
         sender.sendMessage(this.messageService.commandAdminList(tags));
     }
 
+    private void handleSet(final @NonNull CommandContext<CommandUser> context) {
+        CommandUser sender = context.getSender();
+        TagsUser target = context.get("target");
+        Tag tag = context.get("tag");
+
+        target.setCurrent(tag);
+        sender.sendMessage(this.messageService.commandAdminSet(tag, target));
+    }
+
 }
diff --git a/core/src/main/java/broccolai/tags/core/config/LocaleConfiguration.java b/core/src/main/java/broccolai/tags/core/config/LocaleConfiguration.java
@@ -65,6 +65,9 @@ public static final class AdminFeedbackLocaleConfiguration {
             @Setting
             public LocaleEntry remove = new BasicLocaleEntry("<prefix> Tag <tag> has been removed from <target>");
 
+            @Setting
+            public LocaleEntry set = new BasicLocaleEntry("<prefix> <target> had their tag set to <tag>");
+
         }
 
         @ConfigSerializable

diff --git a/core/src/main/java/broccolai/tags/core/service/message/MiniMessageService.java b/core/src/main/java/broccolai/tags/core/service/message/MiniMessageService.java
@@ -99,6 +99,14 @@ public Component commandAdminList(final @NonNull Collection<Tag> tags) {
         return component;
     }
 
+    @Override
+    public Component commandAdminSet(@NonNull final Tag tag, @NonNull final TagsUser target) {
+        Template tagTemplate = Template.of("tag", tag.component());
+        Template targetTemplate = Template.of("target", this.nameFromUser(target));
+
+        return this.locale.commands.admin.list.asComponent(this.prefix(), tagTemplate, targetTemplate);
+    }
+
     @Override
     public Component commandErrorUserNotFound(final @NonNull String input) {
         Template inputTemplate = Template.of("input", input);