diff --git a/handler/app.py b/handler/app.py
index 01e6a63..3234a66 100755
--- a/handler/app.py
+++ b/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 here.'
+ @app.route("/signin/", methods=["GET"])
+ def signin(self, request, account):
+ auth_url = self.sinks.truelayer.create_auth_url(account)
+ return f'Please sign in to {account} here.'
# 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/", methods=["GET"])
+ def balance(self, request, account):
+ accounts = self.sinks.truelayer.get_accounts(account)
return dumps(accounts, indent=2)
diff --git a/handler/sinks/truelayer.py b/handler/sinks/truelayer.py
index c140843..d0e3131 100644
--- a/handler/sinks/truelayer.py
+++ b/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:
diff --git a/handler/ux/commands.py b/handler/ux/commands.py
index 2a3df34..71941c3 100644
--- a/handler/ux/commands.py
+++ b/handler/ux/commands.py
@@ -443,36 +443,45 @@ class IRCCommands(object):
class signin(object):
name = "signin"
authed = True
- helptext = "Generate a TrueLayer signin URL."
-
- @staticmethod
- def run(cmd, spl, length, authed, msg, agora, tx, ux):
- auth_url = tx.truelayer.create_auth_url()
- msg(f"Auth URL: {auth_url}")
-
- class accounts(object):
- name = "accounts"
- authed = True
- helptext = "Get a list of acccounts."
-
- @staticmethod
- def run(cmd, spl, length, authed, msg, agora, tx, ux):
- accounts = tx.sinks.truelayer.get_accounts()
- msg(dumps(accounts))
-
- class transactions(object):
- name = "transactions"
- authed = True
- helptext = "Get a list of transactions. Usage: transactions "
+ helptext = "Generate a TrueLayer signin URL. Usage: signin "
@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)
+ 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. Usage: accounts "
+
+ @staticmethod
+ def run(cmd, spl, length, authed, msg, agora, tx, ux):
+ 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 "
+
+ @staticmethod
+ def run(cmd, spl, length, authed, msg, agora, tx, ux):
+ 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}")