enh: channel notification

backend/open_webui/routers/channels.py

@@ -3,11 +3,11 @@ import logging
 from typing import Optional
 
 
-from fastapi import APIRouter, Depends, HTTPException, Request, status
+from fastapi import APIRouter, Depends, HTTPException, Request, status, BackgroundTasks
 from pydantic import BaseModel
 
 
-from open_webui.socket.main import sio
+from open_webui.socket.main import sio, SESSION_POOL
 from open_webui.models.users import Users, UserNameResponse
 
 from open_webui.models.channels import Channels, ChannelModel, ChannelForm
@@ -16,11 +16,12 @@ from open_webui.models.messages import Messages, MessageModel, MessageForm
 
 from open_webui.config import ENABLE_ADMIN_CHAT_ACCESS, ENABLE_ADMIN_EXPORT
 from open_webui.constants import ERROR_MESSAGES
-from open_webui.env import SRC_LOG_LEVELS
+from open_webui.env import SRC_LOG_LEVELS, WEBUI_URL
 
 
 from open_webui.utils.auth import get_admin_user, get_verified_user
-from open_webui.utils.access_control import has_access
+from open_webui.utils.access_control import has_access, get_users_with_access
+from open_webui.utils.webhook import post_webhook
 
 log = logging.getLogger(__name__)
 log.setLevel(SRC_LOG_LEVELS["MODELS"])
@@ -180,9 +181,39 @@ async def get_channel_messages(
 ############################
 
 
+async def send_notification(channel, message, active_user_ids):
+
+    print(f"Sending notification to {channel=}, {message=}, {active_user_ids=}")
+    users = get_users_with_access("read", channel.access_control)
+
+    for user in users:
+        if user.id in active_user_ids:
+            continue
+        else:
+            if user.settings:
+                webhook_url = user.settings.ui.get("notifications", {}).get(
+                    "webhook_url", None
+                )
+
+                if webhook_url:
+                    post_webhook(
+                        webhook_url,
+                        f"#{channel.name} - {WEBUI_URL}/c/{channel.id}\n\n{message.content}",
+                        {
+                            "action": "channel",
+                            "message": message.content,
+                            "title": channel.name,
+                            "url": f"{WEBUI_URL}/c/{channel.id}",
+                        },
+                    )
+
+
 @router.post("/{id}/messages/post", response_model=Optional[MessageModel])
 async def post_new_message(
-    id: str, form_data: MessageForm, user=Depends(get_verified_user)
+    id: str,
+    form_data: MessageForm,
+    background_tasks: BackgroundTasks,
+    user=Depends(get_verified_user),
 ):
     channel = Channels.get_channel_by_id(id)
     if not channel:
@@ -201,24 +232,44 @@ async def post_new_message(
         message = Messages.insert_new_message(form_data, channel.id, user.id)
 
         if message:
+            event_data = {
+                "channel_id": channel.id,
+                "message_id": message.id,
+                "data": {
+                    "type": "message",
+                    "data": {
+                        **message.model_dump(),
+                        "user": UserNameResponse(**user.model_dump()).model_dump(),
+                    },
+                },
+                "user": UserNameResponse(**user.model_dump()).model_dump(),
+                "channel": channel.model_dump(),
+            }
+
             await sio.emit(
                 "channel-events",
-                {
-                    "channel_id": channel.id,
-                    "message_id": message.id,
-                    "data": {
-                        "type": "message",
-                        "data": {
-                            **message.model_dump(),
-                            "user": UserNameResponse(**user.model_dump()).model_dump(),
-                        },
-                    },
-                    "user": UserNameResponse(**user.model_dump()).model_dump(),
-                    "channel": channel.model_dump(),
-                },
+                event_data,
                 to=f"channel:{channel.id}",
             )
 
+            active_session_ids = sio.manager.get_participants(
+                namespace="/",
+                room=f"channel:{channel.id}",
+            )
+
+            active_user_ids = list(
+                set(
+                    [
+                        SESSION_POOL.get(session_id[0])
+                        for session_id in active_session_ids
+                    ]
+                )
+            )
+
+            background_tasks.add_task(
+                send_notification, channel, message, active_user_ids
+            )
+
         return MessageModel(**message.model_dump())
     except Exception as e:
         log.exception(e)