Track new analytics metrics in backend (#5895)

* Allow filtering by project IDs in analytics route

* Download meta info in header

* add recursion limit

* Track playtime country

* fix clickhouse migrations

apps/labrinth/src/routes/internal/admin.rs

@@ -1,7 +1,7 @@
 use crate::auth::validate::get_user_record_from_bearer_token;
 use crate::database::PgPool;
 use crate::database::redis::RedisPool;
-use crate::models::analytics::Download;
+use crate::models::analytics::{Download, DownloadReason};
 use crate::models::ids::ProjectId;
 use crate::models::pats::Scopes;
 use crate::queue::analytics::AnalyticsQueue;
@@ -35,6 +35,17 @@ pub struct DownloadBody {
     pub headers: HashMap<String, String>,
 }
 
+/// Extra data attached to each download request, transmitted through the
+/// [`DOWNLOAD_META_HEADER`] header.
+#[derive(Debug, Clone, Deserialize)]
+pub struct DownloadMeta {
+    pub reason: DownloadReason,
+    pub game_version: String,
+    pub loader: String,
+}
+
+pub const DOWNLOAD_META_HEADER: &str = "modrinth-download-meta";
+
 // This is an internal route, cannot be used without key
 #[utoipa::path(
     patch,
@@ -118,6 +129,11 @@ pub async fn count_download(
     let ip = crate::util::ip::convert_to_ip_v6(&download_body.ip)
         .unwrap_or_else(|_| Ipv4Addr::new(127, 0, 0, 1).to_ipv6_mapped());
 
+    let meta = download_body
+        .headers
+        .get(DOWNLOAD_META_HEADER)
+        .and_then(|v| serde_json::from_str::<DownloadMeta>(v).ok());
+
     analytics_queue.add_download(Download {
         recorded: get_current_tenths_of_ms(),
         domain: url.host_str().unwrap_or_default().to_string(),
@@ -153,6 +169,9 @@ pub async fn count_download(
                     .contains(&&*x.0.to_lowercase())
             })
             .collect(),
+        reason: meta.as_ref().map(|m| m.reason),
+        game_version: meta.as_ref().map(|m| m.game_version.clone()),
+        loader: meta.as_ref().map(|m| m.loader.clone()),
     });
 
     Ok(HttpResponse::NoContent().body(""))

apps/labrinth/src/routes/internal/mod.rs

@@ -1,4 +1,4 @@
-pub(crate) mod admin;
+pub mod admin;
 pub mod affiliate;
 pub mod billing;
 pub mod delphi;