]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_operprefix.cpp
Minor spelling errors in m_spanningtree.so
[user/henk/code/inspircd.git] / src / modules / m_operprefix.cpp
index b0cb0e4e9d8876b4183528eb5bbf316adec3b61a..7d5e6d1181151f0cfae5c7b243205fed2d910248 100644 (file)
@@ -47,26 +47,70 @@ class OperPrefixMode : public ModeHandler
 
                ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
                {
-                       if (IS_SERVER(source) || (source && ServerInstance->ULine(source->server)))
+                       if (IS_SERVER(source) || ServerInstance->ULine(source->server))
                                return MODEACTION_ALLOW;
                        else
                        {
-                               if (source && channel)
+                               if (channel)
                                        source->WriteNumeric(ERR_CHANOPRIVSNEEDED, "%s %s :Only servers are permitted to change channel mode '%c'", source->nick.c_str(), channel->name.c_str(), 'y');
                                return MODEACTION_DENY;
                        }
                }
 
                bool NeedsOper() { return true; }
+
+               void RemoveMode(Channel* chan, irc::modestacker* stack)
+               {
+                       irc::modestacker modestack(false);
+                       const UserMembList* users = chan->GetUsers();
+                       for (UserMembCIter i = users->begin(); i != users->end(); ++i)
+                       {
+                               if (i->second->hasMode(mode))
+                               {
+                                       if (stack)
+                                               stack->Push(this->GetModeChar(), i->first->nick);
+                                       else
+                                               modestack.Push(this->GetModeChar(), i->first->nick);
+                               }
+                       }
+
+                       if (stack)
+                               return;
+
+                       std::deque<std::string> stackresult;
+                       std::vector<std::string> mode_junk;
+                       mode_junk.push_back(chan->name);
+                       while (modestack.GetStackedLine(stackresult))
+                       {
+                               mode_junk.insert(mode_junk.end(), stackresult.begin(), stackresult.end());
+                               ServerInstance->SendMode(mode_junk, ServerInstance->FakeClient);
+                               mode_junk.erase(mode_junk.begin() + 1, mode_junk.end());
+                       }
+               }
+
+               void RemoveMode(User* user, irc::modestacker* stack)
+               {
+               }
+};
+
+class ModuleOperPrefixMode;
+class HideOperWatcher : public ModeWatcher
+{
+       ModuleOperPrefixMode* parentmod;
+ public:
+       HideOperWatcher(ModuleOperPrefixMode* parent) : ModeWatcher((Module*) parent, 'H', MODETYPE_USER), parentmod(parent) {}
+       void AfterMode(User* source, User* dest, Channel* channel, const std::string &parameter, bool adding, ModeType type);
 };
 
 class ModuleOperPrefixMode : public Module
 {
  private:
        OperPrefixMode opm;
+       bool mw_added;
+       HideOperWatcher hideoperwatcher;
  public:
        ModuleOperPrefixMode()
-               : opm(this)
+               : opm(this), mw_added(false), hideoperwatcher(this)
        {
        }
 
@@ -74,54 +118,102 @@ class ModuleOperPrefixMode : public Module
        {
                ServerInstance->Modules->AddService(opm);
 
-               Implementation eventlist[] = { I_OnPostJoin, I_OnOper };
-               ServerInstance->Modules->Attach(eventlist, this, 2);
+               Implementation eventlist[] = { I_OnUserPreJoin, I_OnPostOper, I_OnLoadModule, I_OnUnloadModule, I_OnPostJoin };
+               ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
+
+               /* To give clients a chance to learn about the new prefix we don't give +y to opers
+                * right now. That means if the module was loaded after opers have joined channels
+                * they need to rejoin them in order to get the oper prefix.
+                */
+
+               if (ServerInstance->Modules->Find("m_hideoper.so"))
+                       mw_added = ServerInstance->Modes->AddModeWatcher(&hideoperwatcher);
+       }
+
+       ModResult OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string& privs, const std::string& keygiven)
+       {
+               /* The user may have the +H umode on himself, but +H does not necessarily correspond
+                * to the +H of m_hideoper.
+                * However we only add the modewatcher when m_hideoper is loaded, so these
+                * conditions (mw_added and the user being +H) together mean the user is a hidden oper.
+                */
+
+               if (IS_OPER(user) && (!mw_added || !user->IsModeSet('H')))
+                       privs.push_back('y');
+               return MOD_RES_PASSTHRU;
+       }
+
+       void OnPostJoin(Membership* memb)
+       {
+               if ((!IS_LOCAL(memb->user)) || (!IS_OPER(memb->user)) || (((mw_added) && (memb->user->IsModeSet('H')))))
+                       return;
+
+               if (memb->hasMode(opm.GetModeChar()))
+                       return;
+
+               // The user was force joined and OnUserPreJoin() did not run. Set the operprefix now.
+               std::vector<std::string> modechange;
+               modechange.push_back(memb->chan->name);
+               modechange.push_back("+y");
+               modechange.push_back(memb->user->nick);
+               ServerInstance->SendGlobalMode(modechange, ServerInstance->FakeClient);
        }
 
