Initial commit: VoIdeaAI - voice-first AI idea assistant

2026-05-13 12:51:42 +03:00
commit 688d043dad
421 changed files with 47915 additions and 0 deletions
@@ -0,0 +1,135 @@
+"""Google OAuth client with Google Drive API support.
+
+Activated when OAUTH_GOOGLE_ID is set in .env.
+"""
+
+from dataclasses import dataclass
+
+import httpx
+
+from app.core.config import get_settings
+
+GOOGLE_OAUTH_URL = "https://accounts.google.com/o/oauth2/v2/auth"
+GOOGLE_TOKEN_URL = "https://oauth2.googleapis.com/token"
+GOOGLE_USERINFO_URL = "https://www.googleapis.com/oauth2/v2/userinfo"
+GOOGLE_DRIVE_URL = "https://www.googleapis.com/drive/v3"
+
+SCOPES = " ".join([
+    "https://www.googleapis.com/auth/userinfo.email",
+    "https://www.googleapis.com/auth/userinfo.profile",
+    "https://www.googleapis.com/auth/drive.file",
+])
+
+VOIDEA_DRIVE_FOLDER = "VoIdeaAI"
+
+
+@dataclass
+class GoogleUserInfo:
+    id: str
+    email: str
+    display_name: str
+    avatar_url: str | None
+
+
+def is_available() -> bool:
+    return get_settings().google_oauth_enabled
+
+
+async def get_authorize_url() -> str:
+    settings = get_settings()
+    params = {
+        "response_type": "code",
+        "client_id": settings.oauth_google_id,
+        "redirect_uri": settings.oauth_google_redirect_uri,
+        "scope": SCOPES,
+        "access_type": "offline",
+        "prompt": "consent",
+    }
+    query = "&".join(f"{k}={v}" for k, v in params.items())
+    return f"{GOOGLE_OAUTH_URL}?{query}"
+
+
+async def exchange_code(code: str) -> dict | None:
+    settings = get_settings()
+    if not settings.google_oauth_enabled:
+        return None
+    async with httpx.AsyncClient() as client:
+        resp = await client.post(
+            GOOGLE_TOKEN_URL,
+            data={
+                "grant_type": "authorization_code",
+                "code": code,
+                "client_id": settings.oauth_google_id,
+                "client_secret": settings.oauth_google_secret,
+                "redirect_uri": settings.oauth_google_redirect_uri,
+            },
+        )
+        if resp.status_code != 200:
+            return None
+        return resp.json()
+
+
+async def get_user_info(access_token: str) -> GoogleUserInfo | None:
+    async with httpx.AsyncClient() as client:
+        resp = await client.get(
+            GOOGLE_USERINFO_URL,
+            headers={"Authorization": f"Bearer {access_token}"},
+        )
+        if resp.status_code != 200:
+            return None
+        data = resp.json()
+        return GoogleUserInfo(
+            id=data["id"],
+            email=data["email"],
+            display_name=data.get("name", data["email"]),
+            avatar_url=data.get("picture"),
+        )
+
+
+async def ensure_app_folder(access_token: str) -> str | None:
+    async with httpx.AsyncClient() as client:
+        resp = await client.post(
+            f"{GOOGLE_DRIVE_URL}/files",
+            headers={
+                "Authorization": f"Bearer {access_token}",
+                "Content-Type": "application/json",
+            },
+            json={
+                "name": VOIDEA_DRIVE_FOLDER,
+                "mimeType": "application/vnd.google-apps.folder",
+            },
+        )
+        if resp.status_code == 200:
+            return resp.json().get("id")
+        return None
+
+
+async def upload_file(
+    access_token: str,
+    file_name: str,
+    file_content: bytes,
+    parent_id: str | None = None,
+) -> bool:
+    metadata = {"name": file_name}
+    if parent_id:
+        metadata["parents"] = [parent_id]
+
+    async with httpx.AsyncClient() as client:
+        resp = await client.post(
+            f"{GOOGLE_DRIVE_URL}/files?uploadType=multipart",
+            headers={"Authorization": f"Bearer {access_token}"},
+            data={"metadata": str(metadata)},
+            files={"file": (file_name, file_content)},
+        )
+        return resp.status_code == 200
+
+
+async def get_disk_info(access_token: str) -> dict | None:
+    async with httpx.AsyncClient() as client:
+        resp = await client.get(
+            f"{GOOGLE_DRIVE_URL}/about?fields=storageQuota",
+            headers={"Authorization": f"Bearer {access_token}"},
+        )
+        if resp.status_code != 200:
+            return None
+        return resp.json()