Implement attachment view

core/realtime/typing_state.py

@@ -0,0 +1,74 @@
+import time
+
+from django.core.cache import cache
+
+TYPING_TTL_SECONDS = 12
+
+
+def _person_key(user_id, person_id):
+    return f"compose:typing:user:{int(user_id)}:person:{int(person_id)}"
+
+
+def set_person_typing_state(
+    *,
+    user_id,
+    person_id,
+    started,
+    source_service="",
+    display_name="",
+):
+    if not user_id or not person_id:
+        return
+    now_ms = int(time.time() * 1000)
+    state = {
+        "typing": bool(started),
+        "source_service": str(source_service or ""),
+        "display_name": str(display_name or ""),
+        "updated_ts": now_ms,
+        "expires_ts": (
+            now_ms + (TYPING_TTL_SECONDS * 1000) if started else now_ms
+        ),
+    }
+    cache.set(
+        _person_key(user_id, person_id),
+        state,
+        timeout=max(TYPING_TTL_SECONDS * 2, 30),
+    )
+
+
+def get_person_typing_state(*, user_id, person_id):
+    if not user_id or not person_id:
+        return {
+            "typing": False,
+            "source_service": "",
+            "display_name": "",
+            "updated_ts": 0,
+            "expires_ts": 0,
+        }
+
+    key = _person_key(user_id, person_id)
+    state = dict(cache.get(key) or {})
+    if not state:
+        return {
+            "typing": False,
+            "source_service": "",
+            "display_name": "",
+            "updated_ts": 0,
+            "expires_ts": 0,
+        }
+
+    now_ms = int(time.time() * 1000)
+    is_typing = bool(state.get("typing"))
+    expires_ts = int(state.get("expires_ts") or 0)
+    if is_typing and expires_ts and now_ms > expires_ts:
+        state["typing"] = False
+        state["updated_ts"] = now_ms
+        cache.set(key, state, timeout=max(TYPING_TTL_SECONDS * 2, 30))
+
+    return {
+        "typing": bool(state.get("typing")),
+        "source_service": str(state.get("source_service") or ""),
+        "display_name": str(state.get("display_name") or ""),
+        "updated_ts": int(state.get("updated_ts") or 0),
+        "expires_ts": int(state.get("expires_ts") or 0),
+    }