95 lines
3.9 KiB
Python
95 lines
3.9 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.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from datetime import datetime, timezone
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
|
||
|
|
from app.core.auth import has_any_permission
|
||
|
|
from app.core.socketio import sio
|
||
|
|
from app.repositories.user import UserRepository
|
||
|
|
from app.services.chat_notifications import UNASSIGNED_CHAT_VIEWERS
|
||
|
|
|
||
|
|
# Per-organization room every inbox page joins (see handle_join_room), for
|
||
|
|
# updates about any of the org's conversations — including AI-owned ones,
|
||
|
|
# which have no handling agent to address.
|
||
|
|
ORG_INBOX_ROOM_PREFIX = "org_inbox_"
|
||
|
|
|
||
|
|
|
||
|
|
# chat_reply `type`s the inbox appends to a conversation
|
||
|
|
USER_MESSAGE = 'user_message'
|
||
|
|
BOT_MESSAGE = 'bot'
|
||
|
|
|
||
|
|
|
||
|
|
def org_inbox_room(organization_id) -> str:
|
||
|
|
return f"{ORG_INBOX_ROOM_PREFIX}{organization_id}"
|
||
|
|
|
||
|
|
|
||
|
|
def is_org_inbox_room(room: str) -> bool:
|
||
|
|
return room.startswith(ORG_INBOX_ROOM_PREFIX)
|
||
|
|
|
||
|
|
|
||
|
|
def may_join_org_inbox(db: Session, socket_session: dict, room: str) -> bool:
|
||
|
|
"""Whether a dashboard socket may join an org inbox room: its own
|
||
|
|
organization's, and only for people who can see chats nobody owns."""
|
||
|
|
if str(socket_session.get('organization_id')) != room.removeprefix(ORG_INBOX_ROOM_PREFIX):
|
||
|
|
return False
|
||
|
|
user_id = socket_session.get('user_id')
|
||
|
|
user = UserRepository(db).get_user(user_id) if user_id else None
|
||
|
|
return (user is not None and user.is_active
|
||
|
|
and has_any_permission(user, UNASSIGNED_CHAT_VIEWERS))
|
||
|
|
|
||
|
|
|
||
|
|
async def emit_to_session_agents(event: str, payload: dict, session_id,
|
||
|
|
user_id: Optional[object] = None) -> None:
|
||
|
|
"""Push a live update to the dashboard for one conversation.
|
||
|
|
|
||
|
|
The inbox listens on the handling agent's personal room (user_<id>); the
|
||
|
|
session room reaches anyone who joined that conversation explicitly.
|
||
|
|
"""
|
||
|
|
if user_id is not None:
|
||
|
|
await sio.emit(event, payload, room=f"user_{user_id}", namespace='/agent')
|
||
|
|
await sio.emit(event, payload, room=str(session_id), namespace='/agent')
|
||
|
|
|
||
|
|
|
||
|
|
async def emit_to_org_inbox(event: str, payload: dict, organization_id,
|
||
|
|
user_id: Optional[object] = None) -> None:
|
||
|
|
"""Push a live update to every inbox open on the organization, and to the
|
||
|
|
handling agent's personal room (where a chat they own is listening)."""
|
||
|
|
await sio.emit(event, payload, room=org_inbox_room(organization_id), namespace='/agent')
|
||
|
|
if user_id is not None:
|
||
|
|
await sio.emit(event, payload, room=f"user_{user_id}", namespace='/agent')
|
||
|
|
|
||
|
|
|
||
|
|
async def relay_stored_message(session_record, stored, message_type: str) -> None:
|
||
|
|
"""Push a stored message into the dashboard conversation live, the way
|
||
|
|
the widget's chat_reply events do — a customer message, or one sent from
|
||
|
|
outside the inbox (e.g. by an external app) into a chat a human holds."""
|
||
|
|
session_id = str(session_record.session_id)
|
||
|
|
payload = {
|
||
|
|
'message': stored.message,
|
||
|
|
'message_id': stored.id,
|
||
|
|
'type': message_type,
|
||
|
|
'transfer_to_human': False,
|
||
|
|
'session_id': session_id,
|
||
|
|
# The inbox appends the message live off this field; without it the
|
||
|
|
# handler builds an invalid Date and the message only shows on refetch.
|
||
|
|
'created_at': (stored.created_at or datetime.now(timezone.utc)).isoformat(),
|
||
|
|
# Carries e.g. the button-reply marker, so tags show live too
|
||
|
|
'attributes': stored.attributes or {},
|
||
|
|
}
|
||
|
|
await emit_to_session_agents('chat_reply', payload, session_id, session_record.user_id)
|