337 lines
11 KiB
TypeScript
337 lines
11 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any */
|
||
|
||
import { NFOMetadata,parseNFO } from './nfo-parser';
|
||
import { parseVideoFileName } from './video-parser';
|
||
import { XiaoyaClient } from './xiaoya.client';
|
||
|
||
export interface XiaoyaMetadata {
|
||
tmdbId?: number;
|
||
title: string;
|
||
year?: string;
|
||
rating?: number;
|
||
genres?: string[];
|
||
plot?: string;
|
||
poster?: string;
|
||
background?: string;
|
||
mediaType: 'movie' | 'tv';
|
||
source: 'folder' | 'nfo' | 'tmdb' | 'file';
|
||
}
|
||
|
||
/**
|
||
* 从文件夹名提取 TMDb ID 和年份
|
||
* 格式: "标题 (年份) {tmdb-id}"
|
||
*/
|
||
function parseFolderName(folderName: string | undefined): {
|
||
title?: string;
|
||
year?: string;
|
||
tmdbId?: number;
|
||
} {
|
||
if (!folderName || typeof folderName !== 'string') {
|
||
return {};
|
||
}
|
||
const match = folderName.match(/^(.+?)\s*\((\d{4})\)\s*\{tmdb-(\d+)\}$/);
|
||
if (match) {
|
||
return {
|
||
title: match[1].trim(),
|
||
year: match[2],
|
||
tmdbId: parseInt(match[3]),
|
||
};
|
||
}
|
||
return {};
|
||
}
|
||
|
||
/**
|
||
* 查找 NFO 文件并解析
|
||
*/
|
||
async function findNFO(
|
||
xiaoyaClient: XiaoyaClient,
|
||
videoPath: string
|
||
): Promise<NFOMetadata | null> {
|
||
const pathParts = videoPath.split('/').filter(Boolean);
|
||
|
||
// 判断是否为文件路径
|
||
const videoExtensions = ['.mp4', '.mkv', '.avi', '.m3u8', '.flv', '.ts', '.mov', '.wmv', '.webm'];
|
||
const isFilePath = videoExtensions.some(ext => videoPath.toLowerCase().endsWith(ext));
|
||
|
||
let isInSeasonDir = false;
|
||
|
||
if (isFilePath) {
|
||
// 文件路径：判断父目录是否为季度目录
|
||
const parentDir = pathParts[pathParts.length - 2];
|
||
isInSeasonDir = /(season\s*\d+|s\d+)/i.test(parentDir);
|
||
} else {
|
||
// 目录路径：判断当前目录是否为季度目录
|
||
const currentDir = pathParts[pathParts.length - 1];
|
||
isInSeasonDir = /(season\s*\d+|s\d+)/i.test(currentDir);
|
||
}
|
||
|
||
const nfoSearchPaths: string[] = [];
|
||
|
||
if (isInSeasonDir) {
|
||
// 电视剧：查父级的 tvshow.nfo
|
||
const grandParentDir = pathParts.slice(0, isFilePath ? -2 : -1).join('/');
|
||
nfoSearchPaths.push(`/${grandParentDir}/tvshow.nfo`);
|
||
} else {
|
||
// 电影：查同级的 movie.nfo
|
||
const parentDir = pathParts.slice(0, isFilePath ? -1 : pathParts.length).join('/');
|
||
nfoSearchPaths.push(`/${parentDir}/movie.nfo`);
|
||
}
|
||
|
||
for (const nfoPath of nfoSearchPaths) {
|
||
try {
|
||
const content = await xiaoyaClient.getFileContent(nfoPath);
|
||
const metadata = await parseNFO(content);
|
||
if (metadata) {
|
||
return metadata;
|
||
}
|
||
} catch (error) {
|
||
continue;
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
/**
|
||
* 获取小雅视频的元数据
|
||
*/
|
||
export async function getXiaoyaMetadata(
|
||
xiaoyaClient: XiaoyaClient,
|
||
videoPath: string,
|
||
tmdbApiKey?: string,
|
||
tmdbProxy?: string,
|
||
tmdbReverseProxy?: string
|
||
): Promise<XiaoyaMetadata> {
|
||
const pathParts = videoPath.split('/').filter(Boolean);
|
||
|
||
// 验证路径格式
|
||
if (pathParts.length < 1) {
|
||
throw new Error(`无效的视频路径格式: ${videoPath}`);
|
||
}
|
||
|
||
// 判断是否为文件路径（包含视频扩展名）
|
||
const videoExtensions = ['.mp4', '.mkv', '.avi', '.m3u8', '.flv', '.ts', '.mov', '.wmv', '.webm'];
|
||
const isFilePath = videoExtensions.some(ext => videoPath.toLowerCase().endsWith(ext));
|
||
|
||
// 如果是文件路径，检查是否在季度目录中
|
||
const isInSeasonDir = isFilePath && pathParts.length >= 2 && /(season\s*\d+|s\d+)/i.test(pathParts[pathParts.length - 2]);
|
||
|
||
// 验证路径长度是否足够
|
||
if (isInSeasonDir && pathParts.length < 3) {
|
||
throw new Error(`Season目录路径格式不正确: ${videoPath}`);
|
||
}
|
||
|
||
// 确定元数据目录和文件夹名
|
||
let metadataDir: string;
|
||
let folderName: string;
|
||
|
||
if (isFilePath) {
|
||
// 文件路径
|
||
metadataDir = isInSeasonDir
|
||
? pathParts.slice(0, -2).join('/')
|
||
: pathParts.slice(0, -1).join('/');
|
||
folderName = pathParts[isInSeasonDir ? pathParts.length - 3 : pathParts.length - 2];
|
||
} else {
|
||
// 目录路径
|
||
if (pathParts.length === 1) {
|
||
// 只有一级目录
|
||
metadataDir = '';
|
||
folderName = pathParts[0];
|
||
} else {
|
||
// 判断当前目录是否为季度目录
|
||
const currentDirName = pathParts[pathParts.length - 1];
|
||
const isSeasonDir = /(season\s*\d+|s\d+)/i.test(currentDirName);
|
||
|
||
if (isSeasonDir && pathParts.length >= 2) {
|
||
// 季度目录：使用父级目录名
|
||
metadataDir = pathParts.slice(0, -2).join('/');
|
||
folderName = pathParts[pathParts.length - 2];
|
||
} else {
|
||
metadataDir = pathParts.slice(0, -1).join('/');
|
||
folderName = pathParts[pathParts.length - 1];
|
||
}
|
||
}
|
||
}
|
||
|
||
// 验证 folderName 是否有效
|
||
if (!folderName) {
|
||
throw new Error(`无法从路径中提取文件夹名: ${videoPath}`);
|
||
}
|
||
|
||
// 优先级 1: 从文件夹名提取 TMDb ID
|
||
const folderInfo = parseFolderName(folderName);
|
||
if (folderInfo.tmdbId) {
|
||
const baseUrl = xiaoyaClient.getBaseURL();
|
||
const posterUrl = `${baseUrl}/d/${metadataDir}/poster.jpg`;
|
||
const backgroundUrl = `${baseUrl}/d/${metadataDir}/background.jpg`;
|
||
|
||
// 尝试读取 NFO 获取详细信息
|
||
const nfoData = await findNFO(xiaoyaClient, videoPath);
|
||
|
||
return {
|
||
tmdbId: folderInfo.tmdbId,
|
||
title: nfoData?.title || folderInfo.title || folderName,
|
||
year: folderInfo.year,
|
||
rating: nfoData?.rating,
|
||
genres: nfoData?.genres,
|
||
plot: nfoData?.plot,
|
||
poster: posterUrl,
|
||
background: backgroundUrl,
|
||
mediaType: isInSeasonDir ? 'tv' : 'movie',
|
||
source: nfoData ? 'nfo' : (isFilePath ? 'file' : 'folder'),
|
||
};
|
||
}
|
||
|
||
// 优先级 2: 读取 NFO 文件
|
||
const nfoData = await findNFO(xiaoyaClient, videoPath);
|
||
if (nfoData && nfoData.tmdbId) {
|
||
const baseUrl = xiaoyaClient.getBaseURL();
|
||
const posterUrl = `${baseUrl}/d/${metadataDir}/poster.jpg`;
|
||
const backgroundUrl = `${baseUrl}/d/${metadataDir}/background.jpg`;
|
||
|
||
return {
|
||
tmdbId: nfoData.tmdbId,
|
||
title: nfoData.title || folderName,
|
||
year: nfoData.year?.toString(),
|
||
rating: nfoData.rating,
|
||
genres: nfoData.genres,
|
||
plot: nfoData.plot,
|
||
poster: posterUrl,
|
||
background: backgroundUrl,
|
||
mediaType: nfoData.mediaType,
|
||
source: 'nfo',
|
||
};
|
||
}
|
||
|
||
// 优先级 3: 实时搜索 TMDb（使用文件名）
|
||
if (tmdbApiKey) {
|
||
const fileName = pathParts[pathParts.length - 1];
|
||
const searchQuery = fileName
|
||
.replace(/\.(mp4|mkv|avi|m3u8|flv|ts)$/i, '')
|
||
.replace(/[[\]()]/g, ' ')
|
||
.trim();
|
||
|
||
// 如果文件名是纯数字（可能带小数点）或者是 SxxExx 格式，跳过文件名搜索，直接使用文件夹名
|
||
const isPureNumber = /^[\d.]+$/.test(searchQuery);
|
||
const isSeasonEpisode = /^S\d+E\d+/i.test(searchQuery);
|
||
|
||
if (!isPureNumber && !isSeasonEpisode) {
|
||
const { searchTMDB, getTMDBImageUrl } = await import('./tmdb.search');
|
||
const tmdbResult = await searchTMDB(tmdbApiKey, searchQuery, tmdbProxy, undefined, tmdbReverseProxy);
|
||
|
||
if (tmdbResult.code === 200 && tmdbResult.result) {
|
||
return {
|
||
tmdbId: tmdbResult.result.id,
|
||
title: tmdbResult.result.title || tmdbResult.result.name || folderName,
|
||
year: tmdbResult.result.release_date?.substring(0, 4) ||
|
||
tmdbResult.result.first_air_date?.substring(0, 4),
|
||
rating: tmdbResult.result.vote_average,
|
||
plot: tmdbResult.result.overview,
|
||
poster: tmdbResult.result.poster_path
|
||
? getTMDBImageUrl(tmdbResult.result.poster_path)
|
||
: undefined,
|
||
mediaType: tmdbResult.result.media_type,
|
||
source: isFilePath ? 'file' : 'tmdb',
|
||
};
|
||
}
|
||
}
|
||
}
|
||
|
||
// 优先级 4: 实时搜索 TMDb（使用文件夹名）
|
||
if (tmdbApiKey) {
|
||
const searchQuery = folderName
|
||
.replace(/[[\](){}]/g, ' ')
|
||
.replace(/\d{4}/g, '')
|
||
.trim();
|
||
|
||
const { searchTMDB, getTMDBImageUrl } = await import('./tmdb.search');
|
||
const tmdbResult = await searchTMDB(tmdbApiKey, searchQuery, tmdbProxy, undefined, tmdbReverseProxy);
|
||
|
||
if (tmdbResult.code === 200 && tmdbResult.result) {
|
||
return {
|
||
tmdbId: tmdbResult.result.id,
|
||
title: tmdbResult.result.title || tmdbResult.result.name || folderName,
|
||
year: tmdbResult.result.release_date?.substring(0, 4) ||
|
||
tmdbResult.result.first_air_date?.substring(0, 4),
|
||
rating: tmdbResult.result.vote_average,
|
||
plot: tmdbResult.result.overview,
|
||
poster: tmdbResult.result.poster_path
|
||
? getTMDBImageUrl(tmdbResult.result.poster_path)
|
||
: undefined,
|
||
mediaType: tmdbResult.result.media_type,
|
||
source: 'tmdb',
|
||
};
|
||
}
|
||
}
|
||
|
||
// 降级：只返回文件夹名
|
||
return {
|
||
title: folderName,
|
||
mediaType: isInSeasonDir ? 'tv' : 'movie',
|
||
source: 'folder',
|
||
};
|
||
}
|
||
|
||
/**
|
||
* 获取视频集数列表
|
||
*/
|
||
export async function getXiaoyaEpisodes(
|
||
xiaoyaClient: XiaoyaClient,
|
||
videoPath: string
|
||
): Promise<Array<{ path: string; title: string }>> {
|
||
const pathParts = videoPath.split('/').filter(Boolean);
|
||
|
||
// 判断是否为文件路径（包含视频扩展名）
|
||
const videoExtensions = ['.mp4', '.mkv', '.avi', '.m3u8', '.flv', '.ts', '.mov', '.wmv', '.webm'];
|
||
const isFilePath = videoExtensions.some(ext => videoPath.toLowerCase().endsWith(ext));
|
||
|
||
// 如果是文件路径，检查是否在季度目录中
|
||
const isInSeasonDir = isFilePath && /(season\s*\d+|s\d+)/i.test(pathParts[pathParts.length - 2]);
|
||
|
||
if (isInSeasonDir) {
|
||
// 电视剧：列出当前季的所有集
|
||
const seasonDir = pathParts.slice(0, -1).join('/');
|
||
const listResponse = await xiaoyaClient.listDirectory(`/${seasonDir}`, 1, 100, false);
|
||
|
||
const videoFiles = listResponse.content
|
||
.filter(item =>
|
||
!item.is_dir &&
|
||
videoExtensions.some(ext => item.name.toLowerCase().endsWith(ext))
|
||
)
|
||
.sort((a, b) => a.name.localeCompare(b.name));
|
||
|
||
return videoFiles.map(file => {
|
||
const parsed = parseVideoFileName(file.name);
|
||
console.log('[xiaoya-metadata] 解析文件名:', file.name, '结果:', parsed);
|
||
let title = file.name;
|
||
|
||
if (parsed.season && parsed.episode) {
|
||
title = `S${parsed.season.toString().padStart(2, '0')}E${parsed.episode.toString().padStart(2, '0')}`;
|
||
} else if (parsed.episode) {
|
||
title = parsed.isOVA ? `OVA ${parsed.episode}` : `第${parsed.episode}集`;
|
||
}
|
||
|
||
return {
|
||
path: `/${seasonDir}/${file.name}`,
|
||
title,
|
||
};
|
||
});
|
||
} else {
|
||
// 目录路径或电影文件路径：列出该目录下的所有视频
|
||
const targetDir = isFilePath ? pathParts.slice(0, -1).join('/') : pathParts.join('/');
|
||
const listResponse = await xiaoyaClient.listDirectory(`/${targetDir}`, 1, 100, false);
|
||
|
||
const videoFiles = listResponse.content
|
||
.filter(item =>
|
||
!item.is_dir &&
|
||
videoExtensions.some(ext => item.name.toLowerCase().endsWith(ext))
|
||
)
|
||
.sort((a, b) => a.name.localeCompare(b.name));
|
||
|
||
return videoFiles.map(file => ({
|
||
path: `/${targetDir}/${file.name}`,
|
||
title: file.name,
|
||
}));
|
||
}
|
||
}
|