From 13254e61aea298b95490f45453083e99e3a04894 Mon Sep 17 00:00:00 2001
From: alemidev <me@alemi.dev>
Date: Wed, 10 Nov 2021 18:58:19 +0100
Subject: [PATCH] added helpers and chat parser

---
 aiocraft/helpers.py | 49 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 aiocraft/helpers.py

diff --git a/aiocraft/helpers.py b/aiocraft/helpers.py
new file mode 100644
index 0000000..a60cab7
--- /dev/null
+++ b/aiocraft/helpers.py
@@ -0,0 +1,49 @@
+from termcolor import colored
+
+_EQUIVALENTS = {
+	"dark_red" : "red",
+	"red" : "red",
+	"gold": "yellow",
+	"yellow": "yellow",
+	"dark_green": "green",
+	"green": "green",
+	"aqua": "cyan",
+	"dark_aqua": "cyan",
+	"dark_blue": "blue",
+	"blue": "blue",
+	"light_purple": "magenta",
+	"dark_purple": "magenta",
+	"white": "white",
+	"gray": "grey",
+	"dark_gray": "grey",
+	"black": "white"
+}
+
+def _parse_formatted_block(msg:dict) -> str:
+	attr = []
+	if "bold" in msg and msg["bold"]:
+		attr.append("bold")
+	if "underlined" in msg and msg["underlined"]:
+		attr.append("underline")
+	if "obfuscated" in msg and msg["obfuscated"]:
+		attr.append("blink")
+	if "color" in msg:
+		return colored(msg["text"], _EQUIVALENTS[msg["color"]], attrs=attr)
+	else:
+		return colored(msg["text"], "white", attrs=attr)
+
+def parse_chat(msg:dict, color:bool=True) -> str:
+	"""Recursive function to parse minecraft chat json, with optional colors"""
+	out = ""
+	if "text" in msg:
+		if color:
+			out += _parse_formatted_block(msg)
+		else:
+			out += msg["text"]
+	if "with" in msg:
+		for elem in msg["with"]:
+			out += parse_chat(elem, color)
+	if "extra" in msg:
+		for elem in msg["extra"]:
+			out += parse_chat(elem, color)
+	return out