]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/cmd_who.cpp
/who tidyup - needs QA'ing
[user/henk/code/inspircd.git] / src / cmd_who.cpp
index 0add5a09f3879321db35b58d47406387c831517d..c4c37e52876d260c0ac32c3af203fbdf8b85695e 100644 (file)
@@ -1,15 +1,12 @@
-/*   +------------------------------------+
- *   | Inspire Internet Relay Chat Daemon |
- *   +------------------------------------+
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *   E-mail:
- *<brain@chatspike.net>
- *<Craig@chatspike.net>
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
  *
- * Written by Craig Edwards, Craig McLure, and others.
  * This program is free but copyrighted software; see
- *the file COPYING for details.
+ *         the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 /* get the last 'visible' chan of a user */
 static char *getlastchanname(userrec *u)
 {
-       for (std::vector<ucrec*>::const_iterator v = u->chans.begin(); v != u->chans.end(); v++)
+       UCListIter i = u->chans.begin();
+       if (i != u->chans.end())
        {
-               ucrec* temp = (ucrec*)*v;
-
-               if (temp->channel)
-               {
-                       if (!temp->channel->IsModeSet('s'))
-                               return temp->channel->name;
-               }
+               if (!i->first->IsModeSet('s'))
+                       return i->first->name;
        }
 
        return "*";
 }
 
-bool whomatch(userrec* user, const char* matchtext, bool opt_realname, bool opt_showrealhost)
+bool whomatch(userrec* user, const char* matchtext, bool opt_realname, bool opt_showrealhost, bool opt_mode)
 {
        bool realhost = false;
        bool realname = false;
+       bool positive = true;
 
        if (user->registered != REG_ALL)
                return false;
 
+       if (opt_mode)
+       {
+               for (const char* n = matchtext; *n; n++)
+               {
+                       if (*n == '+')
+                       {
+                               positive = true;
+                               continue;
+                       }
+                       else if (*n == '-')
+                       {
+                               positive = false;
+                               continue;
+                       }
+                       if (user->IsModeSet(*n) != positive)
+                               return false;
+               }
+               return true;
+       }
+
        if (opt_realname)
                realname = match(user->fullname, matchtext);
 
@@ -61,6 +75,51 @@ extern "C" command_t* init_command(InspIRCd* Instance)
        return new cmd_who(Instance);
 }
 
+bool cmd_who::CanView(chanrec* chan, userrec* user)
+{
+       if (!user || !chan)
+               return false;
+
+       /* Execute items in fastest-to-execute first order */
+       if (*user->oper)
+               return true;
+       else if (!chan->IsModeSet('s') && !chan->IsModeSet('p'))
+               return true;
+       else if (chan->HasUser(user))
+               return true;
+
+       return false;
+}
+
+void cmd_who::SendWhoLine(userrec* user, const std::string &initial, chanrec* ch, userrec* u, std::vector<std::string> &whoresults)
+{
+       std::string lcn = getlastchanname(u);
+       chanrec* chlast = ServerInstance->FindChan(lcn);
+
+       std::string wholine =   initial + (ch ? ch->name : lcn) + " " + u->ident + " " + (opt_showrealhost ? u->host : u->dhost) + " " +
+                               ((*ServerInstance->Config->HideWhoisServer && !*user->oper) ? ServerInstance->Config->HideWhoisServer : u->server) +
+                               " " + u->nick + " ";
+
+       /* away? */
+       if (u->awaymsg)
+       {
+               wholine.append("G");
+       }
+       else
+       {
+               wholine.append("H");
+       }
+
+       /* oper? */
+       if (u->oper)
+       {
+               wholine.append("*");
+       }
+
+       wholine = wholine + (ch ? ch->GetPrefixChar(u) : (chlast ? chlast->GetPrefixChar(u) : "")) + " :0 " + u->fullname;
+       whoresults.push_back(wholine);
+}
+
 CmdResult cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
 {
        /*
@@ -71,10 +130,11 @@ CmdResult cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
         */
 
        /* WHO options */
-       bool opt_viewopersonly = false;
-       bool opt_showrealhost = false;
-       bool opt_unlimit = false;
-       bool opt_realname = false;
+       opt_viewopersonly = false;
+       opt_showrealhost = false;
+       opt_unlimit = false;
+       opt_realname = false;
+       opt_mode = false;
 
        chanrec *ch = NULL;
        std::vector<std::string> whoresults;
@@ -110,6 +170,9 @@ CmdResult cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
                                case 'r':
                                        opt_realname = true;
                                break;
+                               case 'm':
+                                       opt_mode = true;
+                               break;
                        }
 
                        *iter++;
