Remove leftover files
This commit is contained in:
parent
50c71703fd
commit
4033496866
GPG Key ID:
5ACFCEED46C0904F
|
|
@ -1,451 +0,0 @@
|
||||||
# Other library imports
|
|
||||||
from json import dumps, loads
|
|
||||||
|
|
||||||
# Project imports
|
|
||||||
from settings import settings
|
|
||||||
|
|
||||||
|
|
||||||
class IRCCommands(object):
|
|
||||||
class trades(object):
|
|
||||||
name = "trades"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get all open trades."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
"""
|
|
||||||
Get details of open trades and post on IRC.
|
|
||||||
"""
|
|
||||||
# Send IRC - we don't want to automatically send messages on IRC, even though
|
|
||||||
# this variable seems counter-intuitive here, we are doing something with the result
|
|
||||||
# then calling msg() ourselves, and we don't want extra spam in the channel.
|
|
||||||
trades = agora.get_dashboard()
|
|
||||||
if not trades:
|
|
||||||
msg("No open trades.")
|
|
||||||
return
|
|
||||||
for trade_id in trades:
|
|
||||||
msg(trade_id)
|
|
||||||
|
|
||||||
class create(object):
|
|
||||||
name = "create"
|
|
||||||
authed = True
|
|
||||||
helptext = "Create an ad. Usage: create <XMR/BTC> <country> <currency> [<provider>]"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
"""
|
|
||||||
Post an ad on AgoraDesk with the given country and currency code.
|
|
||||||
"""
|
|
||||||
if length == 4:
|
|
||||||
if spl[1] not in loads(settings.Agora.AssetList):
|
|
||||||
msg(f"Not a valid asset: {spl[1]}")
|
|
||||||
return
|
|
||||||
posted = agora.create_ad(spl[1], spl[2], spl[3], "REVOLUT")
|
|
||||||
if posted["success"]:
|
|
||||||
msg(f"{posted['response']['data']['message']}: {posted['response']['data']['ad_id']}")
|
|
||||||
else:
|
|
||||||
msg(dumps(posted["response"]))
|
|
||||||
elif length == 5:
|
|
||||||
if spl[1] not in loads(settings.Agora.AssetList):
|
|
||||||
msg(f"Not a valid asset: {spl[1]}")
|
|
||||||
return
|
|
||||||
if spl[4] not in loads(settings.Agora.ProviderList):
|
|
||||||
msg(f"Not a valid provider: {spl[4]}")
|
|
||||||
return
|
|
||||||
posted = agora.create_ad(spl[1], spl[2], spl[3], spl[4])
|
|
||||||
if posted["success"]:
|
|
||||||
msg(f"{posted['response']['data']['message']}: {posted['response']['data']['ad_id']}")
|
|
||||||
else:
|
|
||||||
msg(dumps(posted["response"]))
|
|
||||||
|
|
||||||
class messages(object):
|
|
||||||
name = "messages"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get messages. Usage: messages [<reference>]"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
"""
|
|
||||||
Get all messages for all open trades or a given trade.
|
|
||||||
"""
|
|
||||||
if length == 1:
|
|
||||||
messages = agora.get_recent_messages()
|
|
||||||
if messages is False:
|
|
||||||
msg("Error getting messages.")
|
|
||||||
return
|
|
||||||
if not messages:
|
|
||||||
msg("No messages.")
|
|
||||||
return
|
|
||||||
for reference in messages:
|
|
||||||
for message in messages[reference]:
|
|
||||||
msg(f"{reference}: {message[0]} {message[1]}")
|
|
||||||
msg("---")
|
|
||||||
|
|
||||||
elif length == 2:
|
|
||||||
tx = tx.ref_to_tx(spl[1])
|
|
||||||
if not tx:
|
|
||||||
msg(f"No such reference: {spl[1]}")
|
|
||||||
return
|
|
||||||
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
|
|
||||||
helptext = "Distribute all our chosen currency and country ad pairs. Usage: dist [<XMR/BTC>]"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
# Distribute out our ad to all countries in the config
|
|
||||||
if length == 2:
|
|
||||||
asset = spl[1]
|
|
||||||
if asset not in loads(settings.Agora.AssetList):
|
|
||||||
msg(f"Not a valid asset: {spl[1]}")
|
|
||||||
return
|
|
||||||
for x in agora.dist_countries(filter_asset=asset):
|
|
||||||
if x["success"]:
|
|
||||||
msg(f"{x['response']['data']['message']}: {x['response']['data']['ad_id']}")
|
|
||||||
else:
|
|
||||||
msg(dumps(x["response"]))
|
|
||||||
elif length == 1:
|
|
||||||
for x in agora.dist_countries():
|
|
||||||
if x["success"]:
|
|
||||||
msg(f"{x['response']['data']['message']}: {x['response']['data']['ad_id']}")
|
|
||||||
else:
|
|
||||||
msg(dumps(x["response"]))
|
|
||||||
|
|
||||||
class redist(object):
|
|
||||||
name = "redist"
|
|
||||||
authed = True
|
|
||||||
helptext = "Update all ads with details."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
for x in agora.redist_countries():
|
|
||||||
if x[0]["success"]:
|
|
||||||
msg(f"{x[0]['response']['data']['message']}: {x[1]}")
|
|
||||||
else:
|
|
||||||
msg(dumps(x[0]["response"]))
|
|
||||||
|
|
||||||
class stripdupes(object):
|
|
||||||
name = "stripdupes"
|
|
||||||
authed = True
|
|
||||||
helptext = "Remove all duplicate adverts."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
rtrn = agora.strip_duplicate_ads()
|
|
||||||
msg(dumps(rtrn))
|
|
||||||
|
|
||||||
class total(object):
|
|
||||||
name = "total"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get total account balance from Sinks and Agora."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
totals_all = tx.get_total()
|
|
||||||
totals = totals_all[0]
|
|
||||||
wallets = totals_all[1]
|
|
||||||
msg(f"Totals: SEK: {totals[0]} | USD: {totals[1]} | GBP: {totals[2]}")
|
|
||||||
msg(f"Wallets: XMR USD: {wallets[0]} | BTC USD: {wallets[1]}")
|
|
||||||
|
|
||||||
class ping(object):
|
|
||||||
name = "ping"
|
|
||||||
authed = False
|
|
||||||
helptext = "Pong!"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg):
|
|
||||||
msg("Pong!")
|
|
||||||
|
|
||||||
class summon(object):
|
|
||||||
name = "summon"
|
|
||||||
authed = True
|
|
||||||
helptext = "Summon all operators."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
notify.sendmsg("You have been summoned!")
|
|
||||||
|
|
||||||
class message(object):
|
|
||||||
name = "msg"
|
|
||||||
authed = True
|
|
||||||
helptext = "Send a message on a trade. Usage: msg <reference> <message...>"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
if length > 2:
|
|
||||||
full_msg = " ".join(spl[2:])
|
|
||||||
reference = tx.ref_to_tx(spl[1])
|
|
||||||
if not reference:
|
|
||||||
msg(f"No such reference: {spl[1]}")
|
|
||||||
return
|
|
||||||
rtrn = agora.agora.contact_message_post(reference, full_msg)
|
|
||||||
msg(f"Sent {full_msg} to {reference}: {rtrn}")
|
|
||||||
|
|
||||||
class refs(object):
|
|
||||||
name = "refs"
|
|
||||||
authed = True
|
|
||||||
helptext = "List all references"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
msg(f"References: {', '.join(tx.get_refs())}")
|
|
||||||
|
|
||||||
class ref(object):
|
|
||||||
name = "ref"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get more information about a reference. Usage: ref <reference>"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
if length == 2:
|
|
||||||
ref_data = tx.get_ref(spl[1])
|
|
||||||
if not ref_data:
|
|
||||||
msg(f"No such reference: {spl[1]}")
|
|
||||||
return
|
|
||||||
msg(f"{spl[1]}: {dumps(ref_data)}")
|
|
||||||
|
|
||||||
class delete(object):
|
|
||||||
name = "del"
|
|
||||||
authed = True
|
|
||||||
helptext = "Delete a reference. Usage: del <reference>"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
if length == 2:
|
|
||||||
ref_data = tx.get_ref(spl[1])
|
|
||||||
if not ref_data:
|
|
||||||
msg(f"No such reference: {spl[1]}")
|
|
||||||
return
|
|
||||||
tx.del_ref(spl[1])
|
|
||||||
msg(f"Deleted reference: {spl[1]}")
|
|
||||||
|
|
||||||
class release(object):
|
|
||||||
name = "release"
|
|
||||||
authed = True
|
|
||||||
helptext = "Release funds for a trade. Usage: release <reference>"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
if length == 2:
|
|
||||||
tx = tx.ref_to_tx(spl[1])
|
|
||||||
if not tx:
|
|
||||||
msg(f"No such reference: {spl[1]}")
|
|
||||||
return
|
|
||||||
rtrn = agora.release_funds(tx)
|
|
||||||
message = rtrn["message"]
|
|
||||||
message_long = rtrn["response"]["data"]["message"]
|
|
||||||
msg(f"{message} - {message_long}")
|
|
||||||
|
|
||||||
class nuke(object):
|
|
||||||
name = "nuke"
|
|
||||||
authed = True
|
|
||||||
helptext = "Delete all our adverts."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
rtrn = agora.nuke_ads()
|
|
||||||
msg(dumps(rtrn))
|
|
||||||
|
|
||||||
class wallet(object):
|
|
||||||
name = "wallet"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get Agora wallet balances."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
rtrn_xmr = agora.agora.wallet_balance_xmr()
|
|
||||||
if not rtrn_xmr["success"]:
|
|
||||||
msg("Error getting XMR wallet details.")
|
|
||||||
return
|
|
||||||
rtrn_btc = agora.agora.wallet_balance()
|
|
||||||
if not rtrn_btc["success"]:
|
|
||||||
msg("Error getting BTC wallet details.")
|
|
||||||
return
|
|
||||||
balance_xmr = rtrn_xmr["response"]["data"]["total"]["balance"]
|
|
||||||
balance_btc = rtrn_btc["response"]["data"]["total"]["balance"]
|
|
||||||
msg(f"XMR wallet balance: {balance_xmr}")
|
|
||||||
msg(f"BTC wallet balance: {balance_btc}")
|
|
||||||
|
|
||||||
class pubads(object):
|
|
||||||
name = "pubads"
|
|
||||||
authed = True
|
|
||||||
helptext = "View public adverts. Usage: pubads <XMR/BTC> <currency> [<provider,...>]"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
if length == 3:
|
|
||||||
asset = spl[1]
|
|
||||||
if asset not in loads(settings.Agora.AssetList):
|
|
||||||
msg(f"Not a valid asset: {spl[1]}")
|
|
||||||
return
|
|
||||||
currency = spl[2]
|
|
||||||
rtrn = agora.get_all_public_ads(assets=[asset], currencies=[currency])
|
|
||||||
if not rtrn:
|
|
||||||
msg("No results.")
|
|
||||||
return
|
|
||||||
for ad in rtrn[currency]:
|
|
||||||
msg(f"({ad[0]}) {ad[1]} {ad[2]} {ad[3]} {ad[4]} {ad[5]} {ad[6]}")
|
|
||||||
elif length == 4:
|
|
||||||
asset = spl[1]
|
|
||||||
if asset not in loads(settings.Agora.AssetList):
|
|
||||||
msg(f"Not a valid asset: {spl[1]}")
|
|
||||||
return
|
|
||||||
providers = spl[3].split(",")
|
|
||||||
currency = spl[2]
|
|
||||||
rtrn = agora.get_all_public_ads(assets=[asset], currencies=[currency], providers=providers)
|
|
||||||
if not rtrn:
|
|
||||||
msg("No results.")
|
|
||||||
return
|
|
||||||
for ad in rtrn[currency]:
|
|
||||||
msg(f"({ad[0]}) {ad[1]} {ad[2]} {ad[3]} {ad[4]} {ad[5]} {ad[6]}")
|
|
||||||
|
|
||||||
class cheat(object):
|
|
||||||
name = "cheat"
|
|
||||||
authed = True
|
|
||||||
helptext = "Cheat the markets by manipulating our prices to exploit people. Usage: cheat [<XMR/BTC>]"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
if length == 1:
|
|
||||||
agora.run_cheat_in_thread()
|
|
||||||
msg("Running cheat in thread.")
|
|
||||||
elif length == 2:
|
|
||||||
asset = spl[1]
|
|
||||||
if asset not in loads(settings.Agora.AssetList):
|
|
||||||
msg(f"Not a valid asset: {spl[1]}")
|
|
||||||
return
|
|
||||||
agora.run_cheat_in_thread([asset])
|
|
||||||
msg(f"Running cheat in thread for {asset}.")
|
|
||||||
|
|
||||||
class cheatnext(object):
|
|
||||||
name = "cheatnext"
|
|
||||||
authed = True
|
|
||||||
helptext = "Run the next currency for cheat."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
if length == 1:
|
|
||||||
asset = agora.run_cheat_in_thread()
|
|
||||||
msg(f"Running next asset for cheat in thread: {asset}")
|
|
||||||
|
|
||||||
class ads(object):
|
|
||||||
name = "ads"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get all our ad regions"
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
ads = agora.enum_ads()
|
|
||||||
for ad in ads:
|
|
||||||
msg(f"({ad[0]}) {ad[1]} {ad[2]} {ad[3]} {ad[4]}")
|
|
||||||
|
|
||||||
class xmr(object):
|
|
||||||
name = "xmr"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get current XMR price."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
xmr_prices = agora.cg.get_price(ids="monero", vs_currencies=["sek", "usd", "gbp"])
|
|
||||||
price_sek = xmr_prices["monero"]["sek"]
|
|
||||||
price_usd = xmr_prices["monero"]["usd"]
|
|
||||||
price_gbp = xmr_prices["monero"]["gbp"]
|
|
||||||
msg(f"SEK: {price_sek} | USD: {price_usd} | GBP: {price_gbp}")
|
|
||||||
|
|
||||||
class btc(object):
|
|
||||||
name = "btc"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get current BTC price."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
xmr_prices = agora.cg.get_price(ids="bitcoin", vs_currencies=["sek", "usd", "gbp"])
|
|
||||||
price_sek = xmr_prices["bitcoin"]["sek"]
|
|
||||||
price_usd = xmr_prices["bitcoin"]["usd"]
|
|
||||||
price_gbp = xmr_prices["bitcoin"]["gbp"]
|
|
||||||
msg(f"SEK: {price_sek} | USD: {price_usd} | GBP: {price_gbp}")
|
|
||||||
|
|
||||||
class withdraw(object):
|
|
||||||
name = "withdraw"
|
|
||||||
authed = True
|
|
||||||
helptext = "Take profit."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
agora.withdraw_funds()
|
|
||||||
|
|
||||||
class remaining(object):
|
|
||||||
name = "r"
|
|
||||||
authed = True
|
|
||||||
helptext = "Show how much is left before we are able to withdraw funds."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
remaining = tx.get_remaining()
|
|
||||||
msg(f"Remaining: {remaining}USD")
|
|
||||||
|
|
||||||
class total_remaining(object):
|
|
||||||
name = "tr"
|
|
||||||
authed = True
|
|
||||||
helptext = "Show how much is left before we are able to withdraw funds (including open trades)."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
remaining = tx.get_total_remaining()
|
|
||||||
msg(f"Total remaining: {remaining}USD")
|
|
||||||
|
|
||||||
class tradetotal(object):
|
|
||||||
name = "tradetotal"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get total value of all open trades in USD."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
total = tx.get_open_trades_usd()
|
|
||||||
msg(f"Total trades: {total}USD")
|
|
||||||
|
|
||||||
class dollar(object):
|
|
||||||
name = "$"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get total value of everything, including open trades."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
total = tx.get_total_with_trades()
|
|
||||||
msg(f"${total}")
|
|
||||||
|
|
||||||
class profit(object):
|
|
||||||
name = "profit"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get total profit."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
total = tx.money.get_profit()
|
|
||||||
msg(f"Profit: {total}USD")
|
|
||||||
|
|
||||||
class tprofit(object):
|
|
||||||
name = "tprofit"
|
|
||||||
authed = True
|
|
||||||
helptext = "Get total profit with open trades."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
total = tx.money.get_profit(True)
|
|
||||||
msg(f"Profit: {total}USD")
|
|
||||||
|
|
||||||
class signin(object):
|
|
||||||
name = "signin"
|
|
||||||
authed = True
|
|
||||||
helptext = "Generate a TrueLayer signin URL."
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(cmd, spl, length, authed, msg, agora, tx, notify):
|
|
||||||
auth_url = agora.truelayer.create_auth_url()
|
|
||||||
msg(f"Auth URL: {auth_url}")
|
|
||||||
235
handler/irc.py
235
handler/irc.py
|
|
@ -1,235 +0,0 @@
|
||||||
# Twisted/Klein imports
|
|
||||||
from twisted.logger import Logger
|
|
||||||
from twisted.words.protocols import irc
|
|
||||||
from twisted.internet import protocol, reactor, ssl
|
|
||||||
from twisted.internet.task import deferLater
|
|
||||||
|
|
||||||
# Project imports
|
|
||||||
from settings import settings
|
|
||||||
from commands import IRCCommands
|
|
||||||
|
|
||||||
|
|
||||||
class IRCBot(irc.IRCClient):
|
|
||||||
def __init__(self, log):
|
|
||||||
"""
|
|
||||||
Initialise IRC bot.
|
|
||||||
:param log: logger instance
|
|
||||||
:type log: Logger
|
|
||||||
"""
|
|
||||||
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.password = settings.IRC.Pass
|
|
||||||
self.realname = self.nickname
|
|
||||||
self.username = self.nickname
|
|
||||||
|
|
||||||
# Don't give away information about our client
|
|
||||||
self.userinfo = None
|
|
||||||
self.fingerReply = None
|
|
||||||
self.versionName = None
|
|
||||||
self.sourceURL = None
|
|
||||||
self.lineRate = None # Don't throttle messages, we may need to send a lot
|
|
||||||
|
|
||||||
self.prefix = settings.IRC.Prefix
|
|
||||||
self.admins = (settings.IRC.Admins).split("\n")
|
|
||||||
self.highlight = (settings.IRC.Highlight).split("\n")
|
|
||||||
|
|
||||||
self.channel = settings.IRC.Channel
|
|
||||||
|
|
||||||
def parse(self, user, host, channel, msg):
|
|
||||||
"""
|
|
||||||
Simple handler for IRC commands.
|
|
||||||
:param user: full user string with host
|
|
||||||
:param host: user's hostname
|
|
||||||
:param channel: channel the message was received on
|
|
||||||
:param msg: the message
|
|
||||||
:type user: string
|
|
||||||
:type host: string
|
|
||||||
:type channel: string
|
|
||||||
:type msg: string
|
|
||||||
"""
|
|
||||||
spl = msg.split()
|
|
||||||
# nick = user.split("!")[0]
|
|
||||||
|
|
||||||
cmd = spl[0]
|
|
||||||
length = len(spl)
|
|
||||||
|
|
||||||
# Check if user is authenticated
|
|
||||||
authed = host in self.admins
|
|
||||||
if cmd == "help" and length == 2 and authed:
|
|
||||||
if spl[1] in self.cmdhash:
|
|
||||||
cmdname = self.cmdhash[spl[1]]
|
|
||||||
obj = getattr(self.cmd, cmdname)
|
|
||||||
helptext = getattr(obj, "helptext")
|
|
||||||
self.msg(channel, helptext)
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
self.msg(channel, f"No such command: {spl[1]}")
|
|
||||||
return
|
|
||||||
if cmd == "helpall" and authed:
|
|
||||||
for command in self.cmdhash:
|
|
||||||
cmdname = self.cmdhash[command]
|
|
||||||
obj = getattr(self.cmd, cmdname)
|
|
||||||
helptext = getattr(obj, "helptext")
|
|
||||||
self.msg(channel, f"{cmdname}: {helptext}")
|
|
||||||
return
|
|
||||||
|
|
||||||
if cmd in self.cmdhash:
|
|
||||||
# Get the class name of the referenced command
|
|
||||||
cmdname = self.cmdhash[cmd]
|
|
||||||
# Get the class name
|
|
||||||
obj = getattr(self.cmd, cmdname)
|
|
||||||
|
|
||||||
def msgl(x):
|
|
||||||
self.msg(channel, x)
|
|
||||||
|
|
||||||
# Check if the command required authentication
|
|
||||||
if obj.authed:
|
|
||||||
if host in self.admins:
|
|
||||||
obj.run(cmd, spl, length, authed, msgl, self.agora, self.tx, self.notify)
|
|
||||||
else:
|
|
||||||
# Handle authentication here instead of in the command module for security
|
|
||||||
self.msg(channel, "Access denied.")
|
|
||||||
else:
|
|
||||||
# Run an unauthenticated command, without passing through secure library calls
|
|
||||||
obj.run(cmd, spl, len(spl), authed, msgl)
|
|
||||||
return
|
|
||||||
self.msg(channel, "Command not found.")
|
|
||||||
if authed:
|
|
||||||
# Give user command hints if they are authenticated
|
|
||||||
self.msg(channel, f"Commands loaded: {', '.join(self.cmdhash.keys())}")
|
|
||||||
|
|
||||||
def signedOn(self):
|
|
||||||
"""
|
|
||||||
Called when we have signed on to IRC.
|
|
||||||
Join our channel.
|
|
||||||
"""
|
|
||||||
self.log.info("Signed on as %s" % (self.nickname))
|
|
||||||
deferLater(reactor, 2, self.join, self.channel)
|
|
||||||
|
|
||||||
def joined(self, channel):
|
|
||||||
"""
|
|
||||||
Called when we have joined a channel.
|
|
||||||
Setup the Agora LoopingCall to get trades.
|
|
||||||
This is here to ensure the IRC client is initialised enough to send the trades.
|
|
||||||
:param channel: channel we joined
|
|
||||||
:type channel: string
|
|
||||||
"""
|
|
||||||
self.agora.setup_loop()
|
|
||||||
self.log.info("Joined channel %s" % (channel))
|
|
||||||
|
|
||||||
def privmsg(self, user, channel, msg):
|
|
||||||
"""
|
|
||||||
Called on received PRIVMSGs.
|
|
||||||
Pass through identified commands to the parse function.
|
|
||||||
:param user: full user string with host
|
|
||||||
:param channel: channel the message was received on
|
|
||||||
:param msg: the message
|
|
||||||
:type user: string
|
|
||||||
:type channel: string
|
|
||||||
:type msg: string
|
|
||||||
"""
|
|
||||||
nick = user.split("!")[0]
|
|
||||||
|
|
||||||
if channel == self.nickname:
|
|
||||||
channel = nick
|
|
||||||
|
|
||||||
host = user.split("!")[1]
|
|
||||||
host = host.split("@")[1]
|
|
||||||
|
|
||||||
ident = user.split("!")[1]
|
|
||||||
ident = ident.split("@")[0]
|
|
||||||
self.log.info("(%s) %s: %s" % (channel, user, msg))
|
|
||||||
if msg[0] == self.prefix:
|
|
||||||
if len(msg) > 1:
|
|
||||||
if msg.split()[0] != "!":
|
|
||||||
self.parse(user, host, channel, msg[1:])
|
|
||||||
elif host in self.admins and channel == nick:
|
|
||||||
if len(msg) > 0:
|
|
||||||
spl = msg.split()
|
|
||||||
if len(spl) > 0:
|
|
||||||
if spl[0] != "!":
|
|
||||||
self.parse(user, host, channel, msg)
|
|
||||||
|
|
||||||
def noticed(self, user, channel, msg):
|
|
||||||
"""
|
|
||||||
Called on received NOTICEs.
|
|
||||||
:param user: full user string with host
|
|
||||||
:param channel: channel the notice was received on
|
|
||||||
:param msg: the message
|
|
||||||
:type user: string
|
|
||||||
:type channel: string
|
|
||||||
:type msg: string
|
|
||||||
"""
|
|
||||||
nick = user.split("!")[0]
|
|
||||||
if channel == self.nickname:
|
|
||||||
channel = nick
|
|
||||||
# self.log.info("[%s] %s: %s" % (channel, user, msg))
|
|
||||||
|
|
||||||
|
|
||||||
class IRCBotFactory(protocol.ClientFactory):
|
|
||||||
def __init__(self):
|
|
||||||
self.log = Logger("irc")
|
|
||||||
|
|
||||||
def sendmsg(self, msg):
|
|
||||||
"""
|
|
||||||
Passthrough function to send a message to the channel.
|
|
||||||
"""
|
|
||||||
if self.client:
|
|
||||||
self.client.msg(self.client.channel, msg)
|
|
||||||
else:
|
|
||||||
self.log.error("Trying to send a message without connected client: {msg}", msg=msg)
|
|
||||||
return
|
|
||||||
|
|
||||||
def buildProtocol(self, addr):
|
|
||||||
"""
|
|
||||||
Custom override for the Twisted buildProtocol so we can access the Protocol instance.
|
|
||||||
Passes through the Agora instance to IRC.
|
|
||||||
:return: IRCBot Protocol instance
|
|
||||||
"""
|
|
||||||
prcol = IRCBot(self.log)
|
|
||||||
self.client = prcol
|
|
||||||
setattr(self.client, "agora", self.agora)
|
|
||||||
setattr(self.client, "truelayer", self.truelayer)
|
|
||||||
setattr(self.client, "nordigen", self.nordigen)
|
|
||||||
setattr(self.client, "tx", self.tx)
|
|
||||||
setattr(self.client, "notify", self.notify)
|
|
||||||
return prcol
|
|
||||||
|
|
||||||
def clientConnectionLost(self, connector, reason):
|
|
||||||
"""
|
|
||||||
Called when connection to IRC server lost. Reconnect.
|
|
||||||
:param connector: connector object
|
|
||||||
:param reason: reason connection lost
|
|
||||||
:type connector: object
|
|
||||||
:type reason: string
|
|
||||||
"""
|
|
||||||
self.log.error("Lost connection: {reason}, reconnecting", reason=reason)
|
|
||||||
connector.connect()
|
|
||||||
|
|
||||||
def clientConnectionFailed(self, connector, reason):
|
|
||||||
"""
|
|
||||||
Called when connection to IRC server failed. Reconnect.
|
|
||||||
:param connector: connector object
|
|
||||||
:param reason: reason connection failed
|
|
||||||
:type connector: object
|
|
||||||
:type reason: string
|
|
||||||
"""
|
|
||||||
self.log.error("Could not connect: {reason}", reason=reason)
|
|
||||||
connector.connect()
|
|
||||||
|
|
||||||
|
|
||||||
def bot():
|
|
||||||
"""
|
|
||||||
Load the certificates, start the Bot Factory and connect it to the IRC server.
|
|
||||||
:return: Factory instance
|
|
||||||
:rtype: Factory
|
|
||||||
"""
|
|
||||||
# Load the certificates
|
|
||||||
context = ssl.DefaultOpenSSLContextFactory(settings.IRC.Cert, settings.IRC.Cert)
|
|
||||||
# Define the factory instance
|
|
||||||
factory = IRCBotFactory()
|
|
||||||
reactor.connectSSL(settings.IRC.Host, int(settings.IRC.Port), factory, context)
|
|
||||||
return factory
|
|
||||||
|
|
@ -1,45 +0,0 @@
|
||||||
# Twisted/Klein imports
|
|
||||||
from twisted.logger import Logger
|
|
||||||
|
|
||||||
# Other library imports
|
|
||||||
import requests
|
|
||||||
|
|
||||||
# Project imports
|
|
||||||
from settings import settings
|
|
||||||
|
|
||||||
|
|
||||||
class Notify(object):
|
|
||||||
"""
|
|
||||||
Class to handle more robust notifications.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
self.log = Logger("notify")
|
|
||||||
|
|
||||||
def sendmsg(self, msg, title=None, priority=None, tags=None):
|
|
||||||
headers = {"Title": "Bot"}
|
|
||||||
if title:
|
|
||||||
headers["Title"] = title
|
|
||||||
if priority:
|
|
||||||
headers["Priority"] = priority
|
|
||||||
if tags:
|
|
||||||
headers["Tags"] = tags
|
|
||||||
requests.post(
|
|
||||||
f"{settings.Notify.Host}/{settings.Notify.Topic}",
|
|
||||||
data=msg,
|
|
||||||
headers=headers,
|
|
||||||
)
|
|
||||||
|
|
||||||
def notify_new_trade(self, amount, currency):
|
|
||||||
amount_usd = self.money.to_usd(amount, currency)
|
|
||||||
self.sendmsg(f"Total: {amount_usd}", title="New trade", tags="trades", priority="2")
|
|
||||||
|
|
||||||
def notify_complete_trade(self, amount, currency):
|
|
||||||
amount_usd = self.money.to_usd(amount, currency)
|
|
||||||
self.sendmsg(f"Total: {amount_usd}", title="Trade complete", tags="trades,profit", priority="3")
|
|
||||||
|
|
||||||
def notify_withdrawal(self, amount_usd):
|
|
||||||
self.sendmsg(f"Total: {amount_usd}", title="Withdrawal", tags="profit", priority="4")
|
|
||||||
|
|
||||||
def notify_need_topup(self, amount_usd_xmr, amount_usd_btc):
|
|
||||||
self.sendmsg(f"XMR: {amount_usd_xmr} | BTC: {amount_usd_btc}", title="Topup needed", tags="admin", priority="5")
|
|
||||||
Loading…
Reference in New Issue