| import json |
| import httpx |
| from config import SUPABASE_URL, SUPABASE_KEY, user_tokens |
|
|
| async def save_token_to_supabase(user_id: str, service: str, token_data: dict): |
| google_services = ["google", "gmail", "calendar", "drive"] |
| ms_services = ["microsoft", "teams", "word", "excel", "powerpoint"] |
| |
| if service in google_services: |
| target_services = google_services |
| elif service in ms_services: |
| target_services = ms_services |
| else: |
| target_services = [service] |
| |
| for s in target_services: |
| user_tokens[(user_id, s)] = token_data |
| try: |
| url = f"{SUPABASE_URL}/rest/v1/user_connectors" |
| headers = { |
| "apikey": SUPABASE_KEY, |
| "Authorization": f"Bearer {SUPABASE_KEY}", |
| "Content-Type": "application/json", |
| "Prefer": "resolution=merge-duplicates" |
| } |
| payload = { |
| "user_id": user_id, |
| "service": s, |
| "token_data": token_data |
| } |
| async with httpx.AsyncClient(timeout=15.0) as c: |
| await c.post(url, headers=headers, json=payload) |
| except Exception: |
| pass |
|
|
| async def get_token_from_supabase(user_id: str, service: str): |
| google_services = ["google", "gmail", "calendar", "drive"] |
| ms_services = ["microsoft", "teams", "word", "excel", "powerpoint"] |
| |
| if service in google_services: |
| check_services = [service] + [x for x in google_services if x != service] |
| elif service in ms_services: |
| check_services = [service] + [x for x in ms_services if x != service] |
| else: |
| check_services = [service] |
|
|
| for s in check_services: |
| if (user_id, s) in user_tokens: |
| return user_tokens[(user_id, s)] |
| try: |
| url = f"{SUPABASE_URL}/rest/v1/user_connectors?user_id=eq.{user_id}&service=eq.{s}&select=token_data" |
| headers = { |
| "apikey": SUPABASE_KEY, |
| "Authorization": f"Bearer {SUPABASE_KEY}" |
| } |
| async with httpx.AsyncClient(timeout=15.0) as c: |
| res = await c.get(url, headers=headers) |
| if res.status_code == 200: |
| data = res.json() |
| if data and len(data) > 0: |
| raw_t = data[0].get("token_data") |
| if isinstance(raw_t, dict): |
| t_data = raw_t |
| elif isinstance(raw_t, str): |
| t_data = json.loads(raw_t) |
| else: |
| t_data = {} |
| if t_data: |
| user_tokens[(user_id, s)] = t_data |
| return t_data |
| except Exception: |
| pass |
| return None |
|
|
| async def delete_token_from_supabase(user_id: str, service: str): |
| google_services = ["google", "gmail", "calendar", "drive"] |
| ms_services = ["microsoft", "teams", "word", "excel", "powerpoint"] |
| |
| if service in google_services: |
| target_services = google_services |
| elif service in ms_services: |
| target_services = ms_services |
| else: |
| target_services = [service] |
|
|
| for s in target_services: |
| user_tokens.pop((user_id, s), None) |
| try: |
| url = f"{SUPABASE_URL}/rest/v1/user_connectors?user_id=eq.{user_id}&service=eq.{s}" |
| headers = { |
| "apikey": SUPABASE_KEY, |
| "Authorization": f"Bearer {SUPABASE_KEY}" |
| } |
| async with httpx.AsyncClient(timeout=15.0) as c: |
| await c.delete(url, headers=headers) |
| except Exception: |
| pass |
|
|
