feat(infra): add persistence and SMS adapters

2026-07-14 09:32:58 +03:30
parent 4fc0ee1107
commit 61f7b5c322
7 changed files with 440 additions and 0 deletions
--- a/src/gapido_auth/infrastructure/mongo_repositories.py
+++ b/src/gapido_auth/infrastructure/mongo_repositories.py
@@ -0,0 +1,135 @@
+from datetime import UTC, datetime
+from typing import Any
+
+from bson import ObjectId
+from motor.motor_asyncio import AsyncIOMotorDatabase
+from pymongo import ASCENDING
+
+from gapido_auth.domain.entities import RefreshSession, Role, User
+from gapido_auth.domain.ports import RefreshSessionRepository, UserRepository
+
+
+def _now() -> datetime:
+    return datetime.now(UTC)
+
+
+def _user_from_doc(doc: dict[str, Any]) -> User:
+    return User(
+        id=str(doc["_id"]),
+        mobile=str(doc["mobile"]),
+        role=Role(str(doc["role"])),
+        is_active=bool(doc["is_active"]),
+        created_at=doc["created_at"],
+        updated_at=doc["updated_at"],
+    )
+
+
+def _session_from_doc(doc: dict[str, Any]) -> RefreshSession:
+    return RefreshSession(
+        id=str(doc["_id"]),
+        user_id=str(doc["user_id"]),
+        token_hash=str(doc["token_hash"]),
+        expires_at=doc["expires_at"],
+        revoked_at=doc.get("revoked_at"),
+        replaced_by_hash=doc.get("replaced_by_hash"),
+        created_at=doc["created_at"],
+    )
+
+
+class MongoUserRepository(UserRepository):
+    def __init__(self, db: AsyncIOMotorDatabase[Any]) -> None:
+        self._collection = db.users
+
+    async def create_indexes(self) -> None:
+        await self._collection.create_index([("mobile", ASCENDING)], unique=True)
+        await self._collection.create_index([("role", ASCENDING)])
+
+    async def get_by_id(self, user_id: str) -> User | None:
+        if not ObjectId.is_valid(user_id):
+            return None
+        doc = await self._collection.find_one({"_id": ObjectId(user_id)})
+        return _user_from_doc(doc) if doc else None
+
+    async def get_by_mobile(self, mobile: str) -> User | None:
+        doc = await self._collection.find_one({"mobile": mobile})
+        return _user_from_doc(doc) if doc else None
+
+    async def get_or_create_user(self, mobile: str, role: Role = Role.USER) -> User:
+        now = _now()
+        await self._collection.update_one(
+            {"mobile": mobile},
+            {
+                "$setOnInsert": {
+                    "mobile": mobile,
+                    "role": role.value,
+                    "is_active": True,
+                    "created_at": now,
+                },
+                "$set": {"updated_at": now},
+            },
+            upsert=True,
+        )
+        user = await self.get_by_mobile(mobile)
+        if user is None:
+            raise RuntimeError("failed to create user")
+        return user
+
+    async def ensure_admin_user(self, mobile: str) -> User:
+        now = _now()
+        await self._collection.update_one(
+            {"mobile": mobile},
+            {
+                "$set": {"role": Role.ADMIN.value, "is_active": True, "updated_at": now},
+                "$setOnInsert": {"mobile": mobile, "created_at": now},
+            },
+            upsert=True,
+        )
+        user = await self.get_by_mobile(mobile)
+        if user is None:
+            raise RuntimeError("failed to seed admin user")
+        return user
+
+
+class MongoRefreshSessionRepository(RefreshSessionRepository):
+    def __init__(self, db: AsyncIOMotorDatabase[Any]) -> None:
+        self._collection = db.refresh_sessions
+
+    async def create_indexes(self) -> None:
+        await self._collection.create_index([("token_hash", ASCENDING)], unique=True)
+        await self._collection.create_index([("user_id", ASCENDING), ("expires_at", ASCENDING)])
+        await self._collection.create_index([("revoked_at", ASCENDING)])
+
+    async def create(self, user_id: str, token_hash: str, expires_at: datetime) -> RefreshSession:
+        now = _now()
+        result = await self._collection.insert_one(
+            {
+                "user_id": user_id,
+                "token_hash": token_hash,
+                "expires_at": expires_at,
+                "revoked_at": None,
+                "replaced_by_hash": None,
+                "created_at": now,
+            }
+        )
+        doc = await self._collection.find_one({"_id": result.inserted_id})
+        if doc is None:
+            raise RuntimeError("failed to create refresh session")
+        return _session_from_doc(doc)
+
+    async def get_active_by_hash(self, token_hash: str, now: datetime) -> RefreshSession | None:
+        doc = await self._collection.find_one(
+            {"token_hash": token_hash, "revoked_at": None, "expires_at": {"$gt": now}}
+        )
+        return _session_from_doc(doc) if doc else None
+
+    async def revoke(
+        self, token_hash: str, now: datetime, replaced_by_hash: str | None = None
+    ) -> None:
+        update: dict[str, Any] = {"revoked_at": now}
+        if replaced_by_hash is not None:
+            update["replaced_by_hash"] = replaced_by_hash
+        await self._collection.update_one(
+            {"token_hash": token_hash, "revoked_at": None},
+            {"$set": update},
+        )
+