add visits counter to homeHEADmain

3 files changed, 25 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index 7247c33..bbfcbff 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -12,6 +12,7 @@ mod scrobbles;
 mod skweets;
 mod templates;
 mod utils;
+mod visits;
 
 use std::{collections::HashSet, time::Duration};
 
@@ -54,17 +55,23 @@ struct Static;
 #[handler]
 async fn home(Data(reqwest): Data<&reqwest::Client>, locale: Locale) -> templates::Home {
     let listenbrainz_client = listenbrainz::raw::Client::new();
-    let (live, listenbrainz, blogposts, poems) = tokio::join!(
+    let (live, listenbrainz, blogposts, poems, visits) = tokio::join!(
         live::get_live_status(reqwest),
         scrobbles::get_now_playing(&listenbrainz_client),
         // skweets::get_recents(&clients.skinnyverse),
         Blogpost::get_articles(),
-        Poem::get_articles()
+        Poem::get_articles(),
+        visits::get_visits(reqwest)
     );
     let is_live = live.unwrap_or_default().online;
     let listenbrainz = listenbrainz.unwrap_or_default();
     let blogposts = blogposts.unwrap_or_default();
     let poems = poems.unwrap_or_default();
+    // let visits = visits.ok().map(|v| v.count);
+    let visits = visits
+        .ok()
+        .map(|v| v.count.replace("\u{202f}", ""))
+        .and_then(|c| c.parse().ok());
 
     let mut posts = blogposts
         .iter()
@@ -96,6 +103,7 @@ async fn home(Data(reqwest): Data<&reqwest::Client>, locale: Locale) -> template
         locale,
         filter_tags: HashSet::new(),
         jiggle: 8,
+        visits,
     }
 }
 
diff --git a/src/templates.rs b/src/templates.rs
index c086487..c0b3c14 100644
--- a/src/templates.rs
+++ b/src/templates.rs
@@ -29,6 +29,7 @@ pub struct Home {
     pub locale: Locale,
     pub filter_tags: HashSet<String>,
     pub jiggle: isize,
+    pub visits: Option<usize>,
 }
 
 #[derive(Template)]
diff --git a/src/visits.rs b/src/visits.rs
new file mode 100644
index 0000000..1e0685c
--- /dev/null
+++ b/src/visits.rs
@@ -0,0 +1,14 @@
+use reqwest::Client;
+use serde::Deserialize;
+
+use crate::Result;
+
+#[derive(Deserialize, Default)]
+pub struct Visits {
+    pub count: String,
+}
+
+pub async fn get_visits(client: &Client) -> Result<Visits> {
+    let endpoint = "https://stats.blos.sm/counter/TOTAL.json";
+    Ok(client.get(endpoint).send().await?.json::<Visits>().await?)
+}