#![allow(dead_code)] use clap::Parser; use std::path::Path; use actix_web::http::StatusCode; use byte_unit::{Byte, ByteUnit}; use meilisearch_auth::AuthController; use meilisearch_http::setup_meilisearch; use meilisearch_lib::options::{IndexerOpts, MaxMemory}; use once_cell::sync::Lazy; use serde_json::Value; use tempfile::TempDir; use meilisearch_http::option::Opt; use super::index::Index; use super::service::Service; pub struct Server { pub service: Service, // hold ownership to the tempdir while we use the server instance. _dir: Option, } pub static TEST_TEMP_DIR: Lazy = Lazy::new(|| TempDir::new().unwrap()); impl Server { pub async fn new() -> Self { let dir = TempDir::new().unwrap(); if cfg!(windows) { std::env::set_var("TMP", TEST_TEMP_DIR.path()); } else { std::env::set_var("TMPDIR", TEST_TEMP_DIR.path()); } let options = default_settings(dir.path()); let meilisearch = setup_meilisearch(&options).unwrap(); let auth = AuthController::new(&options.db_path, &options.master_key).unwrap(); let service = Service { meilisearch, auth, options, api_key: None, }; Server { service, _dir: Some(dir), } } pub async fn new_auth() -> Self { let dir = TempDir::new().unwrap(); if cfg!(windows) { std::env::set_var("TMP", TEST_TEMP_DIR.path()); } else { std::env::set_var("TMPDIR", TEST_TEMP_DIR.path()); } let mut options = default_settings(dir.path()); options.master_key = Some("MASTER_KEY".to_string()); let meilisearch = setup_meilisearch(&options).unwrap(); let auth = AuthController::new(&options.db_path, &options.master_key).unwrap(); let service = Service { meilisearch, auth, options, api_key: None, }; Server { service, _dir: Some(dir), } } pub async fn new_with_options(options: Opt) -> Self { let meilisearch = setup_meilisearch(&options).unwrap(); let auth = AuthController::new(&options.db_path, &options.master_key).unwrap(); let service = Service { meilisearch, auth, options, api_key: None, }; Server { service, _dir: None, } } /// Returns a view to an index. There is no guarantee that the index exists. pub fn index(&self, uid: impl AsRef) -> Index<'_> { Index { uid: uid.as_ref().to_string(), service: &self.service, } } pub async fn list_indexes(&self) -> (Value, StatusCode) { self.service.get("/indexes").await } pub async fn version(&self) -> (Value, StatusCode) { self.service.get("/version").await } pub async fn stats(&self) -> (Value, StatusCode) { self.service.get("/stats").await } pub async fn tasks(&self) -> (Value, StatusCode) { self.service.get("/tasks").await } pub async fn get_dump_status(&self, uid: &str) -> (Value, StatusCode) { self.service.get(format!("/dumps/{}/status", uid)).await } } pub fn default_settings(dir: impl AsRef) -> Opt { Opt { db_path: dir.as_ref().join("db"), dumps_dir: dir.as_ref().join("dump"), env: "development".to_owned(), #[cfg(all(not(debug_assertions), feature = "analytics"))] no_analytics: true, max_index_size: Byte::from_unit(4.0, ByteUnit::GiB).unwrap(), max_task_db_size: Byte::from_unit(4.0, ByteUnit::GiB).unwrap(), http_payload_size_limit: Byte::from_unit(10.0, ByteUnit::MiB).unwrap(), snapshot_dir: ".".into(), ..Parser::parse_from(None as Option<&str>) } }