scope-bca 0.5.6

//! Token discovery API handler.

use crate::AppState;
use axum::Json;
use axum::extract::{Query, State};
use axum::http::StatusCode;
use axum::response::IntoResponse;
use scope::chains::DexClient;
use serde::Deserialize;
use std::sync::Arc;

/// Query parameters for token discovery.
#[derive(Debug, Deserialize)]
pub struct DiscoverQuery {
    /// Source: "profiles", "boosts", "top-boosts".
    #[serde(default = "default_source")]
    pub source: String,
    /// Filter by chain (optional).
    pub chain: Option<String>,
    /// Max results (default: 15).
    #[serde(default = "default_limit")]
    pub limit: u32,
}

fn default_source() -> String {
    "profiles".to_string()
}

fn default_limit() -> u32 {
    15
}

/// GET /api/discover — Browse trending/boosted tokens.
pub async fn handle(
    State(_state): State<Arc<AppState>>,
    Query(params): Query<DiscoverQuery>,
) -> impl IntoResponse {
    let client = DexClient::new();

    let tokens = match params.source.as_str() {
        "boosts" => client.get_token_boosts().await,
        "top-boosts" => client.get_token_boosts_top().await,
        _ => client.get_token_profiles().await,
    };

    match tokens {
        Ok(tokens) => {
            let filtered: Vec<_> = if let Some(ref chain) = params.chain {
                let c = chain.to_lowercase();
                tokens
                    .into_iter()
                    .filter(|t| t.chain_id.to_lowercase() == c)
                    .take(params.limit as usize)
                    .collect()
            } else {
                tokens.into_iter().take(params.limit as usize).collect()
            };
            Json(serde_json::json!(filtered)).into_response()
        }
        Err(e) => (
            StatusCode::INTERNAL_SERVER_ERROR,
            Json(serde_json::json!({ "error": e.to_string() })),
        )
            .into_response(),
    }
}

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

    #[test]
    fn test_deserialize_full() {
        let json = serde_json::json!({
            "source": "boosts",
            "chain": "ethereum",
            "limit": 25
        });
        let req: DiscoverQuery = serde_json::from_value(json).unwrap();
        assert_eq!(req.source, "boosts");
        assert_eq!(req.chain, Some("ethereum".to_string()));
        assert_eq!(req.limit, 25);
    }

    #[test]
    fn test_deserialize_minimal() {
        let json = serde_json::json!({});
        let req: DiscoverQuery = serde_json::from_value(json).unwrap();
        assert_eq!(req.source, "profiles");
        assert_eq!(req.chain, None);
        assert_eq!(req.limit, 15);
    }

    #[test]
    fn test_defaults() {
        assert_eq!(default_source(), "profiles");
        assert_eq!(default_limit(), 15);
    }

    #[test]
    fn test_with_chain_filter() {
        let json = serde_json::json!({
            "chain": "polygon",
            "limit": 10
        });
        let req: DiscoverQuery = serde_json::from_value(json).unwrap();
        assert_eq!(req.source, "profiles");
        assert_eq!(req.chain, Some("polygon".to_string()));
        assert_eq!(req.limit, 10);
    }

    #[tokio::test]
    async fn test_handle_discover_profiles() {
        use crate::AppState;
        use axum::extract::{Query, State};
        use axum::response::IntoResponse;
        use scope::chains::DefaultClientFactory;
        use scope::config::Config;

        let config = Config::default();
        let http: std::sync::Arc<dyn scope::http::HttpClient> =
            std::sync::Arc::new(scope::http::NativeHttpClient::new().unwrap());
        let factory = DefaultClientFactory {
            chains_config: config.chains.clone(),
            http,
        };
        let state = std::sync::Arc::new(AppState { config, factory });

        let params = DiscoverQuery {
            source: "profiles".to_string(),
            chain: None,
            limit: 5,
        };
        let response = handle(State(state), Query(params)).await.into_response();
        let status = response.status();
        assert!(status.is_success() || status.is_server_error());
    }

    #[tokio::test]
    async fn test_handle_discover_boosts() {
        use crate::AppState;
        use axum::extract::{Query, State};
        use axum::response::IntoResponse;
        use scope::chains::DefaultClientFactory;
        use scope::config::Config;

        let config = Config::default();
        let http: std::sync::Arc<dyn scope::http::HttpClient> =
            std::sync::Arc::new(scope::http::NativeHttpClient::new().unwrap());
        let factory = DefaultClientFactory {
            chains_config: config.chains.clone(),
            http,
        };
        let state = std::sync::Arc::new(AppState { config, factory });

        let params = DiscoverQuery {
            source: "boosts".to_string(),
            chain: None,
            limit: 5,
        };
        let response = handle(State(state), Query(params)).await.into_response();
        let status = response.status();
        assert!(status.is_success() || status.is_server_error());
    }

    #[tokio::test]
    async fn test_handle_discover_top_boosts() {
        use crate::AppState;
        use axum::extract::{Query, State};
        use axum::response::IntoResponse;
        use scope::chains::DefaultClientFactory;
        use scope::config::Config;

        let config = Config::default();
        let http: std::sync::Arc<dyn scope::http::HttpClient> =
            std::sync::Arc::new(scope::http::NativeHttpClient::new().unwrap());
        let factory = DefaultClientFactory {
            chains_config: config.chains.clone(),
            http,
        };
        let state = std::sync::Arc::new(AppState { config, factory });

        let params = DiscoverQuery {
            source: "top-boosts".to_string(),
            chain: None,
            limit: 5,
        };
        let response = handle(State(state), Query(params)).await.into_response();
        let status = response.status();
        assert!(status.is_success() || status.is_server_error());
    }

    #[tokio::test]
    async fn test_handle_discover_with_chain_filter() {
        use crate::AppState;
        use axum::extract::{Query, State};
        use axum::response::IntoResponse;
        use scope::chains::DefaultClientFactory;
        use scope::config::Config;

        let config = Config::default();
        let http: std::sync::Arc<dyn scope::http::HttpClient> =
            std::sync::Arc::new(scope::http::NativeHttpClient::new().unwrap());
        let factory = DefaultClientFactory {
            chains_config: config.chains.clone(),
            http,
        };
        let state = std::sync::Arc::new(AppState { config, factory });

        let params = DiscoverQuery {
            source: "profiles".to_string(),
            chain: Some("ethereum".to_string()),
            limit: 10,
        };
        let response = handle(State(state), Query(params)).await.into_response();
        let status = response.status();
        assert!(status.is_success() || status.is_server_error());
    }

    #[tokio::test]
    async fn test_handle_discover_error_path() {
        use crate::AppState;
        use axum::extract::{Query, State};
        use axum::http::StatusCode;
        use axum::response::IntoResponse;
        use scope::chains::DefaultClientFactory;
        use scope::config::Config;

        let config = Config::default();
        let http: std::sync::Arc<dyn scope::http::HttpClient> =
            std::sync::Arc::new(scope::http::NativeHttpClient::new().unwrap());
        let factory = DefaultClientFactory {
            chains_config: config.chains.clone(),
            http,
        };
        let state = std::sync::Arc::new(AppState { config, factory });

        let params = DiscoverQuery {
            source: "unknown-source".to_string(),
            chain: None,
            limit: 5,
        };
        let response = handle(State(state), Query(params)).await.into_response();
        let status = response.status();
        if status == StatusCode::INTERNAL_SERVER_ERROR {
            let body = axum::body::to_bytes(response.into_body(), 1_000_000)
                .await
                .unwrap();
            let json: serde_json::Value = serde_json::from_slice(&body).unwrap();
            assert!(json.get("error").is_some());
        }
    }
}