universalducks/main.py

83 lines
2.9 KiB
Python
Raw Normal View History

2021-07-20 16:53:33 -05:00
import asyncio
import random
2021-07-20 23:06:28 -05:00
import time
2021-07-20 16:53:33 -05:00
from irctokens import build, Line
from ircrobots import Bot as BaseBot
from ircrobots import Server as BaseServer
from ircrobots import ConnectionParams
2021-07-20 23:16:15 -05:00
lang = {
"noduck": "there was no duck! you missed by {} seconds!",
"noduckstart": "there was no duck!",
"duckcought": "duck has been cought by {} in channel {} in {} seconds!",
"duck": "・゜゜・。。・゜゜\_o< QUACK!",
}
2021-07-20 16:53:33 -05:00
class Server(BaseServer):
messages = 0
duckactive = False
2021-07-20 23:16:15 -05:00
duckactivetime = 0
lastduck = 0
2021-07-20 23:06:28 -05:00
async def msg(self, chan, msg, usr=None):
if usr != None:
await self.send(build("PRIVMSG", [chan, usr + ": " + msg]))
else: await self.send(build("PRIVMSG", [chan, msg]))
async def msgall(self, msg):
[await self.msg(channel, msg) for channel in self.channels]
async def new_duck(self):
self.messages = 0
self.duckactive = True
self.duckactivetime = time.time()
2021-07-20 23:16:15 -05:00
await self.msgall(lang["duck"])
async def duck_test(self):
if self.messages > 1 and random.randint(0, 99) < 10: await self.new_duck()
2021-07-20 23:18:31 -05:00
async def misstime(self):
return format(time.time() - self.lastduck, '.2f')
async def coughttime(self):
return format(self.lastduck - self.duckactivetime, '.2f')
2021-07-20 23:06:28 -05:00
async def duck_action(self, user, chan):
if self.duckactive:
self.duckactive = False
self.messages = 0
self.lastduck = time.time()
2021-07-20 23:18:31 -05:00
await self.msgall(lang["duckcought"].format(user, chan, self.coughttime())))
2021-07-20 23:16:15 -05:00
elif self.lastduck != 0:
2021-07-20 23:18:31 -05:00
await self.msg(chan, lang["noduck"].format(self.coughttime())), user)
2021-07-20 23:06:28 -05:00
else:
2021-07-20 23:18:31 -05:00
await self.msg(chan, lang["noduckstart"], user)
2021-07-20 16:53:33 -05:00
async def line_read(self, line: Line):
print(f"{self.name} < {line.format()}")
if line.command == "001":
await self.send(build("JOIN", ["#testchannel"]))
2021-07-20 23:06:28 -05:00
elif line.command == "PRIVMSG":
print(line.params)
print(line.hostmask.nickname)
2021-07-20 16:53:33 -05:00
if line.params[1][0] == '%':
cmd = line.params[1][1:]
2021-07-20 23:06:28 -05:00
chan = line.params[0]
user = line.hostmask.nickname
if cmd == "bef": await self.duck_action(user, chan)
if cmd == "trigger": await self.new_duck()
return
2021-07-20 16:53:33 -05:00
self.messages += 1
self.duck_test()
2021-07-20 23:06:28 -05:00
elif line.command == "INVITE":
await self.send(build("JOIN", [line.params[1]]))
2021-07-20 16:53:33 -05:00
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("test", "beepboop.systems", 6667, False)
await bot.add_server("beep", params)
await bot.run()
if __name__ == "__main__":
asyncio.run(main())