Fix issues with multiplatform conversation displays

2026-02-16 11:42:53 +00:00
parent 15af8af6b2
commit 908516bc94
10 changed files with 489 additions and 78 deletions
--- a/core/views/system.py
+++ b/core/views/system.py
@@ -0,0 +1,152 @@
+from django.shortcuts import render
+from django.views import View
+
+from core.models import (
+    AIRequest,
+    AIResult,
+    AIResultSignal,
+    Chat,
+    ChatSession,
+    Group,
+    MemoryItem,
+    Message,
+    MessageEvent,
+    PatternArtifactExport,
+    PatternMitigationAutoSettings,
+    PatternMitigationCorrection,
+    PatternMitigationGame,
+    PatternMitigationMessage,
+    PatternMitigationPlan,
+    PatternMitigationRule,
+    Person,
+    PersonIdentifier,
+    Persona,
+    QueuedMessage,
+    WorkspaceConversation,
+    WorkspaceMetricSnapshot,
+)
+from core.views.manage.permissions import SuperUserRequiredMixin
+
+
+class SystemSettings(SuperUserRequiredMixin, View):
+    template_name = "pages/system-settings.html"
+
+    def _counts(self, user):
+        return {
+            "chat_sessions": ChatSession.objects.filter(user=user).count(),
+            "messages": Message.objects.filter(user=user).count(),
+            "queued_messages": QueuedMessage.objects.filter(user=user).count(),
+            "message_events": MessageEvent.objects.filter(user=user).count(),
+            "workspace_conversations": WorkspaceConversation.objects.filter(user=user).count(),
+            "workspace_snapshots": WorkspaceMetricSnapshot.objects.filter(
+                conversation__user=user
+            ).count(),
+            "ai_requests": AIRequest.objects.filter(user=user).count(),
+            "ai_results": AIResult.objects.filter(user=user).count(),
+            "ai_result_signals": AIResultSignal.objects.filter(user=user).count(),
+            "memory_items": MemoryItem.objects.filter(user=user).count(),
+            "mitigation_plans": PatternMitigationPlan.objects.filter(user=user).count(),
+            "mitigation_rules": PatternMitigationRule.objects.filter(user=user).count(),
+            "mitigation_games": PatternMitigationGame.objects.filter(user=user).count(),
+            "mitigation_corrections": PatternMitigationCorrection.objects.filter(
+                user=user
+            ).count(),
+            "mitigation_messages": PatternMitigationMessage.objects.filter(
+                user=user
+            ).count(),
+            "mitigation_auto_settings": PatternMitigationAutoSettings.objects.filter(
+                user=user
+            ).count(),
+            "mitigation_exports": PatternArtifactExport.objects.filter(user=user).count(),
+            "osint_people": Person.objects.filter(user=user).count(),
+            "osint_identifiers": PersonIdentifier.objects.filter(user=user).count(),
+            "osint_groups": Group.objects.filter(user=user).count(),
+            "osint_personas": Persona.objects.filter(user=user).count(),
+        }
+
+    def _purge_non_osint(self, user):
+        deleted = 0
+        deleted += PatternArtifactExport.objects.filter(user=user).delete()[0]
+        deleted += PatternMitigationMessage.objects.filter(user=user).delete()[0]
+        deleted += PatternMitigationCorrection.objects.filter(user=user).delete()[0]
+        deleted += PatternMitigationGame.objects.filter(user=user).delete()[0]
+        deleted += PatternMitigationRule.objects.filter(user=user).delete()[0]
+        deleted += PatternMitigationAutoSettings.objects.filter(user=user).delete()[0]
+        deleted += PatternMitigationPlan.objects.filter(user=user).delete()[0]
+        deleted += AIResultSignal.objects.filter(user=user).delete()[0]
+        deleted += AIResult.objects.filter(user=user).delete()[0]
+        deleted += AIRequest.objects.filter(user=user).delete()[0]
+        deleted += MemoryItem.objects.filter(user=user).delete()[0]
+        deleted += WorkspaceMetricSnapshot.objects.filter(conversation__user=user).delete()[0]
+        deleted += MessageEvent.objects.filter(user=user).delete()[0]
+        deleted += Message.objects.filter(user=user).delete()[0]
+        deleted += QueuedMessage.objects.filter(user=user).delete()[0]
+        deleted += WorkspaceConversation.objects.filter(user=user).delete()[0]
+        deleted += ChatSession.objects.filter(user=user).delete()[0]
+        # Chat rows are legacy Signal cache rows and are not user-scoped.
+        deleted += Chat.objects.all().delete()[0]
+        return deleted
+
+    def _purge_osint_people(self, user):
+        return Person.objects.filter(user=user).delete()[0]
+
+    def _purge_osint_identifiers(self, user):
+        return PersonIdentifier.objects.filter(user=user).delete()[0]
+
+    def _purge_osint_groups(self, user):
+        return Group.objects.filter(user=user).delete()[0]
+
+    def _purge_osint_personas(self, user):
+        return Persona.objects.filter(user=user).delete()[0]
+
+    def _handle_action(self, request):
+        action = str(request.POST.get("action") or "").strip().lower()
+        if action == "purge_non_osint":
+            return (
+                "success",
+                f"Purged {self._purge_non_osint(request.user)} non-OSINT row(s).",
+            )
+        if action == "purge_osint_people":
+            return (
+                "warning",
+                f"Purged {self._purge_osint_people(request.user)} OSINT people row(s).",
+            )
+        if action == "purge_osint_identifiers":
+            return (
+                "warning",
+                f"Purged {self._purge_osint_identifiers(request.user)} OSINT identifier row(s).",
+            )
+        if action == "purge_osint_groups":
+            return (
+                "warning",
+                f"Purged {self._purge_osint_groups(request.user)} OSINT group row(s).",
+            )
+        if action == "purge_osint_personas":
+            return (
+                "warning",
+                f"Purged {self._purge_osint_personas(request.user)} OSINT persona row(s).",
+            )
+        return ("danger", "Unknown action.")
+
+    def get(self, request):
+        return render(
+            request,
+            self.template_name,
+            {
+                "counts": self._counts(request.user),
+                "notice_level": "",
+                "notice_message": "",
+            },
+        )
+
+    def post(self, request):
+        notice_level, notice_message = self._handle_action(request)
+        return render(
+            request,
+            self.template_name,
+            {
+                "counts": self._counts(request.user),
+                "notice_level": notice_level,
+                "notice_message": notice_message,
+            },
+        )