use log::debug; use actix_web::{HttpResponse, web}; use meilisearch_lib::MeiliSearch; use meilisearch_lib::index::{Settings, Unchecked}; use meilisearch_lib::index_controller::Update; use crate::extractors::authentication::{policies::*, GuardedData}; use crate::error::ResponseError; #[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 meilisearch_lib::{MeiliSearch, index::Settings, index_controller::Update}; use crate::error::ResponseError; use crate::extractors::authentication::{GuardedData, policies::*}; pub async fn delete( meilisearch: GuardedData, index_uid: web::Path, ) -> Result { let settings = Settings { $attr: Setting::Reset, ..Default::default() }; let update = Update::Settings(settings); let update_status = meilisearch.register_update(index_uid.into_inner(), update).await?; debug!("returns: {:?}", update_status); Ok(HttpResponse::Accepted().json(serde_json::json!({ "updateId": update_status.id() }))) } pub async fn update( meilisearch: 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 = Update::Settings(settings); let update_status = meilisearch.register_update(index_uid.into_inner(), update).await?; debug!("returns: {:?}", update_status); Ok(HttpResponse::Accepted().json(serde_json::json!({ "updateId": update_status.id() }))) } pub async fn get( meilisearch: GuardedData, index_uid: actix_web::web::Path, ) -> std::result::Result { let settings = meilisearch.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::BTreeSet, filterable_attributes, "filterableAttributes" ); make_setting_route!( "/sortable-attributes", std::collections::BTreeSet, sortable_attributes, "sortableAttributes" ); 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, sortable_attributes, displayed_attributes, searchable_attributes, distinct_attribute, stop_words, synonyms, ranking_rules ); pub async fn update_all( meilisearch: GuardedData, index_uid: web::Path, body: web::Json>, ) -> Result { let settings = body.into_inner(); let update = Update::Settings(settings); let update_result = meilisearch .register_update(index_uid.into_inner(), update) .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 = Update::Settings(settings.into_unchecked()); let update_result = data .register_update(index_uid.into_inner(), update) .await?; let json = serde_json::json!({ "updateId": update_result.id() }); debug!("returns: {:?}", json); Ok(HttpResponse::Accepted().json(json)) }