brk_server 0.3.0-beta.9

use std::str::FromStr;

use aide::{
    OperationInput,
    operation::{ParamLocation, add_parameters, parameters_from_schema},
};
use axum::{extract::FromRequestParts, http::request::Parts};
use schemars::JsonSchema;

use brk_types::Txid;

use crate::Error;

const MAX_TXIDS: usize = 250;

/// Query parameter for transaction-times endpoint.
///
/// Extracted manually because `serde_urlencoded` (and serde derive in general)
/// doesn't support repeated keys like `txId[]=a&txId[]=b`. The schema is still
/// declared via `JsonSchema` so the OpenAPI spec lists the parameter and the
/// generated client SDKs see `txids: List[Txid]`.
#[derive(JsonSchema)]
pub struct TxidsParam {
    /// Transaction IDs to look up (max 250 per request).
    #[serde(rename = "txId[]")]
    pub txids: Vec<Txid>,
}

impl TxidsParam {
    /// Parsed manually from URI since serde_urlencoded doesn't support repeated keys.
    /// Rejects unknown keys to prevent cache-busting via injected query params.
    pub fn from_query(query: &str) -> Result<Self, String> {
        if query.is_empty() {
            return Err("missing required query parameter `txId[]`".into());
        }
        let mut txids = Vec::new();
        for pair in query.split('&') {
            let (key, val) = pair.split_once('=').ok_or_else(|| {
                format!("malformed query parameter `{pair}`, expected `txId[]=<txid>`")
            })?;
            if key == "txId[]" || key == "txId%5B%5D" {
                if txids.len() == MAX_TXIDS {
                    return Err(format!("too many txids, max {MAX_TXIDS} per request"));
                }
                let txid = Txid::from_str(val).map_err(|e| format!("invalid txid `{val}`: {e}"))?;
                txids.push(txid);
            } else {
                return Err(format!(
                    "unknown query parameter `{key}`, expected `txId[]`"
                ));
            }
        }
        Ok(Self { txids })
    }
}

impl<S> FromRequestParts<S> for TxidsParam
where
    S: Send + Sync,
{
    type Rejection = Error;

    async fn from_request_parts(parts: &mut Parts, _state: &S) -> Result<Self, Self::Rejection> {
        Self::from_query(parts.uri.query().unwrap_or("")).map_err(Error::bad_request)
    }
}

impl OperationInput for TxidsParam {
    fn operation_input(
        ctx: &mut aide::generate::GenContext,
        operation: &mut aide::openapi::Operation,
    ) {
        let schema = ctx.schema.subschema_for::<Self>();
        let params = parameters_from_schema(ctx, schema, ParamLocation::Query);
        add_parameters(ctx, operation, params);
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    const T1: &str = "0000000000000000000000000000000000000000000000000000000000000001";
    const T2: &str = "0000000000000000000000000000000000000000000000000000000000000002";

    #[test]
    fn parses_single_and_multi() {
        assert_eq!(
            TxidsParam::from_query(&format!("txId[]={T1}"))
                .unwrap()
                .txids
                .len(),
            1
        );
        assert_eq!(
            TxidsParam::from_query(&format!("txId%5B%5D={T1}&txId[]={T2}"))
                .unwrap()
                .txids
                .len(),
            2,
        );
    }

    #[test]
    fn rejects_empty_unknown_key_and_invalid_txid() {
        assert!(TxidsParam::from_query("").is_err());
        assert!(TxidsParam::from_query("foo=bar").is_err());
        assert!(TxidsParam::from_query("txId[]=notahex").is_err());
        assert!(TxidsParam::from_query("noequals").is_err());
    }
}