Implement class-based IRC commands
This commit is contained in:
parent
e2099b1c82
commit
15096be553
|
@ -70,9 +70,10 @@ if __name__ == "__main__":
|
||||||
# Define Transactions
|
# Define Transactions
|
||||||
tx = Transactions()
|
tx = Transactions()
|
||||||
|
|
||||||
# Pass Agora and IRC to Transactions
|
# Pass Agora and IRC to Transactions and Transactions to IRC
|
||||||
tx.set_agora(agora)
|
tx.set_agora(agora)
|
||||||
tx.set_irc(irc)
|
tx.set_irc(irc)
|
||||||
|
irc.set_tx(tx)
|
||||||
|
|
||||||
# Define WebApp
|
# Define WebApp
|
||||||
webapp = WebApp()
|
webapp = WebApp()
|
||||||
|
|
|
@ -0,0 +1,120 @@
|
||||||
|
# Other library imports
|
||||||
|
from json import dumps
|
||||||
|
|
||||||
|
|
||||||
|
class IRCCommands(object):
|
||||||
|
class trades(object):
|
||||||
|
name = "trades"
|
||||||
|
authed = True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def run(cmd, spl, length, authed, msg, agora, revolut, tx):
|
||||||
|
"""
|
||||||
|
Get details of open trades and post on IRC.
|
||||||
|
"""
|
||||||
|
trades = agora.dashboard(send_irc=False)
|
||||||
|
if not trades:
|
||||||
|
msg("No open trades.")
|
||||||
|
for trade_id in trades:
|
||||||
|
msg(trade_id)
|
||||||
|
|
||||||
|
class create(object):
|
||||||
|
name = "create"
|
||||||
|
authed = True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def run(cmd, spl, length, authed, msg, agora, revolut, tx):
|
||||||
|
"""
|
||||||
|
Post an ad on AgoraDesk with the given country and currency code.
|
||||||
|
"""
|
||||||
|
posted = agora.create_ad(spl[1], spl[2])
|
||||||
|
if posted["success"]:
|
||||||
|
msg(f"{posted['response']['data']['message']}: {posted['response']['data']['ad_id']}")
|
||||||
|
else:
|
||||||
|
msg(posted["response"]["data"]["message"])
|
||||||
|
|
||||||
|
class messages(object):
|
||||||
|
name = "messages"
|
||||||
|
authed = True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def run(cmd, spl, length, authed, msg, agora, revolut, tx):
|
||||||
|
"""
|
||||||
|
Get all messages for all open trades or a given trade.
|
||||||
|
"""
|
||||||
|
if length == 1:
|
||||||
|
messages = agora.get_all_messages()
|
||||||
|
if not messages:
|
||||||
|
msg("No messages.")
|
||||||
|
for message_id in messages:
|
||||||
|
for message in messages[message_id]:
|
||||||
|
msg(f"{message_id}: {message}")
|
||||||
|
msg("---")
|
||||||
|
|
||||||
|
elif length == 2:
|
||||||
|
messages = agora.get_messages(spl[1], send_irc=False)
|
||||||
|
if not messages:
|
||||||
|
msg("No messages.")
|
||||||
|
for message in messages:
|
||||||
|
msg(f"{spl[1]}: {message}")
|
||||||
|
|
||||||
|
class dist(object):
|
||||||
|
name = "dist"
|
||||||
|
authed = True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def run(cmd, spl, length, authed, msg, agora, revolut, tx):
|
||||||
|
# Distribute out our ad to all countries in the config
|
||||||
|
rtrn = agora.dist_countries()
|
||||||
|
msg(dumps(rtrn))
|
||||||
|
|
||||||
|
class find(object):
|
||||||
|
name = "find"
|
||||||
|
authed = True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def run(cmd, spl, length, authed, msg, agora, revolut, tx):
|
||||||
|
"""
|
||||||
|
Find a transaction received by Revolut with the given reference and amount.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
int(spl[2])
|
||||||
|
except ValueError:
|
||||||
|
msg("Amount is not an integer.")
|
||||||
|
rtrn = tx.find_tx(spl[1], spl[2])
|
||||||
|
if rtrn == "AMOUNT_INVALID":
|
||||||
|
msg("Reference found but amount invalid.")
|
||||||
|
elif not rtrn:
|
||||||
|
msg("Reference not found.")
|
||||||
|
else:
|
||||||
|
return dumps(rtrn)
|
||||||
|
|
||||||
|
class accounts(object):
|
||||||
|
name = "accounts"
|
||||||
|
authed = True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def run(cmd, spl, length, authed, msg, agora, revolut, tx):
|
||||||
|
accounts = revolut.accounts()
|
||||||
|
for account in accounts:
|
||||||
|
if account["balance"] > 0:
|
||||||
|
msg(f"{account['name']} {account['currency']}: {account['balance']}")
|
||||||
|
|
||||||
|
class total(object):
|
||||||
|
name = "total"
|
||||||
|
authed = True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def run(cmd, spl, length, authed, msg, agora, revolut, tx):
|
||||||
|
total_usd = revolut.get_total_usd()
|
||||||
|
if not total_usd:
|
||||||
|
msg("Error getting total balance.")
|
||||||
|
msg(f"Total: {round(total_usd, 2)}USD")
|
||||||
|
|
||||||
|
class ping(object):
|
||||||
|
name = "ping"
|
||||||
|
authed = False
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def run(cmd, spl, length, authed, msg):
|
||||||
|
msg("Pong!")
|
103
handler/irc.py
103
handler/irc.py
|
@ -3,11 +3,9 @@ from twisted.logger import Logger
|
||||||
from twisted.words.protocols import irc
|
from twisted.words.protocols import irc
|
||||||
from twisted.internet import protocol, reactor, ssl
|
from twisted.internet import protocol, reactor, ssl
|
||||||
|
|
||||||
# Other library imports
|
|
||||||
from json import dumps
|
|
||||||
|
|
||||||
# Project imports
|
# Project imports
|
||||||
from settings import settings
|
from settings import settings
|
||||||
|
from commands import IRCCommands
|
||||||
|
|
||||||
|
|
||||||
class IRCBot(irc.IRCClient):
|
class IRCBot(irc.IRCClient):
|
||||||
|
@ -18,6 +16,9 @@ class IRCBot(irc.IRCClient):
|
||||||
:type log: Logger
|
:type log: Logger
|
||||||
"""
|
"""
|
||||||
self.log = log
|
self.log = log
|
||||||
|
self.cmd = IRCCommands()
|
||||||
|
# Parse the commands into "commandname": "commandclass"
|
||||||
|
self.cmdhash = {getattr(self.cmd, x).name: x for x in dir(self.cmd) if not x.startswith("_")}
|
||||||
self.nickname = settings.IRC.Nick
|
self.nickname = settings.IRC.Nick
|
||||||
self.password = settings.IRC.Pass
|
self.password = settings.IRC.Pass
|
||||||
self.realname = self.nickname
|
self.realname = self.nickname
|
||||||
|
@ -42,6 +43,9 @@ class IRCBot(irc.IRCClient):
|
||||||
def set_revolut(self, revolut):
|
def set_revolut(self, revolut):
|
||||||
self.revolut = revolut
|
self.revolut = revolut
|
||||||
|
|
||||||
|
def set_tx(self, tx):
|
||||||
|
self.tx = tx
|
||||||
|
|
||||||
def parse(self, user, host, channel, msg):
|
def parse(self, user, host, channel, msg):
|
||||||
"""
|
"""
|
||||||
Simple handler for IRC commands.
|
Simple handler for IRC commands.
|
||||||
|
@ -59,71 +63,32 @@ class IRCBot(irc.IRCClient):
|
||||||
|
|
||||||
cmd = spl[0]
|
cmd = spl[0]
|
||||||
|
|
||||||
if cmd == "trades" and host in self.admins:
|
# Check if user is authenticated
|
||||||
# Get details of open trades and post on IRC
|
authed = host in self.admins
|
||||||
trades = self.agora.dashboard(send_irc=False)
|
if cmd in self.cmdhash:
|
||||||
if not trades:
|
# Get the class name of the referenced command
|
||||||
self.msg(channel, "No open trades.")
|
cmdname = self.cmdhash[cmd]
|
||||||
for trade_id in trades:
|
# Get the class name
|
||||||
self.msg(channel, trade_id)
|
obj = getattr(self.cmd, cmdname)
|
||||||
|
|
||||||
elif cmd == "create" and host in self.admins and len(spl) == 3:
|
def msgl(x):
|
||||||
# Post an ad on AgoraDesk with the given country and currency code
|
self.msg(channel, x)
|
||||||
posted = self.agora.create_ad(spl[1], spl[2])
|
|
||||||
if posted["success"]:
|
# Check if the command required authentication
|
||||||
self.msg(channel, f"{posted['response']['data']['message']}: {posted['response']['data']['ad_id']}")
|
if obj.authed:
|
||||||
|
if host in self.admins:
|
||||||
|
obj.run(cmd, spl, len(spl), authed, msgl, self.agora, self.revolut, self.tx)
|
||||||
else:
|
else:
|
||||||
self.msg(channel, posted["response"]["data"]["message"])
|
# Handle authentication here instead of in the command module for security
|
||||||
|
self.msg(channel, "Access denied.")
|
||||||
elif cmd == "messages" and host in self.admins and len(spl) == 1:
|
|
||||||
# Get all messages for all open trades
|
|
||||||
messages = self.agora.get_all_messages()
|
|
||||||
if not messages:
|
|
||||||
self.msg(channel, "No messages.")
|
|
||||||
for message_id in messages:
|
|
||||||
for message in messages[message_id]:
|
|
||||||
self.msg(channel, f"{message_id}: {message}")
|
|
||||||
self.msg(channel, "---")
|
|
||||||
# self.msg(channel, dumps(messages))
|
|
||||||
|
|
||||||
elif cmd == "messages" and host in self.admins and len(spl) == 2:
|
|
||||||
# Get all messages for a given trade
|
|
||||||
messages = self.agora.get_messages(spl[1], send_irc=False)
|
|
||||||
if not messages:
|
|
||||||
self.msg(channel, "No messages.")
|
|
||||||
for message in messages:
|
|
||||||
self.msg(channel, f"{spl[1]}: {message}")
|
|
||||||
|
|
||||||
elif cmd == "dist" and host in self.admins:
|
|
||||||
# Distribute out our ad to all countries in the config
|
|
||||||
rtrn = self.agora.dist_countries()
|
|
||||||
self.msg(channel, dumps(rtrn))
|
|
||||||
|
|
||||||
elif cmd == "find" and host in self.admins and len(spl) == 3:
|
|
||||||
# Find a transaction received by Revolut with the given reference and amount
|
|
||||||
try:
|
|
||||||
int(spl[2])
|
|
||||||
except ValueError:
|
|
||||||
self.msg(channel, "Amount is not an integer.")
|
|
||||||
rtrn = self.tx.find_tx(spl[1], spl[2])
|
|
||||||
if rtrn == "AMOUNT_INVALID":
|
|
||||||
self.msg(channel, "Reference found but amount invalid.")
|
|
||||||
elif not rtrn:
|
|
||||||
self.msg(channel, "Reference not found.")
|
|
||||||
else:
|
else:
|
||||||
return dumps(rtrn)
|
# Run an unauthenticated command, without passing through secure library calls
|
||||||
|
obj.run(cmd, spl, len(spl), authed, msgl)
|
||||||
elif cmd == "accounts" and host in self.admins:
|
return
|
||||||
accounts = self.revolut.accounts()
|
self.msg(channel, "Command not found.")
|
||||||
for account in accounts:
|
if authed:
|
||||||
if account["balance"] > 0:
|
# Give user command hints if they are authenticated
|
||||||
self.msg(channel, f"{account['name']} {account['currency']}: {account['balance']}")
|
self.msg(channel, f"Commands loaded: {', '.join(self.cmdhash.keys())}")
|
||||||
|
|
||||||
elif cmd == "total" and host in self.admins:
|
|
||||||
total_usd = self.revolut.get_total_usd()
|
|
||||||
if not total_usd:
|
|
||||||
self.msg(channel, "Error getting total balance.")
|
|
||||||
self.msg(channel, f"Total: {round(total_usd, 2)}USD")
|
|
||||||
|
|
||||||
def signedOn(self):
|
def signedOn(self):
|
||||||
"""
|
"""
|
||||||
|
@ -170,6 +135,10 @@ class IRCBot(irc.IRCClient):
|
||||||
if len(msg) > 1:
|
if len(msg) > 1:
|
||||||
if msg.split()[0] != "!":
|
if msg.split()[0] != "!":
|
||||||
self.parse(user, host, channel, msg[1:])
|
self.parse(user, host, channel, msg[1:])
|
||||||
|
elif host in self.admins and channel == nick:
|
||||||
|
if len(msg) > 0:
|
||||||
|
if msg.split()[0] != "!":
|
||||||
|
self.parse(user, host, channel, msg)
|
||||||
|
|
||||||
def noticed(self, user, channel, msg):
|
def noticed(self, user, channel, msg):
|
||||||
"""
|
"""
|
||||||
|
@ -197,6 +166,9 @@ class IRCBotFactory(protocol.ClientFactory):
|
||||||
def set_revolut(self, revolut):
|
def set_revolut(self, revolut):
|
||||||
self.revolut = revolut
|
self.revolut = revolut
|
||||||
|
|
||||||
|
def set_tx(self, tx):
|
||||||
|
self.tx = tx
|
||||||
|
|
||||||
def buildProtocol(self, addr):
|
def buildProtocol(self, addr):
|
||||||
"""
|
"""
|
||||||
Custom override for the Twisted buildProtocol so we can access the Protocol instance.
|
Custom override for the Twisted buildProtocol so we can access the Protocol instance.
|
||||||
|
@ -207,6 +179,7 @@ class IRCBotFactory(protocol.ClientFactory):
|
||||||
self.client = prcol
|
self.client = prcol
|
||||||
self.client.set_agora(self.agora)
|
self.client.set_agora(self.agora)
|
||||||
self.client.set_revolut(self.revolut)
|
self.client.set_revolut(self.revolut)
|
||||||
|
self.client.set_tx(self.tx)
|
||||||
return prcol
|
return prcol
|
||||||
|
|
||||||
def clientConnectionLost(self, connector, reason):
|
def clientConnectionLost(self, connector, reason):
|
||||||
|
|
Loading…
Reference in New Issue