m_spanningtree Allow server-to-server command handlers to specify whether they accept servers, remote users or both as the command source

To make life easier for handlers accepting servers only as source, pass them a TreeServer* so they don't have to call FindServer()

1 files changed, 2 insertions, 10 deletions

diff --git a/src/modules/m_spanningtree/misccommands.cpp b/src/modules/m_spanningtree/misccommands.cpp
index 6e66c68a7..5b04c73bc 100644
--- a/src/modules/m_spanningtree/misccommands.cpp
+++ b/src/modules/m_spanningtree/misccommands.cpp
@@ -35,22 +35,14 @@ CmdResult CommandSNONotice::Handle(User* user, std::vector<std::string>& params)
 	return CMD_SUCCESS;
 }
 
-CmdResult CommandBurst::Handle(User* user, std::vector<std::string>& params)
+CmdResult CommandBurst::HandleServer(TreeServer* server, std::vector<std::string>& params)
 {
-	if (!IS_SERVER(user))
-		return CMD_INVALID;
-
-	TreeServer* server = Utils->FindServer(user->server);
 	server->bursting = true;
 	return CMD_SUCCESS;
 }
 
-CmdResult CommandEndBurst::Handle(User* user, std::vector<std::string>& params)
+CmdResult CommandEndBurst::HandleServer(TreeServer* server, std::vector<std::string>& params)
 {
-	if (!IS_SERVER(user))
-		return CMD_INVALID;
-
-	TreeServer* server = Utils->FindServer(user->server);
 	server->FinishBurst();
 	return CMD_SUCCESS;
 }