216 lines
7.5 KiB
Rust
216 lines
7.5 KiB
Rust
use sea_orm::ActiveValue::{NotSet, Set};
|
||
use sea_orm::IntoActiveModel;
|
||
use serde_json::json;
|
||
|
||
use crate::bilibili::VideoInfo;
|
||
use crate::config::CONFIG;
|
||
use crate::utils::id_time_key;
|
||
|
||
impl VideoInfo {
|
||
/// 将 VideoInfo 转换为 ActiveModel
|
||
pub fn to_model(&self, base_model: Option<bili_sync_entity::video::Model>) -> bili_sync_entity::video::ActiveModel {
|
||
let base_model = match base_model {
|
||
Some(base_model) => base_model.into_active_model(),
|
||
None => {
|
||
let mut tmp_model = bili_sync_entity::video::Model::default().into_active_model();
|
||
// 注意此处要把 id 和 created_at 设置为 NotSet，方便在 sql 中忽略这些字段，交由数据库自动生成
|
||
tmp_model.id = NotSet;
|
||
tmp_model.created_at = NotSet;
|
||
tmp_model
|
||
}
|
||
};
|
||
match self {
|
||
VideoInfo::Simple {
|
||
bvid,
|
||
cover,
|
||
ctime,
|
||
pubtime,
|
||
} => bili_sync_entity::video::ActiveModel {
|
||
bvid: Set(bvid.clone()),
|
||
cover: Set(cover.clone()),
|
||
ctime: Set(ctime.naive_utc()),
|
||
pubtime: Set(pubtime.naive_utc()),
|
||
category: Set(2), // 视频合集里的内容类型肯定是视频
|
||
valid: Set(true),
|
||
..base_model
|
||
},
|
||
VideoInfo::Detail {
|
||
title,
|
||
vtype,
|
||
bvid,
|
||
intro,
|
||
cover,
|
||
upper,
|
||
ctime,
|
||
fav_time,
|
||
pubtime,
|
||
attr,
|
||
} => bili_sync_entity::video::ActiveModel {
|
||
bvid: Set(bvid.clone()),
|
||
name: Set(title.clone()),
|
||
category: Set(*vtype),
|
||
intro: Set(intro.clone()),
|
||
cover: Set(cover.clone()),
|
||
ctime: Set(ctime.naive_utc()),
|
||
pubtime: Set(pubtime.naive_utc()),
|
||
favtime: Set(fav_time.naive_utc()),
|
||
download_status: Set(0),
|
||
valid: Set(*attr == 0),
|
||
tags: Set(None),
|
||
single_page: Set(None),
|
||
upper_id: Set(upper.mid),
|
||
upper_name: Set(upper.name.clone()),
|
||
upper_face: Set(upper.face.clone()),
|
||
..base_model
|
||
},
|
||
VideoInfo::View {
|
||
title,
|
||
bvid,
|
||
intro,
|
||
cover,
|
||
upper,
|
||
ctime,
|
||
pubtime,
|
||
state,
|
||
..
|
||
} => bili_sync_entity::video::ActiveModel {
|
||
bvid: Set(bvid.clone()),
|
||
name: Set(title.clone()),
|
||
category: Set(2), // 视频合集里的内容类型肯定是视频
|
||
intro: Set(intro.clone()),
|
||
cover: Set(cover.clone()),
|
||
ctime: Set(ctime.naive_utc()),
|
||
pubtime: Set(pubtime.naive_utc()),
|
||
favtime: Set(pubtime.naive_utc()), // 合集不包括 fav_time，使用发布时间代替
|
||
download_status: Set(0),
|
||
valid: Set(*state == 0),
|
||
tags: Set(None),
|
||
single_page: Set(None),
|
||
upper_id: Set(upper.mid),
|
||
upper_name: Set(upper.name.clone()),
|
||
upper_face: Set(upper.face.clone()),
|
||
..base_model
|
||
},
|
||
VideoInfo::WatchLater {
|
||
title,
|
||
bvid,
|
||
intro,
|
||
cover,
|
||
upper,
|
||
ctime,
|
||
fav_time,
|
||
pubtime,
|
||
state,
|
||
} => bili_sync_entity::video::ActiveModel {
|
||
bvid: Set(bvid.clone()),
|
||
name: Set(title.clone()),
|
||
category: Set(2), // 稍后再看里的内容类型肯定是视频
|
||
intro: Set(intro.clone()),
|
||
cover: Set(cover.clone()),
|
||
ctime: Set(ctime.naive_utc()),
|
||
pubtime: Set(pubtime.naive_utc()),
|
||
favtime: Set(fav_time.naive_utc()),
|
||
download_status: Set(0),
|
||
valid: Set(*state == 0),
|
||
tags: Set(None),
|
||
single_page: Set(None),
|
||
upper_id: Set(upper.mid),
|
||
upper_name: Set(upper.name.clone()),
|
||
upper_face: Set(upper.face.clone()),
|
||
..base_model
|
||
},
|
||
VideoInfo::Submission {
|
||
title,
|
||
bvid,
|
||
intro,
|
||
cover,
|
||
ctime,
|
||
} => bili_sync_entity::video::ActiveModel {
|
||
bvid: Set(bvid.clone()),
|
||
name: Set(title.clone()),
|
||
intro: Set(intro.clone()),
|
||
cover: Set(cover.clone()),
|
||
ctime: Set(ctime.naive_utc()),
|
||
category: Set(2), // 投稿视频的内容类型肯定是视频
|
||
valid: Set(true),
|
||
..base_model
|
||
},
|
||
}
|
||
}
|
||
|
||
pub fn to_fmt_args(&self) -> Option<serde_json::Value> {
|
||
match self {
|
||
VideoInfo::Simple { .. } | VideoInfo::Submission { .. } => None, // 不能从简单视频信息中构造格式化参数
|
||
VideoInfo::Detail {
|
||
title,
|
||
bvid,
|
||
upper,
|
||
pubtime,
|
||
fav_time,
|
||
..
|
||
}
|
||
| VideoInfo::WatchLater {
|
||
title,
|
||
bvid,
|
||
upper,
|
||
pubtime,
|
||
fav_time,
|
||
..
|
||
} => Some(json!({
|
||
"bvid": &bvid,
|
||
"title": &title,
|
||
"upper_name": &upper.name,
|
||
"upper_mid": &upper.mid,
|
||
"pubtime": pubtime.format(&CONFIG.time_format).to_string(),
|
||
"fav_time": fav_time.format(&CONFIG.time_format).to_string(),
|
||
})),
|
||
VideoInfo::View {
|
||
title,
|
||
bvid,
|
||
upper,
|
||
pubtime,
|
||
..
|
||
} => {
|
||
let pubtime = pubtime.format(&CONFIG.time_format).to_string();
|
||
Some(json!({
|
||
"bvid": &bvid,
|
||
"title": &title,
|
||
"upper_name": &upper.name,
|
||
"upper_mid": &upper.mid,
|
||
"pubtime": &pubtime,
|
||
"fav_time": &pubtime,
|
||
}))
|
||
}
|
||
}
|
||
}
|
||
|
||
pub fn video_key(&self) -> String {
|
||
match self {
|
||
// 对于合集没有 fav_time，只能用 pubtime 代替
|
||
VideoInfo::Simple {
|
||
bvid, pubtime: time, ..
|
||
}
|
||
| VideoInfo::Detail {
|
||
bvid, fav_time: time, ..
|
||
}
|
||
| VideoInfo::WatchLater {
|
||
bvid, fav_time: time, ..
|
||
}
|
||
| VideoInfo::Submission { bvid, ctime: time, .. } => id_time_key(bvid, time),
|
||
// 详情接口返回的数据仅用于填充详情，不会被作为 video_key
|
||
_ => unreachable!(),
|
||
}
|
||
}
|
||
|
||
pub fn bvid(&self) -> &str {
|
||
match self {
|
||
VideoInfo::Simple { bvid, .. }
|
||
| VideoInfo::Detail { bvid, .. }
|
||
| VideoInfo::WatchLater { bvid, .. }
|
||
| VideoInfo::Submission { bvid, .. } => bvid,
|
||
// 同上
|
||
_ => unreachable!(),
|
||
}
|
||
}
|
||
}
|