|
| 1 | +"""Operator notifications — pings to the maintainer's Discord server. |
| 2 | +
|
| 3 | +NOT the user-facing webhooks product: this is the internal channel that |
| 4 | +tells the operator a visitor submitted the contact form or reported a |
| 5 | +URL. It happens to deliver over Discord webhook URLs, which is why it |
| 6 | +used to live in ``infrastructure/webhook/`` — that name is reserved for |
| 7 | +the real webhooks system. |
| 8 | +
|
| 9 | +``OpsNotifier`` is semantic: callers state WHAT happened; the |
| 10 | +implementation owns channel routing and every Discord specific (embed |
| 11 | +structure, colors, footer). Send failures return ``False`` and never |
| 12 | +raise — callers decide whether a failed ping is fatal. |
| 13 | +""" |
| 14 | + |
| 15 | +from __future__ import annotations |
| 16 | + |
| 17 | +from datetime import datetime, timezone |
| 18 | +from typing import Any, Protocol |
| 19 | + |
| 20 | +from infrastructure.http_client import HttpClient |
| 21 | +from infrastructure.logging import get_logger |
| 22 | + |
| 23 | +log = get_logger(__name__) |
| 24 | + |
| 25 | +_FOOTER = { |
| 26 | + "text": "spoo-me", |
| 27 | + "icon_url": "https://spoo.me/static/images/favicon.png", |
| 28 | +} |
| 29 | +_CONTACT_COLOR = 9103397 |
| 30 | +_REPORT_COLOR = 14177041 |
| 31 | + |
| 32 | +# Summary embed: list at most this many targets, then "… and N more". |
| 33 | +_SUMMARY_MAX_LISTED = 10 |
| 34 | +_SUMMARY_LINE_MAX = 80 |
| 35 | + |
| 36 | + |
| 37 | +class OpsNotifier(Protocol): |
| 38 | + async def contact_message(self, email: str, message: str) -> bool: ... |
| 39 | + |
| 40 | + async def url_report( |
| 41 | + self, short_code: str, reason: str, ip_address: str, app_url: str |
| 42 | + ) -> bool: ... |
| 43 | + |
| 44 | + async def report_summary( |
| 45 | + self, |
| 46 | + *, |
| 47 | + submission_id: str, |
| 48 | + source: str, |
| 49 | + authenticated: bool, |
| 50 | + accepted: list[tuple[str, str]], |
| 51 | + rejected_count: int, |
| 52 | + reporter_email: str | None, |
| 53 | + reporter_org: str | None, |
| 54 | + ip: str, |
| 55 | + now: datetime, |
| 56 | + ) -> bool: ... |
| 57 | + |
| 58 | + |
| 59 | +class DiscordOpsNotifier: |
| 60 | + """Discord implementation — routes each notification to its channel |
| 61 | + (contact vs reports) and builds the embeds. |
| 62 | +
|
| 63 | + Embed structures are preserved exactly from the previous per-service |
| 64 | + builders (originally ``utils/contact_utils.py``). |
| 65 | + """ |
| 66 | + |
| 67 | + def __init__( |
| 68 | + self, contact_url: str, report_url: str, http_client: HttpClient |
| 69 | + ) -> None: |
| 70 | + self._contact_url = contact_url |
| 71 | + self._report_url = report_url |
| 72 | + self._http = http_client |
| 73 | + |
| 74 | + # ── OpsNotifier ─────────────────────────────────────────────────────────── |
| 75 | + |
| 76 | + async def contact_message(self, email: str, message: str) -> bool: |
| 77 | + payload = { |
| 78 | + "embeds": [ |
| 79 | + { |
| 80 | + "title": "New Contact Message ✉️", |
| 81 | + "color": _CONTACT_COLOR, |
| 82 | + "fields": [ |
| 83 | + {"name": "Email", "value": f"```{email}```"}, |
| 84 | + {"name": "Message", "value": f"```{message}```"}, |
| 85 | + ], |
| 86 | + "timestamp": datetime.now(timezone.utc).isoformat(), |
| 87 | + "footer": _FOOTER, |
| 88 | + } |
| 89 | + ] |
| 90 | + } |
| 91 | + return await self._deliver(self._contact_url, payload, kind="contact_message") |
| 92 | + |
| 93 | + async def url_report( |
| 94 | + self, short_code: str, reason: str, ip_address: str, app_url: str |
| 95 | + ) -> bool: |
| 96 | + payload = { |
| 97 | + "embeds": [ |
| 98 | + { |
| 99 | + "title": f"URL Report for `{short_code}`", |
| 100 | + "color": _REPORT_COLOR, |
| 101 | + "url": f"{app_url}stats/{short_code}", |
| 102 | + "fields": [ |
| 103 | + {"name": "Short Code", "value": f"```{short_code}```"}, |
| 104 | + {"name": "Reason", "value": f"```{reason}```"}, |
| 105 | + {"name": "IP Address", "value": f"```{ip_address}```"}, |
| 106 | + ], |
| 107 | + "timestamp": datetime.now(timezone.utc).isoformat(), |
| 108 | + "footer": _FOOTER, |
| 109 | + } |
| 110 | + ] |
| 111 | + } |
| 112 | + return await self._deliver(self._report_url, payload, kind="url_report") |
| 113 | + |
| 114 | + async def report_summary( |
| 115 | + self, |
| 116 | + *, |
| 117 | + submission_id: str, |
| 118 | + source: str, |
| 119 | + authenticated: bool, |
| 120 | + accepted: list[tuple[str, str]], |
| 121 | + rejected_count: int, |
| 122 | + reporter_email: str | None, |
| 123 | + reporter_org: str | None, |
| 124 | + ip: str, |
| 125 | + now: datetime, |
| 126 | + ) -> bool: |
| 127 | + """ONE embed per submission — counts, source, up to |
| 128 | + ``_SUMMARY_MAX_LISTED`` targets with reasons, submission id. |
| 129 | +
|
| 130 | + ``accepted`` carries ``(display_target, reason)`` pairs; ``now`` |
| 131 | + is the submission timestamp already stamped on the audit record, |
| 132 | + so the embed and the record can never disagree. |
| 133 | + """ |
| 134 | + fields: list[dict[str, Any]] = [ |
| 135 | + {"name": "Submission ID", "value": f"```{submission_id}```"}, |
| 136 | + { |
| 137 | + "name": "Source", |
| 138 | + "value": ( |
| 139 | + f"```{source} · " |
| 140 | + f"{'authenticated' if authenticated else 'anonymous'}```" |
| 141 | + ), |
| 142 | + }, |
| 143 | + { |
| 144 | + "name": "Accepted / Rejected", |
| 145 | + "value": f"```{len(accepted)} / {rejected_count}```", |
| 146 | + }, |
| 147 | + ] |
| 148 | + |
| 149 | + if accepted: |
| 150 | + lines = [] |
| 151 | + for display, reason in accepted[:_SUMMARY_MAX_LISTED]: |
| 152 | + line = f"{display} — {reason}" |
| 153 | + if len(line) > _SUMMARY_LINE_MAX: |
| 154 | + line = line[: _SUMMARY_LINE_MAX - 1] + "…" |
| 155 | + lines.append(line) |
| 156 | + overflow = len(accepted) - _SUMMARY_MAX_LISTED |
| 157 | + if overflow > 0: |
| 158 | + lines.append(f"… and {overflow} more") |
| 159 | + fields.append( |
| 160 | + {"name": "Reported Links", "value": "```" + "\n".join(lines) + "```"} |
| 161 | + ) |
| 162 | + |
| 163 | + if reporter_email or reporter_org: |
| 164 | + fields.append( |
| 165 | + { |
| 166 | + "name": "Reporter", |
| 167 | + "value": f"```{reporter_email or '—'} · {reporter_org or '—'}```", |
| 168 | + } |
| 169 | + ) |
| 170 | + |
| 171 | + fields.append({"name": "IP Address", "value": f"```{ip}```"}) |
| 172 | + |
| 173 | + payload = { |
| 174 | + "embeds": [ |
| 175 | + { |
| 176 | + "title": "New URL Report Submission", |
| 177 | + "color": _REPORT_COLOR, |
| 178 | + "fields": fields, |
| 179 | + "timestamp": now.isoformat(), |
| 180 | + "footer": _FOOTER, |
| 181 | + } |
| 182 | + ] |
| 183 | + } |
| 184 | + return await self._deliver(self._report_url, payload, kind="report_summary") |
| 185 | + |
| 186 | + # ── Delivery ────────────────────────────────────────────────────────────── |
| 187 | + |
| 188 | + async def _deliver(self, url: str, payload: dict[str, Any], *, kind: str) -> bool: |
| 189 | + if not url: |
| 190 | + log.warning("ops_notify_not_configured", kind=kind) |
| 191 | + return False |
| 192 | + try: |
| 193 | + response = await self._http.post(url, json=payload) |
| 194 | + if response.status_code in (200, 204): |
| 195 | + return True |
| 196 | + log.warning( |
| 197 | + "ops_notify_failed", |
| 198 | + kind=kind, |
| 199 | + status_code=response.status_code, |
| 200 | + response_text=response.text[:200], |
| 201 | + ) |
| 202 | + return False |
| 203 | + except Exception as e: |
| 204 | + log.error( |
| 205 | + "ops_notify_request_failed", |
| 206 | + kind=kind, |
| 207 | + error=str(e), |
| 208 | + error_type=type(e).__name__, |
| 209 | + ) |
| 210 | + return False |
0 commit comments