ani-auto 1.3.0 → 1.3.1

package/src/notify.js

@@ -3,83 +3,106 @@
 const { execSync } = require('child_process');
 const { loadConfig } = require('./config');
 
-// Bot token is shared — each user only needs their own chat ID
-// Users get their chat ID by messaging @anime_auto_downloader_bot on Telegram
 const BOT_TOKEN = '8695132702:AAEFDoSL6nkdwDOoz5wmoVkmuhaYa4tNw0o';
 
-// ── Telegram ───────────────────────────────────────────────────────────────
+// ── Telegram helpers ───────────────────────────────────────────────────────
 
-async function sendTelegram(text) {
+async function sendTelegram(payload) {
     const config = loadConfig();
     const chatId = config.telegramChatId;
-    if (!chatId) return;
-    if (config.telegramNotify === false) return;
+    if (!chatId || config.telegramNotify === false) return;
     try {
         const fetch = require('node-fetch');
-        await fetch(`https://api.telegram.org/bot${BOT_TOKEN}/sendMessage`, {
+        const isPhoto = !!payload.photo;
+        const endpoint = isPhoto ? 'sendPhoto' : 'sendMessage';
+        await fetch(`https://api.telegram.org/bot${BOT_TOKEN}/${endpoint}`, {
             method:  'POST',
             headers: { 'Content-Type': 'application/json' },
-            body:    JSON.stringify({
-                chat_id:    String(chatId),
-                text,
-                parse_mode: 'HTML',
-            }),
+            body:    JSON.stringify({ chat_id: String(chatId), parse_mode: 'HTML', ...payload }),
         });
     } catch (e) {
-        // silently fail — never crash the download for a notification
+        // silently fail
     }
 }
 
-// ── KDE Desktop (notify-send) ──────────────────────────────────────────────
+// ── KDE desktop ────────────────────────────────────────────────────────────
 
-function sendDesktop(title, body, urgency = 'normal') {
+async function sendDesktop(title, body, urgency = 'normal', posterUrl = null) {
     try {
-        // KDE uses notify-send from libnotify
-        execSync(`notify-send -u ${urgency} -a "ani-auto" "${title}" "${body}"`, { stdio: 'ignore' });
-    } catch (e) {
-        // silently fail if notify-send not available
-    }
+        let icon = '';
+        if (posterUrl) {
+            const os    = require('os');
+            const path  = require('path');
+            const fs    = require('fs');
+            const fetch = require('node-fetch');
+            const tmp   = path.join(os.tmpdir(), 'ani-auto-poster.jpg');
+            try {
+                const res = await fetch(posterUrl);
+                fs.writeFileSync(tmp, await res.buffer());
+                icon = `-i "${tmp}"`;
+            } catch (e) {}
+        }
+        execSync(`notify-send -u ${urgency} ${icon} -a "ani-auto" "${title}" "${body}"`, { stdio: 'ignore' });
+    } catch (e) {}
 }
 
-// ── Public notification functions ─────────────────────────────────────────
+// ── Public notification functions ──────────────────────────────────────────
 
-async function notifyEpisode(animeTitle, episode) {
-    const msg   = `✅ <b>${animeTitle}</b>\nEpisode ${episode} downloaded`;
-    const short = `Episode ${episode} downloaded`;
-    await sendTelegram(msg);
-    sendDesktop(animeTitle, short);
+async function notifyEpisode(animeTitle, episode, quality, posterUrl) {
+    const caption =
+        `<b>${animeTitle}</b>\n` +
+        `<code>━━━━━━━━━━━━━━━</code>\n` +
+        `✅  Episode ${episode} downloaded\n` +
+        `🎞  Quality: ${quality || 'default'}`;
+
+    if (posterUrl) {
+        await sendTelegram({ photo: posterUrl, caption });
+    } else {
+        await sendTelegram({ text: `🎬 ${caption}` });
+    }
+    await sendDesktop(animeTitle, `Episode ${episode} downloaded`, 'normal', posterUrl);
 }
 
 async function notifyFailed(animeTitle, episode, error) {
-    const msg   = `❌ <b>${animeTitle}</b>\nEpisode ${episode} failed: ${error}`;
-    const short = `Episode ${episode} failed`;
-    await sendTelegram(msg);
-    sendDesktop(animeTitle, short, 'critical');
+    const text =
+        `❌ <b>${animeTitle}</b>\n` +
+        `<code>━━━━━━━━━━━━━━━</code>\n` +
+        `Episode ${episode} failed\n` +
+        `<i>${error}</i>`;
+    await sendTelegram({ text });
+    await sendDesktop(animeTitle, `Episode ${episode} failed`, 'critical');
 }
 
 async function notifySummary(totals) {
-    if (totals.downloaded === 0 && totals.failed === 0) return; // skip empty runs
-    const msg = 
-        `📋 <b>ani-auto — run complete</b>\n` +
-        `✅ Downloaded: ${totals.downloaded}\n` +
-        `⏭  Skipped:    ${totals.skipped}\n` +
-        `❌ Failed:     ${totals.failed}`;
-    const short = `↓${totals.downloaded}  ✗${totals.failed}`;
-    await sendTelegram(msg);
-    sendDesktop('ani-auto run complete', short);
+    if (totals.downloaded === 0 && totals.failed === 0) return;
+    const text =
+        `📊 <b>ani-auto run complete</b>\n` +
+        `<code>━━━━━━━━━━━━━━━</code>\n` +
+        `✅  Downloaded:  ${totals.downloaded}\n` +
+        `⏭   Skipped:     ${totals.skipped}\n` +
+        `❌  Failed:      ${totals.failed}`;
+    await sendTelegram({ text });
+    await sendDesktop('ani-auto run complete', `Downloaded: ${totals.downloaded}  Failed: ${totals.failed}`);
 }
 
-async function notifyDaemonStart(intervalMinutes) {
-    const msg   = `🚀 <b>ani-auto daemon started</b>\nChecking every ${intervalMinutes} minutes`;
-    const short = `Checking every ${intervalMinutes} minutes`;
-    await sendTelegram(msg);
-    sendDesktop('ani-auto daemon started', short);
+async function notifyDaemonStart(intervalMinutes, downloadingCount) {
+    const { version } = require('../package.json');
+    const text =
+        `🚀 <b>ani-auto daemon started</b>\n` +
+        `<code>━━━━━━━━━━━━━━━</code>\n` +
+        `📦  Version:      v${version}\n` +
+        `📺  Downloading:  ${downloadingCount} anime\n` +
+        `⏱  Fallback check every ${intervalMinutes} minutes`;
+    await sendTelegram({ text });
+    await sendDesktop(
+        `ani-auto v${version} started`,
+        `Downloading ${downloadingCount} anime · every ${intervalMinutes} min`
+    );
 }
 
 async function notifyDaemonStop() {
-    const msg = `🛑 <b>ani-auto daemon stopped</b>`;
-    await sendTelegram(msg);
-    sendDesktop('ani-auto daemon stopped', '');
+    await sendTelegram({ text: '🛑 <b>ani-auto daemon stopped</b>' });
+    await sendDesktop('ani-auto daemon stopped', '');
 }
 
 module.exports = {