私人影库小雅初步实现
This commit is contained in:
mtvpls
@@ -0,0 +1,224 @@
|
||||
/* eslint-disable @typescript-eslint/no-explicit-any */
|
||||
|
||||
import { XiaoyaClient } from './xiaoya.client';
|
||||
import { parseNFO, NFOMetadata } from './nfo-parser';
|
||||
import { parseVideoFileName } from './video-parser';
|
||||
|
||||
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';
|
||||
}
|
||||
|
||||
/**
|
||||
* 从文件夹名提取 TMDb ID 和年份
|
||||
* 格式: "标题 (年份) {tmdb-id}"
|
||||
*/
|
||||
function parseFolderName(folderName: string): {
|
||||
title?: string;
|
||||
year?: string;
|
||||
tmdbId?: number;
|
||||
} {
|
||||
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 parentDir = pathParts.slice(0, -1).join('/');
|
||||
|
||||
const isInSeasonDir = /season\s*\d+/i.test(parentDir);
|
||||
|
||||
const nfoSearchPaths: string[] = [];
|
||||
|
||||
if (isInSeasonDir) {
|
||||
// 电视剧：查父级的 tvshow.nfo
|
||||
const grandParentDir = pathParts.slice(0, -2).join('/');
|
||||
nfoSearchPaths.push(`/${grandParentDir}/tvshow.nfo`);
|
||||
} else {
|
||||
// 电影：查同级的 movie.nfo
|
||||
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
|
||||
): Promise<XiaoyaMetadata> {
|
||||
const pathParts = videoPath.split('/').filter(Boolean);
|
||||
const isInSeasonDir = /season\s*\d+/i.test(pathParts[pathParts.length - 2]);
|
||||
|
||||
// 确定元数据目录
|
||||
const metadataDir = isInSeasonDir
|
||||
? pathParts.slice(0, -2).join('/')
|
||||
: pathParts.slice(0, -1).join('/');
|
||||
|
||||
const folderName = pathParts[isInSeasonDir ? pathParts.length - 3 : pathParts.length - 2];
|
||||
|
||||
// 优先级 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' : '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();
|
||||
|
||||
const { searchTMDB, getTMDBImageUrl } = await import('./tmdb.search');
|
||||
const tmdbResult = await searchTMDB(tmdbApiKey, searchQuery, tmdbProxy);
|
||||
|
||||
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 isInSeasonDir = /season\s*\d+/i.test(pathParts[pathParts.length - 2]);
|
||||
|
||||
if (isInSeasonDir) {
|
||||
// 电视剧：列出当前季的所有集
|
||||
const seasonDir = pathParts.slice(0, -1).join('/');
|
||||
const listResponse = await xiaoyaClient.listDirectory(`/${seasonDir}`);
|
||||
|
||||
const videoExtensions = ['.mp4', '.mkv', '.avi', '.m3u8', '.flv', '.ts', '.mov', '.wmv', '.webm'];
|
||||
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);
|
||||
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 fileName = pathParts[pathParts.length - 1];
|
||||
return [{
|
||||
path: videoPath,
|
||||
title: fileName,
|
||||
}];
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user