feat: plausible analytics

src/analytics/plausible.rs

@@ -0,0 +1,55 @@
+use crate::analytics::Analytics;
+use crate::settings::Settings;
+use anyhow::Error;
+use log::{info, warn};
+use rocket::Request;
+use serde::{Deserialize, Serialize};
+use tokio::sync::mpsc::{unbounded_channel, UnboundedSender};
+
+#[derive(Debug, Serialize, Deserialize)]
+struct Event {
+    pub name: String,
+    pub domain: String,
+    pub url: String,
+    pub referrer: Option<String>,
+}
+
+pub struct PlausibleAnalytics {
+    tx: UnboundedSender<Event>,
+}
+
+impl PlausibleAnalytics {
+    pub fn new(settings: &Settings) -> Self {
+        let (tx, mut rx) = unbounded_channel::<Event>();
+        let url = match &settings.plausible_url {
+            Some(s) => s.clone(),
+            _ => "".to_string(),
+        };
+        let pub_url = settings.public_url.clone();
+        tokio::spawn(async move {
+            while let Some(mut msg) = rx.recv().await {
+                msg.url = format!("{}{}", pub_url, msg.url);
+                match ureq::post(&format!("{}/api/event", url)).send_json(&msg) {
+                    Ok(v) => info!("Sent {:?}", msg),
+                    Err(e) => warn!("Failed to track: {}", e),
+                }
+            }
+        });
+
+        Self { tx }
+    }
+}
+
+impl Analytics for PlausibleAnalytics {
+    fn track(&self, req: &Request) -> Result<(), Error> {
+        Ok(self.tx.send(Event {
+            name: "pageview".to_string(),
+            domain: match req.host() {
+                Some(s) => s.to_string(),
+                None => return Ok(()), // ignore request
+            },
+            url: req.uri().to_string(),
+            referrer: None,
+        })?)
+    }
+}