From bb7bc6498cc94c8c0993a4c33b9d274c172502f9 Mon Sep 17 00:00:00 2001 From: Mark Veidemanis Date: Fri, 24 Dec 2021 17:24:45 +0000 Subject: [PATCH] Add IRC module --- handler/irc.py | 139 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 139 insertions(+) create mode 100644 handler/irc.py diff --git a/handler/irc.py b/handler/irc.py new file mode 100644 index 0000000..c794839 --- /dev/null +++ b/handler/irc.py @@ -0,0 +1,139 @@ +# Twisted/Klein imports +from twisted.logger import Logger +from twisted.words.protocols import irc +from twisted.internet import protocol, reactor, ssl + +# Project imports +from settings import settings + + +class IRCBot(irc.IRCClient): + def __init__(self, log): + self.log = log + self.nickname = settings.IRC.Nick + self.password = settings.IRC.Pass + self.realname = self.nickname + self.username = self.nickname + + self.userinfo = None + self.fingerReply = None + self.versionName = None + self.sourceURL = None + + self.prefix = "!" + self.admins = (settings.IRC.Admins).split("\n") + self.highlight = (settings.IRC.Highlight).split("\n") + + def parse(self, user, host, channel, msg): + spl = msg.split() + # nick = user.split("!")[0] + + cmd = spl[0] + + # help command + # if cmd == "help": + # self.msg(channel, "Hey it's me") + + # setprefix command + if cmd == "setprefix" and host in self.admins: + if len(spl) == 2: + if len(spl[1]) == 1: + self.prefix = spl[1] + self.msg(channel, "Prefix set to '%s'." % self.prefix) + else: + self.msg(channel, "Incorrect usage") + + # addmod command + elif cmd == "addmod" and host in self.admins: + if len(spl) == 2: + if not spl[1] in self.admins: + self.admins.append(spl[1]) + self.msg(channel, "%s added to admins." % spl[1]) + else: + self.msg(channel, "%s is already an admin!" % spl[1]) + else: + self.msg(channel, "Incorrect usage") + + # delmod command + elif cmd == "delmod" and host in self.admins: + if len(spl) == 2: + if spl[1] in self.admins: + self.admins.remove(spl[1]) + self.msg(channel, "%s removed from admins." % spl[1]) + else: + self.msg(channel, "%s is not an admin!" % spl[1]) + else: + self.msg(channel, "Incorrect usage") + + # join command + elif cmd == "join" and host in self.admins: + if len(spl) == 2: + self.join(spl[1]) + else: + self.msg(channel, "Incorrect usage") + + # part command + elif cmd == "part" and host in self.admins: + if len(spl) == 2: + self.part(spl[1]) + else: + self.part(channel) + elif cmd == "raw" and host in self.admins: + self.sendLine(" ".join(spl[1:])) + + def stopcall(self, call): + call.stop() + + def signedOn(self): + self.log.info("Signed on as %s" % (self.nickname)) + self.join(settings.IRC.Channel) + + def joined(self, channel): + self.log.info("Joined channel %s" % (channel)) + + def privmsg(self, user, channel, msg): + nick = user.split("!")[0] + + if channel == self.nickname: + channel = nick + + host = user.split("!")[1] + host = host.split("@")[1] + + ident = user.split("!")[1] + ident = ident.split("@")[0] + self.log.info("(%s) %s: %s" % (channel, user, msg)) + if msg[0] == self.prefix: + if len(msg) > 1: + if msg.split()[0] != "!": + self.parse(user, host, channel, msg[1:]) + + def noticed(self, user, channel, message): + nick = user.split("!")[0] + if channel == self.nickname: + channel = nick + # self.log.info("[%s] %s: %s" % (channel, user, message)) + + +class IRCBotFactory(protocol.ClientFactory): + def __init__(self): + self.log = Logger("irc") + + def buildProtocol(self, addr): + prcol = IRCBot(self.log) + self.client = prcol + return prcol + + def clientConnectionLost(self, connector, reason): + self.log.error("Lost connection (%s), reconnecting" % (reason)) + connector.connect() + + def clientConnectionFailed(self, connector, reason): + self.log.error("Could not connect: %s" % (reason)) + + +def bot(): + context = ssl.DefaultOpenSSLContextFactory("./cert.pem", "./cert.pem") + factory = IRCBotFactory() + reactor.connectSSL(settings.IRC.Host, int(settings.IRC.Port), factory, context) + return factory