Handle multiple accounts with TrueLayer

handler/app.py

@@ -4,8 +4,7 @@ from twisted.internet import reactor
 from klein import Klein
 
 # Other library imports
-from json import dumps, loads
-from json.decoder import JSONDecodeError
+from json import dumps
 from signal import signal, SIGINT
 
 # Project imports
@@ -47,23 +46,23 @@ class WebApp(util.Base):
 
     app = Klein()
 
-    @app.route("/callback", methods=["POST"])
-    def callback(self, request):
-        content = request.content.read()
-        try:
-            parsed = loads(content)
-        except JSONDecodeError:
-            self.log.error(f"Failed to parse JSON callback: {content}")
-            return dumps(False)
-        self.log.info("Callback received: {parsed}", parsed=parsed["data"]["id"])
-        # self.tx.transaction(parsed)
-        return dumps(True)
+    # @app.route("/callback", methods=["POST"])
+    # def callback(self, request):
+    #    content = request.content.read()
+    #    try:
+    #        parsed = loads(content)
+    #    except JSONDecodeError:
+    #        self.log.error(f"Failed to parse JSON callback: {content}")
+    #        return dumps(False)
+    #    self.log.info("Callback received: {parsed}", parsed=parsed["data"]["id"])
+    #    # self.tx.transaction(parsed)
+    #    return dumps(True)
 
     # set up another connection to a bank
-    @app.route("/signin", methods=["GET"])
-    def signin(self, request):
-        auth_url = self.sinks.truelayer.create_auth_url()
-        return f'Please sign in <a href="{auth_url}" target="_blank">here.</a>'
+    @app.route("/signin/<string:account>", methods=["GET"])
+    def signin(self, request, account):
+        auth_url = self.sinks.truelayer.create_auth_url(account)
+        return f'Please sign in to {account} <a href="{auth_url}" target="_blank">here.</a>'
 
     # endpoint called after we finish setting up a connection above
     @app.route("/callback-truelayer", methods=["POST"])
@@ -72,9 +71,9 @@ class WebApp(util.Base):
         self.sinks.truelayer.handle_authcode_received(code)
         return dumps(True)
 
-    @app.route("/accounts", methods=["GET"])
-    def balance(self, request):
-        accounts = self.sinks.truelayer.get_accounts()
+    @app.route("/accounts/<string:account>", methods=["GET"])
+    def balance(self, request, account):
+        accounts = self.sinks.truelayer.get_accounts(account)
         return dumps(accounts, indent=2)
 
 

handler/sinks/truelayer.py

@@ -5,6 +5,7 @@ from twisted.internet.task import LoopingCall
 import requests
 from simplejson.errors import JSONDecodeError
 from time import time
+from json import dumps, loads
 import urllib
 
 # Project imports
@@ -19,11 +20,41 @@ class TrueLayer(util.Base):
 
     def __init__(self):
         super().__init__()
-        self.token = None
-        self.lc = LoopingCall(self.get_new_token)
+        self.tokens = {}
+
+        # account we are authenticating - where to store the refresh keys
+        self.current_authcode_account = None
+        self.lc = LoopingCall(self.get_new_tokens_all)
         self.lc.start(int(settings.TrueLayer.RefreshSec))
 
-    def create_auth_url(self):
+    def add_refresh_token(self, refresh_token):
+        """
+        Add an API key to the configuration.
+        Data type: {"monzo":   refresh_token,
+                    "revolut": refresh_token}
+        """
+        print("ADD REFRESH TOKEN", refresh_token)
+        account = self.current_authcode_account
+        if not account:
+            print("CURRENT AUTHCODE ACCOUNT", account)
+            return False
+        existing_entry = loads(settings.TrueLayer.RefreshKeys)
+        print("existing entry", existing_entry)
+        existing_entry[account] = refresh_token
+        settings.TrueLayer.RefreshKeys = dumps(existing_entry)
+        settings.write()
+
+    def get_refresh_tokens(self):
+        existing_entry = loads(settings.TrueLayer.RefreshKeys)
+        return existing_entry
+
+    def get_key(self, account):
+        if account in self.tokens:
+            return self.tokens[account]
+        else:
+            return False
+
+    def create_auth_url(self, account):
         query = urllib.parse.urlencode(
             {
                 "response_type": "code",
@@ -36,6 +67,8 @@ class TrueLayer(util.Base):
             }
         )
         auth_uri = f"{settings.TrueLayer.AuthBase}/?{query}&redirect_uri={settings.TrueLayer.CallbackURL}"
+        print("SETTING AUTHCODE ACCOUNT TO", account)
+        self.current_authcode_account = account
         return auth_uri
 
     def handle_authcode_received(self, authcode):
@@ -52,24 +85,46 @@ class TrueLayer(util.Base):
         except JSONDecodeError:
             return False
         if "error" in parsed:
-            self.log.error("Error requesting refresh token: {error}", error=parsed["error"])
+            self.log.error("Error requesting refresh token: {parsed['error']}")
             return False
-        settings.TrueLayer.RefreshToken = parsed["refresh_token"]
-        settings.TrueLayer.AuthCode = authcode
-        settings.write()
-        self.token = parsed["access_token"]
-        self.log.info("Retrieved access/refresh tokens")
 
