X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fbasile.git;a=blobdiff_plain;f=basile.py;h=ad22ab649f74ded43ca500733377546a6cc20634;hp=c549707765830511058a7962a8984c91ad888ffb;hb=HEAD;hpb=176b7d51c6c719694dd92cba28ab7b0775232e05

diff --git a/basile.py b/basile.py
index c549707..ad22ab6 100755
--- a/basile.py
+++ b/basile.py
@@ -17,6 +17,8 @@ import sys
 # Oui, j'ai recodé ma version d'irclib pour pouvoir rattrapper les SIGHUP
 sys.path.insert(0, "/home/vincent/scripts/python-myirclib")
 import irclib
+# On veut réagir sur la partie du whois qui dit qu'un nick est registered
+irclib.numeric_events['307'] = "whoisregnick"
 import ircbot
 
 from commands import getstatusoutput as ex
@@ -211,6 +213,11 @@ class Basile(ircbot.SingleServerIRCBot):
             self.serv.privmsg(place, message)
         log(self.serveur, place, auteur, something + "%r" % params + ("[successful]" if success else "[failed]"))
     
+    def whois(self, pseudo, askedwhy, askedby):
+        """Demande un whois sur ``pseudo``. La réponse sera handled par une autre fonction."""
+        self.users.pending_whois[pseudo] = ["pending", askedwhy, askedby, None]
+        self.serv.whois([pseudo])
+    
     ### Surcharge des events du Bot
     def on_welcome(self, serv, ev):
         """À l'arrivée sur le serveur."""
@@ -229,6 +236,42 @@ class Basile(ircbot.SingleServerIRCBot):
             for report in self.report_bugs_to:
                 serv.privmsg(report, "Connection to NK2015 failed, invalid password ?")
 
+    def on_whoisregnick(self, serv, ev):
+        """Appelée sur une réponse à un whois au moment où ça dit "is a registered nick".
+           J'ai vérifié, "is a registered nick" ça inclu le fiat qu'il est identified correctement.
+           
+           On stocke l'information comme quoi cette personne est registered, et quand c'était."""
+        pseudo, phrase = ev.arguments()
+        if phrase == 'is a registered nick':
+            # Le whois n'est plus "pending"
+            if self.users.pending_whois.has_key(pseudo):
+                self.users.pending_whois[pseudo][0] = "registered"
+                self.users.pending_whois[pseudo][3] = time.time()
+                _, askedwhy, askedby, _ = self.users.pending_whois[pseudo]
+                if askedwhy == "cmd WHOIS":
+                    # Ce whois a été demandé par quelqu'un, on lui répond
+                    self.serv.privmsg(askedby, "%s is a registered nick" % (pseudo,))
+    
+    def on_endofwhois(self, serv, ev):
+        """Si on arrive à la fin du whois, on va voir si on n'a pas reçu "is a registered nick"
+           c'est que le pseudo n'est pas identifié. """
+        pseudo = ev.arguments()[0]
+        # On laisse le temps au bot de souffler un coup
+        serv.execute_delayed(config.whois_timeout, self.fail_whoisregnick, (pseudo,))
+
+    def fail_whoisregnick(self, pseudo):
+        """Maintenant qu'on a laissé quelques secondes au bot pour gérer les affaires courantes,
+           on considère que le pseudo n'est pas registered. """
+        # Attention, parce qu'il se pourrait qu'on n'ait pas sollicité ce whois
+        # et que donc pending_whois n'ai pas été peuplé en conséquence
+        if self.users.pending_whois.has_key(pseudo):
+            status, askedwhy, askedby, _ = self.users.pending_whois[pseudo]
+            if status == "pending":
+                # Si le status est encore pending, on n'a pas eu de réponse positive, donc elle est négative
+                self.users.pending_whois[pseudo] = "notregistered"
+                if askedwhy == "cmd WHOIS":
+                    self.serv.privmsg(askedby, "%s is NOT a registered nick" % (pseudo,))
+    
     def on_privmsg(self, serv, ev):
         """À la réception d'un message en privé."""
         if ignore_event(serv, ev):
@@ -265,6 +308,8 @@ class Basile(ircbot.SingleServerIRCBot):
                         helpmsg += "\n" + helpmsgs[2]
                     else:
                         helpmsg = helpmsgs[2]
+                if not helpmsg: # Un non-op a demandé de l'aide sur une commande dont il n'est pas censé connaître l'existence
+                    helpmsg = "Commande inacessible."
             for ligne in helpmsg.split("\n"):
                 serv.privmsg(auteur, ligne.encode("utf-8"))
         elif cmd == u"identify":
@@ -290,10 +335,11 @@ class Basile(ircbot.SingleServerIRCBot):
         elif cmd == u"drop":
             if len(message) > 1:
                 if self.users.has(auteur):
+                    idbde = self.users[auteur].idbde
                     password = " ".join(message[1:])
-                    success, _, _ = self.new_connection_NK(serv, "#%s" % self.users[auteur].idbde, password)
+                    success, _, _ = self.new_connection_NK(serv, "#%s" % idbde, password)
                     if success:
-                        self.users.drop(auteur)
+                        self.users.drop(idbde)
                         log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
                         serv.privmsg(auteur, "Pseudo oublié.")
                     else:
@@ -458,7 +504,7 @@ class Basile(ircbot.SingleServerIRCBot):
                         serv.privmsg(auteur, "%.2f (%s)" % (solde/100.0, pseudo.encode("utf8")))
                     log(self.serveur, "priv", auteur, " ".join(message) + ("[successful]" if success else "[failed]"))
                 else:
-                    serv.privmsg(canal, "Je ne connais pas votre note.")
+                    serv.privmsg(auteur, "Je ne connais pas votre note.")
         elif cmd == u"ops":
             if auteur in self.overops:
                 serv.privmsg(auteur, " ".join(self.ops))
@@ -469,6 +515,11 @@ class Basile(ircbot.SingleServerIRCBot):
                 serv.privmsg(auteur, " ".join(self.overops))
             else:
                 notunderstood = True
+        elif cmd == u"whois":
+            if auteur in self.ops and len(message) > 1:
+                self.whois(message[1], askedwhy="cmd WHOIS", askedby=auteur)
+            else:
+                notunderstood = True
         else:
             notunderstood = True
         if notunderstood:
@@ -756,7 +807,7 @@ class Logger(object):
         f.close()
 
 def main():
-    """Exécution principal : lecture des paramètres et lancement du bot."""
+    """Exécution principale : lecture des paramètres et lancement du bot."""
     if len(sys.argv) == 1:
         print "Usage : basile.py <serveur> [--debug] [--no-output] [--daemon [--pidfile]] [--outfile]"
         print "        --outfile sans --no-output ni --daemon n'a aucun effet"