feat(january): fetch embeds for multiple links

src/database/entities/microservice/january.rs

@@ -124,16 +124,28 @@ impl Embed {
             .collect::<Vec<&str>>()
             .join("\n");
 
-        // ! FIXME: allow multiple links
         let mut finder = LinkFinder::new();
         finder.kinds(&[LinkKind::Url]);
-        let links: Vec<_> = finder.links(&content).collect();
+        let links: Vec<_> = finder.links(&content).take(5).collect();
 
         if links.len() == 0 {
             return Err(Error::LabelMe);
         }
 
-        let link = &links[0];
+        let mut embeds: Vec<Embed> = Vec::new();
+
+        // ! FIXME: allow configuration of number of embeds
+        // ! FIXME: batch request to january?
+        let mut link_index = 0;
+
+        while link_index < links.len() {
+            let link = &links[link_index];
+
+            // Check if we did the same link already in for this message.
+            if link_index != 0 && links.iter().take(link_index).any(|x| x.as_str() == link.as_str()) {
+                link_index = link_index + 1;
+                continue;
+            }
 
             let client = reqwest::Client::new();
             let result = client
@@ -142,16 +154,26 @@ impl Embed {
                 .send()
                 .await;
 
-        match result {
+            if result.is_err() {
-            Err(_) => return Err(Error::LabelMe),
+                link_index = link_index + 1;
-            Ok(result) => match result.status() {
+                continue;
-                reqwest::StatusCode::OK => {
-                    let res: Embed = result.json().await.map_err(|_| Error::InvalidOperation)?;
-
-                    Ok(vec![res])
             }
-                _ => return Err(Error::LabelMe),
+
-            },
+            let response = result.unwrap();
+            if response.status().is_success() {
+                let res: Embed = response.json().await.map_err(|_| Error::InvalidOperation)?;
+
+                embeds.push(res);
+            }
+
+            link_index = link_index + 1;
+        }
+
+        // Prevent database update when no embeds are found
+        if embeds.len() > 0 {
+            Ok(embeds)
+        } else {
+            Err(Error::LabelMe)
         }
     }
 }