src/modules/m_chanhistory.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   5  *
   6  * This file is part of InspIRCd.  InspIRCd is free software: you can
   7  * redistribute it and/or modify it under the terms of the GNU General Public
   8  * License as published by the Free Software Foundation, version 2.
   9  *
  10  * This program is distributed in the hope that it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  12  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  13  * details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19
  20 #include "inspircd.h"
  21
  22 /* $ModDesc: Provides channel history for a given number of lines */
  23
  24 struct HistoryItem
  25 {
  26         time_t ts;
  27         std::string line;
  28         HistoryItem(const std::string& Line) : ts(ServerInstance->Time()), line(Line) {}
  29 };
  30
  31 struct HistoryList
  32 {
  33         std::deque<HistoryItem> lines;
  34         unsigned int maxlen, maxtime;
  35         HistoryList(unsigned int len, unsigned int time) : maxlen(len), maxtime(time) {}
  36 };
  37
  38 class HistoryMode : public ModeHandler
  39 {
  40         bool IsValidDuration(const std::string& duration)
  41         {
  42                 for (std::string::const_iterator i = duration.begin(); i != duration.end(); ++i)
  43                 {
  44                         unsigned char c = *i;
  45                         if (((c >= '0') && (c <= '9')) || (c == 's') || (c == 'S'))
  46                                 continue;
  47
  48                         if (duration_multi[c] == 1)
  49                                 return false;
  50                 }
  51                 return true;
  52         }
  53
  54  public:
  55         SimpleExtItem<HistoryList> ext;
  56         unsigned int maxlines;
  57         HistoryMode(Module* Creator) : ModeHandler(Creator, "history", 'H', PARAM_SETONLY, MODETYPE_CHANNEL),
  58                 ext("history", Creator) { }
  59
  60         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
  61         {
  62                 if (adding)
  63                 {
  64                         std::string::size_type colon = parameter.find(':');
  65                         if (colon == std::string::npos)
  66                                 return MODEACTION_DENY;
  67
  68                         std::string duration = parameter.substr(colon+1);
  69                         if ((IS_LOCAL(source)) && ((duration.length() > 10) || (!IsValidDuration(duration))))
  70                                 return MODEACTION_DENY;
  71
  72                         unsigned int len = ConvToInt(parameter.substr(0, colon));
  73                         int time = ServerInstance->Duration(duration);
  74                         if (len == 0 || time < 0)
  75                                 return MODEACTION_DENY;
  76                         if (len > maxlines && IS_LOCAL(source))
  77                                 return MODEACTION_DENY;
  78                         if (len > maxlines)
  79                                 len = maxlines;
  80                         if (parameter == channel->GetModeParameter(this))
  81                                 return MODEACTION_DENY;
  82
  83                         HistoryList* history = ext.get(channel);
  84                         if (history)
  85                         {
  86                                 // Shrink the list if the new line number limit is lower than the old one
  87                                 if (len < history->lines.size())
  88                                         history->lines.erase(history->lines.begin(), history->lines.begin() + (history->lines.size() - len));
  89
  90                                 history->maxlen = len;
  91                                 history->maxtime = time;
  92                         }
  93                         else
  94                         {
  95                                 ext.set(channel, new HistoryList(len, time));
  96                         }
  97                         channel->SetModeParam('H', parameter);
  98                 }
  99                 else
 100                 {
 101                         if (!channel->IsModeSet('H'))
 102                                 return MODEACTION_DENY;
 103                         ext.unset(channel);
 104                         channel->SetModeParam('H', "");
 105                 }
 106                 return MODEACTION_ALLOW;
 107         }
 108 };
 109
 110 class ModuleChanHistory : public Module
 111 {
 112         HistoryMode m;
 113         bool sendnotice;
 114         bool dobots;
 115  public:
 116         ModuleChanHistory() : m(this)
 117         {
 118         }
 119
 120         void init()
 121         {
 122                 ServerInstance->Modules->AddService(m);
 123                 ServerInstance->Modules->AddService(m.ext);
 124
 125                 Implementation eventlist[] = { I_OnPostJoin, I_OnUserMessage, I_OnRehash };
 126                 ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
 127                 OnRehash(NULL);
 128         }
 129
 130         void OnRehash(User*)
 131         {
 132                 ConfigTag* tag = ServerInstance->Config->ConfValue("chanhistory");
 133                 m.maxlines = tag->getInt("maxlines", 50);
 134                 sendnotice = tag->getBool("notice", true);
 135                 dobots = tag->getBool("bots", true);
 136         }
 137
 138         void OnUserMessage(User* user,void* dest,int target_type, const std::string &text, char status, const CUList&)
 139         {
 140                 if (target_type == TYPE_CHANNEL && status == 0)
 141                 {
 142                         Channel* c = (Channel*)dest;
 143                         HistoryList* list = m.ext.get(c);
 144                         if (list)
 145                         {
 146                                 char buf[MAXBUF];
 147                                 snprintf(buf, MAXBUF, ":%s PRIVMSG %s :%s",
 148                                         user->GetFullHost().c_str(), c->name.c_str(), text.c_str());
 149                                 list->lines.push_back(HistoryItem(buf));
 150                                 if (list->lines.size() > list->maxlen)
 151                                         list->lines.pop_front();
 152                         }
 153                 }
 154         }
 155
 156         void OnPostJoin(Membership* memb)
 157         {
 158                 if (IS_REMOTE(memb->user))
 159                         return;
 160
 161                 if (!dobots && ServerInstance->Modules->Find("m_botmode.so") && memb->user->IsModeSet('B'))
 162                         return;
 163
 164                 HistoryList* list = m.ext.get(memb->chan);
 165                 if (!list)
 166                         return;
 167                 time_t mintime = 0;
 168                 if (list->maxtime)
 169                         mintime = ServerInstance->Time() - list->maxtime;
 170
 171                 if (sendnotice)
 172                 {
 173                         memb->user->WriteServ("NOTICE %s :Replaying up to %d lines of pre-join history spanning up to %d seconds",
 174                                 memb->chan->name.c_str(), list->maxlen, list->maxtime);
 175                 }
 176
 177                 for(std::deque<HistoryItem>::iterator i = list->lines.begin(); i != list->lines.end(); ++i)
 178                 {
 179                         if (i->ts >= mintime)
 180                                 memb->user->Write(i->line);
 181                 }
 182         }
 183
 184         Version GetVersion()
 185         {
 186                 return Version("Provides channel history replayed on join", VF_VENDOR);
 187         }
 188 };
 189
 190 MODULE_INIT(ModuleChanHistory)