notifications.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691
  1. """API routes for notification providers."""
  2. import json
  3. import logging
  4. import time
  5. from collections import defaultdict, deque
  6. from datetime import datetime, timedelta, timezone
  7. from fastapi import APIRouter, Depends, HTTPException, Query
  8. from fastapi.responses import FileResponse
  9. from sqlalchemy import delete, desc, func, select, update
  10. from sqlalchemy.ext.asyncio import AsyncSession
  11. from backend.app.core.auth import RequirePermissionIfAuthEnabled, ScopedCaller, require_notification_send
  12. from backend.app.core.database import get_db
  13. from backend.app.core.permissions import Permission
  14. from backend.app.models.notification import NotificationLog, NotificationProvider
  15. from backend.app.models.notification_lock_screen_widget import NotificationLockScreenWidget
  16. from backend.app.models.user import User
  17. from backend.app.schemas.notification import (
  18. AppMessage,
  19. AppMessageChannel,
  20. AppMessageResult,
  21. NotificationLogResponse,
  22. NotificationLogStats,
  23. NotificationProviderCreate,
  24. NotificationProviderResponse,
  25. NotificationProviderUpdate,
  26. NotificationTestRequest,
  27. NotificationTestResponse,
  28. )
  29. from backend.app.services.notification_service import notification_service
  30. from backend.app.services.notify_client import NotifyError, notify_credentials
  31. from backend.app.services.notify_live_activities import notify_live_activities
  32. from backend.app.services.notify_widgets import notify_widgets
  33. from backend.app.services.telegram_reactions import telegram_reaction_poller
  34. from backend.app.utils.notification_photos import find_notification_photo
  35. logger = logging.getLogger(__name__)
  36. router = APIRouter(prefix="/notifications", tags=["notifications"])
  37. def _notify_providers_changed() -> None:
  38. """Wake optional workers after a provider edit, without waiting for their HTTP work."""
  39. for worker in (notify_live_activities, notify_widgets):
  40. try:
  41. worker.providers_changed()
  42. except Exception:
  43. logger.exception("Could not refresh Notify worker configuration")
  44. async def _resync_reaction_poller():
  45. """Start/stop Telegram reaction polls after a provider changed (#3046).
  46. Never fails the request: the provider row is already saved, and the
  47. poller catches up on the next restart at worst.
  48. """
  49. try:
  50. await telegram_reaction_poller.sync()
  51. except Exception as e:
  52. logger.warning("Telegram reaction poller resync failed: %s", e)
  53. def _provider_to_dict(provider: NotificationProvider) -> dict:
  54. """Convert a NotificationProvider model to a response dictionary."""
  55. return {
  56. "id": provider.id,
  57. "name": provider.name,
  58. "provider_type": provider.provider_type,
  59. "enabled": provider.enabled,
  60. "config": json.loads(provider.config) if isinstance(provider.config, str) else provider.config,
  61. "attach_photo": provider.attach_photo,
  62. # Print lifecycle events
  63. "on_print_start": provider.on_print_start,
  64. "on_print_complete": provider.on_print_complete,
  65. "on_print_failed": provider.on_print_failed,
  66. "on_print_stopped": provider.on_print_stopped,
  67. "on_print_progress": provider.on_print_progress,
  68. "on_print_missing_spool_assignment": provider.on_print_missing_spool_assignment,
  69. "on_billing_charge_failed": provider.on_billing_charge_failed,
  70. # Printer status events
  71. "on_printer_offline": provider.on_printer_offline,
  72. "on_printer_error": provider.on_printer_error,
  73. "on_ai_failure_detection": provider.on_ai_failure_detection,
  74. "on_filament_low": provider.on_filament_low,
  75. "on_maintenance_due": provider.on_maintenance_due,
  76. # Home Assistant sensor alerts (#1148, #2824). Both directions of this
  77. # file are hand-maintained field maps, so a column missing here reads
  78. # back as the schema default (False) no matter what the row holds.
  79. "on_ha_sensor_alert": provider.on_ha_sensor_alert,
  80. "on_location_ha_sensor_alert": provider.on_location_ha_sensor_alert,
  81. # AMS environmental alarms (regular AMS)
  82. "on_ams_humidity_high": provider.on_ams_humidity_high,
  83. "on_ams_temperature_high": provider.on_ams_temperature_high,
  84. "on_ams_drying_suspended": provider.on_ams_drying_suspended,
  85. # AMS-HT environmental alarms
  86. "on_ams_ht_humidity_high": provider.on_ams_ht_humidity_high,
  87. "on_ams_ht_temperature_high": provider.on_ams_ht_temperature_high,
  88. # Build plate detection
  89. "on_plate_not_empty": provider.on_plate_not_empty,
  90. "on_plate_clear_required": provider.on_plate_clear_required,
  91. # Post-print outcome confirmation (#1898)
  92. "on_print_confirm_request": provider.on_print_confirm_request,
  93. # Rows from before #3046 hold NULL here; "buttons" is what they did.
  94. "telegram_verdict_mode": provider.telegram_verdict_mode or "buttons",
  95. # Bed cooled
  96. "on_bed_cooled": provider.on_bed_cooled,
  97. # First layer complete
  98. "on_first_layer_complete": provider.on_first_layer_complete,
  99. # Messages from connected apps
  100. "on_app_message": bool(provider.on_app_message),
  101. # Inventory stock alerts. Absent here, the toggles above always read
  102. # back off no matter what the row holds — the same hand-maintained
  103. # field map the Home Assistant comment warns about.
  104. "on_stock_reorder_alert": provider.on_stock_reorder_alert,
  105. "on_stock_break_alert": provider.on_stock_break_alert,
  106. # Print queue events
  107. "on_queue_job_added": provider.on_queue_job_added,
  108. "on_queue_job_assigned": provider.on_queue_job_assigned,
  109. "on_queue_job_started": provider.on_queue_job_started,
  110. "on_queue_job_waiting": provider.on_queue_job_waiting,
  111. "on_queue_job_skipped": provider.on_queue_job_skipped,
  112. "on_queue_job_failed": provider.on_queue_job_failed,
  113. "on_queue_completed": provider.on_queue_completed,
  114. # Quiet hours
  115. "quiet_hours_enabled": provider.quiet_hours_enabled,
  116. "quiet_hours_start": provider.quiet_hours_start,
  117. "quiet_hours_end": provider.quiet_hours_end,
  118. # Daily digest
  119. "daily_digest_enabled": provider.daily_digest_enabled,
  120. "daily_digest_time": provider.daily_digest_time,
  121. # Printer filter
  122. "printer_id": provider.printer_id,
  123. # Status tracking
  124. "last_success": provider.last_success,
  125. "last_error": provider.last_error,
  126. "last_error_at": provider.last_error_at,
  127. # Timestamps
  128. "created_at": provider.created_at,
  129. "updated_at": provider.updated_at,
  130. }
  131. # ============================================================================
  132. # Provider List/Create Routes (no path parameters)
  133. # ============================================================================
  134. @router.get("/", response_model=list[NotificationProviderResponse])
  135. async def list_notification_providers(
  136. db: AsyncSession = Depends(get_db),
  137. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_READ),
  138. ):
  139. """List all notification providers."""
  140. result = await db.execute(select(NotificationProvider).order_by(NotificationProvider.created_at.desc()))
  141. providers = result.scalars().all()
  142. return [_provider_to_dict(provider) for provider in providers]
  143. @router.post("/", response_model=NotificationProviderResponse)
  144. async def create_notification_provider(
  145. provider_data: NotificationProviderCreate,
  146. db: AsyncSession = Depends(get_db),
  147. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_CREATE),
  148. ):
  149. """Create a new notification provider."""
  150. if provider_data.provider_type.value == "notify":
  151. try:
  152. notify_credentials(provider_data.config)
  153. except NotifyError as exc:
  154. raise HTTPException(status_code=422, detail=str(exc)) from None
  155. provider = NotificationProvider(
  156. name=provider_data.name,
  157. provider_type=provider_data.provider_type.value,
  158. enabled=provider_data.enabled,
  159. config=json.dumps(provider_data.config),
  160. attach_photo=provider_data.attach_photo,
  161. # Print lifecycle events
  162. on_print_start=provider_data.on_print_start,
  163. on_print_complete=provider_data.on_print_complete,
  164. on_print_failed=provider_data.on_print_failed,
  165. on_print_stopped=provider_data.on_print_stopped,
  166. on_print_progress=provider_data.on_print_progress,
  167. on_print_missing_spool_assignment=provider_data.on_print_missing_spool_assignment,
  168. on_billing_charge_failed=provider_data.on_billing_charge_failed,
  169. # Printer status events
  170. on_printer_offline=provider_data.on_printer_offline,
  171. on_printer_error=provider_data.on_printer_error,
  172. on_ai_failure_detection=provider_data.on_ai_failure_detection,
  173. on_filament_low=provider_data.on_filament_low,
  174. on_maintenance_due=provider_data.on_maintenance_due,
  175. # Home Assistant sensor alerts (#1148, #2824)
  176. on_ha_sensor_alert=provider_data.on_ha_sensor_alert,
  177. on_location_ha_sensor_alert=provider_data.on_location_ha_sensor_alert,
  178. # AMS environmental alarms (regular AMS)
  179. on_ams_humidity_high=provider_data.on_ams_humidity_high,
  180. on_ams_temperature_high=provider_data.on_ams_temperature_high,
  181. on_ams_drying_suspended=provider_data.on_ams_drying_suspended,
  182. # AMS-HT environmental alarms
  183. on_ams_ht_humidity_high=provider_data.on_ams_ht_humidity_high,
  184. on_ams_ht_temperature_high=provider_data.on_ams_ht_temperature_high,
  185. # Build plate detection
  186. on_plate_not_empty=provider_data.on_plate_not_empty,
  187. on_plate_clear_required=provider_data.on_plate_clear_required,
  188. # Post-print outcome confirmation (#1898)
  189. on_print_confirm_request=provider_data.on_print_confirm_request,
  190. telegram_verdict_mode=provider_data.telegram_verdict_mode,
  191. # Bed cooled
  192. on_bed_cooled=provider_data.on_bed_cooled,
  193. # First layer complete
  194. on_first_layer_complete=provider_data.on_first_layer_complete,
  195. on_app_message=provider_data.on_app_message,
  196. # Inventory stock alerts
  197. on_stock_reorder_alert=provider_data.on_stock_reorder_alert,
  198. on_stock_break_alert=provider_data.on_stock_break_alert,
  199. # Print queue events
  200. on_queue_job_added=provider_data.on_queue_job_added,
  201. on_queue_job_assigned=provider_data.on_queue_job_assigned,
  202. on_queue_job_started=provider_data.on_queue_job_started,
  203. on_queue_job_waiting=provider_data.on_queue_job_waiting,
  204. on_queue_job_skipped=provider_data.on_queue_job_skipped,
  205. on_queue_job_failed=provider_data.on_queue_job_failed,
  206. on_queue_completed=provider_data.on_queue_completed,
  207. # Quiet hours
  208. quiet_hours_enabled=provider_data.quiet_hours_enabled,
  209. quiet_hours_start=provider_data.quiet_hours_start,
  210. quiet_hours_end=provider_data.quiet_hours_end,
  211. # Daily digest
  212. daily_digest_enabled=provider_data.daily_digest_enabled,
  213. daily_digest_time=provider_data.daily_digest_time,
  214. # Printer filter
  215. printer_id=provider_data.printer_id,
  216. )
  217. db.add(provider)
  218. await db.commit()
  219. await db.refresh(provider)
  220. logger.info("Created notification provider: %s (%s)", provider.name, provider.provider_type)
  221. if provider.provider_type == "notify":
  222. _notify_providers_changed()
  223. await _resync_reaction_poller()
  224. return _provider_to_dict(provider)
  225. # ============================================================================
  226. # Static Path Routes (must come BEFORE parameterized routes)
  227. # ============================================================================
  228. @router.post("/test-config", response_model=NotificationTestResponse)
  229. async def test_notification_config(
  230. test_request: NotificationTestRequest,
  231. db: AsyncSession = Depends(get_db),
  232. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_CREATE),
  233. ):
  234. """Test notification configuration before saving."""
  235. success, message = await notification_service.send_test_notification(
  236. test_request.provider_type.value, test_request.config, db, attach_photo=test_request.attach_photo
  237. )
  238. return NotificationTestResponse(success=success, message=message)
  239. @router.post("/test-all")
  240. async def test_all_notification_providers(
  241. db: AsyncSession = Depends(get_db),
  242. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_UPDATE),
  243. ):
  244. """Send a test notification to all enabled providers."""
  245. result = await db.execute(select(NotificationProvider).where(NotificationProvider.enabled.is_(True)))
  246. providers = result.scalars().all()
  247. if not providers:
  248. return {"tested": 0, "success": 0, "failed": 0, "results": []}
  249. results = []
  250. success_count = 0
  251. failed_count = 0
  252. for provider in providers:
  253. config = json.loads(provider.config) if isinstance(provider.config, str) else provider.config
  254. success, message = await notification_service.send_test_notification(
  255. provider.provider_type, config, db, attach_photo=provider.attach_photo
  256. )
  257. # Update provider status
  258. if success:
  259. provider.last_success = datetime.now(timezone.utc)
  260. success_count += 1
  261. else:
  262. provider.last_error = message
  263. provider.last_error_at = datetime.now(timezone.utc)
  264. failed_count += 1
  265. results.append(
  266. {
  267. "provider_id": provider.id,
  268. "provider_name": provider.name,
  269. "provider_type": provider.provider_type,
  270. "success": success,
  271. "message": message,
  272. }
  273. )
  274. await db.commit()
  275. return {
  276. "tested": len(providers),
  277. "success": success_count,
  278. "failed": failed_count,
  279. "results": results,
  280. }
  281. # ============================================================================
  282. # Notification Log Routes (must come BEFORE /{provider_id} routes)
  283. # ============================================================================
  284. @router.get("/logs", response_model=list[NotificationLogResponse])
  285. async def get_notification_logs(
  286. limit: int = Query(default=100, ge=1, le=500),
  287. offset: int = Query(default=0, ge=0),
  288. provider_id: int | None = Query(default=None),
  289. event_type: str | None = Query(default=None),
  290. success: bool | None = Query(default=None),
  291. days: int | None = Query(default=7, ge=1, le=90, description="Filter logs from the last N days"),
  292. db: AsyncSession = Depends(get_db),
  293. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_READ),
  294. ):
  295. """Get notification logs with optional filters."""
  296. query = select(NotificationLog).order_by(desc(NotificationLog.created_at))
  297. # Apply filters
  298. if provider_id is not None:
  299. query = query.where(NotificationLog.provider_id == provider_id)
  300. if event_type is not None:
  301. query = query.where(NotificationLog.event_type == event_type)
  302. if success is not None:
  303. query = query.where(NotificationLog.success == success)
  304. if days is not None:
  305. cutoff = datetime.now(timezone.utc) - timedelta(days=days)
  306. query = query.where(NotificationLog.created_at >= cutoff)
  307. query = query.offset(offset).limit(limit)
  308. result = await db.execute(query)
  309. logs = result.scalars().all()
  310. # Get provider info for each log
  311. response = []
  312. providers_cache: dict[int, NotificationProvider | None] = {}
  313. for log in logs:
  314. if log.provider_id not in providers_cache:
  315. provider_result = await db.execute(
  316. select(NotificationProvider).where(NotificationProvider.id == log.provider_id)
  317. )
  318. providers_cache[log.provider_id] = provider_result.scalar_one_or_none()
  319. provider = providers_cache[log.provider_id]
  320. response.append(
  321. NotificationLogResponse(
  322. id=log.id,
  323. provider_id=log.provider_id,
  324. provider_name=provider.name if provider else None,
  325. provider_type=provider.provider_type if provider else None,
  326. event_type=log.event_type,
  327. title=log.title,
  328. message=log.message,
  329. success=log.success,
  330. error_message=log.error_message,
  331. printer_id=log.printer_id,
  332. printer_name=log.printer_name,
  333. created_at=log.created_at,
  334. )
  335. )
  336. return response
  337. @router.get("/logs/stats", response_model=NotificationLogStats)
  338. async def get_notification_log_stats(
  339. days: int = Query(default=7, ge=1, le=90, description="Statistics for the last N days"),
  340. db: AsyncSession = Depends(get_db),
  341. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_READ),
  342. ):
  343. """Get notification log statistics."""
  344. cutoff = datetime.now(timezone.utc) - timedelta(days=days)
  345. # Total counts
  346. total_result = await db.execute(select(func.count(NotificationLog.id)).where(NotificationLog.created_at >= cutoff))
  347. total = total_result.scalar() or 0
  348. success_result = await db.execute(
  349. select(func.count(NotificationLog.id)).where(
  350. NotificationLog.created_at >= cutoff, NotificationLog.success.is_(True)
  351. )
  352. )
  353. success_count = success_result.scalar() or 0
  354. # By event type
  355. event_result = await db.execute(
  356. select(NotificationLog.event_type, func.count(NotificationLog.id))
  357. .where(NotificationLog.created_at >= cutoff)
  358. .group_by(NotificationLog.event_type)
  359. )
  360. by_event_type = {row[0]: row[1] for row in event_result.fetchall()}
  361. # By provider (need to join to get name)
  362. provider_result = await db.execute(
  363. select(NotificationProvider.name, func.count(NotificationLog.id))
  364. .join(NotificationProvider, NotificationLog.provider_id == NotificationProvider.id)
  365. .where(NotificationLog.created_at >= cutoff)
  366. .group_by(NotificationProvider.name)
  367. )
  368. by_provider = {row[0]: row[1] for row in provider_result.fetchall()}
  369. return NotificationLogStats(
  370. total=total,
  371. success_count=success_count,
  372. failure_count=total - success_count,
  373. by_event_type=by_event_type,
  374. by_provider=by_provider,
  375. )
  376. @router.delete("/logs")
  377. async def clear_notification_logs(
  378. older_than_days: int = Query(default=30, ge=1, description="Delete logs older than N days"),
  379. db: AsyncSession = Depends(get_db),
  380. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_DELETE),
  381. ):
  382. """Clear old notification logs."""
  383. cutoff = datetime.now(timezone.utc) - timedelta(days=older_than_days)
  384. result = await db.execute(delete(NotificationLog).where(NotificationLog.created_at < cutoff))
  385. await db.commit()
  386. deleted_count = result.rowcount
  387. logger.info("Deleted %s notification logs older than %s days", deleted_count, older_than_days)
  388. return {"deleted": deleted_count, "message": f"Deleted {deleted_count} logs older than {older_than_days} days"}
  389. @router.get("/photos/{filename}")
  390. async def get_notification_photo(filename: str):
  391. """Serve an ad-hoc notification snapshot to HA, Bark or Slack.
  392. They fetch this URL themselves with no session, so the unguessable
  393. filename is the credential and opens this one photo only -- see
  394. backend/app/utils/notification_photos.py. Anything that isn't a live
  395. photo of exactly that shape is a 404.
  396. """
  397. photo_path = find_notification_photo(filename)
  398. if photo_path is None:
  399. raise HTTPException(404, "Photo not found")
  400. return FileResponse(path=photo_path, media_type="image/jpeg", headers={"Cache-Control": "private, no-store"})
  401. # ============================================================================
  402. # Provider Instance Routes (parameterized - must come LAST)
  403. # ============================================================================
  404. # Messages from other applications -------------------------------------------
  405. # Per caller, in memory: enough for any real app (Bambuddy Orders sends a few a
  406. # day), and a buggy or hostile one can't flood the channels.
  407. APP_MESSAGE_LIMIT = 20
  408. APP_MESSAGE_WINDOW_SECONDS = 60
  409. _app_message_times: dict[str, deque[float]] = defaultdict(deque)
  410. def _app_sender(caller: ScopedCaller) -> tuple[str, str]:
  411. """(rate-limit key, name shown in the log) for whoever sends the message."""
  412. if caller.api_key is not None:
  413. return f"key:{caller.api_key.id}", caller.api_key.name
  414. if caller.user is not None:
  415. return f"user:{caller.user.id}", caller.user.username
  416. return "anonymous", "app"
  417. def _check_app_message_rate(key: str) -> None:
  418. times = _app_message_times[key]
  419. cutoff = time.monotonic() - APP_MESSAGE_WINDOW_SECONDS
  420. while times and times[0] < cutoff:
  421. times.popleft()
  422. if len(times) >= APP_MESSAGE_LIMIT:
  423. raise HTTPException(status_code=429, detail="Too many messages; try again in a minute")
  424. times.append(time.monotonic())
  425. @router.post("/app-message", response_model=AppMessageResult)
  426. async def send_app_message(
  427. data: AppMessage,
  428. db: AsyncSession = Depends(get_db),
  429. caller: ScopedCaller = Depends(require_notification_send()),
  430. ):
  431. """Send a message through every enabled channel that has "Messages from
  432. connected apps" on. For other applications, e.g. Bambuddy Orders; an API
  433. key needs the "Send notifications" permission."""
  434. key, sender = _app_sender(caller)
  435. _check_app_message_rate(key)
  436. channels = await notification_service.on_app_message(
  437. db, sender=sender, title=data.title, message=data.message, url=data.url
  438. )
  439. return AppMessageResult(channels=channels)
  440. @router.get("/app-message/channels", response_model=list[AppMessageChannel])
  441. async def app_message_channels(
  442. db: AsyncSession = Depends(get_db),
  443. _: ScopedCaller = Depends(require_notification_send()),
  444. ):
  445. """The enabled channels that deliver app messages: names and types only,
  446. so an app can tell its user where its messages will arrive."""
  447. rows = await db.execute(
  448. select(NotificationProvider)
  449. .where(NotificationProvider.enabled.is_(True), NotificationProvider.on_app_message.is_(True))
  450. .order_by(NotificationProvider.name)
  451. )
  452. return [AppMessageChannel(name=p.name, provider_type=p.provider_type) for p in rows.scalars()]
  453. @router.get("/{provider_id}", response_model=NotificationProviderResponse)
  454. async def get_notification_provider(
  455. provider_id: int,
  456. db: AsyncSession = Depends(get_db),
  457. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_READ),
  458. ):
  459. """Get a specific notification provider."""
  460. result = await db.execute(select(NotificationProvider).where(NotificationProvider.id == provider_id))
  461. provider = result.scalar_one_or_none()
  462. if not provider:
  463. raise HTTPException(status_code=404, detail="Notification provider not found")
  464. return _provider_to_dict(provider)
  465. @router.patch("/{provider_id}", response_model=NotificationProviderResponse)
  466. async def update_notification_provider(
  467. provider_id: int,
  468. update_data: NotificationProviderUpdate,
  469. db: AsyncSession = Depends(get_db),
  470. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_UPDATE),
  471. ):
  472. """Update a notification provider."""
  473. result = await db.execute(select(NotificationProvider).where(NotificationProvider.id == provider_id))
  474. provider = result.scalar_one_or_none()
  475. if not provider:
  476. raise HTTPException(status_code=404, detail="Notification provider not found")
  477. old_config = json.loads(provider.config) if isinstance(provider.config, str) else provider.config
  478. old_type = provider.provider_type
  479. # Update only provided fields
  480. update_dict = update_data.model_dump(exclude_unset=True)
  481. effective_type = update_dict.get("provider_type") or provider.provider_type
  482. effective_type = getattr(effective_type, "value", effective_type)
  483. if effective_type == "notify":
  484. try:
  485. effective_config = update_dict.get("config", old_config)
  486. if not isinstance(effective_config, dict):
  487. raise NotifyError("Notify! configuration must be an object")
  488. notify_credentials(effective_config)
  489. except NotifyError as exc:
  490. raise HTTPException(status_code=422, detail=str(exc)) from None
  491. for key, value in update_dict.items():
  492. if key == "config" and value is not None:
  493. setattr(provider, key, json.dumps(value))
  494. elif key == "provider_type" and value is not None:
  495. setattr(provider, key, value.value)
  496. else:
  497. setattr(provider, key, value)
  498. if old_type == "notify":
  499. effective_config = json.loads(provider.config) if isinstance(provider.config, str) else provider.config
  500. remove_widgets = (
  501. provider.provider_type != "notify"
  502. or not provider.enabled
  503. or effective_config.get("lock_screen_widgets") is not True
  504. )
  505. if remove_widgets or provider.printer_id is not None:
  506. condition = NotificationLockScreenWidget.provider_id == provider_id
  507. if not remove_widgets:
  508. condition = condition & (NotificationLockScreenWidget.printer_id != provider.printer_id)
  509. # Persist opt-out before returning, even if the user immediately
  510. # opts back in. The worker finishes cleanup before creating anew.
  511. await db.execute(
  512. update(NotificationLockScreenWidget)
  513. .where(condition)
  514. .values(
  515. state="deleting",
  516. failures=0,
  517. next_attempt_at=None,
  518. )
  519. )
  520. await db.commit()
  521. await db.refresh(provider)
  522. response = _provider_to_dict(provider)
  523. new_config = response["config"]
  524. changed = old_type != provider.provider_type or any(
  525. old_config.get(key) != new_config.get(key) for key in ("device_id", "token")
  526. )
  527. # Refresh opened a new read transaction; close it before remote cleanup.
  528. await db.commit()
  529. if old_type == "notify" and changed:
  530. await notify_live_activities.schedule_cleanup(provider_id, old_config)
  531. await notify_widgets.schedule_cleanup(provider_id, old_config)
  532. if old_type == "notify" or provider.provider_type == "notify":
  533. _notify_providers_changed()
  534. logger.info("Updated notification provider: %s", provider.name)
  535. await _resync_reaction_poller()
  536. return response
  537. @router.delete("/{provider_id}")
  538. async def delete_notification_provider(
  539. provider_id: int,
  540. db: AsyncSession = Depends(get_db),
  541. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_DELETE),
  542. ):
  543. """Delete a notification provider."""
  544. result = await db.execute(select(NotificationProvider).where(NotificationProvider.id == provider_id))
  545. provider = result.scalar_one_or_none()
  546. if not provider:
  547. raise HTTPException(status_code=404, detail="Notification provider not found")
  548. name = provider.name
  549. if provider.provider_type == "notify":
  550. config = json.loads(provider.config) if isinstance(provider.config, str) else provider.config
  551. # Retire this provider and capture owned IDs before FK cascade removes
  552. # them. Queue remote cleanup; HTTP must never delay this request.
  553. provider.enabled = False
  554. await db.commit()
  555. await notify_live_activities.schedule_cleanup(provider_id, config)
  556. await notify_widgets.schedule_cleanup(provider_id, config)
  557. await db.delete(provider)
  558. await db.commit()
  559. if provider.provider_type == "notify":
  560. _notify_providers_changed()
  561. logger.info("Deleted notification provider: %s", name)
  562. await _resync_reaction_poller()
  563. return {"message": f"Notification provider '{name}' deleted"}
  564. @router.post("/{provider_id}/test", response_model=NotificationTestResponse)
  565. async def test_notification_provider(
  566. provider_id: int,
  567. db: AsyncSession = Depends(get_db),
  568. _: User | None = RequirePermissionIfAuthEnabled(Permission.NOTIFICATIONS_UPDATE),
  569. ):
  570. """Send a test notification using an existing provider."""
  571. result = await db.execute(select(NotificationProvider).where(NotificationProvider.id == provider_id))
  572. provider = result.scalar_one_or_none()
  573. if not provider:
  574. raise HTTPException(status_code=404, detail="Notification provider not found")
  575. config = json.loads(provider.config) if isinstance(provider.config, str) else provider.config
  576. success, message = await notification_service.send_test_notification(
  577. provider.provider_type, config, db, attach_photo=provider.attach_photo
  578. )
  579. # Update provider status
  580. if success:
  581. provider.last_success = datetime.now(timezone.utc)
  582. else:
  583. provider.last_error = message
  584. provider.last_error_at = datetime.now(timezone.utc)
  585. await db.commit()
  586. return NotificationTestResponse(success=success, message=message)