ani-auto 1.2.5 → 1.3.0

package/README.md

@@ -82,7 +82,6 @@ Commands:
 
 ```bash
 ani-auto config --set-quality 1080p
-ani-auto config --set-language sub
 ani-auto config --set-concurrency 3
 ani-auto config --set-output ~/videos/anime
 ani-auto config --set-interval 60

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ani-auto",
-  "version": "1.2.5",
+  "version": "1.3.0",
   "description": "Automatic anime episode downloader",
   "bin": {
     "ani-auto": "./src/cli.js"

package/src/cli.js

@@ -31,8 +31,7 @@ program
     .action(async () => {
         banner();
         const current = loadConfig();
-        console.log(chalk.cyan('  Let\'s configure your auto-downloader.\n'));
-
+        console.log(chalk.cyan('  Let\'s configure your auto-downloader.')); 
         const { anilistUsername } = await inquirer.prompt([{
             type: 'input', name: 'anilistUsername',
             message: 'AniList username (leave blank to skip AniList integration):',
@@ -58,8 +57,8 @@ program
                 for (const opener of openers) {
                     try { spawn(opener, [url], { stdio: 'ignore', detached: true, env: process.env }).unref(); opened = true; break; } catch (e) {}
                 }
-                console.log(chalk.cyan(`\n  ${opened ? 'Browser opened!' : 'Visit:'} ${url}`));
-                console.log(chalk.gray('  Authorize, then copy the token from the URL (the part after access_token=).\n'));
+                console.log(chalk.cyan(`  ${opened ? 'Browser opened!' : 'Visit:'} ${url}`));
+                console.log(chalk.gray('  Authorize, then copy the token from the URL (the part after access_token=).'));
                 await new Promise(r => setTimeout(r, 1500));
             }
 
@@ -86,12 +85,11 @@ program
         }
 
         const answers = await inquirer.prompt([
-            { type: 'list',   name: 'quality',               message: 'Default download quality:', choices: ['1080p', '720p', '480p', '360p'], default: current.quality || '1080p' },
-            { type: 'list',   name: 'language',              message: 'Default audio language:',   choices: ['sub', 'dub'], default: current.language || 'sub' },
+            { type: 'list',   name: 'quality',               message: 'Default download quality:', choices: ['1080p', '720p', '360p'], default: current.quality || '1080p' },
             { type: 'number', name: 'concurrency',           message: 'Parallel downloads per run (1-10):', default: current.concurrency || 2, validate: v => (v >= 1 && v <= 10) ? true : 'Enter a number 1–10' },
             { type: 'input',  name: 'outputDir',             message: 'Output directory for downloads:', default: current.outputDir },
             { type: 'number', name: 'daemonIntervalMinutes', message: 'Daemon polling interval (minutes):', default: current.daemonIntervalMinutes || 30 },
-            { type: 'input',  name: 'telegramChatId',        message: 'Telegram chat ID for notifications (leave blank to skip):\n  → Message @userinfobot on Telegram to get your ID:', default: current.telegramChatId || '' },
+            { type: 'input',  name: 'telegramChatId',        message: 'Telegram chat ID for notifications (leave blank to skip):  → Message @userinfobot on Telegram to get your ID:', default: current.telegramChatId || '' },
         ]);
         answers.anilistUsername = anilistUsername;
         answers.anilistToken    = anilistToken;
@@ -100,7 +98,7 @@ program
         if (answers.anilistToken) {
             const spinner = ora('Verifying AniList token...').start();
             try { const viewer = await getViewer(answers.anilistToken); spinner.succeed(`Logged in as ${chalk.bold(viewer.name)}`); }
-            catch { spinner.warn('Could not verify token — saved anyway, check it manually.'); }
+            catch { spinner.warn('Could not verify token - saved anyway, check it manually.'); }
         }
 
         updateConfig({
@@ -108,16 +106,15 @@ program
             anilistToken:     answers.anilistToken    || null,
             watchStatuses:    answers.watchStatuses   || current.watchStatuses,
             quality:          answers.quality,
-            language:         answers.language,
             concurrency:      answers.concurrency,
             outputDir:        answers.outputDir,
             daemonIntervalMinutes: answers.daemonIntervalMinutes,
             telegramChatId:   answers.telegramChatId  || null,
         });
 
-        console.log(chalk.green(`\n  ✔  Config saved to ${CONFIG_FILE}\n`));
+        console.log(chalk.green(`  ✔  Config saved to ${CONFIG_FILE}`));
         console.log(`  Run ${chalk.bold('ani-auto check')} to do a manual download cycle.`);
-        console.log(`  Run ${chalk.bold('ani-auto daemon')} to start the background daemon.\n`);
+        console.log(`  Run ${chalk.bold('ani-auto daemon')} to start the background daemon.`);
     });
 
 program
@@ -139,7 +136,7 @@ program
                         type:    'list',
                         name:    'pickedQuality',
                         message: 'Quality for this run:',
-                        choices: ['1080p', '720p', '480p', '360p'],
+                        choices: ['1080p', '720p', '360p'],
                         default: quality,
                     },
                     {
@@ -152,7 +149,7 @@ program
                 quality = pickedQuality;
                 if (saveAsDefault) {
                     updateConfig({ quality });
-                    console.log(chalk.green(`  ✔  Saved default quality: ${quality}\n`));
+                    console.log(chalk.green(`  ✔  Saved default quality: ${quality}`));
                 }
 
                 const { pickedStatuses } = await inquirer.prompt([{
@@ -214,7 +211,7 @@ program
 
         if (!all.length) {
             console.log(chalk.yellow('  No anime tracked yet.'));
-            console.log(`  Use ${chalk.bold('ani-auto add "<title>"')} to add one manually.\n`);
+            console.log(`  Use ${chalk.bold('ani-auto add "<title>"')} to add one manually.`);
             return;
         }
 
@@ -249,7 +246,7 @@ program
         const { picked } = await inquirer.prompt([{
             type: 'list', name: 'picked', message: 'Select the anime:',
             choices: results.map(r => ({
-                name:  `${pickTitle(r.titles)} (${r.year || '?'}) — ${r.episodes ? r.episodes + ' eps' : 'ongoing'} [${r.status}]`,
+                name:  `${pickTitle(r.titles)} (${r.year || '?'}) - ${r.episodes ? r.episodes + ' eps' : 'ongoing'} [${r.status}]`,
                 value: r,
             })),
         }]);
@@ -279,7 +276,7 @@ program
                     );
                 });
                 if (inAnilist) {
-                    console.log(chalk.yellow(`  "${resolvedTitle}" is already in your AniList list (${inAnilist.status}) — no need to add manually.`));
+                    console.log(chalk.yellow(`  "${resolvedTitle}" is already in your AniList list (${inAnilist.status}) - no need to add manually.`));
                     return;
                 }
             } catch (e) {}
@@ -290,7 +287,7 @@ program
         updateConfig({ manualList });
         db.upsertAnime({ title: resolvedTitle, siteId: resolvedTitle, anilistId: picked.anilistId, source: 'manual', quality: entry.quality, language: entry.language });
 
-        console.log(chalk.green(`\n  ✔  Added ${resolvedTitle} to manual list.`));
+        console.log(chalk.green(`  ✔  Added ${resolvedTitle} to manual list.`));
         if (opts.quality)  console.log(`     Quality:  ${opts.quality}`);
         if (opts.language) console.log(`     Language: ${opts.language}`);
         console.log('');
@@ -305,7 +302,7 @@ program
         const config = loadConfig();
         const manualList = config.manualList || [];
 
-        if (!manualList.length) { console.log(chalk.yellow('  Your manual list is empty.\n')); return; }
+        if (!manualList.length) { console.log(chalk.yellow('  Your manual list is empty.')); return; }
 
         let target = title;
         if (!target) {
@@ -317,14 +314,14 @@ program
         }
 
         const exists = manualList.find(m => m.title.toLowerCase() === target.toLowerCase());
-        if (!exists) { console.log(chalk.yellow(`  "${target}" is not in your manual list.\n`)); return; }
+        if (!exists) { console.log(chalk.yellow(`  "${target}" is not in your manual list.`)); return; }
 
         const { confirm } = await inquirer.prompt([{ type: 'confirm', name: 'confirm', message: `Remove "${target}" and all its episode history?`, default: false }]);
         if (!confirm) return;
 
         updateConfig({ manualList: manualList.filter(m => m.title.toLowerCase() !== target.toLowerCase()) });
         db.removeAnime(target);
-        console.log(chalk.green(`  ✔  Removed "${target}".\n`));
+        console.log(chalk.green(`  ✔  Removed "${target}".`));
     });
 
 async function pickFromWatchList(message) {
@@ -332,9 +329,9 @@ async function pickFromWatchList(message) {
     const config = loadConfig();
     const watchList = await resolveWatchList(config);
 
-    if (!watchList.length) { console.log(chalk.yellow('  Watch list is empty.\n')); return null; }
+    if (!watchList.length) { console.log(chalk.yellow('  Watch list is empty.')); return null; }
 
-    console.log(chalk.bold('\n  Watch list:'));
+    console.log(chalk.bold('  Watch list:'));
     watchList.forEach((item, i) => console.log(`  ${chalk.cyan(i + 1)}. ${item.title}`));
     console.log('');
 
@@ -357,7 +354,7 @@ program
         const item = await pickFromWatchList('Select anime to enable:');
         if (!item) return;
         db.setAnimeEnabled(item.title, true);
-        console.log(chalk.green(`\n  ✔  Enabled "${item.title}".\n`));
+        console.log(chalk.green(`  ✔  Enabled "${item.title}".`));
     });
 
 program
@@ -368,7 +365,7 @@ program
         const item = await pickFromWatchList('Select anime to disable:');
         if (!item) return;
         db.setAnimeEnabled(item.title, false);
-        console.log(chalk.yellow(`\n  ✔  Disabled "${item.title}".\n`));
+        console.log(chalk.yellow(`  ✔  Disabled "${item.title}".`));
     });
 
 program
@@ -379,7 +376,7 @@ program
         banner();
         await db.initDb();
         const runs = db.getRecentRuns(parseInt(opts.runs, 10));
-        if (!runs.length) { console.log(chalk.yellow('  No runs recorded yet. Run `ani-auto check` to start.\n')); return; }
+        if (!runs.length) { console.log(chalk.yellow('  No runs recorded yet. Run `ani-auto check` to start.')); return; }
 
         console.log(chalk.bold(`  ${'Time'.padEnd(20)} ${'Downloaded'.padEnd(12)} ${'Skipped'.padEnd(10)} Failed`));
         console.log('  ' + '─'.repeat(55));
@@ -407,7 +404,7 @@ program
             const { getConfigPath } = require('./config');
             const configPath = getConfigPath();
             const editor = process.env.EDITOR || process.env.VISUAL || 'nano';
-            console.log(chalk.cyan(`  Opening config in ${editor}: ${configPath}\n`));
+            console.log(chalk.cyan(`  Opening config in ${editor}: ${configPath}`));
             try { execSync(`${editor} "${configPath}"`, { stdio: 'inherit' }); }
             catch (e) { console.log(chalk.yellow(`  Could not open editor. Config is at: ${configPath}`)); }
             return;
@@ -423,11 +420,11 @@ program
 
         if (Object.keys(updates).length) {
             updateConfig(updates);
-            console.log(chalk.green('  ✔  Config updated:\n'));
+            console.log(chalk.green('  ✔  Config updated:'));
             for (const [k, v] of Object.entries(updates)) console.log(`     ${chalk.bold(k)}: ${v}`);
         } else {
             const config = loadConfig();
-            console.log(chalk.bold('  Current configuration:\n'));
+            console.log(chalk.bold('  Current configuration:'));
             for (const [k, v] of Object.entries(config)) {
                 if (k === 'anilistToken' && v) console.log(`  ${chalk.cyan(k.padEnd(26))} ${chalk.gray('[set]')}`);
                 else console.log(`  ${chalk.cyan(k.padEnd(26))} ${JSON.stringify(v)}`);
@@ -436,6 +433,43 @@ program
         console.log('');
     });
 
+// ── notify ───────────────────────────────────────────────────────────────
+
+program
+    .command('notify [action]')
+    .description('Enable or disable Telegram notifications (on, off, status)')
+    .action(async (action) => {
+        banner();
+        const config = loadConfig();
+
+        if (!action) {
+            const enabled = !!config.telegramChatId && config.telegramNotify !== false;
+            console.log(`  Telegram notifications: ${enabled ? chalk.green('enabled') : chalk.gray('disabled')}`);
+            if (!config.telegramChatId) console.log(chalk.yellow('  No chat ID set - run ani-auto setup to configure. '));
+            else console.log('');
+            return;
+        }
+
+        switch (action.toLowerCase()) {
+            case 'on':
+                if (!config.telegramChatId) {
+                    console.log(chalk.yellow('  No chat ID set - run ani-auto setup first. '));
+                    return;
+                }
+                updateConfig({ telegramNotify: true });
+                console.log(chalk.green('  ✔  Telegram notifications enabled. '));
+                break;
+
+            case 'off':
+                updateConfig({ telegramNotify: false });
+                console.log(chalk.yellow('  ✔  Telegram notifications disabled. '));
+                break;
+
+            default:
+                console.log(chalk.yellow(`  Unknown action "${action}". Use: on, off`));
+        }
+    });
+
 // ── update ───────────────────────────────────────────────────────────────
 
 program
@@ -454,12 +488,12 @@ program
             spinner.stop();
 
             if (latest === version) {
-                console.log(chalk.green(`  ✔  Already on the latest version (v${version}).\n`));
+                console.log(chalk.green(`  ✔  Already on the latest version (v${version}).`));
                 return;
             }
 
             console.log(chalk.cyan(`  Current version: ${chalk.bold(version)}`));
-            console.log(chalk.cyan(`  Latest version:  ${chalk.bold(latest)}\n`));
+            console.log(chalk.cyan(`  Latest version:  ${chalk.bold(latest)}`));
 
             const { confirm } = await inquirer.prompt([{
                 type:    'confirm',
@@ -473,7 +507,7 @@ program
             const installing = ora(`Installing v${latest}...`).start();
             try {
                 execSync('npm install -g ani-auto@latest', { stdio: 'pipe' });
-                installing.succeed(`Updated to v${latest} — restart your terminal to use the new version.`);
+                installing.succeed(`Updated to v${latest} - restart your terminal to use the new version.`);
             } catch (e) {
                 installing.fail('Update failed. Try manually: npm install -g ani-auto@latest');
             }
@@ -486,15 +520,107 @@ program
 // ── daemon ────────────────────────────────────────────────────────────────
 
 program
-    .command('daemon')
-    .description('Start the background daemon')
-    .option('--auto', 'Skip prompts, auto-select all (default: true for daemon)')
-    .action(() => { require('./daemon'); });
+    .command('daemon [action]')
+    .description('Manage the background daemon (start, stop, restart, status, install)')
+    .action(async (action) => {
+        const { execSync, spawnSync } = require('child_process');
+
+        const SERVICE  = 'ani-auto';
+        const run = (cmd) => {
+            try {
+                const r = spawnSync(cmd, { shell: true, encoding: 'utf8' });
+                return { out: r.stdout?.trim(), err: r.stderr?.trim(), code: r.status };
+            } catch (e) {
+                return { out: '', err: e.message, code: 1 };
+            }
+        };
+
+        // No action - run inline (legacy behaviour)
+        if (!action) {
+            require('./daemon');
+            return;
+        }
+
+        switch (action.toLowerCase()) {
+
+            case 'install': {
+                banner();
+                const nodeBin   = run('which node').out;
+                const daemonPath = run('npm root -g').out + '/ani-auto/src/daemon.js';
+                const serviceContent = [
+                    '[Unit]',
+                    'Description=ani-auto daemon',
+                    'After=network-online.target',
+                    '',
+                    '[Service]',
+                    `ExecStart=${nodeBin} ${daemonPath}`,
+                    'Restart=on-failure',
+                    'RestartSec=10',
+                    '',
+                    '[Install]',
+                    'WantedBy=default.target',
+                ].join('\n');
+
+                const fs   = require('fs');
+                const os   = require('os');
+                const path = require('path');
+                const dir  = path.join(os.homedir(), '.config', 'systemd', 'user');
+                fs.mkdirSync(dir, { recursive: true });
+                fs.writeFileSync(path.join(dir, `${SERVICE}.service`), serviceContent);
+
+                run('systemctl --user daemon-reload');
+                run(`systemctl --user enable ${SERVICE}`);
+                run(`systemctl --user start ${SERVICE}`);
+                console.log(chalk.green(`  ✔  ani-auto daemon installed and started.`));
+                console.log(chalk.gray(`  Run ${chalk.bold('ani-auto daemon status')} to check.
+`));
+                break;
+            }
+
+            case 'start': {
+                const r = run(`systemctl --user start ${SERVICE}`);
+                if (r.code === 0) console.log(chalk.green('  ✔  Daemon started. '));
+                else console.log(chalk.red(`  ✖  Failed to start: ${r.err}
+  Try: ani-auto daemon install`));
+                break;
+            }
+
+            case 'stop': {
+                const r = run(`systemctl --user stop ${SERVICE}`);
+                if (r.code === 0) console.log(chalk.yellow('  ✔  Daemon stopped. '));
+                else console.log(chalk.red(`  ✖  Failed to stop: ${r.err}`));
+                break;
+            }
+
+            case 'restart': {
+                const r = run(`systemctl --user restart ${SERVICE}`);
+                if (r.code === 0) console.log(chalk.green('  ✔  Daemon restarted. '));
+                else console.log(chalk.red(`  ✖  Failed to restart: ${r.err}`));
+                break;
+            }
+
+            case 'status': {
+                const r = run(`systemctl --user status ${SERVICE}`);
+                console.log(r.out || r.err);
+                break;
+            }
+
+            case 'logs': {
+                console.log(chalk.gray('  Streaming logs (Ctrl+C to stop)... '));
+                const { spawn } = require('child_process');
+                spawn('journalctl', ['--user', '-u', SERVICE, '-f'], { stdio: 'inherit' });
+                break;
+            }
+
+            default:
+                console.log(chalk.yellow(`  Unknown action "${action}". Use: install, start, stop, restart, status, logs`));
+        }
+    });
 
 program
     .name('ani-auto')
     .version(version)
-    .description('Automatic anime episode downloader — AniList-aware')
+    .description('Automatic anime episode downloader - AniList-aware')
     .parse(process.argv);
 
 if (!process.argv.slice(2).length) { banner(); program.help(); }

package/src/db.js

@@ -56,6 +56,7 @@ function migrate(db) {
             language             TEXT,
             confirmed_site_id    TEXT,
             confirmed_site_title TEXT,
+            anilist_title        TEXT,
             added_at             TEXT    NOT NULL DEFAULT (datetime('now'))
         )
     `);
