Include explicit routing information in Command, will replace CMD_LOCALONLY return value

git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@11601 e03df62e-2008-0410-955e-edbf42e46eb7

2 files changed, 45 insertions, 61 deletions

diff --git a/src/command_parse.cpp b/src/command_parse.cpp
index b446d2b9a..b063e392c 100644
--- a/src/command_parse.cpp
+++ b/src/command_parse.cpp
@@ -607,20 +607,29 @@ void CommandParser::SetupCommandTable()
 		this->CreateCommand(new CommandReload(ServerInstance));
 }
 
-int CommandParser::TranslateUIDs(const std::vector<TranslateType> to, const std::vector<std::string> &source, std::string &dest)
+int CommandParser::TranslateUIDs(const std::vector<TranslateType> to, const std::vector<std::string> &source, std::string &dest, bool prefix_final, Command* custom_translator)
 {
-	std::vector<std::string>::const_iterator items = source.begin();
 	std::vector<TranslateType>::const_iterator types = to.begin();
 	User* user = NULL;
+	unsigned int i;
 	int translations = 0;
 	dest.clear();
 
-	while (items != source.end() && types != to.end())
+	for(i=0; i < source.size(); i++)
 	{
-		TranslateType t = *types;
-		std::string item = *items;
-		types++;
-		items++;
+		TranslateType t;
+		std::string item = source[i];
+
+		if (types == to.end())
+			t = TR_TEXT;
+		else
+		{
+			t = *types;
+			types++;
+		}
+
+		if (prefix_final && i == source.size() - 1)
+			dest.append(":");
 
 		switch (t)
 		{
@@ -635,6 +644,10 @@ int CommandParser::TranslateUIDs(const std::vector<TranslateType> to, const std:
 				else
 					dest.append(item);
 			break;
+			case TR_CUSTOM:
+				if (custom_translator)
+					custom_translator->EncodeParameter(item, i);
+				dest.append(item);
 			break;
 			case TR_END:
 			case TR_TEXT:
@@ -643,11 +656,10 @@ int CommandParser::TranslateUIDs(const std::vector<TranslateType> to, const std:
 				dest.append(item);
 			break;
 		}
-		dest.append(" ");
+		if (i != source.size() - 1)
+			dest.append(" ");
 	}
 
-	if (!dest.empty())
-		dest.erase(dest.end() - 1);
 	return translations;
 }
 
diff --git a/src/modules/m_spanningtree/postcommand.cpp b/src/modules/m_spanningtree/postcommand.cpp
index 719ff0046..428079c23 100644
--- a/src/modules/m_spanningtree/postcommand.cpp
+++ b/src/modules/m_spanningtree/postcommand.cpp
@@ -29,63 +29,35 @@
 
 void ModuleSpanningTree::OnPostCommand(const std::string &command, const std::vector<std::string>& parameters, User *user, CmdResult result, const std::string &original_line)
 {
-	if ((result == CMD_SUCCESS) && (ServerInstance->IsValidModuleCommand(command, parameters.size(), user)))
-	{
-		/* Safe, we know its non-null because IsValidModuleCommand returned true */
-		Command* thiscmd = ServerInstance->Parser->GetHandler(command);
+	if (result != CMD_SUCCESS)
+		return;
+	if (!ServerInstance->IsValidModuleCommand(command, parameters.size(), user))
+		return;
 
-		Module* srcmodule = ServerInstance->Modules->Find(thiscmd->source);
+	/* We know it's non-null because IsValidModuleCommand returned true */
+	Command* thiscmd = ServerInstance->Parser->GetHandler(command);
 
-		if (srcmodule && !(srcmodule->GetVersion().Flags & VF_COMMON)) {
-			ServerInstance->Logs->Log("m_spanningtree",ERROR,"Routed command %s from non-VF_COMMON module %s",
-				command.c_str(), thiscmd->source.c_str());
-			return;
-		}
+	RouteDescriptor routing = thiscmd->GetRouting(user, parameters);
 
-		// this bit of code cleverly routes all module commands
-		// to all remote severs *automatically* so that modules
-		// can just handle commands locally, without having
-		// to have any special provision in place for remote
-		// commands and linking protocols.
-		parameterlist params;
-		params.clear();
-		unsigned int n_translate = thiscmd->translation.size();
-		TranslateType translate_to;
+	if (routing.type == ROUTE_TYPE_LOCALONLY)
+		return;
 
-		/* To make sure that parameters with spaces, or empty
-		 * parameters, etc, are always sent properly, *always*
-		 * prefix the last parameter with a :. This also removes
-		 * an extra strchr() */
-		for (unsigned int j = 0; j < parameters.size(); j++)
-		{
-			std::string target;
+	Module* srcmodule = ServerInstance->Modules->Find(thiscmd->source);
 
-			/* Map all items to UUIDs where neccessary */
-			if (j < n_translate)
-			{
-				/* We have a translation mapping for this index */
-				translate_to = thiscmd->translation[j] != TR_END ? thiscmd->translation[j] : TR_TEXT;
-			}
-			else
-				translate_to = TR_TEXT;
+	if (srcmodule && !(srcmodule->GetVersion().Flags & VF_COMMON)) {
+		ServerInstance->Logs->Log("m_spanningtree",ERROR,"Routed command %s from non-VF_COMMON module %s",
+			command.c_str(), thiscmd->source.c_str());
+		return;
+	}
+
+	std::string output_text;
+	ServerInstance->Parser->TranslateUIDs(thiscmd->translation, parameters, output_text, true, thiscmd);
 
-			ServerInstance->Logs->Log("m_spanningtree",DEBUG,"TRANSLATION: %s - type is %d", parameters[j].c_str(), translate_to);
-			if (translate_to == TR_CUSTOM)
-			{
-				target = parameters[j];
-				thiscmd->EncodeParameter(target, j);
-			}
-			else
-			{
-				ServerInstance->Parser->TranslateUIDs(translate_to, parameters[j], target);
-			}
+	parameterlist params;
+	params.push_back(output_text);
 
-			if (j == (parameters.size() - 1))
-				params.push_back(":" + target);
-			else
-				params.push_back(target);
-		}
+	if (routing.type == ROUTE_TYPE_BROADCAST)
 		Utils->DoOneToMany(user->uuid, command, params);
-	}
+	else
+		Utils->DoOneToOne(user->uuid, command, params, routing.serverdest);
 }
-