844 lines
34 KiB
Rust
844 lines
34 KiB
Rust
use std::borrow::Cow;
|
||
|
||
use itertools::Itertools;
|
||
use lazy_static::lazy_static;
|
||
use regex::Regex;
|
||
use serde::{Deserialize, Serialize};
|
||
|
||
use crate::extract::defs::{DIGIT_1PLUS_REG, ZH_NUM_MAP, ZH_NUM_RE};
|
||
|
||
const NAME_EXTRACT_REPLACE_ADHOC1_REPLACED: &str = "$1/$2";
|
||
|
||
lazy_static! {
|
||
static ref TITLE_RE: Regex = Regex::new(
|
||
r#"(.*|\[.*])( -? \d+|\[\d+]|\[\d+.?[vV]\d]|第\d+[话話集]|\[第?\d+[话話集]]|\[\d+.?END]|[Ee][Pp]?\d+|\[\s*\d+\s*[\-\~]\s*\d+\s*\p{scx=Han}*[话話集]\s*])(.*)"#
|
||
).unwrap();
|
||
static ref EP_COLLECTION_RE:Regex = Regex::new(r#"\[?\s*\d+\s*[\-\~]\s*\d+\s*\p{scx=Han}*合?[话話集]\s*]?"#).unwrap();
|
||
static ref MOVIE_TITLE_RE:Regex = Regex::new(r#"(.*|\[.*])(剧场版|[Mm]ovie|电影)(.*?)$"#).unwrap();
|
||
static ref RESOLUTION_RE: Regex = Regex::new(r"1080|720|2160|4K|2K").unwrap();
|
||
static ref SOURCE_L1_RE: Regex = Regex::new(r"B-Global|[Bb]aha|[Bb]ilibili|AT-X|W[Ee][Bb][Rr][Ii][Pp]|Sentai|B[Dd][Rr][Ii][Pp]|UHD[Rr][Ii][Pp]|NETFLIX").unwrap();
|
||
static ref SOURCE_L2_RE: Regex = Regex::new(r"AMZ|CR|W[Ee][Bb]|B[Dd]").unwrap();
|
||
static ref SUB_RE: Regex = Regex::new(r"[简繁日字幕]|CH|BIG5|GB").unwrap();
|
||
static ref PREFIX_RE: Regex =
|
||
Regex::new(r"[^\w\s\p{Unified_Ideograph}\p{scx=Han}\p{scx=Hira}\p{scx=Kana}-]").unwrap();
|
||
static ref EN_BRACKET_SPLIT_RE: Regex = Regex::new(r"[\[\]]").unwrap();
|
||
static ref MOVIE_SEASON_EXTRACT_RE: Regex = Regex::new(r"剧场版|Movie|电影").unwrap();
|
||
static ref MAIN_TITLE_PREFIX_PROCESS_RE1: Regex = Regex::new(r"新番|月?番").unwrap();
|
||
static ref MAIN_TITLE_PREFIX_PROCESS_RE2: Regex = Regex::new(r"[港澳台]{1,3}地区").unwrap();
|
||
static ref MAIN_TITLE_PRE_PROCESS_BACKETS_RE: Regex = Regex::new(r"\[.+\]").unwrap();
|
||
static ref MAIN_TITLE_PRE_PROCESS_BACKETS_RE_SUB1: Regex = Regex::new(r"^.*?\[").unwrap();
|
||
static ref SEASON_EXTRACT_SEASON_ALL_RE: Regex = Regex::new(r"S\d{1,2}|Season \d{1,2}|[第].[季期]|1st|2nd|3rd|\d{1,2}th").unwrap();
|
||
static ref SEASON_EXTRACT_SEASON_EN_PREFIX_RE: Regex = Regex::new(r"Season|S").unwrap();
|
||
static ref SEASON_EXTRACT_SEASON_EN_NTH_RE: Regex = Regex::new(r"1st|2nd|3rd|\d{1,2}th").unwrap();
|
||
static ref SEASON_EXTRACT_SEASON_ZH_PREFIX_RE: Regex = Regex::new(r"[第 ].*[季期(部分)]|部分").unwrap();
|
||
static ref SEASON_EXTRACT_SEASON_ZH_PREFIX_SUB_RE: Regex = Regex::new(r"[第季期 ]").unwrap();
|
||
static ref NAME_EXTRACT_REMOVE_RE: Regex = Regex::new(r"[(（]仅限[港澳台]{1,3}地区[）)]").unwrap();
|
||
static ref NAME_EXTRACT_SPLIT_RE: Regex = Regex::new(r"/|\s{2}|-\s{2}|\]\[").unwrap();
|
||
static ref NAME_EXTRACT_REPLACE_ADHOC1_RE: Regex = Regex::new(r"([\p{scx=Han}\s\(\)]{5,})_([a-zA-Z]{2,})").unwrap();
|
||
static ref NAME_JP_TEST: Regex = Regex::new(r"[\p{scx=Hira}\p{scx=Kana}]{2,}").unwrap();
|
||
static ref NAME_ZH_TEST: Regex = Regex::new(r"[\p{scx=Han}]{2,}").unwrap();
|
||
static ref NAME_EN_TEST: Regex = Regex::new(r"[a-zA-Z]{3,}").unwrap();
|
||
static ref TAGS_EXTRACT_SPLIT_RE: Regex = Regex::new(r"[\[\]()（）_]").unwrap();
|
||
static ref CLEAR_SUB_RE: Regex = Regex::new(r"_MP4|_MKV").unwrap();
|
||
}
|
||
|
||
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
|
||
pub struct RawEpisodeMeta {
|
||
pub name_en: Option<String>,
|
||
pub name_en_no_season: Option<String>,
|
||
pub name_jp: Option<String>,
|
||
pub name_jp_no_season: Option<String>,
|
||
pub name_zh: Option<String>,
|
||
pub name_zh_no_season: Option<String>,
|
||
pub season: i32,
|
||
pub season_raw: Option<String>,
|
||
pub episode_index: i32,
|
||
pub subtitle: Option<String>,
|
||
pub source: Option<String>,
|
||
pub fansub: Option<String>,
|
||
pub resolution: Option<String>,
|
||
}
|
||
|
||
fn extract_fansub(raw_name: &str) -> Option<&str> {
|
||
let mut groups = EN_BRACKET_SPLIT_RE.splitn(raw_name, 3);
|
||
groups.nth(1)
|
||
}
|
||
|
||
fn replace_ch_bracket_to_en(raw_name: &str) -> String {
|
||
raw_name.replace('【', "[").replace('】', "]")
|
||
}
|
||
|
||
fn title_body_pre_process(title_body: &str, fansub: Option<&str>) -> eyre::Result<String> {
|
||
let raw_without_fansub = if let Some(fansub) = fansub {
|
||
let fan_sub_re = Regex::new(&format!(".{fansub}."))?;
|
||
fan_sub_re.replace_all(title_body, "")
|
||
} else {
|
||
Cow::Borrowed(title_body)
|
||
};
|
||
let raw_with_prefix_replaced = PREFIX_RE.replace_all(&raw_without_fansub, "/");
|
||
let mut arg_group = raw_with_prefix_replaced
|
||
.split('/')
|
||
.map(|s| s.trim())
|
||
.filter(|s| !s.is_empty())
|
||
.collect::<Vec<_>>();
|
||
|
||
if arg_group.len() == 1 {
|
||
arg_group = arg_group.first_mut().unwrap().split(' ').collect();
|
||
}
|
||
let mut raw = raw_without_fansub.to_string();
|
||
for arg in arg_group.iter() {
|
||
if (arg_group.len() <= 5 && MAIN_TITLE_PREFIX_PROCESS_RE1.is_match(arg))
|
||
|| (MAIN_TITLE_PREFIX_PROCESS_RE2.is_match(arg))
|
||
{
|
||
let sub = Regex::new(&format!(".{arg}."))?;
|
||
raw = sub.replace_all(&raw, "").to_string();
|
||
}
|
||
}
|
||
if let Some(m) = MAIN_TITLE_PRE_PROCESS_BACKETS_RE.find(&raw) {
|
||
if m.len() as f32 > (raw.len() as f32) * 0.5 {
|
||
let mut raw1 = MAIN_TITLE_PRE_PROCESS_BACKETS_RE_SUB1
|
||
.replace(&raw, "")
|
||
.chars()
|
||
.collect_vec();
|
||
while let Some(ch) = raw1.pop() {
|
||
if ch == ']' {
|
||
break;
|
||
}
|
||
}
|
||
raw = raw1.into_iter().collect();
|
||
}
|
||
}
|
||
Ok(raw.to_string())
|
||
}
|
||
|
||
pub fn extract_season_from_title_body(title_body: &str) -> (String, Option<String>, i32) {
|
||
let name_and_season = EN_BRACKET_SPLIT_RE.replace_all(title_body, " ");
|
||
let seasons = SEASON_EXTRACT_SEASON_ALL_RE
|
||
.find(&name_and_season)
|
||
.into_iter()
|
||
.map(|s| s.as_str())
|
||
.collect_vec();
|
||
|
||
if seasons.is_empty() {
|
||
return (title_body.to_string(), None, 1);
|
||
}
|
||
|
||
let mut season = 1;
|
||
let mut season_raw = None;
|
||
let name = SEASON_EXTRACT_SEASON_ALL_RE.replace_all(&name_and_season, "");
|
||
|
||
for s in seasons {
|
||
season_raw = Some(s);
|
||
if let Some(m) = SEASON_EXTRACT_SEASON_EN_PREFIX_RE.find(s) {
|
||
if let Ok(s) = SEASON_EXTRACT_SEASON_ALL_RE
|
||
.replace_all(m.as_str(), "")
|
||
.parse::<i32>()
|
||
{
|
||
season = s;
|
||
break;
|
||
}
|
||
}
|
||
if let Some(m) = SEASON_EXTRACT_SEASON_EN_NTH_RE.find(s) {
|
||
if let Some(s) = DIGIT_1PLUS_REG
|
||
.find(m.as_str())
|
||
.and_then(|s| s.as_str().parse::<i32>().ok())
|
||
{
|
||
season = s;
|
||
break;
|
||
}
|
||
}
|
||
if let Some(m) = SEASON_EXTRACT_SEASON_ZH_PREFIX_RE.find(s) {
|
||
if let Ok(s) = SEASON_EXTRACT_SEASON_ZH_PREFIX_SUB_RE
|
||
.replace(m.as_str(), "")
|
||
.parse::<i32>()
|
||
{
|
||
season = s;
|
||
break;
|
||
}
|
||
if let Some(m) = ZH_NUM_RE.find(m.as_str()) {
|
||
season = ZH_NUM_MAP[m.as_str()];
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
(name.to_string(), season_raw.map(|s| s.to_string()), season)
|
||
}
|
||
|
||
fn extract_name_from_title_body_name_section(
|
||
title_body_name_section: &str,
|
||
) -> (Option<String>, Option<String>, Option<String>) {
|
||
let mut name_en = None;
|
||
let mut name_zh = None;
|
||
let mut name_jp = None;
|
||
let replaced1 = NAME_EXTRACT_REMOVE_RE.replace_all(title_body_name_section, "");
|
||
let replaced2 = NAME_EXTRACT_REPLACE_ADHOC1_RE
|
||
.replace_all(&replaced1, NAME_EXTRACT_REPLACE_ADHOC1_REPLACED);
|
||
let trimmed = replaced2.trim();
|
||
let mut split = NAME_EXTRACT_SPLIT_RE
|
||
.split(trimmed)
|
||
.map(|s| s.trim())
|
||
.filter(|s| !s.is_empty())
|
||
.map(|s| s.to_string())
|
||
.collect_vec();
|
||
if split.len() == 1 {
|
||
let mut split_space = split[0].split(' ').collect_vec();
|
||
let mut search_indices = vec![0];
|
||
if split_space.len() > 1 {
|
||
search_indices.push(split_space.len() - 1);
|
||
}
|
||
for i in search_indices {
|
||
if NAME_ZH_TEST.is_match(split_space[i]) {
|
||
let chs = split_space[i];
|
||
split_space.remove(i);
|
||
split = vec![chs.to_string(), split_space.join(" ")];
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
for item in split {
|
||
if NAME_JP_TEST.is_match(&item) && name_jp.is_none() {
|
||
name_jp = Some(item);
|
||
} else if NAME_ZH_TEST.is_match(&item) && name_zh.is_none() {
|
||
name_zh = Some(item);
|
||
} else if NAME_EN_TEST.is_match(&item) && name_en.is_none() {
|
||
name_en = Some(item);
|
||
}
|
||
}
|
||
(name_en, name_zh, name_jp)
|
||
}
|
||
|
||
fn extract_episode_index_from_title_episode(title_episode: &str) -> Option<i32> {
|
||
DIGIT_1PLUS_REG
|
||
.find(title_episode)?
|
||
.as_str()
|
||
.parse::<i32>()
|
||
.ok()
|
||
}
|
||
|
||
fn clear_sub(sub: Option<String>) -> Option<String> {
|
||
sub.map(|s| CLEAR_SUB_RE.replace_all(&s, "").to_string())
|
||
}
|
||
|
||
fn extract_tags_from_title_extra(
|
||
title_extra: &str,
|
||
) -> (Option<String>, Option<String>, Option<String>) {
|
||
let replaced = TAGS_EXTRACT_SPLIT_RE.replace_all(title_extra, " ");
|
||
let elements = replaced
|
||
.split(' ')
|
||
.map(|s| s.trim())
|
||
.filter(|s| !s.is_empty())
|
||
.collect_vec();
|
||
|
||
let mut sub = None;
|
||
let mut resolution = None;
|
||
let mut source = None;
|
||
for element in elements.iter() {
|
||
if SUB_RE.is_match(element) {
|
||
sub = Some(element.to_string())
|
||
} else if RESOLUTION_RE.is_match(element) {
|
||
resolution = Some(element.to_string())
|
||
} else if SOURCE_L1_RE.is_match(element) {
|
||
source = Some(element.to_string())
|
||
}
|
||
}
|
||
if source.is_none() {
|
||
for element in elements {
|
||
if SOURCE_L2_RE.is_match(element) {
|
||
source = Some(element.to_string())
|
||
}
|
||
}
|
||
}
|
||
(clear_sub(sub), resolution, source)
|
||
}
|
||
|
||
pub fn check_is_movie(title: &str) -> bool {
|
||
MOVIE_TITLE_RE.is_match(title)
|
||
}
|
||
|
||
pub fn parse_episode_meta_from_raw_name(s: &str) -> eyre::Result<RawEpisodeMeta> {
|
||
let raw_title = s.trim();
|
||
let raw_title_without_ch_brackets = replace_ch_bracket_to_en(raw_title);
|
||
let fansub = extract_fansub(&raw_title_without_ch_brackets);
|
||
let movie_capture = check_is_movie(&raw_title_without_ch_brackets);
|
||
if let Some(title_re_match_obj) = MOVIE_TITLE_RE
|
||
.captures(&raw_title_without_ch_brackets)
|
||
.or(TITLE_RE.captures(&raw_title_without_ch_brackets))
|
||
{
|
||
let mut title_body = title_re_match_obj
|
||
.get(1)
|
||
.map(|s| s.as_str().trim())
|
||
.unwrap_or_else(|| unreachable!("TITLE_RE has at least 3 capture groups"))
|
||
.to_string();
|
||
let mut title_episode = title_re_match_obj
|
||
.get(2)
|
||
.map(|s| s.as_str().trim())
|
||
.unwrap_or_else(|| unreachable!("TITLE_RE has at least 3 capture groups"));
|
||
let title_extra = title_re_match_obj
|
||
.get(3)
|
||
.map(|s| s.as_str().trim())
|
||
.unwrap_or_else(|| unreachable!("TITLE_RE has at least 3 capture groups"));
|
||
|
||
if movie_capture {
|
||
title_body += title_episode;
|
||
title_episode = "";
|
||
} else if EP_COLLECTION_RE.is_match(&title_episode) {
|
||
title_episode = "";
|
||
}
|
||
|
||
let title_body = title_body_pre_process(&title_body, fansub)?;
|
||
let (name_without_season, season_raw, season) = extract_season_from_title_body(&title_body);
|
||
let (name_en, name_zh, name_jp) = extract_name_from_title_body_name_section(&title_body);
|
||
let (name_en_no_season, name_zh_no_season, name_jp_no_season) =
|
||
extract_name_from_title_body_name_section(&name_without_season);
|
||
let episode_index = extract_episode_index_from_title_episode(title_episode).unwrap_or(1);
|
||
let (sub, resolution, source) = extract_tags_from_title_extra(title_extra);
|
||
Ok(RawEpisodeMeta {
|
||
name_en,
|
||
name_en_no_season,
|
||
name_jp,
|
||
name_jp_no_season,
|
||
name_zh,
|
||
name_zh_no_season,
|
||
season,
|
||
season_raw,
|
||
episode_index,
|
||
subtitle: sub,
|
||
source,
|
||
fansub: fansub.map(|s| s.to_string()),
|
||
resolution,
|
||
})
|
||
} else {
|
||
Err(eyre::eyre!(
|
||
"Can not parse episode meta from raw filename {}",
|
||
raw_title
|
||
))
|
||
}
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
|
||
use super::{parse_episode_meta_from_raw_name, RawEpisodeMeta};
|
||
|
||
fn test_raw_ep_parser_case(raw_name: &str, expected: &str) {
|
||
let expected: Option<RawEpisodeMeta> = serde_json::from_str(expected).unwrap_or_default();
|
||
let found = parse_episode_meta_from_raw_name(raw_name).ok();
|
||
|
||
if expected != found {
|
||
println!(
|
||
"expected {} and found {} are not equal",
|
||
serde_json::to_string_pretty(&expected).unwrap(),
|
||
serde_json::to_string_pretty(&found).unwrap()
|
||
)
|
||
}
|
||
assert_eq!(expected, found);
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_all_parts_wrapped() {
|
||
test_raw_ep_parser_case(
|
||
r#"[新Sub][1月新番][我心里危险的东西 第二季][05][HEVC][10Bit][1080P][简日双语][招募翻译]"#,
|
||
r#"{
|
||
"name_zh": "我心里危险的东西",
|
||
"name_zh_no_season": "我心里危险的东西",
|
||
"season": 2,
|
||
"season_raw": "第二季",
|
||
"episode_index": 5,
|
||
"subtitle": "简日双语",
|
||
"source": null,
|
||
"fansub": "新Sub",
|
||
"resolution": "1080P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_title_wrapped_by_one_square_bracket_and_season_prefix() {
|
||
test_raw_ep_parser_case(
|
||
r#"【喵萌奶茶屋】★01月新番★[我内心的糟糕念头 / Boku no Kokoro no Yabai Yatsu][18][1080p][简日双语][招募翻译]"#,
|
||
r#"{
|
||
"name_en": "Boku no Kokoro no Yabai Yatsu",
|
||
"name_en_no_season": "Boku no Kokoro no Yabai Yatsu",
|
||
"name_zh": "我内心的糟糕念头",
|
||
"name_zh_no_season": "我内心的糟糕念头",
|
||
"season": 1,
|
||
"season_raw": null,
|
||
"episode_index": 18,
|
||
"subtitle": "简日双语",
|
||
"source": null,
|
||
"fansub": "喵萌奶茶屋",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
);
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_ep_and_version() {
|
||
test_raw_ep_parser_case(
|
||
r#"[LoliHouse] 因为不是真正的伙伴而被逐出勇者队伍，流落到边境展开慢活人生 2nd / Shin no Nakama 2nd - 08v2 [WebRip 1080p HEVC-10bit AAC][简繁内封字幕]"#,
|
||
r#"{
|
||
"name_en": "Shin no Nakama 2nd",
|
||
"name_en_no_season": "Shin no Nakama",
|
||
"name_zh": "因为不是真正的伙伴而被逐出勇者队伍，流落到边境展开慢活人生 2nd",
|
||
"name_zh_no_season": "因为不是真正的伙伴而被逐出勇者队伍，流落到边境展开慢活人生",
|
||
"season": 2,
|
||
"season_raw": "2nd",
|
||
"episode_index": 8,
|
||
"subtitle": "简繁内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_en_title_only() {
|
||
test_raw_ep_parser_case(
|
||
r"[动漫国字幕组&LoliHouse] THE MARGINAL SERVICE - 08 [WebRip 1080p HEVC-10bit AAC][简繁内封字幕]",
|
||
r#"{
|
||
"name_en": "THE MARGINAL SERVICE",
|
||
"name_en_no_season": "THE MARGINAL SERVICE",
|
||
"season": 1,
|
||
"episode_index": 8,
|
||
"subtitle": "简繁内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "动漫国字幕组&LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_two_zh_title() {
|
||
test_raw_ep_parser_case(
|
||
r#"[LoliHouse] 事与愿违的不死冒险者 / 非自愿的不死冒险者 / Nozomanu Fushi no Boukensha - 01 [WebRip 1080p HEVC-10bit AAC][简繁内封字幕]"#,
|
||
r#"{
|
||
"name_en": "Nozomanu Fushi no Boukensha",
|
||
"name_en_no_season": "Nozomanu Fushi no Boukensha",
|
||
"name_zh": "事与愿违的不死冒险者",
|
||
"name_zh_no_season": "事与愿违的不死冒险者",
|
||
"season": 1,
|
||
"season_raw": null,
|
||
"episode_index": 1,
|
||
"subtitle": "简繁内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_en_zh_jp_titles() {
|
||
test_raw_ep_parser_case(
|
||
r#"[喵萌奶茶屋&LoliHouse] 碰之道 / ぽんのみち / Pon no Michi - 07 [WebRip 1080p HEVC-10bit AAC][简繁日内封字幕]"#,
|
||
r#"{
|
||
"name_en": "Pon no Michi",
|
||
"name_jp": "ぽんのみち",
|
||
"name_zh": "碰之道",
|
||
"name_en_no_season": "Pon no Michi",
|
||
"name_jp_no_season": "ぽんのみち",
|
||
"name_zh_no_season": "碰之道",
|
||
"season": 1,
|
||
"season_raw": null,
|
||
"episode_index": 7,
|
||
"subtitle": "简繁日内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "喵萌奶茶屋&LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_nth_season() {
|
||
test_raw_ep_parser_case(
|
||
r#"[ANi] Yowai Character Tomozakikun / 弱角友崎同学 2nd STAGE - 09 [1080P][Baha][WEB-DL][AAC AVC][CHT][MP4]"#,
|
||
r#"{
|
||
"name_en": "Yowai Character Tomozakikun",
|
||
"name_en_no_season": "Yowai Character Tomozakikun",
|
||
"name_zh": "弱角友崎同学 2nd STAGE",
|
||
"name_zh_no_season": "弱角友崎同学",
|
||
"season": 2,
|
||
"season_raw": "2nd",
|
||
"episode_index": 9,
|
||
"subtitle": "CHT",
|
||
"source": "Baha",
|
||
"fansub": "ANi",
|
||
"resolution": "1080P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_season_en_and_season_zh() {
|
||
test_raw_ep_parser_case(
|
||
r#"[豌豆字幕组&LoliHouse] 王者天下 第五季 / Kingdom S5 - 07 [WebRip 1080p HEVC-10bit AAC][简繁外挂字幕]"#,
|
||
r#"{
|
||
"name_en": "Kingdom S5",
|
||
"name_en_no_season": "Kingdom",
|
||
"name_zh": "王者天下 第五季",
|
||
"name_zh_no_season": "王者天下",
|
||
"season": 5,
|
||
"season_raw": "第五季",
|
||
"episode_index": 7,
|
||
"subtitle": "简繁外挂字幕",
|
||
"source": "WebRip",
|
||
"fansub": "豌豆字幕组&LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_airota_fansub_style_case1() {
|
||
test_raw_ep_parser_case(
|
||
r#"【千夏字幕组】【爱丽丝与特蕾丝的虚幻工厂_Alice to Therese no Maboroshi Koujou】[剧场版][WebRip_1080p_HEVC][简繁内封][招募新人]"#,
|
||
r#"{
|
||
"name_en": "Alice to Therese no Maboroshi Koujou",
|
||
"name_en_no_season": "Alice to Therese no Maboroshi Koujou",
|
||
"name_zh": "爱丽丝与特蕾丝的虚幻工厂",
|
||
"name_zh_no_season": "爱丽丝与特蕾丝的虚幻工厂",
|
||
"season": 1,
|
||
"episode_index": 1,
|
||
"subtitle": "简繁内封",
|
||
"source": "WebRip",
|
||
"fansub": "千夏字幕组",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_airota_fansub_style_case2() {
|
||
test_raw_ep_parser_case(
|
||
r#"[千夏字幕组&喵萌奶茶屋][电影 轻旅轻营 (摇曳露营) _Yuru Camp Movie][剧场版][UHDRip_2160p_HEVC][繁体][千夏15周年]"#,
|
||
r#"{
|
||
"name_en": "Yuru Camp Movie",
|
||
"name_en_no_season": "Yuru Camp Movie",
|
||
"name_zh": "电影 轻旅轻营 (摇曳露营)",
|
||
"name_zh_no_season": "电影 轻旅轻营 (摇曳露营)",
|
||
"season": 1,
|
||
"episode_index": 1,
|
||
"subtitle": "繁体",
|
||
"source": "UHDRip",
|
||
"fansub": "千夏字幕组&喵萌奶茶屋",
|
||
"resolution": "2160p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_large_episode_style() {
|
||
test_raw_ep_parser_case(
|
||
r#"[梦蓝字幕组]New Doraemon 哆啦A梦新番[747][2023.02.25][AVC][1080P][GB_JP][MP4]"#,
|
||
r#"{
|
||
"name_en": "New Doraemon",
|
||
"name_en_no_season": "New Doraemon",
|
||
"name_zh": "哆啦A梦新番",
|
||
"name_zh_no_season": "哆啦A梦新番",
|
||
"season": 1,
|
||
"episode_index": 747,
|
||
"subtitle": "GB",
|
||
"fansub": "梦蓝字幕组",
|
||
"resolution": "1080P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_many_square_brackets_split_title() {
|
||
test_raw_ep_parser_case(
|
||
r#"【MCE汉化组】[剧场版-摇曳露营][Yuru Camp][Movie][简日双语][1080P][x264 AAC]"#,
|
||
r#"{
|
||
"name_en": "Yuru Camp",
|
||
"name_en_no_season": "Yuru Camp",
|
||
"name_zh": "剧场版-摇曳露营",
|
||
"name_zh_no_season": "剧场版-摇曳露营",
|
||
"season": 1,
|
||
"episode_index": 1,
|
||
"subtitle": "简日双语",
|
||
"fansub": "MCE汉化组",
|
||
"resolution": "1080P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_implicit_lang_title_sep() {
|
||
test_raw_ep_parser_case(
|
||
r#"[织梦字幕组][尼尔：机械纪元 NieR Automata Ver1.1a][02集][1080P][AVC][简日双语]"#,
|
||
r#"{
|
||
"name_en": "NieR Automata Ver1.1a",
|
||
"name_en_no_season": "NieR Automata Ver1.1a",
|
||
"name_zh": "尼尔：机械纪元",
|
||
"name_zh_no_season": "尼尔：机械纪元",
|
||
"season": 1,
|
||
"episode_index": 2,
|
||
"subtitle": "简日双语",
|
||
"fansub": "织梦字幕组",
|
||
"resolution": "1080P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_square_brackets_wrapped_and_space_split() {
|
||
test_raw_ep_parser_case(
|
||
r#"[天月搬运组][迷宫饭 Delicious in Dungeon][03][日语中字][MKV][1080P][NETFLIX][高画质版]"#,
|
||
r#"
|
||
{
|
||
"name_en": "Delicious in Dungeon",
|
||
"name_en_no_season": "Delicious in Dungeon",
|
||
"name_zh": "迷宫饭",
|
||
"name_zh_no_season": "迷宫饭",
|
||
"season": 1,
|
||
"episode_index": 3,
|
||
"subtitle": "日语中字",
|
||
"source": "NETFLIX",
|
||
"fansub": "天月搬运组",
|
||
"resolution": "1080P"
|
||
}
|
||
"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_start_with_brackets_wrapped_season_info_prefix() {
|
||
test_raw_ep_parser_case(
|
||
r#"[爱恋字幕社][1月新番][迷宫饭][Dungeon Meshi][01][1080P][MP4][简日双语] "#,
|
||
r#"{
|
||
"name_en": "Dungeon Meshi",
|
||
"name_en_no_season": "Dungeon Meshi",
|
||
"name_zh": "迷宫饭",
|
||
"name_zh_no_season": "迷宫饭",
|
||
"season": 1,
|
||
"episode_index": 1,
|
||
"subtitle": "简日双语",
|
||
"fansub": "爱恋字幕社",
|
||
"resolution": "1080P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_small_no_title_extra_brackets_case() {
|
||
test_raw_ep_parser_case(
|
||
r#"[ANi] Mahou Shoujo ni Akogarete / 梦想成为魔法少女 [年龄限制版] - 09 [1080P][Baha][WEB-DL][AAC AVC][CHT][MP4]"#,
|
||
r#"{
|
||
"name_en": "Mahou Shoujo ni Akogarete",
|
||
"name_en_no_season": "Mahou Shoujo ni Akogarete",
|
||
"name_zh": "梦想成为魔法少女 [年龄限制版]",
|
||
"name_zh_no_season": "梦想成为魔法少女 [年龄限制版]",
|
||
"season": 1,
|
||
"episode_index": 9,
|
||
"subtitle": "CHT",
|
||
"source": "Baha",
|
||
"fansub": "ANi",
|
||
"resolution": "1080P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_title_leading_space_style() {
|
||
test_raw_ep_parser_case(
|
||
r#"[ANi] 16bit 的感动 ANOTHER LAYER - 01 [1080P][Baha][WEB-DL][AAC AVC][CHT][MP4]"#,
|
||
r#"{
|
||
"name_zh": "16bit 的感动 ANOTHER LAYER",
|
||
"name_zh_no_season": "16bit 的感动 ANOTHER LAYER",
|
||
"season": 1,
|
||
"season_raw": null,
|
||
"episode_index": 1,
|
||
"subtitle": "CHT",
|
||
"source": "Baha",
|
||
"fansub": "ANi",
|
||
"resolution": "1080P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_title_leading_month_and_wrapped_brackets_style() {
|
||
test_raw_ep_parser_case(
|
||
r#"【喵萌奶茶屋】★07月新番★[银砂糖师与黑妖精 ~ Sugar Apple Fairy Tale ~][13][1080p][简日双语][招募翻译]"#,
|
||
r#"{
|
||
"name_en": "~ Sugar Apple Fairy Tale ~",
|
||
"name_en_no_season": "~ Sugar Apple Fairy Tale ~",
|
||
"name_zh": "银砂糖师与黑妖精",
|
||
"name_zh_no_season": "银砂糖师与黑妖精",
|
||
"season": 1,
|
||
"episode_index": 13,
|
||
"subtitle": "简日双语",
|
||
"fansub": "喵萌奶茶屋",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_title_leading_month_style() {
|
||
test_raw_ep_parser_case(
|
||
r#"【极影字幕社】★4月新番 天国大魔境 Tengoku Daimakyou 第05话 GB 720P MP4（字幕社招人内详）"#,
|
||
r#"{
|
||
"name_en": "Tengoku Daimakyou",
|
||
"name_en_no_season": "Tengoku Daimakyou",
|
||
"name_zh": "天国大魔境",
|
||
"name_zh_no_season": "天国大魔境",
|
||
"season": 1,
|
||
"episode_index": 5,
|
||
"subtitle": "字幕社招人内详",
|
||
"source": null,
|
||
"fansub": "极影字幕社",
|
||
"resolution": "720P"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_tokusatsu_style() {
|
||
test_raw_ep_parser_case(
|
||
r#"[MagicStar] 假面骑士Geats / 仮面ライダーギーツ EP33 [WEBDL] [1080p] [TTFC]【生】"#,
|
||
r#"{
|
||
"name_jp": "仮面ライダーギーツ",
|
||
"name_jp_no_season": "仮面ライダーギーツ",
|
||
"name_zh": "假面骑士Geats",
|
||
"name_zh_no_season": "假面骑士Geats",
|
||
"season": 1,
|
||
"episode_index": 33,
|
||
"source": "WEBDL",
|
||
"fansub": "MagicStar",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_parse_ep_with_multi_lang_zh_title() {
|
||
test_raw_ep_parser_case(
|
||
r#"[百冬练习组&LoliHouse] BanG Dream! 少女乐团派对！☆PICO FEVER！ / Garupa Pico: Fever! - 26 [WebRip 1080p HEVC-10bit AAC][简繁内封字幕][END] [101.69 MB]"#,
|
||
r#"{
|
||
"name_en": "Garupa Pico: Fever!",
|
||
"name_en_no_season": "Garupa Pico: Fever!",
|
||
"name_zh": "BanG Dream! 少女乐团派对！☆PICO FEVER！",
|
||
"name_zh_no_season": "BanG Dream! 少女乐团派对！☆PICO FEVER！",
|
||
"season": 1,
|
||
"episode_index": 26,
|
||
"subtitle": "简繁内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "百冬练习组&LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
)
|
||
}
|
||
|
||
#[test]
|
||
fn test_ep_collections() {
|
||
test_raw_ep_parser_case(
|
||
r#"[奶²&LoliHouse] 蘑菇狗 / Kinokoinu: Mushroom Pup [01-12 精校合集][WebRip 1080p HEVC-10bit AAC][简日内封字幕]"#,
|
||
r#"{
|
||
"name_en": "Kinokoinu: Mushroom Pup",
|
||
"name_en_no_season": "Kinokoinu: Mushroom Pup",
|
||
"name_zh": "蘑菇狗",
|
||
"name_zh_no_season": "蘑菇狗",
|
||
"season": 1,
|
||
"episode_index": 1,
|
||
"subtitle": "简日内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "奶²&LoliHouse",
|
||
"resolution": "1080p",
|
||
"name": " 蘑菇狗 / Kinokoinu: Mushroom Pup [01-12 精校合集]"
|
||
}"#,
|
||
);
|
||
|
||
test_raw_ep_parser_case(
|
||
r#"[LoliHouse] 叹气的亡灵想隐退 / Nageki no Bourei wa Intai shitai [01-13 合集][WebRip 1080p HEVC-10bit AAC][简繁内封字幕][Fin]"#,
|
||
r#"{
|
||
"name_en": "Nageki no Bourei wa Intai shitai",
|
||
"name_en_no_season": "Nageki no Bourei wa Intai shitai",
|
||
"name_jp": null,
|
||
"name_jp_no_season": null,
|
||
"name_zh": "叹气的亡灵想隐退",
|
||
"name_zh_no_season": "叹气的亡灵想隐退",
|
||
"season": 1,
|
||
"season_raw": null,
|
||
"episode_index": 1,
|
||
"subtitle": "简繁内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
);
|
||
|
||
test_raw_ep_parser_case(
|
||
r#"[LoliHouse] 精灵幻想记 第二季 / Seirei Gensouki S2 [01-12 合集][WebRip 1080p HEVC-10bit AAC][简繁内封字幕][Fin]"#,
|
||
r#"{
|
||
"name_en": "Seirei Gensouki S2",
|
||
"name_en_no_season": "Seirei Gensouki",
|
||
"name_zh": "精灵幻想记 第二季",
|
||
"name_zh_no_season": "精灵幻想记",
|
||
"season": 2,
|
||
"season_raw": "第二季",
|
||
"episode_index": 1,
|
||
"subtitle": "简繁内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
);
|
||
|
||
test_raw_ep_parser_case(
|
||
r#"[喵萌奶茶屋&LoliHouse] 超自然武装当哒当 / 胆大党 / Dandadan [01-12 精校合集][WebRip 1080p HEVC-10bit AAC][简繁日内封字幕][Fin]"#,
|
||
r#" {
|
||
"name_en": "Dandadan",
|
||
"name_en_no_season": "Dandadan",
|
||
"name_zh": "超自然武装当哒当",
|
||
"name_zh_no_season": "超自然武装当哒当",
|
||
"season": 1,
|
||
"episode_index": 1,
|
||
"subtitle": "简繁日内封字幕",
|
||
"source": "WebRip",
|
||
"fansub": "喵萌奶茶屋&LoliHouse",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
);
|
||
}
|
||
|
||
// TODO: FIXME
|
||
#[test]
|
||
fn test_bad_cases() {
|
||
test_raw_ep_parser_case(
|
||
r#"[7³ACG x 桜都字幕组] 摇曳露营△ 剧场版/映画 ゆるキャン△/Eiga Yuru Camp△ [简繁字幕] BDrip 1080p x265 FLAC 2.0"#,
|
||
r#"{
|
||
"name_zh": "摇曳露营△剧场版",
|
||
"name_zh_no_season": "摇曳露营△剧场版",
|
||
"season": 1,
|
||
"season_raw": null,
|
||
"episode_index": 1,
|
||
"subtitle": "简繁字幕",
|
||
"source": "BDrip",
|
||
"fansub": "7³ACG x 桜都字幕组",
|
||
"resolution": "1080p"
|
||
}"#,
|
||
);
|
||
|
||
test_raw_ep_parser_case(
|
||
r#"【幻樱字幕组】【4月新番】【古见同学有交流障碍症 第二季 Komi-san wa, Komyushou Desu. S02】【22】【GB_MP4】【1920X1080】"#,
|
||
r#"{
|
||
"name_en": "第二季 Komi-san wa, Komyushou Desu. S02",
|
||
"name_en_no_season": "Komi-san wa, Komyushou Desu.",
|
||
"name_zh": "古见同学有交流障碍症",
|
||
"name_zh_no_season": "古见同学有交流障碍症",
|
||
"season": 2,
|
||
"season_raw": "第二季",
|
||
"episode_index": 22,
|
||
"subtitle": "GB",
|
||
"fansub": "幻樱字幕组",
|
||
"resolution": "1920X1080"
|
||
}"#,
|
||
);
|
||
}
|
||
}
|