@@ -120,7 +183,7 @@ CmdResult cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
        /* who on a channel? */
        ch = ServerInstance->FindChan(matchtext);
 
-       if (ch)
+       if ((ch) && (CanView(ch,user)))
        {
                /* who on a channel. */
                CUList *cu = ch->GetUsers();
@@ -131,30 +194,7 @@ CmdResult cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
                        if (opt_viewopersonly && !*(i->second)->oper)
                                continue;
 
-                       /* XXX - code duplication; this could be more efficient -- w00t */
-                       std::string wholine = initial;
-
-                       wholine = wholine + ch->name + " " + i->second->ident + " " + (opt_showrealhost ? i->second->host : i->second->dhost) + " " + 
-                                       i->second->server + " " + i->second->nick + " ";
-
-                       /* away? */
-                       if (*(i->second)->awaymsg)
-                       {
-                               wholine.append("G");
-                       }
-                       else
-                       {
-                               wholine.append("H");
-                       }
-
-                       /* oper? */
-                       if (*(i->second)->oper)
-                       {
-                               wholine.append("*");
-                       }
-
-                       wholine = wholine + ch->GetPrefixChar(i->second) + " :0 " + i->second->fullname;
-                       whoresults.push_back(wholine);
+                       SendWhoLine(user, initial, ch, i->second, whoresults);
                }
        }
        else
@@ -168,63 +208,19 @@ CmdResult cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
                        {
                                userrec* oper = *i;
 
-                               if (whomatch(oper, matchtext, opt_realname, opt_showrealhost))
+                               if (whomatch(oper, matchtext, opt_realname, opt_showrealhost, opt_mode))
                                {
-                                       std::string wholine = initial;
-       
-                                       wholine = wholine + getlastchanname(oper) + " " + oper->ident + " " + (opt_showrealhost ? oper->host : oper->dhost) + " " + 
-                                                       oper->server + " " + oper->nick + " ";
-
-                                       /* away? */
-                                       if (*oper->awaymsg)
-                                       {
-                                               wholine.append("G");
-                                       }
-                                       else
-                                       {
-                                               wholine.append("H");
-                                       }
-       
-                                       /* oper? */
-                                       if (*oper->oper)
-                                       {
-                                               wholine.append("*");
-                                       }
-       
-                                       wholine = wholine + ch->GetPrefixChar(oper) + " :0 " + oper->fullname;
-                                       whoresults.push_back(wholine);
+                                       SendWhoLine(user, initial, NULL, oper, whoresults);
                                }
                        }
                }
                else
                {
-                       for (user_hash::iterator i = ServerInstance->clientlist.begin(); i != ServerInstance->clientlist.end(); i++)
+                       for (user_hash::iterator i = ServerInstance->clientlist->begin(); i != ServerInstance->clientlist->end(); i++)
                        {
-                               if (whomatch(i->second, matchtext, opt_realname, opt_showrealhost))
+                               if (whomatch(i->second, matchtext, opt_realname, opt_showrealhost, opt_mode))
                                {
-                                       std::string wholine = initial;
-       
-                                       wholine = wholine + getlastchanname(i->second) + " " + i->second->ident + " " + (opt_showrealhost ? i->second->host : i->second->dhost) + " " + 
-                                               i->second->server + " " + i->second->nick + " ";
-       
-                                       /* away? */
-                                       if (*(i->second)->awaymsg)
-                                       {
-                                               wholine.append("G");
-                                       }
-                                       else
-                                       {
-                                               wholine.append("H");
-                                       }
-
-                                       /* oper? */
-                                       if (*(i->second)->oper)
-                                       {
-                                               wholine.append("*");
-                                       }
-
-                                       wholine = wholine + ch->GetPrefixChar(i->second) + " :0 " + i->second->fullname;
-                                       whoresults.push_back(wholine);
+                                       SendWhoLine(user, initial, NULL, i->second, whoresults);
                                }
                        }
                }