scope/web/api/
discover.rs

1//! Token discovery API handler.
2
3use crate::chains::DexClient;
4use crate::web::AppState;
5use axum::Json;
6use axum::extract::{Query, State};
7use axum::http::StatusCode;
8use axum::response::IntoResponse;
9use serde::Deserialize;
10use std::sync::Arc;
11
12/// Query parameters for token discovery.
13#[derive(Debug, Deserialize)]
14pub struct DiscoverQuery {
15    /// Source: "profiles", "boosts", "top-boosts".
16    #[serde(default = "default_source")]
17    pub source: String,
18    /// Filter by chain (optional).
19    pub chain: Option<String>,
20    /// Max results (default: 15).
21    #[serde(default = "default_limit")]
22    pub limit: u32,
23}
24
25fn default_source() -> String {
26    "profiles".to_string()
27}
28
29fn default_limit() -> u32 {
30    15
31}
32
33/// GET /api/discover — Browse trending/boosted tokens.
34pub async fn handle(
35    State(_state): State<Arc<AppState>>,
36    Query(params): Query<DiscoverQuery>,
37) -> impl IntoResponse {
38    let client = DexClient::new();
39
40    let tokens = match params.source.as_str() {
41        "boosts" => client.get_token_boosts().await,
42        "top-boosts" => client.get_token_boosts_top().await,
43        _ => client.get_token_profiles().await,
44    };
45
46    match tokens {
47        Ok(tokens) => {
48            let filtered: Vec<_> = if let Some(ref chain) = params.chain {
49                let c = chain.to_lowercase();
50                tokens
51                    .into_iter()
52                    .filter(|t| t.chain_id.to_lowercase() == c)
53                    .take(params.limit as usize)
54                    .collect()
55            } else {
56                tokens.into_iter().take(params.limit as usize).collect()
57            };
58            Json(serde_json::json!(filtered)).into_response()
59        }
60        Err(e) => (
61            StatusCode::INTERNAL_SERVER_ERROR,
62            Json(serde_json::json!({ "error": e.to_string() })),
63        )
64            .into_response(),
65    }
66}
67
68#[cfg(test)]
69mod tests {
70    use super::*;
71
72    #[test]
73    fn test_deserialize_full() {
74        let json = serde_json::json!({
75            "source": "boosts",
76            "chain": "ethereum",
77            "limit": 25
78        });
79        let req: DiscoverQuery = serde_json::from_value(json).unwrap();
80        assert_eq!(req.source, "boosts");
81        assert_eq!(req.chain, Some("ethereum".to_string()));
82        assert_eq!(req.limit, 25);
83    }
84
85    #[test]
86    fn test_deserialize_minimal() {
87        let json = serde_json::json!({});
88        let req: DiscoverQuery = serde_json::from_value(json).unwrap();
89        assert_eq!(req.source, "profiles");
90        assert_eq!(req.chain, None);
91        assert_eq!(req.limit, 15);
92    }
93
94    #[test]
95    fn test_defaults() {
96        assert_eq!(default_source(), "profiles");
97        assert_eq!(default_limit(), 15);
98    }
99
100    #[test]
101    fn test_with_chain_filter() {
102        let json = serde_json::json!({
103            "chain": "polygon",
104            "limit": 10
105        });
106        let req: DiscoverQuery = serde_json::from_value(json).unwrap();
107        assert_eq!(req.source, "profiles");
108        assert_eq!(req.chain, Some("polygon".to_string()));
109        assert_eq!(req.limit, 10);
110    }
111}
scope/web/api/discover.rs

scope/web/api/
discover.rs
