c762564d40
- Fix 71 invalid-syntax files (class-body newline-broken assignments) - Add from/None chain to 307 B904 raise-without-from sites - Add B008 ignore to ruff.toml (already in pyproject.toml) - Noqa F401 on __init__.py re-exports (137 sites) - Noqa E402 on deferred imports (63 sites) - Bulk-add stdlib/FastAPI/project imports for F821 (127 sites) - Replace ×→x, –→-, …→... in docstrings (4093 chars) - Manual refactor of 5 SIM103/SIM116 patterns Tests: 791 passed (66 deselected due to pre-existing Redis issues in test_rag.py) Co-authored-by: opencode <opencode@rugmunch.io>
135 lines
4.1 KiB
Python
135 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Email API Router - Maildir + SMTP access for all RMI email accounts.
|
|
Maildir is mounted from host at /var/mail/vhosts.
|
|
SMTP uses host gateway for outbound delivery.
|
|
"""
|
|
|
|
import email
|
|
import os
|
|
import smtplib
|
|
from email.mime.multipart import MIMEMultipart
|
|
from email.mime.text import MIMEText
|
|
|
|
from fastapi import APIRouter, HTTPException, Query
|
|
from pydantic import BaseModel
|
|
|
|
router = APIRouter(prefix="/email", tags=["email"])
|
|
|
|
SMTP_HOST = os.getenv("SMTP_HOST", "172.20.0.1")
|
|
SMTP_PORT = int(os.getenv("SMTP_PORT", "25"))
|
|
MAILDIR_BASE = "/var/mail/vhosts"
|
|
|
|
ACCOUNTS = {
|
|
"admin": "admin@rugmunch.io",
|
|
"biz": "biz@rugmunch.io",
|
|
"security": "security@rugmunch.io",
|
|
"hello": "hello@cryptorugmunch.com",
|
|
"media": "media@cryptorugmunch.com",
|
|
}
|
|
|
|
|
|
class SendRequest(BaseModel):
|
|
sender_account: str
|
|
to: str
|
|
subject: str
|
|
body: str
|
|
|
|
|
|
def _maildir_for(account: str) -> str:
|
|
addr = ACCOUNTS.get(account)
|
|
if not addr:
|
|
raise HTTPException(404, f"Unknown account: {account}")
|
|
local, domain = addr.split("@")
|
|
return f"{MAILDIR_BASE}/{domain}/{local}"
|
|
|
|
|
|
def _read_msg(filepath: str, msg_id: int, full: bool = False) -> dict:
|
|
with open(filepath, "rb") as f:
|
|
msg = email.message_from_binary_file(f)
|
|
result = {
|
|
"id": msg_id,
|
|
"subject": str(msg.get("Subject", "")),
|
|
"sender": str(msg.get("From", "")),
|
|
"recipient": str(msg.get("To", "")),
|
|
"date": str(msg.get("Date", "")),
|
|
}
|
|
if full:
|
|
body = ""
|
|
if msg.is_multipart():
|
|
for part in msg.walk():
|
|
if part.get_content_type() == "text/plain":
|
|
payload = part.get_payload(decode=True)
|
|
if payload:
|
|
body = payload.decode(errors="replace")
|
|
break
|
|
else:
|
|
payload = msg.get_payload(decode=True)
|
|
if payload:
|
|
body = payload.decode(errors="replace")
|
|
result["body"] = body[:10000]
|
|
return result
|
|
|
|
|
|
def _list_files(maildir: str) -> list:
|
|
files = []
|
|
for sub in ["new", "cur"]:
|
|
d = os.path.join(maildir, sub)
|
|
if os.path.isdir(d):
|
|
files.extend(os.path.join(d, f) for f in os.listdir(d))
|
|
files.sort(key=lambda p: os.path.getmtime(p), reverse=True)
|
|
return files
|
|
|
|
|
|
@router.get("/accounts")
|
|
def list_accounts():
|
|
return {"accounts": dict(ACCOUNTS.items())}
|
|
|
|
|
|
@router.get("/{account}/inbox")
|
|
def list_inbox(account: str, limit: int = Query(20, le=100)):
|
|
if account not in ACCOUNTS:
|
|
raise HTTPException(404, f"Unknown account. Options: {list(ACCOUNTS.keys())}")
|
|
try:
|
|
maildir = _maildir_for(account)
|
|
files = _list_files(maildir)[:limit]
|
|
msgs = [_read_msg(f, i + 1) for i, f in enumerate(files)]
|
|
return {"account": ACCOUNTS[account], "count": len(msgs), "messages": msgs}
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(500, str(e)) from e
|
|
|
|
|
|
@router.get("/{account}/message/{msg_id}")
|
|
def read_message(account: str, msg_id: int):
|
|
if account not in ACCOUNTS:
|
|
raise HTTPException(404, "Unknown account")
|
|
try:
|
|
maildir = _maildir_for(account)
|
|
files = _list_files(maildir)
|
|
if msg_id < 1 or msg_id > len(files):
|
|
raise HTTPException(404, "Message not found")
|
|
return _read_msg(files[msg_id - 1], msg_id, full=True)
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(500, str(e)) from e
|
|
|
|
|
|
@router.post("/send")
|
|
def send_email(req: SendRequest):
|
|
if req.sender_account not in ACCOUNTS:
|
|
raise HTTPException(404, "Unknown sender account")
|
|
sender_addr = ACCOUNTS[req.sender_account]
|
|
msg = MIMEMultipart()
|
|
msg["From"] = sender_addr
|
|
msg["To"] = req.to
|
|
msg["Subject"] = req.subject
|
|
msg.attach(MIMEText(req.body, "plain"))
|
|
try:
|
|
with smtplib.SMTP(SMTP_HOST, SMTP_PORT, timeout=10) as smtp:
|
|
smtp.send_message(msg)
|
|
return {"status": "sent", "from": sender_addr, "to": req.to}
|
|
except Exception as e:
|
|
raise HTTPException(500, f"Send failed: {e}") from e
|