Files
replypier/backend/app/services/notifications.py
T
WangShen007 20fef2b9f3 chore: settle product naming across app, widget and deployments
One word everywhere: ReplyPier in UI copy, package and image names, the
replypier.site domains (app/api/docs/help), REPLYPIER_* settings and the
--rp-* design tokens.

- public widget API is window.ReplyPier with replypierId / replypierBaseUrl /
  replypierConfig / replypierToken, loader at webclient/replypier.min.js;
  both bundles rebuilt from source
- NOTICE, llms.txt, CONTRIBUTING, HELP_CENTER_INFRA and the demo runtime
  config rewritten; the demo config ships no third-party keys
- migration adds the ReplyPier managed model value to the AI model enum
- local compose project, database, role, volume and Docker network carry the
  same name
2026-10-26 00:57:22 +08:00

115 lines
3.5 KiB
Python

"""
Copyright 2024-2026 ReplyPier
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
import enum
from typing import Iterable, Optional
from sqlalchemy.orm import Session
from app.core.logger import get_logger
from app.models.notification import Notification, NotificationType
from app.models.notification_settings import (
NOTIFY_CHAT_ASSIGNED,
NOTIFY_CHAT_TRANSFER,
NOTIFY_NEW_CHAT,
)
from app.repositories.notification_settings import UserNotificationSettingsRepository
from app.services.user import send_fcm_notification
logger = get_logger(__name__)
class ChatNotificationEvent(str, enum.Enum):
"""Chat events a user can mute. The value is both the preference column
and the `event` discriminator carried in the push metadata.
"""
NEW_CHAT = NOTIFY_NEW_CHAT
CHAT_TRANSFER = NOTIFY_CHAT_TRANSFER
CHAT_ASSIGNED = NOTIFY_CHAT_ASSIGNED
async def notify_user(
db: Session,
user_id,
type_: NotificationType,
title: str,
message: str,
metadata: Optional[dict] = None,
) -> None:
"""Persist an in-app notification and send its FCM push.
Never raises and no-ops without a user (background jobs may have none) —
a notification failure must not fail the work it reports on.
NOTE: this commits (and, on error, rolls back) the caller's `db` session,
so call it only once any work you want persisted is already committed.
Every current caller either commits first or has nothing pending.
"""
if not user_id:
return
try:
notification = Notification(
user_id=user_id,
type=type_,
title=title,
message=message,
notification_metadata=metadata,
)
db.add(notification)
db.commit()
await send_fcm_notification(user_id, notification, db)
except Exception as e:
logger.error(f"Error sending notification '{title}' to user {user_id}: {e}")
try:
db.rollback()
except Exception:
pass
async def notify_chat_event(
db: Session,
user_ids: Iterable,
event: ChatNotificationEvent,
title: str,
message: str,
metadata: Optional[dict] = None,
) -> None:
"""Notify every user in `user_ids` who hasn't muted `event`.
Recipients are resolved in a single query, so fanning out to a whole group
or org costs one lookup rather than one per user. Like notify_user, this
never raises — a notification must not fail the chat it reports on.
"""
try:
recipients = UserNotificationSettingsRepository(db).filter_enabled(
user_ids, event.value
)
if not recipients:
return
payload = {**(metadata or {}), "event": event.value}
for user_id in recipients:
await notify_user(
db=db,
user_id=user_id,
type_=NotificationType.CHAT,
title=title,
message=message,
metadata=payload,
)
except Exception as e:
logger.error(f"Error sending '{event.value}' notification: {e}")