-    def get_new_token(self):
+        # Extract the access tokens
+        refresh_token = parsed["refresh_token"]
+        access_token = parsed["access_token"]
+
+        # Add the refresh token
+        self.add_refresh_token(refresh_token)
+
+        # Add the access
+        if self.current_authcode_account:
+            self.tokens[self.current_authcode_account] = access_token
+        else:
+            self.log.error(f"Received an authcode we didn't ask for")
+            return
+        self.log.info(f"Retrieved access/refresh tokens for {self.current_authcode_account}")
+
+    def get_new_tokens_all(self):
+        print("get new_tokens_all running")
+        refresh_list = loads(settings.TrueLayer.RefreshKeys)
+
+        for account in refresh_list:
+            print("new_new_tokens_all running on", account)
+            self.get_new_token(account)
+
+    def get_new_token(self, account):
         """
         Exchange our refresh token for an access token.
+        :param account: account to refresh the token for
+        :type account:
         """
-        if not settings.TrueLayer.RefreshToken:
+        refresh_tokens = self.get_refresh_tokens()
+        if account not in refresh_tokens:
             return
+
         headers = {"Content-Type": "application/x-www-form-urlencoded"}
         data = {
             "grant_type": "refresh_token",
-            "refresh_token": settings.TrueLayer.RefreshToken,
+            "refresh_token": refresh_tokens[account],
             "client_id": settings.TrueLayer.ID,
             "client_secret": settings.TrueLayer.Key,
         }
@@ -80,8 +135,8 @@ class TrueLayer(util.Base):
             return False
         if r.status_code == 200:
             if "access_token" in parsed.keys():
-                self.token = parsed["access_token"]
-                self.log.info("Refreshed access token")
+                self.tokens[account] = parsed["access_token"]
+                self.log.info(f"Refreshed access token for {account}")
                 return True
             else:
                 self.log.error(f"Token refresh didn't contain access token: {parsed}", parsed=parsed)
@@ -90,11 +145,12 @@ class TrueLayer(util.Base):
             self.log.error(f"Cannot refresh token: {parsed}", parsed=parsed)
             return False
 
-    def get_accounts(self):
+    def get_accounts(self, account):
         """
         Get a list of accounts.
         """
-        headers = {"Authorization": f"Bearer {self.token}"}
+        token = self.get_key(account)
+        headers = {"Authorization": f"Bearer {token}"}
         path = f"{settings.TrueLayer.DataBase}/accounts"
         r = requests.get(path, headers=headers)
         try:
@@ -105,14 +161,15 @@ class TrueLayer(util.Base):
 
         return parsed
 
-    def get_transactions(self, account_id):
+    def get_transactions(self, account, account_id):
         """
         Get a list of transactions from an account.
         :param account_id: account to fetch transactions for
         :return: list of transactions
         :rtype: dict
         """
-        headers = {"Authorization": f"Bearer {self.token}"}
+        token = self.get_key(account)
+        headers = {"Authorization": f"Bearer {token}"}
         path = f"{settings.TrueLayer.DataBase}/accounts/{account_id}/transactions"
         r = requests.get(path, headers=headers)
         try:

handler/ux/commands.py

@@ -443,36 +443,45 @@ class IRCCommands(object):
     class signin(object):
         name = "signin"
         authed = True
-        helptext = "Generate a TrueLayer signin URL."
+        helptext = "Generate a TrueLayer signin URL. Usage: signin <account>"
 
         @staticmethod
         def run(cmd, spl, length, authed, msg, agora, tx, ux):
-            auth_url = tx.truelayer.create_auth_url()
-            msg(f"Auth URL: {auth_url}")
+            if length == 2:
+                account = spl[1]
+                auth_url = tx.truelayer.create_auth_url(account)
+                msg(f"Auth URL for {account}: {auth_url}")
 
     class accounts(object):
         name = "accounts"
         authed = True
-        helptext = "Get a list of acccounts."
+        helptext = "Get a list of acccounts. Usage: accounts <account>"
 
         @staticmethod
         def run(cmd, spl, length, authed, msg, agora, tx, ux):
-            accounts = tx.sinks.truelayer.get_accounts()
-            msg(dumps(accounts))
+            if length == 2:
+                account = spl[1]
+                accounts = tx.sinks.truelayer.get_accounts(account)
+                msg(dumps(accounts))
 
     class transactions(object):
         name = "transactions"
         authed = True
-        helptext = "Get a list of transactions. Usage: transactions <account_id>"
+        helptext = "Get a list of transactions. Usage: transactions <account> <account_id>"
 
         @staticmethod
         def run(cmd, spl, length, authed, msg, agora, tx, ux):
-            if length == 2:
-                account_id = spl[1]
-                transactions = tx.sinks.truelayer.get_transactions(account_id)
+            if length == 3:
+                account = spl[1]
+                account_id = spl[2]
+                transactions = tx.sinks.truelayer.get_transactions(account, account_id)
                 for transaction in transactions:
+                    print(transaction)
+                    txid = transaction["transaction_id"]
+                    ptxid = transaction["provider_transaction_id"]
+                    txtype = transaction["transaction_type"]
                     timestamp = transaction["timestamp"]
                     amount = transaction["amount"]
                     currency = transaction["currency"]
-                    recipient = transaction["counter_party_preferred_name"]
-                    msg(f"{timestamp} {amount}{currency} {recipient}")
+                    description = transaction["description"]
+                    msg(f"{timestamp} {txid} {ptxid} {txtype} {amount}{currency} {description}")