As we have an enum for type, why not ..use it?

[user/henk/code/inspircd.git] / src / modules / m_spanningtree / protocolinterface.cpp

diff --git a/src/modules/m_spanningtree/protocolinterface.cpp b/src/modules/m_spanningtree/protocolinterface.cpp
index bbdca23054da334f8376baaa06a4971197827000..9f32963a02c55adb1e645b8e9b32e675a882eff9 100644 (file)
--- a/src/modules/m_spanningtree/protocolinterface.cpp
+++ b/src/modules/m_spanningtree/protocolinterface.cpp
@@ -1,14 +1,37 @@
 #include "inspircd.h"
 #include "m_spanningtree/main.h"
 #include "m_spanningtree/utils.h"
+#include "m_spanningtree/treeserver.h"
+#include "m_spanningtree/treesocket.h"
 #include "m_spanningtree/protocolinterface.h"
 
+/*
+ * For documentation on this class, see include/protocol.h.
+ */
+
+void SpanningTreeProtocolInterface::GetServerList(ProtoServerList &sl)
+{
+       sl.clear();
+       for (server_hash::iterator i = Utils->serverlist.begin(); i != Utils->serverlist.end(); i++)
+       {
+               ProtoServer ps;
+               ps.servername = i->second->GetName();
+               TreeServer* s = i->second->GetParent();
+               ps.parentname = s ? s->GetName() : "";
+               ps.usercount = i->second->GetUserCount();
+               ps.opercount = i->second->GetOperCount();
+               ps.gecos = i->second->GetDesc();
+               ps.latencyms = i->second->rtt;
+               sl.push_back(ps);
+       }
+}
+
 void SpanningTreeProtocolInterface::SendEncapsulatedData(parameterlist &encap)
 {
        Utils->DoOneToMany(ServerInstance->Config->GetSID(), "ENCAP", encap);
 }
 
-void SpanningTreeProtocolInterface::SendMetaData(void* target, int type, const std::string &key, const std::string &data)
+void SpanningTreeProtocolInterface::SendMetaData(void* target, TargetTypeFlags type, const std::string &key, const std::string &data)
 {
        parameterlist params;
 
@@ -16,13 +39,16 @@ void SpanningTreeProtocolInterface::SendMetaData(void* target, int type, const s
        {
                case TYPE_USER:
                        params.push_back(((User*)target)->uuid);
-               break;
+                       break;
                case TYPE_CHANNEL:
                        params.push_back(((Channel*)target)->name);
-               break;
+                       break;
                case TYPE_SERVER:
-                       params.push_back(ServerInstance->Config->GetSID());
-               break;
+                       params.push_back("*");
+                       break;
+               default:
+                       throw CoreException("I don't know how to handle TYPE_OTHER.");
+                       break;
        }
        params.push_back(key);
        params.push_back(":" + data);
@@ -47,9 +73,14 @@ void SpanningTreeProtocolInterface::SendMode(const std::string &target, paramete
        if (modedata.empty())
                return;
 
+       std::string outdata;
+
        /* Warning: in-place translation is only safe for type TR_NICK */
        for (size_t n = 0; n < modedata.size(); n++)
-               ServerInstance->Parser->TranslateUIDs(TR_NICK, modedata[n], modedata[n]);
+       {
+               ServerInstance->Parser->TranslateUIDs(TR_NICK, modedata[n], outdata);
+               modedata[n] = outdata;
+       }
 
        std::string uidtarget;
        ServerInstance->Parser->TranslateUIDs(TR_NICK, target, uidtarget);
@@ -72,13 +103,6 @@ void SpanningTreeProtocolInterface::SendMode(const std::string &target, paramete
        }
 }
 
-void SpanningTreeProtocolInterface::SendOperNotice(const std::string &text)
-{
-       parameterlist p;
-       p.push_back(":" + text);
-       Utils->DoOneToMany(ServerInstance->Config->GetSID(), "OPERNOTICE", p);
-}
-
 void SpanningTreeProtocolInterface::SendModeNotice(const std::string &modes, const std::string &text)
 {
        parameterlist p;
@@ -103,3 +127,80 @@ void SpanningTreeProtocolInterface::PushToClient(User* target, const std::string
        Utils->DoOneToOne(ServerInstance->Config->GetSID(), "PUSH", p, target->server);
 }
 
+void SpanningTreeProtocolInterface::SendChannel(Channel* target, char status, const std::string &text)
+{
+       std::string cname = target->name;
+       if (status)
+               cname = status + cname;
+       TreeServerList list;
+       CUList exempt_list;
+       Utils->GetListOfServersForChannel(target,list,status,exempt_list);
+       for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
+       {
+               TreeSocket* Sock = i->second->GetSocket();
+               if (Sock)
+                       Sock->WriteLine(text);
+       }
+}
+
+
+void SpanningTreeProtocolInterface::SendChannelPrivmsg(Channel* target, char status, const std::string &text)
+{
+       SendChannel(target, status, ":" + ServerInstance->Config->GetSID()+" PRIVMSG "+target->name+" :"+text);
+}
+
+void SpanningTreeProtocolInterface::SendChannelNotice(Channel* target, char status, const std::string &text)
+{
+       SendChannel(target, status, ":" + ServerInstance->Config->GetSID()+" NOTICE "+target->name+" :"+text);
+}
+
+void SpanningTreeProtocolInterface::SendUserPrivmsg(User* target, const std::string &text)
+{
+       TreeServer* serv = Utils->FindServer(target->server);
+       if (serv)
+       {
+               TreeSocket* sock = serv->GetSocket();
+               if (sock)
+               {
+                       sock->WriteLine(":" + ServerInstance->Config->GetSID() + " PRIVMSG " + target->nick + " :"+text);
+               }
+       }
+}
+
+void SpanningTreeProtocolInterface::SendUserNotice(User* target, const std::string &text)
+{
+       TreeServer* serv = Utils->FindServer(target->server);
+       if (serv)
+       {
+               TreeSocket* sock = serv->GetSocket();
+               if (sock)
+               {
+                       sock->WriteLine(":" + ServerInstance->Config->GetSID() + " NOTICE " + target->nick + " :"+text);
+               }
+       }
+}
+
+void SpanningTreeProtocolInterface::Introduce(User* user)
+{
+       if (IS_LOCAL(user))
+       {
+               std::deque<std::string> params;
+               params.push_back(user->uuid);
+               params.push_back(ConvToStr(user->age));
+               params.push_back(user->nick);
+               params.push_back(user->host);
+               params.push_back(user->dhost);
+               params.push_back(user->ident);
+               params.push_back(user->GetIPString());
+               params.push_back(ConvToStr(user->signon));
+               params.push_back("+"+std::string(user->FormatModes(true)));
+               params.push_back(":"+std::string(user->fullname));
+               Utils->DoOneToMany(ServerInstance->Config->GetSID(), "UID", params);
+       }
+
+       TreeServer* SourceServer = Utils->FindServer(user->server);
+       if (SourceServer)
+       {
+               SourceServer->SetUserCount(1); // increment by 1
+       }
+}