Spaces:
Running
Running
| from __future__ import annotations | |
| from fastapi import APIRouter | |
| from ..services.auth import storage as auth_store | |
| from ..services.drive import service as drive_svc | |
| from ..services.history import storage as hist | |
| router = APIRouter(prefix="/api/drive", tags=["drive"]) | |
| def drive_status(username: str = ""): | |
| configured = drive_svc.is_configured() | |
| if not username: | |
| return {"ok": True, "configured": configured, "linked": False} | |
| fid, link = auth_store.get_drive_folder(username) | |
| return { | |
| "ok": True, | |
| "configured": configured, | |
| "linked": bool(fid), | |
| "folder_id": fid or "", | |
| } | |
| def drive_setup(body: dict): | |
| username = (body.get("username") or "").strip() | |
| folder_link = (body.get("folder_link") or "").strip() | |
| if not username: | |
| return {"ok": False, "error": "Missing username"} | |
| if not folder_link: | |
| return {"ok": False, "error": "Missing folder_link"} | |
| if not drive_svc.is_configured(): | |
| return {"ok": False, "error": "Google Drive chưa được cấu hình trên máy chủ"} | |
| folder_id = drive_svc.extract_folder_id(folder_link) | |
| if not folder_id: | |
| return {"ok": False, "error": "Link Google Drive không hợp lệ"} | |
| # Save to persistent auth storage (PostgreSQL / JSON file) | |
| auth_store.save_drive_folder(username, folder_id, folder_link) | |
| # Also keep history storage in sync for Drive backup mechanism | |
| hist.save_drive_folder(username, folder_id, folder_link) | |
| # Restore history from Drive backup (if exists) | |
| restored = hist.restore_from_drive(username, drive_svc) | |
| return {"ok": True, "folder_id": folder_id, "history_restored": restored} | |
| def drive_remove(username: str = ""): | |
| if not username: | |
| return {"ok": False, "error": "Missing username"} | |
| auth_store.save_drive_folder(username, "", "") | |
| hist.save_drive_folder(username, "", "") | |
| return {"ok": True} | |