ci: init

Signed-off-by: Amneesh Singh <natto@weirdnatto.in>

src/lib/messages.rs

@@ -0,0 +1,53 @@
+use linkify::LinkFinder;
+use serenity::model::channel::Message;
+
+#[derive(Default)]
+pub struct URLExtractInfo {
+    urls: Vec<String>,
+    n_attachments: u8,
+    n_links: u16,
+    rn_attachments: Option<u8>,
+    rn_links: Option<u16>,
+}
+
+pub trait ExtractInfo {
+    fn extract_urls(&self) -> URLExtractInfo;
+}
+
+// Priority: Text > Attachments > Reference
+impl ExtractInfo for Message {
+    fn extract_urls(&self) -> URLExtractInfo {
+        let mut ret = URLExtractInfo::default();
+        let finder = LinkFinder::new();
+        let find_links = |x| finder.links(x).map(|x| x.as_str().to_string()).collect();
+
+        ret.urls = find_links(&self.content);
+
+        ret.n_links = ret.urls.len() as u16;
+
+        ret.urls.extend(
+            self.attachments
+                .iter()
+                .map(|x| x.url.clone())
+                .collect::<Vec<String>>(),
+        );
+
+        ret.n_attachments = self.attachments.len() as u8;
+
+        if let Some(msg) = &self.referenced_message {
+            let msg_links: Vec<String> = find_links(&msg.content);
+
+            ret.rn_attachments = Some(msg.attachments.len() as u8);
+            ret.rn_links = Some(msg_links.len() as u16);
+            ret.urls.extend(msg_links);
+            ret.urls.extend(
+                msg.attachments
+                    .iter()
+                    .map(|x| x.url.clone())
+                    .collect::<Vec<String>>(),
+            );
+        }
+
+        ret
+    }
+}

src/main.rs

@@ -27,7 +27,7 @@ impl TypeMapKey for ShardManagerContainer {
 }
 
 #[group]
-#[commands(remux,speed)]
+#[commands(remux, speed)]
 struct Transcode;
 
 #[group]
@@ -68,7 +68,7 @@ async fn main() {
     };
 
     let framework = StandardFramework::new()
-        .configure(|c| c.owners(owners).prefix("xx"))
+        .configure(|c| c.owners(owners).prefix("xx").delimiters(vec![" ", "\n"]))
         .help(&MY_HELP)
         .group(&GENERAL_GROUP)
         .group(&TRANSCODE_GROUP);