Implement custom teleport click actions

Author: PotatoPresident

src/main/java/com/github/quiltservertools/ledger/mixin/ServerCommonPacketListenerImplMixin.java

@@ -1,12 +1,19 @@
 package com.github.quiltservertools.ledger.mixin;
 
 import com.github.quiltservertools.ledger.commands.subcommands.PageCommand;
+import com.github.quiltservertools.ledger.commands.subcommands.TeleportCommand;
 import com.github.quiltservertools.ledger.utility.MessageUtils;
+import net.minecraft.core.BlockPos;
+import net.minecraft.core.registries.Registries;
 import net.minecraft.nbt.Tag;
 import net.minecraft.network.protocol.common.ServerboundCustomClickActionPacket;
+import net.minecraft.resources.Identifier;
+import net.minecraft.resources.ResourceKey;
+import net.minecraft.server.level.ServerLevel;
 import net.minecraft.server.level.ServerPlayer;
 import net.minecraft.server.network.ServerCommonPacketListenerImpl;
 import net.minecraft.server.network.ServerGamePacketListenerImpl;
+import net.minecraft.world.level.Level;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
@@ -16,14 +23,30 @@
 public class ServerCommonPacketListenerImplMixin {
     @Inject(method = "handleCustomClickAction", at = @At("HEAD"), cancellable = true)
     private void handleLedgerPageChangeClick(ServerboundCustomClickActionPacket packet, CallbackInfo ci) {
-        if (!packet.id().equals(MessageUtils.INSTANCE.getPageChangeAction())) return;
         if (!((Object) this instanceof ServerGamePacketListenerImpl game)) return;
-        ServerPlayer player = game.player;
-        packet.payload().flatMap(Tag::asCompound)
-                .flatMap(root -> root.getInt("page"))
-                .ifPresent(page ->
-                        PageCommand.INSTANCE.page(player.createCommandSourceStack(), page)
+
+        if (packet.id().equals(MessageUtils.INSTANCE.getPageChangeAction())) {
+            ServerPlayer player = game.player;
+            packet.payload().flatMap(Tag::asCompound)
+                    .flatMap(root -> root.getInt("page"))
+                    .ifPresent(page ->
+                            PageCommand.INSTANCE.page(player.createCommandSourceStack(), page)
+                    );
+            ci.cancel();
+        } else if (packet.id().equals(MessageUtils.INSTANCE.getTeleportAction())) {
+            ServerPlayer player = game.player;
+            packet.payload().flatMap(Tag::asCompound).ifPresent(tag -> {
+                var pos = new BlockPos(
+                        tag.getIntOr("x", 0), 
+                        tag.getIntOr("y", 0), 
+                        tag.getIntOr("z", 0)
                 );
-        ci.cancel();
+                var world = Identifier.parse(tag.getStringOr("world", ""));
+                ResourceKey<Level> resourceKey = ResourceKey.create(Registries.DIMENSION, world);
+                ServerLevel serverLevel = game.player.level().getServer().getLevel(resourceKey);
+                TeleportCommand.INSTANCE.teleport(player, serverLevel, pos);
+            });
+            ci.cancel();
+        }
     }
 }

src/main/kotlin/com/github/quiltservertools/ledger/actions/AbstractActionType.kt

@@ -8,6 +8,7 @@ import com.github.quiltservertools.ledger.utility.literal
 import net.minecraft.ChatFormatting
 import net.minecraft.commands.CommandSourceStack
 import net.minecraft.core.BlockPos
+import net.minecraft.nbt.CompoundTag
 import net.minecraft.network.chat.ClickEvent
 import net.minecraft.network.chat.Component
 import net.minecraft.network.chat.HoverEvent
@@ -18,6 +19,7 @@ import net.minecraft.server.players.NameAndId
 import net.minecraft.util.Util
 import net.minecraft.world.level.Level
 import java.time.Instant
+import java.util.*
 import kotlin.time.ExperimentalTime
 
 abstract class AbstractActionType : ActionType {
@@ -98,15 +100,20 @@ abstract class AbstractActionType : ActionType {
     open fun getLocationMessage(): Component = "${pos.x} ${pos.y} ${pos.z}".literal()
         .setStyle(TextColorPallet.secondary)
         .withStyle {
+            val tag: CompoundTag = CompoundTag().apply { 
+                this.putInt("x", pos.x)
+                this.putInt("y", pos.y)
+                this.putInt("z", pos.z)
+                this.putString("world", (world ?: Level.OVERWORLD.identifier()).toString())
+            }
+
             it.withHoverEvent(
                 HoverEvent.ShowText(
                     Component.literal(world?.let { "$it\n" } ?: "")
                         .append(Component.translatable("text.ledger.action_message.location.hover"))
                 )
             ).withClickEvent(
-                ClickEvent.RunCommand(
-                    "/lg tp ${world ?: Level.OVERWORLD.identifier()} ${pos.x} ${pos.y} ${pos.z}"
-                )
+                ClickEvent.Custom(MessageUtils.teleportAction, Optional.of(tag))
             )
         }
 }

src/main/kotlin/com/github/quiltservertools/ledger/commands/subcommands/TeleportCommand.kt

@@ -9,7 +9,9 @@ import net.minecraft.commands.Commands
 import net.minecraft.commands.arguments.DimensionArgument
 import net.minecraft.commands.arguments.coordinates.Coordinates
 import net.minecraft.commands.arguments.coordinates.Vec3Argument
+import net.minecraft.core.BlockPos
 import net.minecraft.server.level.ServerLevel
+import net.minecraft.server.level.ServerPlayer
 
 object TeleportCommand : BuildableCommand {
     private const val BLOCK_CENTER_OFFSET = 0.5
@@ -35,6 +37,14 @@ object TeleportCommand : BuildableCommand {
         val player = context.source.playerOrException
         val pos = posArg.getBlockPos(context.source)
 
+        teleport(player, world, pos)
+
+        return 1
+    }
+
+    fun teleport(player: ServerPlayer, world: ServerLevel, pos: BlockPos) {
+        if (!Permissions.check(player, "ledger.commands.tp", CommandConsts.PERMISSION_LEVEL)) return
+
         val x = pos.x.toDouble() + BLOCK_CENTER_OFFSET
         val z = pos.z.toDouble() + BLOCK_CENTER_OFFSET
 
@@ -48,7 +58,5 @@ object TeleportCommand : BuildableCommand {
             player.xRot,
             true
         )
-
-        return 1
     }
 }

src/main/kotlin/com/github/quiltservertools/ledger/utility/MessageUtils.kt

@@ -25,6 +25,7 @@ import kotlin.time.toKotlinDuration
 
 object MessageUtils {
     val pageChangeAction: Identifier = Ledger.identifier("page-change")
+    val teleportAction: Identifier = Ledger.identifier("teleport")
 
     @OptIn(ExperimentalTime::class)
     suspend fun sendSearchResults(source: CommandSourceStack, results: SearchResults, header: Component) {