Feature: mcp server

src/paperless_mail/mcp.py

@@ -0,0 +1,129 @@
+from mcp_server import ModelQueryToolset
+from mcp_server import drf_publish_create_mcp_tool
+from mcp_server import drf_publish_destroy_mcp_tool
+from mcp_server import drf_publish_list_mcp_tool
+from mcp_server import drf_publish_update_mcp_tool
+
+from documents.permissions import get_objects_for_user_owner_aware
+from paperless_mail.models import MailAccount
+from paperless_mail.models import MailRule
+from paperless_mail.models import ProcessedMail
+from paperless_mail.views import MailAccountViewSet
+from paperless_mail.views import MailRuleViewSet
+from paperless_mail.views import ProcessedMailViewSet
+
+VIEWSET_ACTIONS = {
+    "create": {"post": "create"},
+    "list": {"get": "list"},
+    "update": {"put": "update"},
+    "destroy": {"delete": "destroy"},
+}
+
+BODY_SCHEMA = {"type": "object", "additionalProperties": True}
+
+VIEWSET_INSTRUCTIONS = {
+    MailAccountViewSet: "Manage mail accounts.",
+    MailRuleViewSet: "Manage mail rules.",
+    ProcessedMailViewSet: "List processed mail.",
+}
+
+
+class MailAccountQueryToolset(ModelQueryToolset):
+    model = MailAccount
+
+    def get_queryset(self):
+        user = getattr(self.request, "user", None)
+        if not user or not user.is_authenticated:
+            return MailAccount.objects.none()
+        if user.is_superuser:
+            return MailAccount.objects.all()
+        return get_objects_for_user_owner_aware(
+            user,
+            "paperless_mail.view_mailaccount",
+            MailAccount,
+        )
+
+
+class MailRuleQueryToolset(ModelQueryToolset):
+    model = MailRule
+
+    def get_queryset(self):
+        user = getattr(self.request, "user", None)
+        if not user or not user.is_authenticated:
+            return MailRule.objects.none()
+        if user.is_superuser:
+            return MailRule.objects.all()
+        return get_objects_for_user_owner_aware(
+            user,
+            "paperless_mail.view_mailrule",
+            MailRule,
+        )
+
+
+class ProcessedMailQueryToolset(ModelQueryToolset):
+    model = ProcessedMail
+
+    def get_queryset(self):
+        user = getattr(self.request, "user", None)
+        if not user or not user.is_authenticated:
+            return ProcessedMail.objects.none()
+        if user.is_superuser:
+            return ProcessedMail.objects.all()
+        return get_objects_for_user_owner_aware(
+            user,
+            "paperless_mail.view_processedmail",
+            ProcessedMail,
+        )
+
+
+drf_publish_create_mcp_tool(
+    MailAccountViewSet,
+    actions=VIEWSET_ACTIONS["create"],
+    instructions=VIEWSET_INSTRUCTIONS[MailAccountViewSet],
+    body_schema=BODY_SCHEMA,
+)
+drf_publish_list_mcp_tool(
+    MailAccountViewSet,
+    actions=VIEWSET_ACTIONS["list"],
+    instructions=VIEWSET_INSTRUCTIONS[MailAccountViewSet],
+)
+drf_publish_update_mcp_tool(
+    MailAccountViewSet,
+    actions=VIEWSET_ACTIONS["update"],
+    instructions=VIEWSET_INSTRUCTIONS[MailAccountViewSet],
+    body_schema=BODY_SCHEMA,
+)
+drf_publish_destroy_mcp_tool(
+    MailAccountViewSet,
+    actions=VIEWSET_ACTIONS["destroy"],
+    instructions=VIEWSET_INSTRUCTIONS[MailAccountViewSet],
+)
+
+drf_publish_create_mcp_tool(
+    MailRuleViewSet,
+    actions=VIEWSET_ACTIONS["create"],
+    instructions=VIEWSET_INSTRUCTIONS[MailRuleViewSet],
+    body_schema=BODY_SCHEMA,
+)
+drf_publish_list_mcp_tool(
+    MailRuleViewSet,
+    actions=VIEWSET_ACTIONS["list"],
+    instructions=VIEWSET_INSTRUCTIONS[MailRuleViewSet],
+)
+drf_publish_update_mcp_tool(
+    MailRuleViewSet,
+    actions=VIEWSET_ACTIONS["update"],
+    instructions=VIEWSET_INSTRUCTIONS[MailRuleViewSet],
+    body_schema=BODY_SCHEMA,
+)
+drf_publish_destroy_mcp_tool(
+    MailRuleViewSet,
+    actions=VIEWSET_ACTIONS["destroy"],
+    instructions=VIEWSET_INSTRUCTIONS[MailRuleViewSet],
+)
+
+drf_publish_list_mcp_tool(
+    ProcessedMailViewSet,
+    actions=VIEWSET_ACTIONS["list"],
+    instructions=VIEWSET_INSTRUCTIONS[ProcessedMailViewSet],
+)