use axum::{ extract::{Json, Query, State}, http::StatusCode, response::IntoResponse, }; use serde_json::json; use tracing::{error, info}; use crate::application::dtos::search_dto::SearchCriteriaDto; use crate::common::di::AppState; /** * Handler for search operations through the API. * * All search processing (filtering, scoring, sorting, categorization, * formatting) is performed server-side. These handlers are thin HTTP * adapters that delegate to the SearchUseCase. */ pub struct SearchHandler; impl SearchHandler { /// GET /search — simple query-parameter-based search. pub async fn search_files_get( State(state): State, Query(params): Query, ) -> impl IntoResponse { info!("API: File search with parameters: {:?}", params); let search_service = match &state.applications.search_service { Some(service) => service, None => { error!("Search service not available"); return ( StatusCode::SERVICE_UNAVAILABLE, Json(json!({ "error": "Search service is not available" })), ) .into_response(); } }; let search_criteria = SearchCriteriaDto { name_contains: params.query, file_types: params .type_filter .map(|t| t.split(',').map(|s| s.trim().to_string()).collect()), created_after: params.created_after, created_before: params.created_before, modified_after: params.modified_after, modified_before: params.modified_before, min_size: params.min_size, max_size: params.max_size, folder_id: params.folder_id, recursive: params.recursive.unwrap_or(true), limit: params.limit.unwrap_or(100), offset: params.offset.unwrap_or(0), sort_by: params.sort_by.unwrap_or_else(|| "relevance".to_string()), }; match search_service.search(search_criteria).await { Ok(results) => { info!( "Search completed in {}ms — {} files, {} folders", results.query_time_ms, results.files.len(), results.folders.len() ); (StatusCode::OK, Json(results)).into_response() } Err(err) => { error!("Search error: {}", err); ( StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": format!("Search error: {}", err) })), ) .into_response() } } } /// POST /search/advanced — full criteria in the request body. pub async fn search_files_post( State(state): State, Json(criteria): Json, ) -> impl IntoResponse { info!("API: Advanced file search"); let search_service = match &state.applications.search_service { Some(service) => service, None => { error!("Search service not available"); return ( StatusCode::SERVICE_UNAVAILABLE, Json(json!({ "error": "Search service is not available" })), ) .into_response(); } }; match search_service.search(criteria).await { Ok(results) => { info!( "Advanced search completed in {}ms — {} files, {} folders", results.query_time_ms, results.files.len(), results.folders.len() ); (StatusCode::OK, Json(results)).into_response() } Err(err) => { error!("Search error: {}", err); ( StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": format!("Search error: {}", err) })), ) .into_response() } } } /// GET /search/suggest — lightweight autocomplete suggestions. pub async fn suggest_files( State(state): State, Query(params): Query, ) -> impl IntoResponse { info!("API: Search suggestions for {:?}", params.query); let search_service = match &state.applications.search_service { Some(service) => service, None => { error!("Search service not available"); return ( StatusCode::SERVICE_UNAVAILABLE, Json(json!({ "error": "Search service is not available" })), ) .into_response(); } }; let limit = params.limit.unwrap_or(10).min(20); match search_service .suggest(¶ms.query, params.folder_id.as_deref(), limit) .await { Ok(suggestions) => { info!( "Suggestions completed in {}ms — {} results", suggestions.query_time_ms, suggestions.suggestions.len() ); (StatusCode::OK, Json(suggestions)).into_response() } Err(err) => { error!("Suggestions error: {}", err); ( StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": format!("Suggestions error: {}", err) })), ) .into_response() } } } /// DELETE /search/cache — clears the search results cache. pub async fn clear_search_cache(State(state): State) -> impl IntoResponse { info!("API: Clearing search cache"); let search_service = match &state.applications.search_service { Some(service) => service, None => { error!("Search service not available"); return ( StatusCode::SERVICE_UNAVAILABLE, Json(json!({ "error": "Search service is not available" })), ) .into_response(); } }; match search_service.clear_search_cache().await { Ok(_) => { info!("Search cache cleared successfully"); ( StatusCode::OK, Json(json!({ "message": "Search cache cleared successfully" })), ) .into_response() } Err(err) => { error!("Error clearing search cache: {}", err); ( StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": format!("Error clearing search cache: {}", err) })), ) .into_response() } } } } /// Search parameters for the GET /search endpoint #[derive(Debug, serde::Deserialize)] pub struct SearchParams { /// Text to search in file and folder names pub query: Option, /// Filter by file types (comma-separated extensions) #[serde(rename = "type")] pub type_filter: Option, /// Created after this timestamp pub created_after: Option, /// Created before this timestamp pub created_before: Option, /// Modified after this timestamp pub modified_after: Option, /// Modified before this timestamp pub modified_before: Option, /// Minimum file size in bytes pub min_size: Option, /// Maximum file size in bytes pub max_size: Option, /// Folder ID to limit the search scope pub folder_id: Option, /// Recursive search in subfolders (default: true) pub recursive: Option, /// Result limit for pagination pub limit: Option, /// Offset for pagination pub offset: Option, /// Sort order: relevance | name | name_desc | date | date_desc | size | size_desc pub sort_by: Option, } /// Parameters for the GET /search/suggest endpoint #[derive(Debug, serde::Deserialize)] pub struct SuggestParams { /// Text to search for suggestions pub query: String, /// Folder ID to limit the suggestion scope pub folder_id: Option, /// Maximum number of suggestions (default 10, max 20) pub limit: Option, }