class="kw">package com.hypixel.hytale.server.npc.commands;

class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.server.core.Message;
class="kw">import com.hypixel.hytale.server.core.command.system.CommandContext;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.system.RequiredArg;
class="kw">import com.hypixel.hytale.server.core.universe.world.World;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.BuilderInfo;
class="kw">import com.hypixel.hytale.server.npc.entities.NPCEntity;
class="kw">import com.hypixel.hytale.server.npc.systems.RoleChangeSystem;
class="kw">import javax.annotation.Nonnull;

class="kw">public class NPCRoleCommand class="kw">extends NPCWorldCommandBase {
   @Nonnull
   class="kw">private class="kw">final RequiredArg<BuilderInfo> roleArg = this.withRequiredArg("role", "server.commands.npc.role.role.desc", NPCCommand.NPC_ROLE);

   class="kw">public NPCRoleCommand() {
      super("role", "server.commands.npc.role.desc");
      this.addUsageVariant(new NPCRoleCommand.GetRoleCommand());
   }

   @Override
   class="kw">protected void execute(
      @Nonnull CommandContext var1, @Nonnull NPCEntity var2, @Nonnull World var3, @Nonnull Store<EntityStore> var4, @Nonnull Ref<EntityStore> var5
   ) {
      BuilderInfo var6 = this.roleArg.get(var1);
      if (var2.getRole().isRoleChangeRequested()) {
         var1.sendMessage(Message.translation("server.commands.npc.role.unableToSetRole"));
      } else {
         RoleChangeSystem.requestRoleChange(var5, var2.getRole(), var6.getIndex(), true, var4);
         var1.sendMessage(Message.translation("server.commands.npc.role.roleSet").param("role", var6.getKeyName()));
      }
   }

   class="kw">public class="kw">static class GetRoleCommand class="kw">extends NPCWorldCommandBase {
      class="kw">public GetRoleCommand() {
         super("server.commands.npc.role.get.desc");
      }

      @Override
      class="kw">protected void execute(
         @Nonnull CommandContext var1, @Nonnull NPCEntity var2, @Nonnull World var3, @Nonnull Store<EntityStore> var4, @Nonnull Ref<EntityStore> var5
      ) {
         var1.sendMessage(Message.translation("server.commands.npc.role.npcHasRole").param("role", var2.getRoleName()));
      }
   }
}