Add Telegram notifications, make Discord optional

app/notifier.py

@@ -1,12 +1,18 @@
+import html
 import logging
+import re
 import time
 
 import requests
 
 from env_config import (
+    DISCORD_ENABLED,
     DISCORD_ROLE_MAP,
     DISCORD_SERVER_NAME,
     DISCORD_WEBHOOK_URL,
+    TELEGRAM_BOT_TOKEN,
+    TELEGRAM_CHAT_ID,
+    TELEGRAM_ENABLED,
     TEST_MODE,
     buy_now,
     currency,
@@ -24,8 +30,27 @@ AVATAR = "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img
 THUMBNAIL = "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/RTX5000.jpg"
 
 
+_MD_LINK = re.compile(r"\[([^\]]*)\]\(([^)\s]*)\)")
+_MD_BOLD = re.compile(r"\*\*(.+?)\*\*")
+_MD_CODE = re.compile(r"`(.+?)`")
+_EMOJI_SHORTCODES = {":point_right:": "👉"}
+
+
+def _md_to_telegram_html(text):
+    """Convert the Discord markdown of the translations to Telegram HTML."""
+    text = html.escape(text)
+    text = _MD_LINK.sub(lambda m: f'<a href="{m.group(2)}">{m.group(1)}</a>', text)
+    text = _MD_BOLD.sub(lambda m: f"<b>{m.group(1)}</b>", text)
+    text = _MD_CODE.sub(lambda m: f"<code>{m.group(1)}</code>", text)
+    for shortcode, emoji in _EMOJI_SHORTCODES.items():
+        text = text.replace(shortcode, emoji)
+    return text
+
+
 # In stock
 def send_discord_notification(gpu_name, product_link, products_price):
+    if not DISCORD_ENABLED:
+        return
     timestamp = int(time.time())
     if TEST_MODE:
         logging.info(f"[TEST MODE] Notification: {gpu_name} available!")
@@ -66,6 +91,8 @@ def send_discord_notification(gpu_name, product_link, products_price):
 
 # Out of stock
 def send_out_of_stock_notification(gpu_name, product_link, products_price):
+    if not DISCORD_ENABLED:
+        return
     timestamp = int(time.time())
     if TEST_MODE:
         logging.info(f"[TEST MODE] Out of stock: {gpu_name}")
@@ -100,6 +127,8 @@ def send_out_of_stock_notification(gpu_name, product_link, products_price):
 
 # SKU change
 def send_sku_change_notification(gpu_name, old_sku, new_sku, product_link):
+    if not DISCORD_ENABLED:
+        return
     timestamp = int(time.time())
     if TEST_MODE:
         logging.info(f"[TEST MODE] SKU change: {old_sku} → {new_sku}")
@@ -134,3 +163,82 @@ def send_sku_change_notification(gpu_name, old_sku, new_sku, product_link):
             logging.error(f"❌ Webhook error: {response.status_code} - {response.text}")
     except Exception as e:
         logging.error(f"🚨 Error sending webhook: {e}")
+
+
+# Telegram
+def _send_telegram(text, success_log):
+    if TEST_MODE:
+        logging.info(f"[TEST MODE] Telegram message: {text}")
+        return
+
+    url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
+    payload = {
+        "chat_id": TELEGRAM_CHAT_ID,
+        "text": text,
+        "parse_mode": "HTML",
+        "disable_web_page_preview": True,
+    }
+
+    try:
+        response = requests.post(url, json=payload, timeout=10)
+        if response.status_code == 200:
+            logging.info(success_log)
+        else:
+            logging.error(f"❌ Telegram error: {response.status_code} - {response.text}")
+    except Exception as e:
+        # The exception message may contain the bot token (it is part of the URL)
+        logging.error(f"🚨 Error sending Telegram message: {type(e).__name__}")
+
+
+def _telegram_footer():
+    return html.escape(footer.format(DISCORD_SERVER_NAME=DISCORD_SERVER_NAME))
+
+
+def _telegram_time():
+    return html.escape(time.strftime("%Y-%m-%d %H:%M:%S %Z"))
+
+
+def send_telegram_notification(gpu_name, product_link, products_price):
+    if not TELEGRAM_ENABLED:
+        return
+    title = html.escape(in_stock_title.format(gpu_name=gpu_name))
+    description = _md_to_telegram_html(buy_now.format(product_link=product_link))
+    price = html.escape(f"{currency}{products_price}")
+    text = (
+        f"<b>{title}</b>\n"
+        f"{description}\n\n"
+        f"{html.escape(price_label)}: <code>{price}</code>\n"
+        f"{html.escape(time_label)}: {_telegram_time()}\n\n"
+        f"<i>{_telegram_footer()}</i>"
+    )
+    _send_telegram(text, "✅ Notification sent to Telegram.")
+
+
+def send_telegram_out_of_stock_notification(gpu_name, product_link, products_price):
+    if not TELEGRAM_ENABLED:
+        return
+    title = html.escape(out_of_stock_title.format(gpu_name=gpu_name))
+    text = (
+        f'<b><a href="{html.escape(product_link)}">{title}</a></b>\n'
+        f"{html.escape(time_label)}: {_telegram_time()}\n\n"
+        f"<i>{_telegram_footer()}</i>"
+    )
+    _send_telegram(text, "✅ Out-of-stock notification sent to Telegram.")
+
+
+def send_telegram_sku_change_notification(gpu_name, old_sku, new_sku, product_link):
+    if not TELEGRAM_ENABLED:
+        return
+    alert = html.escape(imminent_drop.format(DISCORD_ROLE="").strip())
+    title = html.escape(sku_change_title.format(gpu_name=gpu_name))
+    description = _md_to_telegram_html(
+        sku_description.format(old_sku=old_sku, new_sku=new_sku)
+    )
+    text = (
+        f"{alert}\n"
+        f'<b><a href="{html.escape(product_link)}">{title}</a></b>\n'
+        f"{description}\n"
+        f"{html.escape(time_label)}: {_telegram_time()}\n\n"
+        f"<i>{_telegram_footer()}</i>"
+    )
+    _send_telegram(text, "✅ SKU change notification sent to Telegram.")