461 lines
14 KiB
TypeScript
461 lines
14 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any */
|
||
import parseTorrentName from 'parse-torrent-name';
|
||
import { parseStringPromise } from 'xml2js';
|
||
|
||
import { getConfig, setCachedConfig } from '@/lib/config';
|
||
import { getMagnetBaseUrl, universalMagnetFetch } from '@/lib/magnet.client';
|
||
import { db, getStorage } from '@/lib/db';
|
||
import { EmailService } from '@/lib/email.service';
|
||
import {
|
||
addOpenListOfflineDownload,
|
||
getOfflineDownloadBasePath,
|
||
joinOpenListPath,
|
||
} from '@/lib/openlist-offline-download';
|
||
import { AnimeSubscription, AnimeSubscriptionDownloadTool } from '@/types/anime-subscription';
|
||
|
||
const downloadTools: AnimeSubscriptionDownloadTool[] = ['aria2', 'qBittorrent', 'Transmission'];
|
||
|
||
const pickRssText = (value: any): string => {
|
||
if (value === undefined || value === null) return '';
|
||
const first = Array.isArray(value) ? value[0] : value;
|
||
if (first === undefined || first === null) return '';
|
||
if (typeof first === 'object') return String(first._ ?? first.$?.url ?? first.$?.href ?? '');
|
||
return String(first);
|
||
};
|
||
|
||
function getAnimeSubscriptionDownloadTool(tool: unknown): AnimeSubscriptionDownloadTool {
|
||
return typeof tool === 'string' && downloadTools.includes(tool as AnimeSubscriptionDownloadTool)
|
||
? tool as AnimeSubscriptionDownloadTool
|
||
: 'aria2';
|
||
}
|
||
|
||
/**
|
||
* 从标题中提取集数
|
||
*/
|
||
export function extractEpisode(title: string): number | null {
|
||
const parsed = parseTorrentName(title);
|
||
|
||
if (parsed.episode) {
|
||
return parsed.episode;
|
||
}
|
||
|
||
// 备用正则匹配
|
||
const patterns = [
|
||
/\[(\d+)\]/, // [01]
|
||
/第(\d+)[集话]/, // 第01集
|
||
/EP?(\d+)/i, // EP01, E01
|
||
/\s(\d+)\s/, // 空格01空格
|
||
];
|
||
|
||
for (const pattern of patterns) {
|
||
const match = title.match(pattern);
|
||
if (match) {
|
||
return parseInt(match[1], 10);
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
/**
|
||
* 解析逗号分隔关键词（兼容中文逗号）
|
||
*/
|
||
function parseKeywords(text: string): string[] {
|
||
return text
|
||
.replace(/，/g, ',')
|
||
.split(',')
|
||
.map((k) => k.trim())
|
||
.filter(Boolean);
|
||
}
|
||
|
||
/**
|
||
* 检查标题是否匹配过滤条件（包含关键词，AND：必须全部命中）
|
||
*/
|
||
export function matchesFilter(title: string, filterText: string): boolean {
|
||
if (!filterText) return true;
|
||
|
||
// 支持多个关键词，用逗号分隔，必须全部匹配
|
||
const keywords = parseKeywords(filterText);
|
||
|
||
return keywords.every((keyword) => title.includes(keyword));
|
||
}
|
||
|
||
/**
|
||
* 检查标题是否命中排除关键词（OR：任一命中即排除）
|
||
*/
|
||
export function matchesExclude(title: string, excludeText?: string): boolean {
|
||
if (!excludeText) return false;
|
||
|
||
const keywords = parseKeywords(excludeText);
|
||
|
||
return keywords.some((keyword) => title.includes(keyword));
|
||
}
|
||
|
||
/**
|
||
* 搜索 ACG 资源（直接调用搜索逻辑，不通过 HTTP）
|
||
*/
|
||
export async function searchACG(
|
||
keyword: string,
|
||
source: 'acgrip' | 'mikan' | 'dmhy' | 'nyaa'
|
||
) {
|
||
const trimmedKeyword = keyword.trim();
|
||
const config = await getConfig();
|
||
|
||
let searchUrl: string;
|
||
|
||
switch (source) {
|
||
case 'mikan': {
|
||
const baseUrl = getMagnetBaseUrl(
|
||
'https://mikanani.me',
|
||
config.SiteConfig.MagnetMikanReverseProxy
|
||
);
|
||
searchUrl = `${baseUrl}/RSS/Search?searchstr=${encodeURIComponent(trimmedKeyword)}`;
|
||
break;
|
||
}
|
||
case 'dmhy': {
|
||
const baseUrl = getMagnetBaseUrl(
|
||
'http://share.dmhy.org',
|
||
config.SiteConfig.MagnetDmhyReverseProxy
|
||
);
|
||
searchUrl = `${baseUrl}/topics/rss/rss.xml?keyword=${encodeURIComponent(trimmedKeyword)}`;
|
||
break;
|
||
}
|
||
case 'nyaa': {
|
||
const baseUrl = getMagnetBaseUrl(
|
||
'https://nyaa.si',
|
||
config.SiteConfig.MagnetNyaaReverseProxy
|
||
);
|
||
searchUrl = `${baseUrl}/?page=rss&q=${encodeURIComponent(trimmedKeyword)}&c=1_0&f=0`;
|
||
break;
|
||
}
|
||
case 'acgrip':
|
||
default: {
|
||
const baseUrl = getMagnetBaseUrl(
|
||
'https://acg.rip',
|
||
config.SiteConfig.MagnetAcgripReverseProxy
|
||
);
|
||
searchUrl = `${baseUrl}/page/1.xml?term=${encodeURIComponent(trimmedKeyword)}`;
|
||
break;
|
||
}
|
||
}
|
||
|
||
const response = await universalMagnetFetch(searchUrl, config.SiteConfig.MagnetProxy, {
|
||
headers: {
|
||
'User-Agent':
|
||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
|
||
},
|
||
});
|
||
|
||
if (!response.ok) {
|
||
throw new Error(`${source} API 请求失败: ${response.status}`);
|
||
}
|
||
|
||
const xmlData = await response.text();
|
||
const parsed = await parseStringPromise(xmlData);
|
||
|
||
if (!parsed?.rss?.channel?.[0]?.item) {
|
||
return [];
|
||
}
|
||
|
||
const items = parsed.rss.channel[0].item;
|
||
|
||
// 统一格式。注意：Nyaa RSS 的 link 是 .torrent 下载地址，guid 才是详情页。
|
||
return items.map((item: any) => {
|
||
const title = pickRssText(item.title);
|
||
const rawLink = pickRssText(item.link);
|
||
const rawGuid = pickRssText(item.guid);
|
||
const pubDate = pickRssText(item.pubDate);
|
||
const description = pickRssText(item.description) || pickRssText(item['content:encoded']);
|
||
const enclosureUrl =
|
||
pickRssText(item.enclosure?.[0]?.$?.url) ||
|
||
pickRssText(item.enclosure?.[0]?.$?.href);
|
||
|
||
const isNyaa = source === 'nyaa';
|
||
const link = isNyaa ? (rawGuid || rawLink) : rawLink;
|
||
const torrentUrl = isNyaa ? rawLink : enclosureUrl;
|
||
const guid = rawGuid || link || torrentUrl || `${title}-${pubDate}`;
|
||
|
||
return {
|
||
title,
|
||
link,
|
||
guid,
|
||
pubDate,
|
||
torrentUrl,
|
||
description,
|
||
};
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 添加离线下载任务
|
||
*/
|
||
export async function addOfflineDownload(
|
||
torrentUrl: string,
|
||
downloadPath: string
|
||
) {
|
||
const config = await getConfig();
|
||
const downloadTool = getAnimeSubscriptionDownloadTool(
|
||
config.AnimeSubscriptionConfig?.DownloadTool
|
||
);
|
||
|
||
await addOpenListOfflineDownload(config, downloadPath, torrentUrl, downloadTool);
|
||
}
|
||
|
||
/**
|
||
* 发送追番更新通知和邮件
|
||
*/
|
||
async function sendAnimeUpdateNotifications(
|
||
subscription: AnimeSubscription,
|
||
episodes: number[]
|
||
) {
|
||
const config = await getConfig();
|
||
const storage = getStorage();
|
||
|
||
// 获取站长用户名 - 从用户列表中查找 owner 角色
|
||
let ownerUsername: string | null = null;
|
||
try {
|
||
const allUsers = await db.getAllUsers();
|
||
for (const username of allUsers) {
|
||
const userInfo = await db.getUserInfoV2(username);
|
||
if (userInfo?.role === 'owner') {
|
||
ownerUsername = username;
|
||
break;
|
||
}
|
||
}
|
||
} catch (error) {
|
||
console.error('[AnimeSubscription] 获取站长用户名失败:', error);
|
||
}
|
||
|
||
if (!ownerUsername) {
|
||
console.warn('[AnimeSubscription] 未找到站长用户，跳过通知');
|
||
return;
|
||
}
|
||
|
||
// 准备通知内容
|
||
const episodeList = episodes.join('、');
|
||
const notificationTitle = `追番更新：${subscription.title}`;
|
||
const notificationMessage = `您订阅的番剧《${subscription.title}》有新集数更新：第 ${episodeList} 集，已下载到私人影库`;
|
||
|
||
// 需要通知的用户列表（去重）
|
||
const usersToNotify: string[] = [ownerUsername];
|
||
|
||
// 如果创建者不是站长，也通知创建者
|
||
if (subscription.createdBy && subscription.createdBy !== ownerUsername) {
|
||
usersToNotify.push(subscription.createdBy);
|
||
}
|
||
|
||
// 发送站内通知
|
||
for (const username of usersToNotify) {
|
||
try {
|
||
await storage.addNotification(username, {
|
||
id: crypto.randomUUID(),
|
||
type: 'anime_subscription_update',
|
||
title: notificationTitle,
|
||
message: notificationMessage,
|
||
timestamp: Date.now(),
|
||
read: false,
|
||
metadata: {
|
||
subscriptionId: subscription.id,
|
||
subscriptionTitle: subscription.title,
|
||
episodes: episodes,
|
||
},
|
||
});
|
||
console.log(`[AnimeSubscription] 已发送站内通知给用户: ${username}`);
|
||
} catch (error) {
|
||
console.error(`[AnimeSubscription] 发送站内通知失败 (${username}):`, error);
|
||
}
|
||
}
|
||
|
||
// 发送邮件通知（如果已启用）
|
||
const emailConfig = config.EmailConfig;
|
||
if (!emailConfig?.enabled) {
|
||
return;
|
||
}
|
||
|
||
// 获取需要发送邮件的用户邮箱
|
||
const emailsToSend: Array<{ username: string; email: string }> = [];
|
||
|
||
for (const username of usersToNotify) {
|
||
try {
|
||
const userInfo = await db.getUserInfoV2(username);
|
||
// 使用可选的 email 字段
|
||
const email = (userInfo as any)?.email;
|
||
if (email) {
|
||
emailsToSend.push({ username, email });
|
||
}
|
||
} catch (error) {
|
||
console.error(`[AnimeSubscription] 获取用户邮箱失败 (${username}):`, error);
|
||
}
|
||
}
|
||
|
||
// 发送邮件
|
||
for (const { username, email } of emailsToSend) {
|
||
try {
|
||
const emailHtml = `
|
||
<div style="font-family: Arial, sans-serif; max-width: 600px; margin: 0 auto;">
|
||
<h2 style="color: #333;">追番更新通知</h2>
|
||
<p>您好，${username}！</p>
|
||
<p>您订阅的番剧有新集数更新：</p>
|
||
<div style="background-color: #f5f5f5; padding: 15px; border-radius: 5px; margin: 20px 0;">
|
||
<h3 style="margin-top: 0; color: #2563eb;">${subscription.title}</h3>
|
||
<p style="margin: 10px 0;">新增集数：第 ${episodeList} 集</p>
|
||
<p style="margin: 10px 0; color: #666;">搜索源：${subscription.source === 'acgrip' ? 'ACG.RIP' : subscription.source === 'mikan' ? '蜜柑' : subscription.source === 'nyaa' ? 'Nyaa' : '动漫花园'}</p>
|
||
</div>
|
||
<p style="color: #666; font-size: 14px;">这些集数已自动添加到 OpenList 离线下载队列。</p>
|
||
<hr style="border: none; border-top: 1px solid #eee; margin: 20px 0;">
|
||
<p style="color: #999; font-size: 12px;">此邮件由系统自动发送，请勿回复。</p>
|
||
</div>
|
||
`;
|
||
|
||
if (emailConfig.provider === 'smtp' && emailConfig.smtp) {
|
||
await EmailService.sendViaSMTP(emailConfig.smtp, {
|
||
to: email,
|
||
subject: notificationTitle,
|
||
html: emailHtml,
|
||
});
|
||
} else if (emailConfig.provider === 'resend' && emailConfig.resend) {
|
||
await EmailService.sendViaResend(emailConfig.resend, {
|
||
to: email,
|
||
subject: notificationTitle,
|
||
html: emailHtml,
|
||
});
|
||
}
|
||
|
||
console.log(`[AnimeSubscription] 已发送邮件通知给: ${email}`);
|
||
} catch (error) {
|
||
console.error(`[AnimeSubscription] 发送邮件失败 (${email}):`, error);
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 检查单个订阅的更新
|
||
*/
|
||
export async function checkSubscription(subscription: AnimeSubscription) {
|
||
const config = await getConfig();
|
||
if (!config.OpenListConfig?.OfflineDownloadPath) {
|
||
throw new Error('OpenList 离线下载路径未配置');
|
||
}
|
||
|
||
// 1. 搜索资源
|
||
const results = await searchACG(subscription.title, subscription.source);
|
||
|
||
// 2. 过滤并解析集数（包含关键词 AND，排除关键词 OR）
|
||
const newEpisodes = results
|
||
.filter((item: any) => matchesFilter(item.title, subscription.filterText))
|
||
.filter((item: any) => !matchesExclude(item.title, subscription.excludeText))
|
||
.map((item: any) => ({
|
||
episode: extractEpisode(item.title),
|
||
...item,
|
||
}))
|
||
.filter((item: any) => item.episode && item.episode > subscription.lastEpisode)
|
||
.sort((a: any, b: any) => a.episode! - b.episode!);
|
||
|
||
// 3. 下载新集数
|
||
const downloaded = [];
|
||
for (const item of newEpisodes) {
|
||
try {
|
||
const downloadPath = joinOpenListPath(
|
||
getOfflineDownloadBasePath(config),
|
||
subscription.title
|
||
);
|
||
await addOfflineDownload(item.torrentUrl, downloadPath);
|
||
|
||
// 成功后更新 lastEpisode
|
||
subscription.lastEpisode = item.episode!;
|
||
downloaded.push(item.episode);
|
||
|
||
console.log(
|
||
`[AnimeSubscription] ${subscription.title}: 已添加第${item.episode}集到下载队列`
|
||
);
|
||
} catch (error) {
|
||
// 失败则停止，下次继续尝试这一集
|
||
console.error(
|
||
`[AnimeSubscription] ${subscription.title}: 下载第${item.episode}集失败`,
|
||
error
|
||
);
|
||
break;
|
||
}
|
||
}
|
||
|
||
// 4. 更新检查时间
|
||
subscription.lastCheckTime = Date.now();
|
||
|
||
// 5. 发送通知和邮件（如果有下载成功的集数）
|
||
if (downloaded.length > 0) {
|
||
try {
|
||
await sendAnimeUpdateNotifications(subscription, downloaded);
|
||
} catch (error) {
|
||
console.error(`[AnimeSubscription] ${subscription.title}: 发送通知失败`, error);
|
||
}
|
||
}
|
||
|
||
return {
|
||
found: newEpisodes.length,
|
||
downloaded: downloaded.length,
|
||
episodes: downloaded,
|
||
};
|
||
}
|
||
|
||
/**
|
||
* 检查所有订阅（定时任务调用）
|
||
*/
|
||
export async function checkAnimeSubscriptions() {
|
||
console.log('[AnimeSubscription] 开始检查动漫订阅');
|
||
|
||
const config = await getConfig();
|
||
const animeConfig = config.AnimeSubscriptionConfig;
|
||
|
||
if (!animeConfig?.Enabled) {
|
||
console.log('[AnimeSubscription] 动漫订阅功能未启用，跳过检查');
|
||
return;
|
||
}
|
||
|
||
const subscriptions = animeConfig.Subscriptions || [];
|
||
console.log(`[AnimeSubscription] 共有 ${subscriptions.length} 个订阅`);
|
||
|
||
const now = Date.now();
|
||
const MIN_CHECK_INTERVAL = 30 * 60 * 1000; // 30分钟
|
||
let configChanged = false;
|
||
let checkedCount = 0;
|
||
let skippedCount = 0;
|
||
let errorCount = 0;
|
||
|
||
for (const sub of subscriptions) {
|
||
if (!sub.enabled) {
|
||
console.log(`[AnimeSubscription] 跳过已禁用的订阅: ${sub.title}`);
|
||
skippedCount++;
|
||
continue;
|
||
}
|
||
|
||
// 检查是否距离上次检查超过30分钟
|
||
const timeSinceLastCheck = now - sub.lastCheckTime;
|
||
if (timeSinceLastCheck < MIN_CHECK_INTERVAL) {
|
||
const remainingMinutes = Math.ceil((MIN_CHECK_INTERVAL - timeSinceLastCheck) / 60000);
|
||
console.log(`[AnimeSubscription] 跳过 ${sub.title}: 距离上次检查仅 ${Math.floor(timeSinceLastCheck / 60000)} 分钟，还需等待 ${remainingMinutes} 分钟`);
|
||
skippedCount++;
|
||
continue;
|
||
}
|
||
|
||
try {
|
||
console.log(`[AnimeSubscription] 检查订阅: ${sub.title} (源: ${sub.source}, 上次集数: ${sub.lastEpisode})`);
|
||
const result = await checkSubscription(sub);
|
||
console.log(`[AnimeSubscription] ${sub.title}: 找到 ${result.found} 个新集数，成功下载 ${result.downloaded} 个`);
|
||
configChanged = true;
|
||
checkedCount++;
|
||
} catch (error) {
|
||
console.error(`[AnimeSubscription] ${sub.title}: 检查失败`, error);
|
||
errorCount++;
|
||
}
|
||
}
|
||
|
||
// 5. 保存配置并刷新缓存
|
||
if (configChanged) {
|
||
await db.saveAdminConfig(config);
|
||
await setCachedConfig(config);
|
||
console.log('[AnimeSubscription] 配置已更新并保存');
|
||
}
|
||
|
||
console.log(`[AnimeSubscription] 检查完成 - 总计: ${subscriptions.length}, 已检查: ${checkedCount}, 跳过: ${skippedCount}, 失败: ${errorCount}`);
|
||
}
|