From 1501cd0db353d550b39faa6616e266b83e95d530 Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Tue, 12 Apr 2022 22:06:45 +0100
Subject: [PATCH] Implement generic command types

---
 handler/ux/commands.py | 60 +++++++++++++++++++++++++++++++++---------
 1 file changed, 47 insertions(+), 13 deletions(-)

diff --git a/handler/ux/commands.py b/handler/ux/commands.py
index b3f72e5..6e20258 100644
--- a/handler/ux/commands.py
+++ b/handler/ux/commands.py
@@ -5,27 +5,47 @@ from json import dumps, loads
 from settings import settings
 
 
-class IRCCommands(object):
+class GenericCommands(object):
     class trades(object):
-        name = "trades"
         authed = True
-        helptext = "Get all open trades."
+        platform = None
+        name = None
+        helptext = None
 
         @staticmethod
-        def run(cmd, spl, length, authed, msg, agora, tx, ux):
+        def run(cmd, spl, length, authed, msg, agora, tx, ux, caller):
             """
             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()
+
+            trades = caller.get_dashboard()
+
             if not trades:
                 msg("No open trades.")
                 return
             for trade_id in trades:
                 msg(trade_id)
 
+
+class IRCCommands(object):
+    class atrades(object):
+        authed = True
+        name = "atrades"
+        helptext = "Get all open trades for Agora."
+
+        @staticmethod
+        def run(cmd, spl, length, authed, msg, agora, tx, ux):
+            GenericCommands.trades.run(cmd, spl, length, authed, msg, agora, tx, ux, agora)
+
+    class ltrades(GenericCommands.trades):
+        authed = True
+        name = "ltrades"
+        helptext = "Get all open trades for LBTC."
+
+        @staticmethod
+        def run(cmd, spl, length, authed, msg, agora, tx, ux):
+            GenericCommands.trades.run(cmd, spl, length, authed, msg, agora, tx, ux, tx.lbtc)
+
     class create(object):
         name = "create"
         authed = True
@@ -44,7 +64,13 @@ class IRCCommands(object):
                     msg(f"Not a valid asset: {asset}")
                     return
                 _, account_info = tx.markets.get_valid_account_details()
-                posted = agora.create_ad(asset, country, currency, "NATIONAL_BANK", payment_details=account_info[currency])
+                posted = agora.create_ad(
+                    asset,
+                    country,
+                    currency,
+                    "NATIONAL_BANK",
+                    payment_details=account_info[currency],
+                )
                 if posted["success"]:
                     msg(f"{posted['response']['data']['message']}: {posted['response']['data']['ad_id']}")
                 else:
@@ -61,7 +87,13 @@ class IRCCommands(object):
                     msg(f"Not a valid provider: {provider}")
                     return
                 _, account_info = tx.markets.get_valid_account_details()
-                posted = agora.create_ad(asset, country, currency, provider, payment_details=account_info[currency])
+                posted = agora.create_ad(
+                    asset,
+                    country,
+                    currency,
+                    provider,
+                    payment_details=account_info[currency],
+                )
                 if posted["success"]:
                     msg(f"{posted['response']['data']['message']}: {posted['response']['data']['ad_id']}")
                 else:
@@ -533,9 +565,9 @@ class IRCCommands(object):
 
         @staticmethod
         def run(cmd, spl, length, authed, msg, agora, tx, ux):
-            if length == 3:
+            if length >= 3:
                 country = spl[1]
-                bank_name = spl[2]
+                bank_name = " ".join(spl[2:])
                 auth_url = tx.sinks.nordigen.create_auth_url(country, bank_name)
                 if not auth_url:
                     msg("Could not find bank.")
@@ -690,7 +722,9 @@ class IRCCommands(object):
                     for account in accounts:
                         accounts_active.append(account)
                 accounts_all = tx.sinks.truelayer.get_accounts(bank)
-                accounts_unmapped = [x["account_id"] for x in accounts_all["results"] if x["account_id"] not in accounts_active]
+                accounts_unmapped = [
+                    x["account_id"] for x in accounts_all["results"] if x["account_id"] not in accounts_active
+                ]
                 msg(f"Unmapped accounts: {', '.join(accounts_unmapped)}")
 
     class distdetails(object):