From 3f88eb397295a76740b6d73d4cfbc418a19197a4 Mon Sep 17 00:00:00 2001
From: Earu <earu.derg@gmail.com>
Date: Sun, 31 Jul 2022 13:05:53 +0200
Subject: [PATCH] clean that up a bit

---
 lua/easychat/easychat.lua | 21 ++++++++-------------
 1 file changed, 8 insertions(+), 13 deletions(-)

diff --git a/lua/easychat/easychat.lua b/lua/easychat/easychat.lua
index 8de243d..9296a21 100644
--- a/lua/easychat/easychat.lua
+++ b/lua/easychat/easychat.lua
@@ -182,24 +182,19 @@ local function check_nick_override_wrapper_status()
 	timer.Simple(1, check_nick_override_wrapper_status)
 end
 
-PLY.Nick = make_nick_override_wrapper()
-PLY.Name = PLY.Nick
-PLY.GetName = PLY.Nick
-PLY.GetNick = PLY.Nick
+local new_wrapper = make_nick_override_wrapper()
+for _, fn_name in ipairs(clean_name_fns) do
+	PLY[fn_name] = new_wrapper
+end
+
+for _, fn_name in ipairs(tagged_name_fns) do
+	PLY[fn_name] = rich_nick_wrapper
+end
 
 PLY.RealNick = PLY.EngineNick
 PLY.RealName = PLY.EngineNick
 PLY.GetRealName = PLY.EngineNick
 
-PLY.RichNick = rich_nick_wrapper
-PLY.RichName = PLY.RichNick
-PLY.GetRichNick = PLY.RichNick
-PLY.GetRichName = PLY.RichNick
-PLY.GetNameDecorated = PLY.RichNick
-PLY.GetNickDecorated = PLY.RichNick
-PLY.NickDecorated = PLY.RichNick
-PLY.NameDecorated = PLY.RichNick
-
 timer.Simple(1, check_nick_override_wrapper_status)
 
 local load_modules, get_modules = include("easychat/autoloader.lua")