Change Membership:hasMode() to accept a PrefixMode

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

diff --git a/src/modules/m_operprefix.cpp b/src/modules/m_operprefix.cpp
index 262c034dba4adb09b7fc1b6b706564357eb02c34..d66f994501389ccad8e29ab6a3440d55f0047223 100644 (file)
--- a/src/modules/m_operprefix.cpp
+++ b/src/modules/m_operprefix.cpp
@@ -72,18 +72,26 @@ class ModuleOperPrefixMode : public Module
                return MOD_RES_PASSTHRU;
        }
 
+       void OnPostJoin(Membership* memb)
+       {
+               if ((!IS_LOCAL(memb->user)) || (!memb->user->IsOper()) || (memb->user->IsModeSet(hideopermode)))
+                       return;
+
+               if (memb->HasMode(&opm))
+                       return;
+
+               // The user was force joined and OnUserPreJoin() did not run. Set the operprefix now.
+               Modes::ChangeList changelist;
+               changelist.push_add(&opm, memb->user->nick);
+               ServerInstance->Modes.Process(ServerInstance->FakeClient, memb->chan, NULL, changelist);
+       }
+
        void SetOperPrefix(User* user, bool add)
        {
-               std::vector<std::string> modechange;
-               modechange.push_back("");
-               modechange.push_back(add ? "+" : "-");
-               modechange[1].push_back(opm.GetModeChar());
-               modechange.push_back(user->nick);
+               Modes::ChangeList changelist;
+               changelist.push(&opm, add, user->nick);
                for (User::ChanList::iterator v = user->chans.begin(); v != user->chans.end(); v++)
-               {
-                       modechange[0] = (*v)->chan->name;
-                       ServerInstance->Modes->Process(modechange, ServerInstance->FakeClient);
-               }
+                       ServerInstance->Modes->Process(ServerInstance->FakeClient, (*v)->chan, NULL, changelist);
        }
 
        void OnPostOper(User* user, const std::string& opername, const std::string& opertype) CXX11_OVERRIDE