266 lines
11 KiB
Rust
266 lines
11 KiB
Rust
use bili_sync_entity::rule::{AndGroup, Condition, Rule, RuleTarget};
|
||
use bili_sync_entity::{page, video};
|
||
use chrono::{Local, NaiveDateTime};
|
||
|
||
pub(crate) trait Evaluatable<T> {
|
||
fn evaluate(&self, value: T) -> bool;
|
||
}
|
||
|
||
pub(crate) trait FieldEvaluatable {
|
||
fn evaluate(&self, video: &video::ActiveModel, pages: &[page::ActiveModel]) -> bool;
|
||
fn evaluate_model(&self, video: &video::Model, pages: &[page::Model]) -> bool;
|
||
}
|
||
|
||
impl Evaluatable<&str> for Condition<String> {
|
||
fn evaluate(&self, value: &str) -> bool {
|
||
match self {
|
||
Condition::Equals(expected) => expected == value,
|
||
Condition::Contains(substring) => value.contains(substring),
|
||
Condition::Prefix(prefix) => value.starts_with(prefix),
|
||
Condition::Suffix(suffix) => value.ends_with(suffix),
|
||
Condition::MatchesRegex(_, regex) => regex.is_match(value),
|
||
_ => false,
|
||
}
|
||
}
|
||
}
|
||
|
||
impl Evaluatable<usize> for Condition<usize> {
|
||
fn evaluate(&self, value: usize) -> bool {
|
||
match self {
|
||
Condition::Equals(expected) => *expected == value,
|
||
Condition::GreaterThan(threshold) => value > *threshold,
|
||
Condition::LessThan(threshold) => value < *threshold,
|
||
Condition::Between(start, end) => value > *start && value < *end,
|
||
_ => false,
|
||
}
|
||
}
|
||
}
|
||
|
||
impl Evaluatable<&NaiveDateTime> for Condition<NaiveDateTime> {
|
||
fn evaluate(&self, value: &NaiveDateTime) -> bool {
|
||
match self {
|
||
Condition::Equals(expected) => expected == value,
|
||
Condition::GreaterThan(threshold) => value > threshold,
|
||
Condition::LessThan(threshold) => value < threshold,
|
||
Condition::Between(start, end) => value > start && value < end,
|
||
_ => false,
|
||
}
|
||
}
|
||
}
|
||
|
||
impl FieldEvaluatable for RuleTarget {
|
||
/// 修改模型后进行评估，此时能访问的是未保存的 activeModel，就地使用 activeModel 评估
|
||
fn evaluate(&self, video: &video::ActiveModel, pages: &[page::ActiveModel]) -> bool {
|
||
match self {
|
||
RuleTarget::Title(cond) => video.name.try_as_ref().is_some_and(|title| cond.evaluate(title)),
|
||
// 目前的所有条件都是分别针对全体标签进行 any 评估的，例如 Prefix("a") && Suffix("b") 意味着 any(tag.Prefix("a")) && any(tag.Suffix("b")) 而非 any(tag.Prefix("a") && tag.Suffix("b"))
|
||
// 这可能不满足用户预期，但应该问题不大，如果真有很多人用复杂标签筛选再单独改
|
||
RuleTarget::Tags(cond) => video
|
||
.tags
|
||
.try_as_ref()
|
||
.and_then(|t| t.as_ref())
|
||
.is_some_and(|tags| tags.0.iter().any(|tag| cond.evaluate(tag))),
|
||
RuleTarget::FavTime(cond) => video
|
||
.favtime
|
||
.try_as_ref()
|
||
.map(|fav_time| fav_time.and_utc().with_timezone(&Local).naive_local()) // 数据库中保存的一律是 utc 时间，转换为 local 时间再比较
|
||
.is_some_and(|fav_time| cond.evaluate(&fav_time)),
|
||
RuleTarget::PubTime(cond) => video
|
||
.pubtime
|
||
.try_as_ref()
|
||
.map(|pub_time| pub_time.and_utc().with_timezone(&Local).naive_local())
|
||
.is_some_and(|pub_time| cond.evaluate(&pub_time)),
|
||
RuleTarget::PageCount(cond) => cond.evaluate(pages.len()),
|
||
RuleTarget::Not(inner) => !inner.evaluate(video, pages),
|
||
}
|
||
}
|
||
|
||
/// 手动触发对历史视频的评估，拿到的是原始 Model，直接使用
|
||
fn evaluate_model(&self, video: &video::Model, pages: &[page::Model]) -> bool {
|
||
match self {
|
||
RuleTarget::Title(cond) => cond.evaluate(&video.name),
|
||
// 目前的所有条件都是分别针对全体标签进行 any 评估的，例如 Prefix("a") && Suffix("b") 意味着 any(tag.Prefix("a")) && any(tag.Suffix("b")) 而非 any(tag.Prefix("a") && tag.Suffix("b"))
|
||
// 这可能不满足用户预期，但应该问题不大，如果真有很多人用复杂标签筛选再单独改
|
||
RuleTarget::Tags(cond) => video
|
||
.tags
|
||
.as_ref()
|
||
.is_some_and(|tags| tags.0.iter().any(|tag| cond.evaluate(tag))),
|
||
RuleTarget::FavTime(cond) => cond.evaluate(&video.favtime.and_utc().with_timezone(&Local).naive_local()),
|
||
RuleTarget::PubTime(cond) => cond.evaluate(&video.pubtime.and_utc().with_timezone(&Local).naive_local()),
|
||
RuleTarget::PageCount(cond) => cond.evaluate(pages.len()),
|
||
RuleTarget::Not(inner) => !inner.evaluate_model(video, pages),
|
||
}
|
||
}
|
||
}
|
||
|
||
impl FieldEvaluatable for AndGroup {
|
||
fn evaluate(&self, video: &video::ActiveModel, pages: &[page::ActiveModel]) -> bool {
|
||
self.iter().all(|target| target.evaluate(video, pages))
|
||
}
|
||
|
||
fn evaluate_model(&self, video: &video::Model, pages: &[page::Model]) -> bool {
|
||
self.iter().all(|target| target.evaluate_model(video, pages))
|
||
}
|
||
}
|
||
|
||
impl FieldEvaluatable for Rule {
|
||
fn evaluate(&self, video: &video::ActiveModel, pages: &[page::ActiveModel]) -> bool {
|
||
if self.0.is_empty() {
|
||
return true;
|
||
}
|
||
self.0.iter().any(|group| group.evaluate(video, pages))
|
||
}
|
||
|
||
fn evaluate_model(&self, video: &video::Model, pages: &[page::Model]) -> bool {
|
||
if self.0.is_empty() {
|
||
return true;
|
||
}
|
||
self.0.iter().any(|group| group.evaluate_model(video, pages))
|
||
}
|
||
}
|
||
|
||
/// 对于 Option<Rule> 如果 rule 不存在应该被认为是通过评估
|
||
impl FieldEvaluatable for Option<Rule> {
|
||
fn evaluate(&self, video: &video::ActiveModel, pages: &[page::ActiveModel]) -> bool {
|
||
self.as_ref().is_none_or(|rule| rule.evaluate(video, pages))
|
||
}
|
||
|
||
fn evaluate_model(&self, video: &video::Model, pages: &[page::Model]) -> bool {
|
||
self.as_ref().is_none_or(|rule| rule.evaluate_model(video, pages))
|
||
}
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use bili_sync_entity::page;
|
||
use chrono::NaiveDate;
|
||
use sea_orm::ActiveValue::Set;
|
||
|
||
use super::*;
|
||
|
||
#[test]
|
||
fn test_display() {
|
||
let test_cases = vec![
|
||
(
|
||
Rule(vec![vec![RuleTarget::Title(Condition::Contains("唐氏".to_string()))]]),
|
||
"「(标题包含“唐氏”)」",
|
||
),
|
||
(
|
||
Rule(vec![vec![
|
||
RuleTarget::Title(Condition::Prefix("街霸".to_string())),
|
||
RuleTarget::Tags(Condition::Contains("套路".to_string())),
|
||
]]),
|
||
"「(标题以“街霸”开头)且(标签包含“套路”)」",
|
||
),
|
||
(
|
||
Rule(vec![
|
||
vec![
|
||
RuleTarget::Title(Condition::Contains("Rust".to_string())),
|
||
RuleTarget::PageCount(Condition::GreaterThan(5)),
|
||
],
|
||
vec![
|
||
RuleTarget::Tags(Condition::Suffix("入门".to_string())),
|
||
RuleTarget::PubTime(Condition::GreaterThan(
|
||
NaiveDate::from_ymd_opt(2023, 1, 1)
|
||
.unwrap()
|
||
.and_hms_opt(0, 0, 0)
|
||
.unwrap(),
|
||
)),
|
||
],
|
||
]),
|
||
"「(标题包含“Rust”)且(视频分页数量大于“5”)」或「(标签以“入门”结尾)且(发布时间大于“2023-01-01 00:00:00”)」",
|
||
),
|
||
(
|
||
Rule(vec![vec![
|
||
RuleTarget::Not(Box::new(RuleTarget::Title(Condition::Contains("广告".to_string())))),
|
||
RuleTarget::PageCount(Condition::LessThan(10)),
|
||
]]),
|
||
"「(标题不包含“广告”)且(视频分页数量小于“10”)」",
|
||
),
|
||
(
|
||
Rule(vec![vec![
|
||
RuleTarget::FavTime(Condition::Between(
|
||
NaiveDate::from_ymd_opt(2023, 6, 1)
|
||
.unwrap()
|
||
.and_hms_opt(0, 0, 0)
|
||
.unwrap(),
|
||
NaiveDate::from_ymd_opt(2023, 12, 31)
|
||
.unwrap()
|
||
.and_hms_opt(23, 59, 59)
|
||
.unwrap(),
|
||
)),
|
||
RuleTarget::Tags(Condition::MatchesRegex(
|
||
"技术|教程".to_string(),
|
||
regex::Regex::new("技术|教程").unwrap(),
|
||
)),
|
||
]]),
|
||
"「(收藏时间在“2023-06-01 00:00:00”和“2023-12-31 23:59:59”之间)且(标签匹配“技术|教程”)」",
|
||
),
|
||
];
|
||
|
||
for (rule, expected) in test_cases {
|
||
assert_eq!(rule.to_string(), expected);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_evaluate() {
|
||
let test_cases = vec![
|
||
(
|
||
(
|
||
video::ActiveModel {
|
||
name: Set("骂谁唐氏呢！！！".to_string()),
|
||
..Default::default()
|
||
},
|
||
vec![],
|
||
),
|
||
Rule(vec![vec![RuleTarget::Title(Condition::Contains("唐氏".to_string()))]]),
|
||
true,
|
||
),
|
||
(
|
||
(
|
||
video::ActiveModel::default(),
|
||
vec![page::ActiveModel::default(); 2],
|
||
),
|
||
Rule(vec![vec![RuleTarget::PageCount(Condition::Equals(1))]]),
|
||
false,
|
||
),
|
||
(
|
||
(
|
||
video::ActiveModel{
|
||
tags: Set(Some(vec!["原神".to_owned(),"永雏塔菲".to_owned(),"虚拟主播".to_owned()].into())),
|
||
..Default::default()
|
||
},
|
||
vec![],
|
||
),
|
||
Rule (vec![vec![RuleTarget::Not(Box::new(RuleTarget::Tags(Condition::Equals(
|
||
"原神".to_string(),
|
||
))))]],
|
||
),
|
||
false,
|
||
),
|
||
(
|
||
(
|
||
video::ActiveModel {
|
||
name: Set(
|
||
"万字怒扒网易《归唐》底裤！中国首款大厂买断制单机，靠谱吗？——全网最全！官方非独家幕后！关于《归唐》PV的所有秘密~都在这里了~".to_owned(),
|
||
),
|
||
..Default::default()
|
||
},
|
||
vec![],
|
||
),
|
||
Rule(vec![vec![RuleTarget::Not(Box::new(RuleTarget::Title(Condition::MatchesRegex(
|
||
r"^\S+字(解析|怒扒|拆解)".to_owned(),
|
||
regex::Regex::new(r"^\S+字(解析|怒扒)").unwrap(),
|
||
))))]],
|
||
),
|
||
false,
|
||
),
|
||
];
|
||
|
||
for ((video, pages), rule, expected) in test_cases {
|
||
assert_eq!(rule.evaluate(&video, &pages), expected);
|
||
}
|
||
}
|
||
}
|