Formatting
This commit is contained in:
parent
ce6dc597ef
commit
b720b5edf8
4 changed files with 135 additions and 91 deletions
7
bot.py
7
bot.py
|
@ -53,7 +53,7 @@ class bot(bare.bot):
|
||||||
else "FireBot"
|
else "FireBot"
|
||||||
)
|
)
|
||||||
self.queue: list[bbytes] = [] # pyright: ignore [reportInvalidTypeForm]
|
self.queue: list[bbytes] = [] # pyright: ignore [reportInvalidTypeForm]
|
||||||
self.statuses = {'firepup': {}}
|
self.statuses = {"firepup": {}}
|
||||||
self.ops = {}
|
self.ops = {}
|
||||||
self.sock = socket(AF_INET, SOCK_STREAM)
|
self.sock = socket(AF_INET, SOCK_STREAM)
|
||||||
self.current = "user"
|
self.current = "user"
|
||||||
|
@ -349,7 +349,10 @@ class bot(bare.bot):
|
||||||
Err = format_exc()
|
Err = format_exc()
|
||||||
for line in Err.split("\n"):
|
for line in Err.split("\n"):
|
||||||
self.log(line, "ERROR")
|
self.log(line, "ERROR")
|
||||||
self.msg("Reload failed, likely partially reloaded. Please check error logs.", chan)
|
self.msg(
|
||||||
|
"Reload failed, likely partially reloaded. Please check error logs.",
|
||||||
|
chan,
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
if ircmsg.startswith("PING "):
|
if ircmsg.startswith("PING "):
|
||||||
self.ping(ircmsg)
|
self.ping(ircmsg)
|
||||||
|
|
55
commands.py
55
commands.py
|
@ -155,19 +155,19 @@ def debug(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
||||||
def debugInternal(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
def debugInternal(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
||||||
things = dir(bot)
|
things = dir(bot)
|
||||||
try:
|
try:
|
||||||
thing = message.split(' ', 1)[1]
|
thing = message.split(" ", 1)[1]
|
||||||
except IndexError:
|
except IndexError:
|
||||||
bot.msg("You can't just ask me to lookup nothing.", chan)
|
bot.msg("You can't just ask me to lookup nothing.", chan)
|
||||||
return
|
return
|
||||||
if thing in things:
|
if thing in things:
|
||||||
bot.msg(f"self.{thing} = {getattr(bot, thing)}", chan)
|
bot.msg(f"self.{thing} = {getattr(bot, thing)}", chan)
|
||||||
else:
|
else:
|
||||||
bot.msg(f"I have nothing called \"{thing}\"", chan)
|
bot.msg(f'I have nothing called "{thing}"', chan)
|
||||||
|
|
||||||
|
|
||||||
def debugEval(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
def debugEval(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
||||||
try:
|
try:
|
||||||
bot.msg(str(eval(message.split(' ', 1)[1])), chan)
|
bot.msg(str(eval(message.split(" ", 1)[1])), chan)
|
||||||
except Exception as E:
|
except Exception as E:
|
||||||
bot.msg(f"Exception: {E}", chan)
|
bot.msg(f"Exception: {E}", chan)
|
||||||
|
|
||||||
|
@ -234,17 +234,23 @@ def markov(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
||||||
|
|
||||||
|
|
||||||
def setStatus(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
def setStatus(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
||||||
user, stat, reas = ('', 0, '')
|
user, stat, reas = ("", 0, "")
|
||||||
try:
|
try:
|
||||||
if message.split(' ')[1] == "help":
|
if message.split(" ")[1] == "help":
|
||||||
bot.msg("Assuming you want help with status codes. 1 is Available, 2 is Busy, 3 is Unavailable, anything else is Unknown.", chan)
|
bot.msg(
|
||||||
|
"Assuming you want help with status codes. 1 is Available, 2 is Busy, 3 is Unavailable, anything else is Unknown.",
|
||||||
|
chan,
|
||||||
|
)
|
||||||
return
|
return
|
||||||
message = message.split(' ', 1)[1]
|
message = message.split(" ", 1)[1]
|
||||||
user = message.split(' ')[0].lower()
|
user = message.split(" ")[0].lower()
|
||||||
stat = int(message.split(' ')[1])
|
stat = int(message.split(" ")[1])
|
||||||
reas = message.split(' ', 2)[2]
|
reas = message.split(" ", 2)[2]
|
||||||
except IndexError:
|
except IndexError:
|
||||||
bot.msg(f"Insufficent information to set a status. Only got {len(message.split(' ')) - (1 if '.sS' in message else 0)}/3 expected args.", chan)
|
bot.msg(
|
||||||
|
f"Insufficent information to set a status. Only got {len(message.split(' ')) - (1 if '.sS' in message else 0)}/3 expected args.",
|
||||||
|
chan,
|
||||||
|
)
|
||||||
return
|
return
|
||||||
except ValueError:
|
except ValueError:
|
||||||
bot.msg("Status parameter must be an int.", chan)
|
bot.msg("Status parameter must be an int.", chan)
|
||||||
|
@ -259,28 +265,31 @@ def setStatus(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
||||||
case _:
|
case _:
|
||||||
stat = "Unknown"
|
stat = "Unknown"
|
||||||
if user in ["me", "my", "I"]:
|
if user in ["me", "my", "I"]:
|
||||||
user = 'firepup'
|
user = "firepup"
|
||||||
bot.statuses[user] = {'status': stat, 'reason': reas}
|
bot.statuses[user] = {"status": stat, "reason": reas}
|
||||||
bot.msg(f"Status set for '{user}'. Raw data: {bot.statuses[user]}", chan)
|
bot.msg(f"Status set for '{user}'. Raw data: {bot.statuses[user]}", chan)
|
||||||
|
|
||||||
|
|
||||||
def getStatus(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
def getStatus(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
||||||
user = ''
|
user = ""
|
||||||
try:
|
try:
|
||||||
user = message.split(' ')[1]
|
user = message.split(" ")[1]
|
||||||
except IndexError:
|
except IndexError:
|
||||||
user = 'firepup'
|
user = "firepup"
|
||||||
if bot.statuses.get(user) is None:
|
if bot.statuses.get(user) is None:
|
||||||
bot.msg("You've gotta provide a nick I actually recognize.", chan)
|
bot.msg("You've gotta provide a nick I actually recognize.", chan)
|
||||||
return
|
return
|
||||||
bot.msg(f"{user}'s status: {'Unknown' if not bot.statuses[user].get('status') else bot.statuses[user]['status']} - {'Reason unset' if not bot.statuses[user].get('reason') else bot.statuses[user]['reason']}", chan)
|
bot.msg(
|
||||||
|
f"{user}'s status: {'Unknown' if not bot.statuses[user].get('status') else bot.statuses[user]['status']} - {'Reason unset' if not bot.statuses[user].get('reason') else bot.statuses[user]['reason']}",
|
||||||
|
chan,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def check(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
def check(bot: bare.bot, chan: str, name: str, message: str) -> None:
|
||||||
try:
|
try:
|
||||||
msg = message.split(' ', 1)[1]
|
msg = message.split(" ", 1)[1]
|
||||||
nick = msg.split('!')[0]
|
nick = msg.split("!")[0]
|
||||||
host = msg.split('@', 1)[1]
|
host = msg.split("@", 1)[1]
|
||||||
dnsbl, raws = conf.dnsblHandler(bot, nick, host, chan)
|
dnsbl, raws = conf.dnsblHandler(bot, nick, host, chan)
|
||||||
bot.msg(f"Blacklist check: {dnsbl if dnsbl else 'Safe.'} ({raws})", chan)
|
bot.msg(f"Blacklist check: {dnsbl if dnsbl else 'Safe.'} ({raws})", chan)
|
||||||
except Exception as E:
|
except Exception as E:
|
||||||
|
@ -301,7 +310,11 @@ data: dict[str, dict[str, Any]] = {
|
||||||
"uptime": {"prefix": True, "aliases": []},
|
"uptime": {"prefix": True, "aliases": []},
|
||||||
"raw": {"prefix": True, "aliases": ["cmd "], "check": checks.admin},
|
"raw": {"prefix": True, "aliases": ["cmd "], "check": checks.admin},
|
||||||
"debug": {"prefix": True, "aliases": ["dbg", "d"], "check": checks.admin},
|
"debug": {"prefix": True, "aliases": ["dbg", "d"], "check": checks.admin},
|
||||||
"debugInternal": {"prefix": True, "aliases": ["dbgInt", "dI"], "check": checks.admin},
|
"debugInternal": {
|
||||||
|
"prefix": True,
|
||||||
|
"aliases": ["dbgInt", "dI"],
|
||||||
|
"check": checks.admin,
|
||||||
|
},
|
||||||
"debugEval": {"prefix": True, "aliases": ["dbgEval", "dE"], "check": checks.admin},
|
"debugEval": {"prefix": True, "aliases": ["dbgEval", "dE"], "check": checks.admin},
|
||||||
"8ball": {"prefix": True, "aliases": ["eightball", "8b"]},
|
"8ball": {"prefix": True, "aliases": ["eightball", "8b"]},
|
||||||
"join": {"prefix": True, "aliases": ["j"], "check": checks.admin},
|
"join": {"prefix": True, "aliases": ["j"], "check": checks.admin},
|
||||||
|
|
75
config.py
75
config.py
|
@ -6,25 +6,27 @@ from typing import Optional, Any, Union
|
||||||
import bare, pylast
|
import bare, pylast
|
||||||
from pydnsbl import DNSBLIpChecker, DNSBLDomainChecker, providers as BL
|
from pydnsbl import DNSBLIpChecker, DNSBLDomainChecker, providers as BL
|
||||||
|
|
||||||
|
|
||||||
class droneBL(BL.Provider):
|
class droneBL(BL.Provider):
|
||||||
def process_response(self, response):
|
def process_response(self, response):
|
||||||
reasons = set()
|
reasons = set()
|
||||||
for result in response:
|
for result in response:
|
||||||
reason = result.host
|
reason = result.host
|
||||||
if reason in ['127.0.0.3']:
|
if reason in ["127.0.0.3"]:
|
||||||
reasons.add('IRC Spambot')
|
reasons.add("IRC Spambot")
|
||||||
elif reason in ['127.0.0.19']:
|
elif reason in ["127.0.0.19"]:
|
||||||
reasons.add('Abused VPN')
|
reasons.add("Abused VPN")
|
||||||
elif reason in ['127.0.0.9', '127.0.0.8']:
|
elif reason in ["127.0.0.9", "127.0.0.8"]:
|
||||||
reasons.add('Open Proxy')
|
reasons.add("Open Proxy")
|
||||||
elif reason in ['127.0.0.13']:
|
elif reason in ["127.0.0.13"]:
|
||||||
reasons.add('Automated Attacks')
|
reasons.add("Automated Attacks")
|
||||||
else:
|
else:
|
||||||
print('Unknown dnsbl reason: ' + reason, flush=True)
|
print("Unknown dnsbl reason: " + reason, flush=True)
|
||||||
reasons.add('unknown')
|
reasons.add("unknown")
|
||||||
return reasons
|
return reasons
|
||||||
|
|
||||||
providers = BL.BASE_PROVIDERS + [droneBL('dnsbl.dronebl.org')]
|
|
||||||
|
providers = BL.BASE_PROVIDERS + [droneBL("dnsbl.dronebl.org")]
|
||||||
|
|
||||||
ipbl = DNSBLIpChecker(providers=providers)
|
ipbl = DNSBLIpChecker(providers=providers)
|
||||||
hsbl = DNSBLDomainChecker(providers=providers)
|
hsbl = DNSBLDomainChecker(providers=providers)
|
||||||
|
@ -44,7 +46,7 @@ servers: dict[str, dict[str, Any]] = {
|
||||||
"channels": {"#random": 0, "#dice": 0, "#offtopic": 0, "#main/replirc": 0},
|
"channels": {"#random": 0, "#dice": 0, "#offtopic": 0, "#main/replirc": 0},
|
||||||
"ignores": ["#main/replirc"],
|
"ignores": ["#main/replirc"],
|
||||||
"hosts": ["9pfs.repl.co"],
|
"hosts": ["9pfs.repl.co"],
|
||||||
"dnsblMode": "kickban"
|
"dnsblMode": "kickban",
|
||||||
},
|
},
|
||||||
"efnet": {
|
"efnet": {
|
||||||
"address": "irc.underworld.no",
|
"address": "irc.underworld.no",
|
||||||
|
@ -126,8 +128,9 @@ def decode_escapes(s: str) -> str:
|
||||||
def cmdFind(message: str, find: list, usePrefix: bool = True) -> bool:
|
def cmdFind(message: str, find: list, usePrefix: bool = True) -> bool:
|
||||||
cmd = None
|
cmd = None
|
||||||
try:
|
try:
|
||||||
cmd = message.split(' ', 1)[0]
|
cmd = message.split(" ", 1)[0]
|
||||||
except IndexError: ...
|
except IndexError:
|
||||||
|
...
|
||||||
if not cmd:
|
if not cmd:
|
||||||
return False
|
return False
|
||||||
if usePrefix:
|
if usePrefix:
|
||||||
|
@ -136,7 +139,6 @@ def cmdFind(message: str, find: list, usePrefix: bool = True) -> bool:
|
||||||
return any(cmd == match for match in find)
|
return any(cmd == match for match in find)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def mfind(message: str, find: list, usePrefix: bool = True) -> bool:
|
def mfind(message: str, find: list, usePrefix: bool = True) -> bool:
|
||||||
if usePrefix:
|
if usePrefix:
|
||||||
return any(message[: len(match) + 1] == prefix + match for match in find)
|
return any(message[: len(match) + 1] == prefix + match for match in find)
|
||||||
|
@ -175,31 +177,50 @@ def dnsbl(hostname: str) -> tuple[str, dict[str, list[str]]]:
|
||||||
return "", hstDT
|
return "", hstDT
|
||||||
hostStr = None
|
hostStr = None
|
||||||
if len(hosts) >= 3:
|
if len(hosts) >= 3:
|
||||||
hostStr = ', and '.join((', '.join(hosts)).rsplit(", ", 1))
|
hostStr = ", and ".join((", ".join(hosts)).rsplit(", ", 1))
|
||||||
else:
|
else:
|
||||||
hostStr = ' and '.join(hosts)
|
hostStr = " and ".join(hosts)
|
||||||
return hostStr, hstDT
|
return hostStr, hstDT
|
||||||
|
|
||||||
def dnsblHandler(bot: bare.bot, nick: str, hostname: str, chan: str) -> tuple[str, dict[str, list[str]]]:
|
|
||||||
dnsblStatus = 'Not enabled'
|
def dnsblHandler(
|
||||||
|
bot: bare.bot, nick: str, hostname: str, chan: str
|
||||||
|
) -> tuple[str, dict[str, list[str]]]:
|
||||||
|
dnsblStatus = "Not enabled"
|
||||||
dnsblResps = {}
|
dnsblResps = {}
|
||||||
if bot.dnsblMode != "none":
|
if bot.dnsblMode != "none":
|
||||||
dnsblStatus, dnsblResps = dnsbl(hostname)
|
dnsblStatus, dnsblResps = dnsbl(hostname)
|
||||||
if dnsblStatus:
|
if dnsblStatus:
|
||||||
match bot.dnsblMode:
|
match bot.dnsblMode:
|
||||||
case "kickban":
|
case "kickban":
|
||||||
bot.sendraw(f"KICK #{chan} {nick} :Sorry, but you're on the {dnsblStatus} blacklist(s).")
|
bot.sendraw(
|
||||||
|
f"KICK #{chan} {nick} :Sorry, but you're on the {dnsblStatus} blacklist(s)."
|
||||||
|
)
|
||||||
bot.sendraw(f"MODE #{chan} +b *!*@{hostname}")
|
bot.sendraw(f"MODE #{chan} +b *!*@{hostname}")
|
||||||
case "akill":
|
case "akill":
|
||||||
bot.sendraw(f"OS AKILL ADD *@{hostname} !P Sorry, but you're on the {dnsblStatus} blacklist(s).")
|
bot.sendraw(
|
||||||
|
f"OS AKILL ADD *@{hostname} !P Sorry, but you're on the {dnsblStatus} blacklist(s)."
|
||||||
|
)
|
||||||
case "kline":
|
case "kline":
|
||||||
bot.sendraw(f"KILL {nick} :Sorry, but you're on the {dnsblStatus} blacklist(s).")
|
bot.sendraw(
|
||||||
bot.sendraw(f"KLINE 524160 *@{hostname} :Sorry, but you're on the {dnsblStatus} blacklist(s).")
|
f"KILL {nick} :Sorry, but you're on the {dnsblStatus} blacklist(s)."
|
||||||
bot.sendraw(f"KLINE *@{hostname} :Sorry, but you're on the {dnsblStatus} blacklist(s).")
|
)
|
||||||
|
bot.sendraw(
|
||||||
|
f"KLINE 524160 *@{hostname} :Sorry, but you're on the {dnsblStatus} blacklist(s)."
|
||||||
|
)
|
||||||
|
bot.sendraw(
|
||||||
|
f"KLINE *@{hostname} :Sorry, but you're on the {dnsblStatus} blacklist(s)."
|
||||||
|
)
|
||||||
case "gline":
|
case "gline":
|
||||||
bot.sendraw(f"KILL {nick} :Sorry, but you're on the {dnsblStatus} blacklist(s).")
|
bot.sendraw(
|
||||||
bot.sendraw(f"GLINE *@{hostname} 524160 :Sorry, but you're on the {dnsblStatus} blacklist(s).")
|
f"KILL {nick} :Sorry, but you're on the {dnsblStatus} blacklist(s)."
|
||||||
bot.sendraw(f"GLINE *@{hostname} :Sorry, but you're on the {dnsblStatus} blacklist(s).")
|
)
|
||||||
|
bot.sendraw(
|
||||||
|
f"GLINE *@{hostname} 524160 :Sorry, but you're on the {dnsblStatus} blacklist(s)."
|
||||||
|
)
|
||||||
|
bot.sendraw(
|
||||||
|
f"GLINE *@{hostname} :Sorry, but you're on the {dnsblStatus} blacklist(s)."
|
||||||
|
)
|
||||||
case _:
|
case _:
|
||||||
bot.log(f'Unknown dnsbl Mode "{bot.dnsblMode}"!', "WARN")
|
bot.log(f'Unknown dnsbl Mode "{bot.dnsblMode}"!', "WARN")
|
||||||
return dnsblStatus, dnsblResps
|
return dnsblStatus, dnsblResps
|
||||||
|
|
13
handlers.py
13
handlers.py
|
@ -117,7 +117,10 @@ def PRIVMSG(bot: bare.bot, msg: str) -> Union[tuple[None, None], tuple[str, str]
|
||||||
Err = format_exc()
|
Err = format_exc()
|
||||||
for line in Err.split("\n"):
|
for line in Err.split("\n"):
|
||||||
bot.log(line, "ERROR")
|
bot.log(line, "ERROR")
|
||||||
bot.msg("Sorry, I had an error trying to execute that command. Please check error logs.", chan)
|
bot.msg(
|
||||||
|
"Sorry, I had an error trying to execute that command. Please check error logs.",
|
||||||
|
chan,
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
try:
|
try:
|
||||||
cmds.call[cmd](bot, chan, name, message)
|
cmds.call[cmd](bot, chan, name, message)
|
||||||
|
@ -125,7 +128,10 @@ def PRIVMSG(bot: bare.bot, msg: str) -> Union[tuple[None, None], tuple[str, str]
|
||||||
Err = format_exc()
|
Err = format_exc()
|
||||||
for line in Err.split("\n"):
|
for line in Err.split("\n"):
|
||||||
bot.log(line, "ERROR")
|
bot.log(line, "ERROR")
|
||||||
bot.msg("Sorry, I had an error trying to execute that command. Please check error logs.", chan)
|
bot.msg(
|
||||||
|
"Sorry, I had an error trying to execute that command. Please check error logs.",
|
||||||
|
chan,
|
||||||
|
)
|
||||||
handled = True
|
handled = True
|
||||||
break
|
break
|
||||||
if not handled:
|
if not handled:
|
||||||
|
@ -213,7 +219,8 @@ def MODE(bot: bare.bot, msg: str) -> tuple[None, None]:
|
||||||
users = ""
|
users = ""
|
||||||
try:
|
try:
|
||||||
users = msg.split("#", 1)[1].split(" ", 2)[2].split()
|
users = msg.split("#", 1)[1].split(" ", 2)[2].split()
|
||||||
except IndexError: ...
|
except IndexError:
|
||||||
|
...
|
||||||
if len(modes) != len(users):
|
if len(modes) != len(users):
|
||||||
bot.log("Refusing to handle modes that do not have corresponding users.")
|
bot.log("Refusing to handle modes that do not have corresponding users.")
|
||||||
return None, None
|
return None, None
|
||||||
|
|
Loading…
Reference in a new issue