feat: 大范围重构，支持视频合集下载 (#97)

2024-07-03 18:57:12 +08:00
parent 097f885050
commit 4c9ad2318c
27 changed files with 1545 additions and 446 deletions
--- a/crates/bili_sync/src/utils/convert.rs
+++ b/crates/bili_sync/src/utils/convert.rs
@@ -0,0 +1,131 @@
+use sea_orm::ActiveValue::NotSet;
+use sea_orm::{IntoActiveModel, Set};
+use serde_json::json;
+
+use crate::bilibili::VideoInfo;
+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 设置成 NotSet，否则 id 会是 Unchanged(0)
+                tmp_model.id = 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
+            },
+        }
+    }
+
+    pub fn to_fmt_args(&self) -> Option<serde_json::Value> {
+        match self {
+            VideoInfo::Simple { .. } => None, // 不能从简单的视频信息中构造格式化参数
+            VideoInfo::Detail { title, bvid, upper, .. } => Some(json!({
+                "bvid": &bvid,
+                "title": &title,
+                "upper_name": &upper.name,
+                "upper_mid": &upper.mid,
+            })),
+            VideoInfo::View { title, bvid, upper, .. } => Some(json!({
+                "bvid": &bvid,
+                "title": &title,
+                "upper_name": &upper.name,
+                "upper_mid": &upper.mid,
+            })),
+        }
+    }
+
+    pub fn video_key(&self) -> String {
+        match self {
+            // 对于合集没有 fav_time，只能用 pubtime 代替
+            VideoInfo::Simple { bvid, pubtime, .. } => id_time_key(bvid, pubtime),
+            VideoInfo::Detail { bvid, fav_time, .. } => id_time_key(bvid, fav_time),
+            // 详情接口返回的数据仅用于填充详情，不会被作为 video_key
+            _ => unreachable!(),
+        }
+    }
+
+    pub fn bvid(&self) -> &str {
+        match self {
+            VideoInfo::Simple { bvid, .. } => bvid,
+            VideoInfo::Detail { bvid, .. } => bvid,
+            // 同上
+            _ => unreachable!(),
+        }
+    }
+}