[bilibili.playback] Send XML metadata file updates

Author: SpriteOvO

src/helper.rs

@@ -120,6 +120,9 @@ pub async fn ffprobe_resolution(video: &Path) -> anyhow::Result<VideoResolution>
     Ok(VideoResolution { width, height })
 }
 
+// ffmpeg -i 录制-5208231-20241209-100743-972-王小桃模式.flv -c copy -fs 100M
+// 1.mp4
+
 #[cfg(test)]
 mod tests {
     use super::*;

src/source/platform/bilibili/playback/bililive_recorder.rs

@@ -8,7 +8,10 @@ use chrono::{DateTime, Local};
 use serde::Deserialize;
 use serde_json as json;
 use spdlog::prelude::*;
-use tokio::sync::{mpsc, Mutex};
+use tokio::{
+    fs,
+    sync::{mpsc, Mutex},
+};
 use warp::Filter;
 
 use super::PLATFORM_METADATA;
@@ -190,49 +193,61 @@ async fn handle(event: data::WebhookV2, params: &Context) -> anyhow::Result<()>
                 );
             }
 
-            let file_path = params.working_directory.join(&file_closed.relative_path);
-
-            enum FileType {
-                Video(PlaybackFormat),
-                Metadata,
-            }
+            let send_update = async |path: PathBuf| {
+                enum FileType {
+                    Video(PlaybackFormat),
+                    Metadata,
+                }
+
+                let file_type = match path
+                    .extension()
+                    .map(|ext| ext.to_string_lossy().to_ascii_lowercase())
+                    .as_deref()
+                {
+                    Some("flv") => FileType::Video(PlaybackFormat::Flv),
+                    Some("xml") => FileType::Metadata,
+                    _ => bail!(
+                        "bililive-recorder closed a file with an unknown extension '{path:?}'"
+                    ),
+                };
+
+                let status = Update::new(
+                    match file_type {
+                        FileType::Video(format) => UpdateKind::Playback(Playback {
+                            live_start_time: session.live_start_time,
+                            file_path: path,
+                            format,
+                        }),
+                        FileType::Metadata => UpdateKind::Document(Document { file_path: path }),
+                    },
+                    StatusSource {
+                        platform: PLATFORM_METADATA,
+                        user: None,
+                    },
+                );
 
-            let file_type = match file_path
-                .extension()
-                .map(|ext| ext.to_string_lossy().to_ascii_lowercase())
-                .as_deref()
-            {
-                Some("flv") => FileType::Video(PlaybackFormat::Flv),
-                Some("xml") => FileType::Metadata,
-                _ => bail!(
-                    "bililive-recorder closed a file with an unknown extension '{file_path:?}'"
-                ),
+                params
+                    .senders
+                    .lock()
+                    .await
+                    .get(&session.room_id)
+                    .ok_or_else(|| anyhow!("room id {} has no sender", session.room_id))?
+                    .send(status)
+                    .await
+                    .map_err(|err| anyhow!("failed to send status: {err}. session: {session:?}"))
             };
 
-            let status = Update::new(
-                match file_type {
-                    FileType::Video(format) => UpdateKind::Playback(Playback {
-                        live_start_time: session.live_start_time,
-                        file_path,
-                        format,
-                    }),
-                    FileType::Metadata => UpdateKind::Document(Document { file_path }),
-                },
-                StatusSource {
-                    platform: PLATFORM_METADATA,
-                    user: None,
-                },
-            );
-
-            params
-                .senders
-                .lock()
-                .await
-                .get(&session.room_id)
-                .ok_or_else(|| anyhow!("room id {} has no sender", session.room_id))?
-                .send(status)
-                .await
-                .map_err(|err| anyhow!("failed to send status: {err}. session: {session:?}"))
+            let playback_file = params.working_directory.join(&file_closed.relative_path);
+            let metadate_file = playback_file.with_extension("xml");
+
+            let ret1 = send_update(playback_file).await;
+            let mut ret2 = Ok(());
+            if let Ok(true) = fs::try_exists(&metadate_file).await {
+                ret2 = send_update(metadate_file).await;
+            }
+            ret1?;
+            ret2?;
+            Ok(())
         }
         data::EventKind::FileOpening {}
         | data::EventKind::StreamStarted {}

src/task/subscription.rs

@@ -101,7 +101,7 @@ impl Task for TaskSubscription {
         match self.sourcer.take().unwrap() {
             Sourcer::Fetcher(fetcher) => Box::pin(self.continuous_fetch(fetcher)),
             Sourcer::Listener(mut listener) => {
-                let (sender, receiver) = mpsc::channel(1);
+                let (sender, receiver) = mpsc::channel(10);
                 // TODO: A bit hacky, improve it?
                 let platform = listener.to_string();
                 Box::pin(async move {