mirror of
https://github.com/truewhile/MeBox.git
synced 2026-11-15 20:34:23 +08:00
truewhile
232 lines
8.7 KiB
TypeScript
232 lines
8.7 KiB
TypeScript
import type { Media } from '../types'
|
||
|
||
const VIDEO_EXTENSIONS = new Set([
|
||
'mkv', 'mp4', 'm4v', 'avi', 'mov', 'webm', 'flv', 'wmv', 'ts', 'm2ts', 'mts', 'vob', 'rmvb', 'rm', 'iso',
|
||
])
|
||
|
||
export function isStrmMedia(media?: Media | null): boolean {
|
||
if (!media) return false
|
||
const path = (media.path || '').toLowerCase()
|
||
const strmUrl = (media.strm_url || '').toLowerCase()
|
||
return path.endsWith('.strm') || path.startsWith('cloud://') || Boolean(strmUrl.trim())
|
||
}
|
||
|
||
/** 从路径 / strm URL 推断真实视频容器扩展名（mkv/mp4…）。 */
|
||
export function mediaVersionContainer(media: Media): string {
|
||
const explicit = (media.container || '').trim().toLowerCase().replace(/^\./, '')
|
||
if (explicit && explicit !== 'strm' && VIDEO_EXTENSIONS.has(explicit)) {
|
||
return explicit
|
||
}
|
||
|
||
const path = (media.path || '').replace(/\\/g, '/')
|
||
const base = path.split('/').pop() || ''
|
||
const ext = base.includes('.') ? base.slice(base.lastIndexOf('.')).toLowerCase() : ''
|
||
const name = ext ? base.slice(0, -ext.length) : base
|
||
|
||
if (ext === '.strm') {
|
||
const second = name.includes('.') ? name.slice(name.lastIndexOf('.')).toLowerCase().replace(/^\./, '') : ''
|
||
if (second && VIDEO_EXTENSIONS.has(second)) {
|
||
return second
|
||
}
|
||
|
||
const strm = (media.strm_url || '').toLowerCase()
|
||
if (strm) {
|
||
// 1. /video.mkv 或 /video.mp4 模式
|
||
const marker = '/video.'
|
||
const idx = strm.lastIndexOf(marker)
|
||
if (idx >= 0) {
|
||
let rest = strm.slice(idx + marker.length)
|
||
const end = rest.search(/[?#&/]/)
|
||
if (end >= 0) rest = rest.slice(0, end)
|
||
rest = rest.replace(/^\./, '').trim()
|
||
if (rest && VIDEO_EXTENSIONS.has(rest)) return rest
|
||
}
|
||
|
||
// 2. 从 query 参数（如 ref=/movies/abc.mkv 或 path=...）或 pathname 里匹配扩展名
|
||
const match = strm.match(/\.(mkv|mp4|m4v|avi|mov|webm|flv|wmv|ts|m2ts|iso)(?=[?#&/]|$)/i)
|
||
if (match?.[1]) {
|
||
return match[1].toLowerCase()
|
||
}
|
||
}
|
||
return 'strm'
|
||
}
|
||
|
||
const cleanExt = ext.replace(/^\./, '')
|
||
return VIDEO_EXTENSIONS.has(cleanExt) ? cleanExt : cleanExt
|
||
}
|
||
|
||
function formatSize(bytes: number): string {
|
||
if (!bytes || bytes < 1024) return ''
|
||
const units = ['KB', 'MB', 'GB', 'TB']
|
||
let value = bytes
|
||
let unit = -1
|
||
do {
|
||
value /= 1024
|
||
unit += 1
|
||
} while (value >= 1024 && unit < units.length - 1)
|
||
return `${value.toFixed(1)} ${units[unit]}`
|
||
}
|
||
|
||
/** 从文件名/路径中提取质量或规格标签（如 4K, 2160p, REMUX, HDR, 1080p, 60fps 等） */
|
||
function extractQualityKeywords(text: string): string[] {
|
||
const result: string[] = []
|
||
const upper = text.toUpperCase()
|
||
|
||
// 分辨率
|
||
if (/\b(4K|2160P|UHD)\b/i.test(upper)) result.push('4K')
|
||
else if (/\b(1080P|FHD)\b/i.test(upper)) result.push('1080p')
|
||
else if (/\b(720P|HD)\b/i.test(upper)) result.push('720p')
|
||
|
||
// 格式/版本规格
|
||
if (/\bREMUX\b/i.test(upper)) result.push('REMUX')
|
||
else if (/\b(BLURAY|BLU-RAY|BD)\b/i.test(upper)) result.push('BluRay')
|
||
else if (/\b(WEB-?DL|WEBRIP)\b/i.test(upper)) result.push('WEB-DL')
|
||
|
||
// 动态范围/画质技术
|
||
if (/\b(DV|DOVI|DOLBY[\s._-]?VISION)\b/i.test(upper)) result.push('DV')
|
||
if (/\b(HDR10\+|HDR10PLUS)\b/i.test(upper)) result.push('HDR10+')
|
||
else if (/\bHDR\b/i.test(upper) && !result.includes('DV')) result.push('HDR')
|
||
|
||
// 帧率
|
||
if (/\b(60FPS|120FPS)\b/i.test(upper)) {
|
||
const fpsMatch = upper.match(/\b(60FPS|120FPS)\b/i)
|
||
if (fpsMatch) result.push(fpsMatch[1])
|
||
}
|
||
|
||
return result
|
||
}
|
||
|
||
/** 清理文件名以用作干净的兜底展示 */
|
||
function cleanFallbackFilename(raw: string): string {
|
||
let name = raw.replace(/\\/g, '/').split('/').pop() || ''
|
||
// 去除 .strm 及前缀扩展名
|
||
name = name.replace(/\.strm$/i, '')
|
||
name = name.replace(/\.(mkv|mp4|m4v|avi|mov|webm|flv|wmv|ts|iso)$/i, '')
|
||
// 替换连续的点或下划线为空格
|
||
name = name.replace(/[._]+/g, ' ').trim()
|
||
return name
|
||
}
|
||
|
||
function isIndistinctVersionLabel(parts: string[]): boolean {
|
||
if (parts.length === 0) return true
|
||
return parts.every((part) => {
|
||
const upper = part.toUpperCase()
|
||
if (upper === '云端直链' || upper === 'STRM') return true
|
||
return VIDEO_EXTENSIONS.has(part.toLowerCase())
|
||
})
|
||
}
|
||
|
||
/** 版本切换展示名：分辨率 · 容器 · 编码 · 体积 / 规格 / 来源标签。 */
|
||
export function mediaVersionLabel(media: Media): string {
|
||
const isStrm = isStrmMedia(media)
|
||
const parts: string[] = []
|
||
|
||
// 1. 分辨率（优先从媒体元数据获取，其次从路径中提取）
|
||
if (media.height > 0) {
|
||
if (media.height >= 2100) parts.push('4K')
|
||
else parts.push(`${media.height}p`)
|
||
} else if (media.width > 0) {
|
||
if (media.width >= 3800) parts.push('4K')
|
||
else parts.push(`${media.width}w`)
|
||
}
|
||
|
||
// 2. 容器
|
||
const container = mediaVersionContainer(media)
|
||
if (container && container !== 'strm') {
|
||
parts.push(container.toUpperCase())
|
||
}
|
||
|
||
// 3. 视频编码
|
||
if (media.video_codec?.trim()) {
|
||
parts.push(media.video_codec.trim().toUpperCase())
|
||
}
|
||
|
||
// 4. 从文件名/路径补充规格特征（针对 STRM 缺乏探测数据时的关键画质提示）
|
||
const pathKeywords = extractQualityKeywords(`${media.path || ''} ${media.strm_url || ''}`)
|
||
for (const kw of pathKeywords) {
|
||
// 避免重复（如已经有了 4K / 1080p）
|
||
if (!parts.some((p) => p.toUpperCase() === kw.toUpperCase())) {
|
||
parts.push(kw)
|
||
}
|
||
}
|
||
|
||
// 5. 文件大小：过滤掉本地 STRM 占位小文件（< 1MB 的 STRM 索引通常不是真实视频大小）
|
||
const isPlaceholderSize = isStrm && media.size_bytes > 0 && media.size_bytes < 1024 * 1024
|
||
if (media.size_bytes > 0 && !isPlaceholderSize) {
|
||
const size = formatSize(media.size_bytes)
|
||
if (size) parts.push(size)
|
||
}
|
||
|
||
// 6. 如果仍然没有解析出足够特征，标记来源（云盘 / STRM）
|
||
if (isStrm && !parts.some((p) => ['4K', '1080P', '720P', 'REMUX', 'WEB-DL', 'MKV', 'MP4'].includes(p.toUpperCase()))) {
|
||
parts.push('云端直链')
|
||
}
|
||
|
||
// 只有容器 /「云端直链」时无法区分多分片，回退文件名（sivr-270-1 / …part1）
|
||
if (parts.length > 0 && !isIndistinctVersionLabel(parts)) {
|
||
return parts.join(' · ')
|
||
}
|
||
|
||
const cleaned = cleanFallbackFilename(media.path || '')
|
||
return cleaned || media.title || media.id
|
||
}
|
||
|
||
export function mediaVersionsOf(media: Media | null | undefined): Media[] {
|
||
if (!media) return []
|
||
if (media.versions && media.versions.length > 1) return media.versions
|
||
return [media]
|
||
}
|
||
|
||
/**
|
||
* media 自身或它的任一版本是否命中 id。
|
||
*
|
||
* 选集列表按「版本组」折叠后，行的 id 是该组首个（最优）版本；用户切到
|
||
* 组内其它版本时播放中的 id 与行 id 不再相等，必须连带比对 versions，
|
||
* 否则当前集高亮、上一集/下一集都会失效。
|
||
*/
|
||
export function mediaVersionMatches(media: Media | null | undefined, id: string): boolean {
|
||
if (!media || !id) return false
|
||
if (media.id === id) return true
|
||
return Boolean(media.versions?.some((version) => version.id === id))
|
||
}
|
||
|
||
/** 版本来源：云端（STRM / 网盘直链）或本地文件。 */
|
||
export function mediaVersionSourceLabel(media: Media): string {
|
||
return isStrmMedia(media) ? '云端直链' : '本地文件'
|
||
}
|
||
|
||
/**
|
||
* 挑出「播放器该自动跳转到的那一集」。
|
||
*
|
||
* 传给 /play/:id 的可能是「整部剧」这类自身没有可播文件的条目——首页轮播、远程 Emby
|
||
* 剧集库的卡片、剧集详情页的播放按钮都会把剧集 ID 直接交给播放器。这种条目的选集接口
|
||
* 返回的是它下面的剧集，列表里没有它自己；若直接拿剧集 ID 去请求
|
||
* /api/stream/<剧集ID>，远程 Emby 会返回 HTTP 500（Unable to cast object of type
|
||
* 'Series' to type 'IHasMediaSources'），用户看到的就是「播放失败」。此时换成第一集，
|
||
* 播放、进度、上下集都落在真正可播的条目上。
|
||
*
|
||
* 返回 null 表示无需跳转：条目还没加载完、条目本身可播（含以其它版本形式出现在列表里）、
|
||
* 或该条目下没有任何可选剧集。
|
||
*/
|
||
export function autoPlayEpisodeFor(
|
||
media: Media | null | undefined,
|
||
requestedId: string,
|
||
episodes: Media[],
|
||
): Media | null {
|
||
if (!media || !requestedId || media.id !== requestedId) return null
|
||
if (episodes.length === 0) return null
|
||
if (episodes.some((episode) => mediaVersionMatches(episode, media.id))) return null
|
||
const first = episodes[0]
|
||
if (!first?.id || first.id === media.id) return null
|
||
return first
|
||
}
|
||
|
||
/** 版本文件名（去掉目录，并剥掉 .strm 外层扩展名），用作版本选项的次要说明。 */
|
||
export function mediaVersionFileName(media: Media): string {
|
||
const path = (media.path || '').replace(/\\/g, '/').trim()
|
||
if (!path) return ''
|
||
const base = path.split('/').filter(Boolean).pop() || ''
|
||
if (!base) return ''
|
||
return base.replace(/\.strm$/i, '')
|
||
}
|