From e630426656b713ec67bfe7ede2f2081751e349db Mon Sep 17 00:00:00 2001 From: Linnea Gräf Date: Wed, 29 May 2024 21:09:07 +0200 Subject: [WIP] Add mod api --- .../apis/ingame/FirmamentCustomPayload.kt | 36 ++++++ .../moe/nea/firmament/apis/ingame/HypixelModAPI.kt | 56 +++++++++ .../firmament/apis/ingame/InGameCodecWrapper.kt | 51 ++++++++ .../firmament/apis/ingame/JoinedCustomPayload.kt | 25 ++++ .../apis/ingame/packets/PartyInfoRequest.kt | 134 +++++++++++++++++++++ .../events/FirmamentCustomPayloadEvent.kt | 15 +++ 6 files changed, 317 insertions(+) create mode 100644 src/main/kotlin/moe/nea/firmament/apis/ingame/FirmamentCustomPayload.kt create mode 100644 src/main/kotlin/moe/nea/firmament/apis/ingame/HypixelModAPI.kt create mode 100644 src/main/kotlin/moe/nea/firmament/apis/ingame/InGameCodecWrapper.kt create mode 100644 src/main/kotlin/moe/nea/firmament/apis/ingame/JoinedCustomPayload.kt create mode 100644 src/main/kotlin/moe/nea/firmament/apis/ingame/packets/PartyInfoRequest.kt create mode 100644 src/main/kotlin/moe/nea/firmament/events/FirmamentCustomPayloadEvent.kt (limited to 'src/main/kotlin/moe') diff --git a/src/main/kotlin/moe/nea/firmament/apis/ingame/FirmamentCustomPayload.kt b/src/main/kotlin/moe/nea/firmament/apis/ingame/FirmamentCustomPayload.kt new file mode 100644 index 0000000..34e39ab --- /dev/null +++ b/src/main/kotlin/moe/nea/firmament/apis/ingame/FirmamentCustomPayload.kt @@ -0,0 +1,36 @@ +/* + * SPDX-FileCopyrightText: 2024 Linnea Gräf + * + * SPDX-License-Identifier: GPL-3.0-or-later + */ + +package moe.nea.firmament.apis.ingame + +import io.netty.buffer.ByteBuf +import net.minecraft.network.codec.PacketCodec +import net.minecraft.network.packet.CustomPayload +import net.minecraft.util.Identifier + +interface FirmamentCustomPayload : CustomPayload { + + class Unhandled private constructor(val identifier: Identifier) : FirmamentCustomPayload { + override fun getId(): CustomPayload.Id { + return CustomPayload.id(identifier.toString()) + } + + companion object { + fun createCodec(identifier: Identifier): PacketCodec { + return object : PacketCodec { + override fun decode(buf: B): Unhandled { + return Unhandled(identifier) + } + + override fun encode(buf: B, value: Unhandled) { + // we will never send an unhandled packet stealthy + } + } + } + } + + } +} diff --git a/src/main/kotlin/moe/nea/firmament/apis/ingame/HypixelModAPI.kt b/src/main/kotlin/moe/nea/firmament/apis/ingame/HypixelModAPI.kt new file mode 100644 index 0000000..460df91 --- /dev/null +++ b/src/main/kotlin/moe/nea/firmament/apis/ingame/HypixelModAPI.kt @@ -0,0 +1,56 @@ +/* + * SPDX-FileCopyrightText: 2024 Linnea Gräf + * + * SPDX-License-Identifier: GPL-3.0-or-later + */ + +package moe.nea.firmament.apis.ingame + +import net.minecraft.network.packet.c2s.common.CustomPayloadC2SPacket +import net.minecraft.text.Text +import moe.nea.firmament.annotations.Subscribe +import moe.nea.firmament.apis.ingame.packets.PartyInfoRequest +import moe.nea.firmament.apis.ingame.packets.PartyInfoResponse +import moe.nea.firmament.commands.thenExecute +import moe.nea.firmament.events.CommandEvent +import moe.nea.firmament.events.FirmamentCustomPayloadEvent +import moe.nea.firmament.events.subscription.SubscriptionOwner +import moe.nea.firmament.features.FirmamentFeature +import moe.nea.firmament.features.debug.DeveloperFeatures +import moe.nea.firmament.util.MC + + +object HypixelModAPI : SubscriptionOwner { + init { + InGameCodecWrapper.Direction.C2S.customCodec = + InGameCodecWrapper.createStealthyCodec( + PartyInfoRequest.intoType() + ) + InGameCodecWrapper.Direction.S2C.customCodec = + InGameCodecWrapper.createStealthyCodec( + PartyInfoResponse.intoType() + ) + } + + @JvmStatic + fun sendRequest(packet: FirmamentCustomPayload) { + MC.networkHandler?.sendPacket(CustomPayloadC2SPacket(packet)) + } + + @Subscribe + fun testCommand(event: CommandEvent.SubCommand) { + event.subcommand("sendpartyrequest") { + thenExecute { + sendRequest(PartyInfoRequest(1)) + } + } + } + + @Subscribe + fun logEvents(event: FirmamentCustomPayloadEvent) { + MC.sendChat(Text.stringifiedTranslatable("firmament.modapi.event", event.toString())) + } + + override val delegateFeature: FirmamentFeature + get() = DeveloperFeatures +} diff --git a/src/main/kotlin/moe/nea/firmament/apis/ingame/InGameCodecWrapper.kt b/src/main/kotlin/moe/nea/firmament/apis/ingame/InGameCodecWrapper.kt new file mode 100644 index 0000000..0720dbf --- /dev/null +++ b/src/main/kotlin/moe/nea/firmament/apis/ingame/InGameCodecWrapper.kt @@ -0,0 +1,51 @@ +/* + * SPDX-FileCopyrightText: 2024 Linnea Gräf + * + * SPDX-License-Identifier: GPL-3.0-or-later + */ + +package moe.nea.firmament.apis.ingame + +import net.minecraft.network.PacketByteBuf +import net.minecraft.network.codec.PacketCodec +import net.minecraft.network.packet.CustomPayload + +class InGameCodecWrapper( + val wrapped: PacketCodec, + val direction: Direction, +) : PacketCodec { + enum class Direction { + S2C, + C2S, + ; + + var customCodec: PacketCodec = createStealthyCodec() + } + + companion object { + fun createStealthyCodec(vararg codecs: CustomPayload.Type): PacketCodec { + return CustomPayload.createCodec( + { FirmamentCustomPayload.Unhandled.createCodec(it) }, + codecs.toList() + ) as PacketCodec + } + + } + + override fun decode(buf: PacketByteBuf): CustomPayload { + val duplicateBuffer = PacketByteBuf(buf.slice()) + val original = wrapped.decode(buf) + val duplicate = direction.customCodec.decode(duplicateBuffer) + if (duplicate is FirmamentCustomPayload.Unhandled) + return original + return JoinedCustomPayload(original, duplicate) + } + + override fun encode(buf: PacketByteBuf, value: CustomPayload) { + if (value is FirmamentCustomPayload) { + direction.customCodec.encode(buf, value) + } else { + wrapped.encode(buf, value) + } + } +} diff --git a/src/main/kotlin/moe/nea/firmament/apis/ingame/JoinedCustomPayload.kt b/src/main/kotlin/moe/nea/firmament/apis/ingame/JoinedCustomPayload.kt new file mode 100644 index 0000000..c673264 --- /dev/null +++ b/src/main/kotlin/moe/nea/firmament/apis/ingame/JoinedCustomPayload.kt @@ -0,0 +1,25 @@ +/* + * SPDX-FileCopyrightText: 2024 Linnea Gräf + * + * SPDX-License-Identifier: GPL-3.0-or-later + */ + +package moe.nea.firmament.apis.ingame + +import net.minecraft.network.packet.CustomPayload + +/** + * A class to smuggle two parsed instances of the same custom payload packet. + */ +class JoinedCustomPayload( + val original: CustomPayload, + val smuggled: FirmamentCustomPayload +) : CustomPayload { + companion object { + val joinedId = CustomPayload.id("firmament:joined") + } + + override fun getId(): CustomPayload.Id { + return joinedId + } +} diff --git a/src/main/kotlin/moe/nea/firmament/apis/ingame/packets/PartyInfoRequest.kt b/src/main/kotlin/moe/nea/firmament/apis/ingame/packets/PartyInfoRequest.kt new file mode 100644 index 0000000..2b3d234 --- /dev/null +++ b/src/main/kotlin/moe/nea/firmament/apis/ingame/packets/PartyInfoRequest.kt @@ -0,0 +1,134 @@ +/* + * SPDX-FileCopyrightText: 2024 Linnea Gräf + * + * SPDX-License-Identifier: GPL-3.0-or-later + */ + +package moe.nea.firmament.apis.ingame.packets + +import io.netty.buffer.ByteBuf +import java.util.UUID +import net.minecraft.network.PacketByteBuf +import net.minecraft.network.codec.PacketCodec +import net.minecraft.network.codec.PacketCodecs +import net.minecraft.network.packet.CustomPayload +import net.minecraft.util.Uuids +import moe.nea.firmament.apis.ingame.FirmamentCustomPayload + +interface FirmamentCustomPayloadMeta { + val ID: CustomPayload.Id + val CODEC: PacketCodec + + fun intoType(): CustomPayload.Type { + return CustomPayload.Type(ID, CODEC) + } +} + +data class PartyInfoRequest(val version: Int) : FirmamentCustomPayload { + companion object : FirmamentCustomPayloadMeta { + override val ID = CustomPayload.id("hypixel:party_info") + override val CODEC = + PacketCodecs.VAR_INT.cast() + .xmap(::PartyInfoRequest, PartyInfoRequest::version) + } + + override fun getId(): CustomPayload.Id { + return ID + } +} + +sealed interface PartyInfoResponseV +sealed interface HypixelVersionedPacketData +data class HypixelSuccessfulResponse(val data: T) : HypixelVersionedPacketData +data class HypixelUnknownVersion(val version: Int) : HypixelVersionedPacketData +data class HypixelApiError(val label: String, val errorId: Int) : HypixelVersionedPacketData { + companion object { + fun createCodec(label: String): PacketCodec { + return PacketCodecs.VAR_INT + .cast() + .xmap({ HypixelApiError(label, it) }, HypixelApiError::errorId) + } + } +} + +object CodecUtils { + fun dispatchVersioned( + versions: Map>, + errorCodec: PacketCodec + ): PacketCodec> { + return object : PacketCodec> { + override fun decode(buf: B): HypixelVersionedPacketData { + if (!buf.readBoolean()) { + return errorCodec.decode(buf) + } + val version = buf.readVarInt() + val versionCodec = versions[version] + ?: return HypixelUnknownVersion(version) + return HypixelSuccessfulResponse(versionCodec.decode(buf)) + } + + override fun encode(buf: B, value: HypixelVersionedPacketData?) { + error("Cannot encode a hypixel packet") + } + } + } + + fun dispatchS2CBoolean( + ifTrue: PacketCodec, + ifFalse: PacketCodec + ): PacketCodec { + return object : PacketCodec { + override fun decode(buf: B): T { + return if (buf.readBoolean()) { + ifTrue.decode(buf) + } else { + ifFalse.decode(buf) + } + } + + override fun encode(buf: B, value: T) { + error("Cannot reverse dispatch boolean") + } + } + } + +} + + +data object PartyInfoResponseVUnknown : PartyInfoResponseV +data class PartyInfoResponseV1( + val leader: UUID?, + val members: Set, +) : PartyInfoResponseV { + data object PartyMember + companion object { + val CODEC: PacketCodec = + CodecUtils.dispatchS2CBoolean( + PacketCodec.tuple( + Uuids.PACKET_CODEC, PartyInfoResponseV1::leader, + Uuids.PACKET_CODEC.collect(PacketCodecs.toCollection(::HashSet)), PartyInfoResponseV1::members, + ::PartyInfoResponseV1 + ), + PacketCodec.unit(PartyInfoResponseV1(null, setOf()))) + } +} + + +data class PartyInfoResponse(val data: HypixelVersionedPacketData) : FirmamentCustomPayload { + companion object : FirmamentCustomPayloadMeta { + override val ID: CustomPayload.Id = CustomPayload.id("hypixel:party_info") + override val CODEC = + CodecUtils + .dispatchVersioned( + mapOf( + 1 to PartyInfoResponseV1.CODEC, + ), + HypixelApiError.createCodec("PartyInfoResponse")) + .xmap(::PartyInfoResponse, PartyInfoResponse::data) + + } + + override fun getId(): CustomPayload.Id { + return ID + } +} diff --git a/src/main/kotlin/moe/nea/firmament/events/FirmamentCustomPayloadEvent.kt b/src/main/kotlin/moe/nea/firmament/events/FirmamentCustomPayloadEvent.kt new file mode 100644 index 0000000..057fcef --- /dev/null +++ b/src/main/kotlin/moe/nea/firmament/events/FirmamentCustomPayloadEvent.kt @@ -0,0 +1,15 @@ +/* + * SPDX-FileCopyrightText: 2024 Linnea Gräf + * + * SPDX-License-Identifier: GPL-3.0-or-later + */ + +package moe.nea.firmament.events + +import moe.nea.firmament.apis.ingame.FirmamentCustomPayload + +data class FirmamentCustomPayloadEvent( + val payload: FirmamentCustomPayload +) : FirmamentEvent() { + companion object : FirmamentEventBus() +} -- cgit