# Un bot IRC qui, un jour, s'interfacera avec la Note Kfet 2015
-import irclib
-import ircbot
import threading
import random
import time
import pickle
import re
import os
+import signal
+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
+import ircbot
+
from commands import getstatusoutput as ex
# on récupère la config
-import config_basile as config
+import config
# la partie qui réfère au fichier lui-même est mieux ici
# sinon on réfère la config et pas le fichier lui-même
sock=socket.socket()
try:
# On établit la connexion sur port 4242
- sock.connect(("127.0.0.1",4242))
+ sock.connect((config.nk_server, config.nk_port))
# On passe en SSL
sock=ssl.wrap_socket(sock,ca_certs='../keys/ca_.crt')
# On fait un hello
- sock.write('hello "Basile"')
+ sock.write('["hello", "Basile"]')
# On récupère la réponse du hello
out=sock.read()
out=json.loads(out)
def login_NK(username,password,typ="bdd"):
sock=connect_NK()
if typ=="special": # ça c'est pour Basile lui-même
- masque='["note"]'
+ masque='[]'
elif typ=="bdd":
- masque='[["all"],["all"],false]'
+ masque='[[], [], true]'
try:
# Basile a un compte special user
- commande='login [%s,%s,"%s",%s]'%(json.dumps(username),json.dumps(password),typ,masque)
+ commande='["login", [%s,%s,"%s",%s]]'%(json.dumps(username),json.dumps(password),typ,masque)
sock.write(commande)
out=sock.read()
except Exception as exc:
class UnicodeBotError(Exception):
pass
+
+class CrashError(Exception):
+ """Pour pouvoir faire crasher Basile, parce que ça a l'air drôle"""
+ def __init__(self, msg=u""):
+ Exception.__init__(self, msg)
+
def bot_unicode(chain):
try:
unicode(chain,"utf8")
except UnicodeDecodeError as exc:
raise UnicodeBotError
+
class Basile(ircbot.SingleServerIRCBot):
def __init__(self,serveur,debug=False):
temporary_pseudo=config.irc_pseudo+str(random.randrange(10000,100000))
log(self.serveur,"Connected")
if self.debug:
self.chanlist=["#bot"]
- else:
- serv.privmsg("ChanServ", "INVITE #note-dev")
for c in self.chanlist:
log(self.serveur,"JOIN %s"%(c))
serv.join(c)
Me fait perdre sur le channel.""",None],
"reconnect": [None,"""RECONNECT
Établit à nouveau la connexion avec le serveur NK2015""",None],
+ "reload": [None,"""RELOAD
+ Recharge la configuration.""",None],
"say": [None,None,"""SAY <channel> <message>
Me fait parler sur le channel."""],
"do": [None,None,"""DO <channel> <action>
"kick": [None,None,"""KICK <channel> <pseudo> [<raison>]
Kicke <pseudo> du channel (Il faut bien entendu que j'y sois op)."""],
"die": [None,None,"""DIE
- Me déconnecte du serveur IRC."""]
+ Me déconnecte du serveur IRC."""],
+ "crash": [None,None,"""CRASH
+ Me fait crasher"""]
}
helpmsg_default="Liste des commandes disponibles :\nHELP IDENTIFY DROP SOLDE"
- helpmsg_ops=" JOIN LEAVE QUIET NOQUIET LOST RECONNECT"
- helpmsg_overops=" SAY DO STAY NOSTAY OPS OVEROPS KICK DIE"
+ helpmsg_ops=" JOIN LEAVE QUIET NOQUIET LOST RECONNECT RELOAD"
+ helpmsg_overops=" SAY DO STAY NOSTAY OPS OVEROPS KICK DIE CRASH"
op,overop=auteur in self.ops, auteur in self.overops
if len(message)==1:
helpmsg=helpmsg_default
self.mourir()
else:
notunderstood=True
+ elif cmd=="crash":
+ if auteur in self.overops:
+ log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
+ self.crash("priv", auteur)
+ else:
+ notunderstood=True
+ elif cmd=="reload":
+ if auteur in self.ops:
+ self.reload(auteur)
+ log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
+ else:
+ notunderstood=True
elif cmd=="reconnect":
if auteur in self.ops:
try:
if len(message)==1:
if self.identities.has_key(auteur):
try:
- self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(self.identities[auteur])))
+ self.nk.write('["search", ["x",["pseudo"],%s]]'%(json.dumps(self.identities[auteur])))
ret=json.loads(self.nk.read())
solde=ret["msg"][0]["solde"]
pseudo=ret["msg"][0]["pseudo"]
serv.privmsg(canal,"Je ne connais pas ton pseudo note.")
elif auteur in self.ops:
try:
- self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(message[1])))
+ self.nk.write('["search", ["x",["pseudo"],%s]]'%(json.dumps(message[1])))
ret=json.loads(self.nk.read())
solde=ret["msg"][0]["solde"]
pseudo=ret["msg"][0]["pseudo"]
else:
serv.privmsg(canal,("%s: %s"%(auteur,random.choice(config.quit_fail_messages))).encode("utf8"))
log(self.serveur,canal,auteur,message+"[failed]")
-
+ elif cmd == "reload":
+ if auteur in self.ops:
+ log(self.serveur, canal, auteur, message+"[successful]")
+ self.reload(canal)
+ elif cmd == "crash":
+ if auteur in self.overops:
+ self.crash(auteur, message)
elif cmd in ["part","leave","dégage","va-t-en","tut'tiresailleurs,c'estmesgalets"]:
if auteur in self.ops and (not (canal in self.stay_channels)
or auteur in self.overops):
if self.identities.has_key(auteur):
pseudo=self.identities[auteur]
try:
- self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(pseudo)))
+ self.nk.write('["search", ["x",["pseudo"],%s]]'%(json.dumps(pseudo)))
ret=json.loads(self.nk.read())
solde=ret["msg"][0]["solde"]
pseudo=ret["msg"][0]["pseudo"]
if not canal in self.quiet_channels:
mypseudo=self.nick
if re.match((u"^("+u"|".join(config.bonjour_triggers)
- +u")( {}| all| tout le monde|(|à) tous)(\.|( |)!|)$"
- ).format(mypseudo).lower(), message.strip().lower()):
+ +ur")( {}| all| tout le monde| (à )?tous)(\.| ?!)?$"
+ ).format(mypseudo).lower(), message.decode("utf8").strip().lower()):
answer=random.choice(config.bonjour_answers)
serv.privmsg(canal,answer.format(auteur).encode("utf8"))
if (is_perdu(message) and not canal in self.quiet_channels):
return self.serv.get_nickname()
nick=property(_getnick)
+ def reload(self, auteur=None):
+ reload(config)
+ if auteur in [None, "SIGHUP"]:
+ towrite = "Config reloaded" + " (SIGHUP received)"*(auteur == "SIGHUP")
+ for to in config.report_bugs_to:
+ self.serv.privmsg(to, towrite)
+ log(self.serveur, towrite)
+ else:
+ self.serv.privmsg(auteur,"Config reloaded")
+
+ def crash(self, chan="nowhere", who="nobody"):
+ where = u"en privé" if chan == "priv" else u"sur le chan %s" % chan
+ raise CrashError(u"Crash demandé par %s %s" % (who, where))
+
+ def start_as_daemon(self, outfile):
+ sys.stderr = Logger(outfile)
+ self.start()
+
+
+class Logger(object):
+ """Pour écrire ailleurs que sur stdout"""
+ def __init__(self, filename="basile.full.log"):
+ self.filename = filename
-if __name__=="__main__":
- import sys
+ def write(self, message):
+ f = open(self.filename, "a")
+ f.write(message)
+ f.close()
+
+def main():
if len(sys.argv)==1:
- print "Usage : basile.py <serveur> [--debug]"
+ print "Usage : basile.py <serveur> [--debug] [--no-output] [--daemon [--pidfile]] [--outfile]"
+ print " --outfile sans --no-output ni --daemon n'a aucun effet"
exit(1)
serveur=sys.argv[1]
+ if "--daemon" in sys.argv:
+ thisfile = os.path.realpath(__file__)
+ thisdirectory = thisfile.rsplit("/", 1)[0]
+ os.chdir(thisdirectory)
+ daemon = True
+ else:
+ daemon = False
if "debug" in sys.argv or "--debug" in sys.argv:
debug=True
else:
config.debug_stdout=False
serveurs={"a♡":"acoeur.crans.org","acoeur":"acoeur.crans.org","acoeur.crans.org":"acoeur.crans.org",
"irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org"}
+ if "--no-output" in sys.argv or "--daemon" in sys.argv:
+ outfile = "/var/log/bots/basile.full.log"
+ for arg in sys.argv:
+ arg = arg.split("=")
+ if arg[0].strip('-') in ["out", "outfile", "logfile"]:
+ outfile = arg[1]
+ sys.stdout = Logger(outfile)
try:
serveur=serveurs[serveur]
except KeyError:
print "Server Unknown : %s"%(serveur)
exit(404)
basile=Basile(serveur,debug)
- basile.start()
+ # Si on reçoit un SIGHUP, on reload la config
+ def sighup_handler(signum, frame):
+ basile.reload("SIGHUP")
+ signal.signal(signal.SIGHUP, sighup_handler)
+ if daemon:
+ child_pid = os.fork()
+ if child_pid == 0:
+ os.setsid()
+ basile.start_as_daemon(outfile)
+ else:
+ # on enregistre le pid de basile
+ pidfile = "/var/run/bots/basile.pid"
+ for arg in sys.argv:
+ arg = arg.split("=")
+ if arg[0].strip('-') in ["pidfile"]:
+ pidfile = arg[1]
+ f = open(pidfile, "w")
+ f.write("%s\n" % child_pid)
+ f.close()
+ else:
+ basile.start()
+
+if __name__ == "__main__":
+ main()