-       void PushChanMode(Channel* channel, User* user)
+       void SetOperPrefix(User* user, bool add)
        {
-               char modeline[] = "+y";
                std::vector<std::string> modechange;
-               modechange.push_back(channel->name);
-               modechange.push_back(modeline);
+               modechange.push_back("");
+               modechange.push_back(add ? "+y" : "-y");
                modechange.push_back(user->nick);
-               ServerInstance->SendMode(modechange,ServerInstance->FakeClient);
+               for (UCListIter v = user->chans.begin(); v != user->chans.end(); v++)
+               {
+                       modechange[0] = (*v)->name;
+                       ServerInstance->SendGlobalMode(modechange, ServerInstance->FakeClient);
+               }
        }
 
-       void OnPostJoin(Membership* memb)
+       void OnPostOper(User* user, const std::string& opername, const std::string& opertype)
        {
-               if (IS_OPER(memb->user) && !memb->user->IsModeSet('H'))
-                       PushChanMode(memb->chan, memb->user);
+               if (IS_LOCAL(user) && (!mw_added || !user->IsModeSet('H')))
+                       SetOperPrefix(user, true);
        }
 
-       // XXX: is there a better way to do this?
-       ModResult OnRawMode(User* user, Channel* chan, const char mode, const std::string &param, bool adding, int pcnt)
+       void OnLoadModule(Module* mod)
        {
-               /* force event propagation to its ModeHandler */
-               if (!IS_SERVER(user) && chan && (mode == 'y'))
-                       return MOD_RES_ALLOW;
-               return MOD_RES_PASSTHRU;
+               if ((!mw_added) && (mod->ModuleSourceFile == "m_hideoper.so"))
+                       mw_added = ServerInstance->Modes->AddModeWatcher(&hideoperwatcher);
        }
 
-       void OnOper(User *user, const std::string&)
+       void OnUnloadModule(Module* mod)
        {
-               if (user && !user->IsModeSet('H'))
-               {
-                       for (UCListIter v = user->chans.begin(); v != user->chans.end(); v++)
-                       {
-                               PushChanMode(*v, user);
-                       }
-               }
+               if ((mw_added) && (mod->ModuleSourceFile == "m_hideoper.so") && (ServerInstance->Modes->DelModeWatcher(&hideoperwatcher)))
+                       mw_added = false;
        }
 
        ~ModuleOperPrefixMode()
        {
+               if (mw_added)
+                       ServerInstance->Modes->DelModeWatcher(&hideoperwatcher);
        }
 
        Version GetVersion()
        {
                return Version("Gives opers cmode +y which provides a staff prefix.", VF_VENDOR);
        }
+
+       void Prioritize()
+       {
+               // m_opermodes may set +H on the oper to hide him, we don't want to set the oper prefix in that case
+               Module* opermodes = ServerInstance->Modules->Find("m_opermodes.so");
+               ServerInstance->Modules->SetPriority(this, I_OnPostOper, PRIORITY_AFTER, opermodes);
+       }
 };
 
+void HideOperWatcher::AfterMode(User* source, User* dest, Channel* channel, const std::string& parameter, bool adding, ModeType type)
+{
+       // If hideoper is being unset because the user is deopering, don't set +y
+       if (IS_LOCAL(dest) && IS_OPER(dest))
+               parentmod->SetOperPrefix(dest, !adding);
+}
+
 MODULE_INIT(ModuleOperPrefixMode)