use std::fmt; use std::path::Path; use base64::Engine; use semver::Version; use tracing::{error, info}; use crate::log_filter::create_log_filter; use flowy_server_pub::af_cloud_config::AFCloudConfiguration; use flowy_user::services::entities::URL_SAFE_ENGINE; use lib_infra::file_util::copy_dir_recursive; use lib_infra::util::OperatingSystem; #[derive(Clone)] pub struct AppFlowyCoreConfig { /// Different `AppFlowyCoreConfig` instance should have different name pub(crate) app_version: Version, pub name: String, pub(crate) device_id: String, pub platform: String, /// Used to store the user data pub storage_path: String, /// Origin application path is the path of the application binary. By default, the /// storage_path is the same as the origin_application_path. However, when the user /// choose a custom path for the user data, the storage_path will be different from /// the origin_application_path. pub application_path: String, pub(crate) log_filter: String, pub cloud_config: Option, } impl fmt::Debug for AppFlowyCoreConfig { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { let mut debug = f.debug_struct("AppFlowy Configuration"); debug.field("app_version", &self.app_version); debug.field("storage_path", &self.storage_path); debug.field("application_path", &self.application_path); if let Some(config) = &self.cloud_config { debug.field("base_url", &config.base_url); debug.field("ws_url", &config.ws_base_url); debug.field("gotrue_url", &config.gotrue_url); debug.field("enable_sync_trace", &config.enable_sync_trace); } debug.finish() } } fn make_user_data_folder(root: &str, url: &str) -> String { // Isolate the user data folder by using the base url of AppFlowy cloud. This is to avoid // the user data folder being shared by different AppFlowy cloud. let storage_path = if !url.is_empty() { let server_base64 = URL_SAFE_ENGINE.encode(url); format!("{}_{}", root, server_base64) } else { root.to_string() }; // Copy the user data folder from the root path to the isolated path // The root path without any suffix is the created by the local version AppFlowy if !Path::new(&storage_path).exists() && Path::new(root).exists() { info!("Copy dir from {} to {}", root, storage_path); let src = Path::new(root); match copy_dir_recursive(src, Path::new(&storage_path)) { Ok(_) => storage_path, Err(err) => { // when the copy dir failed, use the root path as the storage path error!("Copy dir failed: {}", err); root.to_string() }, } } else { storage_path } } impl AppFlowyCoreConfig { pub fn new( app_version: Version, custom_application_path: String, application_path: String, device_id: String, platform: String, name: String, ) -> Self { let cloud_config = AFCloudConfiguration::from_env().ok(); let mut log_crates = vec![]; let storage_path = match &cloud_config { None => custom_application_path, Some(config) => { if config.enable_sync_trace { log_crates.push("sync_trace_log".to_string()); } make_user_data_folder(&custom_application_path, &config.base_url) }, }; let log_filter = create_log_filter( "info".to_owned(), log_crates, OperatingSystem::from(&platform), ); AppFlowyCoreConfig { app_version, name, storage_path, application_path, device_id, platform, log_filter, cloud_config, } } pub fn log_filter(mut self, level: &str, with_crates: Vec) -> Self { self.log_filter = create_log_filter( level.to_owned(), with_crates, OperatingSystem::from(&self.platform), ); self } }