| import json |
| import datetime |
| import httpx |
| from config import SUPABASE_URL, SUPABASE_KEY, user_tokens |
|
|
| GOOGLE_SERVICES = ["google", "gmail", "calendar", "drive", "sheets", "docs", "slides", "youtube"] |
|
|
| async def save_token_to_supabase(user_id: str, service: str, token_data: dict): |
| target_services = GOOGLE_SERVICES if service in GOOGLE_SERVICES else [service] |
| now_iso = datetime.datetime.now(datetime.timezone.utc).isoformat() |
| |
| 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, |
| "updated_at": now_iso |
| } |
| 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): |
| check_services = [service] + [x for x in GOOGLE_SERVICES if x != service] if service in GOOGLE_SERVICES else [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): |
| target_services = GOOGLE_SERVICES if service in GOOGLE_SERVICES else [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 |
|
|
| async def fetch_session_messages(session_id: str): |
| try: |
| url = f"{SUPABASE_URL}/rest/v1/chat_messages?session_id=eq.{session_id}&order=created_at.asc" |
| headers = { |
| "apikey": SUPABASE_KEY, |
| "Authorization": f"Bearer {SUPABASE_KEY}" |
| } |
| async with httpx.AsyncClient(timeout=20.0) as c: |
| res = await c.get(url, headers=headers) |
| if res.status_code == 200: |
| return res.json() |
| except Exception: |
| pass |
| return [] |
|
|
| async def save_assistant_message_to_db(session_id: str, user_id: str, content: str, thought_reasoning: str = "", sources = None, tool_calls = None): |
| try: |
| url = f"{SUPABASE_URL}/rest/v1/chat_messages" |
| headers = { |
| "apikey": SUPABASE_KEY, |
| "Authorization": f"Bearer {SUPABASE_KEY}", |
| "Content-Type": "application/json", |
| "Prefer": "return=minimal" |
| } |
| payload = { |
| "session_id": session_id, |
| "user_id": user_id, |
| "sender": "assistant", |
| "content": content, |
| "thought_reasoning": thought_reasoning |
| } |
| if sources is not None: |
| payload["sources"] = sources |
| if tool_calls is not None: |
| payload["tool_calls"] = tool_calls |
|
|
| async with httpx.AsyncClient(timeout=20.0) as c: |
| await c.post(url, headers=headers, json=payload) |
| except Exception: |
| pass |
|
|
| async def update_user_message_file_content(session_id: str, file_name: str, file_content: str): |
| try: |
| url = f"{SUPABASE_URL}/rest/v1/chat_messages?session_id=eq.{session_id}&sender=eq.user&file_name=eq.{file_name}&order=created_at.desc&limit=1" |
| headers = { |
| "apikey": SUPABASE_KEY, |
| "Authorization": f"Bearer {SUPABASE_KEY}", |
| "Content-Type": "application/json" |
| } |
| payload = { |
| "file_content": file_content |
| } |
| async with httpx.AsyncClient(timeout=20.0) as c: |
| await c.patch(url, headers=headers, json=payload) |
| except Exception: |
| pass |
|
|
