188 lines
8.4 KiB
Python
188 lines
8.4 KiB
Python
import asyncio
|
|
import os
|
|
import random
|
|
import time
|
|
import subprocess
|
|
|
|
from irctokens import build, Line
|
|
from ircrobots import Bot as BaseBot
|
|
from ircrobots import Server as BaseServer
|
|
from ircrobots import ConnectionParams
|
|
|
|
from botany import IRCBotany as Botany
|
|
from admin import Admin
|
|
|
|
from secrets import PASSWORD
|
|
|
|
channels = [
|
|
"#bots",
|
|
"#club",
|
|
"###",
|
|
"#forero",
|
|
]
|
|
helpmessage = "hey, i'm botanybot. i water plants on ~club. my prefix is % and i was made by randomuser. check out https://ttm.sh/Fs4.txt for more information."
|
|
|
|
def userchooser(user):
|
|
return random.choice([i for i in os.listdir(r"/home") if i.lower()[0] == user.lower()[0]])
|
|
|
|
# borrowed from kiedtl/ircbot
|
|
async def _aexec(self, code):
|
|
# Make an async function with the code and `exec` it
|
|
exec(f"async def __ex(self): " + "".join(f"\n {l}" for l in code.split("\n")))
|
|
|
|
# Get `__ex` from local variables, call it and return the result
|
|
return await locals()["__ex"](self)
|
|
|
|
class Server(BaseServer):
|
|
admin = Admin('rndusr')
|
|
async def msg(self, chan, string, user=None):
|
|
if user == None: await self.send(build("PRIVMSG", [chan, string]))
|
|
else: await self.send(build("PRIVMSG", [chan, user + ": " + string]))
|
|
def isDrunk(self):
|
|
if abs(self.drunkentime - int(time.time())) > 60 * 4: return True
|
|
return False
|
|
async def line_read(self, line: Line):
|
|
print(f"{self.name} < {line.format()}")
|
|
if line.command == "001":
|
|
await self.send(build("MODE", ["botanybot", "+B"]))
|
|
await self.msg("nickserv", "identify " + PASSWORD)
|
|
for i in channels:
|
|
await self.send(build("JOIN", [i]))
|
|
self.drunkentime = 0
|
|
if line.command == "PRIVMSG":
|
|
user = line.hostmask.nickname
|
|
if line.params[-1] == "!rollcall":
|
|
await self.msg(user, helpmessage, user)
|
|
if line.params[-1] == "!botlist":
|
|
await self.msg(user, helpmessage, user)
|
|
if line.params[-1][0] == '%':
|
|
commands = line.params[-1][1:].split(' ')
|
|
if commands[0] == "score":
|
|
if len(commands) == 1: commands.append(user)
|
|
if self.isDrunk(): b = Botany(commands[1])
|
|
else:
|
|
b = Botany(userchooser(commands[1]))
|
|
while b.getInfo() == []:
|
|
b = Botany(userchooser(commands[1]))
|
|
i = b.getInfo()
|
|
await self.msg(user, "{}'s score: {}".format(b.user, str(int(b.score()))), user)
|
|
elif commands[0] == "online":
|
|
proc = subprocess.Popen('w', stdout=subprocess.PIPE)
|
|
try: out, err = proc.communicate(timeout=10)
|
|
except:
|
|
proc.kill()
|
|
return
|
|
out = out.decode("utf-8")
|
|
out = out.split("\n")
|
|
out = len(set([i.split(" ")[0] for i in out][2:-1]))
|
|
await self.msg(user, "{} users are currently online".format(out), user)
|
|
|
|
elif commands[0] == "pct":
|
|
if len(commands) == 1: commands.append(user)
|
|
b = Botany(commands[1])
|
|
i = b.getInfo()
|
|
await self.msg(user, "warning: the %pct command is experimental and could possibly not work. you have been warned.", user)
|
|
if len(i) > 1:
|
|
pct = max((1 - ((time.time() - i['last_watered'])/86400)) * 100, 0)
|
|
await self.msg(user, "plant percentage for {}: {}%".format(b.user, int(pct)), user)
|
|
else:
|
|
await self.msg(user, "couldn't find plant for {}".format(commands[1]), user)
|
|
elif commands[0] == "vodka":
|
|
if self.isDrunk():
|
|
self.drunkentime = int(time.time())
|
|
await self.msg(user, "glug glug glug", user)
|
|
else:
|
|
await self.msg(user, "vodka? what's vodka? *burp*", user)
|
|
elif commands[0] == "desc":
|
|
if len(commands) == 1: commands.append(user)
|
|
if self.isDrunk():
|
|
b = Botany(commands[1])
|
|
else:
|
|
b = Botany(userchooser(commands[1]))
|
|
await self.msg(user, b.plantDescription(), user)
|
|
elif commands[0] == "water":
|
|
if len(commands) == 1: commands.append(user)
|
|
if self.isDrunk():
|
|
b = Botany(commands[1])
|
|
if b.water("{} (via IRC)".format(user)):
|
|
await self.msg(user, b.watered(), user)
|
|
else:
|
|
await self.msg(user, b.cantWater(), user)
|
|
else:
|
|
b = Botany(userchooser(commands[1]))
|
|
while not b.water("{} (via IRC)".format(user)):
|
|
b = Botany(userchooser(commands[1]))
|
|
await self.msg(user, b.watered(), user)
|
|
elif commands[0] == "help":
|
|
await self.msg(user, helpmessage, user)
|
|
elif commands[0] == "join":
|
|
if len(commands) == 2:
|
|
if user == self.admin:
|
|
await self.send(build("JOIN", [commands[1]]))
|
|
await self.msg(user, "joined the user {}".format(commands[1]), user)
|
|
else:
|
|
await self.msg(user, "you're not an admin", user)
|
|
else:
|
|
await self.msg(user, "specify the user", user)
|
|
elif commands[0] == "addowner":
|
|
if len(commands) == 2:
|
|
if user == self.admin:
|
|
self.admin.append(commands[1])
|
|
await self.msg(user, "admin added: {}".format(commands[1]), user)
|
|
return
|
|
else:
|
|
await self.msg(user, "error: you must be an admin!", user)
|
|
return
|
|
else:
|
|
await self.msg(user, "two arguments required", user)
|
|
return
|
|
elif commands[0] == "delowner":
|
|
if len(commands) == 2:
|
|
if user == self.admin:
|
|
try: self.admin.remove(commands[1])
|
|
except:
|
|
await self.msg(user, "problem with removing admin", user)
|
|
return
|
|
await self.msg(user, "admin deleted: {}".format(commands[1]), user)
|
|
return
|
|
else:
|
|
await self.msg(user, "error: you must be an admin!", user)
|
|
return
|
|
else:
|
|
await self.msg(user, "two arguments required", user)
|
|
return
|
|
elif commands[0] == "amowner":
|
|
if user == self.admin:
|
|
await self.msg(user, "you're an admin", user)
|
|
else:
|
|
await self.msg(user, "you're not an admin", user)
|
|
elif commands[0] == "ping": await self.msg(user, "pong", user)
|
|
elif commands[0] == "eval":
|
|
if user == self.admin:
|
|
text = ' '.join(line.params[-1][1:].split(' ')[1:])
|
|
try:
|
|
result = await _aexec(self, text)
|
|
except Exception as e:
|
|
await self.msg(user, "segfault: {}".format(repr(e)), user)
|
|
return
|
|
await self.msg(user, "{}".format(result), user)
|
|
else:
|
|
await self.msg(user, "must have admin to execute", user)
|
|
|
|
|
|
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()
|
|
params = ConnectionParams("botanybot", "tilde.chat", 6697, True)
|
|
await bot.add_server("tilde", params)
|
|
await bot.run()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|