notification_service.py 69 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812
  1. """Notification service for sending push notifications via various providers."""
  2. import asyncio
  3. import json
  4. import logging
  5. import re
  6. import smtplib
  7. from datetime import datetime, timedelta, timezone
  8. from email.mime.multipart import MIMEMultipart
  9. from email.mime.text import MIMEText
  10. from typing import Any
  11. from urllib.parse import quote
  12. import httpx
  13. from sqlalchemy import select
  14. from sqlalchemy.ext.asyncio import AsyncSession
  15. from backend.app.models.notification import NotificationDigestQueue, NotificationLog, NotificationProvider
  16. from backend.app.models.notification_template import NotificationTemplate
  17. logger = logging.getLogger(__name__)
  18. class NotificationService:
  19. """Service for sending notifications through various providers."""
  20. def __init__(self):
  21. self._http_client: httpx.AsyncClient | None = None
  22. self._template_cache: dict[str, NotificationTemplate] = {}
  23. self._digest_scheduler_task: asyncio.Task | None = None
  24. self._last_digest_check: str = "" # "HH:MM" to avoid duplicate checks
  25. async def _get_client(self) -> httpx.AsyncClient:
  26. """Get or create HTTP client."""
  27. if self._http_client is None or self._http_client.is_closed:
  28. self._http_client = httpx.AsyncClient(timeout=30.0)
  29. return self._http_client
  30. async def close(self):
  31. """Close HTTP client."""
  32. if self._http_client and not self._http_client.is_closed:
  33. await self._http_client.aclose()
  34. def _is_in_quiet_hours(self, provider: NotificationProvider) -> bool:
  35. """Check if current time is within provider's quiet hours."""
  36. if not provider.quiet_hours_enabled:
  37. return False
  38. if not provider.quiet_hours_start or not provider.quiet_hours_end:
  39. return False
  40. try:
  41. now = datetime.now()
  42. current_time = now.hour * 60 + now.minute
  43. start_parts = provider.quiet_hours_start.split(":")
  44. end_parts = provider.quiet_hours_end.split(":")
  45. start_minutes = int(start_parts[0]) * 60 + int(start_parts[1])
  46. end_minutes = int(end_parts[0]) * 60 + int(end_parts[1])
  47. # Handle overnight quiet hours (e.g., 22:00 to 07:00)
  48. if start_minutes > end_minutes:
  49. # Quiet hours span midnight
  50. return current_time >= start_minutes or current_time < end_minutes
  51. else:
  52. # Same day quiet hours
  53. return start_minutes <= current_time < end_minutes
  54. except (ValueError, TypeError, AttributeError):
  55. logger.warning("Invalid quiet hours format for provider %s", provider.name)
  56. return False
  57. async def _get_template(self, db: AsyncSession, event_type: str) -> NotificationTemplate | None:
  58. """Get a notification template by event type."""
  59. # Check cache first
  60. if event_type in self._template_cache:
  61. return self._template_cache[event_type]
  62. result = await db.execute(select(NotificationTemplate).where(NotificationTemplate.event_type == event_type))
  63. template = result.scalar_one_or_none()
  64. if template:
  65. self._template_cache[event_type] = template
  66. return template
  67. def _render_template(self, template_str: str, variables: dict[str, Any]) -> str:
  68. """Render a template string with variables. Missing variables become empty."""
  69. result = template_str
  70. for key, value in variables.items():
  71. result = result.replace("{" + key + "}", str(value) if value is not None else "")
  72. # Remove any remaining unreplaced placeholders
  73. result = re.sub(r"\{[a-z_]+\}", "", result)
  74. return result
  75. async def _format_eta(self, seconds: int | None, db: AsyncSession) -> str:
  76. """Format ETA as wall-clock time, respecting user's time_format setting."""
  77. if not seconds or seconds <= 0:
  78. return "Unknown"
  79. from backend.app.api.routes.settings import get_setting
  80. eta_time = datetime.now() + timedelta(seconds=seconds)
  81. time_format = await get_setting(db, "time_format")
  82. if time_format == "12h":
  83. return eta_time.strftime("%I:%M %p").lstrip("0")
  84. # Default to 24h for "24h", "system", or unset
  85. return eta_time.strftime("%H:%M")
  86. def _format_duration(self, seconds: int | None) -> str:
  87. """Format duration in seconds to human-readable string."""
  88. if seconds is None:
  89. return "Unknown"
  90. hours = seconds // 3600
  91. minutes = (seconds % 3600) // 60
  92. if hours > 0:
  93. return f"{hours}h {minutes}m"
  94. return f"{minutes}m"
  95. def _clean_filename(self, filename: str) -> str:
  96. """Extract filename and remove file extensions."""
  97. import os
  98. # Strip path prefix (e.g., /data/Metadata/plate_5.gcode -> plate_5.gcode)
  99. filename = os.path.basename(filename)
  100. # Remove common extensions
  101. if filename.endswith(".gcode.3mf"):
  102. return filename[:-10]
  103. elif filename.endswith(".gcode"):
  104. return filename[:-6]
  105. elif filename.endswith(".3mf"):
  106. return filename[:-4]
  107. return filename
  108. async def _build_message_from_template(
  109. self, db: AsyncSession, event_type: str, variables: dict[str, Any]
  110. ) -> tuple[str, str]:
  111. """Build notification title and body from template."""
  112. # Add common variables
  113. variables["timestamp"] = datetime.now().strftime("%Y-%m-%d %H:%M")
  114. variables["app_name"] = "Bambuddy"
  115. template = await self._get_template(db, event_type)
  116. if not template:
  117. # Fallback to simple message
  118. logger.warning("Template not found for event type: %s", event_type)
  119. return event_type.replace("_", " ").title(), str(variables)
  120. title = self._render_template(template.title_template, variables)
  121. body = self._render_template(template.body_template, variables)
  122. return title, body
  123. async def send_test_notification(
  124. self, provider_type: str, config: dict[str, Any], db: AsyncSession | None = None
  125. ) -> tuple[bool, str]:
  126. """Send a test notification to verify configuration."""
  127. if db:
  128. title, message = await self._build_message_from_template(db, "test", {})
  129. else:
  130. title = "Bambuddy Test"
  131. message = "This is a test notification. If you see this, notifications are working!"
  132. try:
  133. if provider_type == "callmebot":
  134. return await self._send_callmebot(config, f"{title}\n{message}")
  135. elif provider_type == "ntfy":
  136. return await self._send_ntfy(config, title, message)
  137. elif provider_type == "pushover":
  138. return await self._send_pushover(config, title, message)
  139. elif provider_type == "telegram":
  140. return await self._send_telegram(config, f"*{title}*\n{message}")
  141. elif provider_type == "email":
  142. return await self._send_email(config, title, message)
  143. elif provider_type == "discord":
  144. return await self._send_discord(config, title, message)
  145. elif provider_type == "webhook":
  146. return await self._send_webhook(config, title, message)
  147. elif provider_type == "homeassistant":
  148. return await self._send_homeassistant(config, title, message, db=db)
  149. else:
  150. return False, f"Unknown provider type: {provider_type}"
  151. except Exception as e:
  152. logger.exception("Error sending test notification via %s", provider_type)
  153. return False, str(e)
  154. async def _send_callmebot(self, config: dict, message: str) -> tuple[bool, str]:
  155. """Send notification via CallMeBot (WhatsApp)."""
  156. phone = config.get("phone", "").strip()
  157. apikey = config.get("apikey", "").strip()
  158. if not phone or not apikey:
  159. return False, "Phone number and API key are required"
  160. # URL encode the message
  161. encoded_message = quote(message)
  162. url = f"https://api.callmebot.com/whatsapp.php?phone={phone}&text={encoded_message}&apikey={apikey}"
  163. client = await self._get_client()
  164. response = await client.get(url)
  165. if response.status_code == 200:
  166. return True, "Message sent successfully"
  167. else:
  168. return False, f"HTTP {response.status_code}: {response.text[:200]}"
  169. async def _send_ntfy(
  170. self,
  171. config: dict,
  172. title: str,
  173. message: str,
  174. image_data: bytes | None = None,
  175. event_type: str | None = None,
  176. ) -> tuple[bool, str]:
  177. """Send notification via ntfy."""
  178. server = config.get("server", "https://ntfy.sh").rstrip("/")
  179. topic = config.get("topic", "").strip()
  180. auth_token = config.get("auth_token", "").strip()
  181. if not topic:
  182. return False, "Topic is required"
  183. url = f"{server}/{topic}"
  184. # ntfy reads Title/Message from HTTP headers. httpx enforces ASCII
  185. # for str header values, but printer names and filenames can contain
  186. # non-ASCII characters (e.g. accented letters, CJK). Passing bytes
  187. # bypasses the ASCII check — ntfy handles UTF-8 headers correctly.
  188. headers: dict[str, str | bytes] = {"Title": title.encode("utf-8")}
  189. # Per-event Priority header (#990). Only set when the user has
  190. # explicitly mapped this event to a 1-5 value; otherwise fall through
  191. # to the ntfy server's default so existing setups stay unchanged.
  192. event_priorities = config.get("event_priorities") or {}
  193. if event_type and isinstance(event_priorities, dict):
  194. raw = event_priorities.get(event_type)
  195. try:
  196. priority = int(raw) if raw is not None else None
  197. except (TypeError, ValueError):
  198. priority = None
  199. if priority is not None and 1 <= priority <= 5:
  200. headers["Priority"] = str(priority)
  201. if auth_token:
  202. headers["Authorization"] = f"Bearer {auth_token}"
  203. client = await self._get_client()
  204. if image_data:
  205. # ntfy supports image attachments via multipart form-data.
  206. # HTTP headers cannot contain newlines, but ntfy interprets
  207. # literal \n (backslash-n) as newlines in the Message header.
  208. headers["Filename"] = "photo.jpg"
  209. headers["Message"] = message.replace("\n", "\\n").encode("utf-8")
  210. response = await client.put(url, content=image_data, headers=headers)
  211. if response.status_code == 400 and "attachments not allowed" in response.text:
  212. # Server has attachments disabled — retry without the image
  213. headers.pop("Filename", None)
  214. headers.pop("Message", None)
  215. response = await client.post(url, content=message.encode("utf-8"), headers=headers)
  216. else:
  217. response = await client.post(url, content=message.encode("utf-8"), headers=headers)
  218. if response.status_code in (200, 204):
  219. return True, "Message sent successfully"
  220. else:
  221. return False, f"HTTP {response.status_code}: {response.text[:200]}"
  222. async def _send_pushover(
  223. self, config: dict, title: str, message: str, image_data: bytes | None = None
  224. ) -> tuple[bool, str]:
  225. """Send notification via Pushover.
  226. Args:
  227. config: Provider configuration with user_key, app_token, priority
  228. title: Notification title
  229. message: Notification body
  230. image_data: Optional JPEG image bytes to attach (max 2.5MB)
  231. """
  232. user_key = config.get("user_key", "").strip()
  233. app_token = config.get("app_token", "").strip()
  234. priority = config.get("priority", 0)
  235. if not user_key or not app_token:
  236. return False, "User key and app token are required"
  237. url = "https://api.pushover.net/1/messages.json"
  238. data = {
  239. "token": app_token,
  240. "user": user_key,
  241. "title": title,
  242. "message": message,
  243. "priority": priority,
  244. }
  245. client = await self._get_client()
  246. if image_data:
  247. # Pushover supports image attachments via multipart form-data
  248. files = {"attachment": ("photo.jpg", image_data, "image/jpeg")}
  249. response = await client.post(url, data=data, files=files)
  250. else:
  251. response = await client.post(url, data=data)
  252. if response.status_code == 200:
  253. return True, "Message sent successfully"
  254. else:
  255. try:
  256. error_data = response.json()
  257. errors = error_data.get("errors", [])
  258. return False, f"Pushover error: {', '.join(errors)}"
  259. except Exception:
  260. return False, f"HTTP {response.status_code}: {response.text[:200]}"
  261. async def _send_telegram(self, config: dict, message: str, image_data: bytes | None = None) -> tuple[bool, str]:
  262. """Send notification via Telegram bot."""
  263. bot_token = config.get("bot_token", "").strip()
  264. chat_id = config.get("chat_id", "").strip()
  265. if not bot_token or not chat_id:
  266. return False, "Bot token and chat ID are required"
  267. # Escape underscores in the message body so Telegram Markdown
  268. # parsing doesn't break on job names like "A1_plate_8" or error
  269. # codes like "0300_0001". The title is already wrapped in *bold*
  270. # markers, so only escape after the first newline.
  271. if "\n" in message:
  272. title_part, body_part = message.split("\n", 1)
  273. body_part = body_part.replace("_", "\\_")
  274. message = f"{title_part}\n{body_part}"
  275. client = await self._get_client()
  276. if image_data:
  277. # Use sendPhoto to attach the thumbnail with the caption
  278. url = f"https://api.telegram.org/bot{bot_token}/sendPhoto"
  279. response = await client.post(
  280. url,
  281. data={"chat_id": chat_id, "caption": message, "parse_mode": "Markdown"},
  282. files={"photo": ("photo.jpg", image_data, "image/jpeg")},
  283. )
  284. else:
  285. url = f"https://api.telegram.org/bot{bot_token}/sendMessage"
  286. data = {
  287. "chat_id": chat_id,
  288. "text": message,
  289. "parse_mode": "Markdown",
  290. }
  291. response = await client.post(url, json=data)
  292. if response.status_code == 200:
  293. result = response.json()
  294. if result.get("ok"):
  295. return True, "Message sent successfully"
  296. else:
  297. return False, f"Telegram error: {result.get('description', 'Unknown error')}"
  298. else:
  299. return False, f"HTTP {response.status_code}: {response.text[:200]}"
  300. async def _send_email(self, config: dict, subject: str, body: str) -> tuple[bool, str]:
  301. """Send notification via email (SMTP)."""
  302. smtp_server = config.get("smtp_server", "").strip()
  303. smtp_port = int(config.get("smtp_port", 587))
  304. username = config.get("username", "").strip()
  305. password = config.get("password", "").strip()
  306. from_email = config.get("from_email", "").strip()
  307. to_email = config.get("to_email", "").strip()
  308. # Security: "starttls" (port 587), "ssl" (port 465), "none" (port 25)
  309. security = config.get("security", "starttls")
  310. # Authentication: "true" or "false"
  311. auth_enabled = config.get("auth_enabled", "true").lower() == "true"
  312. if not all([smtp_server, from_email, to_email]):
  313. return False, "SMTP server, from email, and to email are required"
  314. if auth_enabled and not all([username, password]):
  315. return False, "Username and password are required when authentication is enabled"
  316. try:
  317. msg = MIMEMultipart()
  318. msg["From"] = from_email
  319. msg["To"] = to_email
  320. msg["Subject"] = f"[Bambuddy] {subject}"
  321. msg.attach(MIMEText(body, "plain"))
  322. if security == "ssl":
  323. # Direct SSL connection (typically port 465)
  324. server = smtplib.SMTP_SSL(smtp_server, smtp_port)
  325. elif security == "starttls":
  326. # STARTTLS upgrade (typically port 587)
  327. server = smtplib.SMTP(smtp_server, smtp_port)
  328. server.starttls()
  329. else:
  330. # No encryption (typically port 25) - use with caution
  331. server = smtplib.SMTP(smtp_server, smtp_port)
  332. if auth_enabled:
  333. server.login(username, password)
  334. server.sendmail(from_email, to_email, msg.as_string())
  335. server.quit()
  336. return True, "Email sent successfully"
  337. except smtplib.SMTPAuthenticationError:
  338. return False, "SMTP authentication failed - check username/password"
  339. except smtplib.SMTPException as e:
  340. return False, f"SMTP error: {str(e)}"
  341. except Exception as e:
  342. return False, f"Email error: {str(e)}"
  343. async def _send_discord(
  344. self, config: dict, title: str, message: str, image_data: bytes | None = None
  345. ) -> tuple[bool, str]:
  346. """Send notification via Discord webhook."""
  347. webhook_url = config.get("webhook_url", "").strip()
  348. if not webhook_url:
  349. return False, "Webhook URL is required"
  350. if not webhook_url.startswith("https://discord.com/api/webhooks/"):
  351. return False, "Invalid Discord webhook URL"
  352. # Discord embed format for nicer messages
  353. embed = {
  354. "title": title,
  355. "description": message,
  356. "color": 0x00AE42, # Bambu green
  357. }
  358. client = await self._get_client()
  359. if image_data:
  360. # Attach image via multipart form-data and reference in embed
  361. embed["image"] = {"url": "attachment://photo.jpg"}
  362. payload = {"embeds": [embed]}
  363. response = await client.post(
  364. webhook_url,
  365. data={"payload_json": json.dumps(payload)},
  366. files={"files[0]": ("photo.jpg", image_data, "image/jpeg")},
  367. )
  368. else:
  369. response = await client.post(webhook_url, json={"embeds": [embed]})
  370. if response.status_code in (200, 204):
  371. return True, "Message sent successfully"
  372. else:
  373. return False, f"HTTP {response.status_code}: {response.text[:200]}"
  374. async def _send_webhook(
  375. self,
  376. config: dict,
  377. title: str,
  378. message: str,
  379. image_data: bytes | None = None,
  380. event_type: str | None = None,
  381. variables: dict | None = None,
  382. ) -> tuple[bool, str]:
  383. """Send notification via generic webhook (POST JSON).
  384. Supports two payload formats:
  385. - generic: Custom field names with timestamp/source metadata + structured event data
  386. - slack: Slack/Mattermost compatible format (just {"text": "..."})
  387. """
  388. webhook_url = config.get("webhook_url", "").strip()
  389. auth_header = config.get("auth_header", "").strip()
  390. payload_format = config.get("payload_format", "generic").strip()
  391. if not webhook_url:
  392. return False, "Webhook URL is required"
  393. # Build payload based on format
  394. if payload_format == "slack":
  395. # Slack/Mattermost format - just text field
  396. data = {"text": f"*{title}*\n{message}"}
  397. else:
  398. # Generic format with custom field names
  399. custom_field_title = config.get("field_title", "title").strip() or "title"
  400. custom_field_message = config.get("field_message", "message").strip() or "message"
  401. data = {
  402. custom_field_title: title,
  403. custom_field_message: message,
  404. "timestamp": datetime.now().isoformat(),
  405. "source": "Bambuddy",
  406. }
  407. # For generic format, include structured event data for automation tools
  408. if payload_format != "slack":
  409. if event_type:
  410. data["event"] = event_type
  411. if variables:
  412. for key, value in variables.items():
  413. if key not in data: # Don't overwrite title/message/timestamp/source
  414. data[key] = value
  415. # Attach base64-encoded image when available (generic format only)
  416. if image_data and payload_format != "slack":
  417. import base64
  418. data["image"] = base64.b64encode(image_data).decode("ascii")
  419. headers = {"Content-Type": "application/json"}
  420. if auth_header:
  421. # Support "Bearer token" or just "token" format
  422. if " " in auth_header:
  423. headers["Authorization"] = auth_header
  424. else:
  425. headers["Authorization"] = f"Bearer {auth_header}"
  426. client = await self._get_client()
  427. try:
  428. response = await client.post(webhook_url, json=data, headers=headers)
  429. if response.status_code in (200, 201, 202, 204):
  430. return True, "Webhook delivered successfully"
  431. else:
  432. return False, f"HTTP {response.status_code}: {response.text[:200]}"
  433. except Exception as e:
  434. return False, f"Webhook error: {str(e)}"
  435. async def _send_homeassistant(
  436. self, config: dict, title: str, message: str, db: AsyncSession | None = None
  437. ) -> tuple[bool, str]:
  438. """Send notification via Home Assistant.
  439. Uses the globally configured HA URL/token from settings.
  440. Defaults to persistent_notification/create, but supports
  441. custom services via config["service"] (e.g. notify.mobile_app_myphone).
  442. """
  443. # Get HA connection settings from global config
  444. ha_url = ""
  445. ha_token = ""
  446. if db:
  447. from backend.app.api.routes.settings import get_homeassistant_settings
  448. try:
  449. ha_settings = await get_homeassistant_settings(db)
  450. ha_url = ha_settings.get("ha_url", "")
  451. ha_token = ha_settings.get("ha_token", "")
  452. except Exception as e:
  453. logger.warning("Failed to read HA settings from database: %s", e)
  454. else:
  455. # Fallback: read directly from environment if no DB session
  456. import os
  457. ha_url = os.environ.get("HA_URL", "")
  458. ha_token = os.environ.get("HA_TOKEN", "")
  459. if not ha_url or not ha_token:
  460. return False, (
  461. "Home Assistant is not configured. Please set HA URL and token in Settings → Network → Home Assistant."
  462. )
  463. # Determine which HA service to call - Default: persistent_notification.create
  464. service = (config.get("service") or "").strip()
  465. if service:
  466. # Allow in different forms:
  467. # - notify.mobile_app_<device>
  468. # - notify/mobile_app_<device>
  469. # - api/services/notify/mobile_app_<device>
  470. service_str = service.lstrip("/")
  471. if service_str.startswith("api/services/"):
  472. endpoint = service_str
  473. elif "/" in service_str:
  474. endpoint = f"api/services/{service_str}"
  475. elif "." in service_str:
  476. domain, svc = service_str.split(".", 1)
  477. endpoint = f"api/services/{domain}/{svc}"
  478. else:
  479. return False, (
  480. "Invalid Home Assistant service name. Use e.g. 'notify.mobile_app_yourdevice' or 'notify/your_service'."
  481. )
  482. if not re.match(r"^api/services/[a-zA-Z0-9_]+/[a-zA-Z0-9_]+$", endpoint):
  483. return False, (
  484. "Invalid Home Assistant service name. Domain and service must only contain letters, numbers, and underscores."
  485. )
  486. else:
  487. endpoint = "api/services/persistent_notification/create"
  488. url = f"{ha_url.rstrip('/')}/{endpoint}"
  489. headers = {
  490. "Authorization": f"Bearer {ha_token}",
  491. "Content-Type": "application/json",
  492. }
  493. payload = {
  494. "title": title,
  495. "message": message,
  496. }
  497. client = await self._get_client()
  498. response = await client.post(url, json=payload, headers=headers)
  499. if response.status_code in (200, 201):
  500. return True, "Notification sent via Home Assistant"
  501. elif response.status_code == 401:
  502. return False, "Home Assistant authentication failed - check your token"
  503. else:
  504. return False, f"HTTP {response.status_code}: {response.text[:200]}"
  505. async def _send_to_provider(
  506. self,
  507. provider: NotificationProvider,
  508. title: str,
  509. message: str,
  510. db: AsyncSession | None = None,
  511. image_data: bytes | None = None,
  512. event_type: str | None = None,
  513. variables: dict | None = None,
  514. ) -> tuple[bool, str]:
  515. """Send notification to a specific provider."""
  516. # Check quiet hours
  517. if self._is_in_quiet_hours(provider):
  518. logger.info("Skipping notification to %s - quiet hours active", provider.name)
  519. return True, "Skipped - quiet hours"
  520. config = json.loads(provider.config) if isinstance(provider.config, str) else provider.config
  521. try:
  522. if provider.provider_type == "callmebot":
  523. return await self._send_callmebot(config, f"{title}\n{message}")
  524. elif provider.provider_type == "ntfy":
  525. return await self._send_ntfy(config, title, message, image_data=image_data, event_type=event_type)
  526. elif provider.provider_type == "pushover":
  527. return await self._send_pushover(config, title, message, image_data=image_data)
  528. elif provider.provider_type == "telegram":
  529. return await self._send_telegram(config, f"*{title}*\n{message}", image_data=image_data)
  530. elif provider.provider_type == "email":
  531. return await self._send_email(config, title, message)
  532. elif provider.provider_type == "discord":
  533. return await self._send_discord(config, title, message, image_data=image_data)
  534. elif provider.provider_type == "webhook":
  535. return await self._send_webhook(
  536. config, title, message, image_data=image_data, event_type=event_type, variables=variables
  537. )
  538. elif provider.provider_type == "homeassistant":
  539. return await self._send_homeassistant(config, title, message, db=db)
  540. else:
  541. return False, f"Unknown provider type: {provider.provider_type}"
  542. except Exception as e:
  543. logger.exception("Error sending notification via %s", provider.provider_type)
  544. return False, str(e)
  545. async def _update_provider_status(
  546. self, db: AsyncSession, provider_id: int, success: bool, error: str | None = None
  547. ):
  548. """Update provider status after sending notification."""
  549. result = await db.execute(select(NotificationProvider).where(NotificationProvider.id == provider_id))
  550. provider = result.scalar_one_or_none()
  551. if provider:
  552. if success:
  553. provider.last_success = datetime.now(timezone.utc)
  554. else:
  555. provider.last_error = error
  556. provider.last_error_at = datetime.now(timezone.utc)
  557. await db.commit()
  558. async def _get_providers_for_event(
  559. self,
  560. db: AsyncSession,
  561. event_field: str,
  562. printer_id: int | None = None,
  563. ) -> list[NotificationProvider]:
  564. """Get all enabled providers that want a specific event type."""
  565. # Build the query dynamically based on event field
  566. query = select(NotificationProvider).where(
  567. NotificationProvider.enabled.is_(True),
  568. getattr(NotificationProvider, event_field).is_(True),
  569. )
  570. if printer_id is not None:
  571. query = query.where(
  572. (NotificationProvider.printer_id.is_(None)) | (NotificationProvider.printer_id == printer_id)
  573. )
  574. result = await db.execute(query)
  575. return list(result.scalars().all())
  576. async def _log_notification(
  577. self,
  578. db: AsyncSession,
  579. provider_id: int,
  580. event_type: str,
  581. title: str,
  582. message: str,
  583. success: bool,
  584. error_message: str | None = None,
  585. printer_id: int | None = None,
  586. printer_name: str | None = None,
  587. ):
  588. """Create a log entry for a sent notification."""
  589. try:
  590. log = NotificationLog(
  591. provider_id=provider_id,
  592. event_type=event_type,
  593. title=title,
  594. message=message,
  595. success=success,
  596. error_message=error_message,
  597. printer_id=printer_id,
  598. printer_name=printer_name,
  599. )
  600. db.add(log)
  601. await db.commit()
  602. except Exception as e:
  603. logger.warning("Failed to log notification: %s", e)
  604. # Don't fail the notification just because logging failed
  605. async def _send_to_providers(
  606. self,
  607. providers: list[NotificationProvider],
  608. title: str,
  609. message: str,
  610. db: AsyncSession,
  611. event_type: str = "unknown",
  612. printer_id: int | None = None,
  613. printer_name: str | None = None,
  614. force_immediate: bool = False,
  615. image_data: bytes | None = None,
  616. variables: dict | None = None,
  617. ):
  618. """Send notification to multiple providers and log the results.
  619. All notifications are always sent immediately. If digest mode is enabled,
  620. the notification is ALSO queued for the daily digest summary.
  621. """
  622. for provider in providers:
  623. try:
  624. # Always send notification immediately
  625. success, error = await self._send_to_provider(
  626. provider, title, message, db, image_data=image_data, event_type=event_type, variables=variables
  627. )
  628. # Also queue for digest if enabled (digest is a summary, not a queue)
  629. if provider.daily_digest_enabled and provider.daily_digest_time:
  630. await self._queue_for_digest(
  631. provider=provider,
  632. event_type=event_type,
  633. title=title,
  634. message=message,
  635. db=db,
  636. printer_id=printer_id,
  637. printer_name=printer_name,
  638. )
  639. await self._update_provider_status(db, provider.id, success, error if not success else None)
  640. await self._log_notification(
  641. db=db,
  642. provider_id=provider.id,
  643. event_type=event_type,
  644. title=title,
  645. message=message,
  646. success=success,
  647. error_message=error if not success else None,
  648. printer_id=printer_id,
  649. printer_name=printer_name,
  650. )
  651. if success:
  652. logger.info("Sent notification via %s", provider.name)
  653. else:
  654. logger.warning("Failed to send notification via %s: %s", provider.name, error)
  655. except Exception as e:
  656. logger.exception("Error sending notification via %s", provider.name)
  657. await self._update_provider_status(db, provider.id, False, str(e))
  658. await self._log_notification(
  659. db=db,
  660. provider_id=provider.id,
  661. event_type=event_type,
  662. title=title,
  663. message=message,
  664. success=False,
  665. error_message=str(e),
  666. printer_id=printer_id,
  667. printer_name=printer_name,
  668. )
  669. async def on_print_start(
  670. self,
  671. printer_id: int,
  672. printer_name: str,
  673. data: dict,
  674. db: AsyncSession,
  675. archive_data: dict | None = None,
  676. ):
  677. """Handle print start event - send notifications to relevant providers.
  678. Args:
  679. printer_id: The printer ID
  680. printer_name: The printer name
  681. data: MQTT event data with filename, subtask_name, remaining_time, raw_data
  682. db: Database session
  683. archive_data: Optional archive data with print_time_seconds from 3MF parsing
  684. """
  685. logger.info("on_print_start called for printer %s (%s)", printer_id, printer_name)
  686. providers = await self._get_providers_for_event(db, "on_print_start", printer_id)
  687. if not providers:
  688. logger.info("No notification providers configured for print_start event on printer %s", printer_id)
  689. return
  690. # Use subtask_name (project name) if available, otherwise use filename
  691. subtask_name = data.get("subtask_name")
  692. if subtask_name:
  693. # Replace underscores with spaces for readability
  694. filename = subtask_name.replace("_", " ")
  695. else:
  696. filename = self._clean_filename(data.get("filename", "Unknown"))
  697. # Priority for estimated_time:
  698. # 1. Archive's print_time_seconds from 3MF parsing (most reliable)
  699. # 2. MQTT remaining_time (may be 0 at print start)
  700. # 3. raw_data mc_remaining_time
  701. estimated_time = None
  702. # Try archive data first (from 3MF parsing - most reliable)
  703. if archive_data and archive_data.get("print_time_seconds"):
  704. estimated_time = archive_data["print_time_seconds"]
  705. logger.debug("Using print_time_seconds from archive: %s", estimated_time)
  706. # Fall back to MQTT remaining_time
  707. if estimated_time is None:
  708. estimated_time = data.get("remaining_time")
  709. if estimated_time:
  710. logger.debug("Using remaining_time from MQTT: %s", estimated_time)
  711. # Last resort: raw_data mc_remaining_time (in minutes, convert to seconds)
  712. if estimated_time is None:
  713. raw_time = data.get("raw_data", {}).get("mc_remaining_time")
  714. if raw_time:
  715. estimated_time = raw_time * 60
  716. logger.debug("Using mc_remaining_time from raw_data: %s", estimated_time)
  717. time_str = self._format_duration(estimated_time)
  718. eta_str = await self._format_eta(estimated_time, db)
  719. variables = {
  720. "printer": printer_name,
  721. "filename": filename,
  722. "estimated_time": time_str,
  723. "eta": eta_str,
  724. }
  725. # Extract image data for providers that support attachments (e.g. Pushover)
  726. image_data = None
  727. if archive_data:
  728. image_data = archive_data.get("image_data")
  729. logger.info("Found %s providers for print_start: %s", len(providers), [p.name for p in providers])
  730. title, message = await self._build_message_from_template(db, "print_start", variables)
  731. await self._send_to_providers(
  732. providers,
  733. title,
  734. message,
  735. db,
  736. "print_start",
  737. printer_id,
  738. printer_name,
  739. image_data=image_data,
  740. variables=variables,
  741. )
  742. async def on_print_complete(
  743. self,
  744. printer_id: int,
  745. printer_name: str,
  746. status: str,
  747. data: dict,
  748. db: AsyncSession,
  749. archive_data: dict | None = None,
  750. ):
  751. """Handle print complete event - send notifications to relevant providers."""
  752. logger.info("on_print_complete called for printer %s (%s), status=%s", printer_id, printer_name, status)
  753. # Determine event type based on status
  754. if status == "completed":
  755. event_field = "on_print_complete"
  756. event_type = "print_complete"
  757. elif status in ("failed",):
  758. event_field = "on_print_failed"
  759. event_type = "print_failed"
  760. elif status in ("aborted", "stopped", "cancelled"):
  761. event_field = "on_print_stopped"
  762. event_type = "print_stopped"
  763. else:
  764. logger.warning("Unknown print status '%s', defaulting to on_print_complete", status)
  765. event_field = "on_print_complete"
  766. event_type = "print_complete"
  767. providers = await self._get_providers_for_event(db, event_field, printer_id)
  768. if not providers:
  769. logger.info("No notification providers configured for %s event on printer %s", event_field, printer_id)
  770. return
  771. # Use subtask_name (project name) if available, otherwise use filename
  772. subtask_name = data.get("subtask_name")
  773. if subtask_name:
  774. filename = subtask_name.replace("_", " ")
  775. else:
  776. filename = self._clean_filename(data.get("filename", "Unknown"))
  777. variables = {
  778. "printer": printer_name,
  779. "filename": filename,
  780. "duration": "Unknown",
  781. "filament_grams": "Unknown",
  782. "reason": "Unknown",
  783. }
  784. if archive_data:
  785. if archive_data.get("print_time_seconds"):
  786. variables["duration"] = self._format_duration(archive_data["print_time_seconds"])
  787. if archive_data.get("actual_filament_grams"):
  788. variables["filament_grams"] = f"{archive_data['actual_filament_grams']:.1f}"
  789. if status == "failed" and archive_data.get("failure_reason"):
  790. variables["reason"] = archive_data["failure_reason"]
  791. if archive_data.get("finish_photo_url"):
  792. variables["finish_photo_url"] = archive_data["finish_photo_url"]
  793. # Build per-slot breakdown string with AMS info when available
  794. if archive_data.get("usage_results"):
  795. parts = []
  796. for u in archive_data["usage_results"]:
  797. ams_id = u.get("ams_id", 0)
  798. tray_id = u.get("tray_id", 0)
  799. material = u.get("material", "Unknown") or "Unknown"
  800. used = u.get("weight_used", 0)
  801. if ams_id >= 128:
  802. slot_label = "Ext"
  803. else:
  804. slot_label = f"AMS-{chr(65 + ams_id)} T{tray_id + 1}"
  805. parts.append(f"{slot_label} {material}: {used:.1f}g")
  806. variables["filament_details"] = " | ".join(parts)
  807. elif archive_data.get("filament_slots"):
  808. parts = []
  809. for slot in archive_data["filament_slots"]:
  810. ftype = slot.get("type", "Unknown") or "Unknown"
  811. used = slot.get("used_g", 0)
  812. parts.append(f"{ftype}: {used:.1f}g")
  813. variables["filament_details"] = " | ".join(parts)
  814. # Add progress for partial prints
  815. if archive_data.get("progress") is not None:
  816. variables["progress"] = str(archive_data["progress"])
  817. # Extract image data for providers that support attachments (e.g. Pushover)
  818. image_data = None
  819. if archive_data:
  820. image_data = archive_data.get("image_data")
  821. logger.info("Found %s providers for %s: %s", len(providers), event_field, [p.name for p in providers])
  822. title, message = await self._build_message_from_template(db, event_type, variables)
  823. await self._send_to_providers(
  824. providers,
  825. title,
  826. message,
  827. db,
  828. event_type,
  829. printer_id,
  830. printer_name,
  831. image_data=image_data,
  832. variables=variables,
  833. )
  834. async def on_print_progress(
  835. self,
  836. printer_id: int,
  837. printer_name: str,
  838. filename: str,
  839. progress: int,
  840. db: AsyncSession,
  841. remaining_time: int | None = None,
  842. image_data: bytes | None = None,
  843. ):
  844. """Handle print progress milestone (25%, 50%, 75%)."""
  845. providers = await self._get_providers_for_event(db, "on_print_progress", printer_id)
  846. if not providers:
  847. return
  848. eta_str = await self._format_eta(remaining_time, db)
  849. variables = {
  850. "printer": printer_name,
  851. "filename": self._clean_filename(filename),
  852. "progress": str(progress),
  853. "remaining_time": self._format_duration(remaining_time) if remaining_time else "Unknown",
  854. "eta": eta_str,
  855. }
  856. title, message = await self._build_message_from_template(db, "print_progress", variables)
  857. await self._send_to_providers(
  858. providers,
  859. title,
  860. message,
  861. db,
  862. "print_progress",
  863. printer_id,
  864. printer_name,
  865. image_data=image_data,
  866. variables=variables,
  867. )
  868. async def on_print_missing_spool_assignment(
  869. self,
  870. printer_id: int,
  871. printer_name: str,
  872. missing_slots: list[dict[str, str]],
  873. db: AsyncSession,
  874. ):
  875. """Handle print-start event when required trays are missing spool assignments."""
  876. if not missing_slots:
  877. return
  878. providers = await self._get_providers_for_event(db, "on_print_missing_spool_assignment", printer_id)
  879. if not providers:
  880. return
  881. missing_slot_names = ", ".join(slot.get("slot", "Unknown") for slot in missing_slots)
  882. detail_lines = []
  883. for slot in missing_slots:
  884. slot_name = slot.get("slot", "Unknown")
  885. profile = slot.get("profile", "Unknown")
  886. detail_lines.append(f"- {slot_name}: {profile}")
  887. missing_profile_details = "\n".join(detail_lines)
  888. variables = {
  889. "printer": printer_name,
  890. "missing_slots": missing_slot_names,
  891. "missing_slot_details": missing_profile_details,
  892. }
  893. title, message = await self._build_message_from_template(db, "print_missing_spool_assignment", variables)
  894. await self._send_to_providers(
  895. providers,
  896. title,
  897. message,
  898. db,
  899. "print_missing_spool_assignment",
  900. printer_id,
  901. printer_name,
  902. force_immediate=True,
  903. variables=variables,
  904. )
  905. async def on_printer_offline(self, printer_id: int, printer_name: str, db: AsyncSession):
  906. """Handle printer offline event."""
  907. providers = await self._get_providers_for_event(db, "on_printer_offline", printer_id)
  908. if not providers:
  909. return
  910. variables = {"printer": printer_name}
  911. title, message = await self._build_message_from_template(db, "printer_offline", variables)
  912. await self._send_to_providers(
  913. providers, title, message, db, "printer_offline", printer_id, printer_name, variables=variables
  914. )
  915. async def on_printer_error(
  916. self,
  917. printer_id: int,
  918. printer_name: str,
  919. error_type: str,
  920. db: AsyncSession,
  921. error_detail: str | None = None,
  922. image_data: bytes | None = None,
  923. ):
  924. """Handle printer error event (AMS issues, etc.)."""
  925. providers = await self._get_providers_for_event(db, "on_printer_error", printer_id)
  926. if not providers:
  927. return
  928. variables = {
  929. "printer": printer_name,
  930. "error_type": error_type,
  931. "error_detail": error_detail or "No details available",
  932. }
  933. title, message = await self._build_message_from_template(db, "printer_error", variables)
  934. await self._send_to_providers(
  935. providers,
  936. title,
  937. message,
  938. db,
  939. "printer_error",
  940. printer_id,
  941. printer_name,
  942. image_data=image_data,
  943. variables=variables,
  944. )
  945. async def on_plate_not_empty(
  946. self,
  947. printer_id: int,
  948. printer_name: str,
  949. db: AsyncSession,
  950. difference_percent: float | None = None,
  951. ):
  952. """Handle plate not empty event - objects detected on build plate before print."""
  953. providers = await self._get_providers_for_event(db, "on_plate_not_empty", printer_id)
  954. if not providers:
  955. return
  956. variables = {
  957. "printer": printer_name,
  958. "difference_percent": f"{difference_percent:.1f}" if difference_percent else "N/A",
  959. }
  960. title, message = await self._build_message_from_template(db, "plate_not_empty", variables)
  961. await self._send_to_providers(
  962. providers,
  963. title,
  964. message,
  965. db,
  966. "plate_not_empty",
  967. printer_id,
  968. printer_name,
  969. force_immediate=True,
  970. variables=variables,
  971. )
  972. async def on_filament_low(
  973. self,
  974. printer_id: int,
  975. printer_name: str,
  976. slot: int,
  977. remaining_percent: int,
  978. db: AsyncSession,
  979. color: str | None = None,
  980. ):
  981. """Handle low filament event."""
  982. providers = await self._get_providers_for_event(db, "on_filament_low", printer_id)
  983. if not providers:
  984. return
  985. variables = {
  986. "printer": printer_name,
  987. "slot": str(slot),
  988. "remaining_percent": str(remaining_percent),
  989. "color": color or "",
  990. }
  991. title, message = await self._build_message_from_template(db, "filament_low", variables)
  992. await self._send_to_providers(
  993. providers, title, message, db, "filament_low", printer_id, printer_name, variables=variables
  994. )
  995. async def on_maintenance_due(
  996. self,
  997. printer_id: int,
  998. printer_name: str,
  999. maintenance_items: list[dict],
  1000. db: AsyncSession,
  1001. ):
  1002. """Handle maintenance due event - sends notification when maintenance is due or warning."""
  1003. if not maintenance_items:
  1004. return
  1005. providers = await self._get_providers_for_event(db, "on_maintenance_due", printer_id)
  1006. if not providers:
  1007. logger.info("No notification providers configured for maintenance_due event on printer %s", printer_id)
  1008. return
  1009. # Format maintenance items list
  1010. items_list = []
  1011. for item in maintenance_items:
  1012. status = "OVERDUE" if item.get("is_due") else "Soon"
  1013. items_list.append(f"- {item['name']} ({status})")
  1014. items_str = "\n".join(items_list)
  1015. variables = {
  1016. "printer": printer_name,
  1017. "items": items_str,
  1018. }
  1019. logger.info("Found %s providers for maintenance_due: %s", len(providers), [p.name for p in providers])
  1020. title, message = await self._build_message_from_template(db, "maintenance_due", variables)
  1021. await self._send_to_providers(
  1022. providers, title, message, db, "maintenance_due", printer_id, printer_name, variables=variables
  1023. )
  1024. async def on_ams_humidity_high(
  1025. self,
  1026. printer_id: int,
  1027. printer_name: str,
  1028. ams_label: str,
  1029. humidity: float,
  1030. threshold: float,
  1031. db: AsyncSession,
  1032. ):
  1033. """Handle AMS high humidity alarm event. Always sends immediately (bypasses digest)."""
  1034. providers = await self._get_providers_for_event(db, "on_ams_humidity_high", printer_id)
  1035. if not providers:
  1036. return
  1037. variables = {
  1038. "printer": printer_name,
  1039. "ams_label": ams_label,
  1040. "humidity": f"{humidity:.0f}",
  1041. "threshold": f"{threshold:.0f}",
  1042. }
  1043. title, message = await self._build_message_from_template(db, "ams_humidity_high", variables)
  1044. # Alarms always send immediately, bypassing digest mode
  1045. await self._send_to_providers(
  1046. providers,
  1047. title,
  1048. message,
  1049. db,
  1050. "ams_humidity_high",
  1051. printer_id,
  1052. printer_name,
  1053. force_immediate=True,
  1054. variables=variables,
  1055. )
  1056. async def on_ams_temperature_high(
  1057. self,
  1058. printer_id: int,
  1059. printer_name: str,
  1060. ams_label: str,
  1061. temperature: float,
  1062. threshold: float,
  1063. db: AsyncSession,
  1064. ):
  1065. """Handle AMS high temperature alarm event. Always sends immediately (bypasses digest)."""
  1066. providers = await self._get_providers_for_event(db, "on_ams_temperature_high", printer_id)
  1067. if not providers:
  1068. return
  1069. variables = {
  1070. "printer": printer_name,
  1071. "ams_label": ams_label,
  1072. "temperature": f"{temperature:.1f}",
  1073. "threshold": f"{threshold:.1f}",
  1074. }
  1075. title, message = await self._build_message_from_template(db, "ams_temperature_high", variables)
  1076. # Alarms always send immediately, bypassing digest mode
  1077. await self._send_to_providers(
  1078. providers,
  1079. title,
  1080. message,
  1081. db,
  1082. "ams_temperature_high",
  1083. printer_id,
  1084. printer_name,
  1085. force_immediate=True,
  1086. variables=variables,
  1087. )
  1088. async def on_ams_ht_humidity_high(
  1089. self,
  1090. printer_id: int,
  1091. printer_name: str,
  1092. ams_label: str,
  1093. humidity: float,
  1094. threshold: float,
  1095. db: AsyncSession,
  1096. ):
  1097. """Handle AMS-HT high humidity alarm event. Always sends immediately (bypasses digest)."""
  1098. providers = await self._get_providers_for_event(db, "on_ams_ht_humidity_high", printer_id)
  1099. if not providers:
  1100. return
  1101. variables = {
  1102. "printer": printer_name,
  1103. "ams_label": ams_label,
  1104. "humidity": f"{humidity:.0f}",
  1105. "threshold": f"{threshold:.0f}",
  1106. }
  1107. # Use the same template as regular AMS (can create separate templates later if needed)
  1108. title, message = await self._build_message_from_template(db, "ams_humidity_high", variables)
  1109. # Alarms always send immediately, bypassing digest mode
  1110. await self._send_to_providers(
  1111. providers,
  1112. title,
  1113. message,
  1114. db,
  1115. "ams_ht_humidity_high",
  1116. printer_id,
  1117. printer_name,
  1118. force_immediate=True,
  1119. variables=variables,
  1120. )
  1121. async def on_ams_ht_temperature_high(
  1122. self,
  1123. printer_id: int,
  1124. printer_name: str,
  1125. ams_label: str,
  1126. temperature: float,
  1127. threshold: float,
  1128. db: AsyncSession,
  1129. ):
  1130. """Handle AMS-HT high temperature alarm event. Always sends immediately (bypasses digest)."""
  1131. providers = await self._get_providers_for_event(db, "on_ams_ht_temperature_high", printer_id)
  1132. if not providers:
  1133. return
  1134. variables = {
  1135. "printer": printer_name,
  1136. "ams_label": ams_label,
  1137. "temperature": f"{temperature:.1f}",
  1138. "threshold": f"{threshold:.1f}",
  1139. }
  1140. # Use the same template as regular AMS (can create separate templates later if needed)
  1141. title, message = await self._build_message_from_template(db, "ams_temperature_high", variables)
  1142. # Alarms always send immediately, bypassing digest mode
  1143. await self._send_to_providers(
  1144. providers,
  1145. title,
  1146. message,
  1147. db,
  1148. "ams_ht_temperature_high",
  1149. printer_id,
  1150. printer_name,
  1151. force_immediate=True,
  1152. variables=variables,
  1153. )
  1154. async def on_bed_cooled(
  1155. self,
  1156. printer_id: int,
  1157. printer_name: str,
  1158. bed_temp: float,
  1159. threshold: float,
  1160. filename: str,
  1161. db: AsyncSession,
  1162. ):
  1163. """Handle bed cooled event - bed temperature dropped below threshold after print."""
  1164. providers = await self._get_providers_for_event(db, "on_bed_cooled", printer_id)
  1165. if not providers:
  1166. return
  1167. variables = {
  1168. "printer": printer_name,
  1169. "bed_temp": f"{bed_temp:.0f}",
  1170. "threshold": f"{threshold:.0f}",
  1171. "filename": self._clean_filename(filename) if filename else "Unknown",
  1172. }
  1173. title, message = await self._build_message_from_template(db, "bed_cooled", variables)
  1174. await self._send_to_providers(
  1175. providers, title, message, db, "bed_cooled", printer_id, printer_name, variables=variables
  1176. )
  1177. async def on_first_layer_complete(
  1178. self,
  1179. printer_id: int,
  1180. printer_name: str,
  1181. filename: str,
  1182. total_layers: int,
  1183. db: AsyncSession,
  1184. image_data: bytes | None = None,
  1185. ):
  1186. """Handle first layer complete event."""
  1187. providers = await self._get_providers_for_event(db, "on_first_layer_complete", printer_id)
  1188. if not providers:
  1189. return
  1190. variables = {
  1191. "printer": printer_name,
  1192. "filename": self._clean_filename(filename),
  1193. "total_layers": str(total_layers),
  1194. }
  1195. title, message = await self._build_message_from_template(db, "first_layer_complete", variables)
  1196. await self._send_to_providers(
  1197. providers,
  1198. title,
  1199. message,
  1200. db,
  1201. "first_layer_complete",
  1202. printer_id,
  1203. printer_name,
  1204. image_data=image_data,
  1205. variables=variables,
  1206. )
  1207. def clear_template_cache(self):
  1208. """Clear the template cache. Call this when templates are updated."""
  1209. self._template_cache.clear()
  1210. async def send_user_print_email(
  1211. self,
  1212. event_type: str,
  1213. created_by_id: int | None,
  1214. printer_name: str,
  1215. filename: str,
  1216. db: AsyncSession,
  1217. ) -> None:
  1218. """Send a print event email notification to the user who submitted the job.
  1219. Args:
  1220. event_type: 'user_print_start', 'user_print_complete', 'user_print_failed', or 'user_print_stopped'
  1221. created_by_id: User ID who submitted the print job (from archive)
  1222. printer_name: Name of the printer
  1223. filename: Raw filename or subtask name
  1224. db: Database session
  1225. """
  1226. if created_by_id is None:
  1227. logger.debug("[EMAIL] Skipping user print email (%s): no created_by_id", event_type)
  1228. return
  1229. try:
  1230. # Check if advanced auth is enabled - required for user email notifications
  1231. from backend.app.models.settings import Settings
  1232. result = await db.execute(select(Settings).where(Settings.key == "advanced_auth_enabled"))
  1233. setting = result.scalar_one_or_none()
  1234. if not setting or setting.value.lower() != "true":
  1235. logger.debug("[EMAIL] Skipping user print email (%s): advanced_auth not enabled", event_type)
  1236. return
  1237. # Check if user notifications are enabled (admin-controlled toggle)
  1238. notif_enabled_result = await db.execute(
  1239. select(Settings).where(Settings.key == "user_notifications_enabled")
  1240. )
  1241. notif_enabled_setting = notif_enabled_result.scalar_one_or_none()
  1242. if notif_enabled_setting and notif_enabled_setting.value.lower() == "false":
  1243. logger.debug("[EMAIL] Skipping user print email (%s): user_notifications_enabled is false", event_type)
  1244. return
  1245. # Check SMTP settings are configured - required for sending emails
  1246. from backend.app.services.email_service import get_smtp_settings, send_user_print_notification
  1247. smtp_settings = await get_smtp_settings(db)
  1248. if not smtp_settings:
  1249. logger.debug("[EMAIL] Skipping user print email (%s): SMTP settings not configured", event_type)
  1250. return
  1251. # Load user preferences
  1252. from backend.app.models.user import User
  1253. from backend.app.models.user_email_pref import UserEmailPreference
  1254. user_result = await db.execute(select(User).where(User.id == created_by_id))
  1255. user = user_result.scalar_one_or_none()
  1256. if user is None or not user.email:
  1257. logger.debug(
  1258. "[EMAIL] Skipping user print email (%s): user %s not found or has no email address",
  1259. event_type,
  1260. created_by_id,
  1261. )
  1262. return
  1263. # Load user's notification preferences
  1264. pref_result = await db.execute(
  1265. select(UserEmailPreference).where(UserEmailPreference.user_id == created_by_id)
  1266. )
  1267. pref = pref_result.scalar_one_or_none()
  1268. # Determine if this event type should be sent
  1269. should_send = False
  1270. if event_type == "user_print_start":
  1271. should_send = pref is None or pref.notify_print_start
  1272. elif event_type == "user_print_complete":
  1273. should_send = pref is None or pref.notify_print_complete
  1274. elif event_type == "user_print_failed":
  1275. should_send = pref is None or pref.notify_print_failed
  1276. elif event_type == "user_print_stopped":
  1277. should_send = pref is None or pref.notify_print_stopped
  1278. if not should_send:
  1279. logger.debug(
  1280. "[EMAIL] Skipping user print email (%s): user %s has notifications disabled for this event",
  1281. event_type,
  1282. created_by_id,
  1283. )
  1284. return
  1285. logger.info(
  1286. "[EMAIL] Sending user print email: event=%s, user=%s (%s), printer=%s, file=%s",
  1287. event_type,
  1288. user.username,
  1289. user.email,
  1290. printer_name,
  1291. filename,
  1292. )
  1293. # Build variables
  1294. variables = {
  1295. "printer": printer_name,
  1296. "filename": self._clean_filename(filename),
  1297. }
  1298. # Send the email
  1299. await send_user_print_notification(
  1300. db=db,
  1301. event_type=event_type,
  1302. user_email=user.email,
  1303. username=user.username,
  1304. variables=variables,
  1305. )
  1306. logger.info("[EMAIL] User print email sent: event=%s → %s", event_type, user.email)
  1307. except Exception as e:
  1308. logger.warning("Failed to send user print email notification: %s", e, exc_info=True)
  1309. # ==================== Queue Notifications ====================
  1310. async def on_queue_job_added(
  1311. self,
  1312. job_name: str,
  1313. target: str,
  1314. db: AsyncSession,
  1315. printer_id: int | None = None,
  1316. printer_name: str | None = None,
  1317. ):
  1318. """Handle queue job added event."""
  1319. providers = await self._get_providers_for_event(db, "on_queue_job_added", printer_id)
  1320. if not providers:
  1321. return
  1322. variables = {
  1323. "job_name": job_name,
  1324. "target": target, # e.g., "Printer1" or "Any X1C"
  1325. "printer": printer_name or target,
  1326. }
  1327. title, message = await self._build_message_from_template(db, "queue_job_added", variables)
  1328. await self._send_to_providers(
  1329. providers, title, message, db, "queue_job_added", printer_id, printer_name, variables=variables
  1330. )
  1331. async def on_queue_job_assigned(
  1332. self,
  1333. job_name: str,
  1334. printer_id: int,
  1335. printer_name: str,
  1336. target_model: str,
  1337. db: AsyncSession,
  1338. ):
  1339. """Handle model-based job assigned to printer event."""
  1340. providers = await self._get_providers_for_event(db, "on_queue_job_assigned", printer_id)
  1341. if not providers:
  1342. return
  1343. variables = {
  1344. "job_name": job_name,
  1345. "printer": printer_name,
  1346. "target_model": target_model,
  1347. }
  1348. title, message = await self._build_message_from_template(db, "queue_job_assigned", variables)
  1349. await self._send_to_providers(
  1350. providers, title, message, db, "queue_job_assigned", printer_id, printer_name, variables=variables
  1351. )
  1352. async def on_queue_job_started(
  1353. self,
  1354. job_name: str,
  1355. printer_id: int,
  1356. printer_name: str,
  1357. db: AsyncSession,
  1358. estimated_time: int | None = None,
  1359. ):
  1360. """Handle queue job started printing event."""
  1361. providers = await self._get_providers_for_event(db, "on_queue_job_started", printer_id)
  1362. if not providers:
  1363. return
  1364. eta_str = await self._format_eta(estimated_time, db)
  1365. variables = {
  1366. "job_name": job_name,
  1367. "printer": printer_name,
  1368. "estimated_time": self._format_duration(estimated_time),
  1369. "eta": eta_str,
  1370. }
  1371. title, message = await self._build_message_from_template(db, "queue_job_started", variables)
  1372. await self._send_to_providers(
  1373. providers, title, message, db, "queue_job_started", printer_id, printer_name, variables=variables
  1374. )
  1375. async def on_queue_job_waiting(
  1376. self,
  1377. job_name: str,
  1378. target_model: str,
  1379. waiting_reason: str,
  1380. db: AsyncSession,
  1381. ):
  1382. """Handle job waiting for filament event."""
  1383. providers = await self._get_providers_for_event(db, "on_queue_job_waiting", None)
  1384. if not providers:
  1385. return
  1386. variables = {
  1387. "job_name": job_name,
  1388. "target_model": target_model,
  1389. "waiting_reason": waiting_reason,
  1390. }
  1391. title, message = await self._build_message_from_template(db, "queue_job_waiting", variables)
  1392. await self._send_to_providers(providers, title, message, db, "queue_job_waiting", variables=variables)
  1393. async def on_queue_job_skipped(
  1394. self,
  1395. job_name: str,
  1396. printer_id: int,
  1397. printer_name: str,
  1398. reason: str,
  1399. db: AsyncSession,
  1400. ):
  1401. """Handle job skipped event (e.g., previous print failed)."""
  1402. providers = await self._get_providers_for_event(db, "on_queue_job_skipped", printer_id)
  1403. if not providers:
  1404. return
  1405. variables = {
  1406. "job_name": job_name,
  1407. "printer": printer_name,
  1408. "reason": reason,
  1409. }
  1410. title, message = await self._build_message_from_template(db, "queue_job_skipped", variables)
  1411. await self._send_to_providers(
  1412. providers, title, message, db, "queue_job_skipped", printer_id, printer_name, variables=variables
  1413. )
  1414. async def on_queue_job_failed(
  1415. self,
  1416. job_name: str,
  1417. printer_id: int | None,
  1418. printer_name: str | None,
  1419. reason: str,
  1420. db: AsyncSession,
  1421. ):
  1422. """Handle job failed to start event (upload error, etc.)."""
  1423. providers = await self._get_providers_for_event(db, "on_queue_job_failed", printer_id)
  1424. if not providers:
  1425. return
  1426. variables = {
  1427. "job_name": job_name,
  1428. "printer": printer_name or "Unknown",
  1429. "reason": reason,
  1430. }
  1431. title, message = await self._build_message_from_template(db, "queue_job_failed", variables)
  1432. await self._send_to_providers(
  1433. providers, title, message, db, "queue_job_failed", printer_id, printer_name, variables=variables
  1434. )
  1435. async def on_queue_completed(
  1436. self,
  1437. completed_count: int,
  1438. db: AsyncSession,
  1439. ):
  1440. """Handle all queue jobs completed event."""
  1441. providers = await self._get_providers_for_event(db, "on_queue_completed", None)
  1442. if not providers:
  1443. return
  1444. variables = {
  1445. "completed_count": str(completed_count),
  1446. }
  1447. title, message = await self._build_message_from_template(db, "queue_completed", variables)
  1448. await self._send_to_providers(providers, title, message, db, "queue_completed", variables=variables)
  1449. async def _queue_for_digest(
  1450. self,
  1451. provider: NotificationProvider,
  1452. event_type: str,
  1453. title: str,
  1454. message: str,
  1455. db: AsyncSession,
  1456. printer_id: int | None = None,
  1457. printer_name: str | None = None,
  1458. ):
  1459. """Queue a notification for later delivery in the daily digest."""
  1460. try:
  1461. queue_entry = NotificationDigestQueue(
  1462. provider_id=provider.id,
  1463. event_type=event_type,
  1464. title=title,
  1465. message=message,
  1466. printer_id=printer_id,
  1467. printer_name=printer_name,
  1468. )
  1469. db.add(queue_entry)
  1470. await db.commit()
  1471. logger.info("Queued notification for digest: %s for provider %s", event_type, provider.name)
  1472. except Exception as e:
  1473. logger.warning("Failed to queue notification for digest: %s", e)
  1474. async def send_digest(self, provider_id: int):
  1475. """Send all queued notifications as a single digest for a provider."""
  1476. from backend.app.core.database import async_session
  1477. async with async_session() as db:
  1478. # Get the provider
  1479. result = await db.execute(select(NotificationProvider).where(NotificationProvider.id == provider_id))
  1480. provider = result.scalar_one_or_none()
  1481. if not provider or not provider.enabled:
  1482. return
  1483. # Get all queued notifications for this provider
  1484. result = await db.execute(
  1485. select(NotificationDigestQueue)
  1486. .where(NotificationDigestQueue.provider_id == provider_id)
  1487. .order_by(NotificationDigestQueue.created_at)
  1488. )
  1489. queue_entries = list(result.scalars().all())
  1490. if not queue_entries:
  1491. logger.debug("No queued notifications for provider %s", provider.name)
  1492. return
  1493. # Build digest message
  1494. title = f"Daily Digest - {len(queue_entries)} Events"
  1495. # Group by event type
  1496. events_by_type: dict[str, list] = {}
  1497. for entry in queue_entries:
  1498. if entry.event_type not in events_by_type:
  1499. events_by_type[entry.event_type] = []
  1500. events_by_type[entry.event_type].append(entry)
  1501. # Format the digest body
  1502. body_parts = []
  1503. for event_type, entries in events_by_type.items():
  1504. event_label = event_type.replace("_", " ").title()
  1505. body_parts.append(f"== {event_label} ({len(entries)}) ==")
  1506. for entry in entries:
  1507. time_str = entry.created_at.strftime("%H:%M")
  1508. printer_info = f"[{entry.printer_name}] " if entry.printer_name else ""
  1509. body_parts.append(f" {time_str} {printer_info}{entry.title}")
  1510. body_parts.append("")
  1511. body = "\n".join(body_parts)
  1512. # Send the digest
  1513. success, error = await self._send_to_provider(provider, title, body, db)
  1514. # Log the digest
  1515. await self._log_notification(
  1516. db=db,
  1517. provider_id=provider.id,
  1518. event_type="daily_digest",
  1519. title=title,
  1520. message=body,
  1521. success=success,
  1522. error_message=error if not success else None,
  1523. )
  1524. # Clear the queue
  1525. for entry in queue_entries:
  1526. await db.delete(entry)
  1527. await db.commit()
  1528. if success:
  1529. logger.info("Sent daily digest with %s events to %s", len(queue_entries), provider.name)
  1530. else:
  1531. logger.warning("Failed to send daily digest to %s: %s", provider.name, error)
  1532. async def check_and_send_digests(self):
  1533. """Check all providers and send digests if it's their scheduled time."""
  1534. from backend.app.core.database import async_session
  1535. current_time = datetime.now().strftime("%H:%M")
  1536. # Avoid duplicate checks within the same minute
  1537. if current_time == self._last_digest_check:
  1538. return
  1539. self._last_digest_check = current_time
  1540. async with async_session() as db:
  1541. # Find all providers with digest enabled at this time
  1542. result = await db.execute(
  1543. select(NotificationProvider).where(
  1544. NotificationProvider.enabled.is_(True),
  1545. NotificationProvider.daily_digest_enabled.is_(True),
  1546. NotificationProvider.daily_digest_time == current_time,
  1547. )
  1548. )
  1549. providers = result.scalars().all()
  1550. for provider in providers:
  1551. try:
  1552. await self.send_digest(provider.id)
  1553. except Exception as e:
  1554. logger.error("Error sending digest for provider %s: %s", provider.id, e)
  1555. def start_digest_scheduler(self):
  1556. """Start the background scheduler for daily digest notifications."""
  1557. if self._digest_scheduler_task is None:
  1558. self._digest_scheduler_task = asyncio.create_task(self._digest_scheduler_loop())
  1559. logger.info("Notification digest scheduler started")
  1560. def stop_digest_scheduler(self):
  1561. """Stop the background scheduler for daily digests."""
  1562. if self._digest_scheduler_task:
  1563. self._digest_scheduler_task.cancel()
  1564. self._digest_scheduler_task = None
  1565. logger.info("Notification digest scheduler stopped")
  1566. async def _digest_scheduler_loop(self):
  1567. """Background loop that checks for scheduled digests every minute."""
  1568. while True:
  1569. try:
  1570. await self.check_and_send_digests()
  1571. except Exception as e:
  1572. logger.error("Error in digest scheduler: %s", e)
  1573. # Wait until the next minute
  1574. await asyncio.sleep(60)
  1575. # Global instance
  1576. notification_service = NotificationService()