Skip to content

Commit

Permalink
ChatImage Protocol (#32)
Browse files Browse the repository at this point in the history 
Add ChatImage protocol
  • Loading branch information
@HaHaWTH
HaHaWTH authored Mar 13, 2024
1 parent de147d2 commit e80f541
Showing 1 changed file with 195 additions and 0 deletions.
195 changes: 195 additions & 0 deletions patches/server/0076-Leaf-chat-image-protocol.patch
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,195 @@
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
From: HaHaWTH <[email protected]>
Date: Wed, 13 Mar 2024 18:11:10 +0800
Subject: [PATCH] Leaf-chat-image-protocol


diff --git a/src/main/java/org/dreeam/leaf/config/modules/network/ProtocolSupport.java b/src/main/java/org/dreeam/leaf/config/modules/network/ProtocolSupport.java
index 3e5e27efac1791cce961ec144405d8215ab87996..1c1abbb3a74a519520d2c1229721887c19d9df2e 100644
--- a/src/main/java/org/dreeam/leaf/config/modules/network/ProtocolSupport.java
+++ b/src/main/java/org/dreeam/leaf/config/modules/network/ProtocolSupport.java
@@ -25,6 +25,9 @@ public class ProtocolSupport implements IConfigModule {
@ConfigInfo(baseName = "appleskin-protocol")
public static boolean appleskinProtocol = false;

+ @ConfigInfo(baseName = "chatImage-protocol")
+ public static boolean chatImageProtocol = false;
+
@ConfigInfo(baseName = "xaero-map-protocol")
public static boolean xaeroMapProtocol = false;
@ConfigInfo(baseName = "xaero-map-server-id")
diff --git a/src/main/java/top/leavesmc/leaves/protocol/ChatImageProtocol.java b/src/main/java/top/leavesmc/leaves/protocol/ChatImageProtocol.java
new file mode 100644
index 0000000000000000000000000000000000000000..5e97da51249a45db93e4b5d2921784d58596abde
--- /dev/null
+++ b/src/main/java/top/leavesmc/leaves/protocol/ChatImageProtocol.java
@@ -0,0 +1,149 @@
+package top.leavesmc.leaves.protocol;
+
+import com.google.common.collect.Lists;
+import com.google.gson.Gson;
+import net.minecraft.network.FriendlyByteBuf;
+import net.minecraft.network.protocol.Packet;
+import net.minecraft.network.protocol.common.custom.CustomPacketPayload;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.server.level.ServerPlayer;
+import org.dreeam.leaf.config.modules.network.ProtocolSupport;
+import org.jetbrains.annotations.Contract;
+import org.jetbrains.annotations.NotNull;
+import top.leavesmc.leaves.protocol.chatimage.ChatImageIndex;
+import top.leavesmc.leaves.protocol.core.LeavesProtocol;
+import top.leavesmc.leaves.protocol.core.ProtocolHandler;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.UUID;
+
+@LeavesProtocol(namespace = "chatimage")
+public class ChatImageProtocol {
+ public static final String PROTOCOL_ID = "chatimage";
+ private static final Map<String, HashMap<Integer, String>> SERVER_BLOCK_CACHE = new HashMap<>();
+ private static final Map<String, Integer> FILE_COUNT_MAP = new HashMap<>();
+ private static final Map<String, List<String>> USER_CACHE_MAP = new HashMap<>();
+ public static int MAX_STRING = 532767;
+ private static final Gson gson = new Gson();
+
+ public record FileInfoChannelPacket(String message) implements CustomPacketPayload {
+ private static final ResourceLocation FILE_INFO = ChatImageProtocol.id("file_info");
+
+ public FileInfoChannelPacket(ResourceLocation id,FriendlyByteBuf buffer) {
+ this(buffer.readUtf(MAX_STRING));
+ }
+
+ @Override
+ public void write(final FriendlyByteBuf buffer) {
+ buffer.writeUtf(message(), MAX_STRING);
+ }
+
+ @Override
+ public @NotNull ResourceLocation id() {
+ return FILE_INFO;
+ }
+ }
+
+ public record DownloadFileChannelPacket(String message) implements CustomPacketPayload {
+ /**
+ * 发送文件分块到客户端通道(Map)
+ */
+ private static final ResourceLocation DOWNLOAD_FILE_CHANNEL = ChatImageProtocol.id("download_file_channel");
+
+ public DownloadFileChannelPacket(ResourceLocation id,FriendlyByteBuf buffer) {
+ this(buffer.readUtf(MAX_STRING));
+ }
+ @Override
+ public void write(final FriendlyByteBuf buffer) {
+ buffer.writeUtf(message(),MAX_STRING);
+ }
+
+ @Override
+ public @NotNull ResourceLocation id() {
+ return DOWNLOAD_FILE_CHANNEL;
+ }
+
+ }
+ public record FileChannelPacket(String message) implements CustomPacketPayload {
+ /**
+ * 客户端发送文件分块到服务器通道(Map)
+ */
+ private static final ResourceLocation FILE_CHANNEL = ChatImageProtocol.id( "file_channel");
+ public FileChannelPacket(ResourceLocation id,FriendlyByteBuf buffer) {
+ this(buffer.readUtf(MAX_STRING)) ;
+ }
+ @Override
+ public void write(final FriendlyByteBuf buffer) {
+ buffer.writeUtf(message(), MAX_STRING);
+ }
+
+ @Override
+ public @NotNull ResourceLocation id() {
+ return FILE_CHANNEL;
+ }
+
+ }
+ @ProtocolHandler.PayloadReceiver(payload = FileChannelPacket.class, payloadId = "file_channel")
+ public static void serverFileChannelReceived(ServerPlayer player, String res) {
+ if (!ProtocolSupport.chatImageProtocol) return;
+ ChatImageIndex title = gson.fromJson(res, ChatImageIndex.class);
+ HashMap<Integer, String> blocks = SERVER_BLOCK_CACHE.containsKey(title.url) ? SERVER_BLOCK_CACHE.get(title.url) : new HashMap<>();
+ blocks.put(title.index, res);
+ SERVER_BLOCK_CACHE.put(title.url, blocks);
+ FILE_COUNT_MAP.put(title.url, title.total);
+ //System.out.println("[FileChannel->Server:" + title.index + "/" + title.total + "]" + title.url);
+ if (title.total == blocks.size()) {
+ if (USER_CACHE_MAP.containsKey(title.url)) {
+ // 通知之前请求但是没图片的客户端
+ List<String> names = USER_CACHE_MAP.get(title.url);
+ for (String uuid : names) {
+ ServerPlayer serverPlayer = player.server.getPlayerList().getPlayer(UUID.fromString(uuid));
+ if (serverPlayer != null) {
+ sendToPlayer(new FileInfoChannelPacket("true->" + title.url), serverPlayer);
+ }
+ //System.out.println("[echo to client(" + uuid + ")]" + title.url);
+ }
+ USER_CACHE_MAP.put(title.url, Lists.newArrayList());
+ }
+ //System.out.println("[FileChannel->Server]" + title.url);
+ }
+ }
+ @ProtocolHandler.PayloadReceiver(payload = FileInfoChannelPacket.class, payloadId = "file_info")
+ public static void serverFileInfoChannelReceived(ServerPlayer player, String url) {
+ if (!ProtocolSupport.chatImageProtocol) return;
+ if (SERVER_BLOCK_CACHE.containsKey(url) && FILE_COUNT_MAP.containsKey(url)) {
+ HashMap<Integer, String> list = SERVER_BLOCK_CACHE.get(url);
+ Integer total = FILE_COUNT_MAP.get(url);
+ if (total == list.size()) {
+ // 服务器存在缓存图片,直接发送给客户端
+ for (Map.Entry<Integer, String> entry : list.entrySet()) {
+ //System.out.println("[GetFileChannel->Client:" + entry.getKey() + "/" + (list.size() - 1) + "]" + url);
+ sendToPlayer(new DownloadFileChannelPacket(entry.getValue()), player);
+ }
+ //System.out.println("[GetFileChannel->Client]" + url);
+ return;
+ }
+ }
+ //通知客户端无文件
+ sendToPlayer(new FileInfoChannelPacket("null->" + url), player);
+ //System.out.println("[GetFileChannel]not found in server:" + url);
+ // 记录uuid,后续有文件了推送
+ List<String> names = USER_CACHE_MAP.containsKey(url) ? USER_CACHE_MAP.get(url) : Lists.newArrayList();
+ names.add(player.getStringUUID());
+ USER_CACHE_MAP.put(url, names);
+ //System.out.println("[GetFileChannel]记录uuid:" + player.getStringUUID());
+ //System.out.println("[not found in server]" + url);
+ }
+
+ @Contract("_ -> new")
+ public static @NotNull ResourceLocation id(String path) {
+ return new ResourceLocation(PROTOCOL_ID, path);
+ }
+
+ public static void sendToPlayer(CustomPacketPayload payload, ServerPlayer player) {
+ player.connection.send((Packet<?>) payload);
+ }
+
+}
diff --git a/src/main/java/top/leavesmc/leaves/protocol/chatimage/ChatImageIndex.java b/src/main/java/top/leavesmc/leaves/protocol/chatimage/ChatImageIndex.java
new file mode 100644
index 0000000000000000000000000000000000000000..bd4f2f605590653d0da61dc64d62420bda06ceee
--- /dev/null
+++ b/src/main/java/top/leavesmc/leaves/protocol/chatimage/ChatImageIndex.java
@@ -0,0 +1,13 @@
+package top.leavesmc.leaves.protocol.chatimage;
+public class ChatImageIndex {
+ public int index;
+ public int total;
+ public String url;
+ public String bytes;
+ public ChatImageIndex(int index, int total, String url, String bytes){
+ this.index = index;
+ this.total = total;
+ this.url = url;
+ this.bytes = bytes;
+ }
+}
\ No newline at end of file

0 comments on commit e80f541

Please sign in to comment.