From 8c553b92f4b1ab8270523a8be4a8790a1e7fbd44 Mon Sep 17 00:00:00 2001
From: Ashhhleyyy <ash@ashhhleyyy.dev>
Date: Mon, 11 Mar 2024 18:19:17 +0000
Subject: [PATCH] feat(backend): proxy endpoint for checkout.theshrine.net

---
 backend/aci/proxy.py | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/backend/aci/proxy.py b/backend/aci/proxy.py
index 4f6850c..5168188 100644
--- a/backend/aci/proxy.py
+++ b/backend/aci/proxy.py
@@ -13,11 +13,14 @@ router = APIRouter(prefix='/3p-proxy')
 def reformat_reject_date(date: str):
     return format_checkin_date(datetime.strptime(date, '%a %b %d %Y'))
 
+def make_client_session():
+    ssl_context = ssl.create_default_context(cafile=certifi.where())
+    return aiohttp.ClientSession(headers={'User-Agent': 'Mozilla 5.0 (compatible); aci-backend/1.0 (+https://git.ashhhleyyy.dev/ash/aci)'}, connector=aiohttp.TCPConnector(ssl=ssl_context))
+
 @router.get('/reject')
 async def reject_proxy(date: str, time: str, activity: str, space: str):
     """We proxy the rejectdopamine API to maintain user pivacy (the extension does not transmit user IP addresses to third-party services)"""
-    ssl_context = ssl.create_default_context(cafile=certifi.where())
-    async with aiohttp.ClientSession(headers={'User-Agent': 'Mozilla 5.0 (compatible); aci-backend/1.0 (+https://git.ashhhleyyy.dev/ash/aci)'}, connector=aiohttp.TCPConnector(ssl=ssl_context)) as session:
+    async with make_client_session() as session:
         async with session.get('https://rejectdopamine.com/api/app/active/yrk/cs/1') as resp:
             response = await resp.json()
             activities = []
@@ -46,3 +49,25 @@ async def reject_proxy(date: str, time: str, activity: str, space: str):
         'activity': activity,
         'codes': [],
     }
+
+@router.get('/shrine-checkout')
+async def shrine_checkout_proxy(date: str, time: str, activity: str, space: str):
+    """We proxy the checkout.theshrine.net API to maintain user privacy (the extension does not transmit user IP address to third-party services)"""
+    async with make_client_session() as session:
+        async with session.post('https://checkout.theshrine.net/api/activity/ext/id', json={
+                'date': date,
+                'time': time,
+                'activity': activity,
+                'space': space,
+        }) as resp:
+            response = await resp.json()
+            external_id = response['externalId']
+        async with session.get(f'https://checkout.theshrine.net/api/activity/ext/{external_id}/codes') as resp:
+            response = await resp.json()
+            return {
+                'date': date,
+                'time': time,
+                'space': space,
+                'activity': activity,
+                'codes': list(map(lambda c: {'code': c['code'], 'score': 0}, response['codes']))
+            }