linkletter 0.1.0

URL replacement and analytics service
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use axum::Json;
use axum::extract::{State, Path};
use axum::http::StatusCode;

use crate::service;
use crate::database;

pub(crate) type VisitsResponse = Vec<VisitsResult>;

#[derive(serde::Serialize)]
#[serde(rename_all = "camelCase")]
pub(crate) struct VisitsResult {
    #[serde(with = "time::serde::rfc3339")]
    timestamp: time::OffsetDateTime,
    ip_addr: Option<std::net::SocketAddr>,
    user_agent: Option<String>,
}

impl From<&database::visit::Visit> for VisitsResult {
    fn from(visit: &database::visit::Visit) -> Self {
        Self {
            timestamp: visit.visit_timestamp.assume_utc(),
            ip_addr: visit.visitor_ip_addr,
            user_agent: visit.visitor_user_agent.clone(),
        }
    }
}

/// Return a list of information for each intercepted visit of URL contained.
/// Information includes time of visit, connecting IP and value of `User-Agent` header.
pub(crate) async fn visits(
    State(state): State<service::State>,
    Path(url_id_simple): Path<uuid::fmt::Simple>,
) -> Result<Json<VisitsResponse>, StatusCode> {
    let visits = database::visit::get_url_visits(&url_id_simple.as_uuid(), &state.pool)
        .await
        .map_err(|err| service::server_error(err, "Could not get URL visits"))?;

    Ok(Json(visits.iter().map(VisitsResult::from).collect()))
}