Added helperfuncs.h

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

diff --git a/src/modules.cpp b/src/modules.cpp
index 1e92e2bda99488d33dfd42cd1505704229bbca33..7214a9673c8351d1c2125679582a4ba066cfaad0 100644 (file)
--- a/src/modules.cpp
+++ b/src/modules.cpp
@@ -43,6 +43,7 @@
 #include "typedefs.h"
 #include "modules.h"
 #include "command_parse.h"
+#include "dns.h"
 
 extern ServerConfig *Config;
 extern InspIRCd* ServerInstance;
@@ -606,6 +607,11 @@ bool Server::DelModeWatcher(ModeWatcher* mw)
        return ServerInstance->ModeGrok->DelModeWatcher(mw);
 }
 
+bool Server::AddResolver(Resolver* r)
+{
+       return dns_add_class(r);
+}
+
 int Server::CountUsers(chanrec* c)
 {
        return usercount(c);
@@ -931,17 +937,43 @@ FileReader::FileReader(const std::string &filename)
        file_cache c;
        readfile(c,filename.c_str());
        this->fc = c;
+       this->CalcSize();
 }
 
 FileReader::FileReader()
 {
 }
 
+std::string FileReader::Contents()
+{
+       std::string x = "";
+       for (file_cache::iterator a = this->fc.begin(); a != this->fc.end(); a++)
+       {
+               x.append(*a);
+               x.append("\r\n");
+       }
+       return x;
+}
+
+unsigned long FileReader::ContentSize()
+{
+       return this->contentsize;
+}
+
+void FileReader::CalcSize()
+{
+       unsigned long n = 0;
+       for (file_cache::iterator a = this->fc.begin(); a != this->fc.end(); a++)
+               n += (a->length() + 2);
+       this->contentsize = n;
+}
+
 void FileReader::LoadFile(const std::string &filename)
 {
        file_cache c;
        readfile(c,filename.c_str());
        this->fc = c;
+       this->CalcSize();
 }