108 lines
3.9 KiB
Python
108 lines
3.9 KiB
Python
from dataclasses import dataclass, field
|
|
|
|
from app.services.handoff_email_templates import (
|
|
build_mirror_handoff_html,
|
|
build_mirror_handoff_text,
|
|
build_mirror_update_html,
|
|
build_mirror_update_text,
|
|
format_sender_line,
|
|
)
|
|
from app.services.html_sanitizer import plain_text_to_html, sanitize_email_html
|
|
from app.services.mime_body_extractor import ExtractedMimePart, MirroredMimeContent, extract_mirror_content
|
|
from app.services.raw_mime_storage import get_raw_mime_storage
|
|
|
|
|
|
@dataclass
|
|
class HandoffMirrorPayload:
|
|
body_text: str
|
|
body_html: str | None
|
|
inline_parts: list[ExtractedMimePart] = field(default_factory=list)
|
|
file_attachments: list[ExtractedMimePart] = field(default_factory=list)
|
|
|
|
|
|
async def load_mirror_payload(external_message: dict) -> HandoffMirrorPayload:
|
|
sender = external_message.get("sender") or {}
|
|
sender_line = format_sender_line(sender)
|
|
external_subject = external_message.get("subject", "")
|
|
agent_email = external_message.get("_agent_email", "")
|
|
|
|
body_text = (external_message.get("body_text") or external_message.get("snippet") or "").strip()
|
|
body_html = external_message.get("body_html")
|
|
inline_parts: list[ExtractedMimePart] = []
|
|
file_attachments: list[ExtractedMimePart] = []
|
|
|
|
if external_message.get("raw_mime"):
|
|
try:
|
|
raw = await get_raw_mime_storage().get(external_message["raw_mime"])
|
|
extracted: MirroredMimeContent = extract_mirror_content(raw)
|
|
if extracted.body_text:
|
|
body_text = extracted.body_text
|
|
if extracted.body_html:
|
|
body_html = extracted.body_html
|
|
inline_parts = extracted.inline_parts
|
|
file_attachments = extracted.file_attachments
|
|
except Exception:
|
|
pass
|
|
|
|
if body_html:
|
|
original_html = sanitize_email_html(body_html)
|
|
elif body_text:
|
|
original_html = plain_text_to_html(body_text)
|
|
else:
|
|
original_html = plain_text_to_html("(No message body)")
|
|
|
|
if not body_text:
|
|
body_text = external_message.get("snippet") or "(No message body)"
|
|
|
|
return HandoffMirrorPayload(
|
|
body_text=build_mirror_handoff_text(
|
|
sender_line=sender_line,
|
|
external_subject=external_subject,
|
|
original_body_text=body_text,
|
|
agent_email=agent_email,
|
|
),
|
|
body_html=build_mirror_handoff_html(
|
|
sender_line=sender_line,
|
|
external_subject=external_subject,
|
|
original_html=original_html,
|
|
agent_email=agent_email,
|
|
),
|
|
inline_parts=inline_parts,
|
|
file_attachments=file_attachments,
|
|
)
|
|
|
|
|
|
async def load_mirror_update_payload(
|
|
external_message: dict,
|
|
*,
|
|
agent_email: str,
|
|
) -> HandoffMirrorPayload:
|
|
external_message = {**external_message, "_agent_email": agent_email}
|
|
payload = await load_mirror_payload(external_message)
|
|
sender = external_message.get("sender") or {}
|
|
sender_line = format_sender_line(sender)
|
|
external_subject = external_message.get("subject", "")
|
|
body_text = (external_message.get("body_text") or external_message.get("snippet") or "").strip()
|
|
|
|
body_html = external_message.get("body_html")
|
|
if body_html:
|
|
original_html = sanitize_email_html(body_html)
|
|
else:
|
|
original_html = plain_text_to_html(body_text or "(No message body)")
|
|
|
|
return HandoffMirrorPayload(
|
|
body_text=build_mirror_update_text(
|
|
sender_line=sender_line,
|
|
external_subject=external_subject,
|
|
original_body_text=body_text,
|
|
agent_email=agent_email,
|
|
),
|
|
body_html=build_mirror_update_html(
|
|
sender_line=sender_line,
|
|
external_subject=external_subject,
|
|
original_html=original_html,
|
|
agent_email=agent_email,
|
|
),
|
|
inline_parts=payload.inline_parts,
|
|
file_attachments=payload.file_attachments,
|
|
)
|