@@ -74,6 +75,7 @@ function migrate(db) {
     // Add confirmed_site columns if upgrading from older DB
     try { db.run('ALTER TABLE anime ADD COLUMN confirmed_site_id TEXT'); } catch(e) {}
     try { db.run('ALTER TABLE anime ADD COLUMN confirmed_site_title TEXT'); } catch(e) {}
+    try { db.run('ALTER TABLE anime ADD COLUMN anilist_title TEXT'); } catch(e) {}
 
     db.run(`
         CREATE TABLE IF NOT EXISTS run_log (
@@ -168,6 +170,13 @@ function isEpisodeDownloaded(animeDbId, episodeNum) {
     );
 }
 
+function getEpisodeRow(animeDbId, episodeNum) {
+    return _get(
+        'SELECT * FROM episodes WHERE anime_id = ? AND episode_num = ?',
+        [animeDbId, episodeNum]
+    );
+}
+
 function markEpisodeStatus(animeDbId, episodeNum, status, filename = null, error = null) {
     _run(`
         INSERT INTO episodes (anime_id, episode_num, status, filename, downloaded_at, error)
@@ -209,22 +218,27 @@ function getDbInstance() {
 }
 
 function getAnimeBySiteTitle(title) {
-    return _get('SELECT * FROM anime WHERE LOWER(title) = LOWER(?)', [title]);
+    // Check by anilist_title first, then fall back to title column
+    return _get('SELECT * FROM anime WHERE LOWER(anilist_title) = LOWER(?)', [title])
+        || _get('SELECT * FROM anime WHERE LOWER(title) = LOWER(?)', [title]);
 }
 
-function saveConfirmedSiteMatch(title, confirmedSiteId, confirmedSiteTitle) {
-    // Try to update existing row first
-    const existing = _get('SELECT id FROM anime WHERE LOWER(title) = LOWER(?)', [title]);
+function saveConfirmedSiteMatch(anilistTitle, confirmedSiteId, confirmedSiteTitle) {
+    // Look up by anilist_title, AniList title matching title col, or site title matching title col
+    const existing =
+        _get('SELECT id FROM anime WHERE LOWER(anilist_title) = LOWER(?)', [anilistTitle]) ||
+        _get('SELECT id FROM anime WHERE LOWER(title) = LOWER(?)', [anilistTitle]) ||
+        _get('SELECT id FROM anime WHERE LOWER(title) = LOWER(?)', [confirmedSiteTitle]);
+
     if (existing) {
         _run(
-            'UPDATE anime SET confirmed_site_id = ?, confirmed_site_title = ? WHERE id = ?',
-            [confirmedSiteId, confirmedSiteTitle, existing.id]
+            'UPDATE anime SET confirmed_site_id = ?, confirmed_site_title = ?, anilist_title = ? WHERE id = ?',
+            [confirmedSiteId, confirmedSiteTitle, anilistTitle, existing.id]
         );
     } else {
-        // No row yet — insert one using confirmedSiteId as the site_id
         _run(
-            'INSERT OR IGNORE INTO anime (title, site_id, confirmed_site_id, confirmed_site_title, source) VALUES (?, ?, ?, ?, ?)',
-            [title, confirmedSiteId, confirmedSiteId, confirmedSiteTitle, 'anilist']
+            'INSERT OR IGNORE INTO anime (title, anilist_title, site_id, confirmed_site_id, confirmed_site_title, source) VALUES (?, ?, ?, ?, ?, ?)',
+            [anilistTitle, anilistTitle, confirmedSiteId, confirmedSiteId, confirmedSiteTitle, 'anilist']
         );
     }
     _save();
@@ -234,6 +248,6 @@ module.exports = {
     initDb,
     getDb: getDbInstance,
     upsertAnime, getAnime, getAnimeBySiteTitle, saveConfirmedSiteMatch, getAllEnabledAnime, setAnimeEnabled, removeAnime,
-    isEpisodeDownloaded, markEpisodeStatus, getEpisodeStats,
+    isEpisodeDownloaded, getEpisodeRow, markEpisodeStatus, getEpisodeStats,
     logRun, getRecentRuns,
 };

package/src/engine.js

@@ -155,7 +155,16 @@ async function processAnime(workItem, config, multiBar, auto = false, runQuality
         info(`[${workItem.title}] Using saved site match: ${chalk.bold(siteAnime.title)}`);
         db.saveConfirmedSiteMatch(workItem.title, siteAnime.session, siteAnime.title);
     } else if (auto) {
-        siteAnime = siteResults[0];
+        // Try to find best match — prefer exact title match, then season match, then first result
+        const normalize = s => s.toLowerCase().replace(/[^a-z0-9]/g, '');
+        const normTitle = normalize(workItem.title);
+        const season    = detectSeason(workItem.title);
+
+        const bestMatch = siteResults.find(r => normalize(r.title) === normTitle)
+            || (season ? siteResults.find(r => detectSeason(r.title) === season) : null)
+            || siteResults[0];
+
+        siteAnime = bestMatch;
         db.saveConfirmedSiteMatch(workItem.title, siteAnime.session, siteAnime.title);
         info(`[${workItem.title}] Auto-matched to: ${chalk.bold(siteAnime.title)}`);
     } else {

package/src/notify.js

@@ -13,6 +13,7 @@ async function sendTelegram(text) {
     const config = loadConfig();
     const chatId = config.telegramChatId;
     if (!chatId) return;
+    if (config.telegramNotify === false) return;
     try {
         const fetch = require('node-fetch');
         await fetch(`https://api.telegram.org/bot${BOT_TOKEN}/sendMessage`, {