115 lines
5.1 KiB
Python
115 lines
5.1 KiB
Python
import sys, os
|
|
import pydle
|
|
import requests
|
|
import random
|
|
import config
|
|
import wownero
|
|
import db
|
|
|
|
|
|
class IRCBot(pydle.Client):
|
|
async def on_connect(self):
|
|
for room in config.ROOMS:
|
|
await self.join(room)
|
|
|
|
async def on_message(self, target, source, message):
|
|
if source == self.nickname:
|
|
return
|
|
else:
|
|
print(f"Target: {target} - Source: {source} - Message: {message}")
|
|
|
|
if self.nickname in message:
|
|
await self.message(target, f"Sup. I'm not very helpful yet, but getting there.")
|
|
elif message.startswith("meme"):
|
|
await self.message(target, random.choice(config.IMAGES))
|
|
elif message.startswith("register"):
|
|
wallet = wownero.Wallet()
|
|
if not wallet.connected:
|
|
await self.message(target, f"My connection to the wallet RPC endpoint is broken so I can't do that right now. Paging: {config.ADMIN_NICKNAMES}")
|
|
if db.User.filter(irc_nick=source):
|
|
await self.message(target, f"You are already registered. FOH.")
|
|
else:
|
|
new_address = wallet.new_address(account=config.ACTIVE_WALLET_ACCOUNT, label=source)
|
|
u = db.User(
|
|
irc_nick=source,
|
|
account_index=config.ACTIVE_WALLET_ACCOUNT,
|
|
address_index=new_address[0]
|
|
)
|
|
u.save()
|
|
print({"type": "registration", "user": source, "address_index": new_address[0]})
|
|
await self.message(target, f"Grats bra, you're locked and loaded. Send Wownero to {new_address[1]}")
|
|
elif message.startswith("deposit"):
|
|
wallet = wownero.Wallet()
|
|
if not wallet.connected:
|
|
await self.message(target, f"My connection to the wallet RPC endpoint is broken so I can't do that right now. Paging: {config.ADMIN_NICKNAMES}")
|
|
if db.User.filter(irc_nick=source):
|
|
u = db.User.get(db.User.irc_nick == source)
|
|
address = wallet.addresses()
|
|
await self.message(target, f"Here's your deposit address {source}: {address[u.address_index]}")
|
|
else:
|
|
await self.message(target, f"You aren't registered. Try saying: `register`")
|
|
elif message.startswith("tip"):
|
|
msg_split = message.split()
|
|
l = len(msg_split)
|
|
if l == 1:
|
|
await self.message(target, "Need more info. `tip <recipient_name> <amount>`")
|
|
elif l == 2:
|
|
await self.message(target, f"Need more info. `tip {msg_split[1]} <amount>`")
|
|
elif l == 3:
|
|
await self.message(target, f"Normally I'd send a tip of {msg_split[2]} Wownero to {msg_split[1]}, but I'm not ready to do that just yet.")
|
|
else:
|
|
await self.message(target, "You're passing way too much shit bro. `tip <recipient_name> <amount>`")
|
|
# elif message.startswith("!proposal "):
|
|
# split_msg = message.split()
|
|
# param = split_msg[1]
|
|
# print(param)
|
|
# print(split_msg)
|
|
# if isinstance(param, int):
|
|
# r = requests.get("https://funding.wownero.com/api/1/proposals?status=2")
|
|
# if r.status_code == 200:
|
|
# found = False
|
|
# for prop in r.json()["data"]:
|
|
# if prop["id"] == param:
|
|
# found = True
|
|
# await self.message(target, f"{prop}")
|
|
# if not found:
|
|
# await self.message(target, f"There's no proposal with that ID")
|
|
# else:
|
|
# await self.message(target, f"I didn't get a valid response from the WFS API. Debug: {r.content}")
|
|
# else:
|
|
# await self.message(target, f"You're supposed to provide a number ya dingus!")
|
|
elif source == config.TELEGRAM_BRIDGE_NAME and config.TELEGRAM_JOIN_SUBSTR in message:
|
|
# ex = 'DudeMan (@DudeManBro) has joined the Telegram Group!'
|
|
strip_last = message.split()[1][:-1]
|
|
strip_first = strip_last[1:]
|
|
await self.message(target, f"WTF is up {strip_first}! Be sure to give us all your money here (https://funding.wownero.com/proposals) and start shitposting here (https://forum.wownero.com/). Stay tuned for more cool ass bot shit goin down!")
|
|
|
|
# isadmin = await self.is_admin(source)
|
|
# print(isadmin)
|
|
|
|
async def is_admin(self, nickname):
|
|
admin = False
|
|
if nickname in config.ADMIN_NICKNAMES:
|
|
info = await self.whois(nickname)
|
|
admin = info['identified']
|
|
print("info: ", info)
|
|
return admin
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
print(f"[+] Starting IRC bot connecting to {config.IRC_HOST}...\n")
|
|
client = IRCBot(
|
|
nickname=config.BOT_NICKNAME,
|
|
sasl_mechanism = "EXTERNAL",
|
|
tls_client_cert = "./freenode.pem",
|
|
)
|
|
client.run(config.IRC_HOST, tls=True, tls_verify=False)
|
|
except KeyboardInterrupt:
|
|
print(' - Adios')
|
|
try:
|
|
sys.exit(0)
|
|
except SystemExit:
|
|
os._exit(0)
|