opencode-semantic-memory 0.1.0__py3-none-any.whl

opencode_memory/session/__init__.py

@@ -0,0 +1,5 @@
+"""Session management modules for opencode-memory."""
+
+from opencode_memory.session.registry import SessionRegistry
+
+__all__ = ["SessionRegistry"]

opencode_memory/session/registry.py

@@ -0,0 +1,57 @@
+"""Session registry for tracking active OpenCode sessions."""
+
+from datetime import UTC, datetime
+
+from opencode_memory.models import Session
+from opencode_memory.storage.sqlite import SQLiteStorage
+
+
+class SessionRegistry:
+    """Manage active sessions and their claims."""
+
+    def __init__(self, storage: SQLiteStorage):
+        self.storage = storage
+
+    def register(self, session_id: str, working_on: str | None = None) -> Session:
+        """Register a new session."""
+        session = Session(
+            id=session_id,
+            started_at=datetime.now(UTC),
+            last_heartbeat=datetime.now(UTC),
+            working_on=working_on,
+        )
+        self.storage.upsert_session(session)
+        return session
+
+    def heartbeat(self, session_id: str) -> None:
+        """Update session heartbeat."""
+        sessions = self.storage.get_active_sessions(stale_minutes=60 * 24)
+        for s in sessions:
+            if s.id == session_id:
+                s.last_heartbeat = datetime.now(UTC)
+                self.storage.upsert_session(s)
+                return
+
+    def claim_item(self, session_id: str, item_ref: str) -> bool:
+        """Claim an item for a session. Returns False if already claimed.
+
+        Uses atomic database operation to prevent race conditions.
+        """
+        success, _ = self.storage.atomic_claim_item(session_id, item_ref)
+        return success
+
+    def release_item(self, session_id: str, item_ref: str) -> None:
+        """Release a claimed item."""
+        self.storage.atomic_release_item(session_id, item_ref)
+
+    def get_active_sessions(self) -> list[Session]:
+        """Get all active sessions."""
+        return self.storage.get_active_sessions()
+
+    def unregister(self, session_id: str) -> None:
+        """Remove a session."""
+        self.storage.delete_session(session_id)
+
+    def get_item_owner(self, item_ref: str) -> str | None:
+        """Get the session that owns an item."""
+        return self.storage.get_item_owner(item_ref)

opencode_memory/storage/__init__.py

@@ -0,0 +1,6 @@
+"""Storage modules for opencode-memory."""
+
+from opencode_memory.storage.sqlite import SQLiteStorage
+from opencode_memory.storage.vectors import VectorStorage
+
+__all__ = ["SQLiteStorage", "VectorStorage"]