Check in backend in working state

packages/sshecret-backend/src/sshecret_backend/api/common.py

@@ -0,0 +1,38 @@
+"""Common helpers."""
+
+import re
+import uuid
+import bcrypt
+
+from sqlmodel import Session, select
+
+from sshecret_backend.models import Client
+
+RE_UUID = re.compile("^[0-9a-f]{8}-[0-9a-f]{4}-[0-5][0-9a-f]{3}-[089ab][0-9a-f]{3}-[0-9a-f]{12}$")
+
+def verify_token(token: str, stored_hash: str) -> bool:
+    """Verify token."""
+    token_bytes = token.encode("utf-8")
+    stored_bytes = stored_hash.encode("utf-8")
+    return bcrypt.checkpw(token_bytes, stored_bytes)
+
+
+async def get_client_by_name(session: Session, name: str) -> Client | None:
+    """Get client by name."""
+    client_filter = select(Client).where(Client.name == name)
+    client_results = session.exec(client_filter)
+    return client_results.first()
+
+async def get_client_by_id(session: Session, id: uuid.UUID) -> Client | None:
+    """Get client by name."""
+    client_filter = select(Client).where(Client.id == id)
+    client_results = session.exec(client_filter)
+    return client_results.first()
+
+async def get_client_by_id_or_name(session: Session, id_or_name: str) -> Client | None:
+    """Get client either by id or name."""
+    if RE_UUID.match(id_or_name):
+        id = uuid.UUID(id_or_name)
+        return await get_client_by_id(session, id)
+
+    return await get_client_by_name(session, id_or_name)