Make bot extend a bare-bones version of itself so type-hinting works as intended
This commit is contained in:
parent
b90d9c02f3
commit
53efdfb830
3 changed files with 90 additions and 118 deletions
74
bare.py
Normal file
74
bare.py
Normal file
|
@ -0,0 +1,74 @@
|
||||||
|
#!/usr/bin/python3
|
||||||
|
from socket import socket
|
||||||
|
from overrides import bytes, bbytes
|
||||||
|
logs = ...
|
||||||
|
re = ...
|
||||||
|
from typing import NoReturn, Union
|
||||||
|
cmds = ...
|
||||||
|
conf = ...
|
||||||
|
sleep = ...
|
||||||
|
reload = ...
|
||||||
|
r = ...
|
||||||
|
handlers = ...
|
||||||
|
|
||||||
|
|
||||||
|
def mfind(message: str, find: list, usePrefix: bool = True) -> bool:
|
||||||
|
...
|
||||||
|
|
||||||
|
|
||||||
|
class bot:
|
||||||
|
gmode: bool
|
||||||
|
server: str
|
||||||
|
nicklen: int
|
||||||
|
address: str
|
||||||
|
port: int
|
||||||
|
channels: dict[str, int]
|
||||||
|
interval: int
|
||||||
|
__version__: str
|
||||||
|
nick: str
|
||||||
|
adminnames: list[str]
|
||||||
|
queue: list[bbytes]
|
||||||
|
sock: socket
|
||||||
|
npallowed: list[str]
|
||||||
|
|
||||||
|
def __init__(self, server: str):
|
||||||
|
...
|
||||||
|
|
||||||
|
def connect(self) -> None:
|
||||||
|
...
|
||||||
|
|
||||||
|
def join(self, chan: str, origin: str, lock: bool = True) -> None:
|
||||||
|
...
|
||||||
|
|
||||||
|
def ping(self, ircmsg: str) -> int:
|
||||||
|
...
|
||||||
|
|
||||||
|
def send(self, command: str) -> int:
|
||||||
|
...
|
||||||
|
|
||||||
|
def recv(self) -> bytes:
|
||||||
|
...
|
||||||
|
|
||||||
|
def log(self, message: str, level: str = "LOG") -> None:
|
||||||
|
...
|
||||||
|
|
||||||
|
def exit(self, message: str) -> NoReturn:
|
||||||
|
...
|
||||||
|
|
||||||
|
def CTCP(self, msg: str, sender: str = "", isRaw: bool = False) -> bool:
|
||||||
|
...
|
||||||
|
|
||||||
|
def msg(self, msg: str, target: str) -> None:
|
||||||
|
...
|
||||||
|
|
||||||
|
def op(self, name: str, chan: str) -> Union[int, None]:
|
||||||
|
...
|
||||||
|
|
||||||
|
def notice(self, msg: str, target: str, silent: bool = False) -> int:
|
||||||
|
...
|
||||||
|
|
||||||
|
def sendraw(self, command: str) -> int:
|
||||||
|
...
|
||||||
|
|
||||||
|
def mainloop(self) -> NoReturn:
|
||||||
|
...
|
117
bot.py
117
bot.py
|
@ -9,6 +9,8 @@ import config as conf
|
||||||
from time import sleep
|
from time import sleep
|
||||||
from importlib import reload
|
from importlib import reload
|
||||||
import random as r
|
import random as r
|
||||||
|
import handlers
|
||||||
|
import bare
|
||||||
|
|
||||||
|
|
||||||
def mfind(message: str, find: list, usePrefix: bool = True) -> bool:
|
def mfind(message: str, find: list, usePrefix: bool = True) -> bool:
|
||||||
|
@ -18,7 +20,7 @@ def mfind(message: str, find: list, usePrefix: bool = True) -> bool:
|
||||||
return any(message[: len(match)] == match for match in find)
|
return any(message[: len(match)] == match for match in find)
|
||||||
|
|
||||||
|
|
||||||
class bot:
|
class bot(bare.bot):
|
||||||
def __init__(self, server: str):
|
def __init__(self, server: str):
|
||||||
self.gmode = False
|
self.gmode = False
|
||||||
self.server = server
|
self.server = server
|
||||||
|
@ -243,118 +245,11 @@ class bot:
|
||||||
except IndexError:
|
except IndexError:
|
||||||
pass
|
pass
|
||||||
if action == "PRIVMSG":
|
if action == "PRIVMSG":
|
||||||
# Format of ":[Nick]![ident]@[host|vhost] PRIVMSG [channel] :[message]”
|
res = handlers.PRIVMSG(self, ircmsg)
|
||||||
name = ircmsg.split("!", 1)[0][1:]
|
if res == "reload":
|
||||||
helpErr = False
|
|
||||||
if (name.startswith("saxjax") and self.server == "efnet") or (
|
|
||||||
name == "ReplIRC" and self.server == "replirc"
|
|
||||||
):
|
|
||||||
if ircmsg.find("<") != -1 and ircmsg.find(">") != -1:
|
|
||||||
Nname = ircmsg.split("<", 1)[1].split(">", 1)[0].strip()
|
|
||||||
if name == "ReplIRC":
|
|
||||||
name = Nname[4:]
|
|
||||||
else:
|
|
||||||
name = Nname
|
|
||||||
message = ircmsg.split(">", 1)[1].strip()
|
|
||||||
helpErr = True
|
|
||||||
else:
|
|
||||||
message = (
|
|
||||||
ircmsg.split("PRIVMSG", 1)[1].split(":", 1)[1].strip()
|
|
||||||
)
|
|
||||||
elif name == self.nick:
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
message = ircmsg.split("PRIVMSG", 1)[1].split(":", 1)[1].strip()
|
|
||||||
if name.endswith("dsc"):
|
|
||||||
helpErr = True
|
|
||||||
chan = ircmsg.split("PRIVMSG", 1)[1].split(":", 1)[0].strip()
|
|
||||||
self.log(
|
|
||||||
f'Got "{bytes(message).lazy_decode()}" from "{name}" in "{chan}"',
|
|
||||||
)
|
|
||||||
if len(name) > self.nicklen:
|
|
||||||
self.log(f"Name too long ({len(name)} > {self.nicklen})")
|
|
||||||
continue
|
|
||||||
elif chan not in self.channels:
|
|
||||||
self.log(
|
|
||||||
f"Channel not in channels ({chan} not in {self.channels})",
|
|
||||||
"WARN",
|
|
||||||
)
|
|
||||||
if not chan.startswith(("#", "+", "&")):
|
|
||||||
chan = name
|
|
||||||
else:
|
|
||||||
self.channels[chan] += 1
|
|
||||||
if "goat" in name.lower() and self.gmode == True:
|
|
||||||
cmds.goat(self, chan, name, message)
|
|
||||||
handled = False
|
|
||||||
for cmd in cmds.data:
|
|
||||||
triggers = [cmd]
|
|
||||||
triggers.extend(cmds.data[cmd]["aliases"])
|
|
||||||
triggers = list(
|
|
||||||
call.replace("$BOTNICK", self.nick.lower())
|
|
||||||
for call in triggers
|
|
||||||
)
|
|
||||||
if mfind(
|
|
||||||
message.lower(),
|
|
||||||
triggers,
|
|
||||||
cmds.data[cmd]["prefix"],
|
|
||||||
):
|
|
||||||
if (
|
|
||||||
"admin" in cmds.data[cmd] and cmds.data[cmd]["admin"]
|
|
||||||
) and name not in self.adminnames:
|
|
||||||
self.msg(
|
|
||||||
f"Sorry {name}, you don't have permission to use {cmd.strip()}.",
|
|
||||||
chan,
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
cmds.call[cmd](self, chan, name, message)
|
|
||||||
handled = True
|
|
||||||
break
|
|
||||||
if not handled:
|
|
||||||
for check in cmds.checks:
|
|
||||||
if re.search(
|
|
||||||
check.replace("$MAX", str(self.nicklen)).replace(
|
|
||||||
"$BOTNICK", self.nick
|
|
||||||
),
|
|
||||||
message,
|
|
||||||
):
|
|
||||||
cmds.call[check](self, chan, name, message)
|
|
||||||
handled = True
|
|
||||||
break
|
|
||||||
if not handled and mfind(message, ["reload"]):
|
|
||||||
if name in self.adminnames:
|
|
||||||
reload(conf)
|
reload(conf)
|
||||||
reload(cmds)
|
reload(cmds)
|
||||||
self.__version__ = conf.__version__
|
reload(handlers)
|
||||||
self.msg("Reloaded config and commands", chan)
|
|
||||||
else:
|
|
||||||
self.msg(
|
|
||||||
f"Sorry {name}, you don't have permission to use reload.",
|
|
||||||
chan,
|
|
||||||
)
|
|
||||||
handled = True
|
|
||||||
if not handled and len(message.split("\x01")) == 3:
|
|
||||||
if not self.CTCP(message, name):
|
|
||||||
CTCP = message.split("\x01")[1]
|
|
||||||
if CTCP == "ACTION ducks":
|
|
||||||
self.msg("\x01ACTION gets hit by a duck\x01", chan)
|
|
||||||
elif CTCP.startswith("ACTION ducks"):
|
|
||||||
self.msg(
|
|
||||||
f"\x01ACTION gets hit by {CTCP.split(' ', 2)[2]}\x01",
|
|
||||||
chan,
|
|
||||||
)
|
|
||||||
if chan in self.channels and self.channels[chan] >= self.interval:
|
|
||||||
r.seed()
|
|
||||||
self.channels[chan] = 0
|
|
||||||
mm = open("mastermessages.txt", "r")
|
|
||||||
q = mm.readlines()
|
|
||||||
sel = conf.decode_escapes(
|
|
||||||
str(r.sample(q, 1))
|
|
||||||
.strip("[]'")
|
|
||||||
.replace("\\n", "")
|
|
||||||
.strip('"')
|
|
||||||
)
|
|
||||||
self.msg(f"[QUOTE] {sel}", chan)
|
|
||||||
mm.close()
|
|
||||||
elif action == "NICK":
|
elif action == "NICK":
|
||||||
name = ircmsg.split("!", 1)[0][1:]
|
name = ircmsg.split("!", 1)[0][1:]
|
||||||
if name == self.nick:
|
if name == self.nick:
|
||||||
|
|
11
handlers.py
11
handlers.py
|
@ -5,11 +5,10 @@ import commands as cmds
|
||||||
from typing import Union
|
from typing import Union
|
||||||
from overrides import bytes, bbytes
|
from overrides import bytes, bbytes
|
||||||
from importlib import reload
|
from importlib import reload
|
||||||
|
import bare
|
||||||
|
|
||||||
def CTCP(bot, msg: str, sender: str = "", isRaw: bool = False) -> bool:
|
def CTCP(bot: bare.bot, msg: str) -> bool:
|
||||||
if isRaw:
|
|
||||||
sender = msg.split("!", 1)[0][1:]
|
sender = msg.split("!", 1)[0][1:]
|
||||||
message = msg.split("PRIVMSG", 1)[1].split(":", 1)[1].strip()
|
|
||||||
kind = msg.split("\x01")[1].split(" ", 1)[0]
|
kind = msg.split("\x01")[1].split(" ", 1)[0]
|
||||||
bot.log(f'Responding to CTCP "{kind}" from {sender}')
|
bot.log(f'Responding to CTCP "{kind}" from {sender}')
|
||||||
if kind == "VERSION":
|
if kind == "VERSION":
|
||||||
|
@ -40,7 +39,7 @@ def CTCP(bot, msg: str, sender: str = "", isRaw: bool = False) -> bool:
|
||||||
bot.log(f'Unknown CTCP "{kind}"', "WARN")
|
bot.log(f'Unknown CTCP "{kind}"', "WARN")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
def PRIVMSG(bot, msg: str) -> Union[None, str]:
|
def PRIVMSG(bot: bare.bot, msg: str) -> Union[None, str]:
|
||||||
# Format of ":[Nick]![ident]@[host|vhost] PRIVMSG [channel] :[message]”
|
# Format of ":[Nick]![ident]@[host|vhost] PRIVMSG [channel] :[message]”
|
||||||
name = msg.split("!", 1)[0][1:]
|
name = msg.split("!", 1)[0][1:]
|
||||||
if (name.startswith("saxjax") and bot.server == "efnet") or (
|
if (name.startswith("saxjax") and bot.server == "efnet") or (
|
||||||
|
@ -147,3 +146,7 @@ def PRIVMSG(bot, msg: str) -> Union[None, str]:
|
||||||
bot.msg(f"[QUOTE] {sel}", chan)
|
bot.msg(f"[QUOTE] {sel}", chan)
|
||||||
mm.close()
|
mm.close()
|
||||||
return # type: ignore
|
return # type: ignore
|
||||||
|
|
||||||
|
handles = {
|
||||||
|
"PRIVMSG": PRIVMSG
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue