Added an option for only allowing authorized users to add things to the info plugin database.

To use authoroization change the line that reads:
    self.require_auth = False
to:
    self.require_auth = True

in the plugins/info.py file.

Signed-off-by: gry <Gry@Gshellz.org>

plugins/info.py

@@ -7,12 +7,12 @@
 #  it under the terms of the GNU General Public License as published by
 #  the Free Software Foundation; either version 2 of the License, or
 #  (at your option) any later version.
-#  
+#
 #  This program is distributed in the hope that it will be useful,
 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 #  GNU General Public License for more details.
-#  
+#
 #  You should have received a copy of the GNU General Public License
 #  along with this program; if not, write to the Free Software
 #  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
@@ -25,129 +25,150 @@ import configparser
 class Info(object):
     """Infobot. Use 'learn X as Y' and 'what is X'"""
     def __init__(self, server):
+        self.require_auth = False
         self.server = server
+        self.server.pluginManager.loadPlugin("Auth")
         self.commands = [ "what", "who", "learn", "forget", "fulldb" ]
         self.server.handle("command", self.handle_command, self.commands)
         self.dbfile = self.server.config["confdir"] + "info.cfg"
         self.infodb = {}
         self.badchars = re.compile("[?]")
-        
+
         self.configParser = configparser.RawConfigParser()
         if os.path.isfile(self.dbfile):
             self.configParser.read(self.dbfile)
             for k,v in self.configParser.items(self.server.config["network"]):
                 self.infodb[k] = v
-    
+
     def destroy(self):
         self.configParser = configparser.RawConfigParser()
-        
+
         if os.path.isfile(self.dbfile):
             self.configParser.read(self.dbfile)
-            
+
         fh = open(self.dbfile, "w")
         network = self.server.config["network"]
         if not self.configParser.has_section(network):
             self.configParser.add_section(network)
-        
+
         for k,v in list(self.infodb.items()):
             self.configParser.set(network,k,v)
-        
+
         self.configParser.write(fh)
         fh.close()
-    
+
     def getInfo(self, key):
         for k,v in list(self.infodb.items()):
             if k.lower() == key.lower():
-                return k,v 
-        
-        return [None, None]
-        
-    def handle_command(self, channel, user, cmd, args, info_silent = False):
-        if cmd == "what" or cmd == "who":
-            for i in range(0, len(args)):
-                args[i] = self.badchars.sub("", args[i])
-            
-            ind = 0
-            for i in range(0, len(args)):
-                if args[i] == "is" or args[i] == "are" or args[i] == "a" or args[i] == "an" or args[i] == "":
-                    ind += 1
-                else:
-                    break
-            
-            nargs = []
-            for i in range(ind, len(args)):
-                if args[i] != "":
-                    nargs.append(args[i])
-                
-            args = nargs
-            
-            if len(args) < 1:
-                if not info_silent:
-                    self.server.doMessage(channel, "...what should I tell you about?")
-                return
-            
-            tmp = " ".join(args)
-            k,v = self.getInfo(tmp)
+                return k,v
 
-            if k is not None and v is not None:
-                self.server.doMessage(channel, user + ": " + k + ": " + v)
-            else:
-                if not info_silent:
-                    self.server.doMessage(channel, user + ", I don't know what " + tmp + " is.")
+        return [None, None]
 
-        elif cmd == "learn":
+    def handle_command(self, channel, user, cmd, args, info_silent = False):
+        def learn():
             if len(args) < 1:
                 if not info_silent:
                     self.server.doMessage(channel, "...what should I learn?")
                 return
-                
+
             for i in range(0, len(args)):
                 args[i] = self.badchars.sub("", args[i])
-            
+
             tmp = " ".join(args)
             ind = tmp.find(" as ")
             size = 4
-            if ind == -1: 
+            if ind == -1:
                 ind = tmp.find(" as: ")
                 size = 5
                 if ind == -1:
                     if not info_silent:
                         self.server.doMessage(channel, "...what should I learn " + tmp + " as?")
                     return
-                    
+
             val = tmp[ind+size:]
             key = tmp[0:ind]
             key = key.replace("\t", "")
-            k,v = self.getInfo(key) 
+            k,v = self.getInfo(key)
             if k is not None and v is not None:
                 if not info_silent:
                     self.server.doMessage(channel, "I already know what " + k + " is.")
                 return
-                
+
             self.infodb[key] = val
             if not info_silent:
                 self.server.doMessage(channel, "Ok, I learned what " + key + " is.")
-        elif cmd == "forget":
+
+        def forget():
             if len(args) < 1:
                 if not info_silent:
                     self.server.doMessage(channel, "...what should I forget?")
                 return
-                
+
             for i in range(0, len(args)):
                 args[i] = self.badchars.sub("", args[i])
-            
+
             key = ""
             tmp = " ".join(args).strip(" ")
             for k,v in list(self.infodb.items()):
                 if k.lower() == tmp.lower():
                     del self.infodb[k]
-                    key = k
-                    
+                key = k
             if key != "":
                 if not info_silent:
                     self.server.doMessage(channel, "Hmm, I can't seem to remember what "+key+" is anymore.")
             else:
                 if not info_silent:
                     self.server.doMessage(channel, "I don't know what "+tmp+" is.")
+
+        if cmd == "what" or cmd == "who":
+            for i in range(0, len(args)):
+                args[i] = self.badchars.sub("", args[i])
+
+            ind = 0
+            for i in range(0, len(args)):
+                if args[i] == "is" or args[i] == "are" or args[i] == "a" or args[i] == "an" or args[i] == "":
+                    ind += 1
+                else:
+                    break
+
+            nargs = []
+            for i in range(ind, len(args)):
+                if args[i] != "":
+                    nargs.append(args[i])
+
+            args = nargs
+
+            if len(args) < 1:
+                if not info_silent:
+                    self.server.doMessage(channel, "...what should I tell you about?")
+                return
+
+            tmp = " ".join(args)
+            k,v = self.getInfo(tmp)
+
+            if k is not None and v is not None:
+                self.server.doMessage(channel, user + ": " + k + ": " + v)
+            else:
+                if not info_silent:
+                    self.server.doMessage(channel, user + ", I don't know what " + tmp + " is.")
+
+        elif cmd == "learn":
+            if not self.require_auth:
+                learn()
+            else:
+                if self.server.getPlugin("auth").isMod(user):
+                    learn()
+                else:
+                    self.server.doMessage(channel, user+": You are not authorized to use this function. ")
+
+        elif cmd == "forget":
+            if not self.require_auth:
+                forget()
+            else:
+                if self.server.getPlugin("auth").isMod(user):
+                        forget()
+                else:
+                    self.server.doMessage(channel, user+": You are not authorized to use this function. ")
+
         elif cmd == "fulldb":
             self.server.doMessage(channel, user+": My database contains (separated by '|'): "+" | ".join(list(self.infodb.keys())))