use actix_web::{web, HttpResponse}; use log::debug; use crate::extractors::authentication::{policies::*, GuardedData}; use crate::index::Settings; use crate::Data; use crate::{error::ResponseError, index::Unchecked}; #[macro_export] macro_rules! make_setting_route { ($route:literal, $type:ty, $attr:ident, $camelcase_attr:literal) => { pub mod $attr { use log::debug; use actix_web::{web, HttpResponse, Resource}; use milli::update::Setting; use crate::data; use crate::error::ResponseError; use crate::index::Settings; use crate::extractors::authentication::{GuardedData, policies::*}; pub async fn delete( data: GuardedData, index_uid: web::Path, ) -> Result { use crate::index::Settings; let settings = Settings { $attr: Setting::Reset, ..Default::default() }; let update_status = data.update_settings(index_uid.into_inner(), settings, false).await?; debug!("returns: {:?}", update_status); Ok(HttpResponse::Accepted().json(serde_json::json!({ "updateId": update_status.id() }))) } pub async fn update( data: GuardedData, index_uid: actix_web::web::Path, body: actix_web::web::Json>, ) -> std::result::Result { let settings = Settings { $attr: match body.into_inner() { Some(inner_body) => Setting::Set(inner_body), None => Setting::Reset }, ..Default::default() }; let update_status = data.update_settings(index_uid.into_inner(), settings, true).await?; debug!("returns: {:?}", update_status); Ok(HttpResponse::Accepted().json(serde_json::json!({ "updateId": update_status.id() }))) } pub async fn get( data: GuardedData, index_uid: actix_web::web::Path, ) -> std::result::Result { let settings = data.settings(index_uid.into_inner()).await?; debug!("returns: {:?}", settings); let mut json = serde_json::json!(&settings); let val = json[$camelcase_attr].take(); Ok(HttpResponse::Ok().json(val)) } pub fn resources() -> Resource { Resource::new($route) .route(web::get().to(get)) .route(web::post().to(update)) .route(web::delete().to(delete)) } } }; } make_setting_route!( "/filterable-attributes", std::collections::HashSet, filterable_attributes, "filterableAttributes" ); make_setting_route!( "/displayed-attributes", Vec, displayed_attributes, "displayedAttributes" ); make_setting_route!( "/searchable-attributes", Vec, searchable_attributes, "searchableAttributes" ); make_setting_route!( "/stop-words", std::collections::BTreeSet, stop_words, "stopWords" ); make_setting_route!( "/synonyms", std::collections::BTreeMap>, synonyms, "synonyms" ); make_setting_route!( "/distinct-attribute", String, distinct_attribute, "distinctAttribute" ); make_setting_route!("/ranking-rules", Vec, ranking_rules, "rankingRules"); macro_rules! generate_configure { ($($mod:ident),*) => { pub fn configure(cfg: &mut web::ServiceConfig) { cfg.service( web::resource("") .route(web::post().to(update_all)) .route(web::get().to(get_all)) .route(web::delete().to(delete_all))) $(.service($mod::resources()))*; } }; } generate_configure!( filterable_attributes, displayed_attributes, searchable_attributes, distinct_attribute, stop_words, synonyms, ranking_rules ); pub async fn update_all( data: GuardedData, index_uid: web::Path, body: web::Json>, ) -> Result { let settings = body.into_inner().check(); let update_result = data .update_settings(index_uid.into_inner(), settings, true) .await?; let json = serde_json::json!({ "updateId": update_result.id() }); debug!("returns: {:?}", json); Ok(HttpResponse::Accepted().json(json)) } pub async fn get_all( data: GuardedData, index_uid: web::Path, ) -> Result { let settings = data.settings(index_uid.into_inner()).await?; debug!("returns: {:?}", settings); Ok(HttpResponse::Ok().json(settings)) } pub async fn delete_all( data: GuardedData, index_uid: web::Path, ) -> Result { let settings = Settings::cleared(); let update_result = data .update_settings(index_uid.into_inner(), settings, false) .await?; let json = serde_json::json!({ "updateId": update_result.id() }); debug!("returns: {:?}", json); Ok(HttpResponse::Accepted().json(json)) }