chore: add new option with_torrent. (#28)

RinChanNOWWW · Nov 4, 2023 · 413c2f6 · 413c2f6
1 parent be7d772
commit 413c2f6
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 11 deletions.
diff --git a/examples/config.toml b/examples/config.toml
@@ -20,3 +20,4 @@ api = "http://localhost:8000"
 dms = [114514, 1919810]
 groups = [114514, 1919810]
 delay = 200
+with_torrent = true
diff --git a/src/config/mod.rs b/src/config/mod.rs
@@ -34,6 +34,8 @@ pub struct QQBotConfig {
     pub groups: Vec<u64>,
     /// The delay between two messages. (ms)
     pub delay: u64,
+    /// If the message contains torrents' URLs.
+    pub with_torrent: bool,
 }
 
 #[derive(Debug, Default, Serialize, Deserialize)]

diff --git a/src/notifier/qq.rs b/src/notifier/qq.rs
@@ -116,24 +116,25 @@ impl QQNotifier {
     }
 
     fn wrap_item(notifier: &Notifier, source: &str, item: &Item) -> Vec<Message> {
-        vec![
-            Message {
-                msg_type: "node".to_string(),
-                data: Data {
-                    sender_name: notifier.conf.name.clone(),
-                    sender_uin: notifier.conf.uin.clone(),
-                    content: format!("{}:\n{} ({})", source, item.title, item.pub_date),
-                },
+        let mut messages = vec![Message {
+            msg_type: "node".to_string(),
+            data: Data {
+                sender_name: notifier.conf.name.clone(),
+                sender_uin: notifier.conf.uin.clone(),
+                content: format!("{}:\n{} ({})", source, item.title, item.pub_date),
             },
-            Message {
+        }];
+        if notifier.conf.with_torrent {
+            messages.push(Message {
                 msg_type: "node".to_string(),
                 data: Data {
                     sender_name: notifier.conf.name.clone(),
                     sender_uin: notifier.conf.uin.clone(),
                     content: item.url.clone(),
                 },
-            },
-        ]
+            });
+        }
+        messages
     }
 
     async fn send_private_msg(notifier: &Notifier, msg: Vec<Message>) -> Result<()> {