cryptgeon/packages/backend/src/note/routes.rs

use axum::{
    extract::Path,
    http::StatusCode,
    response::{IntoResponse, Response},
    body::Bytes,
};
use serde::Deserialize;
use std::time::SystemTime;

use crate::note::{CreateRequest, generate_id};
use crate::store;
use crate::config;

use super::{CreateResponse, MetaResponse, NoteResponse, NoteMeta};

pub fn now() -> u64 {
    SystemTime::now()
        .duration_since(SystemTime::UNIX_EPOCH)
        .unwrap()
        .as_secs()
}

#[derive(Deserialize)]
pub struct NoteParams {
    id: String,
}

pub async fn create(body: Bytes) -> Response {
    let req: CreateRequest = match rmp_serde::from_slice(&body) {
        Ok(r) => r,
        Err(_) => return (StatusCode::BAD_REQUEST, "Invalid msgpack").into_response(),
    };

    if req.meta.views.is_none() && req.meta.expiration.is_none() {
        return (StatusCode::BAD_REQUEST, "At least views or expiration must be set").into_response();
    }

    if req.meta.extra.len() > *config::EXTRA_SIZE_LIMIT {
        return (StatusCode::BAD_REQUEST, "Extra data too large").into_response();
    }

    let mut meta = req.meta;

    if !*config::ALLOW_ADVANCED {
        meta.views = Some(1);
        meta.expiration = None;
    }

    match meta.views {
        Some(v) => {
            if v > *config::MAX_VIEWS || v < 1 {
                return (StatusCode::BAD_REQUEST, "Invalid views").into_response();
            }
        }
        None => {}
    }

    let expiration_ts = match meta.expiration {
        Some(e) => {
            if e > *config::MAX_EXPIRATION || e < 1 {
                return (StatusCode::BAD_REQUEST, "Invalid expiration").into_response();
            }
            Some(now() + (e as u64 * 60))
        }
        None => None,
    };

    let id = generate_id();
    let views = meta.views.map(|v| v as i64);

    match store::set(&id, &req.data, views, expiration_ts, &meta.extra) {
        Ok(_) => {
            let resp = CreateResponse { id };
            let bytes = rmp_serde::to_vec_named(&resp).unwrap();
            (StatusCode::OK, Bytes::from(bytes)).into_response()
        }
        Err(e) => (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()).into_response(),
    }
}

pub async fn preview(Path(NoteParams { id }): Path<NoteParams>) -> Response {
    match store::get_meta(&id) {
        Ok(Some((views, expiration, extra))) => {
            let meta = NoteMeta {
                views: views.map(|v| v as u32),
                expiration: expiration.map(|e| e as u32),
                extra,
            };
            let resp = MetaResponse { meta };
            let bytes = rmp_serde::to_vec_named(&resp).unwrap();
            (StatusCode::OK, Bytes::from(bytes)).into_response()
        }
        Ok(None) => (StatusCode::NOT_FOUND).into_response(),
        Err(e) => (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()).into_response(),
    }
}

pub async fn view(Path(NoteParams { id }): Path<NoteParams>) -> Response {
    let (views, expiration, extra) = match store::get_meta(&id) {
        Ok(Some(v)) => v,
        _ => return (StatusCode::NOT_FOUND).into_response(),
    };

    let has_views = views.is_some();

    if has_views {
        let remaining = match store::decrement_views(&id) {
            Ok(r) => r,
            Err(_) => return (StatusCode::NOT_FOUND).into_response(),
        };

        let data = match store::get_data(&id) {
            Ok(Some(d)) => d,
            _ => return (StatusCode::NOT_FOUND).into_response(),
        };

        if remaining <= 0 {
            let _ = store::del(&id);
        }

        let meta = NoteMeta {
            views: Some(if remaining > 0 { remaining as u32 } else { 0 }),
            expiration: expiration.map(|e| e as u32),
            extra,
        };
        let resp = NoteResponse { meta, data };
        let bytes = rmp_serde::to_vec_named(&resp).unwrap();
        (StatusCode::OK, Bytes::from(bytes)).into_response()
    } else {
        let data = match store::get_data(&id) {
            Ok(Some(d)) => d,
            _ => return (StatusCode::NOT_FOUND).into_response(),
        };

        let meta = NoteMeta {
            views: None,
            expiration: expiration.map(|e| e as u32),
            extra,
        };
        let resp = NoteResponse { meta, data };
        let bytes = rmp_serde::to_vec_named(&resp).unwrap();
        (StatusCode::OK, Bytes::from(bytes)).into_response()
    }
}