From a909bf763a661401cb03c6aac04effa958a4e632 Mon Sep 17 00:00:00 2001
From: Sadie Powell
Date: Mon, 17 Jul 2023 10:06:39 +0100
Subject: Move checks for whether a user is cloakable to {Get,Reset}Cloaks.

---
 src/modules/m_cloak.cpp | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'src/modules/m_cloak.cpp')

diff --git a/src/modules/m_cloak.cpp b/src/modules/m_cloak.cpp
index 52545552a..51c599742 100644
--- a/src/modules/m_cloak.cpp
+++ b/src/modules/m_cloak.cpp
@@ -106,6 +106,9 @@ public:
 
 	Cloak::List* GetCloaks(LocalUser* user) override
 	{
+		if (user->quitting || !(user->connected & User::CONN_NICKUSER))
+			return nullptr; // This user isn't at a point where they can change their cloak.
+
 		if (!user->GetClass()->config->getBool("usecloak", true))
 			return nullptr;
 
@@ -148,6 +151,9 @@ public:
 
 	void ResetCloaks(LocalUser* user, bool resetdisplay) override
 	{
+		if (user->quitting || !(user->connected & User::CONN_NICKUSER))
+			return; // This user isn't at a point where they can change their cloak.
+
 		const std::string oldcloak = GetFrontCloak(user);
 		ext.Unset(user);
 
@@ -368,10 +374,6 @@ public:
 
 	void OnChangeRemoteAddress(LocalUser* user) override
 	{
-		// Connecting users are handled in OnUserConnect not here.
-		if (!user->IsFullyConnected() || user->quitting)
-			return;
-
 		// Remove the cloaks so we can generate new ones.
 		cloakapi.ResetCloaks(user, false);
 
-- 
cgit v1.3.1-10-gc9f91