120 lines
3.3 KiB
Python
Executable File
120 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# modbot is copyright rndusr/randomuser 2021
|
|
# This file is part of modbot.
|
|
|
|
# modbot is free software: you can redistribute it and/or
|
|
# modify it under the terms of the GNU Affero General Public
|
|
# License as published by the Free Software Foundation,
|
|
# either version 3 of the License, or (at your option) any
|
|
# later version.
|
|
|
|
# modbot 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 Affero General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Affero
|
|
# General Public License along with modbot. If not, see
|
|
# <https://www.gnu.org/licenses/>.
|
|
|
|
import asyncio
|
|
from importlib import import_module as mod_include
|
|
import os
|
|
import sys
|
|
|
|
from irctokens import build, Line
|
|
from ircrobots import Bot as BaseBot
|
|
from ircrobots import Server as BaseServer
|
|
from ircrobots import ConnectionParams
|
|
|
|
SERVERS = [
|
|
("localhost", "localhost")
|
|
]
|
|
EVENTS = [
|
|
"cmd",
|
|
"privmsg",
|
|
"topic",
|
|
"mode",
|
|
"kick",
|
|
"join",
|
|
"part",
|
|
"invite",
|
|
]
|
|
ADMINS = ['rndusr']
|
|
PREFIX = '*'
|
|
|
|
class Admin:
|
|
def __init__(self, nick):
|
|
self.nicks = nick
|
|
def __eq__(self, val):
|
|
return val in self.nicks
|
|
def __contains__(self, val):
|
|
return val in self.nicks
|
|
def append(self, nick):
|
|
self.nicks.append(nick)
|
|
def remove(self, nick):
|
|
self.nicks.remove(nick)
|
|
|
|
class Server(BaseServer):
|
|
handlers = {i: [] for i in EVENTS}
|
|
admins = Admin(ADMINS)
|
|
states = {}
|
|
async def line_read(self, line: Line):
|
|
print(f"{self.name} < {line.format()}")
|
|
if line.command == "001":
|
|
await self.send(build("JOIN", ["#chaos"]))
|
|
self.load_mod("default")
|
|
self.event_handle(line)
|
|
|
|
def load_mod(self, name):
|
|
mod = mod_include(name)
|
|
for i in EVENTS:
|
|
if hasattr(mod, i):
|
|
self.handlers[i].append([name, getattr(mod, i)])
|
|
self.states[name] = None
|
|
|
|
def unload_mod(self, name):
|
|
for i in EVENTS:
|
|
self.handlers[i] = [j for j in self.handlers[i] if not j[0] == name]
|
|
|
|
def event_get(self, line: Line):
|
|
if line.command.lower() in EVENTS[2:]:
|
|
return line.command.lower()
|
|
elif line.command.lower() == "privmsg":
|
|
if line.params[1][0] == PREFIX: return "cmd"
|
|
else: return "privmsg"
|
|
else:
|
|
# something has gone horribly, horribly wrong
|
|
return None
|
|
|
|
def event_handle(self, line: Line):
|
|
event = self.event_get(line)
|
|
if event == None: return False
|
|
try:
|
|
for i in self.handlers[event]:
|
|
ret = i[1](line, self)
|
|
if not ret == None:
|
|
self.states[i[0]] = ret
|
|
except IndexError: return False
|
|
return True
|
|
|
|
async def line_send(self, line: Line):
|
|
print(f"{self.name} > {line.format()}")
|
|
|
|
class Bot(BaseBot):
|
|
def create_server(self, name: str):
|
|
return Server(self, name)
|
|
|
|
async def main():
|
|
bot = Bot()
|
|
for name, host in SERVERS:
|
|
params = ConnectionParams("BitBotNewTest", host, 6667, False)
|
|
await bot.add_server(name, params)
|
|
|
|
await bot.run()
|
|
|
|
if __name__ == "__main__":
|
|
os.chdir("mods")
|
|
sys.path = [os.getcwd()] + sys.path
|
|
asyncio.run(main())
|