From 6b2ead46c83122edfab4b93eab293ed492d11360 Mon Sep 17 00:00:00 2001
From: replaceitem <janne.gruhler@gmail.com>
Date: Sun, 28 Feb 2021 13:50:54 +0100
Subject: [PATCH] Added list header and motd option to display_title()

---
 docs/scarpet/api/Auxiliary.md                 | 13 +++++-
 .../java/carpet/script/api/Auxiliary.java     | 45 ++++++++++++++++++-
 2 files changed, 55 insertions(+), 3 deletions(-)

diff --git a/docs/scarpet/api/Auxiliary.md b/docs/scarpet/api/Auxiliary.md
index cede1808..4b54660d 100644
--- a/docs/scarpet/api/Auxiliary.md
+++ b/docs/scarpet/api/Auxiliary.md
@@ -264,9 +264,20 @@ Sends the player (or players if `players` is a list) a title of a specific type,
    Executing with those will set the times to the specified ones.
    Note that `actionbar` type doesn't support changing times (vanilla bug, see [MC-106167](https://bugs.mojang.com/browse/MC-106167)).
 
+### `display_title(players, 'motd', motd),`
+
+Changes the motd (message of the day) of the server. The `player` value is ignored for the `'motd'` text.
+Note that when the `customMOTD` carpet rule is active, this will not work and instead use the MOTD from the rule.
+
+### `display_title(players, 'header', header, footer),`
+
+Changes the header and footer of the player list.
+Keep in mind this would interfere with carpet's loggers, so if you are using this, you need to keep them off.
+
 ### `logger(msg), logger(type, msg)`
 
-Prints the message to system logs, and not to chat. By default prints an info, unless you specify otherwise in the `type` parameter.
+Prints the message to system logs, and not to chat.
+By default prints an info, unless you specify otherwise in the `type` parameter.
 
 Available output types:
 
diff --git a/src/main/java/carpet/script/api/Auxiliary.java b/src/main/java/carpet/script/api/Auxiliary.java
index a0268cc7..1f1f16a2 100644
--- a/src/main/java/carpet/script/api/Auxiliary.java
+++ b/src/main/java/carpet/script/api/Auxiliary.java
@@ -6,6 +6,7 @@ import carpet.fakes.MinecraftServerInterface;
 import carpet.fakes.StatTypeInterface;
 import carpet.fakes.ThreadedAnvilChunkStorageInterface;
 import carpet.helpers.FeatureGenerator;
+import carpet.mixins.PlayerListHeaderS2CPacketMixin;
 import carpet.script.bundled.Module;
 import carpet.script.CarpetContext;
 import carpet.script.CarpetEventServer;
@@ -48,6 +49,7 @@ import net.minecraft.nbt.NbtHelper;
 import net.minecraft.nbt.StringTag;
 import net.minecraft.nbt.Tag;
 import net.minecraft.network.packet.s2c.play.PlaySoundIdS2CPacket;
+import net.minecraft.network.packet.s2c.play.PlayerListHeaderS2CPacket;
 import net.minecraft.network.packet.s2c.play.TitleS2CPacket;
 import net.minecraft.network.packet.s2c.play.TitleS2CPacket.Action;
 import net.minecraft.particle.ParticleEffect;
@@ -508,7 +510,8 @@ public class Auxiliary {
                 return player;
             });
             TitleS2CPacket.Action action;
-            switch (lv.get(1).evalValue(c).getString().toLowerCase(Locale.ROOT))
+            String actionString = lv.get(1).evalValue(c).getString().toLowerCase(Locale.ROOT);
+            switch (actionString)
             {
                 case "title":
                     action = Action.TITLE;
@@ -522,8 +525,46 @@ public class Auxiliary {
                 case "clear":
                     action = Action.CLEAR;
                     break;
+                case "header":
+                    Text header;
+                    Text footer;
+                    if (lv.size() > 3)
+                    {
+                        pVal = lv.get(2).evalValue(c);
+                        if (pVal instanceof FormattedTextValue)
+                            header = ((FormattedTextValue) pVal).getText();
+                        else
+                            header = new LiteralText(pVal.getString());
+                        pVal = lv.get(3).evalValue(c);
+                        if (pVal instanceof FormattedTextValue)
+                            footer = ((FormattedTextValue) pVal).getText();
+                        else
+                            footer = new LiteralText(pVal.getString());
+                    } else throw new InternalExpressionException("Third and fourth argument of 'display_title' must be present for 'header' type");
+                    PlayerListHeaderS2CPacket packet = new PlayerListHeaderS2CPacket();
+                    ((PlayerListHeaderS2CPacketMixin) packet).setFooter(footer);
+                    ((PlayerListHeaderS2CPacketMixin) packet).setHeader(header);
+                    AtomicInteger total = new AtomicInteger(0);
+                    targets.forEach(p -> {
+                        p.networkHandler.sendPacket(packet);
+                        total.getAndIncrement();
+                    });
+                    Value ret = NumericValue.of(total.get());
+                    return (cc, tt) -> ret;
+                case "motd":
+                    Text title;
+                    if (lv.size() > 2)
+                    {
+                        pVal = lv.get(2).evalValue(c);
+                        if (pVal instanceof FormattedTextValue)
+                            title = ((FormattedTextValue) pVal).getText();
+                        else
+                            title = new LiteralText(pVal.getString());
+                    } else throw new InternalExpressionException("Third argument of 'display_title' must be present except for 'clear' type");
+                    server.getServerMetadata().setDescription(title);
+                    return LazyValue.TRUE;
                 default:
-                    throw new InternalExpressionException("'display_title' requires 'title', 'subtitle', 'actionbar' or 'clear' as second argument");
+                    throw new InternalExpressionException("'display_title' requires 'title', 'subtitle', 'actionbar', 'clear', 'header' or 'motd' as second argument");
             }
             if (action != Action.CLEAR && lv.size() < 3)
                 throw new InternalExpressionException("Third argument of 'display_title' must be present except for 'clear' type");