I'll give you ##TOAST, :p

[user/henk/code/inspircd.git] / src / cmd_who.cpp

diff --git a/src/cmd_who.cpp b/src/cmd_who.cpp
index 6f52ebb6343d66ceca81597f09d35b3d791e5be0..367bdb326715335ccd4570d4ae165381c39badab 100644 (file)
--- a/src/cmd_who.cpp
+++ b/src/cmd_who.cpp
@@ -17,13 +17,9 @@
 #include "configreader.h"
 #include "users.h"
 #include "modules.h"
-#include "commands.h"
-
 #include "wildcard.h"
 #include "commands/cmd_who.h"
 
-
-
 /* get the last 'visible' chan of a user */
 static char *getlastchanname(userrec *u)
 {
@@ -58,7 +54,14 @@ bool whomatch(userrec* user, const char* matchtext, bool opt_realname, bool opt_
        return ((realname) || (realhost) || (match(user->dhost, matchtext)) || (match(user->nick, matchtext)) || (match(user->server, matchtext)));
 }
 
-void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
+
+
+extern "C" command_t* init_command(InspIRCd* Instance)
+{
+       return new cmd_who(Instance);
+}
+
+CmdResult cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
 {
        /*
         * XXX - RFC says:
@@ -172,6 +175,8 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
                                        wholine = wholine + getlastchanname(oper) + " " + oper->ident + " " + (opt_showrealhost ? oper->host : oper->dhost) + " " + 
                                                        oper->server + " " + oper->nick + " ";
 
+                                       ch = ServerInstance->FindChan(getlastchanname(oper));
+
                                        /* away? */
                                        if (*oper->awaymsg)
                                        {
@@ -188,7 +193,7 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
                                                wholine.append("*");
                                        }
        
-                                       wholine = wholine + ch->GetPrefixChar(oper) + " :0 " + oper->fullname;
+                                       wholine = wholine + (ch ? ch->GetPrefixChar(oper) : "") + " :0 " + oper->fullname;
                                        whoresults.push_back(wholine);
                                }
                        }
@@ -204,6 +209,8 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
                                        wholine = wholine + getlastchanname(i->second) + " " + i->second->ident + " " + (opt_showrealhost ? i->second->host : i->second->dhost) + " " + 
                                                i->second->server + " " + i->second->nick + " ";
        
+                                       ch = ServerInstance->FindChan(getlastchanname(i->second));
+
                                        /* away? */
                                        if (*(i->second)->awaymsg)
                                        {
@@ -220,7 +227,7 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
                                                wholine.append("*");
                                        }
 
-                                       wholine = wholine + ch->GetPrefixChar(i->second) + " :0 " + i->second->fullname;
+                                       wholine = wholine + (ch ? ch->GetPrefixChar(i->second) : "") + " :0 " + i->second->fullname;
                                        whoresults.push_back(wholine);
                                }
                        }
@@ -232,10 +239,12 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
                for (std::vector<std::string>::const_iterator n = whoresults.begin(); n != whoresults.end(); n++)
                        user->WriteServ(*n);
                user->WriteServ("315 %s %s :End of /WHO list.",user->nick, *parameters[0] ? parameters[0] : "*");
+               return CMD_SUCCESS;
        }
        else
        {
                /* BZZT! Too many results. */
                user->WriteServ("315 %s %s :Too many results",user->nick, parameters[0]);
+               return CMD_FAILURE;
        }
 }